AntonVoronko commited on
Commit
f4da3ca
·
verified ·
1 Parent(s): dbc0fe2

Model change

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -29,7 +29,8 @@ class BasicAgent:
29
  wiki_tool = FunctionTool.from_defaults(wiki_spec.search_data)
30
  arxiv_spec = ArxivToolSpec()
31
  arxiv_tool = FunctionTool.from_defaults(arxiv_spec.arxiv_query)
32
- llm = Groq(model="deepseek-r1-distill-llama-70b", api_key=api_key)
 
33
  self.agent = FunctionAgent(tools=[search_tool, wiki_tool, arxiv_tool], llm=llm, system_prompt='''
34
  You are a general AI assistant.
35
  I will ask you a question. You can use the tools at your disposal to help you with an answer.
 
29
  wiki_tool = FunctionTool.from_defaults(wiki_spec.search_data)
30
  arxiv_spec = ArxivToolSpec()
31
  arxiv_tool = FunctionTool.from_defaults(arxiv_spec.arxiv_query)
32
+ # llm = Groq(model="deepseek-r1-distill-llama-70b", api_key=api_key)
33
+ llm = Groq(model="qwen-qwq-32b", api_key=api_key)
34
  self.agent = FunctionAgent(tools=[search_tool, wiki_tool, arxiv_tool], llm=llm, system_prompt='''
35
  You are a general AI assistant.
36
  I will ask you a question. You can use the tools at your disposal to help you with an answer.