add finetuned

Author: aboutphilippe

agent_video/pipecat/agent/src/agents/agent.py

@@ -19,12 +19,7 @@
         DailySendDataInput,
         daily_send_data,
     )
-    from src.functions.llm_chat import (
-        LlmChatInput,
-        Message,
-        llm_chat,
-    )
-    from src.functions.llm_talk import LlmTalkInput, llm_talk
+    from src.functions.llm_talk import LlmTalkInput, llm_talk, Message, ModelType
 
 
 class MessagesEvent(BaseModel):
@@ -34,10 +29,9 @@ class MessagesEvent(BaseModel):
 class EndEvent(BaseModel):
     end: bool
 
-
 class AgentInput(BaseModel):
     room_url: str
-    model: Literal["restack", "gpt-4o-mini", "gpt-4o", "openpipe:twenty-lions-fall"] = "restack"
+    model: ModelType
     interactive_prompt: str | None = None
     reasoning_prompt: str | None = None
 

agent_video/pipecat/agent/src/functions/llm_chat.py

@@ -5,7 +5,7 @@
 from pydantic import BaseModel
 from restack_ai.function import NonRetryableError, function
 
-from src.functions.llm_chat import Message
+from src.functions.llm_talk import Message
 
 
 class LlmLogicResponse(BaseModel):

agent_video/pipecat/agent/src/functions/llm_logic.py

@@ -10,19 +10,19 @@
 )
 
 from src.client import api_address
-from src.functions.llm_chat import Message
 
+class Message(BaseModel):
+    role: str
+    content: str
+
+ModelType = Literal["gpt-4o-mini", "ft:gpt-4o-mini-2024-07-18:restack::BJymdMm8", "openpipe:twenty-lions-fall"]
 
 class LlmTalkInput(BaseModel):
     messages: list[Message] = Field(default_factory=list)
     context: str | None = None  # Updated context from Slow AI
     mode: Literal["default", "interrupt"]
     stream: bool = True
-    model: Literal[
-        "gpt-4o-mini",
-        "ft:gpt-4o-mini-2024-07-18:restack::BJymdMm8",
-        "openpipe:twenty-lions-fall"
-    ]
+    model: ModelType = "gpt-4o-mini"
     interactive_prompt: str | None = None
 
 

agent_video/pipecat/agent/src/functions/llm_talk.py

@@ -11,7 +11,6 @@
 from src.functions.context_docs import context_docs
 from src.functions.daily_create_room import daily_create_room
 from src.functions.daily_send_data import daily_send_data
-from src.functions.llm_chat import llm_chat
 from src.functions.llm_logic import llm_logic
 from src.functions.llm_talk import llm_talk
 from src.functions.send_agent_event import send_agent_event
@@ -25,7 +24,6 @@ async def main() -> None:
         agents=[AgentVideo],
         workflows=[RoomWorkflow, LogicWorkflow],
         functions=[
-            llm_chat,
             llm_logic,
             llm_talk,
             context_docs,

agent_video/pipecat/agent/src/services.py

@@ -25,6 +25,8 @@
         LlmTalkInput,
         Message,
         llm_talk,
+        LlmTalkInput,
+        ModelType
     )
     from src.functions.send_agent_event import (
         SendAgentEventInput,
@@ -38,7 +40,7 @@ class LogicWorkflowInput(BaseModel):
     room_url: str
     interactive_prompt: str | None = None
     reasoning_prompt: str | None = None
-    model: Literal["gpt-4o-mini", "openpipe:twenty-lions-fall", "ft:gpt-4o-mini-2024-07-18:restack::BJymdMm8"] = "gpt-4o-mini"
+    model: ModelType
 
 
 class LogicWorkflowOutput(BaseModel):