qver3nc1a commited on
Commit
51172b8
·
verified ·
1 Parent(s): efbf423

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -38,14 +38,13 @@ def wikipedia_search(query: str):
38
  return wikipedia.run(query)
39
 
40
  class BasicAgent:
41
- def __init__(self, model_id="ollama/qwen2:7b", api_base="http://192.168.1.77:11434"):
42
  print('BasicAgent initialized.')
43
  self.tool_node = [web_search, wikipedia_search]
44
  self.agent = ToolCallingAgent(
45
  tools=[self.tool_node],
46
  model=LiteLLMModel(
47
  model_id=model_id,
48
- api_base=api_base
49
  ),
50
  max_steps=15,
51
  verbosity_level=2,
 
38
  return wikipedia.run(query)
39
 
40
  class BasicAgent:
41
+ def __init__(self, model_id="ollama/qwen2:7b"):
42
  print('BasicAgent initialized.')
43
  self.tool_node = [web_search, wikipedia_search]
44
  self.agent = ToolCallingAgent(
45
  tools=[self.tool_node],
46
  model=LiteLLMModel(
47
  model_id=model_id,
 
48
  ),
49
  max_steps=15,
50
  verbosity_level=2,