Add 'enabled' field validation to endpoint configuration and filter disabled endpoints

Author: blzjzhu

code_assistant_manager/config.py

@@ -2,7 +2,6 @@
 
 import json
 import logging
-import os
 import time
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Tuple
@@ -475,7 +474,7 @@ def validate_config(self) -> Tuple[bool, List[str]]:
         # Return cached result if still valid
         if self._is_validation_cache_valid(current_time):
             logger.debug("Using cached validation result")
-            return self._validation_cache
+            return self._validation_cache  # type: ignore
 
         logger.debug("Performing fresh config validation")
         errors = []
@@ -592,7 +591,7 @@ def _validate_endpoint(endpoint_name: str, endpoint_config: dict) -> List[str]:
                     errors.append(f"Invalid model ID in list_of_models for {endpoint_name}: {model}")
 
     # Validate boolean fields
-    boolean_fields = ["keep_proxy_config", "use_proxy"]
+    boolean_fields = ["keep_proxy_config", "use_proxy", "enabled"]
     for field_name in boolean_fields:
         value = endpoint_config.get(field_name, "")
         if value and not validate_boolean(value):

code_assistant_manager/endpoints.py

@@ -165,6 +165,19 @@ def select_endpoint(
             print("Error: No endpoints configured in settings.conf")
             return False, None
 
+        # Filter out disabled endpoints
+        enabled_endpoints = []
+        for ep in endpoints:
+            ep_config = self.config.get_endpoint_config(ep)
+            enabled = ep_config.get("enabled", "true").lower() in ("true", "1", "yes")
+            if enabled:
+                enabled_endpoints.append(ep)
+        endpoints = enabled_endpoints
+
+        if not endpoints:
+            print("Error: No enabled endpoints configured in settings.conf")
+            return False, None
+
         # Filter endpoints by client if specified
         if client_name:
             filtered = []

code_assistant_manager/providers.json

@@ -0,0 +1,39 @@
+{
+  "endpoints": {
+    "example-openai": {
+      "endpoint": "https://api.openai.com/v1",
+      "api_key_env": "OPENAI_API_KEY",
+      "supported_client": "openai",
+      "list_models_cmd": "curl -s -H \"Authorization: Bearer $API_KEY\" $endpoint/models | jq -r '.data[].id'",
+      "list_of_models": ["gpt-4", "gpt-3.5-turbo"],
+      "keep_proxy_config": false,
+      "use_proxy": true,
+      "enabled": true
+    },
+    "example-anthropic": {
+      "endpoint": "https://api.anthropic.com",
+      "api_key_env": "ANTHROPIC_API_KEY",
+      "supported_client": "claude",
+      "list_models_cmd": "curl -s -H \"x-api-key: $API_KEY\" -H \"anthropic-version: 2023-06-01\" $endpoint/v1/messages --data '{\"model\": \"claude-3-haiku-20240307\", \"max_tokens\": 1}' | jq -r '.model // \"claude-3-haiku-20240307\"'",
+      "list_of_models": ["claude-3-opus-20240229", "claude-3-sonnet-20240229", "claude-3-haiku-20240307"],
+      "keep_proxy_config": false,
+      "use_proxy": false,
+      "enabled": true
+    },
+    "disabled-example": {
+      "endpoint": "https://disabled.example.com",
+      "api_key_env": "DISABLED_API_KEY",
+      "supported_client": "openai",
+      "list_models_cmd": "echo 'gpt-4'",
+      "list_of_models": ["gpt-4"],
+      "keep_proxy_config": false,
+      "use_proxy": false,
+      "enabled": false
+    }
+  },
+  "common": {
+    "http_proxy": "",
+    "https_proxy": "",
+    "cache_ttl_seconds": "3600"
+  }
+}

code_assistant_manager/tools/codex.py

@@ -112,9 +112,14 @@ def run(self, args: List[str] = None) -> int:
             return 1
 
         endpoints = self.config.get_sections(exclude_common=True)
-        endpoints = [
-            ep for ep in endpoints if self.endpoint_manager._is_client_supported(ep, "codex")
-        ]
+        filtered_endpoints = []
+        for ep in endpoints:
+            # Check if endpoint is enabled
+            ep_config = self.config.get_endpoint_config(ep)
+            enabled = ep_config.get("enabled", "true").lower() in ("true", "1", "yes")
+            if enabled and self.endpoint_manager._is_client_supported(ep, "codex"):
+                filtered_endpoints.append(ep)
+        endpoints = filtered_endpoints
         if not endpoints:
             return self._handle_error("No endpoints configured for codex")
 

code_assistant_manager/tools/continue_tool.py

@@ -17,11 +17,14 @@ class ContinueTool(CLITool):
     def _get_filtered_endpoints(self) -> List[str]:
         """Collect endpoints that support the continue client."""
         endpoints = self.config.get_sections(exclude_common=True)
-        return [
-            ep
-            for ep in endpoints
-            if self.endpoint_manager._is_client_supported(ep, "continue")
-        ]
+        filtered = []
+        for ep in endpoints:
+            # Check if endpoint is enabled
+            ep_config = self.config.get_endpoint_config(ep)
+            enabled = ep_config.get("enabled", "true").lower() in ("true", "1", "yes")
+            if enabled and self.endpoint_manager._is_client_supported(ep, "continue"):
+                filtered.append(ep)
+        return filtered
 
     def _process_endpoint(self, endpoint_name: str) -> Optional[List[str]]:
         """Process a single endpoint and return selected models if successful."""

code_assistant_manager/tools/endpoint_display.py

@@ -23,7 +23,10 @@ def display_tool_endpoints(config_manager: ConfigManager, client_name: str = Non
     if client_name:
         filtered = []
         for ep in endpoints:
-            if endpoint_manager._is_client_supported(ep, client_name):
+            # Check if endpoint is enabled
+            ep_config = config_manager.get_endpoint_config(ep)
+            enabled = ep_config.get("enabled", "true").lower() in ("true", "1", "yes")
+            if enabled and endpoint_manager._is_client_supported(ep, client_name):
                 filtered.append(ep)
         endpoints = filtered
 
@@ -77,7 +80,10 @@ def display_all_tool_endpoints(config_manager: ConfigManager):
         # Filter endpoints by client
         if tool != "copilot":  # Copilot doesn't use endpoint selection
             for ep in endpoints:
-                if endpoint_manager._is_client_supported(ep, tool):
+                # Check if endpoint is enabled
+                ep_config = config_manager.get_endpoint_config(ep)
+                enabled = ep_config.get("enabled", "true").lower() in ("true", "1", "yes")
+                if enabled and endpoint_manager._is_client_supported(ep, tool):
                     tool_endpoints.append(ep)
 
         if tool_endpoints:

code_assistant_manager/tools/goose.py

@@ -117,11 +117,14 @@ def _determine_engine_type(self, endpoint_config: Dict[str, str], endpoint_name:
     def _get_filtered_endpoints(self) -> List[str]:
         """Collect endpoints that support the goose client."""
         endpoints = self.config.get_sections(exclude_common=True)
-        return [
-            ep
-            for ep in endpoints
-            if self.endpoint_manager._is_client_supported(ep, "goose")
-        ]
+        filtered = []
+        for ep in endpoints:
+            # Check if endpoint is enabled
+            ep_config = self.config.get_endpoint_config(ep)
+            enabled = ep_config.get("enabled", "true").lower() in ("true", "1", "yes")
+            if enabled and self.endpoint_manager._is_client_supported(ep, "goose"):
+                filtered.append(ep)
+        return filtered
 
     def _validate_and_fix_config(self) -> None:
         """Validate that config.yaml references valid custom providers.