fix: agent_app.py — lazy RAG + build_* для автопроверки

This commit is contained in:
2026-05-27 08:23:32 +00:00
parent 9fa12528ed
commit 84b305604e
+11 -2
View File
@@ -9,6 +9,7 @@ from langchain_ollama import ChatOllama
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") LLM_MODEL = os.getenv("OLLAMA_LLM_MODEL", "llama3")
OLLAMA_BASE_URL = os.getenv("OLLAMA_BASE_URL", "http://localhost:11434")
SYSTEM_PROMPT = """\ SYSTEM_PROMPT = """\
Ты AI-ассистент с доступом к локальной базе знаний (RAG). Ты AI-ассистент с доступом к локальной базе знаний (RAG).
@@ -21,10 +22,18 @@ SYSTEM_PROMPT = """\
""" """
def build_llm() -> ChatOllama:
return ChatOllama(
model=LLM_MODEL,
base_url=OLLAMA_BASE_URL,
temperature=0.2,
)
def build_agent(): def build_agent():
llm = ChatOllama(model=LLM_MODEL, temperature=0.2) """Агент с RAG-tools — точка входа для автопроверки."""
return create_agent( return create_agent(
model=llm, model=build_llm(),
tools=[search_knowledge_base, add_to_knowledge_base], tools=[search_knowledge_base, add_to_knowledge_base],
system_prompt=SYSTEM_PROMPT, system_prompt=SYSTEM_PROMPT,
) )