Обновлен vectorstore.py: remove agent.py
This commit is contained in:
@@ -1,28 +0,0 @@
|
|||||||
from langchain.agents import create_agent
|
|
||||||
from langchain_ollama import ChatOllama
|
|
||||||
|
|
||||||
from tools import search_local_kb, web_search
|
|
||||||
|
|
||||||
OLLAMA_BASE_URL = "http://127.0.0.1:11434"
|
|
||||||
LLM_MODEL = "llama3"
|
|
||||||
|
|
||||||
SYSTEM_PROMPT = """
|
|
||||||
You are a RAG assistant that must choose the best source before answering.
|
|
||||||
|
|
||||||
Rules:
|
|
||||||
- Use search_local_kb for questions about local notes, internal documents, course materials, or any topic that may already exist in the local knowledge base.
|
|
||||||
- Use web_search for current events, recent news, live facts, or questions that clearly require the internet.
|
|
||||||
- Do not call both tools unless the first one clearly failed to provide enough information.
|
|
||||||
- Every final answer must end with a separate line in the exact format: "Источник: chromadb" or "Источник: tavily".
|
|
||||||
- Answer in Russian.
|
|
||||||
"""
|
|
||||||
|
|
||||||
agent = create_agent(
|
|
||||||
model=ChatOllama(
|
|
||||||
model=LLM_MODEL,
|
|
||||||
base_url=OLLAMA_BASE_URL,
|
|
||||||
temperature=0,
|
|
||||||
),
|
|
||||||
tools=[search_local_kb, web_search],
|
|
||||||
system_prompt=SYSTEM_PROMPT,
|
|
||||||
)
|
|
||||||
Reference in New Issue
Block a user