Adding signed byte support from cohere integration (#2398)

elastic · Jan 30, 2024 · 203108f · 203108f
1 parent e66a88c
commit 203108f
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 12 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/_types/Results.ts b/specification/inference/_types/Results.ts
@@ -17,7 +17,7 @@
  * under the License.
  */
 
-import { float } from '@_types/Numeric'
+import { float, byte } from '@_types/Numeric'
 import { Dictionary } from '@spec_utils/Dictionary'
 
 /**
@@ -28,9 +28,9 @@ export type SparseVector = Dictionary<string, float>
 
 /**
  * Text Embedding results are represented as Dense Vectors
- * of floats.
+ * of floats or signed bytes.
  */
-export type DenseVector = Array<float>
+export type DenseVector = Array<float> | Array<byte>
 
 export class SparseEmbeddingResult {
   embedding: SparseVector