@j-o-r/hello-dave 0.0.0

package/lib/Cli.js

@@ -0,0 +1,256 @@
+import cli from '@j-o-r/cli';
+import { SH } from '@j-o-r/sh';
+// import audio2text from './tts/openai.com/generate.js';
+/**
+* @typedef {import('./API/openai.com/reponses/text.js').request} OARequest
+* @typedef {import('./API/x.ai/text.js').request} XRequest
+* @typedef {import('./API/anthropic.com/text.js').request} ANTHRequest
+* 
+* @typedef {import('./API/x.ai/text.js').XOptions} XOptions
+* @typedef {import('./API/openai.com/reponses/text.js').OAOptions} OAOptions
+* @typedef {import('./API/anthropic.com/text.js').ANTHOptions} ANTHOptions
+*
+* @typedef {import('./Prompt.js').default} Prompt
+* @typedef {import('./Session.js').default} Session
+* @typedef {import('./ToolSet.js').default} ToolSet
+*/
+/**
+ * @typedef {Object} CLIOptions
+ * @property {Prompt} prompt - The prompt session
+ * @property {Session} session - Session  Cache
+ * @property {ToolSet} [toolset] - The toolset
+ * @property {string} [description] - Custom introduction message.
+ * @property {string} [help] - Custom help message.
+ */
+
+
+const INTRO = `
+Use:
+  - "ALT + d" to exit.
+  - "ALT + k" Show shortcuts /help
+`;
+
+const ERROR = `
+`
+
+const HELP = `
+
+Use:
+  - "ALT + d" to exit.
+  - "ALT + r" to reset and start a new conversation.
+  - "ALT + s" to load a previous session.
+  - "ALT + k" Show shortcuts keys /help
+  - "ALT + i" Show info about the current session and prompt
+
+prompt:
+
+Special prompt commands for editing your prompt:
+Commands are placed between ">" and "<" characters and are only parsed when on a single line.
+
+e.g. >edit< , >paste< , >#!bash [command]<
+
+ > "Edit this text >edit<" : Opens an editor with the text before submitting 
+ > ">edit<" : Opens an editor to formulate your question 
+ > "Text with .. >paste< more text" : Opens an editor with the clipboard as content for further editing 
+ > ">paste<" : Opens an editor with the clipboard as content for further editing 
+ > "Question  >#!bash ls -Fla< remark" : Executes a bash command and fill the editor with the response 
+`;
+
+/**
+* Copy text to the clipboard
+* @param {string} text
+* @retruns {Promise<string>}
+*/
+const copyToClipboard = async (text) => {
+	if (typeof text !== 'string') return;
+	if (text === '') text = ' ';
+	const prams = [
+		'-selection',
+		'clipboard'
+	]
+	return SH`xclip ${prams}`.options({ stdio: 'inherit' }).run(text);
+}
+
+
+/**
+ * CLILoader class for handling CLI-based interactions with AI models.
+ */
+class Cli {
+	#HELP = HELP;
+	#ERROR = ERROR;
+	/** @type {Session} */
+	#session;
+	/** @type {Prompt} */
+	#prompt;
+	#description = '';
+	#lastMessage = '';
+	/**
+	 * Creates an instance of CLILoader.
+	 * @param {CLIOptions} options - The options to configure the CLILoader.
+	 */
+	constructor(options) {
+		this.#prompt = options.prompt;
+		this.#session = options.session;
+		this.#description = (options.description) ? options.description + INTRO : INTRO;
+		this.#HELP = (options.help) ? options.help : HELP;
+		this.#setupCLI();
+	}
+
+	#setupCLI = () => {
+		cli.setRole('assistant', `${this.#session.name}: > `, 'brightGreen');
+		cli.setRole('util', ``, 'brightYellow');
+		cli.setRole('error', ``, 'red');
+		// @ts-ignore
+		cli.inputHandler = async (s, role) => await this.#handleInput(s, role);
+		cli.registerKeyMappings([
+			{ // reset the session, empty the context from the prompt
+				name: 'r', ctrl: false, meta: true, shift: false, handler: async () => {
+					cli.clear();
+					await this.reset();
+				}
+			},
+			{ // load a previous ssession
+				name: 's', ctrl: false, meta: true, shift: false, handler: async () => {
+					await this.loadSession();
+				}
+			},
+			{ // load a previous ssession
+				name: 'i', ctrl: false, meta: true, shift: false, handler: async () => {
+					this.printInfo();
+					// cli.focus('user', true);
+				}
+			},
+			{ // Copy last message to clipboard
+				name: 'm', ctrl: false, meta: true, shift: false, handler: async () => {
+					await copyToClipboard(this.#lastMessage);
+					cli.focus('util');
+					cli.write('copied');
+					// cli.focus('user', true);
+				}
+			},
+			{ // Show help /keys
+				name: 'k', ctrl: false, meta: true, shift: false, handler: async () => {
+					cli.focus('util')
+					cli.write(HELP)
+					// cli.focus('user', true);
+				}
+			}
+		]);
+		// Write messages
+		this.#prompt.on('message', (msg) => {
+			if (msg.role === 'reasoning') {
+				const content = this.#prompt.contentToString(msg.content);
+				cli.focus('util');
+				cli.write(`REASONING: ---------\n${content}\n---------\n`);
+			} else if (msg.role === 'log') {
+				const content = this.#prompt.contentToString(msg.content);
+				cli.focus('log');
+				cli.write(content);
+			} else if (msg.role === 'assistant') {
+				if (msg.content[0].type === 'text') {
+					const content = this.#prompt.contentToString(msg.content);
+					if (content.trim() !== '') {
+						cli.focus('assistant');
+						cli.write(content);
+					}
+				}
+			}
+		});
+		this.#prompt.on('truncated', () => {
+			cli.focus('log');
+			cli.write(`--- prompt truncated ---`);
+			// cli.startSpinner();
+		});
+		this.#prompt.on('finished', () => {
+			cli.focus('user', true);
+		});
+		// console.log(this.#prompt);
+	}
+
+	/**
+	 * @param {string} s
+	 * @param {string} role
+	 */
+	async #handleInput(s, role) {
+		if (!s || s === '') {
+			return;
+		}
+		// Only accept a users input 
+		if (role === 'user') {
+			cli.startSpinner();
+			try {
+				this.#lastMessage = await this.#prompt.call(s)
+			} catch (error) {
+				// @ts-ignore
+				await this.#handleError(error);
+			}
+		}
+	}
+
+	/**
+	 * @param {Error} error
+	 */
+	async #handleError(error) {
+		cli.error(error);
+		cli.focus('util');
+		cli.write(this.#ERROR);
+		// this.reset();
+		cli.focus('user', true);
+	}
+
+  printInfo() {
+		const sessionInfo = this.#session.info();
+		const promptInfo = this.#prompt.info();
+		const info = `\nSESSION\n${sessionInfo}\nPROMPT\n${promptInfo}\n${this.#description}`;
+		cli.focus('util');
+		cli.write(info);
+  }
+
+	/**
+	 * Starts the CLI interaction.
+	 * @param {string} [s] - optional user string/question
+	 * @returns {Promise<void>}
+	 */
+	async start(s) {
+		cli.focus('util');
+		cli.write(this.#description);
+		cli.focus('user', true);
+		if (s && s.trim() !== '') {
+			await this.#handleInput(s, 'user');
+		}
+	}
+	async loadSession() {
+		// Last session last
+		const sess = this.#session.sessionList().reverse();
+		sess.push('NONE');
+		const selected = await cli.select('Select your session: ', sess);
+		if (!selected || selected === 'NONE') {
+			cli.focus('user', true);
+			return;
+		}
+		cli.clear();
+		const messages = this.#session.set(selected);
+		this.#prompt.messages = messages;
+		messages.forEach((msg) => {
+			if (!msg.sticky) {
+				if (['user', 'assistant', 'reasoning', 'log'].includes(msg.role)) {
+					const content = this.#prompt.contentToString(msg.content);
+					// @ts-ignore
+					cli.focus(msg.role);
+					cli.write(content);
+				}
+			}
+		});
+		cli.startSpinner();
+		await this.#prompt.triggerRequest();
+		// cli.focus('user', true);
+	}
+	async reset() {
+		this.#prompt.reset();
+		// this.start();
+		this.printInfo();
+		// cli.focus('user', true);
+	}
+}
+
+export default Cli;