diff --git a/flo_ai/flo_ai/__init__.py b/flo_ai/flo_ai/__init__.py index 307561dd..80e5a364 100644 --- a/flo_ai/flo_ai/__init__.py +++ b/flo_ai/flo_ai/__init__.py @@ -8,7 +8,7 @@ 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 @@ 'OpenAI', 'OllamaLLM', 'Gemini', + 'OpenAIVLLM', # LLM DataClass 'ImageMessage', # Tools