Improve API endpoint error handling

Author: sylwia-budzynska

README.md

@@ -47,7 +47,8 @@ Example:
 AI_API_TOKEN=<your_github_token>
 # MCP configs
 GITHUB_PERSONAL_ACCESS_TOKEN=<your_github_token>
-CODEQL_DBS_BASE_PATH="/app/my_data/codeql_databases"
+CODEQL_DBS_BASE_PATH="/app/data/codeql_databases"
+AI_API_ENDPOINT="https://models.github.ai/inference"
 ```
 
 ## Deploying from Source

src/seclab_taskflow_agent/agent.py

@@ -27,7 +27,8 @@
     case AI_API_ENDPOINT_ENUM.AI_API_MODELS_GITHUB:
         default_model = 'openai/gpt-4o'
     case _:
-        raise ValueError(f"Unsupported Model Endpoint: {api_endpoint}")
+        raise ValueError(f"Unsupported Model Endpoint: {api_endpoint}\n"
+                            f"Supported endpoints: {[e.to_url() for e in AI_API_ENDPOINT_ENUM]}")
 
 DEFAULT_MODEL = os.getenv('COPILOT_DEFAULT_MODEL', default=default_model)
 

src/seclab_taskflow_agent/capi.py

@@ -11,8 +11,20 @@
 
 # Enumeration of currently supported API endpoints.
 class AI_API_ENDPOINT_ENUM(StrEnum):
-  AI_API_MODELS_GITHUB = 'models.github.ai'
-  AI_API_GITHUBCOPILOT = 'api.githubcopilot.com'
+    AI_API_MODELS_GITHUB = 'models.github.ai'
+    AI_API_GITHUBCOPILOT = 'api.githubcopilot.com'
+
+    def to_url(self):
+        """
+        Convert the endpoint to its full URL.
+        """
+        match self:
+            case self.AI_API_GITHUBCOPILOT:
+                return f"https://{self}"
+            case self.AI_API_MODELS_GITHUB:
+                return f"https://{self}/inference"
+            case _:
+                raise ValueError(f"Unsupported Model Endpoint: {self}")
 
 COPILOT_INTEGRATION_ID = 'vscode-chat'
 
@@ -21,7 +33,7 @@ class AI_API_ENDPOINT_ENUM(StrEnum):
 # since different APIs use their own id schema, use -l with your desired
 # endpoint to retrieve the correct id names to use for your taskflow
 def get_AI_endpoint():
-  return os.getenv('AI_API_ENDPOINT', default='https://models.github.ai/inference')
+    return os.getenv('AI_API_ENDPOINT', default='https://models.github.ai/inference')
 
 def get_AI_token():
     """
@@ -65,7 +77,8 @@ def list_capi_models(token: str) -> dict[str, dict]:
             case AI_API_ENDPOINT_ENUM.AI_API_MODELS_GITHUB:
                 models_list = r.json()
             case _:
-                raise ValueError(f"Unsupported Model Endpoint: {api_endpoint}")
+                raise ValueError(f"Unsupported Model Endpoint: {api_endpoint}\n"
+                                 f"Supported endpoints: {[e.to_url() for e in AI_API_ENDPOINT_ENUM]}")
         for model in models_list:
             models[model.get('id')] = dict(model)
     except httpx.RequestError as e:
@@ -88,7 +101,8 @@ def supports_tool_calls(model: str, models: dict) -> bool:
             return 'tool-calling' in models.get(model, {}).\
                 get('capabilities', [])
         case _:
-            raise ValueError(f"Unsupported Model Endpoint: {api_endpoint}")
+            raise ValueError(f"Unsupported Model Endpoint: {api_endpoint}\n"
+                             f"Supported endpoints: {[e.to_url() for e in AI_API_ENDPOINT_ENUM]}")
 
 def list_tool_call_models(token: str) -> dict[str, dict]:
     models = list_capi_models(token)