peterpeter8585 commited on
Commit
7b89462
·
verified ·
1 Parent(s): f26b648

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -12,6 +12,7 @@ from langchain.schema import SystemMessage as SM,HumanMessage as HM, AIMessage a
12
  from langchain import hub
13
  import os
14
  import torch
 
15
  torch.set_flush_denormal(True)
16
  import importlib.util
17
  import logging
@@ -73,6 +74,7 @@ prompt = ChatPromptTemplate.from_messages(
73
  ("human", human),
74
  ]
75
  )
 
76
  from typing import Any, Dict, List, Optional
77
  from langchain_core.language_models import BaseChatModel
78
  from langchain_core.messages import AIMessage, BaseMessage, HumanMessage
@@ -106,6 +108,7 @@ tools.extend(load_tools(["requests_all"],allow_dangerous_tools=True))
106
  tools.extend(load_tools(["llm-math","ddg-search"],llm=llm))
107
  tools.append(Tool.from_function(func=terminal,name="terminal",description="터미널 명령어실행에 적합함"))
108
  tools.append(crt(name="wiki",description="위키 백과를 검색하여 정보를 가져온다",retriever=Wiki(lang="ko",top_k_results=1)))
 
109
  def chat(message,
110
  history: list[tuple[str, str]],
111
  system_message,
@@ -120,9 +123,7 @@ def chat(message,
120
  messages.append(val[1])
121
 
122
  messages.append(message)
123
- memory=MEM(memory_key="history")
124
- agent=Ex(agent=Agent(llm,tools),tools=tools,verbose=True,handle_parsing_errors=True,memory=memory)
125
- return agent.invoke({"input":messages,"chat_history":memory.buffer_as_messages})
126
  ai1=gr.ChatInterface(
127
  chat,
128
  chatbot=chatbot,
 
12
  from langchain import hub
13
  import os
14
  import torch
15
+ from langchain import hub
16
  torch.set_flush_denormal(True)
17
  import importlib.util
18
  import logging
 
74
  ("human", human),
75
  ]
76
  )
77
+ prompt=hub.pull("hwchase17/structured-chat-agent")
78
  from typing import Any, Dict, List, Optional
79
  from langchain_core.language_models import BaseChatModel
80
  from langchain_core.messages import AIMessage, BaseMessage, HumanMessage
 
108
  tools.extend(load_tools(["llm-math","ddg-search"],llm=llm))
109
  tools.append(Tool.from_function(func=terminal,name="terminal",description="터미널 명령어실행에 적합함"))
110
  tools.append(crt(name="wiki",description="위키 백과를 검색하여 정보를 가져온다",retriever=Wiki(lang="ko",top_k_results=1)))
111
+ agent=Ex(agent=Agent(llm,tools,prompt),tools=tools,verbose=True,handle_parsing_errors=True)
112
  def chat(message,
113
  history: list[tuple[str, str]],
114
  system_message,
 
123
  messages.append(val[1])
124
 
125
  messages.append(message)
126
+ return agent.invoke({"input":messages)
 
 
127
  ai1=gr.ChatInterface(
128
  chat,
129
  chatbot=chatbot,