symbolicai 1.0.0__py3-none-any.whl → 1.1.1__py3-none-any.whl

symai/backend/mixin/anthropic.py

@@ -1,58 +1,66 @@
 # https://docs.anthropic.com/en/docs/about-claude/models
 SUPPORTED_CHAT_MODELS = [
-    'claude-3-5-sonnet-latest',
-    'claude-3-5-haiku-latest',
-    'claude-3-5-sonnet-20241022',
-    'claude-3-5-sonnet-20240620',
-    'claude-3-opus-latest',
-    'claude-3-opus-20240229',
-    'claude-3-sonnet-20240229',
-    'claude-3-haiku-20240307',
+    "claude-3-5-sonnet-latest",
+    "claude-3-5-haiku-latest",
+    "claude-3-5-sonnet-20241022",
+    "claude-3-5-sonnet-20240620",
+    "claude-3-opus-latest",
+    "claude-3-opus-20240229",
+    "claude-3-sonnet-20240229",
+    "claude-3-haiku-20240307",
 ]
 SUPPORTED_REASONING_MODELS = [
     "claude-opus-4-1",
     "claude-opus-4-0",
     "claude-sonnet-4-0",
-    'claude-3-7-sonnet-latest',
-    'claude-haiku-4-5',
-    'claude-sonnet-4-5',
+    "claude-3-7-sonnet-latest",
+    "claude-haiku-4-5",
+    "claude-sonnet-4-5",
 ]
 
+
 class AnthropicMixin:
     def api_max_context_tokens(self):
-        if self.model == 'claude-opus-4-1' or \
-           self.model == 'claude-opus-4-0' or \
-           self.model == 'claude-sonnet-4-0' or \
-           self.model == 'claude-3-7-sonnet-latest' or \
-           self.model == 'claude-haiku-4-5' or \
-           self.model == 'claude-sonnet-4-5' or \
-           self.model == 'claude-3-5-sonnet-latest' or \
-           self.model == 'claude-3-5-sonnet-20241022' or \
-           self.model == 'claude-3-5-sonnet-20240620' or \
-           self.model == 'claude-3-opus-latest' or \
-           self.model == 'claude-3-opus-20240229' or \
-           self.model == 'claude-3-sonnet-20240229' or \
-           self.model == 'claude-3-haiku-20240307':
-           return 200_000
+        if (
+            self.model == "claude-opus-4-1"
+            or self.model == "claude-opus-4-0"
+            or self.model == "claude-sonnet-4-0"
+            or self.model == "claude-3-7-sonnet-latest"
+            or self.model == "claude-haiku-4-5"
+            or self.model == "claude-sonnet-4-5"
+            or self.model == "claude-3-5-sonnet-latest"
+            or self.model == "claude-3-5-sonnet-20241022"
+            or self.model == "claude-3-5-sonnet-20240620"
+            or self.model == "claude-3-opus-latest"
+            or self.model == "claude-3-opus-20240229"
+            or self.model == "claude-3-sonnet-20240229"
+            or self.model == "claude-3-haiku-20240307"
+        ):
+            return 200_000
         return None
 
     def api_max_response_tokens(self):
-        if self.model == 'claude-sonnet-4-0' or \
-           self.model == 'claude-3-7-sonnet-latest' or \
-           self.model == 'claude-haiku-4-5' or \
-           self.model == 'claude-sonnet-4-5':
+        if (
+            self.model == "claude-sonnet-4-0"
+            or self.model == "claude-3-7-sonnet-latest"
+            or self.model == "claude-haiku-4-5"
+            or self.model == "claude-sonnet-4-5"
+        ):
             return 64_000
-        if self.model == 'claude-opus-4-1' or \
-           self.model == 'claude-opus-4-0':
+        if self.model == "claude-opus-4-1" or self.model == "claude-opus-4-0":
             return 32_000
-        if self.model == 'claude-3-5-sonnet-latest' or \
-           self.model == 'claude-3-5-sonnet-20241022' or \
-           self.model == 'claude-3-5-haiku-latest':
+        if (
+            self.model == "claude-3-5-sonnet-latest"
+            or self.model == "claude-3-5-sonnet-20241022"
+            or self.model == "claude-3-5-haiku-latest"
+        ):
             return 8_192
-        if self.model == 'claude-3-5-sonnet-20240620' or \
-           self.model == 'claude-3-opus-latest' or \
-           self.model == 'clade-3-opus-20240229' or \
-           self.model == 'claude-3-sonnet-20240229' or \
-           self.model == 'claude-3-haiku-20240307':
-           return 4_096
+        if (
+            self.model == "claude-3-5-sonnet-20240620"
+            or self.model == "claude-3-opus-latest"
+            or self.model == "clade-3-opus-20240229"
+            or self.model == "claude-3-sonnet-20240229"
+            or self.model == "claude-3-haiku-20240307"
+        ):
+            return 4_096
         return None

symai/backend/mixin/cerebras.py

@@ -0,0 +1,9 @@
+SUPPORTED_CHAT_MODELS: list[str] = [
+    "cerebras:qwen-3-235b-a22b-instruct-2507",
+]
+
+SUPPORTED_REASONING_MODELS: list[str] = [
+    "cerebras:zai-glm-4.6",
+    "cerebras:gpt-oss-120b",
+    "cerebras:qwen-3-32b",
+]

symai/backend/mixin/deepseek.py

@@ -1,16 +1,15 @@
 # https://api-docs.deepseek.com/quick_start/pricing
 SUPPORTED_CHAT_MODELS = []
-SUPPORTED_REASONING_MODELS = [
-    'deepseek-reasoner'
-]
+SUPPORTED_REASONING_MODELS = ["deepseek-reasoner"]
+
 
 class DeepSeekMixin:
     def api_max_context_tokens(self):
-        if self.model == 'deepseek-reasoner':
+        if self.model == "deepseek-reasoner":
             return 64_000
         return None
 
     def api_max_response_tokens(self):
-        if self.model == 'deepseek-reasoner':
+        if self.model == "deepseek-reasoner":
             return 8_000
         return None

symai/backend/mixin/google.py

@@ -2,17 +2,18 @@
 SUPPORTED_CHAT_MODELS = []
 SUPPORTED_REASONING_MODELS = [
     # Check the latest snapshots; ie. *-06-05, etc
-    'gemini-2.5-pro',
-    'gemini-2.5-flash',
+    "gemini-2.5-pro",
+    "gemini-2.5-flash",
 ]
 
+
 class GoogleMixin:
     def api_max_context_tokens(self):
-        if self.model.startswith('gemini-2.5-'):
+        if self.model.startswith("gemini-2.5-"):
             return 1_048_576
         return None
 
     def api_max_response_tokens(self):
-        if self.model == 'gemini-2.5-':
+        if self.model == "gemini-2.5-":
             return 65_536
         return None

symai/backend/mixin/groq.py

@@ -1,10 +1,8 @@
-SUPPORTED_CHAT_MODELS = [
-    "groq:moonshotai/kimi-k2-instruct"
-]
+SUPPORTED_CHAT_MODELS = ["groq:moonshotai/kimi-k2-instruct"]
 
 SUPPORTED_REASONING_MODELS = [
     "groq:openai/gpt-oss-120b",
     "groq:openai/gpt-oss-20b",
     "groq:qwen/qwen3-32b",
-    "groq:deepseek-r1-distill-llama-70b"
+    "groq:deepseek-r1-distill-llama-70b",
 ]

symai/backend/mixin/openai.py

@@ -1,142 +1,164 @@
 from ...utils import UserMessage
 
 SUPPORTED_COMPLETION_MODELS = [
-    'davinci-002',
+    "davinci-002",
 ]
 SUPPORTED_CHAT_MODELS = [
-    'gpt-3.5-turbo',
-    'gpt-3.5-turbo-16k',
-    'gpt-3.5-turbo-1106',
-    'gpt-3.5-turbo-0613',
-    'gpt-4',
-    'gpt-4-0613',
-    'gpt-4-1106-preview', # @NOTE: probabily obsolete; same price as 'gpt-4-turbo-2024-04-09' but no vision
-    'gpt-4-turbo',
-    'gpt-4-turbo-2024-04-09',
-    'gpt-4o',
-    'gpt-4o-2024-11-20',
-    'gpt-4o-mini',
-    'chatgpt-4o-latest',
-    'gpt-4.1',
-    'gpt-4.1-mini',
-    'gpt-4.1-nano',
-    'gpt-5-chat-latest',
+    "gpt-3.5-turbo",
+    "gpt-3.5-turbo-16k",
+    "gpt-3.5-turbo-1106",
+    "gpt-3.5-turbo-0613",
+    "gpt-4",
+    "gpt-4-0613",
+    "gpt-4-1106-preview",  # @NOTE: probabily obsolete; same price as 'gpt-4-turbo-2024-04-09' but no vision
+    "gpt-4-turbo",
+    "gpt-4-turbo-2024-04-09",
+    "gpt-4o",
+    "gpt-4o-2024-11-20",
+    "gpt-4o-mini",
+    "chatgpt-4o-latest",
+    "gpt-4.1",
+    "gpt-4.1-mini",
+    "gpt-4.1-nano",
+    "gpt-5-chat-latest",
+    "gpt-5.1-chat-latest",
 ]
 SUPPORTED_REASONING_MODELS = [
-    'o3-mini',
-    'o4-mini',
-    'o1',
-    'o3',
-    'gpt-5',
-    'gpt-5-mini',
-    'gpt-5-nano',
+    "o3-mini",
+    "o4-mini",
+    "o1",
+    "o3",
+    "gpt-5",
+    "gpt-5.1",
+    "gpt-5-mini",
+    "gpt-5-nano",
 ]
 SUPPORTED_EMBEDDING_MODELS = [
-    'text-embedding-ada-002',
-    'text-embedding-3-small',
-    'text-embedding-3-large'
+    "text-embedding-ada-002",
+    "text-embedding-3-small",
+    "text-embedding-3-large",
 ]
 
 
 class OpenAIMixin:
     def api_max_context_tokens(self):
-        if self.model == 'text-curie-001' or \
-           self.model == 'text-babbage-001' or \
-           self.model == 'text-ada-001' or \
-           self.model == 'davinci' or \
-           self.model == 'curie' or \
-           self.model == 'babbage' or \
-           self.model == 'ada':
-               return 2_049
-        if self.model == 'gpt-3.5-turbo' or \
-           self.model == 'gpt-3.5-turbo-0613' or \
-           self.model == 'gpt-3.5-turbo-1106':
-               return 4_096
-        if self.model == 'gpt-4' or \
-           self.model == 'gpt-4-0613' or \
-           self.model == 'text-embedding-ada-002' or \
-           self.model == 'text-embedding-3-small' or \
-           self.model == 'text-embedding-3-large':
-               return 8_192
-        if self.model == 'gpt-3.5-turbo-16k' or \
-           self.model == 'gpt-3.5-turbo-16k-0613' or \
-           self.model == 'davinci-002':
-               return 16_384
-        if self.model == 'gpt-4-32k' or \
-           self.model == 'gpt-4-32k-0613':
-               return 32_768
-        if self.model == 'gpt-4-1106-preview' or \
-           self.model == 'gpt-4-turbo-2024-04-09' or \
-           self.model == 'gpt-4-turbo' or \
-           self.model == 'gpt-4-1106' or \
-           self.model == 'gpt-4o' or \
-           self.model == 'gpt-4o-2024-11-20' or \
-           self.model == 'gpt-4o-mini' or \
-           self.model == 'chatgpt-4o-latest':
-               return 128_000
-        if self.model == 'o1' or \
-           self.model == 'o3' or \
-           self.model == 'o3-mini' or \
-           self.model == 'o4-mini':
-               return 200_000
-        if self.model == 'gpt-5' or \
-           self.model == 'gpt-5-mini' or \
-           self.model == 'gpt-5-nano' or \
-           self.model == 'gpt-5-chat-latest':
+        if (
+            self.model == "text-curie-001"
+            or self.model == "text-babbage-001"
+            or self.model == "text-ada-001"
+            or self.model == "davinci"
+            or self.model == "curie"
+            or self.model == "babbage"
+            or self.model == "ada"
+        ):
+            return 2_049
+        if (
+            self.model == "gpt-3.5-turbo"
+            or self.model == "gpt-3.5-turbo-0613"
+            or self.model == "gpt-3.5-turbo-1106"
+        ):
+            return 4_096
+        if (
+            self.model == "gpt-4"
+            or self.model == "gpt-4-0613"
+            or self.model == "text-embedding-ada-002"
+            or self.model == "text-embedding-3-small"
+            or self.model == "text-embedding-3-large"
+        ):
+            return 8_192
+        if (
+            self.model == "gpt-3.5-turbo-16k"
+            or self.model == "gpt-3.5-turbo-16k-0613"
+            or self.model == "davinci-002"
+        ):
+            return 16_384
+        if self.model == "gpt-4-32k" or self.model == "gpt-4-32k-0613":
+            return 32_768
+        if (
+            self.model == "gpt-4-1106-preview"
+            or self.model == "gpt-4-turbo-2024-04-09"
+            or self.model == "gpt-4-turbo"
+            or self.model == "gpt-4-1106"
+            or self.model == "gpt-4o"
+            or self.model == "gpt-4o-2024-11-20"
+            or self.model == "gpt-4o-mini"
+            or self.model == "chatgpt-4o-latest"
+        ):
+            return 128_000
+        if (
+            self.model == "o1"
+            or self.model == "o3"
+            or self.model == "o3-mini"
+            or self.model == "o4-mini"
+            or self.model == "gpt-5-chat-latest"
+            or self.model == "gpt-5.1-chat-latest"
+        ):
+            return 200_000
+        if (
+            self.model == "gpt-5"
+            or self.model == "gpt-5.1"
+            or self.model == "gpt-5-mini"
+            or self.model == "gpt-5-nano"
+        ):
             return 400_000
-        if self.model == 'gpt-4.1' or \
-           self.model == 'gpt-4.1-mini' or \
-           self.model == 'gpt-4.1-nano':
+        if self.model == "gpt-4.1" or self.model == "gpt-4.1-mini" or self.model == "gpt-4.1-nano":
             return 1_047_576
-        msg = f'Unsupported model: {self.model}'
+        msg = f"Unsupported model: {self.model}"
         UserMessage(msg)
         raise ValueError(msg)
 
     def api_max_response_tokens(self):
-        if self.model == 'davinci-002':
+        if self.model == "davinci-002":
             return 2_048
-        if self.model == 'gpt-4-turbo' or \
-           self.model == 'gpt-4-turbo-2024-04-09' or \
-           self.model == 'gpt-4-1106-preview' or \
-           self.model == 'gpt-3.5-turbo-1106' or \
-           self.model == 'gpt-3.5-turbo-0613' or \
-           self.model == 'gpt-3.5-turbo':
-               return 4_096
-        if self.model == 'gpt-4-0613' or \
-           self.model == 'gpt-4':
-               return 8_192
-        if self.model == 'gpt-3.5-turbo-16k-0613' or \
-           self.model == 'gpt-3.5-turbo-16k' or \
-           self.model == 'gpt-4o-mini' or \
-           self.model == 'gpt-4o' or \
-           self.model == 'gpt-4o-2024-11-20' or \
-           self.model == 'chatgpt-4o-latest':
-               return 16_384
-        if self.model == 'gpt-4.1' or \
-           self.model == 'gpt-4.1-mini' or \
-           self.model == 'gpt-4.1-nano':
+        if (
+            self.model == "gpt-4-turbo"
+            or self.model == "gpt-4-turbo-2024-04-09"
+            or self.model == "gpt-4-1106-preview"
+            or self.model == "gpt-3.5-turbo-1106"
+            or self.model == "gpt-3.5-turbo-0613"
+            or self.model == "gpt-3.5-turbo"
+        ):
+            return 4_096
+        if self.model == "gpt-4-0613" or self.model == "gpt-4":
+            return 8_192
+        if (
+            self.model == "gpt-3.5-turbo-16k-0613"
+            or self.model == "gpt-3.5-turbo-16k"
+            or self.model == "gpt-4o-mini"
+            or self.model == "gpt-4o"
+            or self.model == "gpt-4o-2024-11-20"
+            or self.model == "chatgpt-4o-latest"
+            or self.model == "gpt-5-chat-latest"
+            or self.model == "gpt-5.1-chat-latest"
+        ):
+            return 16_384
+        if self.model == "gpt-4.1" or self.model == "gpt-4.1-mini" or self.model == "gpt-4.1-nano":
             return 32_768
-        if self.model == 'o1' or \
-           self.model == 'o3' or \
-           self.model == 'o3-mini' or \
-           self.model == 'o4-mini':
-               return 100_000
-        if self.model == 'gpt-5' or \
-           self.model == 'gpt-5-mini' or \
-           self.model == 'gpt-5-nano' or \
-           self.model == 'gpt-5-chat-latest':
+        if (
+            self.model == "o1"
+            or self.model == "o3"
+            or self.model == "o3-mini"
+            or self.model == "o4-mini"
+        ):
+            return 100_000
+        if (
+            self.model == "gpt-5"
+            or self.model == "gpt-5.1"
+            or self.model == "gpt-5-mini"
+            or self.model == "gpt-5-nano"
+        ):
             return 128_000
-        msg = f'Unsupported model: {self.model}'
+        msg = f"Unsupported model: {self.model}"
         UserMessage(msg)
         raise ValueError(msg)
 
     def api_embedding_dims(self):
-        if self.model == 'text-embedding-ada-002':
+        if self.model == "text-embedding-ada-002":
             return 1_536
-        if self.model == 'text-embedding-3-small':
+        if self.model == "text-embedding-3-small":
             return 1_536
-        if self.model == 'text-embedding-3-large':
+        if self.model == "text-embedding-3-large":
             return 3_072
-        msg = f'Unsupported model: {self.model}'
+        msg = f"Unsupported model: {self.model}"
         UserMessage(msg)
         raise ValueError(msg)

symai/backend/settings.py

@@ -14,8 +14,8 @@ class SymAIConfig:
     def __init__(self):
         """Initialize configuration paths based on current Python environment."""
         self._env_path = Path(sys.prefix)
-        self._env_config_dir = self._env_path / '.symai'
-        self._home_config_dir = Path.home() / '.symai'
+        self._env_config_dir = self._env_path / ".symai"
+        self._home_config_dir = Path.home() / ".symai"
         self._debug_dir = Path.cwd()  # Current working directory for debug mode
         self._active_paths: dict[str, Path] = {}
 
@@ -31,7 +31,8 @@ class SymAIConfig:
         target_path = Path(key)
         target_name = target_path.name or key
         stale_keys: list[Path] = [
-            existing_key for existing_key in self._active_paths
+            existing_key
+            for existing_key in self._active_paths
             if isinstance(existing_key, Path)
             and (existing_key.name == target_name or str(existing_key) == key)
         ]
@@ -42,8 +43,8 @@ class SymAIConfig:
     def config_dir(self) -> Path:
         """Returns the active configuration directory based on priority system."""
         # Debug mode takes precedence
-        if (self._debug_dir / 'symai.config.json').exists():
-            return self._debug_dir / '.symai'
+        if (self._debug_dir / "symai.config.json").exists():
+            return self._debug_dir / ".symai"
         # Then environment config
         if self._env_config_dir.exists():
             return self._env_config_dir
@@ -60,11 +61,11 @@ class SymAIConfig:
         # Only use the basename for managed directories
         normalized_filename = Path(normalized_filename).name
         debug_config = self._debug_dir / normalized_filename
-        env_config   = self._env_config_dir / normalized_filename
-        home_config  = self._home_config_dir / normalized_filename
+        env_config = self._env_config_dir / normalized_filename
+        home_config = self._home_config_dir / normalized_filename
 
         # Check debug first (only valid for symai.config.json)
-        if normalized_filename == 'symai.config.json' and debug_config.exists():
+        if normalized_filename == "symai.config.json" and debug_config.exists():
             return debug_config
 
         # If forced to fallback, return home config if it exists, otherwise environment
@@ -88,7 +89,7 @@ class SymAIConfig:
             self._remove_legacy_path_keys(key)
             self._active_paths.pop(key, None)
             return {}
-        with config_path.open(encoding='utf-8') as f:
+        with config_path.open(encoding="utf-8") as f:
             config = json.load(f)
         self._remove_legacy_path_keys(key)
         self._active_paths[key] = config_path
@@ -99,7 +100,7 @@ class SymAIConfig:
         config_path = self.get_config_path(filename, fallback_to_home=fallback_to_home)
         key = self._canonical_key(filename)
         config_path.parent.mkdir(parents=True, exist_ok=True)
-        with config_path.open('w', encoding='utf-8') as f:
+        with config_path.open("w", encoding="utf-8") as f:
             json.dump(data, f, indent=4)
         self._remove_legacy_path_keys(key)
         self._active_paths[key] = config_path
@@ -117,9 +118,7 @@ class SymAIConfig:
         if cached is not None:
             return cached
         for legacy_key, cached_path in list(self._active_paths.items()):
-            if isinstance(legacy_key, Path) and (
-                legacy_key.name == key or str(legacy_key) == key
-            ):
+            if isinstance(legacy_key, Path) and (legacy_key.name == key or str(legacy_key) == key):
                 self._active_paths.pop(legacy_key, None)
                 self._active_paths[key] = cached_path
                 return cached_path
@@ -127,7 +126,7 @@ class SymAIConfig:
 
     def get_active_config_dir(self) -> Path:
         """Returns the directory backing the active symai configuration."""
-        symai_key = self._canonical_key('symai.config.json')
+        symai_key = self._canonical_key("symai.config.json")
         cached = self._active_paths.get(symai_key)
         if cached is not None:
             return cached.parent
@@ -140,6 +139,7 @@ class SymAIConfig:
                 return cached_path.parent
         return self.config_dir
 
+
 SYMAI_CONFIG = {}
 SYMSH_CONFIG = {}
 SYMSERVER_CONFIG = {}