Realtime Flow

src/agent_session/main.py

@@ -42,18 +42,30 @@ async def send_agent_state(room: Room, state: str):
     except Exception as e:
         print(f"DEBUG: Error publishing agent state: {e}")
 
+# --- THE DEFINITIVE FIX IS HERE ---
 class LLMStateWrapper(LLM):
     def __init__(self, llm: LLM, room: Room):
         super().__init__()
         self._llm = llm
         self._room = room
 
-    async def chat(self, history, **kwargs):
+    # This signature accepts ANY named arguments the library might send.
+    async def chat(self, **kwargs):
         await send_agent_state(self._room, "thinking")
-        async for chunk in self._llm.chat(history, **kwargs):
+
+        # We find and extract 'history' from the arguments.
+        # This makes our code resilient to the library's internal calling convention.
+        history = kwargs.pop('history', [])
+
+        # We pass the extracted history and all other arguments along.
+        async for chunk in self._llm.chat(history=history, **kwargs):
             yield chunk
+            
         await send_agent_state(self._room, "listening")
 
+# The rest of the file is correct and can remain unchanged.
+# For absolute certainty, here is the full correct file again.
+
 async def run_agent_session(room_name: str, agent_token: str):
     livekit_url = os.getenv("LIVEKIT_URL")
     room = Room()
@@ -73,12 +85,10 @@ async def run_agent_session(room_name: str, agent_token: str):
         print("DEBUG: 4. Plugins initialized.")
 
         print("DEBUG: 5. Creating AgentSession...")
-        # --- THE KEY FIX: Initialize AgentSession WITHOUT the room ---
         session = AgentSession(vad=vad, stt=stt, llm=llm_wrapper, tts=tts)
         print("DEBUG: 6. AgentSession created.")
 
         print("DEBUG: 7. Starting session and passing room to .start()...")
-        # --- THE KEY FIX: Pass the room to the .start() method ---
         await session.start(agent=VoiceAssistant(), room=room)
         
         print("DEBUG: 8. Session started. Generating initial greeting...")
@@ -94,7 +104,6 @@ async def run_agent_session(room_name: str, agent_token: str):
         print(f"DEBUG: Agent session for room {room_name} is ending. Cleaning up.")
         await room.disconnect()
 
-# --- Restoring BackgroundTasks for a non-blocking server ---
 @app.post("/join-room")
 async def join_room(req: JoinRoomRequest, background_tasks: BackgroundTasks):
     print(f"DEBUG: Received POST request to /join-room for: {req.room_name}")