improved lt memory

index.ts

@@ -2,35 +2,38 @@ import Perplexity from "./libs/Perplexity";
 import "dotenv/config";
 import required from "./libs/required";
 import readin from "./libs/readin";
-import fs from "fs"
+import fs from "fs";
-import parseCommands from "./libs/parseCommands";
+import config from "./config";
-import config from "./config"
+import { ContextPart, parseCommands } from "./libs/parseCommands";
-import type Message from "./libs/Perplexity"
-import type { ListFormat } from "typescript";
 
 required(process.env.PPLX_API_KEY);
 const PPLX_API_KEY = process.env.PPLX_API_KEY as string;
 let perplexity: Perplexity;
 
 // NOTE Helper to load context from file
-function inject_context() {
+function inject_context(verbose: boolean = true) {
-    let loaded_context = fs.readFileSync("./context.json", {encoding: "utf-8"})
+  let loaded_context = fs.readFileSync("./context.json", { encoding: "utf-8" });
-    let list_context = JSON.parse(loaded_context)
+  let list_context = JSON.parse(loaded_context) as ContextPart[];
-    for (let i=0; i<list_context.length; i++) {
+  for (let i = 0; i < list_context.length; i++) {
-        perplexity.add_to_context(list_context[i])
+    let msg = list_context[i] as ContextPart;
-    }
+    perplexity.add_to_context(msg);
+    if (verbose) console.log("[" + msg.role + "]> " + msg.content);
+  }
 }
 
 // NOTE Helper to save context long term
 function save_in_context(role: string, message: string) {
-    let loaded_context = fs.readFileSync("./context.json", {encoding: "utf-8"})
+  let loaded_context = fs.readFileSync("./context.json", { encoding: "utf-8" });
-    let list_context = JSON.parse(loaded_context) as Array<{"role": string, "content": string}>
+  let list_context = JSON.parse(loaded_context) as Array<{
-    let new_insertion = {
+    role: string;
-        role: role,
+    content: string;
-        content: message
+  }>;
-    }
+  let new_insertion = {
-    list_context.push(new_insertion)
+    role: role,
-    fs.writeFileSync("./context.json", JSON.stringify(list_context))
+    content: message,
+  };
+  list_context.push(new_insertion);
+  fs.writeFileSync("./context.json", JSON.stringify(list_context));
 }
 
 async function main() {
@@ -39,14 +42,14 @@ async function main() {
   perplexity = new Perplexity(
     PPLX_API_KEY,
     "https://api.perplexity.ai",
-    "sonar-medium-chat",
+    "sonar-small-chat",
     false,
     {}
   );
   // Setting our ai personality
   perplexity.add_to_context(config);
   // If any, inject context
-  inject_context()
+  inject_context();
   // Chatting
   await chat();
 }
@@ -55,18 +58,18 @@ async function chat(loop: boolean = true, save: boolean = true) {
   var proceed = true;
   while (proceed) {
     proceed = loop;
-    let question = await readin("You: ");
+    let question = await readin("[You]> ");
     let parsed = parseCommands(question);
-    proceed = parsed.proceed
+    proceed = parsed.proceed;
-    let logpart = parsed.logpart
+    let logpart = parsed.logpart;
     let response = await perplexity.ask(question);
-    console.log("Assistant: " + response);
+    console.log("[Assistant]> " + response);
     if (save) {
-        save_in_context("user", question)
+      save_in_context("user", question);
-        if (logpart) {
+      if (logpart) {
-            response += +"\n[" + logpart + "]"
+        response += " " + logpart;
-        }
+      }
-        save_in_context("assistant", response as string)
+      save_in_context("assistant", response as string);
     }
   }
 }

libs/parseCommands.ts

@@ -3,7 +3,7 @@ export interface ContextPart {
     content: string
 }
 
-export default function parseCommands(input: string): {proceed: boolean, logpart: string | null} {
+export function parseCommands(input: string): {proceed: boolean, logpart: string | null} {
     let proceed = true
     let logpart = null
     switch (input) {
@@ -13,7 +13,7 @@ export default function parseCommands(input: string): {proceed: boolean, logpart
         case "end":
         case "bye":
             proceed = false
-            logpart = "Last conversation ended at: " + String(Date.now())
+            logpart = "{ Memory: Last conversation ended at: " + String(Date.now() + " }")
             break
         default:
             break

tsconfig.json

@@ -11,7 +11,7 @@
     // Bundler mode
     "moduleResolution": "bundler",
     "allowImportingTsExtensions": true,
-    "verbatimModuleSyntax": true,
+    "verbatimModuleSyntax": false,
     "noEmit": true,
 
     // Best practices