Update garak/generators/openrouter.py

Co-authored-by: Jeffrey Martin <[email protected]> Signed-off-by: Pedram Amini <[email protected]>
NVIDIA · Jan 6, 2025 · 2d38ced · 2d38ced
1 parent 78bb00f
commit 2d38ced
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/garak/generators/openrouter.py b/garak/generators/openrouter.py
@@ -41,9 +41,9 @@ class OpenRouterGenerator(OpenAICompatible):
     ENV_VAR = "OPENROUTER_API_KEY"
     active = True
     generator_family_name = "OpenRouter"
-    DEFAULT_PARAMS = {
-        **{k: val for k, val in OpenAICompatible.DEFAULT_PARAMS.items() if k != "uri"},
-        "max_tokens": 2000
+    DEFAULT_PARAMS = OpenAICompatible.DEFAULT_PARAMS | {
+        "uri": "https://openrouter.ai/api/v1",
+        "max_tokens": 2000,
     }
 
     def __init__(self, name="", config_root=_config):