okai 0.0.28 → 0.0.29

package/dist/index.js

@@ -51,6 +51,9 @@ function parseArgs(...args) {
                 case "/cached":
                     ret.cached = true;
                     break;
+                case "/system":
+                    ret.system = args[++i];
+                    break;
                 default:
                     ret.unknown = ret.unknown || [];
                     ret.unknown.push(arg);
@@ -95,6 +98,10 @@ function parseArgs(...args) {
                 ret.accept = args[++i];
             }
         }
+        else if (arg == "chat") {
+            ret.type = "chat";
+            ret.chat = args[++i];
+        }
         else if (arg.endsWith('.d.ts')) {
             if (ret.type == "help")
                 ret.type = "update";
@@ -174,6 +181,8 @@ ${bin} ls models            Display list of available premium LLM models
 ${bin} init                 Initialize okai.json with project info to override default paths
 ${bin} init <model>         Create an empty <model>.d.ts file for the specified model
 ${bin} info                 Display current project info
+${bin} chat <prompt>        Submit a new OpenAI chat request with the specified prompt
+    -system <prompt>         Specify a system prompt
 
 Options:
     -v, -verbose             Display verbose logging
@@ -416,6 +425,41 @@ Options:
         await acceptGist(command, command.accept);
         process.exit(0);
     }
+    if (command.type === "chat") {
+        try {
+            const url = new URL('/chat', command.baseUrl);
+            const formData = new FormData();
+            formData.append('prompt', command.chat);
+            if (command.system) {
+                formData.append('system', command.system);
+            }
+            if (command.models) {
+                formData.append('model', command.models);
+                if (command.license) {
+                    formData.append('license', command.license);
+                }
+            }
+            if (command.verbose)
+                console.log(`POST: ${url}`);
+            const res = await fetch(url, {
+                method: 'POST',
+                body: formData,
+            });
+            if (!res.ok) {
+                console.log(`Failed to chat: ${res.statusText}`);
+                process.exit(1);
+            }
+            const response = await res.json();
+            if (command.verbose)
+                console.log(JSON.stringify(response, undefined, 2));
+            const content = response.choices[response.choices.length - 1]?.message?.content;
+            console.log(content);
+        }
+        catch (err) {
+            console.error(err);
+        }
+        process.exit(0);
+    }
     if (command.type === "prompt") {
         try {
             if (!info.serviceModelDir)

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "okai",
   "type": "module",
-  "version": "0.0.28",
+  "version": "0.0.29",
   "bin": "./dist/okai.js",
   "main": "./dist/index.js",
   "module": "./dist/index.js",