feat: adjust prompter

Author: pixelass

src/index.ts

@@ -54,6 +54,10 @@ export class Agent<
 		if (options.after) {
 			this.after = options.after;
 		}
+
+		if (options.finally) {
+			this.finally = options.finally;
+		}
 	}
 
 	/**
@@ -80,8 +84,11 @@ export class Agent<
 			})
 		);
 		const modifiedMessage = await this.after(message);
+		console.log("modifiedMessage");
+		console.log(modifiedMessage);
 
-		return this.#store.set(modifiedMessage);
+		const messageId = await this.#store.set(modifiedMessage);
+		return this.finally(messageId, modifiedMessage);
 	}
 
 	/**

src/openai/index.ts

@@ -26,6 +26,8 @@ export class GPTModelAdapter<Options extends GPTOptions> implements ModelAdapter
 	 * @param {OpenAIApi} openai - A configured openai API instance.
 	 */
 	constructor(options: Options, openai: OpenAIApi) {
+		console.log("systemInstruction");
+		console.log(options.systemInstruction);
 		this.#options = options;
 		this.#openai = openai;
 		this.history = [];
@@ -71,6 +73,8 @@ export class GPTModelAdapter<Options extends GPTOptions> implements ModelAdapter
 			});
 
 			const { content } = completion.data.choices[0].message;
+			console.log("RAW");
+			console.log(content);
 			const jsonString = extractCode(content);
 			this.addMessageToHistory({ role: "assistant", content: jsonString });
 			return JSON.parse(jsonString);

src/utils.ts

@@ -107,6 +107,20 @@ export function minify(strings: TemplateStringsArray, ...values: string[]) {
 	return output.replace(/^\s+/gm, "").trim();
 }
 
+export function maybePeriod(text: string) {
+	if (
+		text.endsWith(".") ||
+		text.endsWith("!") ||
+		text.endsWith("?") ||
+		text.endsWith(":") ||
+		text.endsWith(";")
+	) {
+		return "";
+	}
+
+	return ". ";
+}
+
 /**
  * Creates an instruction string for the AI agent.
  *
@@ -117,11 +131,12 @@ export function minify(strings: TemplateStringsArray, ...values: string[]) {
  */
 export function createInstruction(role: string, tasks: string, format: Record<string, unknown>) {
 	return minify`
-		You are a ${role}.
-		Your tasks: ${tasks}.
+		You are a ${role}${maybePeriod(role)}
+		Your tasks: ${tasks}${maybePeriod(tasks)}
 		You NEVER explain or add notes.
-		You ONLY communicate **valid JSON**.
-		**You answer ONLY!!! as valid  JSON in this Format:**
+		ALL content will be inserted in JSON.
+		You EXCLUSIVELY communicate **valid JSON**.
+		**You answer EXCLUSIVELY!!! as valid  JSON in this Format**:
 		${""}
 		\`\`\`json
 		${JSON.stringify(format)}