From a623762b9df952b46c415a4adedfcf06e7591e69 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=A0=D0=B8=D0=BD=D0=B0=D1=80=20=D0=9C=D0=B8=D1=80=D0=B7?= =?UTF-8?q?=D0=B0=D0=B3=D0=B8=D1=82=D0=BE=D0=B2?= Date: Thu, 28 May 2026 12:49:05 +0000 Subject: [PATCH] Remove extra stream_agent.py --- stream_agent.py | 26 -------------------------- 1 file changed, 26 deletions(-) delete mode 100644 stream_agent.py diff --git a/stream_agent.py b/stream_agent.py deleted file mode 100644 index 303b471..0000000 --- a/stream_agent.py +++ /dev/null @@ -1,26 +0,0 @@ -import sys -from typing import Iterable -from langchain_core.messages import HumanMessage, AIMessage -from langchain_openai import ChatOpenAI -from rich.console import Console - -console = Console() - -def stream_response(messages: list[HumanMessage | AIMessage]) -> Iterable[str]: - """Yield chunks of the LLM response using streaming.""" - llm = ChatOpenAI(streaming=True, model="gpt-4o-mini") - chat_history = [msg.model_dump() for msg in messages] - stream = llm.invoke(chat_history) - # The stream yields dicts with 'content' key - for chunk in stream: - # Each chunk is a dict with 'content' - content = chunk["content"] if isinstance(chunk, dict) else None - if content: - yield content - console.print(content, end="") - -if __name__ == "__main__": - user_input = sys.argv[1] if len(sys.argv) > 1 else "Hello" - messages: list[HumanMessage | AIMessage] = [HumanMessage(content=user_input)] - for _ in stream_response(messages): - pass