peterpeter8585 commited on
Commit
adab5f4
Β·
verified Β·
1 Parent(s): 6c0e0df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -2
app.py CHANGED
@@ -205,12 +205,24 @@ class GitHubModelLLM(LLM):
205
  # ──────────────────────────────
206
  # βœ… LangChain 도ꡬ & μ—μ΄μ „νŠΈ ꡬ성
207
  # ──────────────────────────────
 
208
 
209
- tools = load_tools(["ddg-search", "llm-math","arxiv","requests_all"], llm=llm,allow_dangerous_tools=True)
 
 
 
 
 
 
 
 
 
 
 
210
  tools.append(Tool(name="python_repl", func=PythonREPLTool().run, description="Python μ½”λ“œ μ‹€ν–‰ 도ꡬ"))
211
  retriever = WikipediaRetriever(lang="ko")
212
  tools.append(Tool(name="wiki", func=retriever.get_relevant_documents, description="μœ„ν‚€λ°±κ³Ό 검색"))
213
- tools.append(Tool(name="time_now", func=lambda _: f"ν˜„μž¬ μ‹œκ°: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')} (Asia/Seoul)", description="ν˜„μž¬ μ‹œκ°„μ„ λ°˜ν™˜ν•©λ‹ˆλ‹€."))
214
  # βœ… λŒ€ν™” κΈ°μ–΅ λ©”λͺ¨λ¦¬
215
  from langchain_community.tools import ShellTool
216
  #tools+=[YTS()]
 
205
  # ──────────────────────────────
206
  # βœ… LangChain 도ꡬ & μ—μ΄μ „νŠΈ ꡬ성
207
  # ──────────────────────────────
208
+ import requests
209
 
210
+ def get_current_time(_=""):
211
+ timezone='Asia/Seoul'
212
+ url = f'https://worldtimeapi.org/api/timezone/{timezone}'
213
+ try:
214
+ resp = requests.get(url)
215
+ resp.raise_for_status()
216
+ obj = resp.json()
217
+ return f"ν˜„μž¬ μ‹œκ°„ ({timezone}): {obj['datetime']}"
218
+ except Exception as e:
219
+ return "API 호좜 λ˜λŠ” 처리 였λ₯˜"
220
+
221
+ tools = load_tools(["ddg-search", "llm-math","arxiv"], llm=llm,allow_dangerous_tools=True)
222
  tools.append(Tool(name="python_repl", func=PythonREPLTool().run, description="Python μ½”λ“œ μ‹€ν–‰ 도ꡬ"))
223
  retriever = WikipediaRetriever(lang="ko")
224
  tools.append(Tool(name="wiki", func=retriever.get_relevant_documents, description="μœ„ν‚€λ°±κ³Ό 검색"))
225
+ tools.append(Tool(name="time_now", func=get_current_time, description="ν˜„μž¬ μ‹œκ°„μ„ λ°˜ν™˜ν•©λ‹ˆλ‹€.μž…λ ₯:''(μ—†μŒ)"))
226
  # βœ… λŒ€ν™” κΈ°μ–΅ λ©”λͺ¨λ¦¬
227
  from langchain_community.tools import ShellTool
228
  #tools+=[YTS()]