10 lines
477 B
Python
10 lines
477 B
Python
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)
|