symposium 2.0.0 → 2.1.0

package/Agent.js

@@ -58,7 +58,7 @@ export default class Agent {
 
 	async resetState(thread) {
 		thread.state = await this.getDefaultState();
-		thread.state.model = Symposium.getModelByLabel(this.default_model).name;
+		thread.state.model = this.default_model;
 	}
 
 	async getDefaultState() {
@@ -171,7 +171,7 @@ export default class Agent {
 		if (typeof thread !== 'object')
 			thread = await this.getThread(thread);
 
-		const model = Symposium.getModelByName(thread.state.model);
+		const model = Symposium.getModel(thread.state.model);
 		if (!model.audio && typeof content !== 'string') {
 			for (let c of content) {
 				if (c.type === 'audio' && !c.content?.transcription) {
@@ -203,7 +203,7 @@ export default class Agent {
 				if (counter === 0)
 					thread = await this.beforeExecute(thread, emitter);
 
-				const model = Symposium.getModelByName(thread.state.model);
+				const model = Symposium.getModel(thread.state.model);
 
 				const completion_options = {};
 				if (this.type === 'utility') {
@@ -340,7 +340,7 @@ export default class Agent {
 
 	async generateCompletion(thread, options = {}, retry_counter = 1) {
 		try {
-			const model = Symposium.getModelByName(thread.state.model);
+			const model = Symposium.getModel(thread.state.model);
 			const messages = await model.class.generate(model, thread, await this.getFunctions(), options);
 			return model.tools ? messages : messages.map(m => this.parseFunctions(m));
 		} catch (error) {
@@ -391,7 +391,7 @@ export default class Agent {
 	}
 
 	async handleCompletion(thread, completion, emitter) {
-		const model = Symposium.getModelByName(thread.state.model);
+		const model = Symposium.getModel(thread.state.model);
 
 		const functions = [];
 		for (let message of completion) {
@@ -542,7 +542,7 @@ export default class Agent {
 	}
 
 	async setModel(thread, label) {
-		const model_to_switch = Symposium.getModelByLabel(label);
+		const model_to_switch = Symposium.getModel(label);
 		if (model_to_switch && model_to_switch.type === 'llm')
 			await thread.setState({model: model_to_switch.name});
 		else

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/Summarizer.js

@@ -9,7 +9,7 @@ export default class Summarizer extends ContextHandler {
 	}
 
 	async handle(thread) {
-		const model = Symposium.getModelByName(thread.state.model);
+		const model = Symposium.getModel(thread.state.model);
 		if (!model)
 			return thread;
 

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,
@@ -47,14 +50,10 @@ export default class Symposium {
 		}
 	}
 
-	static getModelByName(name) {
+	static getModel(name) {
 		return this.models.get(name);
 	}
 
-	static getModelByLabel(label) {
-		return Array.from(this.models.values()).find(model => model.label === label);
-	}
-
 	static extractFunctionsFromResponse(messages) {
 		const functions = [];
 		for (let message of messages) {
@@ -76,7 +75,7 @@ export default class Symposium {
 			throw new Error('Transcription model not specified');
 
 		if (!this.transcription_model)
-			this.transcription_model = Symposium.getModelByName(model);
+			this.transcription_model = Symposium.getModel(model);
 		if (!this.transcription_model.type !== 'stt')
 			throw new Error('Specified model is not a transcription model');
 
@@ -107,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.0",
+  "version": "2.1.0",
   "description": "Agents",
   "main": "index.js",
   "author": "Domenico Giambra",