huytofu92 commited on
Commit
733585a
·
1 Parent(s): 92a4c5d

Remove arg

Browse files
Files changed (1) hide show
  1. mini_agents.py +6 -12
mini_agents.py CHANGED
@@ -20,8 +20,7 @@ MODEL_CHOICES = {
20
 
21
  audio_model = InferenceClientModel(
22
  model_id=MODEL_CHOICES["audio"][0],
23
- token=os.getenv("HUGGINGFACE_API_KEY"),
24
- api_url="https://api.openai.com/v1/audio/transcriptions"
25
  )
26
 
27
  audio_agent = CodeAgent(
@@ -34,8 +33,7 @@ audio_agent = CodeAgent(
34
 
35
  vlm_model = InferenceClientModel(
36
  model_id=MODEL_CHOICES["vlm"][0],
37
- token=os.getenv("HUGGINGFACE_API_KEY"),
38
- api_url="https://api.openai.com/v1/images/generations"
39
  )
40
 
41
  vlm_agent = CodeAgent(
@@ -48,8 +46,7 @@ vlm_agent = CodeAgent(
48
 
49
  arithmetic_model = InferenceClientModel(
50
  model_id=MODEL_CHOICES["arithmetic"][0],
51
- token=os.getenv("HUGGINGFACE_API_KEY"),
52
- api_url="https://api.openai.com/v1/chat/completions"
53
  )
54
 
55
  arithmetic_agent = CodeAgent(
@@ -62,8 +59,7 @@ arithmetic_agent = CodeAgent(
62
 
63
  pandas_model = InferenceClientModel(
64
  model_id=MODEL_CHOICES["pandas"][0],
65
- token=os.getenv("HUGGINGFACE_API_KEY"),
66
- api_url="https://api.openai.com/v1/chat/completions"
67
  )
68
 
69
  pandas_agent = CodeAgent(
@@ -76,8 +72,7 @@ pandas_agent = CodeAgent(
76
 
77
  code_model = InferenceClientModel(
78
  model_id=MODEL_CHOICES["code"][0],
79
- token=os.getenv("HUGGINGFACE_API_KEY"),
80
- api_url="https://api.openai.com/v1/chat/completions"
81
  )
82
 
83
  multimodal_manager = CodeAgent(
@@ -102,8 +97,7 @@ operation_manager = CodeAgent(
102
 
103
  master_model = InferenceClientModel(
104
  model_id=MODEL_CHOICES["master"][0],
105
- token=os.getenv("HUGGINGFACE_API_KEY"),
106
- api_url="https://api.openai.com/v1/chat/completions"
107
  )
108
 
109
  master_agent = CodeAgent(
 
20
 
21
  audio_model = InferenceClientModel(
22
  model_id=MODEL_CHOICES["audio"][0],
23
+ token=os.getenv("HUGGINGFACE_API_KEY")
 
24
  )
25
 
26
  audio_agent = CodeAgent(
 
33
 
34
  vlm_model = InferenceClientModel(
35
  model_id=MODEL_CHOICES["vlm"][0],
36
+ token=os.getenv("HUGGINGFACE_API_KEY")
 
37
  )
38
 
39
  vlm_agent = CodeAgent(
 
46
 
47
  arithmetic_model = InferenceClientModel(
48
  model_id=MODEL_CHOICES["arithmetic"][0],
49
+ token=os.getenv("HUGGINGFACE_API_KEY")
 
50
  )
51
 
52
  arithmetic_agent = CodeAgent(
 
59
 
60
  pandas_model = InferenceClientModel(
61
  model_id=MODEL_CHOICES["pandas"][0],
62
+ token=os.getenv("HUGGINGFACE_API_KEY")
 
63
  )
64
 
65
  pandas_agent = CodeAgent(
 
72
 
73
  code_model = InferenceClientModel(
74
  model_id=MODEL_CHOICES["code"][0],
75
+ token=os.getenv("HUGGINGFACE_API_KEY")
 
76
  )
77
 
78
  multimodal_manager = CodeAgent(
 
97
 
98
  master_model = InferenceClientModel(
99
  model_id=MODEL_CHOICES["master"][0],
100
+ token=os.getenv("HUGGINGFACE_API_KEY")
 
101
  )
102
 
103
  master_agent = CodeAgent(