Bandook01 commited on
Commit
5d657b0
·
1 Parent(s): 582cff4
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -18,6 +18,8 @@ from llama_index.core.agent.workflow import FunctionAgent
18
  from llama_index.core.tools.tool_spec.load_and_search import (
19
  LoadAndSearchToolSpec,
20
  )
 
 
21
 
22
 
23
  # (Keep Constants as is)
@@ -52,15 +54,15 @@ Tool Use Guidelines:
52
  class BasicAgent:
53
  def __init__(self):
54
  try:
55
- gemini_api_key = os.getenv("GEMINI_API_KEY")
56
- if not gemini_api_key:
57
- raise ValueError("GEMINI_API_KEY not set in environment variables.")
58
- llm = GoogleGenAI(
59
- model="gemini-2.0-flash",
60
- api_key=gemini_api_key,
61
  )
62
  self.agent = FunctionAgent(
63
- tools=[ReverseTextTool,RunPythonFileTool,download_server,wiki_tool,YoutubeTranscript],
64
  llm=llm,
65
  system_prompt=SYSTEM_PROMPT,
66
  )
 
18
  from llama_index.core.tools.tool_spec.load_and_search import (
19
  LoadAndSearchToolSpec,
20
  )
21
+ from smolagents import DuckDuckGoSearchTool
22
+
23
 
24
 
25
  # (Keep Constants as is)
 
54
  class BasicAgent:
55
  def __init__(self):
56
  try:
57
+ hf_api_key = os.getenv("HF_API_KEY")
58
+ if not hf_api_key:
59
+ raise ValueError("hf_api_key not set in environment variables.")
60
+ llm = HuggingFaceInferenceAPI(
61
+ model="Qwen/Qwen2.5-Coder-32B-Instruct",
62
+ api_key=hf_api_key,
63
  )
64
  self.agent = FunctionAgent(
65
+ tools=[ReverseTextTool,RunPythonFileTool,download_server,wiki_tool,YoutubeTranscript, DuckDuckGoSearchTool()],
66
  llm=llm,
67
  system_prompt=SYSTEM_PROMPT,
68
  )