Clean repository: delete agent.py

2026-05-26 10:35:59 +00:00
parent d74ef05c9e
commit ca14f3e654
1 changed files with 0 additions and 97 deletions
@@ -1,97 +0,0 @@
-from langchain.agents import create_agent
-from langchain.tools import tool
-from langchain_openai import ChatOpenAI
-from langgraph.checkpoint.memory import MemorySaver
-from pydantic import SecretStr
-
-
-llm = ChatOpenAI(
-    model="local-model",
-    base_url="http://localhost:1234/v1",
-    api_key=SecretStr("fake"),
-    temperature=0.7,
-)
-
-
-@tool
-def get_price(product: str, city: str) -> str:
-    """Возвращает примерную цену продукта в указанном городе."""
-
-    price_agent = create_agent(
-        model=llm,
-        tools=[],
-        system_prompt=(
-            "Ты субагент оценки цен. Верни одну markdown-таблицу "
-            "с колонками: Продукт, Цена (руб.), Магазин."
-        ),
-    )
-    answer = price_agent.invoke(
-        {
-            "messages": [
-                {
-                    "role": "human",
-                    "content": (
-                        f"Оцени реалистичную цену продукта '{product}' "
-                        f"в городе {city}, опираясь на исторические данные о ценах."
-                    ),
-                }
-            ]
-        }
-    )
-    return answer["messages"][-1].content
-
-
-def format_message(message) -> str:
-    if message.content:
-        return message.content
-    if message.tool_calls:
-        call = message.tool_calls[0]
-        return f"{call['name']}({call['args']})"
-    return ""
-
-
-def stream_shopping_agent() -> None:
-    memory = MemorySaver()
-    agent = create_agent(
-        model=llm,
-        tools=[get_price],
-        system_prompt="Ты помощник по планированию покупок",
-        checkpointer=memory,
-    )
-
-    stream = agent.stream(
-        {
-            "messages": [
-                {
-                    "role": "human",
-                    "content": (
-                        "Помоги составить список покупок: молоко, хлеб, яблоки. "
-                        "Я нахожусь в Казани."
-                    ),
-                }
-            ]
-        },
-        stream_mode=["messages", "updates"],
-        config={"configurable": {"thread_id": "shopping-stream-demo"}},
-    )
-
-    step = None
-    for chunk_type, chunk_data in stream:
-        if chunk_type == "messages":
-            message, meta = chunk_data
-            current_step = meta.get("langgraph_step")
-            if current_step != step:
-                step = current_step
-                print("\n --- --- --- \n")
-            if message.content:
-                print(message.content, end="", flush=True)
-
-        if chunk_type == "updates" and chunk_data.get("model"):
-            last_message = chunk_data["model"]["messages"][-1]
-            formatted = format_message(last_message)
-            if formatted:
-                print(formatted)
-
-
-if __name__ == "__main__":
-    stream_shopping_agent()