feedback 2

ai/ai_utils/get_available_apis.py

@@ -2,7 +2,10 @@
 from ai.providers.openai import OpenAI_API
 
 """
-New providers need to be added below
+This file defines a function to retrieve available API models from different AI providers.
+It imports specific AI provider classes and combines their available models into a single dictionary.
+
+New AI providers must be added below
 """
 
 

ai/ai_utils/handle_response.py

@@ -5,7 +5,11 @@
 from state_store.get_user_state import get_user_state
 
 """
-New providers need to be added below
+This file defines functions to handle AI responses by interacting with different AI providers.
+The `_get_provider()` function returns an instance of the appropriate API provider based on the given API name. 
+The `get_ai_response` function retrieves the user's selected API provider and model, sets the model, and generates a response.
+
+New AI providers must be added below
 """
 
 

ai/providers/anthropic.py

@@ -28,7 +28,7 @@ def get_models(self):
         if self.api_key is not None:
             return self.MODELS
         else:
-            return None
+            return {}
 
     def generate_response(self, prompt: str, system_content: str) -> str:
         try:

ai/providers/base_provider.py

@@ -1,13 +1,14 @@
+# A base class for API providers, defining the interface and common properties for subclasses.
+
+
 class BaseProvider:
     MODELS = {}
-    BASELINE_MAX_TOKENS: int
-    API_NAME: str
 
     def __init__(self):
-        self.current_model = None
         self.api_key: str
 
     def set_model(self, model_name: str):
+        self.current_model = model_name
         raise NotImplementedError("Subclass must implement set_model")
 
     def get_models(self):

ai/providers/openai.py

@@ -25,7 +25,7 @@ def get_models(self):
         if self.api_key is not None:
             return self.MODELS
         else:
-            return None
+            return {}
 
     def generate_response(self, prompt: str, system_content: str) -> str:
         try:

manifest.json

@@ -105,4 +105,4 @@
       "token_rotation_enabled": false,
       "function_runtime": "remote"
   }
-}
+}

requirements.txt

@@ -4,4 +4,4 @@ flake8==7.1.0
 black==24.4.2
 slack-cli-hooks==0.0.2
 openai==1.37.1
-anthropic==0.32.0
+anthropic==0.32.0