downcity 1.0.5 → 1.0.7

package/dist/cli/HeroManager.d.ts

@@ -0,0 +1,9 @@
+import { Hero } from "../hero/Hero.js";
+/**
+ * 英雄管理器:
+ */
+export declare class HeroManager {
+    private _hero;
+    constructor(hero: Hero);
+}
+//# sourceMappingURL=HeroManager.d.ts.map

package/dist/cli/HeroManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"HeroManager.d.ts","sourceRoot":"","sources":["../../src/cli/HeroManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,iBAAiB,CAAC;AAIvC;;GAEG;AACH,qBAAa,WAAW;IACtB,OAAO,CAAC,KAAK,CAAO;gBACR,IAAI,EAAE,IAAI;CAGvB"}

package/dist/cli/HeroManager.js

@@ -0,0 +1,12 @@
+import { HEROS_PATH } from "./const.js";
+const path = HEROS_PATH;
+/**
+ * 英雄管理器:
+ */
+export class HeroManager {
+    _hero;
+    constructor(hero) {
+        this._hero = hero;
+    }
+}
+//# sourceMappingURL=HeroManager.js.map

package/dist/cli/HeroManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"HeroManager.js","sourceRoot":"","sources":["../../src/cli/HeroManager.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAExC,MAAM,IAAI,GAAG,UAAU,CAAC;AACxB;;GAEG;AACH,MAAM,OAAO,WAAW;IACd,KAAK,CAAO;IACpB,YAAY,IAAU;QACpB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;IACpB,CAAC;CACF"}

package/dist/cli/ToolManager.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ToolManager.d.ts.map

package/dist/cli/ToolManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolManager.d.ts","sourceRoot":"","sources":["../../src/cli/ToolManager.ts"],"names":[],"mappings":""}

package/dist/cli/ToolManager.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ToolManager.js.map

package/dist/cli/ToolManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolManager.js","sourceRoot":"","sources":["../../src/cli/ToolManager.ts"],"names":[],"mappings":""}

package/dist/{const.d.ts → cli/const.d.ts}

@@ -1,4 +1,5 @@
 export declare const BASE_PATH: string;
 export declare const CODEX_PATH: string;
 export declare const CODEX_FILE_PATH: string;
+export declare const HEROS_PATH: string;
 //# sourceMappingURL=const.d.ts.map

package/dist/cli/const.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"const.d.ts","sourceRoot":"","sources":["../../src/cli/const.ts"],"names":[],"mappings":"AAGA,eAAO,MAAM,SAAS,QAAmC,CAAC;AAE1D,eAAO,MAAM,UAAU,QAA4C,CAAC;AACpE,eAAO,MAAM,eAAe,QAAiD,CAAC;AAE9E,eAAO,MAAM,UAAU,QAA+B,CAAC"}

package/dist/{const.js → cli/const.js}

@@ -3,4 +3,5 @@ import path from "path";
 export const BASE_PATH = path.join(homedir(), "downcity");
 export const CODEX_PATH = path.join(BASE_PATH, "codex", "codex_db");
 export const CODEX_FILE_PATH = path.join(BASE_PATH, "base", "knowledge.json");
+export const HEROS_PATH = path.join(BASE_PATH, "hero");
 //# sourceMappingURL=const.js.map

package/dist/cli/const.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"const.js","sourceRoot":"","sources":["../../src/cli/const.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAC7B,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB,MAAM,CAAC,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,UAAU,CAAC,CAAC;AAE1D,MAAM,CAAC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;AACpE,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,gBAAgB,CAAC,CAAC;AAE9E,MAAM,CAAC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC"}

package/dist/cli/index.js

@@ -2,19 +2,17 @@
 import { Command } from "commander";
 import pkg from "../../package.json" with { type: "json" };
 import chalk from "chalk";
-import { Hero, Vault } from "../index.js";
+import { Hero, Room } from "../index.js";
 import readline from "readline";
-import { createOpenAI } from "@ai-sdk/openai";
 import { log } from "console";
 import { tool } from "ai";
 import z from "zod";
-import { SQLiteVaultPersistor } from "../vault/Vault.js";
+import { SQLiteRoomPersistor } from "../room/Room.js";
 const program = new Command();
 // 设置基本信息
 program
     .name("downcity")
     .description("downcity - An open world productive game in terminal")
-    // 支持 -v (小写) 与 --version，便于用户输入
     .version(pkg.version, "-v, --version", "显示版本号");
 // 添加欢迎信息
 program.addHelpText("before", chalk.blue.bold(`
@@ -24,23 +22,16 @@ program.addHelpText("before", chalk.blue.bold(`
 program.addCommand(new Command("enter").description("enter in this game").action(async () => {
     console.log("welcome to downcity!");
     console.log("letus punk!");
-    if (!process.env.API_KEY) {
-        console.log(chalk.red("请设置 API_KEY 环境变量"));
+    if (!process.env.DOWNCITY_API_KEY) {
+        console.log(chalk.red("请设置 DOWNCITY_API_KEY 环境变量"));
         return;
     }
-    if (!process.env.BASE_URL) {
-        console.log(chalk.red("请设置 BASE_URL 环境变量"));
+    if (!process.env.DOWNCITY_BASE_URL) {
+        console.log(chalk.red("请设置 DOWNCITY_BASE_URL 环境变量"));
         return;
     }
-    // 创建一个provider
-    const provider = createOpenAI({
-        apiKey: process.env.API_KEY,
-        baseURL: process.env.BASE_URL,
-    });
-    // 一个model chat
-    const model = provider.chat("qwen-turbo");
-    // 创建一个vault: 用来控制session
-    const vault = new Vault(new SQLiteVaultPersistor());
+    // 创建一个room: 用来控制shot
+    const room = new Room(new SQLiteRoomPersistor());
     const tools = {
         get_current_time: tool({
             description: "获取当前时间",
@@ -51,10 +42,8 @@ program.addCommand(new Command("enter").description("enter in this game").action
         }),
     };
     const hero = Hero.create()
-        .model(model)
-        .vault(vault)
-        .study(tools)
-        .avatar("你是一个助手");
+        .room(room)
+        .study(tools);
     const rl = readline.createInterface({
         input: process.stdin,
         output: process.stdout,
@@ -62,19 +51,19 @@ program.addCommand(new Command("enter").description("enter in this game").action
     const chatLoop = async () => {
         rl.question(chalk.green("You: "), async (input) => {
             if (input.trim() === "ls") {
-                log(hero.sessions());
+                log(hero.shots());
                 chatLoop();
                 return;
             }
             if (input.trim() === "current") {
-                console.log(hero.session.id);
-                console.log(hero.session.messages);
+                console.log(hero.shot.id);
+                console.log(hero.shot.messages);
                 chatLoop();
                 return;
             }
             if (input.trim() === "new") {
-                const session = hero.renew();
-                console.log(session);
+                const shot = hero.renew();
+                console.log(shot);
                 chatLoop();
                 return;
             }
@@ -84,8 +73,8 @@ program.addCommand(new Command("enter").description("enter in this game").action
                 return;
             }
             if (input.startsWith("switch")) {
-                const sessionId = input.split(" ")[1];
-                hero.switch(sessionId);
+                const shotId = input.split(" ")[1];
+                hero.switch(shotId);
                 chatLoop();
                 return;
             }

package/dist/cli/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/cli/index.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,GAAG,MAAM,oBAAoB,CAAC,OAAO,IAAI,EAAE,MAAM,EAAE,CAAC;AAC3D,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,QAAQ,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,GAAG,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAC1B,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAEzD,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;AAE9B,SAAS;AACT,OAAO;KACJ,IAAI,CAAC,UAAU,CAAC;KAChB,WAAW,CAAC,sDAAsD,CAAC;IACpE,gCAAgC;KAC/B,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;AAElD,SAAS;AACT,OAAO,CAAC,WAAW,CACjB,QAAQ,EACR,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;;CAEjB,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAC7C,CAAC;AAEF,OAAO;AACP,OAAO,CAAC,UAAU,CAChB,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC,WAAW,CAAC,oBAAoB,CAAC,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE;IACvE,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;IACpC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAE3B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACzB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAC3C,OAAO;IACT,CAAC;IAED,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC1B,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAC5C,OAAO;IACT,CAAC;IAED,eAAe;IACf,MAAM,QAAQ,GAAG,YAAY,CAAC;QAC5B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,OAAO;QAC3B,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,QAAQ;KAC9B,CAAC,CAAC;IACH,eAAe;IACf,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAE1C,yBAAyB;IACzB,MAAM,KAAK,GAAG,IAAI,KAAK,CACrB,IAAI,oBAAoB,EAAE,CAC3B,CAAC;IAEF,MAAM,KAAK,GAAG;QACZ,gBAAgB,EAAE,IAAI,CAAC;YACrB,WAAW,EAAE,QAAQ;YACrB,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,GAAG,EAAE;gBACZ,OAAO,IAAI,IAAI,EAAE,CAAC,cAAc,EAAE,CAAC;YACrC,CAAC;SACF,CAAC;KACH,CAAC;IAEF,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,EAAE;SACvB,KAAK,CAAC,KAAK,CAAC;SACZ,KAAK,CAAC,KAAK,CAAC;SACZ,KAAK,CAAC,KAAK,CAAC;SACZ,MAAM,CAAC,QAAQ,CAAC,CAAC;IAEpB,MAAM,EAAE,GAAG,QAAQ,CAAC,eAAe,CAAC;QAClC,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,MAAM,EAAE,OAAO,CAAC,MAAM;KACvB,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE;QAC1B,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAChD,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;gBAC1B,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;gBACrB,QAAQ,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,SAAS,EAAE,CAAC;gBAC/B,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC7B,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBACnC,QAAQ,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,KAAK,EAAE,CAAC;gBAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;gBAC7B,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACrB,QAAQ,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,OAAO,EAAE,CAAC;gBAC7B,IAAI,CAAC,KAAK,EAAE,CAAC;gBACb,QAAQ,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,IAAI,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC/B,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBACtC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBACvB,QAAQ,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,IAAI,KAAK,CAAC,WAAW,EAAE,KAAK,MAAM,EAAE,CAAC;gBACnC,EAAE,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,QAAQ,CAAC,CAAC;YAC1C,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,QAAQ,EAAE,CAAC;AACb,CAAC,CAAC,CACH,CAAC;AAEF,UAAU;AACV,OAAO,CAAC,KAAK,EAAE,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/cli/index.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,GAAG,MAAM,oBAAoB,CAAC,OAAO,IAAI,EAAE,MAAM,EAAE,CAAC;AAC3D,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,QAAQ,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,GAAG,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAC1B,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAEtD,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;AAE9B,SAAS;AACT,OAAO;KACJ,IAAI,CAAC,UAAU,CAAC;KAChB,WAAW,CAAC,sDAAsD,CAAC;KACnE,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;AAElD,SAAS;AACT,OAAO,CAAC,WAAW,CACjB,QAAQ,EACR,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;;CAEjB,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAC7C,CAAC;AAEF,OAAO;AACP,OAAO,CAAC,UAAU,CAChB,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC,WAAW,CAAC,oBAAoB,CAAC,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE;IACvE,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;IACpC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAE3B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC,CAAC;QACpD,OAAO;IACT,CAAC;IAED,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC,CAAC;QACrD,OAAO;IACT,CAAC;IAED,qBAAqB;IACrB,MAAM,IAAI,GAAG,IAAI,IAAI,CACnB,IAAI,mBAAmB,EAAE,CAC1B,CAAC;IAEF,MAAM,KAAK,GAAG;QACZ,gBAAgB,EAAE,IAAI,CAAC;YACrB,WAAW,EAAE,QAAQ;YACrB,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,GAAG,EAAE;gBACZ,OAAO,IAAI,IAAI,EAAE,CAAC,cAAc,EAAE,CAAC;YACrC,CAAC;SACF,CAAC;KACH,CAAC;IAEF,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,EAAE;SACvB,IAAI,CAAC,IAAI,CAAC;SACV,KAAK,CAAC,KAAK,CAAC,CAAA;IAEf,MAAM,EAAE,GAAG,QAAQ,CAAC,eAAe,CAAC;QAClC,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,MAAM,EAAE,OAAO,CAAC,MAAM;KACvB,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE;QAC1B,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAChD,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;gBAC1B,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;gBAClB,QAAQ,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,SAAS,EAAE,CAAC;gBAC/B,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC1B,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAChC,QAAQ,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,KAAK,EAAE,CAAC;gBAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;gBAC1B,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAClB,QAAQ,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,OAAO,EAAE,CAAC;gBAC7B,IAAI,CAAC,KAAK,EAAE,CAAC;gBACb,QAAQ,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,IAAI,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC/B,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBACnC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACpB,QAAQ,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,IAAI,KAAK,CAAC,WAAW,EAAE,KAAK,MAAM,EAAE,CAAC;gBACnC,EAAE,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,QAAQ,CAAC,CAAC;YAC1C,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,QAAQ,EAAE,CAAC;AACb,CAAC,CAAC,CACH,CAAC;AAEF,UAAU;AACV,OAAO,CAAC,KAAK,EAAE,CAAC"}

package/dist/codex/Codex.d.ts

@@ -1,139 +1,101 @@
-import { EmbeddingModel, Tool } from "ai";
-export interface VectorModelConfig {
-    model: any;
-    dimensions?: number;
+import { EmbeddingModel } from "ai";
+import { IndexConfig } from "@lancedb/lancedb";
+import { Volume } from "./Volume.js";
+/**
+ * Codex 配置选项
+ */
+export interface CodexConfig {
+    model: EmbeddingModel;
+    path?: string;
 }
-export interface ChromaConfig {
-    url?: string;
-    collection?: string;
-    persistPath?: string;
+/**
+ * 创建索引参数
+ */
+export interface CreateIndexParams {
+    tableName: string;
+    indexName: string;
+    dimension: number;
+    metric?: "cosine" | "euclidean" | "dotproduct";
+    indexConfig?: IndexConfig;
 }
-export interface KnowledgeOptions {
-    autoLoad?: boolean;
-    categories?: string[];
-    vectorDimension?: number;
-    similarityThreshold?: number;
-    vectorModel?: VectorModelConfig;
-    chroma?: ChromaConfig;
-    localFile?: string;
-}
-export interface KnowledgeItem {
-    id: string;
-    title: string;
-    content: string;
-    category: string;
-    tags: string[];
-    vector?: number[];
-    createdAt: Date;
-    updatedAt: Date;
+/**
+ * 索引统计信息
+ */
+export interface IndexStats {
+    dimension: number;
+    metric?: "cosine" | "euclidean" | "dotproduct";
+    count: number;
 }
+/**
+ * Codex 类，用于管理和检索向量化信息。
+ * 内部使用 LanceDB 进行数据存储。
+ * codex.volume().
+ */
 export declare class Codex {
-    private knowledgeBase;
-    private categories;
-    private options;
-    private chromaClient?;
-    private collection?;
-    private embeddingModel?;
-    private embeddingFunction?;
-    constructor(options?: KnowledgeOptions);
-    /**
-     * 创建Knowledge实例
-     */
-    static create(options?: KnowledgeOptions): Codex;
-    /**
-     * 初始化向量数据库
-     */
-    private initializeVectorDB;
-    /**
-     * 配置向量模型
-     */
-    model(model: EmbeddingModel): Codex;
-    /**
-     * 保存数据到本地文件
-     */
-    private saveToFile;
-    /**
-     * 从本地文件加载数据
-     */
-    private loadFromFile;
-    /**
-     * 获取知识库操作工具
-     * 将知识库的操作方法包装成工具供Hero调用
-     */
-    tools(): Record<string, Tool>;
-    /**
-     * 添加知识条目
-     */
-    addKnowledge(item: Omit<KnowledgeItem, "id" | "createdAt" | "updatedAt">): Promise<KnowledgeItem>;
-    /**
-     * 更新知识条目
-     */
-    updateKnowledge(id: string, updates: Partial<Omit<KnowledgeItem, "id" | "createdAt">>): KnowledgeItem | undefined;
-    /**
-     * 获取知识条目
-     */
-    getKnowledge(id: string): KnowledgeItem | undefined;
-    /**
-     * 搜索知识库（支持向量搜索和文本匹配）
-     */
-    searchKnowledge(query: string, category?: string, useVector?: boolean): Promise<KnowledgeItem[]>;
-    /**
-     * 文本搜索知识库
-     */
-    private textSearchKnowledge;
-    /**
-     * 向量相似度搜索
-     */
-    searchByVector(queryVector: number[], category?: string, limit?: number): Array<KnowledgeItem & {
-        similarity: number;
-    }>;
+    private volumes;
+    private db;
+    private model;
+    private constructor();
     /**
-     * 计算余弦相似度
+     * 创建并初始化一个 Codex 实例。
+     * @param config - Codex 配置
+     * @returns 一个初始化的 Codex 实例
      */
-    private cosineSimilarity;
+    static create(config: CodexConfig): Promise<Codex>;
     /**
-     * 为知识条目生成向量
+     * 关闭数据库连接并清理资源。
      */
-    generateEmbedding(text: string): Promise<number[]>;
+    close(): void;
     /**
-     * 添加知识条目并生成向量
+     * 检查连接是否有效。
+     * @returns 连接状态
      */
-    addKnowledgeWithVector(item: Omit<KnowledgeItem, "id" | "createdAt" | "updatedAt" | "vector">): Promise<KnowledgeItem>;
+    private checkConnection;
     /**
-     * 根据分类获取知识
+     * 列出所有可用的表（volumes）。
+     * @returns 表名列表
      */
-    getKnowledgeByCategory(category: string): KnowledgeItem[];
+    list(): Promise<string[]>;
     /**
-     * 获取所有分类
+     * 检查表是否存在。
+     * @param name - 表名
+     * @returns 表是否存在
      */
-    getCategories(): string[];
+    hasTable(name: string): Promise<boolean>;
     /**
-     * 删除知识条目
+     * 删除指定的表。
+     * @param name - 表名
      */
-    removeKnowledge(id: string): boolean;
+    dropTable(name: string): Promise<void>;
     /**
-     * 清空知识库
+     * 获取或创建一个 Volume 实例。
+     * @param name - Volume 名称
+     * @returns Volume 实例
      */
-    clearKnowledge(): void;
+    volume(name?: string): Promise<Volume>;
     /**
-     * 获取统计信息
+     * 创建向量索引。
+     * @param params - 创建索引参数
      */
-    getStats(): {
-        totalKnowledge: number;
-        totalCategories: number;
-        knowledgeByCategory: Record<string, number>;
-    };
+    createIndex(params: CreateIndexParams): Promise<void>;
     /**
-     * 导出数据
+     * 列出表的所有索引。
+     * @param tableName - 表名
+     * @returns 索引列表
      */
-    export(): any;
+    listIndexes(tableName: string): Promise<IndexConfig[]>;
     /**
-     * 导入数据
+     * 获取索引统计信息。
+     * @param tableName - 表名
+     * @param indexName - 索引名
+     * @returns 索引统计信息
      */
-    import(data: any): void;
+    getIndexStats(tableName: string, indexName: string): Promise<IndexStats>;
     /**
-     * 生成唯一ID
+     * 删除索引。
+     * @param tableName - 表名
+     * @param indexName - 索引名
      */
-    private generateId;
+    dropIndex(tableName: string, indexName: string): Promise<void>;
 }
 //# sourceMappingURL=Codex.d.ts.map

package/dist/codex/Codex.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Codex.d.ts","sourceRoot":"","sources":["../../src/codex/Codex.ts"],"names":[],"mappings":"AAAA,OAAO,EAAS,cAAc,EAAQ,IAAI,EAAE,MAAM,IAAI,CAAC;AAMvD,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,GAAG,CAAC;IACX,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,YAAY;IAC3B,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,WAAW,CAAC,EAAE,iBAAiB,CAAC;IAChC,MAAM,CAAC,EAAE,YAAY,CAAC;IACtB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;CACjB;AAED,qBAAa,KAAK;IAChB,OAAO,CAAC,aAAa,CAAyC;IAC9D,OAAO,CAAC,UAAU,CAA0B;IAC5C,OAAO,CAAC,OAAO,CAAmB;IAClC,OAAO,CAAC,YAAY,CAAC,CAAe;IACpC,OAAO,CAAC,UAAU,CAAC,CAAa;IAChC,OAAO,CAAC,cAAc,CAAC,CAAM;IAC7B,OAAO,CAAC,iBAAiB,CAAC,CAA0B;gBAExC,OAAO,GAAE,gBAAqB;IA0B1C;;OAEG;IACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,gBAAgB,GAAG,KAAK;IAIhD;;OAEG;YACW,kBAAkB;IA6ChC;;OAEG;IACH,KAAK,CAAC,KAAK,EAAE,cAAc,GAAG,KAAK;IAKnC;;OAEG;YACW,UAAU;IAWxB;;OAEG;YACW,YAAY;IAW1B;;;OAGG;IACH,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC;IAsC7B;;OAEG;IACG,YAAY,CAChB,IAAI,EAAE,IAAI,CAAC,aAAa,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAC1D,OAAO,CAAC,aAAa,CAAC;IAiDzB;;OAEG;IACH,eAAe,CACb,EAAE,EAAE,MAAM,EACV,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GACxD,aAAa,GAAG,SAAS;IAiB5B;;OAEG;IACH,YAAY,CAAC,EAAE,EAAE,MAAM,GAAG,aAAa,GAAG,SAAS;IAInD;;OAEG;IACG,eAAe,CACnB,KAAK,EAAE,MAAM,EACb,QAAQ,CAAC,EAAE,MAAM,EACjB,SAAS,GAAE,OAAc,GACxB,OAAO,CAAC,aAAa,EAAE,CAAC;IAsC3B;;OAEG;IACH,OAAO,CAAC,mBAAmB;IA6B3B;;OAEG;IACH,cAAc,CACZ,WAAW,EAAE,MAAM,EAAE,EACrB,QAAQ,CAAC,EAAE,MAAM,EACjB,KAAK,GAAE,MAAW,GACjB,KAAK,CAAC,aAAa,GAAG;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,CAAC;IAuBhD;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAyBxB;;OAEG;IACG,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAWxD;;OAEG;IACG,sBAAsB,CAC1B,IAAI,EAAE,IAAI,CAAC,aAAa,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,GAAG,QAAQ,CAAC,GACrE,OAAO,CAAC,aAAa,CAAC;IAOzB;;OAEG;IACH,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,aAAa,EAAE;IAMzD;;OAEG;IACH,aAAa,IAAI,MAAM,EAAE;IAIzB;;OAEG;IACH,eAAe,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAIpC;;OAEG;IACH,cAAc,IAAI,IAAI;IAItB;;OAEG;IACH,QAAQ,IAAI;QACV,cAAc,EAAE,MAAM,CAAC;QACvB,eAAe,EAAE,MAAM,CAAC;QACxB,mBAAmB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KAC7C;IAeD;;OAEG;IACH,MAAM,IAAI,GAAG;IAQb;;OAEG;IACH,MAAM,CAAC,IAAI,EAAE,GAAG,GAAG,IAAI;IAqBvB;;OAEG;IACH,OAAO,CAAC,UAAU;CAGnB"}
+{"version":3,"file":"Codex.d.ts","sourceRoot":"","sources":["../../src/codex/Codex.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AACpC,OAAO,EAAuB,WAAW,EAAS,MAAM,kBAAkB,CAAC;AAC3E,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAIrC;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,cAAc,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,QAAQ,GAAG,WAAW,GAAG,YAAY,CAAC;IAC/C,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAED;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,QAAQ,GAAG,WAAW,GAAG,YAAY,CAAC;IAC/C,KAAK,EAAE,MAAM,CAAC;CACf;AAED;;;;GAIG;AACH,qBAAa,KAAK;IAChB,OAAO,CAAC,OAAO,CAAyB;IACxC,OAAO,CAAC,EAAE,CAAc;IACxB,OAAO,CAAC,KAAK,CAAiB;IAE9B,OAAO;IAMP;;;;OAIG;WACiB,MAAM,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC;IAkC/D;;OAEG;IACH,KAAK,IAAI,IAAI;IAMb;;;OAGG;IACH,OAAO,CAAC,eAAe;IAWvB;;;OAGG;IACG,IAAI,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAoB/B;;;;OAIG;IACG,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAsB9C;;;OAGG;IACG,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAkB5C;;;;OAIG;IACG,MAAM,CAAC,IAAI,GAAE,MAAkB,GAAG,OAAO,CAAC,MAAM,CAAC;IAoDvD;;;OAGG;IACG,WAAW,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAqB3D;;;;OAIG;IACG,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAmB5D;;;;;OAKG;IACG,aAAa,CACjB,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,UAAU,CAAC;IAwBtB;;;;OAIG;IACG,SAAS,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAkBrE"}