symposium 0.11.3 → 0.12.0

package/Agent.js

@@ -11,6 +11,7 @@ export default class Agent {
 	default_model = 'gpt-4o';
 	max_retries = 5;
 	callbacks = {};
+	utility = null;
 
 	constructor(options) {
 		this.options = {
@@ -58,7 +59,7 @@ export default class Agent {
 	async doInitThread(thread) {
 	}
 
-	async getThread(id, i) {
+	async getThread(id, i = 'default') {
 		let thread = this.threads.get(id);
 		if (thread) {
 			if (thread.interface !== i)
@@ -105,13 +106,37 @@ export default class Agent {
 		if (counter === 0)
 			thread = await this.beforeExecute(thread);
 
-		const completion = await this.generateCompletion(thread);
+		const completion_options = {};
+		if (this.utility) {
+			if (!['function', 'text'].includes(this.utility.type))
+				throw new Error('Bad utility definition');
+
+			if (this.utility.type === 'function') {
+				if (!this.utility.function || !this.utility.function.name || !this.utility.function.parameters)
+					throw new Error('Bad function definition');
+
+				completion_options.functions = [
+					this.utility.function,
+				];
+				completion_options.force_function = this.utility.function.name;
+			}
+		}
+
+		const completion = await this.generateCompletion(thread, completion_options);
 		if (completion) {
 			try {
 				thread = await this.afterExecute(thread, completion);
-				const interrupt = await this.handleCompletion(thread, completion);
-				if (!interrupt)
-					await this.execute(thread);
+				const response = await this.handleCompletion(thread, completion);
+				switch (response.type) {
+					case 'return':
+						return response.value;
+
+					case 'continue':
+						return await this.execute(thread);
+
+					default:
+						throw new Error('Unknown response type');
+				}
 			} catch (e) {
 				console.error(e);
 
@@ -161,15 +186,14 @@ export default class Agent {
 	}
 
 	async output(thread, msg) {
-		const i = this.options.interfaces.find(i => i.name === thread.interface);
-		if (i) {
-			if (this.callbacks.hasOwnProperty(i.name + '-' + thread.id) && this.callbacks[i.name + '-' + thread.id].length) {
-				const callback = this.callbacks[i.name + '-' + thread.id].shift();
-				await callback(msg);
-			}
+		if (this.callbacks.hasOwnProperty(thread.interface + '-' + thread.id) && this.callbacks[thread.interface + '-' + thread.id].length) {
+			const callback = this.callbacks[thread.interface + '-' + thread.id].shift();
+			await callback(msg);
+		}
 
+		const i = this.options.interfaces.find(i => i.name === thread.interface);
+		if (i)
 			return i.output(thread, msg);
-		}
 	}
 
 	parseFunctions(message) {
@@ -206,6 +230,9 @@ export default class Agent {
 			for (let m of message.content) {
 				switch (m.type) {
 					case 'text':
+						if (this.utility && this.utility.type === 'text')
+							return m.content;
+
 						await this.output(thread, m.content);
 						break;
 
@@ -221,6 +248,9 @@ export default class Agent {
 			for (let f of functions) {
 				const response = await this.callFunction(thread, f);
 
+				if (this.utility && this.utility.type === 'function')
+					return response;
+
 				thread.addMessage('tool', [
 					{
 						type: 'function_response',

package/package.json

@@ -1,14 +1,14 @@
 {
   "type": "module",
   "name": "symposium",
-  "version": "0.11.3",
+  "version": "0.12.0",
   "description": "Agents",
   "main": "index.js",
   "author": "Domenico Giambra",
   "license": "ISC",
   "dependencies": {
-    "@anthropic-ai/sdk": "^0.33.0",
-    "groq-sdk": "^0.12.0",
+    "@anthropic-ai/sdk": "^0.36.0",
+    "groq-sdk": "^0.13.0",
     "openai": "^4.12.1",
     "tiktoken": "^1.0.10"
   }