flo-ai 1.0.0.dev1__tar.gz → 1.0.0.dev3__tar.gz

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

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

flo_ai-1.0.0.dev3/flo_ai/__init__.py

@@ -0,0 +1,66 @@
+"""
+flo_ai - A flexible agent framework for LLM-powered applications
+"""
+
+# Models package - Agent framework components
+from .models import Agent, AgentError, BaseAgent, AgentType, ReasoningPattern
+
+from .builder.agent_builder import AgentBuilder
+
+# LLM package - Language model integrations
+from .llm import BaseLLM, Anthropic, OpenAI, OllamaLLM, Gemini
+
+# Tool package - Tool framework components
+from .tool import Tool, ToolExecutionError, flo_tool, create_tool_from_function
+
+# Arium package - Workflow and memory components
+from .arium import (
+    Arium,
+    BaseArium,
+    create_arium,
+    MessageMemory,
+    BaseMemory,
+    StartNode,
+    EndNode,
+    Edge,
+)
+
+# Utils package - Utility functions
+from .utils import FloUtils
+
+__all__ = [
+    # Models
+    'Agent',
+    'AgentError',
+    'BaseAgent',
+    'AgentType',
+    'ReasoningPattern',
+    # Utils
+    'FloUtils',
+    # LLM
+    'BaseLLM',
+    'Anthropic',
+    'OpenAI',
+    'OllamaLLM',
+    'Gemini',
+    # Tools
+    'Tool',
+    'ToolExecutionError',
+    'flo_tool',
+    'create_tool_from_function',
+    # Arium
+    'Arium',
+    'BaseArium',
+    'AriumBuilder',
+    'create_arium',
+    'MessageMemory',
+    'BaseMemory',
+    'StartNode',
+    'EndNode',
+    'Edge',
+    # Builder
+    'AgentBuilder',
+    'AriumBuilder',
+]
+
+__version__ = '1.0.0'

{flo_ai-1.0.0.dev1 → flo_ai-1.0.0.dev3}/flo_ai/llm/base_llm.py

@@ -7,6 +7,7 @@ from dataclasses import dataclass
 @dataclass
 class ImageMessage:
     image_url: Optional[str] = None
+    image_bytes: Optional[bytes] = None
     image_file_path: Optional[str] = None
     image_base64: Optional[str] = None
     mime_type: Optional[str] = None

{flo_ai-1.0.0.dev1 → flo_ai-1.0.0.dev3}/flo_ai/llm/gemini_llm.py

@@ -12,6 +12,7 @@ class Gemini(BaseLLM):
         model: str = 'gemini-2.5-flash',
         temperature: float = 0.7,
         api_key: Optional[str] = None,
+        base_url: str = None,
         **kwargs,
     ):
         super().__init__(model, api_key, temperature, **kwargs)
@@ -153,6 +154,11 @@ class Gemini(BaseLLM):
                 data=image_bytes,
                 mime_type=image.mime_type,
             )
+        elif image.image_bytes:
+            return genai.types.Part.from_bytes(
+                data=image.image_bytes,
+                mime_type=image.mime_type,
+            )
         raise NotImplementedError(
             'Not other way other than file path has been implemented'
         )

{flo_ai-1.0.0.dev1 → flo_ai-1.0.0.dev3}/flo_ai/models/__init__.py

@@ -6,10 +6,4 @@ from .agent import Agent
 from .agent_error import AgentError
 from .base_agent import BaseAgent, AgentType, ReasoningPattern
 
-__all__ = [
-    "Agent",
-    "AgentError", 
-    "BaseAgent",
-    "AgentType",
-    "ReasoningPattern"
-]
+__all__ = ['Agent', 'AgentError', 'BaseAgent', 'AgentType', 'ReasoningPattern']

{flo_ai-1.0.0.dev1 → flo_ai-1.0.0.dev3}/flo_ai/utils/__init__.py

@@ -1,5 +1,3 @@
 from .flo_utils import FloUtils
 
-__all__ = [
-    'FloUtils'
-]
+__all__ = ['FloUtils']

{flo_ai-1.0.0.dev1 → flo_ai-1.0.0.dev3}/flo_ai/utils/flo_utils.py

@@ -1,11 +1,9 @@
 from typing import Dict, Any
-import re
 import json
 from flo_ai.utils.logger import logger
 
 
 class FloUtils:
-
     @staticmethod
     def extract_jsons_from_string(data: str, strict: bool = False) -> Dict[str, Any]:
         """
@@ -14,6 +12,7 @@ class FloUtils:
         3) Merge into one dict (later keys override earlier)
         4) On strict mode, raise FloException if no JSON found
         """
+
         # Custom function to find balanced braces since (?R) is not supported in Python re
         def find_balanced_braces(text):
             matches = []
@@ -25,10 +24,10 @@ class FloUtils:
                     j = i + 1
                     in_string = False
                     escape_next = False
-                    
+
                     while j < len(text) and brace_count > 0:
                         char = text[j]
-                        
+
                         if escape_next:
                             escape_next = False
                         elif char == '\\' and in_string:
@@ -40,9 +39,9 @@ class FloUtils:
                                 brace_count += 1
                             elif char == '}':
                                 brace_count -= 1
-                        
+
                         j += 1
-                    
+
                     if brace_count == 0:
                         candidate = text[start:j]
                         # Try to validate it's actually JSON by attempting to parse
@@ -77,7 +76,7 @@ class FloUtils:
             raise ValueError(f'No JSON found in strict mode: {data}')
 
         return merged
-    
+
     @staticmethod
     def strip_comments_from_string(data: str) -> str:
         """Remove JS-style comments (// and /*…*/) so json.loads() will succeed."""

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

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

flo_ai-1.0.0.dev1/flo_ai/__init__.py

@@ -1,92 +0,0 @@
-"""
-flo_ai - A flexible agent framework for LLM-powered applications
-"""
-
-# Models package - Agent framework components
-from .models import (
-    Agent,
-    AgentError,
-    BaseAgent,
-    AgentType,
-    ReasoningPattern
-)
-
-from .builder.agent_builder import (
-    AgentBuilder
-)
-
-# LLM package - Language model integrations
-from .llm import (
-    BaseLLM,
-    Anthropic,
-    OpenAI,
-    OllamaLLM,
-    Gemini
-)
-
-# Tool package - Tool framework components
-from .tool import (
-    Tool,
-    ToolExecutionError,
-    flo_tool,
-    create_tool_from_function
-)
-
-# Arium package - Workflow and memory components
-from .arium import (
-    Arium,
-    BaseArium,
-    create_arium,
-    MessageMemory,
-    BaseMemory,
-    StartNode,
-    EndNode,
-    Edge
-)
-
-# Utils package - Utility functions
-from .utils import (
-    FloUtils
-)
-
-__all__ = [
-    # Models
-    "Agent",
-    "AgentError",
-    "BaseAgent", 
-    "AgentType",
-    "ReasoningPattern",
-    
-    # Utils
-    "FloUtils",
-    
-    # LLM
-    "BaseLLM",
-    "Anthropic",
-    "OpenAI",
-    "OllamaLLM",
-    "Gemini",
-    
-    # Tools
-    "Tool",
-    "ToolExecutionError",
-    "flo_tool",
-    "create_tool_from_function",
-    
-    # Arium
-    "Arium",
-    "BaseArium",
-    "AriumBuilder",
-    "create_arium",
-    "MessageMemory",
-    "BaseMemory",
-    "StartNode",
-    "EndNode",
-    "Edge"
-
-    # Builder
-    "AgentBuilder",
-    "AriumBuilder"
-]
-
-__version__ = "1.0.0"