symbolicai 1.1.0__tar.gz → 1.1.1__tar.gz

{symbolicai-1.1.0 → symbolicai-1.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: symbolicai
-Version: 1.1.0
+Version: 1.1.1
 Summary: A Neurosymbolic Perspective on Large Language Models
 Author-email: Marius-Constantin Dinu <marius@extensity.ai>, Leoveanu-Condrei Claudiu <leo@extensity.ai>
 License: BSD 3-Clause License

{symbolicai-1.1.0 → symbolicai-1.1.1}/symai/__init__.py

@@ -33,7 +33,7 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
 # Create singleton instance
 config_manager = settings.SymAIConfig()
 
-SYMAI_VERSION = "1.1.0"
+SYMAI_VERSION = "1.1.1"
 __version__ = SYMAI_VERSION
 __root_dir__ = config_manager.config_dir
 

{symbolicai-1.1.0 → symbolicai-1.1.1}/symai/backend/engines/neurosymbolic/__init__.py

@@ -1,6 +1,8 @@
 from ...mixin import (
     ANTHROPIC_CHAT_MODELS,
     ANTHROPIC_REASONING_MODELS,
+    CEREBRAS_CHAT_MODELS,
+    CEREBRAS_REASONING_MODELS,
     DEEPSEEK_CHAT_MODELS,
     DEEPSEEK_REASONING_MODELS,
     GOOGLE_CHAT_MODELS,
@@ -12,6 +14,7 @@ from ...mixin import (
 )
 from .engine_anthropic_claudeX_chat import ClaudeXChatEngine
 from .engine_anthropic_claudeX_reasoning import ClaudeXReasoningEngine
+from .engine_cerebras import CerebrasEngine
 from .engine_deepseekX_reasoning import DeepSeekXReasoningEngine
 from .engine_google_geminiX_reasoning import GeminiXReasoningEngine
 from .engine_groq import GroqEngine
@@ -22,6 +25,8 @@ from .engine_openai_gptX_reasoning import GPTXReasoningEngine
 ENGINE_MAPPING = {
     **dict.fromkeys(ANTHROPIC_CHAT_MODELS, ClaudeXChatEngine),
     **dict.fromkeys(ANTHROPIC_REASONING_MODELS, ClaudeXReasoningEngine),
+    **dict.fromkeys(CEREBRAS_CHAT_MODELS, CerebrasEngine),
+    **dict.fromkeys(CEREBRAS_REASONING_MODELS, CerebrasEngine),
     **dict.fromkeys(DEEPSEEK_REASONING_MODELS, DeepSeekXReasoningEngine),
     **dict.fromkeys(GOOGLE_REASONING_MODELS, GeminiXReasoningEngine),
     **dict.fromkeys(OPENAI_CHAT_MODELS, GPTXChatEngine),
@@ -33,6 +38,8 @@ ENGINE_MAPPING = {
 __all__ = [
     "ANTHROPIC_CHAT_MODELS",
     "ANTHROPIC_REASONING_MODELS",
+    "CEREBRAS_CHAT_MODELS",
+    "CEREBRAS_REASONING_MODELS",
     "DEEPSEEK_CHAT_MODELS",
     "DEEPSEEK_REASONING_MODELS",
     "ENGINE_MAPPING",

{symbolicai-1.1.0 → symbolicai-1.1.1}/symai/backend/mixin/__init__.py

@@ -1,5 +1,7 @@
 from .anthropic import SUPPORTED_CHAT_MODELS as ANTHROPIC_CHAT_MODELS
 from .anthropic import SUPPORTED_REASONING_MODELS as ANTHROPIC_REASONING_MODELS
+from .cerebras import SUPPORTED_CHAT_MODELS as CEREBRAS_CHAT_MODELS
+from .cerebras import SUPPORTED_REASONING_MODELS as CEREBRAS_REASONING_MODELS
 from .deepseek import SUPPORTED_CHAT_MODELS as DEEPSEEK_CHAT_MODELS
 from .deepseek import SUPPORTED_REASONING_MODELS as DEEPSEEK_REASONING_MODELS
 from .google import SUPPORTED_CHAT_MODELS as GOOGLE_CHAT_MODELS
@@ -12,6 +14,8 @@ from .openai import SUPPORTED_REASONING_MODELS as OPENAI_REASONING_MODELS
 __all__ = [
     "ANTHROPIC_CHAT_MODELS",
     "ANTHROPIC_REASONING_MODELS",
+    "CEREBRAS_CHAT_MODELS",
+    "CEREBRAS_REASONING_MODELS",
     "DEEPSEEK_CHAT_MODELS",
     "DEEPSEEK_REASONING_MODELS",
     "GOOGLE_CHAT_MODELS",

symbolicai-1.1.1/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",
+]

{symbolicai-1.1.0 → symbolicai-1.1.1}/symai/components.py

@@ -1330,6 +1330,25 @@ class MetadataTracker(Expression):
                     token_details[(engine_name, model_name)]["completion_breakdown"][
                         "reasoning_tokens"
                     ] += usage.output_tokens_details.reasoning_tokens
+                elif engine_name == "CerebrasEngine":
+                    usage = metadata["raw_output"].usage
+                    token_details[(engine_name, model_name)]["usage"]["completion_tokens"] += (
+                        usage.completion_tokens
+                    )
+                    token_details[(engine_name, model_name)]["usage"]["prompt_tokens"] += (
+                        usage.prompt_tokens
+                    )
+                    token_details[(engine_name, model_name)]["usage"]["total_tokens"] += (
+                        usage.total_tokens
+                    )
+                    token_details[(engine_name, model_name)]["usage"]["total_calls"] += 1
+                    #!: Backward compatibility for components like `RuntimeInfo`
+                    token_details[(engine_name, model_name)]["prompt_breakdown"][
+                        "cached_tokens"
+                    ] += 0  # Assignment not allowed with defualtdict
+                    token_details[(engine_name, model_name)]["completion_breakdown"][
+                        "reasoning_tokens"
+                    ] += 0
                 else:
                     logger.warning(f"Tracking {engine_name} is not supported.")
                     continue
@@ -1475,9 +1494,12 @@ class DynamicEngine(Expression):
         """Create an engine instance based on the model name."""
         # Deferred to avoid components <-> neurosymbolic engine circular imports.
         from .backend.engines.neurosymbolic import ENGINE_MAPPING  # noqa
+        from .backend.engines.neurosymbolic.engine_cerebras import CerebrasEngine  # noqa
 
         try:
             engine_class = ENGINE_MAPPING.get(self.model)
+            if engine_class is None and self.model.startswith("cerebras:"):
+                engine_class = CerebrasEngine
             if engine_class is None:
                 UserMessage(f"Unsupported model '{self.model}'", raise_with=ValueError)
             return engine_class(api_key=self.api_key, model=self.model)

{symbolicai-1.1.0 → symbolicai-1.1.1}/symbolicai.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: symbolicai
-Version: 1.1.0
+Version: 1.1.1
 Summary: A Neurosymbolic Perspective on Large Language Models
 Author-email: Marius-Constantin Dinu <marius@extensity.ai>, Leoveanu-Condrei Claudiu <leo@extensity.ai>
 License: BSD 3-Clause License

{symbolicai-1.1.0 → symbolicai-1.1.1}/symbolicai.egg-info/SOURCES.txt

@@ -168,6 +168,7 @@ symai/backend/engines/text_vision/engine_clip.py
 symai/backend/engines/userinput/engine_console.py
 symai/backend/mixin/__init__.py
 symai/backend/mixin/anthropic.py
+symai/backend/mixin/cerebras.py
 symai/backend/mixin/deepseek.py
 symai/backend/mixin/google.py
 symai/backend/mixin/groq.py