peterpeter8585 commited on
Commit
4175aba
·
verified ·
1 Parent(s): d1067eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,4 +1,4 @@
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_chat_agent as Agent,AgentExecutor as Ex, AgentType as Type
@@ -65,6 +65,7 @@ prompt = ChatPromptTemplate.from_messages(
65
  from transformers import pipeline
66
  pipe=pipeline(model="peterpeter8585/syai4.3",task="question-answering")
67
  llm=HF(pipeline=pipe)
 
68
  from langchain.retrievers import WikipediaRetriever as Wiki
69
  import gradio as gr
70
  chatbot = gr.Chatbot(
@@ -96,7 +97,7 @@ def chat(message,
96
 
97
  messages.append(HM(content=message))
98
  memory=MEM(memory_key="history")
99
- agent=Agent(llm,tools,prompt,memory=memory)
100
  return agent.invoke({"input":messages,"chat_history":memory.buffer_as_messages})
101
  ai1=gr.ChatInterface(
102
  chat,
 
1
+ from langchain_huggingface import HuggingFacePipeline as HF,ChatHuggingFace
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_chat_agent as Agent,AgentExecutor as Ex, AgentType as Type
 
65
  from transformers import pipeline
66
  pipe=pipeline(model="peterpeter8585/syai4.3",task="question-answering")
67
  llm=HF(pipeline=pipe)
68
+ llm=ChatHuggingFace(llm=llm)
69
  from langchain.retrievers import WikipediaRetriever as Wiki
70
  import gradio as gr
71
  chatbot = gr.Chatbot(
 
97
 
98
  messages.append(HM(content=message))
99
  memory=MEM(memory_key="history")
100
+ agent=Ex(agent=Agent(llm,tools,prompt),tools=tools,verbose=True,handle_parsing_errors=True,memory=memory)
101
  return agent.invoke({"input":messages,"chat_history":memory.buffer_as_messages})
102
  ai1=gr.ChatInterface(
103
  chat,