agnt5 0.2.4__cp39-abi3-macosx_11_0_arm64.whl → 0.2.5__cp39-abi3-macosx_11_0_arm64.whl

agnt5/__init__.py

@@ -32,7 +32,7 @@ from .tool import Tool, ToolRegistry, tool
 from .types import BackoffPolicy, BackoffType, FunctionConfig, RetryPolicy, WorkflowConfig
 from .version import _get_version
 from .worker import Worker
-from .workflow import WorkflowRegistry, workflow
+from .workflow import WorkflowRegistry, chatflow, workflow
 
 # Expose simplified language model API (recommended)
 from . import lm
@@ -57,6 +57,7 @@ __all__ = [
     "with_entity_context",
     "create_entity_context",
     "workflow",
+    "chatflow",
     "WorkflowRegistry",
     "tool",
     "Tool",

agnt5/agent.py

@@ -377,13 +377,16 @@ class Agent:
             if handoff_config.pass_full_history:
                 # Get current conversation from the agent's run loop
                 # (This will be set when we detect the handoff in run())
-                conversation_history = ctx.get("_current_conversation", [])
+                conversation_history = getattr(ctx, '_agent_data', {}).get("_current_conversation", [])
+
                 if conversation_history:
                     ctx.logger.info(
                         f"Passing {len(conversation_history)} messages to target agent"
                     )
                     # Store in context for target agent to optionally use
-                    ctx.set("_handoff_conversation_history", conversation_history)
+                    if not hasattr(ctx, '_agent_data'):
+                        ctx._agent_data = {}
+                    ctx._agent_data["_handoff_conversation_history"] = conversation_history
 
             # Execute target agent with the message and shared context
             result = await target_agent.run(message, context=ctx)
@@ -398,7 +401,9 @@ class Agent:
                 "tool_calls": result.tool_calls,
             }
 
-            ctx.set("_handoff_result", handoff_data)
+            if not hasattr(ctx, '_agent_data'):
+                ctx._agent_data = {}
+            ctx._agent_data["_handoff_result"] = handoff_data
 
             # Return the handoff data (will be detected in run() loop)
             return handoff_data
@@ -449,7 +454,6 @@ class Agent:
 
             context = Context(
                 run_id=f"agent-{self.name}-{uuid.uuid4().hex[:8]}",
-                component_type="agent",
             )
 
         # Create span for agent execution with trace linking
@@ -471,8 +475,6 @@ class Agent:
 
             # Reasoning loop
             for iteration in range(self.max_iterations):
-                self.logger.info(f"Agent iteration {iteration + 1}/{self.max_iterations}")
-
                 # Build tool definitions for LLM
                 tool_defs = [
                     ToolDefinition(
@@ -517,10 +519,13 @@ class Agent:
 
                 # Check if LLM wants to use tools
                 if response.tool_calls:
-                    self.logger.info(f"Agent calling {len(response.tool_calls)} tool(s)")
+                    self.logger.debug(f"Agent calling {len(response.tool_calls)} tool(s)")
 
                     # Store current conversation in context for potential handoffs
-                    context.set("_current_conversation", messages)
+                    # Use a simple dict attribute since we don't need full state persistence for this
+                    if not hasattr(context, '_agent_data'):
+                        context._agent_data = {}
+                    context._agent_data["_current_conversation"] = messages
 
                     # Execute tool calls
                     tool_results = []
@@ -586,13 +591,13 @@ class Agent:
                             for tr in tool_results
                         ]
                     )
-                    messages.append(Message.user(f"Tool results:\n{results_text}"))
+                    messages.append(Message.user(f"Tool results:\n{results_text}\n\nPlease provide your final answer based on these results."))
 
                     # Continue loop for agent to process results
 
                 else:
                     # No tool calls - agent is done
-                    self.logger.info(f"Agent completed after {iteration + 1} iterations")
+                    self.logger.debug(f"Agent completed after {iteration + 1} iterations")
                     return AgentResult(
                         output=response.text,
                         tool_calls=all_tool_calls,
@@ -636,7 +641,6 @@ class Agent:
 
             context = Context(
                 run_id=f"agent-chat-{self.name}-{uuid.uuid4().hex[:8]}",
-                component_type="agent",
             )
 
         # Add user message

agnt5/lm.py

@@ -312,11 +312,23 @@ class _LanguageModel:
         if request.response_schema is not None:
             kwargs["response_schema_kw"] = request.response_schema
 
-        # TODO: Add tools and tool_choice support when needed
-        # if request.tools:
-        #     kwargs["tools"] = self._serialize_tools(request.tools)
-        # if request.tool_choice:
-        #     kwargs["tool_choice"] = request.tool_choice.value
+        # Pass tools and tool_choice to Rust
+        if request.tools:
+            # Serialize tools to JSON for Rust
+            tools_list = [
+                {
+                    "name": tool.name,
+                    "description": tool.description,
+                    "parameters": tool.parameters,
+                }
+                for tool in request.tools
+            ]
+            tools_json = json.dumps(tools_list)
+            kwargs["tools"] = tools_json
+
+        if request.tool_choice:
+            # Serialize tool_choice to JSON for Rust
+            kwargs["tool_choice"] = json.dumps(request.tool_choice.value)
 
         # Call Rust implementation - it returns a proper Python coroutine now
         # Using pyo3-async-runtimes for truly async HTTP calls without blocking
@@ -362,11 +374,22 @@ class _LanguageModel:
         if request.config.top_p is not None:
             kwargs["top_p"] = request.config.top_p
 
-        # TODO: Add tools and tool_choice support when needed
-        # if request.tools:
-        #     kwargs["tools"] = self._serialize_tools(request.tools)
-        # if request.tool_choice:
-        #     kwargs["tool_choice"] = request.tool_choice.value
+        # Pass tools and tool_choice to Rust
+        if request.tools:
+            # Serialize tools to JSON for Rust
+            tools_list = [
+                {
+                    "name": tool.name,
+                    "description": tool.description,
+                    "parameters": tool.parameters,
+                }
+                for tool in request.tools
+            ]
+            kwargs["tools"] = json.dumps(tools_list)
+
+        if request.tool_choice:
+            # Serialize tool_choice to JSON for Rust
+            kwargs["tool_choice"] = json.dumps(request.tool_choice.value)
 
         # Call Rust implementation - it returns a proper Python coroutine now
         # Using pyo3-async-runtimes for truly async streaming without blocking
@@ -416,11 +439,16 @@ class _LanguageModel:
                 total_tokens=rust_response.usage.total_tokens,
             )
 
+        # Extract tool_calls from Rust response
+        tool_calls = None
+        if hasattr(rust_response, 'tool_calls') and rust_response.tool_calls:
+            tool_calls = rust_response.tool_calls
+
         return GenerateResponse(
             text=rust_response.content,
             usage=usage,
             finish_reason=None,  # TODO: Add finish_reason to Rust response
-            tool_calls=None,  # TODO: Add tool calls support
+            tool_calls=tool_calls,
             _rust_response=rust_response,  # Store for .structured_output access
         )
 

agnt5/workflow.py

@@ -565,3 +565,121 @@ def workflow(
         return decorator
     else:
         return decorator(_func)
+
+
+def chatflow(
+    _func: Optional[Callable[..., Any]] = None,
+    *,
+    name: Optional[str] = None,
+) -> Callable[..., Any]:
+    """
+    Decorator to mark a function as an AGNT5 chat-enabled workflow.
+
+    Identical to @workflow but adds metadata {"chat": "true"} to indicate
+    this workflow is designed for multi-turn conversation scenarios.
+
+    The platform can use this metadata to:
+    - Enable session affinity and sticky routing
+    - Apply conversation-specific optimizations
+    - Track chat-specific metrics (turn count, conversation length)
+
+    Args:
+        name: Custom workflow name (default: function's __name__)
+
+    Example:
+        @chatflow
+        async def customer_support_chat(ctx: WorkflowContext, message: str) -> dict:
+            # Initialize conversation state
+            if not ctx.state.get("messages"):
+                ctx.state.set("messages", [])
+
+            # Add user message
+            messages = ctx.state.get("messages")
+            messages.append({"role": "user", "content": message})
+            ctx.state.set("messages", messages)
+
+            # Generate AI response
+            response = await ctx.task(generate_response, messages=messages)
+
+            # Add assistant response
+            messages.append({"role": "assistant", "content": response})
+            ctx.state.set("messages", messages)
+
+            return {"response": response, "turn_count": len(messages) // 2}
+    """
+
+    def decorator(func: Callable[..., Any]) -> Callable[..., Any]:
+        # Get workflow name
+        workflow_name = name or func.__name__
+
+        # Validate function signature
+        sig = inspect.signature(func)
+        params = list(sig.parameters.values())
+
+        if not params or params[0].name != "ctx":
+            raise ValueError(
+                f"Chatflow '{workflow_name}' must have 'ctx: WorkflowContext' as first parameter"
+            )
+
+        # Convert sync to async if needed
+        if inspect.iscoroutinefunction(func):
+            handler_func = cast(HandlerFunc, func)
+        else:
+            # Wrap sync function in async
+            @functools.wraps(func)
+            async def async_wrapper(*args: Any, **kwargs: Any) -> Any:
+                return func(*args, **kwargs)
+
+            handler_func = cast(HandlerFunc, async_wrapper)
+
+        # Extract schemas from type hints
+        input_schema, output_schema = extract_function_schemas(func)
+
+        # Extract metadata (description, etc.)
+        metadata = extract_function_metadata(func)
+
+        # Add chat metadata - THIS IS THE KEY DIFFERENCE FROM @workflow
+        metadata["chat"] = "true"
+
+        # Register as workflow (chatflows are workflows with chat metadata)
+        config = WorkflowConfig(
+            name=workflow_name,
+            handler=handler_func,
+            input_schema=input_schema,
+            output_schema=output_schema,
+            metadata=metadata,
+        )
+        WorkflowRegistry.register(config)
+
+        # Create wrapper that provides context
+        @functools.wraps(func)
+        async def wrapper(*args: Any, **kwargs: Any) -> Any:
+            # Create WorkflowEntity and WorkflowContext if not provided
+            if not args or not isinstance(args[0], WorkflowContext):
+                # Auto-create workflow entity and context for direct chatflow calls
+                run_id = f"chatflow-{uuid.uuid4().hex[:8]}"
+
+                # Create WorkflowEntity to manage state
+                workflow_entity = WorkflowEntity(run_id=run_id)
+
+                # Create WorkflowContext that wraps the entity
+                ctx = WorkflowContext(
+                    workflow_entity=workflow_entity,
+                    run_id=run_id,
+                )
+
+                # Execute chatflow
+                return await handler_func(ctx, *args, **kwargs)
+            else:
+                # WorkflowContext provided - use it
+                return await handler_func(*args, **kwargs)
+
+        # Store config on wrapper for introspection
+        wrapper._agnt5_config = config  # type: ignore
+        return wrapper
+
+    # Handle both @chatflow and @chatflow(...) syntax
+    if _func is None:
+        return decorator
+    else:
+        return decorator(_func)

{agnt5-0.2.4.dist-info → agnt5-0.2.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agnt5
-Version: 0.2.4
+Version: 0.2.5
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Programming Language :: Python :: 3

{agnt5-0.2.4.dist-info → agnt5-0.2.5.dist-info}/RECORD

@@ -1,22 +1,22 @@
-agnt5-0.2.4.dist-info/METADATA,sha256=0WFxoXI9jA26vzUKQp2Oqj8ZCb7H50v9PR7pxpXNXgo,994
-agnt5-0.2.4.dist-info/WHEEL,sha256=vpqC0tRn_8bTHidvtrPbrnFQPZnrhuKzsjDdeKwCd58,102
-agnt5/__init__.py,sha256=NAziyM0ZKahdzqAP2edFHTbVHYVdIjEGXm0oEHdmiuo,2011
+agnt5-0.2.5.dist-info/METADATA,sha256=zHVYaWjmmQbnDeJvr59rQNan70yGDUzjKg4cyPLfGr8,994
+agnt5-0.2.5.dist-info/WHEEL,sha256=vpqC0tRn_8bTHidvtrPbrnFQPZnrhuKzsjDdeKwCd58,102
+agnt5/__init__.py,sha256=XUixnW140lI-_9h1rLtN4jBWXIDL3EgnM7jmiNzGrYk,2037
 agnt5/_compat.py,sha256=BGuy3v5VDOHVa5f3Z-C22iMN19lAt0mPmXwF3qSSWxI,369
-agnt5/_core.abi3.so,sha256=_eS8cS8moUInKso8deBYKQBY56Mwt2gOUw3xCkvzMBs,11889296
+agnt5/_core.abi3.so,sha256=HGQaXAplREgIbUBvtNhmrmHyTkDfh3inERsj90LiSVM,11928992
 agnt5/_retry_utils.py,sha256=loHsWY5BR4wZy57IzcDEjQAy88DHVwVIr25Cn1d9GPA,5801
 agnt5/_schema_utils.py,sha256=MR67RW757T4Oq2Jqf4kB61H_b51zwaf3CLWELnkngRo,9572
 agnt5/_telemetry.py,sha256=bIY9AvBRjJBTHoBPbfR6X1OgaiUf-T0vCoi0_snsWXA,5957
-agnt5/agent.py,sha256=VQqNSd9o3tezxEo6jpWQyk399Wue7NgCpZvuQ-LtiTI,27494
+agnt5/agent.py,sha256=BAhYHKD5YuZXhNZaqeoN7EXKTtzF8OPhCJhpcWQS1YM,27837
 agnt5/client.py,sha256=kXksazgxdVXWaG9OkjJA4cWruNtcS-ENhtnkrIdw-Nk,23212
 agnt5/context.py,sha256=S2OzPkhn_jnqSWfT21mSYOux8vHaLKQxcAvggZDHQek,2378
 agnt5/entity.py,sha256=dhdxXUxED79u3OlX9yw-2TLCC9VqBcJqES2kx-fDChs,19041
 agnt5/exceptions.py,sha256=mZ0q-NK6OKhYxgwBJpIbgpgzk-CJaFIHDbp1EE-pS7I,925
 agnt5/function.py,sha256=f1vaAlJRwuo8cxCOGEd8XPido00mOhlPS8UJJx-6hJI,11041
-agnt5/lm.py,sha256=AQ8xz5u1mywBKIGwT3AJKtChVdxo5HOCvzdwZ4Ilbb0,21177
+agnt5/lm.py,sha256=1ufT0TGj_Ra1FXCflgxKfMh1qPQ_DV9p7BLlZGWI4m4,22062
 agnt5/tool.py,sha256=uc4L-Q9QyLzQDe-MZKk2Wo3o5e-mK8tfaQwVDgQdouQ,13133
 agnt5/tracing.py,sha256=Mh2-OfnQM61lM_P8gxJstafdsUA8Gxoo1lP-Joxhub8,5980
 agnt5/types.py,sha256=Zb71ZMwvrt1p4SH18cAKunp2y5tao_W5_jGYaPDejQo,2840
 agnt5/version.py,sha256=rOq1mObLihnnKgKqBrwZA0zwOPudEKVFcW1a48ynkqc,573
 agnt5/worker.py,sha256=_BnqqqvQE16FuezFbPda9FoiKAwcezSaxxNjU0rTjhs,33576
-agnt5/workflow.py,sha256=sve85o4N16PrlIwtQOqJPsHHWgl388SV702IPEXO7A8,18528
-agnt5-0.2.4.dist-info/RECORD,,
+agnt5/workflow.py,sha256=GGCyt34W1DTSL8zjcj54qI7SA6Fn3HyQZ1XcMTlCMdI,22748
+agnt5-0.2.5.dist-info/RECORD,,