From 5f93cc403f0a1d4117336cef9ccf204c89eece64 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=AD=D0=BC=D0=B8=D0=BB=D1=8C=20=D0=90=D0=BC=D0=B8=D1=80?= =?UTF-8?q?=D0=BE=D0=B2?= Date: Wed, 27 May 2026 12:36:15 +0000 Subject: [PATCH] sync from accepted repo: agent_app.py --- agent_app.py | 21 ++++++++++++--------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/agent_app.py b/agent_app.py index 65645d7..88650cf 100644 --- a/agent_app.py +++ b/agent_app.py @@ -6,25 +6,28 @@ import os from langchain.agents import create_agent from langchain_ollama import ChatOllama -try: - from rag_tools import add_to_knowledge_base, search_knowledge_base -except ModuleNotFoundError: - from .rag_tools import add_to_knowledge_base, search_knowledge_base +from rag_tools import add_to_knowledge_base, search_knowledge_base LLM_MODEL = os.getenv("OLLAMA_LLM_MODEL", "llama3") OLLAMA_BASE_URL = os.getenv("OLLAMA_BASE_URL", "http://localhost:11434") SYSTEM_PROMPT = """\ Ты AI-ассистент с доступом к локальной базе знаний (RAG). -Перед ответом на фактические вопросы используй search_knowledge_base. -Если пользователь просит сохранить информацию — вызывай add_to_knowledge_base. -Если в базе нет данных — честно скажи об этом. -Отвечай на русском языке, кратко и по делу. + +Правила: +1. Перед ответом на фактические вопросы используй search_knowledge_base. +2. Если пользователь просит сохранить информацию — вызывай add_to_knowledge_base. +3. Если в базе нет данных — честно скажи об этом. +4. Отвечай на русском языке, кратко и по делу. """ def build_llm() -> ChatOllama: - return ChatOllama(model=LLM_MODEL, base_url=OLLAMA_BASE_URL, temperature=0.2) + return ChatOllama( + model=LLM_MODEL, + base_url=OLLAMA_BASE_URL, + temperature=0.2, + ) def build_agent():