Update garak/generators/watsonx.py

Added version parameter Co-authored-by: Jeffrey Martin <[email protected]> Signed-off-by: iamnotcj <[email protected]>
NVIDIA · Jan 30, 2025 · 4b252fe · 4b252fe
1 parent fefcd5e
commit 4b252fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/garak/generators/watsonx.py b/garak/generators/watsonx.py
@@ -53,7 +53,7 @@ def _set_bearer_token(self, iam_url="https://iam.cloud.ibm.com/identity/token"):
     def _generate_with_project(self, payload):
         # Generation via Project ID.
 
-        url = self.uri + "/ml/v1/text/generation?version=2023-05-29"
+        url = self.uri + f"/ml/v1/text/generation?version={self.version}"
 
         body = {
             "input": payload,