Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
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,
|