@akanjs/cli 3.0.0-beta.9 → 3.0.0

package/index-te54x9zm.js

@@ -1,338 +0,0 @@
-// @bun
-import {
-  GlobalConfig
-} from "./index-m0xccmmm.js";
-import {
-  Spinner
-} from "./index-2tv3fdx4.js";
-
-// pkgs/@akanjs/devkit/aiEditor.ts
-import { input, select } from "@inquirer/prompts";
-import {
-  AIMessage,
-  HumanMessage,
-  mapChatMessagesToStoredMessages,
-  mapStoredMessagesToChatMessages
-} from "@langchain/core/messages";
-import { ChatDeepSeek } from "@langchain/deepseek";
-import { ChatOpenAI } from "@langchain/openai";
-import { Logger } from "akanjs/common";
-import chalk from "chalk";
-var MAX_ASK_TRY = 300;
-var deepSeekLlmModels = ["deepseek-v4-flash", "deepseek-v4-pro"];
-var openAiLlmModels = ["gpt-5.5"];
-var supportedLlmModels = [...deepSeekLlmModels, ...openAiLlmModels];
-var isOpenAiLlmModel = (model) => openAiLlmModels.includes(model);
-var parseTypescriptFileBlocks = (text) => {
-  const fileBlocks = [];
-  const codeBlockRegex = /```(?:typescript|ts|tsx)\s*\n([\s\S]*?)```/gi;
-  const filePathRegex = /^\s*\/\/\s*File:\s*(.+?)\s*$/im;
-  for (const codeBlock of text.matchAll(codeBlockRegex)) {
-    const content = codeBlock[1]?.trim();
-    if (!content)
-      continue;
-    const filePath = filePathRegex.exec(content)?.[1]?.trim();
-    if (!filePath)
-      continue;
-    fileBlocks.push({
-      filePath,
-      content: content.replace(filePathRegex, "").trim()
-    });
-  }
-  return fileBlocks;
-};
-var preserveTypescriptResponseContent = (previousContent, nextContent) => {
-  const previousWrites = parseTypescriptFileBlocks(previousContent);
-  const nextWrites = parseTypescriptFileBlocks(nextContent);
-  if (previousWrites.length > 0 && nextWrites.length === 0)
-    return previousContent;
-  return nextContent;
-};
-
-class AiSession {
-  static #cacheDir = "node_modules/.cache/akan/aiSession";
-  static #chat = null;
-  static async init({ temperature = 0, useExisting = true } = {}) {
-    if (useExisting) {
-      const llmConfig = await AiSession.getLlmConfig();
-      if (llmConfig) {
-        AiSession.#setChatModel(llmConfig.model, llmConfig.apiKey);
-        Logger.rawLog(chalk.dim(`\uD83E\uDD16akan editor uses existing LLM config (${llmConfig.model})`));
-        return AiSession;
-      }
-    } else
-      Logger.rawLog(chalk.yellow("\uD83E\uDD16akan-editor is not initialized. LLM configuration should be set first."));
-    const llmConfig = await AiSession.#requestLlmConfig();
-    const { model, apiKey } = llmConfig;
-    await AiSession.#validateApiKey(model, apiKey);
-    const session = AiSession.#setChatModel(model, apiKey, { temperature });
-    await session.setLlmConfig({ model, apiKey });
-    return session;
-  }
-  static #setChatModel(model, apiKey, { temperature = 0 } = {}) {
-    AiSession.#chat = AiSession.#createChatModel(model, apiKey, {
-      temperature,
-      streaming: true
-    });
-    return AiSession;
-  }
-  static #createChatModel(model, apiKey, { temperature = 0, streaming = false } = {}) {
-    if (isOpenAiLlmModel(model))
-      return new ChatOpenAI({
-        modelName: model,
-        temperature,
-        streaming,
-        openAIApiKey: apiKey
-      });
-    return new ChatDeepSeek({
-      modelName: model,
-      temperature,
-      streaming,
-      apiKey
-    });
-  }
-  static async getLlmConfig() {
-    return await GlobalConfig.getLlmConfig();
-  }
-  static async setLlmConfig(llmConfig) {
-    await GlobalConfig.setLlmConfig(llmConfig);
-    return AiSession;
-  }
-  static async#requestLlmConfig() {
-    const model = await select({
-      message: "Select a LLM model",
-      choices: supportedLlmModels
-    });
-    const apiKey = await input({ message: "Enter your API key" });
-    return { model, apiKey };
-  }
-  static async#validateApiKey(modelName, apiKey) {
-    const spinner = new Spinner("Validating LLM API key...", {
-      prefix: `\uD83E\uDD16akan-editor`
-    }).start();
-    const chat = AiSession.#createChatModel(modelName, apiKey);
-    try {
-      await chat.invoke("Hi, and just say 'ok'");
-      spinner.succeed("LLM API key is valid");
-      return true;
-    } catch (error) {
-      spinner.fail(chalk.red(`LLM API key is invalid. Please check your API key and try again. You can set it again by running "akan set-llm" or reset by running "akan reset-llm"`));
-      throw error;
-    }
-  }
-  static async clearCache(workspaceRoot) {
-    const cacheDir = `${workspaceRoot}/${AiSession.#cacheDir}`;
-    await Bun.$`rm -rf ${cacheDir}`;
-  }
-  messageHistory = [];
-  sessionKey;
-  isCacheLoaded = false;
-  workspace;
-  constructor(type, {
-    workspace,
-    cacheKey,
-    isContinued
-  }) {
-    this.workspace = workspace;
-    this.sessionKey = `${type}${cacheKey ? `-${cacheKey}` : ""}`;
-    if (isContinued)
-      this.#cacheLoadPromise = this.#loadCache();
-  }
-  #cacheLoadPromise = null;
-  async#loadCache() {
-    const cacheFile = `${AiSession.#cacheDir}/${this.sessionKey}.json`;
-    const isCacheExists = await this.workspace.exists(cacheFile);
-    if (isCacheExists)
-      this.messageHistory = mapStoredMessagesToChatMessages(await this.workspace.readJson(cacheFile));
-    else
-      this.messageHistory = [];
-    this.isCacheLoaded = isCacheExists;
-  }
-  async#saveCache() {
-    const cacheFilePath = `${AiSession.#cacheDir}/${this.sessionKey}.json`;
-    await this.workspace.writeJson(cacheFilePath, mapChatMessagesToStoredMessages(this.messageHistory));
-  }
-  async ask(question, {
-    onReasoning = (reasoning) => {
-      Logger.raw(chalk.dim(reasoning));
-    },
-    onChunk = (chunk) => {
-      Logger.raw(chunk);
-    }
-  } = {}) {
-    if (!AiSession.#chat)
-      await AiSession.init();
-    if (this.#cacheLoadPromise)
-      await this.#cacheLoadPromise;
-    if (!AiSession.#chat)
-      throw new Error("Failed to initialize the AI session");
-    const loader = new Spinner(`${AiSession.#chat.model} is thinking...`, {
-      prefix: `\uD83E\uDD16akan-editor`
-    }).start();
-    try {
-      const humanMessage = new HumanMessage(question);
-      this.messageHistory.push(humanMessage);
-      const stream = await AiSession.#chat.stream(this.messageHistory);
-      let reasoningResponse = "", fullResponse = "";
-      for await (const chunk of stream) {
-        if (loader.isSpinning())
-          loader.succeed(`${AiSession.#chat.model} responded`);
-        if (!fullResponse.length) {
-          const reasoningContent = chunk.additional_kwargs.reasoning_content ?? "";
-          if (reasoningContent.length) {
-            reasoningResponse += reasoningContent;
-            onReasoning(reasoningContent);
-            continue;
-          } else if (chunk.content.length) {
-            reasoningResponse += `
-`;
-            onReasoning(reasoningResponse);
-          }
-        }
-        const content = chunk.content;
-        if (typeof content === "string") {
-          fullResponse += content;
-          onChunk(content);
-        }
-      }
-      fullResponse += `
-`;
-      onChunk(`
-`);
-      this.messageHistory.push(new AIMessage(fullResponse));
-      return { content: fullResponse, messageHistory: this.messageHistory };
-    } catch {
-      loader.fail(`${AiSession.#chat.model} failed to respond`);
-      throw new Error("Failed to stream response");
-    }
-  }
-  async edit(question, { onChunk, onReasoning, maxTry = MAX_ASK_TRY, validate, approve, fallbackToPreviousTypescript } = {}) {
-    for (let tryCount = 0;tryCount < maxTry; tryCount++) {
-      let response = await this.ask(question, { onChunk, onReasoning });
-      if (validate?.length && tryCount === 0) {
-        const validateQuestion = `Double check if the response meets the requirements and conditions, and follow the instructions. If not, rewrite it.
-${validate.map((v) => `- ${v}`).join(`
-`)}`;
-        const validateResponse = await this.ask(validateQuestion, {
-          onChunk,
-          onReasoning
-        });
-        response = {
-          ...validateResponse,
-          content: fallbackToPreviousTypescript ? preserveTypescriptResponseContent(response.content, validateResponse.content) : validateResponse.content
-        };
-      }
-      const isConfirmed = approve ? true : await select({
-        message: "Do you want to edit the response?",
-        choices: [
-          { name: "\u2705 Yes, confirm and apply this result", value: true },
-          { name: "\uD83D\uDD04 No, I want to edit it more", value: false }
-        ]
-      });
-      if (isConfirmed) {
-        await this.#saveCache();
-        return response.content;
-      }
-      question = await input({ message: "What do you want to change?" });
-      tryCount++;
-    }
-    throw new Error("Failed to edit");
-  }
-  async editTypescript(question, options = {}) {
-    const content = await this.edit(question, options);
-    return this.#getTypescriptCode(content);
-  }
-  #getTypescriptCode(content) {
-    //! will be deprecated
-    const code = /```(typescript|tsx)([\s\S]*?)```/.exec(content);
-    return code?.[2] ?? content;
-  }
-  addToolMessgaes(messages) {
-    const toolMessages = messages.map((message) => new HumanMessage(message.content));
-    this.messageHistory.push(...toolMessages);
-    return this;
-  }
-  async writeTypescripts(question, executor, options = {}) {
-    const content = await this.edit(question, {
-      ...options,
-      fallbackToPreviousTypescript: true
-    });
-    const writes = this.#getTypescriptCodes(content);
-    if (!writes.length)
-      throw new Error("No parseable TypeScript file blocks were found in the AI response. Include `// File: <path>` in each code block.");
-    for (const write of writes)
-      await executor.writeFile(write.filePath, write.content);
-    return await this.#tryFixTypescripts(writes, executor, options);
-  }
-  async#editTypescripts(question, options = {}, fallbackWrites) {
-    const content = await this.edit(question, {
-      ...options,
-      fallbackToPreviousTypescript: true
-    });
-    const writes = this.#getTypescriptCodes(content);
-    if (!writes.length && fallbackWrites?.length)
-      return fallbackWrites;
-    if (!writes.length)
-      throw new Error("No parseable TypeScript file blocks were found in the AI response. Include `// File: <path>` in each code block.");
-    return writes;
-  }
-  async#tryFixTypescripts(writes, executor, options = {}) {
-    const MAX_EDIT_TRY = 5;
-    for (let tryCount = 0;tryCount < MAX_EDIT_TRY; tryCount++) {
-      const loader = new Spinner(`Type checking and linting...`, {
-        prefix: `\uD83E\uDD16akan-editor`
-      }).start();
-      const fileChecks = await Promise.all(writes.map(async ({ filePath }) => {
-        const lintResult = await executor.lint(filePath, { fix: true });
-        const typeCheckResult = await executor.typeCheckAsync(filePath);
-        const hasTypeErrors = typeCheckResult.fileErrors.length > 0;
-        const hasLintErrors = lintResult.errors.length > 0;
-        const needFix = hasTypeErrors || hasLintErrors;
-        return { filePath, typeCheckResult, lintResult, needFix };
-      }));
-      const hasAnyFix = fileChecks.some((fileCheck) => fileCheck.needFix);
-      if (hasAnyFix) {
-        loader.fail("Type checking and linting has some errors, try to fix them");
-        fileChecks.forEach((fileCheck) => {
-          Logger.rawLog(`TypeCheck Result 
-${fileCheck.typeCheckResult.message}
-Lint Result 
-${fileCheck.lintResult.message}`);
-          this.addToolMessgaes([
-            { type: "typescript", content: fileCheck.typeCheckResult.message },
-            { type: "eslint", content: fileCheck.lintResult.message }
-          ]);
-        });
-        writes = await this.#editTypescripts("Fix the typescript and eslint errors", {
-          ...options,
-          validate: undefined,
-          approve: true
-        }, writes);
-        for (const write of writes)
-          await executor.writeFile(write.filePath, write.content);
-      } else {
-        loader.succeed("Type checking and linting has no errors");
-        return writes;
-      }
-    }
-    throw new Error("Failed to create scalar");
-  }
-  #getTypescriptCodes(text) {
-    return parseTypescriptFileBlocks(text);
-  }
-  async editMarkdown(request, options = {}) {
-    const content = await this.edit(request, options);
-    return this.#getMarkdownContent(content);
-  }
-  #getMarkdownContent(text) {
-    const searchText = "```markdown";
-    const firstIndex = text.indexOf("```markdown");
-    const lastIndex = text.lastIndexOf("```");
-    if (firstIndex === -1)
-      return text;
-    else
-      return text.slice(firstIndex + searchText.length, lastIndex).trim();
-  }
-}
-
-export { AiSession };