Update model.py

model.py

@@ -49,12 +49,12 @@ def get_openai_server_model(model_id: str, **kwargs) -> OpenAIServerModel:
     if not api_key:
         raise ValueError("OPENAI_API_KEY is not set")
 
-    # api_base = os.getenv("OPENAI_API_BASE")
-    # if not api_base:
-    #     raise ValueError("OPENAI_API_BASE is not set")
+    api_base = os.getenv("OPENAI_API_BASE")
+    if not api_base:
+        raise ValueError("OPENAI_API_BASE is not set")
 
     return OpenAIServerModel(
-        model_id=model_id, api_key=api_key,  **kwargs
+        model_id=model_id, api_key=api_key, api_base=api_base, **kwargs
     )
 
 
@@ -90,4 +90,4 @@ def get_model(model_type: str, model_id: str, **kwargs) -> Any:
     if model_type not in models:
         raise ValueError(f"Unknown model type: {model_type}")
 
-    return models[model_type](model_id, **kwargs)
+    return models[model_type](model_id, **kwargs)