diff --git a/rag_agent.py b/rag_agent.py new file mode 100644 index 0000000..33a21a0 --- /dev/null +++ b/rag_agent.py @@ -0,0 +1,9 @@ +from langchain.agents import create_openai_functions_agent, AgentExecutor +from langchain.llms import Ollama +from tools import search_knowledge_base, add_to_knowledge_base + +def get_agent(): + llm = Ollama(model="llama3") + tools_list = [search_knowledge_base, add_to_knowledge_base] + agent = create_openai_functions_agent(llm=llm, tools=tools_list, system_prompt="You are an assistant that uses a knowledge base.") + return AgentExecutor(agent=agent, tools=tools_list)