agnt5 0.2.2__cp39-abi3-macosx_11_0_arm64.whl → 0.2.3__cp39-abi3-macosx_11_0_arm64.whl

agnt5/lm.py

@@ -277,8 +277,9 @@ class _LanguageModel:
         # if request.tool_choice:
         #     kwargs["tool_choice"] = request.tool_choice.value
 
-        # Call Rust implementation
-        rust_response = self._rust_lm.generate(prompt=prompt, **kwargs)
+        # Call Rust implementation - it returns a proper Python coroutine now
+        # Using pyo3-async-runtimes for truly async HTTP calls without blocking
+        rust_response = await self._rust_lm.generate(prompt=prompt, **kwargs)
 
         # Convert Rust response to Python
         return self._convert_response(rust_response)
@@ -326,8 +327,9 @@ class _LanguageModel:
         # if request.tool_choice:
         #     kwargs["tool_choice"] = request.tool_choice.value
 
-        # Call Rust implementation (it returns a list of chunks)
-        rust_chunks = self._rust_lm.stream(prompt=prompt, **kwargs)
+        # Call Rust implementation - it returns a proper Python coroutine now
+        # Using pyo3-async-runtimes for truly async streaming without blocking
+        rust_chunks = await self._rust_lm.stream(prompt=prompt, **kwargs)
 
         # Yield each chunk
         for chunk in rust_chunks:

agnt5/worker.py

@@ -305,67 +305,66 @@ class Worker:
         """Create the message handler that will be called by Rust worker."""
 
         def handle_message(request):
-            """Handle incoming execution requests."""
-            try:
-                # Extract request details
-                component_name = request.component_name
-                component_type = request.component_type
-                input_data = request.input_data
-
-                logger.debug(
-                    f"Handling {component_type} request: {component_name}, input size: {len(input_data)} bytes"
-                )
+            """Handle incoming execution requests - returns coroutine for Rust to await."""
+            # Extract request details
+            component_name = request.component_name
+            component_type = request.component_type
+            input_data = request.input_data
+
+            logger.debug(
+                f"Handling {component_type} request: {component_name}, input size: {len(input_data)} bytes"
+            )
 
-                # Import all registries
-                from .tool import ToolRegistry
-                from .entity import EntityRegistry
-                from .agent import AgentRegistry
-
-                # Route based on component type
-                if component_type == "tool":
-                    tool = ToolRegistry.get(component_name)
-                    if tool:
-                        logger.debug(f"Found tool: {component_name}")
-                        result = asyncio.run(self._execute_tool(tool, input_data, request))
-                        return result
-
-                elif component_type == "entity":
-                    entity_type = EntityRegistry.get(component_name)
-                    if entity_type:
-                        logger.debug(f"Found entity: {component_name}")
-                        result = asyncio.run(self._execute_entity(entity_type, input_data, request))
-                        return result
-
-                elif component_type == "agent":
-                    agent = AgentRegistry.get(component_name)
-                    if agent:
-                        logger.debug(f"Found agent: {component_name}")
-                        result = asyncio.run(self._execute_agent(agent, input_data, request))
-                        return result
-
-                elif component_type == "workflow":
-                    workflow_config = WorkflowRegistry.get(component_name)
-                    if workflow_config:
-                        logger.debug(f"Found workflow: {component_name}")
-                        result = asyncio.run(self._execute_workflow(workflow_config, input_data, request))
-                        return result
-
-                elif component_type == "function":
-                    function_config = FunctionRegistry.get(component_name)
-                    if function_config:
-                        logger.info(f"🔥 WORKER: Received request for function: {component_name}")
-                        result = asyncio.run(self._execute_function(function_config, input_data, request))
-                        return result
-
-                # Not found
-                error_msg = f"Component '{component_name}' of type '{component_type}' not found"
-                logger.error(error_msg)
+            # Import all registries
+            from .tool import ToolRegistry
+            from .entity import EntityRegistry
+            from .agent import AgentRegistry
+
+            # Route based on component type and return coroutines
+            if component_type == "tool":
+                tool = ToolRegistry.get(component_name)
+                if tool:
+                    logger.debug(f"Found tool: {component_name}")
+                    # Return coroutine, don't await it
+                    return self._execute_tool(tool, input_data, request)
+
+            elif component_type == "entity":
+                entity_type = EntityRegistry.get(component_name)
+                if entity_type:
+                    logger.debug(f"Found entity: {component_name}")
+                    # Return coroutine, don't await it
+                    return self._execute_entity(entity_type, input_data, request)
+
+            elif component_type == "agent":
+                agent = AgentRegistry.get(component_name)
+                if agent:
+                    logger.debug(f"Found agent: {component_name}")
+                    # Return coroutine, don't await it
+                    return self._execute_agent(agent, input_data, request)
+
+            elif component_type == "workflow":
+                workflow_config = WorkflowRegistry.get(component_name)
+                if workflow_config:
+                    logger.debug(f"Found workflow: {component_name}")
+                    # Return coroutine, don't await it
+                    return self._execute_workflow(workflow_config, input_data, request)
+
+            elif component_type == "function":
+                function_config = FunctionRegistry.get(component_name)
+                if function_config:
+                    logger.info(f"🔥 WORKER: Received request for function: {component_name}")
+                    # Return coroutine, don't await it
+                    return self._execute_function(function_config, input_data, request)
+
+            # Not found - need to return an async error response
+            error_msg = f"Component '{component_name}' of type '{component_type}' not found"
+            logger.error(error_msg)
+
+            # Create async wrapper for error response
+            async def error_response():
                 return self._create_error_response(request, error_msg)
 
-            except Exception as e:
-                error_msg = f"Handler error: {e}"
-                logger.error(error_msg, exc_info=True)
-                return self._create_error_response(request, error_msg)
+            return error_response()
 
         return handle_message
 

{agnt5-0.2.2.dist-info → agnt5-0.2.3.dist-info}/METADATA

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

{agnt5-0.2.2.dist-info → agnt5-0.2.3.dist-info}/RECORD

@@ -1,8 +1,8 @@
-agnt5-0.2.2.dist-info/METADATA,sha256=lGJESH9_HBx_FmDngvJl_cDBCF02SfJsL_mJlWusN1E,965
-agnt5-0.2.2.dist-info/WHEEL,sha256=vpqC0tRn_8bTHidvtrPbrnFQPZnrhuKzsjDdeKwCd58,102
+agnt5-0.2.3.dist-info/METADATA,sha256=vqyABE681kP5zw7uwyqo_LLKPtgv7-KLBP-IcDW5xD4,965
+agnt5-0.2.3.dist-info/WHEEL,sha256=vpqC0tRn_8bTHidvtrPbrnFQPZnrhuKzsjDdeKwCd58,102
 agnt5/__init__.py,sha256=dQ83SuDrAU83leQ38mDVdthZFP0SWJMK67xq1d4uGtg,1933
 agnt5/_compat.py,sha256=BGuy3v5VDOHVa5f3Z-C22iMN19lAt0mPmXwF3qSSWxI,369
-agnt5/_core.abi3.so,sha256=cWTAlgAIyMzQ4Gmp-Cq-5lnlZ1BTYg4IEWhgxv3WvjA,11895200
+agnt5/_core.abi3.so,sha256=xzAeLfCVjH22YDGMB10y_u_C_HGUxyf5pxw-2TJHEIo,12011968
 agnt5/_telemetry.py,sha256=ZeABlTMoEnXYnBFVOn6ruaSvfeS7r_czI_Pk_tiCfhY,4899
 agnt5/agent.py,sha256=sU0qSZyXl_D6YrQKxtlyR6gggPJ0zZKC63bWtJvMzJI,26557
 agnt5/client.py,sha256=Zesl-TIie8Gwq-ajTauPlAsF7jZm5TPk307iHdxQ2SM,21214
@@ -10,10 +10,10 @@ agnt5/context.py,sha256=fF_eL_rY-GFnGFpL6wqPERykluGf2obpD-mYrGpsUTc,23817
 agnt5/entity.py,sha256=kLMBlHSQJeRqNwfWcXeLcM5UF9JcU8aET6TQBh6tSBA,43966
 agnt5/exceptions.py,sha256=mZ0q-NK6OKhYxgwBJpIbgpgzk-CJaFIHDbp1EE-pS7I,925
 agnt5/function.py,sha256=hH7rH9c_ZohUuNoLlBJHuiR5qPhw0L0pkrVVrpL_g2M,11273
-agnt5/lm.py,sha256=VvKVm_7VneBggLoCZEXfTavpPJJXNpbARrgHPCqhrX8,18915
+agnt5/lm.py,sha256=ryEYWAD9KOszN9EjUX33_d2RGVqIYXn5xwqo4f4DRxg,19142
 agnt5/tool.py,sha256=5s0YK5-BFIlQNUJOQHbejQsWW17r43rAC4zRvJnh9qg,12513
 agnt5/types.py,sha256=Zb71ZMwvrt1p4SH18cAKunp2y5tao_W5_jGYaPDejQo,2840
 agnt5/version.py,sha256=rOq1mObLihnnKgKqBrwZA0zwOPudEKVFcW1a48ynkqc,573
-agnt5/worker.py,sha256=n4zL4DUocRlcpitsOPbrZ4frz-E_ZoVgIhKH_76iq00,31502
+agnt5/worker.py,sha256=fCgJdT2gUrTctr1d3FvU8v1eQIyZ_jhm2xRatI7Qga0,31356
 agnt5/workflow.py,sha256=1sFAvtdZn6n7nDDb0YyE6LGJGuN2KVFNVo9Q42N_7_k,4529
-agnt5-0.2.2.dist-info/RECORD,,
+agnt5-0.2.3.dist-info/RECORD,,