symposium 2.0.1 → 2.1.0

package/Models/OpenAIEmbedding.js

@@ -0,0 +1,18 @@
+import OpenAIModel from "./OpenAIModel.js";
+
+export default class OpenAIEmbedding extends OpenAIModel {
+	type = 'embedding';
+	models = new Map([
+		['openai-text-embedding-3-large', {name: 'text-embedding-3-large'}],
+		['openai-text-embedding-3-small', {name: 'text-embedding-3-small'}],
+	]);
+
+	async embed(input, model) {
+		const response = await this.getOpenAi().embeddings.create({
+			model: model.name,
+			input,
+		});
+
+		return response.data[0].embedding;
+	}
+}

package/Symposium.js

@@ -39,6 +39,9 @@ export default class Symposium {
 
 	static loadModel(model_class) {
 		for (let [key, model] of model_class.models.entries()) {
+			if (this.models.has(key))
+				throw new Error(`Duplicate model with key "${key}"`);
+
 			this.models.set(key, {
 				...model,
 				type: model_class.type,
@@ -103,6 +106,18 @@ export default class Symposium {
 		return this.transcription_model.transcribe(file, model, prompt);
 	}
 
+	static async embed(input, model = null) {
+		model = model || process.env.EMBEDDING_MODEL;
+		if (!model)
+			throw new Error('Embedding model not specified');
+
+		const embedding_model = Symposium.getModel(model);
+		if (!embedding_model || embedding_model.type !== 'embedding')
+			throw new Error('Specified model is not an embedding model');
+
+		return embedding_model.embed(input, model);
+	}
+
 	static getExtFromMime(mime) {
 		const mimeToExt = {
 			'audio/mpeg': 'mp3',

package/package.json

@@ -1,7 +1,7 @@
 {
   "type": "module",
   "name": "symposium",
-  "version": "2.0.1",
+  "version": "2.1.0",
   "description": "Agents",
   "main": "index.js",
   "author": "Domenico Giambra",