Psiska commited on
Commit
a121bc4
·
1 Parent(s): 28b236b

Changed the use of model

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -15,7 +15,7 @@ class BasicAgent:
15
 
16
  agent = CodeAgent(
17
  tools=[tls.search_tool, tls.calculate_cargo_travel_time],
18
- model=InferenceClientModel(),
19
  additional_authorized_imports=["pandas"],
20
  max_steps=20,
21
  )
 
15
 
16
  agent = CodeAgent(
17
  tools=[tls.search_tool, tls.calculate_cargo_travel_time],
18
+ model=model,
19
  additional_authorized_imports=["pandas"],
20
  max_steps=20,
21
  )