symposium 0.5.3 → 0.5.5

package/Agent.js

@@ -179,7 +179,7 @@ export default class Agent {
 		let call_function = null;
 		for (let message of completion) {
 			thread.addDirectMessage(message);
-			await this.log('ai_message', message);
+			await this.log('ai_message', message.content);
 
 			for (let m of message.content) {
 				switch (m.type) {

package/Summarizer.js

@@ -78,11 +78,11 @@ export default class Summarizer extends MemoryHandler {
 			if (message.role === 'system' && !message.tags.includes('summary')) {
 				summarizedThread.messages.push(message);
 			} else {
-				const functionResponse = summary.content.filter(c => c.type === 'function');
-				if (!functionResponse.length)
+				const functionResponse = Symposium.extractFunctionFromResponse(summary);
+				if (functionResponse)
 					throw new Error('Errore durante la generazione di un riassunto interno');
 
-				summarizedThread.addMessage('system', "This is what happened until now:\n" + functionResponse[0].arguments.summary, undefined, ['summary']);
+				summarizedThread.addMessage('system', "This is what happened until now:\n" + functionResponse.summary, undefined, ['summary']);
 				break;
 			}
 		}

package/Symposium.js

@@ -36,4 +36,14 @@ export default class Symposium {
 	static getModelByLabel(label) {
 		return Array.from(this.models.values()).find(model => model.label === label);
 	}
+
+	static extractFunctionFromResponse(messages) {
+		for (let message of messages) {
+			const functionResponse = message.content.filter(c => c.type === 'function');
+			if (functionResponse.length)
+				return functionResponse[0].content.arguments;
+		}
+
+		return null;
+	}
 }

package/models/OpenAIModel.js

@@ -8,12 +8,6 @@ export default class OpenAIModel extends Model {
 	name_for_tiktoken;
 	supports_functions = true;
 
-	constructor() {
-		super();
-		if (!this.name_for_tiktoken)
-			this.name_for_tiktoken = this.name;
-	}
-
 	getOpenAi() {
 		if (!this.openai)
 			this.openai = new OpenAI({apiKey: process.env.OPENAI_API_KEY});
@@ -83,12 +77,16 @@ export default class OpenAIModel extends Model {
 	}
 
 	async countTokens(thread) {
-		const encoder = encoding_for_model(this.name_for_tiktoken);
-
-		const texts = [];
-		for (let message of thread.messages)
-			texts.push(message.content.map(m => typeof m.content === 'string' ? m.content : JSON.stringify(m.content)).join(''));
-		return encoder.encode(texts.join('')).length;
+		try {
+			const encoder = encoding_for_model(this.name_for_tiktoken || this.name);
+
+			const texts = [];
+			for (let message of thread.messages)
+				texts.push(message.content.map(m => typeof m.content === 'string' ? m.content : JSON.stringify(m.content)).join(''));
+			return encoder.encode(texts.join('')).length;
+		} catch (e) {
+			throw new Error('Error while counting tokens');
+		}
 	}
 
 	convertMessage(message) {

package/package.json

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