[ML] Adding timeout field (#2501)

* Adding timeout field * fixing formatting
elastic · Apr 11, 2024 · b14d22e · b14d22e
1 parent 09e00d5
commit b14d22e
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 2 deletions.
diff --git a/output/schema/schema.json b/output/schema/schema.json
diff --git a/output/typescript/types.ts b/output/typescript/types.ts
diff --git a/specification/inference/inference/InferenceRequest.ts b/specification/inference/inference/InferenceRequest.ts
@@ -17,6 +17,7 @@
  * under the License.
  */
 
+import { Duration } from '@_types/Time'
 import { RequestBase } from '@_types/Base'
 import { Id } from '@_types/common'
 import { TaskType } from '@inference/_types/TaskType'
@@ -39,6 +40,13 @@ export interface Request extends RequestBase {
      */
     inference_id: Id
   }
+  query_parameters: {
+    /**
+     * Specifies the amount of time to wait for the inference request to complete.
+     * @server_default 30s
+     */
+    timeout?: Duration
+  }
   body: {
     /**
      * Query input, required for rerank task.