symposium 0.12.3 → 0.12.5

package/Agent.js

@@ -104,6 +104,8 @@ export default class Agent {
 		if (counter === 0)
 			thread = await this.beforeExecute(thread);
 
+		const model = Symposium.getModelByName(thread.state.model);
+
 		const completion_options = {};
 		if (this.utility) {
 			if (!['function', 'text'].includes(this.utility.type))
@@ -113,10 +115,22 @@ export default class Agent {
 				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;
+				if (model.supports_structured_output) {
+					// TODO: 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,
+							strict: true,
+						},
+					};
+				} else {
+					completion_options.functions = [
+						this.utility.function,
+					];
+					completion_options.force_function = this.utility.function.name;
+				}
 			}
 		}
 
@@ -223,6 +237,8 @@ export default class Agent {
 	}
 
 	async handleCompletion(thread, completion) {
+		const model = Symposium.getModelByName(thread.state.model);
+
 		const functions = [];
 		for (let message of completion) {
 			thread.addDirectMessage(message);
@@ -231,9 +247,12 @@ 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);
-
+						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)
+								return this.afterHandle(thread, completion, 'return', JSON.parse(m.content));
+						}
 						await this.output(thread, m.content);
 						break;
 

package/Model.js

@@ -4,6 +4,7 @@ export default class Model {
 	label;
 	tokens;
 	supports_functions = false;
+	supports_structured_output = false;
 
 	constructor() {
 		if (!this.label)

package/Symposium.js

@@ -2,6 +2,8 @@ import Gpt35 from "./models/Gpt35.js";
 import Gpt4 from "./models/Gpt4.js";
 import Gpt4Turbo from "./models/Gpt4Turbo.js";
 import Gpt4O from "./models/Gpt4O.js";
+import GptO1 from "./models/GptO1.js";
+import GptO1Mini from "./models/GptO1Mini.js";
 import Whisper from "./models/Whisper.js";
 import Claude3Haiku from "./models/Claude3Haiku.js";
 import Claude3Sonnet from "./models/Claude3Sonnet.js";
@@ -9,6 +11,8 @@ import Claude3Opus from "./models/Claude3Opus.js";
 import Claude35Sonnet from "./models/Claude35Sonnet.js";
 import Llama3 from "./models/Llama3.js";
 import Mixtral8 from "./models/Mixtral8.js";
+import DeepSeekChat from "./models/DeepSeekChat.js";
+import DeepSeekReasoner from "./models/DeepSeekReasoner.js";
 
 export default class Symposium {
 	static models = new Map();
@@ -25,6 +29,8 @@ export default class Symposium {
 		this.loadModel(new Gpt4());
 		this.loadModel(new Gpt4Turbo());
 		this.loadModel(new Gpt4O());
+		this.loadModel(new GptO1());
+		this.loadModel(new GptO1Mini());
 		this.loadModel(new Whisper());
 
 		this.loadModel(new Claude3Haiku());
@@ -36,6 +42,9 @@ export default class Symposium {
 		this.loadModel(new Llama3());
 		this.loadModel(new Mixtral8());
 
+		this.loadModel(new DeepSeekChat());
+		this.loadModel(new DeepSeekReasoner());
+
 		this.storage = storage;
 		await this.storage.init();
 	}

package/models/DeepSeekChat.js

@@ -0,0 +1,7 @@
+import DeepSeekModel from "./DeepSeekModel.js";
+
+export default class DeepSeekChat extends DeepSeekModel {
+	name = 'deepseek-chat';
+	label = 'deepseek-chat';
+	tokens = 64000;
+}

package/models/DeepSeekModel.js

@@ -0,0 +1,15 @@
+import OpenAIModel from "./OpenAIModel.js";
+import OpenAI from "openai";
+
+export default class DeepSeekModel extends OpenAIModel {
+	getOpenAi() {
+		if (!this.openai) {
+			this.openai = new OpenAI({
+				baseURL: 'https://api.deepseek.com',
+				apiKey: process.env.DEEPSEEK_API_KEY,
+			});
+		}
+
+		return this.openai;
+	}
+}

package/models/DeepSeekReasoner.js

@@ -0,0 +1,7 @@
+import DeepSeekModel from "./DeepSeekModel.js";
+
+export default class DeepSeekReasoner extends DeepSeekModel {
+	name = 'deepseek-reasoner';
+	label = 'deepseek-reasoner';
+	tokens = 64000;
+}

package/models/Gpt4O.js

@@ -5,4 +5,5 @@ export default class Gpt4O extends OpenAIModel {
 	label = 'gpt-4o';
 	name_for_tiktoken = 'gpt-4';
 	tokens = 128000;
+	supports_structured_output = true;
 }

package/models/GptO1.js

@@ -0,0 +1,9 @@
+import OpenAIModel from "./OpenAIModel.js";
+
+export default class GptO1 extends OpenAIModel {
+	name = 'o1';
+	label = 'gpt-o1';
+	name_for_tiktoken = 'o1-mini';
+	tokens = 200000;
+	supports_structured_output = true;
+}

package/models/GptO1Mini.js

@@ -0,0 +1,9 @@
+import OpenAIModel from "./OpenAIModel.js";
+
+export default class GptO1Mini extends OpenAIModel {
+	name = 'o1-mini';
+	label = 'gpt-o1-mini';
+	name_for_tiktoken = 'o1-mini';
+	tokens = 128000;
+	supports_structured_output = true;
+}

package/package.json

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