symposium 0.15.2 → 0.15.4

package/Agent.js

@@ -114,15 +114,17 @@ export default class Agent {
 		return this.execute(thread);
 	}
 
-	async beforeExecute(thread) {
+	async beforeExecute(thread, emitter) {
 		if (this.options.memory_handler)
 			thread = await this.options.memory_handler.handle(thread);
 		return thread;
 	}
 
 	async execute(thread, counter = 0, existing_emitter = null) {
+		const emitter = existing_emitter || new BufferedEventEmitter();
+
 		if (counter === 0)
-			thread = await this.beforeExecute(thread);
+			thread = await this.beforeExecute(thread, emitter);
 
 		const model = Symposium.getModelByName(thread.state.model);
 
@@ -161,10 +163,7 @@ export default class Agent {
 			console.error(e.message);
 			switch (this.type) {
 				case 'chat':
-					const emitter = existing_emitter || new BufferedEventEmitter();
 					emitter.emit('error', e.message);
-					if (!existing_emitter)
-						emitter.emit('end');
 					return emitter;
 
 				case 'utility':
@@ -177,18 +176,11 @@ export default class Agent {
 
 		try {
 			thread = await this.afterExecute(thread, completion);
-			const emitter = this.handleCompletion(thread, completion, existing_emitter);
+			const response = await this.handleCompletion(thread, completion, emitter);
 
 			switch (this.type) {
 				case 'utility':
-					return new Promise((resolve, reject) => {
-						emitter.on('data', data => {
-							if (data.type === 'response')
-								resolve(data.content);
-						});
-
-						emitter.on('error', error => reject(error));
-					});
+					return response;
 
 				case 'chat':
 					if (!existing_emitter) {
@@ -207,7 +199,7 @@ export default class Agent {
 			console.error(e);
 
 			if (counter < this.max_retries)
-				await this.execute(thread, counter + 1, existing_emitter);
+				await this.execute(thread, counter + 1, emitter);
 		}
 	}
 
@@ -306,11 +298,9 @@ export default class Agent {
 		return message;
 	}
 
-	async handleCompletion(thread, completion, existing_emitter = null) {
+	async handleCompletion(thread, completion, emitter) {
 		const model = Symposium.getModelByName(thread.state.model);
 
-		const emitter = existing_emitter || new BufferedEventEmitter();
-
 		const functions = [];
 		for (let message of completion) {
 			thread.addDirectMessage(message);
@@ -320,20 +310,13 @@ export default class Agent {
 				switch (m.type) {
 					case 'text':
 						if (this.type === 'utility') {
-							let response = null;
 							if (this.utility.type === 'text')
-								response = await this.afterHandle(thread, completion, 'return', m.content);
+								return this.afterHandle(thread, completion, 'return', m.content);
 							if (this.utility.type === 'json' && model.supports_structured_output)
-								response = await this.afterHandle(thread, completion, 'return', JSON.parse(m.content));
-
-							if (response) {
-								emitter.emit('data', {type: 'response', content: response});
-								emitter.emit('end');
-								return emitter;
-							}
+								return this.afterHandle(thread, completion, 'return', JSON.parse(m.content));
 						}
 
-						emitter.emit('data', {type: 'output', content: m.content});
+						emitter.emit('output', m.content);
 						break;
 
 					case 'function':
@@ -344,13 +327,10 @@ export default class Agent {
 			}
 		}
 
-		let response;
 		if (functions.length) {
 			for (let f of functions) {
-				if (this.utility && ['function', 'json'].includes(this.utility.type)) {
-					response = this.afterHandle(thread, completion, 'return', f.arguments);
-					break;
-				}
+				if (this.utility && ['function', 'json'].includes(this.utility.type))
+					return this.afterHandle(thread, completion, 'return', f.arguments);
 
 				const function_response = await this.callFunction(thread, f, emitter);
 
@@ -364,23 +344,11 @@ export default class Agent {
 				await this.log('function_response', function_response);
 			}
 
-			if (!response)
-				response = this.afterHandle(thread, completion, 'continue');
+			return this.afterHandle(thread, completion, 'continue');
 		} else {
 			await thread.storeState();
-			response = this.afterHandle(thread, completion, 'void');
+			return this.afterHandle(thread, completion, 'void');
 		}
-
-		response
-			.then(content => {
-				emitter.emit('data', {type: 'response', content});
-				emitter.emit('end');
-			})
-			.catch(error => {
-				emitter.emit('error', error);
-			});
-
-		return emitter;
 	}
 
 	async afterHandle(thread, completion, type, value = null) {
@@ -420,16 +388,16 @@ export default class Agent {
 
 		const func = functions.get(function_call.name);
 		const partialOutput = func.partialOutput ? ((typeof func.partialOutput) === 'text' ? func.partialOutput : func.partialOutput.call(this, function_call.arguments)) : 'Uso lo strumento ' + function_call.name + '...';
-		emitter.emit('data', {type: 'partial', content: partialOutput});
+		emitter.emit('partial', partialOutput);
 
 		await this.log('function_call', function_call);
 
 		try {
 			const response = await func.tool.callFunction(thread, function_call.name, function_call.arguments);
-			emitter.emit('data', {type: 'partial', content: 'Risposta ricevuta da ' + func.tool.name});
+			emitter.emit('partial', 'Risposta ricevuta da ' + func.tool.name);
 			return response;
 		} catch (error) {
-			emitter.emit('data', {type: 'partial', content: 'Ricevuto errore da ' + func.tool.name});
+			emitter.emit('partial', 'Ricevuto errore da ' + func.tool.name);
 			return {error};
 		}
 	}
@@ -451,6 +419,7 @@ export default class Agent {
 		return [this.name];
 	}
 
+	// Currently specific for OpenAI Realtime API
 	async createRealtimeSession(thread_id = null, options = {}) {
 		options = {
 			include_thread: true,
@@ -458,7 +427,7 @@ export default class Agent {
 			...options,
 		};
 
-		// Se viene passato un thread esistente, lo si usa, altrimenti si crea un nuovo thread temporaneo
+		// If a thread is passed, it is used, otherwise a temporary thread is created
 		const thread = await this.getThread(thread_id || uuid());
 
 		const system_message = [], conversation = [];

package/package.json

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