Spaces:
Sleeping
Sleeping
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,
|