flo-ai 1.0.0.dev5__tar.gz → 1.0.0.dev6__tar.gz

{flo_ai-1.0.0.dev5 → flo_ai-1.0.0.dev6}/PKG-INFO

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

{flo_ai-1.0.0.dev5 → flo_ai-1.0.0.dev6}/flo_ai/__init__.py

@@ -8,7 +8,7 @@ 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, ImageMessage
+from .llm import BaseLLM, Anthropic, OpenAI, OllamaLLM, Gemini, OpenAIVLLM, ImageMessage
 
 # Tool package - Tool framework components
 from .tool import Tool, ToolExecutionError, flo_tool, create_tool_from_function
@@ -43,6 +43,7 @@ __all__ = [
     'OpenAI',
     'OllamaLLM',
     'Gemini',
+    'OpenAIVLLM',
     # LLM DataClass
     'ImageMessage',
     # Tools

{flo_ai-1.0.0.dev5 → flo_ai-1.0.0.dev6}/flo_ai/llm/openai_llm.py

@@ -28,11 +28,11 @@ class OpenAI(BaseLLM):
             kwargs['response_format'] = {'type': 'json_object'}
             kwargs['functions'] = [
                 {
-                    'name': output_schema.get('name', 'default'),
+                    'name': output_schema.get('title', 'default'),
                     'parameters': output_schema.get('schema', output_schema),
                 }
             ]
-            kwargs['function_call'] = {'name': output_schema.get('name', 'default')}
+            kwargs['function_call'] = {'name': output_schema.get('title', 'default')}
 
             # Add JSON format instruction to the system prompt
             if messages and messages[0]['role'] == 'system':

{flo_ai-1.0.0.dev5 → flo_ai-1.0.0.dev6}/flo_ai/llm/openai_vllm.py

@@ -25,7 +25,13 @@ class OpenAIVLLM(OpenAI):
     ) -> Any:
         # Convert output_schema to OpenAI format if provided
         if output_schema:
-            kwargs['extra_body'] = {'guided_json': output_schema.get('schema')}
+            kwargs['response_format'] = {
+                'type': 'json_schema',
+                'json_schema': {
+                    'name': output_schema.get('title', 'default'),
+                    'schema': output_schema.get('schema', output_schema),
+                },
+            }
 
             # Add JSON format instruction to the system prompt
             if messages and messages[0]['role'] == 'system':

{flo_ai-1.0.0.dev5 → flo_ai-1.0.0.dev6}/pyproject.toml

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