selectools 0.16.3__tar.gz → 0.16.4__tar.gz

{selectools-0.16.3/src/selectools.egg-info → selectools-0.16.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: selectools
-Version: 0.16.3
+Version: 0.16.4
 Summary: Production-ready AI agents with tool calling, structured output, execution traces, and RAG. Provider-agnostic (OpenAI, Anthropic, Gemini, Ollama) with fallback chains, batch processing, tool policies, streaming, caching, and cost tracking.
 Author-email: John <johnnichev@gmail.com>
 License-Expression: LGPL-3.0-or-later

{selectools-0.16.3 → selectools-0.16.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "selectools"
-version = "0.16.3"
+version = "0.16.4"
 description = "Production-ready AI agents with tool calling, structured output, execution traces, and RAG. Provider-agnostic (OpenAI, Anthropic, Gemini, Ollama) with fallback chains, batch processing, tool policies, streaming, caching, and cost tracking."
 readme = "README.md"
 requires-python = ">=3.9"

{selectools-0.16.3 → selectools-0.16.4}/src/selectools/__init__.py

@@ -1,6 +1,6 @@
 """Public exports for the selectools package."""
 
-__version__ = "0.16.3"
+__version__ = "0.16.4"
 
 # Import submodules (lazy loading for optional dependencies)
 from . import embeddings, guardrails, models, rag, toolbox

{selectools-0.16.3 → selectools-0.16.4}/src/selectools/agent/core.py

@@ -548,10 +548,12 @@ class Agent:
         self._wire_fallback_observer(run_id)
         self._notify_observers("on_run_start", run_id, messages, self._system_prompt)
 
-        # Input guardrails
-        for msg in messages:
-            if msg.role == Role.USER and msg.content:
-                msg.content = self._run_input_guardrails(msg.content, trace)
+        # Input guardrails (operate on copies to avoid mutating caller's objects)
+        if self.config.guardrails and self.config.guardrails.input:
+            messages = [copy.copy(msg) for msg in messages]
+            for msg in messages:
+                if msg.role == Role.USER and msg.content:
+                    msg.content = self._run_input_guardrails(msg.content, trace)
 
         # Memory / session loading
         if self.memory:
@@ -962,6 +964,7 @@ class Agent:
         prompt: str,
         stream_handler: Optional[Callable[[str], None]] = None,
         response_format: Optional[ResponseFormat] = None,
+        parent_run_id: Optional[str] = None,
     ) -> AgentResult:
         """
         Send a single user prompt and return the agent's response.
@@ -975,6 +978,7 @@ class Agent:
             response_format: Optional Pydantic model class or JSON Schema dict.
                 When provided the LLM is instructed to return valid JSON and
                 the result is validated. Access via ``result.parsed``.
+            parent_run_id: Optional run-ID of a parent agent for trace linking.
 
         Returns:
             AgentResult with the response and metadata.
@@ -988,6 +992,7 @@ class Agent:
             [Message(role=Role.USER, content=prompt)],
             stream_handler=stream_handler,
             response_format=response_format,
+            parent_run_id=parent_run_id,
         )
 
     async def aask(
@@ -995,6 +1000,7 @@ class Agent:
         prompt: str,
         stream_handler: Optional[Callable[[str], None]] = None,
         response_format: Optional[ResponseFormat] = None,
+        parent_run_id: Optional[str] = None,
     ) -> AgentResult:
         """
         Async version of :meth:`ask`.
@@ -1003,6 +1009,7 @@ class Agent:
             prompt: Plain-text question or instruction.
             stream_handler: Optional callback for streaming responses.
             response_format: Optional Pydantic model class or JSON Schema dict.
+            parent_run_id: Optional run-ID of a parent agent for trace linking.
 
         Returns:
             AgentResult with the response and metadata.
@@ -1015,6 +1022,7 @@ class Agent:
             [Message(role=Role.USER, content=prompt)],
             stream_handler=stream_handler,
             response_format=response_format,
+            parent_run_id=parent_run_id,
         )
 
     def batch(
@@ -1284,6 +1292,7 @@ class Agent:
                         response_text,
                         trace=ctx.trace,
                         run_id=ctx.run_id,
+                        user_text_for_coherence=ctx.user_text_for_coherence,
                     )
                     ctx.last_tool_name = _last_name
                     ctx.last_tool_args = _last_args
@@ -1717,6 +1726,7 @@ class Agent:
         response_text: str,
         trace: Optional[AgentTrace] = None,
         run_id: Optional[str] = None,
+        user_text_for_coherence: str = "",
     ) -> tuple:
         """Execute multiple tool calls concurrently using ThreadPoolExecutor.
 
@@ -1749,6 +1759,10 @@ class Agent:
             if policy_error:
                 return _Result(tc, policy_error, True, 0.0, tool, 0)
 
+            coherence_error = self._check_coherence(user_text_for_coherence, tool_name, parameters)
+            if coherence_error:
+                return _Result(tc, coherence_error, True, 0.0, tool, 0)
+
             call_id = tc.id or ""
             start = time.time()
             self._call_hook("on_tool_start", tool_name, parameters)
@@ -1771,6 +1785,7 @@ class Agent:
 
             try:
                 result = self._execute_tool_with_timeout(tool, parameters, chunk_cb)
+                result = self._screen_tool_result(tool_name, result)
                 dur = time.time() - start
                 self._call_hook("on_tool_end", tool_name, result, dur)
                 if run_id:
@@ -1880,6 +1895,7 @@ class Agent:
         response_text: str,
         trace: Optional[AgentTrace] = None,
         run_id: Optional[str] = None,
+        user_text_for_coherence: str = "",
     ) -> tuple:
         """Execute multiple tool calls concurrently using asyncio.gather.
 
@@ -1913,6 +1929,12 @@ class Agent:
             if policy_error:
                 return _Result(tc, policy_error, True, 0.0, tool, 0)
 
+            coherence_error = await self._acheck_coherence(
+                user_text_for_coherence, tool_name, parameters
+            )
+            if coherence_error:
+                return _Result(tc, coherence_error, True, 0.0, tool, 0)
+
             start = time.time()
             self._call_hook("on_tool_start", tool_name, parameters)
             if run_id:
@@ -1934,6 +1956,7 @@ class Agent:
 
             try:
                 result = await self._aexecute_tool_with_timeout(tool, parameters, chunk_cb)
+                result = self._screen_tool_result(tool_name, result)
                 dur = time.time() - start
                 self._call_hook("on_tool_end", tool_name, result, dur)
                 if run_id:
@@ -2136,6 +2159,8 @@ class Agent:
                                 timeout=self.config.request_timeout,
                             ),
                         )
+                    if _usage:
+                        self.usage.add_usage(_usage, tool_name=None)
                     self._call_hook("on_llm_end", response_msg.content, _usage)
                     self._notify_observers("on_llm_end", ctx.run_id, response_msg.content, _usage)
                     if _usage:
@@ -2276,6 +2301,7 @@ class Agent:
                         full_content,
                         trace=ctx.trace,
                         run_id=ctx.run_id,
+                        user_text_for_coherence=ctx.user_text_for_coherence,
                     )
                     ctx.last_tool_name = _last_name
                     ctx.last_tool_args = _last_args
@@ -2613,6 +2639,7 @@ class Agent:
                         response_text,
                         trace=ctx.trace,
                         run_id=ctx.run_id,
+                        user_text_for_coherence=ctx.user_text_for_coherence,
                     )
                     ctx.last_tool_name = _last_name
                     ctx.last_tool_args = _last_args

{selectools-0.16.3 → selectools-0.16.4/src/selectools.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: selectools
-Version: 0.16.3
+Version: 0.16.4
 Summary: Production-ready AI agents with tool calling, structured output, execution traces, and RAG. Provider-agnostic (OpenAI, Anthropic, Gemini, Ollama) with fallback chains, batch processing, tool policies, streaming, caching, and cost tracking.
 Author-email: John <johnnichev@gmail.com>
 License-Expression: LGPL-3.0-or-later