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

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
 

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",

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",

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/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.dist-info → symbolicai-1.1.1.dist-info}/METADATA

@@ -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.dist-info → symbolicai-1.1.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 symai/TERMS_OF_SERVICE.md,sha256=HN42UXVI_wAVDHjMShzy_k7xAsbjXaATNeMKcIte_eg,91409
-symai/__init__.py,sha256=tfavYhEZTrbPwBKJWYPhxGs1ceV-HDIvPNbxkp8KlKI,18530
+symai/__init__.py,sha256=KnIPNxN20P6-2mEa3c9mbeCNI9aV7w0_HJEjMZmMOjU,18530
 symai/chat.py,sha256=DCEbmZ96wv-eitAVt6-oF6PT3JM3cT59Iy3r2Hucd_M,14100
-symai/components.py,sha256=b8B8AvjbMMEgmFSqpzvLYLUZu6zwRnRc1r2hJWSbf1s,62584
+symai/components.py,sha256=iMpuICXkfXKA2j2u__92COQ10BmqNKHH4GginihD614,63906
 symai/constraints.py,sha256=ljjB9p0qK4DrDl_u5G_Y-Y6WAH5ZHANIqLLxRtwcORs,1980
 symai/context.py,sha256=4M69MJOeWSdPTr2Y9teoNTs-nEvpzcAcr7900UgORXA,189
 symai/core.py,sha256=gI9qvTT0Skq2D0izdhAoN3RdwBtWei59KO52mKN1Sos,70420
@@ -44,7 +44,7 @@ symai/backend/engines/index/engine_pinecone.py,sha256=fxCew1ldUdjd9UtqnMuWFDiVz5
 symai/backend/engines/index/engine_qdrant.py,sha256=f2lguAgCTZz2p6UI__u7puxH-X9UteWxBIzQnOUX1Xk,37748
 symai/backend/engines/index/engine_vectordb.py,sha256=xXU8QaC2BX9O4dDjDCVYgWO4PxQMpmNlhtal6UVtV0o,8541
 symai/backend/engines/lean/engine_lean4.py,sha256=ln5nbQn5szq8nRulbREPLCPQ5bwjM_A5XAGMkfzPdT8,10102
-symai/backend/engines/neurosymbolic/__init__.py,sha256=TKPs76bp726JQnHpPy-iLGUquZSf5SiUsui7UTLRAxs,1777
+symai/backend/engines/neurosymbolic/__init__.py,sha256=txvvXueinXrY1wq2Fp9xYqb5eLbkxXIJ82whhphKqyE,2062
 symai/backend/engines/neurosymbolic/engine_anthropic_claudeX_chat.py,sha256=WyuskobMjf9ynxRWUnXk55DUMUN7qv7jT1nbZP3Bx9o,21054
 symai/backend/engines/neurosymbolic/engine_anthropic_claudeX_reasoning.py,sha256=thEPDh8H-5XrsADF0mVTWB6m_dJgKeTT49HvyCGJcQM,21291
 symai/backend/engines/neurosymbolic/engine_cerebras.py,sha256=ki84Qh7hdxaKn--UgMMUvAEoqJos7VeKtkka6XpHI3g,13336
@@ -68,8 +68,9 @@ symai/backend/engines/symbolic/engine_wolframalpha.py,sha256=mTH0N4rA0gMffSBLjf2
 symai/backend/engines/text_to_speech/engine_openai.py,sha256=AtY0mDvIM_yZQ6AgYNXuyinZr_OaMK7XiPLQ6fe6RBo,2013
 symai/backend/engines/text_vision/engine_clip.py,sha256=hU9vsHtKPpQYEoESyjuGXOzMhUNhvspYMCNkCAqn2x8,3648
 symai/backend/engines/userinput/engine_console.py,sha256=fDO6PRQI3NYZ_nHVXDFIsS9cFDRv3aTOfv8h5a360jc,743
-symai/backend/mixin/__init__.py,sha256=yGo_NQjG5r7Y1_fXCXPMvLFxNrqB_dA60rQ0CXITLRo,1008
+symai/backend/mixin/__init__.py,sha256=KamwgTNRE63ro-uy5QGmPWkmSMKNxm6pRiRfG_LSic0,1215
 symai/backend/mixin/anthropic.py,sha256=kTHJnFlmgqN6X5fjsAnwSwZ6Qdhr-Zo5vI46_D9Ke3Y,2286
+symai/backend/mixin/cerebras.py,sha256=MEc9vQ6G4KWWrt0NFjdt2y0rojhtBidwa_n4M8Z5EKI,215
 symai/backend/mixin/deepseek.py,sha256=7TnyqXQb2t6r6-hzOClPzxfO2d7TShYC989Lmn_YTzM,414
 symai/backend/mixin/google.py,sha256=N1xxrrTcQkcKJtdPbRorev6dfJ1F65I5XavrGR06GN4,494
 symai/backend/mixin/groq.py,sha256=at6yFLa35Js8o7D8p_-Y4NjOPJI-lH8yx6tsCDrEy6M,227
@@ -160,9 +161,9 @@ symai/server/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 symai/server/huggingface_server.py,sha256=wSAVqFiKQsCu5UB2YYVpxJBhJ7GgQBBfePxNi265yP8,9039
 symai/server/llama_cpp_server.py,sha256=-WPTNB2cbnwtnpES4AtPM__MCasDKl83jr94JGS9tmI,2144
 symai/server/qdrant_server.py,sha256=pOm382ZmPJ2jlJiLIxt_KUJqISOgWiKhs3Ojbp5REOI,7420
-symbolicai-1.1.0.dist-info/licenses/LICENSE,sha256=9vRFudlJ1ghVfra5lcCUIYQCqnZSYcBLjLHbGRsrQCs,1505
-symbolicai-1.1.0.dist-info/METADATA,sha256=IrMaEQXGdoacrMYuIiWkqOuCoEdCo7s80Igr0rGrjI4,23603
-symbolicai-1.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-symbolicai-1.1.0.dist-info/entry_points.txt,sha256=JV5sdydIfUZdDF6QBEQHiZHod6XNPjCjpWQrXh7gTAw,261
-symbolicai-1.1.0.dist-info/top_level.txt,sha256=bOoIDfpDIvCQtQgXcwVKJvxAKwsxpxo2IL4z92rNJjw,6
-symbolicai-1.1.0.dist-info/RECORD,,
+symbolicai-1.1.1.dist-info/licenses/LICENSE,sha256=9vRFudlJ1ghVfra5lcCUIYQCqnZSYcBLjLHbGRsrQCs,1505
+symbolicai-1.1.1.dist-info/METADATA,sha256=NZ8WBjmA0arIzmZQPnBRE6-SOgDtCvMI6VHOhLzzOGU,23603
+symbolicai-1.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+symbolicai-1.1.1.dist-info/entry_points.txt,sha256=JV5sdydIfUZdDF6QBEQHiZHod6XNPjCjpWQrXh7gTAw,261
+symbolicai-1.1.1.dist-info/top_level.txt,sha256=bOoIDfpDIvCQtQgXcwVKJvxAKwsxpxo2IL4z92rNJjw,6
+symbolicai-1.1.1.dist-info/RECORD,,