shubham7080 commited on
Commit
f03d6ef
·
verified ·
1 Parent(s): 6cb4b8e

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +3 -3
model.py CHANGED
@@ -49,9 +49,9 @@ def get_openai_server_model(model_id: str, **kwargs) -> OpenAIServerModel:
49
  if not api_key:
50
  raise ValueError("OPENAI_API_KEY is not set")
51
 
52
- api_base = os.getenv("OPENAI_API_BASE")
53
- if not api_base:
54
- raise ValueError("OPENAI_API_BASE is not set")
55
 
56
  return OpenAIServerModel(
57
  model_id=model_id, api_key=api_key, api_base=api_base, **kwargs
 
49
  if not api_key:
50
  raise ValueError("OPENAI_API_KEY is not set")
51
 
52
+ # api_base = os.getenv("OPENAI_API_BASE")
53
+ # if not api_base:
54
+ # raise ValueError("OPENAI_API_BASE is not set")
55
 
56
  return OpenAIServerModel(
57
  model_id=model_id, api_key=api_key, api_base=api_base, **kwargs