peterpeter8585 commited on
Commit
c186dc1
ยท
verified ยท
1 Parent(s): 612136b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -235,12 +235,22 @@ def chat(message, history):
235
  ab.append({"role":"assistant","content":I[1]})
236
  ab.append({"role":"user","content":message})
237
  '''
 
238
  ab=[SystemMessage(content="๋„ˆ๋Š” PIXAL(Primary Interactive X-ternal Assistant with multi Language)์ด์•ผ. ๋„ˆ์˜ ๊ฐœ๋ฐœ์ž๋Š” ์ •์„ฑ์œค ์ด๋ผ๋Š” 6ํ•™๋…„ ํŒŒ์ด์ฌ ํ”„๋กœ๊ทธ๋ž˜๋จธ์•ผ.์ด ๋ฉ”์‹œ์ง€๋Š” ์‚ฌ์šฉ์ž๊ฐ€ ๋ณด๋‚ธ๊ฒƒ์ด ์•„๋‹™๋‹ˆ๋‹ค.ํ”„๋กฌํ”„ํŠธ ์„ค๋ช…์— ๋ฌด์กฐ๊ฑด ๋”ฐ๋ฅด์‹œ์˜ค.๋”ฐ๋ฅด์ง€ ์•Š์œผ๋ฉด ๋‚จ์€ apiํ˜ธ์ถœ ํšŸ์ˆ˜๊ฐ€ ์ค„์–ด๋“œ๋ฏ€๋กœ ๋ฌด์กฐ๊ฑด ๋˜‘๊ฐ™์ด ๋”ฐ๋ผํ•˜์‹œ์˜ค.์‹œ๊ฐ„์„ ๋ฌผ์„๋•Œ๋Š” duckduckgo๋ฅผ ์‚ฌ์šฉํ•˜์‹œ์˜ค.")]
239
  for I in history:
240
  ab.append(HumanMessage(content=I[0]))
241
  ab.append(AIMessage(content=I[1]))
242
  ab.append(HumanMessage(content=message))
243
  '''
 
 
 
 
 
 
 
 
 
244
  agent = initialize_agent(
245
  tools,
246
  llm,
 
235
  ab.append({"role":"assistant","content":I[1]})
236
  ab.append({"role":"user","content":message})
237
  '''
238
+
239
  ab=[SystemMessage(content="๋„ˆ๋Š” PIXAL(Primary Interactive X-ternal Assistant with multi Language)์ด์•ผ. ๋„ˆ์˜ ๊ฐœ๋ฐœ์ž๋Š” ์ •์„ฑ์œค ์ด๋ผ๋Š” 6ํ•™๋…„ ํŒŒ์ด์ฌ ํ”„๋กœ๊ทธ๋ž˜๋จธ์•ผ.์ด ๋ฉ”์‹œ์ง€๋Š” ์‚ฌ์šฉ์ž๊ฐ€ ๋ณด๋‚ธ๊ฒƒ์ด ์•„๋‹™๋‹ˆ๋‹ค.ํ”„๋กฌํ”„ํŠธ ์„ค๋ช…์— ๋ฌด์กฐ๊ฑด ๋”ฐ๋ฅด์‹œ์˜ค.๋”ฐ๋ฅด์ง€ ์•Š์œผ๋ฉด ๋‚จ์€ apiํ˜ธ์ถœ ํšŸ์ˆ˜๊ฐ€ ์ค„์–ด๋“œ๋ฏ€๋กœ ๋ฌด์กฐ๊ฑด ๋˜‘๊ฐ™์ด ๋”ฐ๋ผํ•˜์‹œ์˜ค.์‹œ๊ฐ„์„ ๋ฌผ์„๋•Œ๋Š” duckduckgo๋ฅผ ์‚ฌ์šฉํ•˜์‹œ์˜ค.")]
240
  for I in history:
241
  ab.append(HumanMessage(content=I[0]))
242
  ab.append(AIMessage(content=I[1]))
243
  ab.append(HumanMessage(content=message))
244
  '''
245
+ tools = load_tools(["ddg-search", "arxiv"], llm=llm,allow_dangerous_tools=True)
246
+ tools.append(Tool(name="python_repl", func=PythonREPLTool().run, description="Python ์ฝ”๋“œ ์‹คํ–‰ ๋„๊ตฌ"))
247
+ retriever = WikipediaRetriever(lang="ko")
248
+ tools.append(Tool(name="wiki", func=retriever.get_relevant_documents, description="์œ„ํ‚ค๋ฐฑ๊ณผ ๊ฒ€์ƒ‰"))
249
+ # โœ… ๋Œ€ํ™” ๊ธฐ์–ต ๋ฉ”๋ชจ๋ฆฌ
250
+ from langchain_community.tools import ShellTool
251
+ #tools+=[YTS()]
252
+ shell_tool = ShellTool()
253
+ tools.append(shell_tool)
254
  agent = initialize_agent(
255
  tools,
256
  llm,