symposium 1.1.0 → 1.2.0

package/Agent.js

@@ -6,6 +6,8 @@ import Symposium from "./Symposium.js";
 import Thread from "./Thread.js";
 import Tool from "./Tool.js";
 import Context from "./Context.js";
+import Text from "./Contexts/Text.js";
+import GetContextTool from "./GetContextTool.js";
 
 export default class Agent {
 	name = 'Agent';
@@ -72,28 +74,56 @@ export default class Agent {
 		this.tools.set(tool.name, tool);
 	}
 
-	addContext(context) {
+	async addContext(context, options = {}) {
+		if (typeof context === 'string')
+			context = new Text(context);
 		if (!(context instanceof Context))
 			throw new Error('Context must be an instance of Context class');
 
-		// TODO: on-request contexts
+		options = {
+			type: 'always', // always, on_request
+			description: null,
+			...options,
+		};
+
 		// TODO: summarization based on tokens
 		// TODO: RAG
 
-		this.context.push(context);
+		const title = await context.getTitle();
+		this.context.push({title, context, options});
 	}
 
 	async initThread(thread) {
 		await this.doInitThread(thread);
 
-		const context_texts = [];
-		for (let context of this.context) {
-			const text = await context.getText();
-			if (text)
-				context_texts.push('<context>' + text + '</context>');
+		let context_texts = [], is_there_on_request = false;
+		for (let {title, context, options} of this.context) {
+			switch (options.type) {
+				case 'always':
+					const text = await context.getText();
+					if (text)
+						context_texts.push('<context>' + text + '</context>');
+					break;
+
+				case 'on_request':
+					is_there_on_request = true;
+					context_texts.push('<context_on_request><name>' + title + '</name><description>' + options.description + '</description></context_on_request>');
+					break;
+
+				default:
+					throw new Error('Bad context option type ' + options.type);
+			}
 		}
 
 		if (context_texts.length) {
+			let context_string = context_texts.join('\n');
+			if (is_there_on_request) {
+				context_string = '<important>Some of the context is available to you immediately here, while longer texts may be available only on request; you are provided with a title and a description. If you think it may be useful for your current task, you can request the text via the get_context tool</important>';
+				if (!this.tools.has('get_context'))
+					this.addTool(new GetContextTool(this));
+			}
+			context_string = '\n<context_info>' + context_string + '</context_info>';
+
 			let system_message_found = null;
 			for (let messages of thread.messages) {
 				if (messages.role === 'system')
@@ -101,9 +131,9 @@ export default class Agent {
 			}
 
 			if (system_message_found)
-				system_message_found.content[0].content += '<context_info>' + context_texts.join('\n') + '</context_info>';
+				system_message_found.content[0].content += context_string;
 			else
-				thread.addMessage('system', '<context_info>' + context_texts.join('\n') + '</context_info>');
+				thread.addMessage('system', context_string);
 		}
 
 		await thread.storeState();

package/Context.js

@@ -1,4 +1,11 @@
 export default class Context {
+	/**
+	 * @returns {Promise<string|null>}
+	 */
+	async getTitle() {
+		return null;
+	}
+
 	/**
 	 * @returns {Promise<string|null>}
 	 */

package/Contexts/File.js

@@ -7,6 +7,13 @@ export default class File extends Context {
 		this.file = file;
 	}
 
+	async getTitle() {
+		if (this.file.startsWith('http://') || this.file.startsWith('https://'))
+			return this.file;
+		else
+			return this.file.split('/').pop();
+	}
+
 	async getText() {
 		if (this.file.startsWith('http://') || this.file.startsWith('https://')) {
 			return fetch(this.file);

package/Contexts/Text.js

@@ -0,0 +1,17 @@
+import Context from "../Context.js";
+
+export default class Text extends Context {
+	constructor(text, title = null) {
+		super();
+		this.text = text;
+		this.title = title;
+	}
+
+	async getTitle() {
+		return this.title;
+	}
+
+	async getText() {
+		return this.text;
+	}
+}

package/GetContextTool.js

@@ -0,0 +1,42 @@
+import Tool from "./Tool.js";
+
+export default class GetContextTool extends Tool {
+	name = 'get_context';
+
+	constructor(agent) {
+		super();
+		this.agent = agent;
+	}
+
+	async getFunctions() {
+		return [
+			{
+				name: 'get_context',
+				description: 'Get the text from a specific context snippet',
+				parameters: {
+					type: 'object',
+					properties: {
+						title: {
+							type: 'string',
+						},
+					},
+					required: ['title'],
+				},
+			}
+		];
+	}
+
+	async callFunction(thread, name, payload) {
+		if (name !== 'get_context')
+			return {error: `Function ${name} not found`};
+
+		const title = payload.title;
+		const context = this.agent.context.find(c => c.title === title && c.options.type === 'on_request');
+		if (!context)
+			return {error: `Context with title ${title} not found`};
+
+		return {
+			context: await context.context.getText(),
+		};
+	}
+}

package/package.json

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