symposium 2.0.0 → 2.0.1

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/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

@@ -47,14 +47,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 +72,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');
 

package/package.json

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