lite-agent 0.12.0__py3-none-any.whl → 0.13.0__py3-none-any.whl

lite_agent/runner.py

@@ -1,9 +1,10 @@
 import json
+import inspect
 from collections.abc import AsyncGenerator, Sequence
 from datetime import datetime, timedelta, timezone
 from os import PathLike
 from pathlib import Path
-from typing import Any, Literal, cast
+from typing import Any, Literal, cast, get_args, get_origin
 
 from funcall import Context
 
@@ -112,6 +113,59 @@ class Runner:
         """Normalize record_to parameter to Path object if provided."""
         return Path(record_to) if record_to else None
 
+    def _tool_expects_history_context(self, tool_calls: Sequence["ToolCall"]) -> bool:
+        """Check if any of the tool calls expect HistoryContext in their signatures.
+        
+        Returns True if any tool function has a Context[HistoryContext[...]] parameter,
+        False if they expect Context[...] without HistoryContext wrapper.
+        """
+        if not tool_calls:
+            return False
+            
+        for tool_call in tool_calls:
+            tool_func = self.agent.fc.function_registry.get(tool_call.function.name)
+            if not tool_func:
+                continue
+                
+            # Get function signature
+            sig = inspect.signature(tool_func)
+            
+            # Check each parameter for Context annotation
+            for param in sig.parameters.values():
+                if param.annotation == inspect.Parameter.empty:
+                    continue
+                    
+                # Check if parameter is Context[...]
+                origin = get_origin(param.annotation)
+                if origin is not None:
+                    # Check if it's Context type (compare by string name to handle import differences)
+                    origin_name = getattr(origin, '__name__', str(origin))
+                    if 'Context' in origin_name:
+                        args = get_args(param.annotation)
+                        if args:
+                            # Check if the Context contains HistoryContext
+                            inner_type = args[0]
+                            inner_origin = get_origin(inner_type)
+                            if inner_origin is not None:
+                                inner_name = getattr(inner_origin, '__name__', str(inner_origin))
+                                if 'HistoryContext' in inner_name:
+                                    logger.debug(f"Tool {tool_call.function.name} expects HistoryContext")
+                                    return True
+                            # Also check for direct HistoryContext class
+                            elif hasattr(inner_type, '__name__') and 'HistoryContext' in inner_type.__name__:
+                                logger.debug(f"Tool {tool_call.function.name} expects HistoryContext")
+                                return True
+                
+                # Also handle direct annotation checking
+                if hasattr(param.annotation, '__name__'):
+                    annotation_str = str(param.annotation)
+                    if 'HistoryContext' in annotation_str:
+                        logger.debug(f"Tool {tool_call.function.name} expects HistoryContext (direct)")
+                        return True
+                        
+        logger.debug("No tools expect HistoryContext")
+        return False
+
     async def _handle_tool_calls(self, tool_calls: "Sequence[ToolCall] | None", includes: Sequence[AgentChunkType], context: "Any | None" = None) -> AsyncGenerator[AgentChunk, None]:  # noqa: ANN401
         """Handle tool calls and yield appropriate chunks."""
         if not tool_calls:
@@ -184,23 +238,33 @@ class Runner:
                         )
             return  # Stop processing other tool calls after transfer
 
-        # Auto-inject history messages into context
-        if context is not None and not isinstance(context, Context):
-            # If user provided a plain object, wrap it in Context first
+        # Check if tools expect HistoryContext wrapper
+        expects_history = self._tool_expects_history_context(tool_calls)
+        
+        if expects_history:
+            # Auto-inject history messages into context for tools that expect HistoryContext
+            if context is not None and not isinstance(context, Context):
+                # If user provided a plain object, wrap it in Context first
+                context = Context(context)
+
+            if isinstance(context, Context):
+                # Extract original value and wrap in HistoryContext
+                original_value = context.value
+                wrapped = HistoryContext(
+                    history_messages=self.messages.copy(),
+                    data=original_value,
+                )
+                context = Context(wrapped)
+            else:
+                # No context provided, create HistoryContext with only history messages
+                wrapped = HistoryContext(history_messages=self.messages.copy())
+                context = Context(wrapped)
+        elif context is not None and not isinstance(context, Context):
+            # Tools don't expect HistoryContext, wrap user object in Context
             context = Context(context)
-
-        if isinstance(context, Context):
-            # Extract original value and wrap in HistoryContext
-            original_value = context.value
-            wrapped = HistoryContext(
-                history_messages=self.messages.copy(),
-                data=original_value,
-            )
-            context = Context(wrapped)
-        else:
-            # No context provided, create HistoryContext with only history messages
-            wrapped = HistoryContext(history_messages=self.messages.copy())
-            context = Context(wrapped)
+        elif context is None:
+            # Provide empty context for tools that don't expect HistoryContext
+            context = Context(None)
 
         async for tool_call_chunk in self.agent.handle_tool_calls(tool_calls, context=context):
             # if tool_call_chunk.type == "function_call" and tool_call_chunk.type in includes:
@@ -511,9 +575,10 @@ class Runner:
         max_steps: int = 20,
         includes: list[AgentChunkType] | None = None,
         record_to: PathLike | str | None = None,
+        context: Context | None = None,
     ) -> list[AgentChunk]:
         """Run the agent until it completes and return the final message."""
-        resp = self.run(user_input, max_steps, includes, record_to=record_to)
+        resp = self.run(user_input, max_steps, includes, record_to=record_to, context=context)
         return await self._collect_all_chunks(resp)
 
     def _analyze_last_assistant_message(self) -> tuple[list[AssistantToolCall], dict[str, str]]:

{lite_agent-0.12.0.dist-info → lite_agent-0.13.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lite-agent
-Version: 0.12.0
+Version: 0.13.0
 Summary: A lightweight, extensible framework for building AI agent.
 Author-email: Jianqi Pan <jannchie@gmail.com>
 License: MIT

{lite_agent-0.12.0.dist-info → lite_agent-0.13.0.dist-info}/RECORD

@@ -7,7 +7,7 @@ lite_agent/context.py,sha256=qnqj4JVYL0pkle1YptwM_CXJUWTeLaYtI1_MGtJYaHI,1225
 lite_agent/loggers.py,sha256=XkNkdqwD_nQGfhQJ-bBWT7koci_mMkNw3aBpyMhOICw,57
 lite_agent/message_transfers.py,sha256=N9ViK7Gxqqa1sd3V_hkNuQ9fUipg7M95l-sVBBG2Id4,5357
 lite_agent/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lite_agent/runner.py,sha256=p-Dw4RNYfRnIwnbt8oJv8e4qI84X7OUWLxgyITMnGe8,45090
+lite_agent/runner.py,sha256=H6gOeynLyLFr-ZTeEvBReuGdURkCVuP6T919_3AQ1CE,48568
 lite_agent/processors/__init__.py,sha256=ybpAzpMBIE9v5I24wIBZRXeaOaPNTmoKH13aofgNI6Q,234
 lite_agent/processors/completion_event_processor.py,sha256=zoWvs8dfrIkCSITGtS-4Hpve3WFCA0UUsMvYifL2fw0,13010
 lite_agent/processors/response_event_processor.py,sha256=Jr3cj1ItJ8aq9UBhEEjDwWDnPNOZ2ZXjWJ3-g4ghkhM,8514
@@ -30,6 +30,6 @@ lite_agent/utils/message_builder.py,sha256=J-yycL9pXSO9MbgC5NEGqvoP1LC2Nxe9r2YRW
 lite_agent/utils/message_converter.py,sha256=5HmNncTl71TD2M_6Ezz1Tnfavzna8DQYb4-D47Du7mA,9165
 lite_agent/utils/message_state_manager.py,sha256=rFUyqyd_7NdJRtyqsAWGcfwrDIlD6gK2dBDSDx1eGBs,5766
 lite_agent/utils/metrics.py,sha256=RzOEhCWxbLmmIEkzaxOJ6tAdthI8dv2Foc98Lq8afOQ,1915
-lite_agent-0.12.0.dist-info/METADATA,sha256=CvsMgd3DNYoZTvD0bTirNibch5S9eH8K-hjwWvEjFZo,3538
-lite_agent-0.12.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lite_agent-0.12.0.dist-info/RECORD,,
+lite_agent-0.13.0.dist-info/METADATA,sha256=dw2mkbA3X3ZY8bV1xxeW9ybNZnHcFFFXSeTdLl4vuyc,3538
+lite_agent-0.13.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lite_agent-0.13.0.dist-info/RECORD,,