refactor: used recommended_models function as default model (#209)

packages/exchange/src/exchange/providers/anthropic.py

@@ -157,6 +157,11 @@ class AnthropicProvider(Provider):
 
         return message, usage
 
+    @staticmethod
+    def recommended_models() -> tuple[str, str]:
+        """Return the recommended model and processor for this provider"""
+        return "claude-3-5-sonnet-20241022", "claude-3-5-sonnet-20241022"
+
     @retry_procedure
     def _post(self, payload: dict) -> httpx.Response:
         response = self.client.post(ANTHROPIC_HOST, json=payload)

packages/exchange/src/exchange/providers/base.py

@@ -50,6 +50,11 @@ class Provider(ABC):
         """Generate the next message using the specified model"""
         pass
 
+    @staticmethod
+    def recommended_models() -> tuple[str, str]:
+        """Return the recommended model and processor for this provider"""
+        return "gpt-4o", "gpt-4o-mini"
+
 
 class MissingProviderEnvVariableError(Exception):
     def __init__(self, env_variable: str, provider: str, instructions_url: Optional[str] = None) -> None:

packages/exchange/src/exchange/providers/databricks.py

@@ -90,6 +90,11 @@ class DatabricksProvider(Provider):
         usage = self.get_usage(response)
         return message, usage
 
+    @staticmethod
+    def recommended_models() -> tuple[str, str]:
+        """Return the recommended model and processor for this provider"""
+        return "databricks-meta-llama-3-1-70b-instruct", "databricks-meta-llama-3-1-70b-instruct"
+
     @retry_procedure
     def _post(self, model: str, payload: dict) -> httpx.Response:
         response = self.client.post(

packages/exchange/src/exchange/providers/google.py

@@ -165,3 +165,8 @@ class GoogleProvider(Provider):
     def _post(self, payload: dict, model: str) -> httpx.Response:
         response = self.client.post("models/" + model + ":generateContent", json=payload)
         return raise_for_status(response).json()
+
+    @staticmethod
+    def recommended_models() -> tuple[str, str]:
+        """Return the recommended model and processor for this provider"""
+        return "gemini-1.5-flash", "gemini-1.5-flash"

packages/exchange/src/exchange/providers/ollama.py

@@ -43,3 +43,8 @@ ollama:
         # When served by Ollama, the OpenAI API is available at the path "v1/".
         client = httpx.Client(base_url=ollama_url + "v1/", timeout=timeout)
         return cls(client)
+
+    @staticmethod
+    def recommended_models() -> tuple[str, str]:
+        """Return the recommended model and processor for this provider"""
+        return OLLAMA_MODEL, OLLAMA_MODEL