Add rag_agent

This commit is contained in:
2026-05-28 13:24:24 +00:00
parent 6195155d67
commit 1a1c746db5
+16
View File
@@ -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()