symposium 0.12.5 → 0.12.7

package/Agent.js

@@ -108,20 +108,23 @@ export default class Agent {
 
 		const completion_options = {};
 		if (this.utility) {
-			if (!['function', 'text'].includes(this.utility.type))
+			if (!['text', 'function', 'json'].includes(this.utility.type))
 				throw new Error('Bad utility definition');
 
-			if (this.utility.type === 'function') {
+			if (['function', 'json'].includes(this.utility.type)) {
 				if (!this.utility.function || !this.utility.function.name || !this.utility.function.parameters)
 					throw new Error('Bad function definition');
 
-				if (model.supports_structured_output) {
-					// TODO: se ci sono più di 100 parametri, OpenAI non supporta gli structured output
+				let response_format = null;
+				if (this.utility.type === 'json' && model.supports_structured_output)
+					response_format = this.convertFunctionToResponseFormat(this.utility.function.parameters);
+
+				if (response_format && response_format.count <= 100) { // Se ci sono più di 100 parametri, OpenAI non supporta gli structured output
 					completion_options.response_format = {
 						type: 'json_schema',
 						json_schema: {
 							name: this.utility.function.name,
-							schema: this.utility.function.parameters,
+							schema: response_format.obj,
 							strict: true,
 						},
 					};
@@ -161,6 +164,40 @@ export default class Agent {
 		}
 	}
 
+	convertFunctionToResponseFormat(obj) {
+		if (obj.type !== 'object')
+			return {obj, count: 0};
+
+		let properties_count = 0, required = [];
+
+		for (let [key, property] of Object.entries(obj.properties || {})) {
+			properties_count++;
+			required.push(key);
+
+			if (property.type === 'object') {
+				const {obj: subobj, count} = this.convertFunctionToResponseFormat(property);
+				obj.properties[key] = subobj;
+				properties_count += count;
+			} else if (property.type === 'array' && property.items.type === 'object') {
+				const {obj: subobj, count} = this.convertFunctionToResponseFormat(property.items);
+				obj.properties[key] = {
+					type: 'array',
+					items: subobj,
+				};
+				properties_count += count;
+			}
+		}
+
+		return {
+			obj: {
+				...obj,
+				additionalProperties: false,
+				required,
+			},
+			count: properties_count,
+		};
+	}
+
 	async afterExecute(thread, completion) {
 		return thread;
 	}
@@ -250,7 +287,7 @@ export default class Agent {
 						if (this.utility) {
 							if (this.utility.type === 'text')
 								return this.afterHandle(thread, completion, 'return', m.content);
-							if (this.utility.type === 'function' && model.supports_structured_output)
+							if (this.utility.type === 'json' && model.supports_structured_output)
 								return this.afterHandle(thread, completion, 'return', JSON.parse(m.content));
 						}
 						await this.output(thread, m.content);
@@ -266,7 +303,7 @@ export default class Agent {
 
 		if (functions.length) {
 			for (let f of functions) {
-				if (this.utility && this.utility.type === 'function')
+				if (this.utility && ['function', 'json'].includes(this.utility.type))
 					return this.afterHandle(thread, completion, 'return', f.arguments);
 
 				const response = await this.callFunction(thread, f);

package/models/OpenAIModel.js

@@ -62,6 +62,9 @@ export default class OpenAIModel extends Model {
 			};
 		}
 
+		if (options.response_format)
+			completion_payload.response_format = options.response_format;
+
 		if (!completion_payload.tools.length)
 			delete completion_payload.tools;
 

package/package.json

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