symbolicai 1.6.0__py3-none-any.whl → 1.7.0__py3-none-any.whl

symai/__init__.py

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

symai/backend/engines/neurosymbolic/engine_anthropic_claudeX_chat.py

@@ -64,6 +64,7 @@ class ClaudeXChatEngine(Engine, AnthropicMixin):
                 and "4-0" not in self.config.get("NEUROSYMBOLIC_ENGINE_MODEL")
                 and "4-1" not in self.config.get("NEUROSYMBOLIC_ENGINE_MODEL")
                 and "4-5" not in self.config.get("NEUROSYMBOLIC_ENGINE_MODEL")
+                and "4-6" not in self.config.get("NEUROSYMBOLIC_ENGINE_MODEL")
             )
         ):
             return "neurosymbolic"

symai/backend/engines/neurosymbolic/engine_anthropic_claudeX_reasoning.py

@@ -66,6 +66,7 @@ class ClaudeXReasoningEngine(Engine, AnthropicMixin):
                 or "4-0" in self.config.get("NEUROSYMBOLIC_ENGINE_MODEL")
                 or "4-1" in self.config.get("NEUROSYMBOLIC_ENGINE_MODEL")
                 or "4-5" in self.config.get("NEUROSYMBOLIC_ENGINE_MODEL")
+                or "4-6" in self.config.get("NEUROSYMBOLIC_ENGINE_MODEL")
             )
         ):
             return "neurosymbolic"
@@ -348,9 +349,48 @@ class ClaudeXReasoningEngine(Engine, AnthropicMixin):
 
         return {"role": "user", "content": user_text}
 
+    def _build_output_config(self, response_format):
+        if response_format is None:
+            return NOT_GIVEN
+
+        if response_format["type"] == "json_schema":
+            schema = response_format.get("schema")
+            if schema is None and response_format.get("json_schema") is not None:
+                schema = response_format["json_schema"].get("schema", response_format["json_schema"])
+            if schema is None:
+                return NOT_GIVEN
+            return {"format": {"type": "json_schema", "schema": schema}}
+
+        if response_format["type"] == "json_object":
+            schema = response_format.get("schema")
+            if schema is None:
+                return NOT_GIVEN
+            return {"format": {"type": "json_schema", "schema": schema}}
+
+        return NOT_GIVEN
+
     def _prepare_request_payload(self, argument):
         kwargs = argument.kwargs
         model = kwargs.get("model", self.model)
+        long_context_1m = bool(kwargs.get("long_context_1m", False))
+        use_long_context_1m = long_context_1m and self.supports_long_context_1m(model)
+        effective_context_tokens = self.api_max_context_tokens(model=model)
+        if effective_context_tokens is None:
+            effective_context_tokens = 200_000
+        if use_long_context_1m:
+            effective_context_tokens = self.api_max_context_tokens(
+                long_context_1m=True, model=model
+            )
+        if long_context_1m and not use_long_context_1m:
+            UserMessage(
+                "long_context_1m is only supported for claude-opus-4-6 and claude-sonnet-4-5; "
+                f"falling back to {effective_context_tokens} token context."
+            )
+
+        extra_headers = None
+        if use_long_context_1m:
+            extra_headers = {"anthropic-beta": self.long_context_beta_header()}
+
         stop = kwargs.get("stop", NOT_GIVEN)
         temperature = kwargs.get("temperature", 1)
         thinking_arg = kwargs.get("thinking", NOT_GIVEN)
@@ -368,6 +408,8 @@ class ClaudeXReasoningEngine(Engine, AnthropicMixin):
         tool_choice = kwargs.get("tool_choice", NOT_GIVEN)
         metadata_anthropic = kwargs.get("metadata", NOT_GIVEN)
         max_tokens = kwargs.get("max_tokens", self.max_response_tokens)
+        response_format = kwargs.get("response_format", argument.prop.response_format)
+        output_config = self._build_output_config(response_format)
 
         if stop != NOT_GIVEN and not isinstance(stop, list):
             stop = [stop]
@@ -377,7 +419,7 @@ class ClaudeXReasoningEngine(Engine, AnthropicMixin):
         if stop != NOT_GIVEN:
             stop = [r"{s}" for s in stop]
 
-        return {
+        payload = {
             "model": model,
             "max_tokens": max_tokens,
             "stop_sequences": stop,
@@ -389,7 +431,12 @@ class ClaudeXReasoningEngine(Engine, AnthropicMixin):
             "metadata": metadata_anthropic,
             "tools": tools,
             "tool_choice": tool_choice,
+            "output_config": output_config,
         }
+        if extra_headers is not None:
+            payload["extra_headers"] = extra_headers
+
+        return payload
 
     def _collect_response(self, res):
         if isinstance(res, list):

symai/backend/engines/neurosymbolic/engine_cerebras.py

@@ -16,6 +16,7 @@ logging.getLogger("requests").setLevel(logging.ERROR)
 logging.getLogger("urllib").setLevel(logging.ERROR)
 logging.getLogger("httpx").setLevel(logging.ERROR)
 logging.getLogger("httpcore").setLevel(logging.ERROR)
+logging.getLogger("hpack").setLevel(logging.ERROR)
 
 
 _NON_VERBOSE_OUTPUT = (

symai/backend/mixin/anthropic.py

@@ -10,6 +10,7 @@ SUPPORTED_CHAT_MODELS = [
     "claude-3-haiku-20240307",
 ]
 SUPPORTED_REASONING_MODELS = [
+    "claude-opus-4-6",
     "claude-opus-4-5",
     "claude-opus-4-1",
     "claude-opus-4-0",
@@ -19,29 +20,49 @@ SUPPORTED_REASONING_MODELS = [
     "claude-sonnet-4-5",
 ]
 
+LONG_CONTEXT_1M_TOKENS = 1_000_000
+LONG_CONTEXT_1M_BETA_HEADER = "context-1m-2025-08-07"
+LONG_CONTEXT_1M_MODELS = {
+    "claude-opus-4-6",
+    "claude-sonnet-4-5",
+}
+
 
 class AnthropicMixin:
-    def api_max_context_tokens(self):
+    def supports_long_context_1m(self, model: str) -> bool:
+        return model in LONG_CONTEXT_1M_MODELS
+
+    def long_context_beta_header(self) -> str:
+        return LONG_CONTEXT_1M_BETA_HEADER
+
+    def api_max_context_tokens(self, long_context_1m: bool = False, model: str | None = None):
+        selected_model = self.model if model is None else model
+        if long_context_1m and self.supports_long_context_1m(selected_model):
+            return LONG_CONTEXT_1M_TOKENS
         if (
-            self.model == "claude-opus-4-5"
-            or 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"
+            selected_model == "claude-opus-4-6"
+            or selected_model == "claude-opus-4-5"
+            or selected_model == "claude-opus-4-1"
+            or selected_model == "claude-opus-4-0"
+            or selected_model == "claude-sonnet-4-0"
+            or selected_model == "claude-3-7-sonnet-latest"
+            or selected_model == "claude-haiku-4-5"
+            or selected_model == "claude-sonnet-4-5"
+            or selected_model == "claude-3-5-sonnet-latest"
+            or selected_model == "claude-3-5-sonnet-20241022"
+            or selected_model == "claude-3-5-sonnet-20240620"
+            or selected_model == "claude-3-opus-latest"
+            or selected_model == "claude-3-opus-20240229"
+            or selected_model == "claude-3-sonnet-20240229"
+            or selected_model == "claude-3-haiku-20240307"
         ):
             return 200_000
         return None
 
     def api_max_response_tokens(self):
+        if self.model == "claude-opus-4-6":
+            return 128_000
+
         if (
             self.model == "claude-opus-4-5"
             or self.model == "claude-sonnet-4-0"
@@ -61,7 +82,7 @@ class AnthropicMixin:
         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-opus-20240229"
             or self.model == "claude-3-sonnet-20240229"
             or self.model == "claude-3-haiku-20240307"
         ):

symai/strategy.py

@@ -3,9 +3,11 @@ import logging
 import time
 from collections import defaultdict
 from collections.abc import Callable
+from contextlib import nullcontext
 from typing import Any, ClassVar
 
 import numpy as np
+from anthropic import transform_schema
 from beartype import beartype
 from loguru import logger
 from pydantic import BaseModel, ValidationError
@@ -14,7 +16,9 @@ from rich.markup import escape
 from rich.panel import Panel
 from rich.table import Table
 
+from .backend.settings import SYMAI_CONFIG
 from .components import Function
+from .context import CURRENT_ENGINE_VAR
 from .models import LLMDataModel, TypeValidationError, build_dynamic_llm_datamodel
 from .symbol import Expression
 from .utils import UserMessage
@@ -182,6 +186,7 @@ class TypeValidationFunction(ValidationFunction):
         self.output_data_model = None
         self.accumulate_errors = accumulate_errors
         self.verbose = verbose
+        self.dynamic_engine = self.retry_params.get("dynamic_engine")
 
     def register_expected_data_model(
         self, data_model: LLMDataModel, attach_to: str, override: bool = False
@@ -361,7 +366,8 @@ Important guidelines:
         if self.verbose:
             self.display_panel(self.remedy_function.dynamic_context, title="New Context")
 
-        json_str = self.remedy_function(seed=remedy_seeds[attempt_index], **kwargs).value
+        with (self.dynamic_engine if self.dynamic_engine is not None else nullcontext()):
+            json_str = self.remedy_function(seed=remedy_seeds[attempt_index], **kwargs).value
         logger.info("Applied remedy function with updated context!")
         return json_str
 
@@ -418,17 +424,51 @@ Important guidelines:
             violations=errors,
         )
 
+    def _build_response_format(self, kwargs: dict) -> dict:
+        response_format = kwargs.get("response_format")
+        if response_format is None:
+            response_format = {"type": "json_object"}
+        # Resolve the effective model from the most specific runtime source first.
+        # This keeps contracts working when engines are swapped dynamically and no static
+        # model is registered in config.
+        selected_model = kwargs.get("model")
+        if selected_model is None:
+            active_engine = CURRENT_ENGINE_VAR.get()
+            if active_engine is not None:
+                selected_model = getattr(active_engine, "model", None)
+        if selected_model is None:
+            selected_model = SYMAI_CONFIG.get("NEUROSYMBOLIC_ENGINE_MODEL", "")
+
+        if "claude-opus" not in selected_model:
+            return response_format
+
+        if response_format.get("type") != "json_object":
+            return response_format
+
+        # At the contract layer, `json_object` stays the generic trigger for structured output.
+        # Opus requires a concrete JSON Schema payload, so we derive it from the output model.
+        return {"type": "json_schema", "schema": transform_schema(self.output_data_model)}
+
     def forward(
         self, prompt: str, f_semantic_conditions: list[Callable] | None = None, *args, **kwargs
     ):
         self._ensure_output_model()
         validation_context = kwargs.pop("validation_context", {})
-        kwargs["response_format"] = {"type": "json_object"}
+        kwargs["response_format"] = self._build_response_format(kwargs)
         logger.info("Initializing validation…")
         self._display_verbose_panels(prompt)
 
         context = self.zero_shot_prompt(prompt=prompt)
-        json_str = super().forward(context, *args, **kwargs).value
+
+        # Zero-shot attempt; fall back to dynamic engine on failure
+        try:
+            json_str = super().forward(context, *args, **kwargs).value
+        except Exception:
+            if self.dynamic_engine is None:
+                raise
+            logger.info("Zero-shot failed; retrying with dynamic engine…")
+            with self.dynamic_engine:
+                json_str = super().forward(context, *args, **kwargs).value
 
         remedy_seeds = self.prepare_seeds(self.retry_params["tries"] + 1, **kwargs)
         logger.info(f"Prepared {len(remedy_seeds)} remedy seeds for validation attempts…")
@@ -468,7 +508,7 @@ class contract:
         post_remedy: bool = False,
         accumulate_errors: bool = False,
         verbose: bool = False,
-        remedy_retry_params: dict[str, int | float | bool] = _default_remedy_retry_params,
+        remedy_retry_params: dict[str, Any] = _default_remedy_retry_params,
     ):
         """
         A contract class decorator inspired by DbC principles. It ensures that the function's input and output

{symbolicai-1.6.0.dist-info → symbolicai-1.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: symbolicai
-Version: 1.6.0
+Version: 1.7.0
 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
@@ -52,7 +52,7 @@ Requires-Dist: tqdm>=4.66.3
 Requires-Dist: python-box>=7.1.1
 Requires-Dist: pandas>=2.2.2
 Requires-Dist: scikit-learn>=1.5.0
-Requires-Dist: torch>=2.2.2
+Requires-Dist: torch<2.10.0
 Requires-Dist: torchaudio>=2.2.2
 Requires-Dist: torchvision>=0.17.2
 Requires-Dist: PyYAML>=6.0.1

{symbolicai-1.6.0.dist-info → symbolicai-1.7.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 symai/TERMS_OF_SERVICE.md,sha256=HN42UXVI_wAVDHjMShzy_k7xAsbjXaATNeMKcIte_eg,91409
-symai/__init__.py,sha256=P9fADfIIeSF_rI1YQUhBHQIMieDgvLt-QM4KvOKtvt4,15612
+symai/__init__.py,sha256=j5bwtmnvxtMH6h5LsdMB5hgynMwXRVH5tWJ5rp49Am0,15612
 symai/chat.py,sha256=DCEbmZ96wv-eitAVt6-oF6PT3JM3cT59Iy3r2Hucd_M,14100
 symai/components.py,sha256=SsGl8Cs-AQcPZjFyTv4IB3ATz8evRiVwTBH7k2umVSo,75391
 symai/constraints.py,sha256=ljjB9p0qK4DrDl_u5G_Y-Y6WAH5ZHANIqLLxRtwcORs,1980
@@ -17,7 +17,7 @@ symai/processor.py,sha256=FsR7rIjm6NlestgX3dVOLtvQJcGNpJaogtPyi3I1PR4,1648
 symai/prompts.py,sha256=cF3FauHfl0ZiBW-6DEE50wx_elNN7PBTbbmlVCr7nzQ,93743
 symai/shell.py,sha256=iEQ-dGGPSm7ZDaaC3I7RwnzgA8X9yHBzPD0TouBk3AU,6487
 symai/shellsv.py,sha256=rdKwBRqTT0k81LgMi0qshD4z8ucYL5iDCN4MOCgvEa8,39581
-symai/strategy.py,sha256=BQTXRnBv57fYO47A--WA6KK1oqGmf9Aijm0p4a_vvqY,45004
+symai/strategy.py,sha256=8-BlvTQ-o1XlVbHiu5LE7wtYRyqwAJPv3qgN6UHnuN0,46932
 symai/symbol.py,sha256=s5CYwP5SGcRUzZ7TlakZFpKBX_Q0mwPQKRbv4pC3sxM,40443
 symai/symsh.md,sha256=QwY_-fX0Ge7Aazul0xde2DuF2FZLw_elxrkXR3kuKDQ,1245
 symai/utils.py,sha256=m4iQzxclkPAUSDderTO_OK2fKznJ69pLfbBcTYq4p70,10824
@@ -39,9 +39,9 @@ symai/backend/engines/index/engine_qdrant.py,sha256=U9p0kzYvHE4DjFgxnvnG_8xfEoP_
 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=zEjkuxtzBQdxx-b1aXzooCatA43Rrlo9jAyZZfRYNzM,2608
-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
+symai/backend/engines/neurosymbolic/engine_anthropic_claudeX_chat.py,sha256=Y5AOCmJgL51QgrVbezSd0pe8GgMYNCuWWDIDES1_eaM,21133
+symai/backend/engines/neurosymbolic/engine_anthropic_claudeX_reasoning.py,sha256=PaIN_C6xMhAUw51vGkI774q-PMnq30mofWsOtwYrIc8,23393
+symai/backend/engines/neurosymbolic/engine_cerebras.py,sha256=zFnKKd_9iEL2S-4hKFMG4vPAK-fkSOkaUMJlZ5EvoF4,13387
 symai/backend/engines/neurosymbolic/engine_deepseekX_reasoning.py,sha256=_k0hy5dUxkbNP9vgF4R780A6THtjfcCJ_xUnkPvPRNw,9299
 symai/backend/engines/neurosymbolic/engine_google_geminiX_reasoning.py,sha256=pvsUNtWYwvFXt8EEy5ed0V4xE4EE0IZ4f7Uw5jV3OF0,28333
 symai/backend/engines/neurosymbolic/engine_groq.py,sha256=pliezi4X1LP-XdnJirjlI2nCXC8S2pex8fZalNlkKWw,13122
@@ -66,7 +66,7 @@ symai/backend/engines/text_to_speech/engine_openai.py,sha256=AtY0mDvIM_yZQ6AgYNX
 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=LMeNSxy0Uwn3CSSImgGcRal7JpZqIxP9I1lw0C4sLJ0,1539
-symai/backend/mixin/anthropic.py,sha256=GdHimGqiJcA21Jo797ZEeFzotRpCOJdBJQIChl_6NJI,2403
+symai/backend/mixin/anthropic.py,sha256=b06RGsEa_lvn1VBKRtLt1BPROCnR6g_lPO0BQtU2s0o,3213
 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
@@ -143,9 +143,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=l4r4rz29c7cO1dapXO0LQ4sHW4WF44keuz7j8v5azMc,9854
-symbolicai-1.6.0.dist-info/licenses/LICENSE,sha256=9vRFudlJ1ghVfra5lcCUIYQCqnZSYcBLjLHbGRsrQCs,1505
-symbolicai-1.6.0.dist-info/METADATA,sha256=efGH2KBd8r1A30XkrgOgIHHF3tg8KwgPFG_TtgOtOCc,22946
-symbolicai-1.6.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-symbolicai-1.6.0.dist-info/entry_points.txt,sha256=JV5sdydIfUZdDF6QBEQHiZHod6XNPjCjpWQrXh7gTAw,261
-symbolicai-1.6.0.dist-info/top_level.txt,sha256=bOoIDfpDIvCQtQgXcwVKJvxAKwsxpxo2IL4z92rNJjw,6
-symbolicai-1.6.0.dist-info/RECORD,,
+symbolicai-1.7.0.dist-info/licenses/LICENSE,sha256=9vRFudlJ1ghVfra5lcCUIYQCqnZSYcBLjLHbGRsrQCs,1505
+symbolicai-1.7.0.dist-info/METADATA,sha256=leIkcs9g6yfwn3e7qqRGnZHUL_Co6M9G5W9YddvE1QM,22946
+symbolicai-1.7.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+symbolicai-1.7.0.dist-info/entry_points.txt,sha256=JV5sdydIfUZdDF6QBEQHiZHod6XNPjCjpWQrXh7gTAw,261
+symbolicai-1.7.0.dist-info/top_level.txt,sha256=bOoIDfpDIvCQtQgXcwVKJvxAKwsxpxo2IL4z92rNJjw,6
+symbolicai-1.7.0.dist-info/RECORD,,