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

Update agent.py

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