Matthew commited on
Commit
cf2a952
·
1 Parent(s): df25518

Add multimodal model

Browse files
Files changed (1) hide show
  1. agent.py +2 -4
agent.py CHANGED
@@ -1,15 +1,13 @@
1
  from smolagents import (
2
  CodeAgent,
3
- InferenceClientModel,
4
  )
5
 
6
  class DRAgent:
7
  def __init__(self):
8
  self.agent = CodeAgent(
9
  tools=[],
10
- model=InferenceClientModel(
11
- model_id="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"
12
- ),
13
  )
14
  print("BasicAgent initialized.")
15
  def __call__(self, question: str) -> str:
 
1
  from smolagents import (
2
  CodeAgent,
3
+ OpenAIServerModel,
4
  )
5
 
6
  class DRAgent:
7
  def __init__(self):
8
  self.agent = CodeAgent(
9
  tools=[],
10
+ model = OpenAIServerModel(model_id="gpt-4o")
 
 
11
  )
12
  print("BasicAgent initialized.")
13
  def __call__(self, question: str) -> str: