remove mirrored file agent_app.py

This commit is contained in:
2026-05-27 13:28:09 +00:00
parent 86f77b8095
commit 30709662d6
-39
View File
@@ -1,39 +0,0 @@
"""Агент с RAG-инструментами."""
from __future__ import annotations
import os
from langchain.agents import create_agent
from langchain_ollama import ChatOllama
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).
Правила:
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,
)
def build_agent():
"""Агент с RAG-tools — точка входа для автопроверки."""
return create_agent(
model=build_llm(),
tools=[search_knowledge_base, add_to_knowledge_base],
system_prompt=SYSTEM_PROMPT,
)