peterpeter8585 commited on
Commit
18f646d
·
verified ·
1 Parent(s): 426f195

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from langchain_huggingface import HuggingFacePipeline 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
@@ -19,8 +19,8 @@ chatbot = gr.Chatbot(
19
  def terminal(c):
20
  a=Popen(c,shell=True,stdin=P,stdout=P,stderr=P)
21
  return a.stdout.read()+a.stderr.read()
22
- llm=HF.from_model_id(model_id="peterpeter8585/syai4.1")
23
- tools=[]
24
  tools.append(PYT())
25
  tools.extend(load_tools(["requests_all"],allow_dangerous_tools=True))
26
  tools.extend(load_tools(["llm-math","ddg-search"],llm=llm))
@@ -41,7 +41,7 @@ def chat(message,
41
 
42
  messages.append(HM(content=message))
43
  memory=MEM(memory_key="history")
44
- agent=Agent(llm=llm,tools=tools,memory=memory,verbose=True,agent=Type.CHAT_ZERO_SHOT_REACT_DESCRIPTION)
45
  yield agent.invoke(messages)
46
  ai1=gr.ChatInterface(
47
  chat,
 
1
  from langchain_huggingface import HuggingFacePipeline 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, create_structured_react_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
 
19
  def terminal(c):
20
  a=Popen(c,shell=True,stdin=P,stdout=P,stderr=P)
21
  return a.stdout.read()+a.stderr.read()
22
+ llm=HF(repo_id="peterpeter8585/syai4.1",HUGGINGFACEHUB_API_TOKEN=os.environ["key"])
23
+ tools=FMT().get_tools()
24
  tools.append(PYT())
25
  tools.extend(load_tools(["requests_all"],allow_dangerous_tools=True))
26
  tools.extend(load_tools(["llm-math","ddg-search"],llm=llm))
 
41
 
42
  messages.append(HM(content=message))
43
  memory=MEM(memory_key="history")
44
+ agent=Ex(agent=Agent(llm=llm,tools=tools),tools=tools,memory=memory,verbose=True,handle_parsing_errors=True)
45
  yield agent.invoke(messages)
46
  ai1=gr.ChatInterface(
47
  chat,