Add modellabs gif to playground app

cookbook/agents/15_generate_video.py

@@ -7,7 +7,7 @@
     tools=[ModelsLabs()],
     description="You are an AI agent that can generate videos using the ModelsLabs API.",
     instructions=[
-        "When the user asks you to create a video, use the `create_video` tool to create the video.",
+        "When the user asks you to create a video, use the `generate_media` tool to create the video.",
         "The video will be displayed in the UI automatically below your response, so you don't need to show the video URL in your response.",
         "Politely and courteously let the user know that the video has been generated and will be displayed below as soon as its ready.",
     ],

cookbook/playground/multimodal_agent.py

@@ -10,6 +10,7 @@
 from phi.model.openai import OpenAIChat
 from phi.tools.dalle import Dalle
 from phi.tools.models_labs import ModelsLabs
+from phi.model.response import FileType
 from phi.playground import Playground, serve_playground_app
 from phi.storage.agent.sqlite import SqlAgentStorage
 from phi.tools.fal_tools import FalTools
@@ -34,22 +35,40 @@
     storage=SqlAgentStorage(table_name="image_agent", db_file="tmp/image_agent.db"),
 )
 
-video_agent = Agent(
+ml_gif_agent = Agent(
+    name="ModelsLab GIF Agent",
+    agent_id="ml_gif_agent",
+    model=OpenAIChat(id="gpt-4o"),
+    tools=[ModelsLabs(wait_for_completion=True, file_type=FileType.GIF)],
+    description="You are an AI agent that can generate gifs using the ModelsLabs API.",
+    instructions=[
+        "When the user asks you to create an image, use the `generate_media` tool to create the image.",
+        "The image will be displayed in the UI automatically below your response, so you don't need to show the image URL in your response.",
+        "Politely and courteously let the user know that the gif has been generated and will be displayed below as soon as its ready.",
+    ],
+    markdown=True,
+    debug_mode=True,
+    add_history_to_messages=True,
+    add_datetime_to_instructions=True,
+    storage=SqlAgentStorage(table_name="ml_gif_agent", db_file="tmp/ml_gif_agent.db"),
+)
+
+ml_video_agent = Agent(
     name="ModelsLab Video Agent",
-    agent_id="video_agent",
+    agent_id="ml_video_agent",
     model=OpenAIChat(id="gpt-4o"),
-    tools=[ModelsLabs(wait_for_completion=True)],
+    tools=[ModelsLabs(wait_for_completion=True, file_type=FileType.MP4)],
     description="You are an AI agent that can generate videos using the ModelsLabs API.",
     instructions=[
-        "When the user asks you to create a video, use the `create_video` tool to create the video.",
+        "When the user asks you to create a video, use the `generate_media` tool to create the video.",
         "The video will be displayed in the UI automatically below your response, so you don't need to show the video URL in your response.",
         "Politely and courteously let the user know that the video has been generated and will be displayed below as soon as its ready.",
     ],
     markdown=True,
     debug_mode=True,
     add_history_to_messages=True,
     add_datetime_to_instructions=True,
-    storage=SqlAgentStorage(table_name="video_agent", db_file="tmp/video_agent.db"),
+    storage=SqlAgentStorage(table_name="ml_video_agent", db_file="tmp/ml_video_agent.db"),
 )
 
 fal_agent = Agent(
@@ -71,7 +90,9 @@
     storage=SqlAgentStorage(table_name="fal_agent", db_file="tmp/fal_agent.db"),
 )
 
-app = Playground(agents=[image_agent, video_agent, fal_agent]).get_app()
+
+
+app = Playground(agents=[image_agent, ml_gif_agent, ml_video_agent, fal_agent]).get_app()
 
 if __name__ == "__main__":
     serve_playground_app("multimodal_agent:app", reload=True)

phi/model/response.py

@@ -23,3 +23,8 @@ class ModelResponse:
     tool_call: Optional[Dict[str, Any]] = None
     event: str = ModelResponseEvent.assistant_response.value
     created_at: int = int(time())
+
+
+class FileType(str, Enum):
+    MP4 = "mp4"
+    GIF = "gif"

phi/tools/models_labs.py

@@ -6,21 +6,16 @@
 
 from phi.agent import Agent
 from phi.model.content import Video, Image
+from phi.model.response import FileType
 from phi.tools import Toolkit
 from phi.utils.log import logger
-from enum import Enum
 
 try:
     import requests
 except ImportError:
     raise ImportError("`requests` not installed. Please install using `pip install requests`")
 
 
-class FileType(str, Enum):
-    MP4 = "mp4"
-    GIF = "gif"
-
-
 class ModelsLabs(Toolkit):
     def __init__(
         self,