stephane.david/agenticSeek
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #460 from octo-patch/fix/provider-registration-bugs

Browse Source 
fix: register anthropic provider, fix togetherAI alias, fix dsk_deepseek exception handling
This commit is contained in:
Martin
2026-04-05 16:39:32 +02:00
committed by GitHub
parent 7b8a2271a3 1fbd8b135b
commit 31205febfe
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sources/llm_provider.py
+8 -5
View File
@@ -17,6 +17,9 @@ from sources.utility import pretty_print, animate_thinking
class Provider: class Provider:
def __init__(self, provider_name, model, server_address="127.0.0.1:5000", is_local=False): def __init__(self, provider_name, model, server_address="127.0.0.1:5000", is_local=False):
self.provider_name = provider_name.lower() self.provider_name = provider_name.lower()
# Normalize provider name aliases (e.g. README documents 'togetherAI' but canonical key is 'together')
_aliases = {"togetherai": "together"}
self.provider_name = _aliases.get(self.provider_name, self.provider_name)
self.model = model self.model = model
self.is_local = is_local self.is_local = is_local
self.server_ip = server_address self.server_ip = server_address
@@ -32,13 +35,14 @@ class Provider:
"together": self.together_fn, "together": self.together_fn,
"dsk_deepseek": self.dsk_deepseek, "dsk_deepseek": self.dsk_deepseek,
"openrouter": self.openrouter_fn, "openrouter": self.openrouter_fn,
"anthropic": self.anthropic_fn,
"minimax": self.minimax_fn, "minimax": self.minimax_fn,
"test": self.test_fn "test": self.test_fn
} }
self.logger = Logger("provider.log") self.logger = Logger("provider.log")
self.api_key = None self.api_key = None
self.internal_url, self.in_docker = self.get_internal_url() self.internal_url, self.in_docker = self.get_internal_url()
self.unsafe_providers = ["openai", "deepseek", "dsk_deepseek", "together", "google", "openrouter", "minimax"] self.unsafe_providers = ["openai", "deepseek", "dsk_deepseek", "together", "google", "openrouter", "anthropic", "minimax"]
if self.provider_name not in self.available_providers: if self.provider_name not in self.available_providers:
raise ValueError(f"Unknown provider: {provider_name}") raise ValueError(f"Unknown provider: {provider_name}")
if self.provider_name in self.unsafe_providers and self.is_local == False: if self.provider_name in self.unsafe_providers and self.is_local == False:
@@ -388,7 +392,6 @@ class Provider:
if "LM Studio" in str(e): if "LM Studio" in str(e):
raise # Re-raise our custom exceptions raise # Re-raise our custom exceptions
raise Exception(f"Unexpected error: {str(e)}") from e raise Exception(f"Unexpected error: {str(e)}") from e
return thought
def openrouter_fn(self, history, verbose=False): def openrouter_fn(self, history, verbose=False):
""" """
@@ -470,13 +473,13 @@ class Provider:
if chunk['type'] == 'text': if chunk['type'] == 'text':
thought += chunk['content'] thought += chunk['content']
return thought return thought
except AuthenticationError: except AuthenticationError as e:
raise AuthenticationError("Authentication failed. Please check your token.") from e raise AuthenticationError("Authentication failed. Please check your token.") from e
except RateLimitError: except RateLimitError as e:
raise RateLimitError("Rate limit exceeded. Please wait before making more requests.") from e raise RateLimitError("Rate limit exceeded. Please wait before making more requests.") from e
except CloudflareError as e: except CloudflareError as e:
raise CloudflareError(f"Cloudflare protection encountered: {str(e)}") from e raise CloudflareError(f"Cloudflare protection encountered: {str(e)}") from e
except NetworkError: except NetworkError as e:
raise NetworkError("Network error occurred. Check your internet connection.") from e raise NetworkError("Network error occurred. Check your internet connection.") from e
except APIError as e: except APIError as e:
raise APIError(f"API error occurred: {str(e)}") from e raise APIError(f"API error occurred: {str(e)}") from e