Fix tests

CharlesDuboisSAP · CharlesDuboisSAP · commit 5956cfc64436 · 2025-09-02T13:15:01.000+02:00
diff --git a/orchestration/src/test/java/com/sap/ai/sdk/orchestration/OrchestrationUnitTest.java b/orchestration/src/test/java/com/sap/ai/sdk/orchestration/OrchestrationUnitTest.java
@@ -1400,6 +1400,8 @@ void testEmbeddingCallWithMasking() {
                             "model": {
                               "name": "text-embedding-3-large",
                               "version": "latest",
+                              "timeout" : 600,
+                              "max_retries" : 2,
                               "params": {
                                 "encoding_format": "float",
                                 "dimensions": 5,
diff --git a/orchestration/src/test/resources/chatMemory.json b/orchestration/src/test/resources/chatMemory.json
@@ -5,7 +5,9 @@
         "model": {
           "name": "gpt-4o",
           "params": {},
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -31,4 +33,4 @@
   },
   "placeholder_values": {},
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/filteringLooseRequest.json b/orchestration/src/test/resources/filteringLooseRequest.json
@@ -12,7 +12,9 @@
             "top_p": 1,
             "n": 1
           },
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -63,4 +65,4 @@
   },
   "placeholder_values": {},
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/groundingHelpSapComRequest.json b/orchestration/src/test/resources/groundingHelpSapComRequest.json
@@ -12,7 +12,9 @@
             "top_p": 1,
             "n": 1
           },
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -53,4 +55,4 @@
     "userMessage": "What is a fuzzy search?"
   },
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/groundingRequest.json b/orchestration/src/test/resources/groundingRequest.json
@@ -12,7 +12,9 @@
             "top_p": 1,
             "n": 1
           },
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -94,4 +96,4 @@
     "query": "String used for similarity search in database"
   },
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/jsonObjectRequest.json b/orchestration/src/test/resources/jsonObjectRequest.json
@@ -5,7 +5,9 @@
         "model": {
           "name": "gpt-4o-mini",
           "params": {},
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -29,4 +31,4 @@
   },
   "placeholder_values": {},
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/jsonSchemaRequest.json b/orchestration/src/test/resources/jsonSchemaRequest.json
@@ -5,7 +5,9 @@
         "model": {
           "name": "gpt-4o-mini",
           "params": {},
-          "version": "latest"
+          "version": "latest",
+          "timeout": 600,
+          "max_retries": 2
         },
         "prompt": {
           "template": [
@@ -50,4 +52,4 @@
   },
   "placeholder_values": {},
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/localTemplateRequest.json b/orchestration/src/test/resources/localTemplateRequest.json
@@ -12,7 +12,9 @@
             "top_p": 1,
             "n": 1
           },
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -86,4 +88,4 @@
     "language": "German"
   },
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/maskingRequest.json b/orchestration/src/test/resources/maskingRequest.json
@@ -12,7 +12,9 @@
             "top_p": 1,
             "n": 1
           },
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -46,4 +48,4 @@
   },
   "placeholder_values": {},
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/messagesHistoryRequest.json b/orchestration/src/test/resources/messagesHistoryRequest.json
@@ -12,7 +12,9 @@
             "top_p": 1,
             "n": 1
           },
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -48,4 +50,4 @@
       "tool_calls": []
     }
   ]
-}
+}
diff --git a/orchestration/src/test/resources/multiMessageRequest.json b/orchestration/src/test/resources/multiMessageRequest.json
@@ -5,7 +5,9 @@
         "model": {
           "name": "gpt-4o-mini",
           "params": {},
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -52,4 +54,4 @@
       ]
     }
   ]
-}
+}
diff --git a/orchestration/src/test/resources/responseFormatTextRequest.json b/orchestration/src/test/resources/responseFormatTextRequest.json
@@ -5,7 +5,9 @@
         "model": {
           "name": "gpt-4o-mini",
           "params": {},
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -29,4 +31,4 @@
   },
   "placeholder_values": {},
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/templateReferenceByIdRequest.json b/orchestration/src/test/resources/templateReferenceByIdRequest.json
@@ -12,7 +12,9 @@
             "top_p": 1,
             "n": 1
           },
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template_ref": {
@@ -27,4 +29,4 @@
     "language": "Italian"
   },
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/templateReferenceByScenarioRequest.json b/orchestration/src/test/resources/templateReferenceByScenarioRequest.json
@@ -12,7 +12,9 @@
             "top_p": 1,
             "n": 1
           },
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template_ref": {
@@ -29,4 +31,4 @@
     "language": "Italian"
   },
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/templatingRequest.json b/orchestration/src/test/resources/templatingRequest.json
@@ -12,7 +12,9 @@
             "top_p": 1,
             "n": 1
           },
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -31,4 +33,4 @@
     "input": "Reply with 'Orchestration Service is working!' in German"
   },
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/toolCallsRequest.json b/orchestration/src/test/resources/toolCallsRequest.json
@@ -5,7 +5,9 @@
         "model": {
           "name": "gpt-4o",
           "params": {},
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -60,4 +62,4 @@
   },
   "placeholder_values": {},
   "messages_history": []
-}
+}
diff --git a/orchestration/src/test/resources/toolCallsRequest2.json b/orchestration/src/test/resources/toolCallsRequest2.json
@@ -5,7 +5,9 @@
         "model": {
           "name": "gpt-4o",
           "params": {},
-          "version": "latest"
+          "version": "latest",
+          "timeout" : 600,
+          "max_retries" : 2
         },
         "prompt": {
           "template": [
@@ -91,4 +93,4 @@
   },
   "placeholder_values": {},
   "messages_history": []
-}
+}
