peterpeter8585 commited on
Commit
73e0190
·
verified ·
1 Parent(s): 569411f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -75,7 +75,7 @@ prompt = ChatPromptTemplate.from_messages(
75
  #m=M.from_pretrained("peterpeter8585/syai4.3")
76
  #t=T.from_pretrained("peterpeter8585/syai4.3")
77
  #pipe=pipeline(model=m,tokenizer=t,task="text-generation")
78
- llm=HF(repo_id="peterpeter8585/syai4.5",task="textgeneration",stream=False)
79
  from langchain.retrievers import WikipediaRetriever as Wiki
80
  import gradio as gr
81
  chatbot = gr.Chatbot(
@@ -97,7 +97,7 @@ def chat(message,
97
  system_message,
98
  max_tokens,
99
  temperature,
100
- top_p, chat_session=""):
101
  messages=[SM(content=system_message+"And, Your name is Chatchat")]
102
  for val in history:
103
  if val[0]:
@@ -106,7 +106,8 @@ def chat(message,
106
  messages.append(AM(content=val[1]))
107
 
108
  messages.append(HM(content=message))
109
- memory=MEM(memory_key="history")
 
110
  agent=Ex(agent=Agent(llm,tools,prompt),tools=tools,verbose=True,handle_parsing_errors=True,memory=memory)
111
  return agent.invoke({"input":messages,"chat_history":memory.buffer_as_messages})
112
  ai1=gr.ChatInterface(
 
75
  #m=M.from_pretrained("peterpeter8585/syai4.3")
76
  #t=T.from_pretrained("peterpeter8585/syai4.3")
77
  #pipe=pipeline(model=m,tokenizer=t,task="text-generation")
78
+ llm=HF(repo_id="peterpeter8585/syai4.5",task="textgeneration")
79
  from langchain.retrievers import WikipediaRetriever as Wiki
80
  import gradio as gr
81
  chatbot = gr.Chatbot(
 
97
  system_message,
98
  max_tokens,
99
  temperature,
100
+ top_p, chat_session):
101
  messages=[SM(content=system_message+"And, Your name is Chatchat")]
102
  for val in history:
103
  if val[0]:
 
106
  messages.append(AM(content=val[1]))
107
 
108
  messages.append(HM(content=message))
109
+ his=HIS(url=os.environ["URL"],session_id=chat_session)
110
+ memory=MEM(memory_key="history",chat_memory=his)
111
  agent=Ex(agent=Agent(llm,tools,prompt),tools=tools,verbose=True,handle_parsing_errors=True,memory=memory)
112
  return agent.invoke({"input":messages,"chat_history":memory.buffer_as_messages})
113
  ai1=gr.ChatInterface(