flo_ai 1.0.0.dev1__tar.gz → 1.0.0.dev2__tar.gz

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

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

{flo_ai-1.0.0.dev1 → flo_ai-1.0.0.dev2}/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.dev2}/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.dev2}/pyproject.toml

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