This commit is contained in:
lidia9 2025-05-08 16:44:16 +02:00
parent c464ac190a
commit 0c81fa66c5
2 changed files with 1 additions and 1 deletions

View File

@ -49,6 +49,7 @@ import requests
# Default API bases
OPENROUTER_API_BASE = "https://openrouter.ai/api/v1"
OLLAMA_LITELLM_API_BASE = "http://localhost:8000/v1"
os.environ["OPENAI_API_KEY"] = "test_key_for_ci_environment"
class CyberMetricEvaluator:
def __init__(self, model_name, file_path, api_key=None, openrouter_api_base=None, ollama_litellm_api_base=None):

View File

@ -44,7 +44,6 @@ benchmarks-test-cybermetric:
variables:
OLLAMA_API_BASE: "http://localhost:8000"
OPENROUTER_API_BASE: "https://openrouter.ai/api/v1"
OPENAI_API_KEY: "fake-api-key"
script:
- pip3 install -e .
- python3 benchmarks/cybermetric/CyberMetric_evaluator.py --model_name ollama/qwen2.5:14b --file_path benchmarks/cybermetric/CyberMetric-2-v1.json