symposium 0.3.4 → 0.3.5

package/Agent.js

@@ -119,10 +119,10 @@ export default class Agent {
 		}
 	}
 
-	async generateCompletion(thread, retry_counter = 1) {
+	async generateCompletion(thread, payload = {}, retry_counter = 1) {
 		try {
 			const model = Symposium.getModelByName(thread.state.model);
-			return model.generate(thread, await this.getFunctions())
+			return model.generate(thread, payload, await this.getFunctions())
 		} catch (error) {
 			if (error.response) {
 				console.error(error.response.status);
@@ -133,7 +133,7 @@ export default class Agent {
 						setTimeout(resolve, 1000);
 					});
 
-					return this.generateCompletion(thread, retry_counter + 1);
+					return this.generateCompletion(thread, payload, retry_counter + 1);
 				}
 
 				await thread.reply('# Errore ' + error.response.status + ': ' + JSON.stringify(error.response.data));
@@ -206,6 +206,14 @@ export default class Agent {
 		await this.execute(thread);
 	}
 
+	async setModel(thread, label) {
+		const model_to_switch = Symposium.getModelByLabel(label);
+		if (model_to_switch && model_to_switch.type === 'llm')
+			await thread.setState({model: model_to_switch.name});
+		else
+			throw new Error("Versione modello non riconosciuta!\nModelli disponibili:\n" + Array.from(Symposium.models.values()).filter(m => m.type === 'llm').map(m => m.label).join("\n"));
+	}
+
 	async log(type, payload) {
 		if (this.options.logger)
 			return this.options.logger.log(this.name, type, payload);

package/Model.js

@@ -1,5 +1,6 @@
 export default class Model {
 	vendor;
+	type = 'llm';
 	name;
 	name_for_tiktoken;
 	label;

package/Symposium.js

@@ -1,9 +1,9 @@
 import Redis from "@travio/redis";
-import OpenAI from "openai";
 import Gpt35 from "./models/Gpt35.js";
 import Gpt4 from "./models/Gpt4.js";
 import Gpt4Turbo from "./models/Gpt4Turbo.js";
 import Gpt4Vision from "./models/Gpt4Vision.js";
+import Whisper from "./models/Whisper.js";
 
 export default class Symposium {
 	static models = new Map();
@@ -13,6 +13,7 @@ export default class Symposium {
 		this.loadModel(new Gpt4());
 		this.loadModel(new Gpt4Turbo());
 		this.loadModel(new Gpt4Vision());
+		this.loadModel(new Whisper());
 
 		return Redis.init();
 	}
@@ -22,21 +23,10 @@ export default class Symposium {
 	}
 
 	static getModelByName(name) {
-		return this.models.find(model => model.name === name);
+		return this.models.get(name);
 	}
 
 	static getModelByLabel(label) {
-		return this.models.find(model => model.label === label);
+		return Array.from(this.models.values()).find(model => model.label === label);
 	}
-
-	/*static async transcribe(agent, file, thread) {
-		const words = await agent.getPromptWordsForTranscription(thread);
-
-		const response = await this.getOpenAi().then(openai => openai.audio.transcriptions.create({
-			file,
-			model: 'whisper-1',
-			prompt: words.join(', '),
-		}));
-		return response.text;
-	}*/
 }

package/models/OpenAIModel.js

@@ -13,11 +13,12 @@ export default class OpenAIModel extends Model {
 		return this.openai;
 	}
 
-	async generate(thread, functions = []) {
+	async generate(thread, payload = {}, functions = []) {
 		const completion_payload = {
 			model: this.name,
 			messages: thread.getMessagesJson(),
 			functions,
+			...payload,
 		};
 
 		if (!completion_payload.functions?.length) {

package/models/Whisper.js

@@ -0,0 +1,16 @@
+import OpenAIModel from "./OpenAIModel.js";
+
+export default class Whisper extends OpenAIModel {
+	type = 'stt';
+
+	async transcribe(agent, thread, file) {
+		const words = await agent.getPromptWordsForTranscription(thread);
+
+		const response = await this.getOpenAi().then(openai => openai.audio.transcriptions.create({
+			file,
+			model: 'whisper-1',
+			prompt: words.join(', '),
+		}));
+		return response.text;
+	}
+}

package/package.json

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