peterpeter8585 commited on
Commit
dedbb83
Β·
verified Β·
1 Parent(s): 01e820b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -8,7 +8,14 @@ from langchain.llms.base import LLM
8
  from langchain.agents import initialize_agent, AgentType,load_tools
9
  from langchain.tools import Tool
10
  from langchain_experimental.tools.python.tool import PythonREPLTool
11
-
 
 
 
 
 
 
 
12
  # ──────────────────────────────
13
  # βœ… GitHub Models LLM
14
  # ──────────────────────────────
@@ -49,9 +56,9 @@ llm = GitHubModelLLM(model="openai/gpt-4.1", token=token)
49
  # βœ… LangChain κΈ°λ³Έ 도ꡬ 뢈러였기
50
  # ──────────────────────────────
51
  tools = load_tools(
52
- ["ddg-search", "requests_all", "llm-math", "wikipedia", "youtube-search"],
53
  llm=llm,allow_dangerous_tools=True
54
- )
55
 
56
  # ──────────────────────────────
57
  # βœ… Python μ‹€ν–‰ 도ꡬ (LangChain λ‚΄μž₯)
 
8
  from langchain.agents import initialize_agent, AgentType,load_tools
9
  from langchain.tools import Tool
10
  from langchain_experimental.tools.python.tool import PythonREPLTool
11
+ from langchain_community.retrievers import WikipediaRetriever
12
+ from langchain.tools.retriever import create_retriever_tool
13
+ retriever = WikipediaRetriever()
14
+ retriever_tool = create_retriever_tool(
15
+ retriever,
16
+ name="wiki_search",
17
+ description="μœ„ν‚€λ°±κ³Όμ—μ„œ ν•„μš”ν•œ 정보λ₯Ό λΆˆλŸ¬μ˜΅λ‹ˆλ‹€.κ²°κ΄΄λ₯Ό κ²€μ¦ν•˜μ—¬ μ‚¬μš©ν•˜μ‹œμ˜€.",
18
+ )
19
  # ──────────────────────────────
20
  # βœ… GitHub Models LLM
21
  # ──────────────────────────────
 
56
  # βœ… LangChain κΈ°λ³Έ 도ꡬ 뢈러였기
57
  # ──────────────────────────────
58
  tools = load_tools(
59
+ ["ddg-search", "requests_all", "llm-math", "youtube-search"],
60
  llm=llm,allow_dangerous_tools=True
61
+ )+[retriever_tool]
62
 
63
  # ──────────────────────────────
64
  # βœ… Python μ‹€ν–‰ 도ꡬ (LangChain λ‚΄μž₯)