Support arrays input for OpenAI embeddings (#6)

Author: mattjohnsonpint

CHANGELOG.md

@@ -2,4 +2,16 @@
 
 ## UNRELEASED
 
+- Update OpenAI Embeddings model to support all allowed types of input
+
+## 2024-06-28 - Version 0.1.6
+
+- Interface changes and fixes
+
+## 2024-06-28 - Version 0.1.5
+
+- Interface changes and fixes
+
+## 2024-06-27 - Version 0.1.2
+
 - Initial Release

src/models/openai/embeddings.ts

@@ -3,16 +3,38 @@ import { Model } from "../..";
 // Reference: https://platform.openai.com/docs/api-reference/embeddings
 
 export class EmbeddingsModel extends Model<EmbeddingsInput, EmbeddingsOutput> {
-  createInput(text: string): EmbeddingsInput {
+  createInput<T>(content: T): EmbeddingsInput {
     const model = this.info.fullName;
-    return <EmbeddingsInput>{ model, input: text };
+
+    switch (idof<T>()) {
+      case idof<string>():
+      case idof<string[]>():
+      case idof<i64[]>():
+      case idof<i32[]>():
+      case idof<i16[]>():
+      case idof<i8[]>():
+      case idof<u64[]>():
+      case idof<u32[]>():
+      case idof<u16[]>():
+      case idof<u8[]>():
+      case idof<i64[][]>():
+      case idof<i32[][]>():
+      case idof<i16[][]>():
+      case idof<i8[][]>():
+      case idof<u64[][]>():
+      case idof<u32[][]>():
+      case idof<u16[][]>():
+      case idof<u8[][]>():
+        return <TypedEmbeddingsInput<T>>{ model, input: content };
+    }
+
+    throw new Error("Unsupported input content type.");
   }
 }
 
 
 @json
 class EmbeddingsInput {
-  input!: string; // todo: support other types of input (arrays, etc.)
   model!: string;
 
 
@@ -29,6 +51,12 @@ class EmbeddingsInput {
 }
 
 
+@json
+class TypedEmbeddingsInput<T> extends EmbeddingsInput {
+  input!: T;
+}
+
+
 @json
 class EmbeddingsOutput {
   object!: string;