diff --git a/rag_agent.py b/rag_agent.py new file mode 100644 index 0000000..956fb95 --- /dev/null +++ b/rag_agent.py @@ -0,0 +1,16 @@ +from langchain.agents import create_openai_functions_agent, AgentExecutor +from rag_tools import search_knowledge_base, add_to_knowledge_base +from langchain.schema import HumanMessage + +def run_agent(): + tools=[search_knowledge_base,add_to_knowledge_base] + agent=create_openai_functions_agent(tools=tools, llm=None) + executor=AgentExecutor(agent=agent, tools=tools, verbose=True) + while True: + inp=input("> ") + if inp.strip().lower()=="/quit": + break + response=executor.invoke({"input":inp}) + print(response["output"]) +if __name__=="__main__": + run_agent()