flo_ai 0.0.6.dev1__tar.gz → 0.0.6.dev3__tar.gz

{flo_ai-0.0.6.dev1 → flo_ai-0.0.6.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flo_ai
-Version: 0.0.6.dev1
+Version: 0.0.6.dev3
 Summary: A easy way to create structured AI agents
 License: MIT
 Author: rootflo

{flo_ai-0.0.6.dev1 → flo_ai-0.0.6.dev3}/flo_ai/models/flo_node.py

@@ -36,7 +36,7 @@ class FloNode:
         self.async_func = async_func
         self.agent_executable = agent_executable
 
-    def invoke(self, query, config):
+    def invoke(self, query, config, chat_history = []):
         return self.func(
             {STATE_NAME_MESSAGES: [HumanMessage(content=query)]}, config=config
         )

{flo_ai-0.0.6.dev1 → flo_ai-0.0.6.dev3}/flo_ai/parsers/flo_json_parser.py

@@ -96,6 +96,9 @@ class FloJsonParser(FloParser):
         literals = [literal_value['value'] for literal_value in literal_values]
         return Literal[tuple(literals)]
 
+    def get_format(self) -> BaseModel:
+        return self.__create_contract_from_json()
+
     def __create_contract_from_json(self) -> BaseModel:
         pydantic_fields = {}
         for field in self.contract.fields:

{flo_ai-0.0.6.dev1 → flo_ai-0.0.6.dev3}/flo_ai/parsers/flo_parser.py

@@ -1,7 +1,12 @@
 from abc import ABC, abstractmethod
+from pydantic import BaseModel
 
 
 class FloParser(ABC):
     @abstractmethod
     def get_format_instructions(self):
         pass
+
+    @abstractmethod
+    def get_format(self) -> BaseModel:
+        pass

{flo_ai-0.0.6.dev1 → flo_ai-0.0.6.dev3}/flo_ai/parsers/flo_pydantic_parser.py

@@ -13,6 +13,9 @@ class FloPydanticParser(FloParser):
             pydantic_object=self.model
         ).get_format_instructions()
 
+    def get_format(self):
+        return self.model
+
     def create(output_model: BaseModel):
         return FloPydanticParser.Builder(output_model).build()
 

{flo_ai-0.0.6.dev1 → flo_ai-0.0.6.dev3}/flo_ai/yaml/config.py

@@ -30,13 +30,6 @@ class ToolConfig(BaseModel):
     filters: Optional[List[FilterArgs]] = None
 
 
-class PromptStrategy(BaseModel):
-    kind: str
-    agent_name: str
-    retries: int
-    next: Union[str, None] = None
-
-
 class MemberKey(BaseModel):
     name: str
 

{flo_ai-0.0.6.dev1 → flo_ai-0.0.6.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "flo_ai"
-version = "0.0.6-dev1"
+version = "0.0.6-dev3"
 description = "A easy way to create structured AI agents"
 authors = ["rootflo <*@rootflo.ai>"]
 license = "MIT"