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 |  |