symposium 0.5.4 → 0.5.6

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/Symposium.js

@@ -41,7 +41,7 @@ export default class Symposium {
 		for (let message of messages) {
 			const functionResponse = message.content.filter(c => c.type === 'function');
 			if (functionResponse.length)
-				return functionResponse[0].arguments;
+				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});
@@ -26,7 +20,7 @@ export default class OpenAIModel extends Model {
 
 		if (functions.length && !this.supports_functions) {
 			// Se il modello non supporta nativamente le funzioni, inserisco il prompt ad hoc come ultimo messaggio di sistema
-			const functions_prompt = this.promptFromFunctions(functions);
+			const functions_prompt = this.promptFromFunctions(payload, functions);
 			let system_messages = [], other_messages = [], first_found = false;
 			for (let message of messages) {
 				if (!first_found && message.role !== 'system')
@@ -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.4",
+  "version": "0.5.6",
   "description": "Agents",
   "main": "index.js",
   "author": "Domenico Giambra",