Add Completion result and task type to the Inference API spec (#2471)

elastic · Mar 28, 2024 · 8b12f48 · 8b12f48
1 parent 071b2f7
commit 8b12f48
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 1 deletion.
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/_types/Results.ts b/specification/inference/_types/Results.ts
@@ -56,6 +56,13 @@ export class TextEmbeddingResult {
   embedding: DenseVector
 }
 
+/**
+ * The completion result object
+ */
+export class CompletionResult {
+  result: string
+}
+
 /**
  * InferenceResult is an aggregation of mutually exclusive variants
  * @variants container
@@ -64,4 +71,5 @@ export class InferenceResult {
   text_embedding_bytes?: Array<TextEmbeddingByteResult>
   text_embedding?: Array<TextEmbeddingResult>
   sparse_embedding?: Array<SparseEmbeddingResult>
+  completion?: Array<CompletionResult>
 }