symposium 0.1.0 → 0.1.2

package/Agent.js

@@ -1,11 +1,10 @@
-import OpenAI from "openai";
+import {Symposium} from "./Symposium.js";
 import {Conversation} from "./Conversation.js";
 
 class Agent {
 	name = 'Agent';
 	descriptionForFront = '';
 	options = {};
-	openai;
 	conversations;
 	functions = null;
 	middlewares = new Map();
@@ -99,9 +98,6 @@ class Agent {
 			}
 		});
 
-		this.openai = new OpenAI({
-			apiKey: process.env.OPENAI_API_KEY,
-		});
 		this.conversations = new Map();
 	}
 
@@ -258,7 +254,8 @@ class Agent {
 					delete completion_payload.function_call;
 			}
 
-			const chatCompletion = await this.openai.chat.completions.create(completion_payload);
+			const openai = await Symposium.getOpenAi();
+			const chatCompletion = await openai.chat.completions.create(completion_payload);
 
 			let completion = chatCompletion.choices[0].message;
 			if (completion.function_call && completion.function_call.arguments)
@@ -355,17 +352,6 @@ class Agent {
 	async getPromptWordsForTranscription(conversation) {
 		return [this.name];
 	}
-
-	async transcribe(file, conversation) {
-		let words = await this.getPromptWordsForTranscription(conversation);
-
-		let response = await this.openai.audio.transcriptions.create({
-			file,
-			model: 'whisper-1',
-			prompt: words.join(', '),
-		});
-		return response.text;
-	}
 }
 
 export {Agent};

package/Summarizer.js

@@ -9,11 +9,11 @@ class Summarizer extends MemoryHandler {
 	}
 
 	async handle(conversation) {
-		const model = this.agent.models.find(model => model.key === conversation.state.model);
+		const model = this.agent.models.find(model => model.name === conversation.state.model);
 		if (!model)
 			return conversation;
 
-		const encoder = encoding_for_model(model.base_model);
+		const encoder = encoding_for_model(model.name);
 		const tokens = this.countTokens(encoder, conversation);
 		if (tokens >= model.tokens * this.threshold)
 			return await this.summarize(encoder, conversation, model.tokens * this.summary_length);

package/Symposium.js

@@ -0,0 +1,30 @@
+import Redis from "@travio/redis";
+import OpenAI from "openai";
+
+class Symposium {
+	static openai;
+
+	static async init() {
+		return Redis.init();
+	}
+
+	static async getOpenAi() {
+		if (!this.openai)
+			this.openai = new OpenAI({apiKey: process.env.OPENAI_API_KEY});
+
+		return this.openai;
+	}
+
+	static async transcribe(agent, file, conversation) {
+		let words = await agent.getPromptWordsForTranscription(conversation);
+
+		let response = await this.openai.audio.transcriptions.create({
+			file,
+			model: 'whisper-1',
+			prompt: words.join(', '),
+		});
+		return response.text;
+	}
+}
+
+export {Symposium};

package/index.js

@@ -1,5 +1,6 @@
 "use strict";
 
+import {Symposium} from "./Symposium.js";
 import {Agent} from "./Agent.js";
 import {Conversation} from "./Conversation.js";
 import {Message} from "./Message.js";
@@ -10,6 +11,7 @@ import {Middleware} from "./Middleware.js";
 import {Summarizer} from "./Summarizer.js";
 
 export {
+	Symposium,
 	Agent,
 	Conversation,
 	Message,

package/package.json

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