From 2d38ced34af3ff50070bd362f0267f8a81239d0b Mon Sep 17 00:00:00 2001
From: Pedram Amini <pedram.amini@gmail.com>
Date: Mon, 6 Jan 2025 17:45:30 +0900
Subject: [PATCH] Update garak/generators/openrouter.py

Co-authored-by: Jeffrey Martin <jmartin@Op3n4M3.dev>
Signed-off-by: Pedram Amini <pedram.amini@gmail.com>
---
 garak/generators/openrouter.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/garak/generators/openrouter.py b/garak/generators/openrouter.py
index 32cad5711..6fc0ff296 100644
--- 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):