Hugging Face's logo

Spaces:
sirine1712
/
Final_Assignment_Template
Sleeping

App Files Community
sirine1712 commited on
Commit
7f7882b
·
verified ·
1 Parent(s): 147fb58

Update multiagents.py

Browse files
Files changed (1) hide show
  1. multiagents.py +8 -6
multiagents.py CHANGED
@@ -28,15 +28,17 @@ vllm_model = OpenAIServerModel(
28
  )
29
 
30
  openai_41nano_model =OpenAIServerModel(
31
- model_id="Qwen/Qwen2.5-1.5B-Instruct",
32
- api_base="http://192.168.1.39:18000/v1",
33
- api_key="token-abc123",
 
34
  )
35
 
36
  openai_41mini_model = OpenAIServerModel(
37
- model_id="Qwen/Qwen2.5-1.5B-Instruct",
38
- api_base="http://192.168.1.39:18000/v1",
39
- api_key="token-abc123",
 
40
  )
41
 
42
 
 
28
  )
29
 
30
  openai_41nano_model =OpenAIServerModel(
31
+ model_id="gemini-2.0-flash",
32
+ api_key=os.environ["GEMINI_API_KEY"],
33
+ # Google Gemini OpenAI-compatible API base URL
34
+ api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
35
  )
36
 
37
  openai_41mini_model = OpenAIServerModel(
38
+ model_id="gemini-2.0-flash",
39
+ api_key=os.environ["GEMINI_API_KEY"],
40
+ # Google Gemini OpenAI-compatible API base URL
41
+ api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
42
  )
43
 
44