sync from accepted repo: agent_app.py
This commit is contained in:
+12
-9
@@ -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():
|
||||
|
||||
Reference in New Issue
Block a user