flo-ai 0.0.5.dev5__tar.gz → 0.0.5.dev7__tar.gz

{flo_ai-0.0.5.dev5 → flo_ai-0.0.5.dev7}/PKG-INFO

@@ -1,18 +1,18 @@
 Metadata-Version: 2.1
 Name: flo-ai
-Version: 0.0.5.dev5
+Version: 0.0.5.dev7
 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.dev5 → flo_ai-0.0.5.dev7}/flo_ai/factory/agent_factory.py

@@ -33,15 +33,16 @@ class AgentFactory:
             if agent_kind is None:
                 raise FloException(f"""Unknown agent kind: `{kind}`. The supported types are llm, tool, reflection, delegator or agentic. 
                             Check the documentation @ {DOCUMENTATION_AGENT_ANCHOR}""")
-            match agent_kind:
-                case AgentKinds.llm:
-                    return AgentFactory.__create_llm_agent(session, agent)
-                case AgentKinds.tool:
-                    return AgentFactory.__create_runnable_agent(session, agent)
-                case AgentKinds.reflection:
-                    return AgentFactory.__create_reflection_agent(session, agent)
-                case AgentKinds.delegator:
-                    return AgentFactory.__create_delegator_agent(session, agent)
+            if agent_kind == AgentKinds.llm:
+                return AgentFactory.__create_llm_agent(session, agent)
+            elif agent_kind == AgentKinds.tool:
+                return AgentFactory.__create_runnable_agent(session, agent)
+            elif agent_kind == AgentKinds.reflection:
+                return AgentFactory.__create_reflection_agent(session, agent)
+            elif agent_kind == AgentKinds.delegator:
+                return AgentFactory.__create_delegator_agent(session, agent)
+            else:
+                raise ValueError(f'Unsupported agent kind: {agent_kind}')
         return AgentFactory.__create_agentic_agent(session, agent, tool_map)
 
     @staticmethod

{flo_ai-0.0.5.dev5 → flo_ai-0.0.5.dev7}/flo_ai/models/flo_executable.py

@@ -16,14 +16,7 @@ class ExecutableType(Enum):
 
     @staticmethod
     def isAgent(type: 'ExecutableType'):
-        match type:
-            case ExecutableType.agentic:
-                return True
-            case ExecutableType.llm:
-                return True
-            case ExecutableType.tool:
-                return True
-        return False
+        return type in [ExecutableType.agentic, ExecutableType.llm, ExecutableType.tool]
 
 
 class ExecutableFlo(FloMember):

{flo_ai-0.0.5.dev5 → flo_ai-0.0.5.dev7}/flo_ai/router/flo_router.py

@@ -78,7 +78,7 @@ class FloRouter(ABC):
         workflow: StateGraph,
         parent: FloNode,
         delegation_node: FloNode,
-        nextNode: Union[FloNode | str],
+        nextNode: Union[FloNode, str],
     ):
         to_agent_names = delegation_node.delegate.to
         delegation_node_name = delegation_node.name
@@ -129,7 +129,7 @@ class FloRouter(ABC):
         self,
         workflow: StateGraph,
         reflection_node: FloNode,
-        nextNode: Union[FloNode | str],
+        nextNode: Union[FloNode, str],
     ):
         to_agent_name = reflection_node.delegate.to[0]
         retry = reflection_node.delegate.retry or 1

{flo_ai-0.0.5.dev5 → flo_ai-0.0.5.dev7}/flo_ai/yaml/config.py

@@ -34,7 +34,7 @@ class PromptStrategy(BaseModel):
     kind: str
     agent_name: str
     retries: int
-    next: str | None = None
+    next: Union[str, None] = None
 
 
 class MemberKey(BaseModel):

{flo_ai-0.0.5.dev5 → flo_ai-0.0.5.dev7}/pyproject.toml

@@ -1,16 +1,15 @@
 [tool.poetry]
 name = "flo-ai"
-version = "0.0.5-dev5"
+version = "0.0.5-dev7"
 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"
@@ -19,7 +18,7 @@ langchain-community = "0.3.2"
 
 [tool.poetry.group.dev.dependencies]
 langchain-mongodb = "^0.2.0"
-langchain-chroma = "^0.1.4"
+# langchain-chroma = "^0.1.4"
 langchainhub = "^0.1.17"
 pypdf = "^4.2.0"
 ipykernel = "^6.29.5"