From c50671a80370b3593998d4d393163d3087b7a2e7 Mon Sep 17 00:00:00 2001
From: RomanKurlygin <ksu23182002@mail.ru>
Date: Tue, 26 May 2026 07:24:24 +0000
Subject: [PATCH] add agent_app.py

---
 agent_app.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 agent_app.py

diff --git a/agent_app.py b/agent_app.py
new file mode 100644
index 0000000..bfe1813
--- /dev/null
+++ b/agent_app.py
@@ -0,0 +1,30 @@
+"""Агент с 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")
+
+SYSTEM_PROMPT = """\
+Ты AI-ассистент с доступом к локальной базе знаний (RAG).
+
+Правила:
+1. Перед ответом на фактические вопросы используй search_knowledge_base.
+2. Если пользователь просит сохранить информацию — вызывай add_to_knowledge_base.
+3. Если в базе нет данных — честно скажи об этом.
+4. Отвечай на русском языке, кратко и по делу.
+"""
+
+
+def build_agent():
+    llm = ChatOllama(model=LLM_MODEL, temperature=0.2)
+    return create_agent(
+        model=llm,
+        tools=[search_knowledge_base, add_to_knowledge_base],
+        system_prompt=SYSTEM_PROMPT,
+    )