peterpeter8585 commited on
Commit
63dc225
·
verified ·
1 Parent(s): 2c6ec22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,4 +1,5 @@
1
  # app_pixal_chat.py
 
2
  import os, re, json, gradio as gr
3
  from langchain.memory import ConversationBufferMemory
4
  from langchain.agents import initialize_agent, AgentType, load_tools
@@ -260,10 +261,11 @@ def chat(message, history):
260
  ab.append(AIMessage(content=I[1]))
261
  ab.append(HumanMessage(content=message))
262
  '''
263
- tools = load_tools(["ddg-search", "arxiv"], llm=llm,allow_dangerous_tools=True)
264
  tools.append(Tool(name="python_repl", func=PythonREPLTool().run, description="Python 코드 실행 도구"))
 
265
  retriever = WikipediaRetriever(lang="ko")
266
- tools.append(Tool(name="wiki", func=retriever.get_relevant_documents, description="위키백과 검색"))
267
  # ✅ 대화 기억 메모리
268
  from langchain_community.tools import ShellTool
269
  #tools+=[YTS()]
 
1
  # app_pixal_chat.py
2
+ from langchain_community.tools import DuckDuckGoSearchRun as DDGS
3
  import os, re, json, gradio as gr
4
  from langchain.memory import ConversationBufferMemory
5
  from langchain.agents import initialize_agent, AgentType, load_tools
 
261
  ab.append(AIMessage(content=I[1]))
262
  ab.append(HumanMessage(content=message))
263
  '''
264
+ tools = load_tools(["arxiv"], llm=llm,allow_dangerous_tools=True)
265
  tools.append(Tool(name="python_repl", func=PythonREPLTool().run, description="Python 코드 실행 도구"))
266
+ tools.append(Tool(name="duckduckgo_search", func=DDGS().run, description="인터넷 검색 도구"))
267
  retriever = WikipediaRetriever(lang="ko")
268
+ tools.append(Tool(name="wiki", func=retriever.get_relevant_documents, description="위키백과 검색도구"))
269
  # ✅ 대화 기억 메모리
270
  from langchain_community.tools import ShellTool
271
  #tools+=[YTS()]