symposium 0.1.2 → 0.1.4

package/Agent.js

@@ -20,19 +20,6 @@ class Agent {
 		if (this.options.memory_handler)
 			this.options.memory_handler.setAgent(this);
 
-		this.models = [
-			{
-				label: 'gpt-3.5',
-				name: 'gpt-3.5-turbo-16k',
-				tokens: 16384,
-			},
-			{
-				label: 'gpt-4',
-				name: 'gpt-4',
-				tokens: 8192
-			}
-		];
-
 		this.commands = new Map();
 
 		this.commands.set('start', {
@@ -48,15 +35,15 @@ class Agent {
 			show_in_help: true,
 			exec: async (conversation, args) => {
 				if (args) {
-					const model_to_switch = this.getModelFromLabel(args);
+					const model_to_switch = Symposium.getModelByLabel(args);
 					if (model_to_switch) {
 						await conversation.setState({model: model_to_switch.name});
 						await conversation.reply('# Da ora in poi uso ' + model_to_switch.label + '!');
 					} else {
-						await conversation.reply("# Versione modello non riconosciuta!\nModelli disponibili:\n" + this.models.map(m => m.label).join("\n"));
+						await conversation.reply("# Versione modello non riconosciuta!\nModelli disponibili:\n" + Symposium.models.map(m => m.label).join("\n"));
 					}
 				} else {
-					const currentModel = this.getModelFromName(conversation.state.model);
+					const currentModel = Symposium.getModelByName(conversation.state.model);
 					await conversation.reply('# Il modello attualmente in uso è ' + currentModel.label);
 				}
 			},
@@ -112,14 +99,6 @@ class Agent {
 		await conversation.storeState();
 	}
 
-	getModelFromLabel(label) {
-		return this.models.find(model => model.label === label);
-	}
-
-	getModelFromName(name) {
-		return this.models.find(model => model.name === name);
-	}
-
 	async resetState(conversation) {
 		conversation.state = await this.getDefaultState();
 	}

package/Model.js

@@ -0,0 +1,13 @@
+class Model {
+	name;
+	label;
+	tokens;
+
+	constructor(name, label, tokens) {
+		this.name = name;
+		this.label = label;
+		this.tokens = tokens;
+	}
+}
+
+export {Model};

package/Summarizer.js

@@ -1,3 +1,4 @@
+import {Symposium} from "./Symposium.js";
 import {MemoryHandler} from "./MemoryHandler.js";
 import {encoding_for_model} from "tiktoken";
 
@@ -9,7 +10,7 @@ class Summarizer extends MemoryHandler {
 	}
 
 	async handle(conversation) {
-		const model = this.agent.models.find(model => model.name === conversation.state.model);
+		const model = Symposium.getModelByName(conversation.state.model);
 		if (!model)
 			return conversation;
 

package/Symposium.js

@@ -1,13 +1,23 @@
 import Redis from "@travio/redis";
 import OpenAI from "openai";
+import {Model} from "./Model.js";
 
 class Symposium {
 	static openai;
+	static models = [];
 
 	static async init() {
+		this.loadModel(new Model('gpt-3.5-turbo-16k', 'gpt-3.5', 16384));
+		this.loadModel(new Model('gpt-4', 'gpt-4', 8192));
+		this.loadModel(new Model('gpt-4-1106-preview', 'gpt-4-turbo', 128000));
+
 		return Redis.init();
 	}
 
+	static loadModel(model) {
+		this.models.push(model);
+	}
+
 	static async getOpenAi() {
 		if (!this.openai)
 			this.openai = new OpenAI({apiKey: process.env.OPENAI_API_KEY});
@@ -15,6 +25,14 @@ class Symposium {
 		return this.openai;
 	}
 
+	static getModelByLabel(label) {
+		return this.models.find(model => model.label === label);
+	}
+
+	static getModelByName(name) {
+		return this.models.find(model => model.name === name);
+	}
+
 	static async transcribe(agent, file, conversation) {
 		let words = await agent.getPromptWordsForTranscription(conversation);
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "type": "module",
   "name": "symposium",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "Agents",
   "main": "index.js",
   "scripts": {