symposium 0.11.4 → 0.12.1

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 = {
@@ -105,13 +106,40 @@ 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);
+
+					case 'void':
+						return;
+
+					default:
+						throw new Error('Unknown response type');
+				}
 			} catch (e) {
 				console.error(e);
 
@@ -205,6 +233,9 @@ export default class Agent {
 			for (let m of message.content) {
 				switch (m.type) {
 					case 'text':
+						if (this.utility && this.utility.type === 'text')
+							return this.afterHandle(thread, completion, 'return', m.content);
+
 						await this.output(thread, m.content);
 						break;
 
@@ -220,6 +251,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 this.afterHandle(thread, completion, 'return', response);
+
 				thread.addMessage('tool', [
 					{
 						type: 'function_response',
@@ -230,15 +264,18 @@ export default class Agent {
 				await this.log('function_response', response);
 			}
 
-			return this.afterHandle(thread, completion, true);
+			return this.afterHandle(thread, completion, 'continue');
 		} else {
 			await thread.storeState();
-			return this.afterHandle(thread, completion, false);
+			return this.afterHandle(thread, completion, 'void');
 		}
 	}
 
-	async afterHandle(thread, completion, executed_function) {
-		return !executed_function;
+	async afterHandle(thread, completion, type, value = null) {
+		return {
+			type,
+			value,
+		};
 	}
 
 	async getFunctions(parsed = true) {

package/package.json

@@ -1,14 +1,14 @@
 {
   "type": "module",
   "name": "symposium",
-  "version": "0.11.4",
+  "version": "0.12.1",
   "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"
   }