flo-ai 0.0.5.dev4__tar.gz → 0.0.5.dev6__tar.gz

{flo_ai-0.0.5.dev4 → flo_ai-0.0.5.dev6}/PKG-INFO

@@ -1,18 +1,18 @@
 Metadata-Version: 2.1
 Name: flo-ai
-Version: 0.0.5.dev4
+Version: 0.0.5.dev6
 Summary: A easy way to create structured AI agents
 License: MIT
 Author: vizsatiz
 Author-email: vishnu@rootfor.xyz
-Requires-Python: >=3.10,<4.0
+Requires-Python: >=3.9,<4.0
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: httpx (==0.27.0)
-Requires-Dist: ipython (==8.24.0)
 Requires-Dist: langchain (==0.3.3)
 Requires-Dist: langchain-community (==0.3.2)
 Requires-Dist: langgraph (==0.2.38)

{flo_ai-0.0.5.dev4 → flo_ai-0.0.5.dev6}/flo_ai/models/flo_node.py

@@ -25,15 +25,22 @@ class FloNode:
         name: str,
         kind: ExecutableType,
         delegate: Optional[Delegate] = None,
+        async_func: functools.partial = None,
     ) -> None:
         self.name = name
         self.func = func
         self.kind: ExecutableType = kind
         self.delegate = delegate
+        self.async_func = async_func
 
     def invoke(self, query, config):
         return self.func({STATE_NAME_MESSAGES: [HumanMessage(content=query)]})
 
+    async def ainvoke(self, query, config):
+        return await self.async_func(
+            {STATE_NAME_MESSAGES: [HumanMessage(content=query)]}
+        )
+
     class Builder:
         def __init__(self, session: FloSession) -> None:
             self.session = session
@@ -47,7 +54,17 @@ class FloNode:
                 model_name=flo_agent.model_name,
                 data_collector=flo_agent.data_collector,
             )
-            return FloNode(agent_func, flo_agent.name, flo_agent.type)
+            agent_func_async = functools.partial(
+                FloNode.Builder.__async_teamflo_agent_node,
+                agent=flo_agent.runnable,
+                name=flo_agent.name,
+                session=self.session,
+                model_name=flo_agent.model_name,
+                data_collector=flo_agent.data_collector,
+            )
+            return FloNode(
+                agent_func, flo_agent.name, flo_agent.type, async_func=agent_func_async
+            )
 
         def build_from_reflection(self, flo_agent: FloReflectionAgent) -> 'FloNode':
             agent_func = functools.partial(
@@ -154,6 +171,57 @@ class FloNode:
             ]
             return {STATE_NAME_MESSAGES: [AIMessage(content=output, name=name)]}
 
+        @staticmethod
+        async def __async_teamflo_agent_node(
+            state: TeamFloAgentState,
+            agent: AgentExecutor,
+            name: str,
+            session: FloSession,
+            model_name: str,
+            data_collector: Optional[FloDataCollector] = None,
+        ):
+            agent_cbs: List[FloAgentCallback] = FloNode.Builder.__filter_callbacks(
+                session, FloAgentCallback
+            )
+            flo_cbs: List[FloCallback] = FloNode.Builder.__filter_callbacks(
+                session, FloCallback
+            )
+            [
+                callback.on_agent_start(name, model_name, state['messages'], **{})
+                for callback in agent_cbs
+            ]
+            [
+                callback.on_agent_start(name, model_name, state['messages'], **{})
+                for callback in flo_cbs
+            ]
+            try:
+                result = await agent.ainvoke(state)
+                output = result if isinstance(result, str) else result['output']
+                if data_collector is not None:
+                    get_logger().info(
+                        'appending output to data collector', session=session
+                    )
+                    data_collector.append(output)
+            except Exception as e:
+                [
+                    callback.on_agent_error(name, model_name, e, **{})
+                    for callback in agent_cbs
+                ]
+                [
+                    callback.on_agent_error(name, model_name, e, **{})
+                    for callback in flo_cbs
+                ]
+                raise e
+            [
+                callback.on_agent_end(name, model_name, output, **{})
+                for callback in agent_cbs
+            ]
+            [
+                callback.on_agent_start(name, model_name, output, **{})
+                for callback in flo_cbs
+            ]
+            return {STATE_NAME_MESSAGES: [AIMessage(content=output, name=name)]}
+
         @staticmethod
         def __filter_callbacks(session: FloSession, type: Type):
             cbs = session.callbacks

{flo_ai-0.0.5.dev4 → flo_ai-0.0.5.dev6}/pyproject.toml

@@ -1,16 +1,15 @@
 [tool.poetry]
 name = "flo-ai"
-version = "0.0.5-dev4"
+version = "0.0.5-dev6"
 description = "A easy way to create structured AI agents"
 authors = ["vizsatiz <vishnu@rootfor.xyz>"]
 license = "MIT"
 readme = "README.md"
 
 [tool.poetry.dependencies]
-python = ">=3.10,<4.0"
+python = ">=3.9,<4.0"
 langchain = "0.3.3"
 langgraph = "0.2.38"
-ipython = "8.24.0"
 httpx = "0.27.0"
 pillow = "^10.3.0"
 pydantic = "^2.9.2"