Fixed error handling
Browse files- L3Score.py +1 -5
L3Score.py
CHANGED
|
@@ -122,8 +122,7 @@ class L3Score(evaluate.Metric):
|
|
| 122 |
)
|
| 123 |
|
| 124 |
# Check whether the model is available
|
| 125 |
-
|
| 126 |
-
|
| 127 |
if provider == "openai":
|
| 128 |
client = openai.OpenAI(api_key=api_key)
|
| 129 |
model_names = set([model.id for model in client.models.list()])
|
|
@@ -131,17 +130,14 @@ class L3Score(evaluate.Metric):
|
|
| 131 |
raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
|
| 132 |
|
| 133 |
elif provider == "deepseek":
|
| 134 |
-
print("Checking DeepSeek model")
|
| 135 |
client = openai.OpenAI(api_key=api_key,base_url="https://api.deepseek.com")
|
| 136 |
model_names = [model.id for model in client.models.list()]
|
| 137 |
-
print(model_names)
|
| 138 |
if model not in model_names:
|
| 139 |
raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
|
| 140 |
|
| 141 |
elif provider == "xai":
|
| 142 |
client = openai.OpenAI(api_key=api_key, base_url="https://api.xai.com")
|
| 143 |
model_names = [model.id for model in client.models.list()]
|
| 144 |
-
print(model_names)
|
| 145 |
if model not in model_names:
|
| 146 |
raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
|
| 147 |
|
|
|
|
| 122 |
)
|
| 123 |
|
| 124 |
# Check whether the model is available
|
| 125 |
+
|
|
|
|
| 126 |
if provider == "openai":
|
| 127 |
client = openai.OpenAI(api_key=api_key)
|
| 128 |
model_names = set([model.id for model in client.models.list()])
|
|
|
|
| 130 |
raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
|
| 131 |
|
| 132 |
elif provider == "deepseek":
|
|
|
|
| 133 |
client = openai.OpenAI(api_key=api_key,base_url="https://api.deepseek.com")
|
| 134 |
model_names = [model.id for model in client.models.list()]
|
|
|
|
| 135 |
if model not in model_names:
|
| 136 |
raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
|
| 137 |
|
| 138 |
elif provider == "xai":
|
| 139 |
client = openai.OpenAI(api_key=api_key, base_url="https://api.xai.com")
|
| 140 |
model_names = [model.id for model in client.models.list()]
|
|
|
|
| 141 |
if model not in model_names:
|
| 142 |
raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
|
| 143 |
|