arial-cli 0.1.6 → 0.1.7

package/dist/cli.js

@@ -2795,9 +2795,10 @@ function isUnicodeSupported() {
 }
 
 // src/lib/config.ts
-var VERSION = "0.1.6";
+var VERSION = "0.1.7";
 var PROMPT_LINE_PREFIX = "●";
 var PROMPT_LINE_PREFIX_SAFE = "ai>";
+var DEFAULT_PROVIDER = "byok";
 var DEFAULT_MODEL = "gpt-4o";
 var DEFAULT_CONFIG_DIR = ".arial";
 var DEFAULT_CONFIG_FILE = "config.json";
@@ -3096,6 +3097,7 @@ function getConfigDir() {
 }
 
 class ConfigManager {
+  provider;
   model;
   temperature;
   maxTokens;
@@ -3103,6 +3105,7 @@ class ConfigManager {
   systemPrompt;
   toolsEnabled;
   constructor(options = {}) {
+    this.provider = options.provider || DEFAULT_PROVIDER;
     this.model = options.model || DEFAULT_MODEL;
     this.temperature = options.temperature || null;
     this.maxTokens = options.maxTokens || 1024;
@@ -3114,6 +3117,7 @@ class ConfigManager {
     this.getOrCreateConfig();
     const config = await this.getConfig();
     this.validateConfig(config);
+    this.provider = config.provider;
     this.model = config.model;
     this.temperature = config.temperature;
     this.maxTokens = config.maxTokens;
@@ -3124,6 +3128,7 @@ class ConfigManager {
   async save() {
     const configPath = getConfigPath();
     const config = {
+      provider: this.provider,
       model: this.model,
       temperature: this.temperature,
       maxTokens: this.maxTokens,
@@ -3155,6 +3160,7 @@ class ConfigManager {
     } catch {
       console.log(`Creating default config file at: ${configPath}`);
       const defaultConfig = {
+        provider: DEFAULT_PROVIDER,
         model: DEFAULT_MODEL,
         temperature: null,
         maxTokens: 1024,
@@ -31777,17 +31783,20 @@ function createOpenAI(options = {}) {
 }
 var openai = createOpenAI();
 
-// src/lib/ai.ts
-class Ai {
+// src/lib/providers/byok-provider.ts
+class ByokProvider {
+  getClient(session) {
+    const key = session.keys.get("OPENAI_API_KEY");
+    if (!key) {
+      throw new Error("No API key found. Please add an OpenAI API key using the /keys command.");
+    }
+    return createOpenAI({
+      apiKey: key.key
+    });
+  }
   async generateWithContext(session) {
     try {
-      const key = session.keys.get("OPENAI_API_KEY");
-      if (!key) {
-        throw new Error("No API key found. Please add an OpenAI API key using the /keys command.");
-      }
-      const openai2 = createOpenAI({
-        apiKey: key.key
-      });
+      const openai2 = this.getClient(session);
       const toolsEnabled = await session.tools.getEnabledTools();
       const params = {
         model: openai2(session.model),
@@ -31807,15 +31816,9 @@ class Ai {
       throw error46;
     }
   }
-  static async* streamWithContext(session) {
+  async* streamWithContext(session) {
     try {
-      const key = session.keys.get("OPENAI_API_KEY");
-      if (!key) {
-        throw new Error("No API key found. Please add an OpenAI API key using the /keys command.");
-      }
-      const openai2 = createOpenAI({
-        apiKey: key.key
-      });
+      const openai2 = this.getClient(session);
       const toolsEnabled = await session.tools.getEnabledTools();
       const params = {
         model: openai2(session.model),
@@ -31841,6 +31844,37 @@ class Ai {
   }
 }
 
+// src/lib/providers/arial-provider.ts
+class ArialProvider {
+  async generateWithContext(session) {
+    return "Generated text from ArialProvider";
+  }
+  async* streamWithContext(session) {
+    yield "Streaming text from ArialProvider";
+  }
+}
+
+// src/lib/ai.ts
+class Ai {
+  selectProvider(session) {
+    if (session.provider === "byok") {
+      return new ByokProvider;
+    } else if (session.provider === "arial") {
+      return new ArialProvider;
+    } else {
+      throw new Error(`Unsupported provider: ${session.provider}`);
+    }
+  }
+  async generateWithContext(session) {
+    const provider = this.selectProvider(session);
+    return provider.generateWithContext(session);
+  }
+  async* streamWithContext(session) {
+    const provider = this.selectProvider(session);
+    yield* provider.streamWithContext(session);
+  }
+}
+
 // node_modules/.pnpm/uuid@13.0.0/node_modules/uuid/dist-node/stringify.js
 var byteToHex = [];
 for (let i = 0;i < 256; ++i) {
@@ -31899,6 +31933,7 @@ class SessionManager {
   currentConversationId = null;
   currentConversationTimestamp = null;
   messages = [];
+  provider;
   model;
   streaming;
   systemPrompt;
@@ -31908,6 +31943,7 @@ class SessionManager {
   storage;
   config;
   constructor(config2) {
+    this.provider = config2.provider;
     this.model = config2.model;
     this.streaming = config2.streaming;
     this.systemPrompt = config2.systemPrompt;
@@ -31974,7 +32010,8 @@ class SessionManager {
   }
   streamResponse(prompt) {
     this.addMessage({ role: "user", content: prompt });
-    return Ai.streamWithContext(this);
+    const ai = new Ai;
+    return ai.streamWithContext(this);
   }
   generateResponse(prompt) {
     this.addMessage({ role: "user", content: prompt });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "arial-cli",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "description": "A CLI utility to interact with LLMs",
   "type": "module",
   "bin": {