peterpeter8585 commited on
Commit
7233fbc
·
verified ·
1 Parent(s): 24e92f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -2,7 +2,7 @@ from langchain_huggingface import HuggingFaceEndpoint as HF
2
  from subprocess import Popen, PIPE as P
3
  from langchain_experimental.tools.python.tool import PythonREPLTool as PYT
4
  from langchain.agents import load_tools, initialize_agent as Agent,AgentExecutor as Ex, AgentType as Type
5
- from langchain.agents.agent_toolkits import create_retriever_tool as crt
6
  from langchain_community.agent_toolkits import FileManagementToolkit as FMT
7
  from langchain.tools import Tool
8
  from langchain.memory import ConversationBufferMemory as MEM,RedisChatMessageHistory as HIS
@@ -40,9 +40,8 @@ def chat(message,
40
  messages.append(AM(content=val[1]))
41
 
42
  messages.append(HM(content=message))
43
- history1=HIS(session_id=chat_session, url=os.environ["URL"])
44
- memory=MEM(chat_memory=history1,memory_key="history")
45
- agent=Agent(tools=tools,llm=llm,memory=memory,agent=Type.CHAT_ZERO_SHOT_REACT_DESCRIPTION, verbose=True)
46
  yield agent.invoke(messages)
47
  ai1=gr.ChatInterface(
48
  chat,
 
2
  from subprocess import Popen, PIPE as P
3
  from langchain_experimental.tools.python.tool import PythonREPLTool as PYT
4
  from langchain.agents import load_tools, initialize_agent as Agent,AgentExecutor as Ex, AgentType as Type
5
+ from langchain.agents.agent_toolkits import create_retriever_tool as crt,create_structed_chat_agent as A
6
  from langchain_community.agent_toolkits import FileManagementToolkit as FMT
7
  from langchain.tools import Tool
8
  from langchain.memory import ConversationBufferMemory as MEM,RedisChatMessageHistory as HIS
 
40
  messages.append(AM(content=val[1]))
41
 
42
  messages.append(HM(content=message))
43
+ memory=MEM(memory_key="history")
44
+ agent=Ex(agent=A(tools=tools,llm=llm),tools=tools,verbose=True, handle_parsing_errors=True,memory_key="history")
 
45
  yield agent.invoke(messages)
46
  ai1=gr.ChatInterface(
47
  chat,