Skip to content

Commit

Permalink
VideoSurferAgent -> VideoSurfer (#4521)
Browse files Browse the repository at this point in the history
  • Loading branch information
gagb authored Dec 4, 2024
1 parent 3022369 commit abb0a77
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 12 deletions.
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
from ._video_surfer import VideoSurferAgent
from ._video_surfer import VideoSurfer

__all__ = ["VideoSurferAgent"]
__all__ = ["VideoSurfer"]
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,9 @@
)


class VideoSurferAgent(AssistantAgent):
class VideoSurfer(AssistantAgent):
"""
VideoSurferAgent is a specialized agent designed to answer questions about a local video file.
VideoSurfer is a specialized agent designed to answer questions about a local video file.
This agent utilizes various tools to extract information from the video, such as its length, screenshots at specific timestamps, and audio transcriptions. It processes these elements to provide detailed answers to user queries.
Expand All @@ -43,15 +43,15 @@ class VideoSurferAgent(AssistantAgent):
from autogen_agentchat.conditions import TextMentionTermination
from autogen_agentchat.teams import RoundRobinGroupChat
from autogen_ext.models import OpenAIChatCompletionClient
from autogen_ext.agents.video_surfer import VideoSurferAgent
from autogen_ext.agents.video_surfer import VideoSurfer
async def main() -> None:
\"\"\"
Main function to run the video agent.
\"\"\"
# Define an agent
video_agent = VideoSurferAgent(
name="VideoSurferAgent",
video_agent = VideoSurfer(
name="VideoSurfer",
model_client=OpenAIChatCompletionClient(model="gpt-4o-2024-08-06")
)
Expand All @@ -67,7 +67,7 @@ async def main() -> None:
asyncio.run(main())
The following example demonstrates how to create and use a VideoSurferAgent and UserProxyAgent with MagenticOneGroupChat.
The following example demonstrates how to create and use a VideoSurfer and UserProxyAgent with MagenticOneGroupChat.
.. code-block:: python
Expand All @@ -77,7 +77,7 @@ async def main() -> None:
from autogen_agentchat.teams import MagenticOneGroupChat
from autogen_agentchat.agents import UserProxyAgent
from autogen_ext.models import OpenAIChatCompletionClient
from autogen_ext.agents.video_surfer import VideoSurferAgent
from autogen_ext.agents.video_surfer import VideoSurfer
async def main() -> None:
\"\"\"
Expand All @@ -87,8 +87,8 @@ async def main() -> None:
model_client = OpenAIChatCompletionClient(model="gpt-4o-2024-08-06")
# Define an agent
video_agent = VideoSurferAgent(
name="VideoSurferAgent",
video_agent = VideoSurfer(
name="VideoSurfer",
model_client=model_client
)
Expand Down Expand Up @@ -128,7 +128,7 @@ def __init__(
system_message: Optional[str] = None,
):
"""
Initialize the VideoSurferAgent.
Initialize the VideoSurfer.
Args:
name (str): The name of the agent.
Expand Down

0 comments on commit abb0a77

Please sign in to comment.