cowrycode commited on
Commit
74afc0f
·
verified ·
1 Parent(s): dad6e91

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +7 -8
agent.py CHANGED
@@ -5,15 +5,14 @@ from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
5
  from youtube_tool import youtube_transcript_tool, youtube_transcript_snippet_tool
6
  from multiple_tools import round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool, wikipedia_search_tool
7
 
 
 
 
 
 
 
 
8
  async def smart_agent(question : str, api_key : str) -> str:
9
- llm = HuggingFaceInferenceAPI(
10
- model_name="deepseek-ai/DeepSeek-R1-0528",
11
- token=api_key,
12
- provider="auto",
13
- max_iterations=10,
14
- max_execution_time=60
15
- )
16
-
17
  # Create Alfred, our gala agent, with the guest info tool
18
  llamaindex_agent = AgentWorkflow.from_tools_or_functions(
19
  [wikipedia_search_tool, youtube_transcript_tool, youtube_transcript_snippet_tool, round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool],
 
5
  from youtube_tool import youtube_transcript_tool, youtube_transcript_snippet_tool
6
  from multiple_tools import round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool, wikipedia_search_tool
7
 
8
+ llm = HuggingFaceInferenceAPI(
9
+ model_name="deepseek-ai/DeepSeek-R1-0528",
10
+ token=api_key,
11
+ provider="auto",
12
+ max_iterations=10,
13
+ max_execution_time=60
14
+ )
15
  async def smart_agent(question : str, api_key : str) -> str:
 
 
 
 
 
 
 
 
16
  # Create Alfred, our gala agent, with the guest info tool
17
  llamaindex_agent = AgentWorkflow.from_tools_or_functions(
18
  [wikipedia_search_tool, youtube_transcript_tool, youtube_transcript_snippet_tool, round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool],