downcity 1.0.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) [year] [fullname]
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,59 @@
+# 🏰 DownCity
+
+> _"在数字世界的城市中，每个 AI 都是一位英雄"_
+
+一个基于游戏化思维设计的 AI 智能体框架，让 AI 开发变得像 RPG 冒险一样有趣！
+
+## 🚀 安装
+
+```bash
+npm install downcity
+```
+
+## 🎮 快速开始
+
+### 创造你的第一个英雄
+
+```ts
+import { Hero, Knowledge, Model } from "downcity";
+
+const hero = Hero.create();
+
+// system prompt
+hero.avatar("我是一个新的英雄");
+// 设置ai-sdk的tool
+hero.model();
+// 学习 ai-sdk的tool
+hero.study();
+
+// 准备好: 直接暴露一个端口
+await hero.ready(5000);
+
+const memory = new Memory();
+hero.memory(memory);
+hero.session(memory.lastSession());
+const res = await hero.chat("你好");
+console.log(res);
+```
+
+```typescript
+import { Knowledge, Memory } from "downcity";
+
+const knowledge = Knowledge.create();
+
+const tools = knowledge.tool();
+
+const memory = new Memory();
+
+hero.memory(memory);
+
+// ai-sdk 的tool
+const tool = createTool({});
+
+hero.study(tools);
+hero.study(tool);
+
+const hero2 = new Hero();
+hero2.memory(memory);
+hero2.session(memory.newSession());
+```

package/dist/Hero.d.ts

@@ -0,0 +1,64 @@
+import { LanguageModel, Tool } from "ai";
+import { Memory } from "./Memory.js";
+export interface Session {
+    id: string;
+    messages: Array<{
+        role: "user" | "assistant" | "system";
+        content: string;
+        timestamp: Date;
+    }>;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare class Hero {
+    private _model;
+    private _system;
+    private _tools;
+    private _memory?;
+    private _currentSession?;
+    private constructor();
+    /**
+     * 创建一个新的英雄实例
+     */
+    static create(): Hero;
+    /**
+     * 设置英雄的身份和系统提示词
+     */
+    avatar(prompt: string): Hero;
+    /**
+     * 设置语言模型
+     */
+    model(model: LanguageModel): Hero;
+    /**
+     * 学习工具
+     */
+    study(tools: Record<string, Tool>): Hero;
+    /**
+     * 设置记忆系统
+     */
+    memory(memory: Memory): Hero;
+    /**
+     * 设置当前会话
+     */
+    session(session: Session): Hero;
+    /**
+     * 与英雄对话
+     */
+    chat(message: string): Promise<string>;
+    /**
+     * 生成JSON对象
+     */
+    json<T>(prompt: string, schema: any): Promise<T>;
+    /**
+     * 基于特定标准进行判断
+     */
+    check(content: string, criteria: string): Promise<boolean>;
+    /**
+     * 启动服务器
+     */
+    ready(port?: number): Promise<void>;
+    get systemPrompt(): string;
+    get tools(): string[];
+    get currentSession(): Session | undefined;
+}
+//# sourceMappingURL=Hero.d.ts.map

package/dist/Hero.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"Hero.d.ts","sourceRoot":"","sources":["../src/Hero.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAgC,IAAI,EAAe,MAAM,IAAI,CAAC;AACpF,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAGrC,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,KAAK,CAAC;QACd,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;QACtC,OAAO,EAAE,MAAM,CAAC;QAChB,SAAS,EAAE,IAAI,CAAC;KACjB,CAAC,CAAC;IACH,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;CACjB;AAED,qBAAa,IAAI;IACf,OAAO,CAAC,MAAM,CAAgD;IAC9D,OAAO,CAAC,OAAO,CAAsB;IACrC,OAAO,CAAC,MAAM,CAA4B;IAC1C,OAAO,CAAC,OAAO,CAAC,CAAS;IACzB,OAAO,CAAC,eAAe,CAAC,CAAU;IAElC,OAAO;IAEP;;OAEG;IACH,MAAM,CAAC,MAAM,IAAI,IAAI;IAIrB;;OAEG;IACH,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAK5B;;OAEG;IACH,KAAK,CAAC,KAAK,EAAE,aAAa,GAAG,IAAI;IAKjC;;OAEG;IACH,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,IAAI;IAKxC;;OAEG;IACH,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAK5B;;OAEG;IACH,OAAO,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAK/B;;OAEG;IACG,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IA0C5C;;OAEG;IACG,IAAI,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC;IAoBtD;;OAEG;IACG,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAqBhE;;OAEG;IACG,KAAK,CAAC,IAAI,GAAE,MAAa,GAAG,OAAO,CAAC,IAAI,CAAC;IAS/C,IAAI,YAAY,IAAI,MAAM,CAEzB;IAED,IAAI,KAAK,IAAI,MAAM,EAAE,CAEpB;IAED,IAAI,cAAc,IAAI,OAAO,GAAG,SAAS,CAExC;CACF"}

package/dist/Hero.js

@@ -0,0 +1,155 @@
+import { generateText, generateObject, stepCountIs } from "ai";
+import { createOpenAI } from "@ai-sdk/openai";
+export class Hero {
+    _model = createOpenAI().chat("gpt-4o");
+    _system = "你是一个智能助手";
+    _tools = {};
+    _memory;
+    _currentSession;
+    constructor() { }
+    /**
+     * 创建一个新的英雄实例
+     */
+    static create() {
+        return new Hero();
+    }
+    /**
+     * 设置英雄的身份和系统提示词
+     */
+    avatar(prompt) {
+        this._system = prompt;
+        return this;
+    }
+    /**
+     * 设置语言模型
+     */
+    model(model) {
+        this._model = model;
+        return this;
+    }
+    /**
+     * 学习工具
+     */
+    study(tools) {
+        this._tools = { ...this._tools, ...tools };
+        return this;
+    }
+    /**
+     * 设置记忆系统
+     */
+    memory(memory) {
+        this._memory = memory;
+        return this;
+    }
+    /**
+     * 设置当前会话
+     */
+    session(session) {
+        this._currentSession = session;
+        return this;
+    }
+    /**
+     * 与英雄对话
+     */
+    async chat(message) {
+        if (!this._model) {
+            throw new Error("请先设置语言模型");
+        }
+        try {
+            // 添加消息到当前会话
+            if (this._currentSession) {
+                this._currentSession.messages.push({
+                    role: "user",
+                    content: message,
+                    timestamp: new Date(),
+                });
+                this._currentSession.updatedAt = new Date();
+            }
+            // 调用AI生成回复，启用多步调用来自动处理工具调用
+            const result = await generateText({
+                model: this._model,
+                system: this._system,
+                prompt: message,
+                tools: this._tools,
+                stopWhen: stepCountIs(5), // 允许最多5步的工具调用
+            });
+            // 添加回复到当前会话
+            if (this._currentSession) {
+                this._currentSession.messages.push({
+                    role: "assistant",
+                    content: result.text,
+                    timestamp: new Date(),
+                });
+                this._currentSession.updatedAt = new Date();
+            }
+            return result.text;
+        }
+        catch (error) {
+            console.error("Hero chat error:", error);
+            throw error;
+        }
+    }
+    /**
+     * 生成JSON对象
+     */
+    async json(prompt, schema) {
+        if (!this._model) {
+            throw new Error("请先设置语言模型");
+        }
+        try {
+            const { object } = await generateObject({
+                model: this._model,
+                system: this._system,
+                prompt,
+                schema,
+            });
+            return object;
+        }
+        catch (error) {
+            console.error("Hero json error:", error);
+            throw error;
+        }
+    }
+    /**
+     * 基于特定标准进行判断
+     */
+    async check(content, criteria) {
+        if (!this._model) {
+            throw new Error("请先设置语言模型");
+        }
+        try {
+            const prompt = `请根据以下标准判断内容是否符合要求：\n\n标准：${criteria}\n\n内容：${content}\n\n请回答：符合 或 不符合`;
+            const { text } = await generateText({
+                model: this._model,
+                system: "你是一个严格的内容审核员，只能回答'符合'或'不符合'。",
+                prompt,
+            });
+            return text.includes("符合") && !text.includes("不符合");
+        }
+        catch (error) {
+            console.error("Hero check error:", error);
+            throw error;
+        }
+    }
+    /**
+     * 启动服务器
+     */
+    async ready(port = 3000) {
+        // TODO: 实现HTTP服务器
+        console.log(`🏰 DownCity Hero is ready on port ${port}`);
+        console.log(`🦸 Avatar: ${this._system}`);
+        console.log(`🧠 Model: ${this._model ? "Configured" : "Not configured"}`);
+        console.log(`🛠️  Tools: ${this._tools.length} tools loaded`);
+    }
+    // Getters for debugging and inspection
+    get systemPrompt() {
+        return this._system;
+    }
+    get tools() {
+        return Object.keys(this._tools);
+    }
+    get currentSession() {
+        return this._currentSession;
+    }
+}
+//# sourceMappingURL=Hero.js.map

package/dist/Hero.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Hero.js","sourceRoot":"","sources":["../src/Hero.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,YAAY,EAAE,cAAc,EAAQ,WAAW,EAAE,MAAM,IAAI,CAAC;AAEpF,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAa9C,MAAM,OAAO,IAAI;IACP,MAAM,GAAkB,YAAY,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACtD,OAAO,GAAW,UAAU,CAAC;IAC7B,MAAM,GAAyB,EAAE,CAAC;IAClC,OAAO,CAAU;IACjB,eAAe,CAAW;IAElC,gBAAuB,CAAC;IAExB;;OAEG;IACH,MAAM,CAAC,MAAM;QACX,OAAO,IAAI,IAAI,EAAE,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,MAAc;QACnB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAoB;QACxB,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAA2B;QAC/B,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;QAC3C,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,MAAc;QACnB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,OAAO,CAAC,OAAgB;QACtB,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC;QAC/B,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,OAAe;QACxB,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;QAC9B,CAAC;QAED,IAAI,CAAC;YACH,YAAY;YACZ,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBACzB,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC;oBACjC,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO;oBAChB,SAAS,EAAE,IAAI,IAAI,EAAE;iBACtB,CAAC,CAAC;gBACH,IAAI,CAAC,eAAe,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;YAC9C,CAAC;YAED,2BAA2B;YAC3B,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,OAAO;gBACpB,MAAM,EAAE,OAAO;gBACf,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,QAAQ,EAAE,WAAW,CAAC,CAAC,CAAC,EAAE,cAAc;aACzC,CAAC,CAAC;YAEH,YAAY;YACZ,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBACzB,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC;oBACjC,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,MAAM,CAAC,IAAI;oBACpB,SAAS,EAAE,IAAI,IAAI,EAAE;iBACtB,CAAC,CAAC;gBACH,IAAI,CAAC,eAAe,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;YAC9C,CAAC;YAED,OAAO,MAAM,CAAC,IAAI,CAAC;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;YACzC,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CAAI,MAAc,EAAE,MAAW;QACvC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;QAC9B,CAAC;QAED,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,cAAc,CAAC;gBACtC,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,OAAO;gBACpB,MAAM;gBACN,MAAM;aACP,CAAC,CAAC;YAEH,OAAO,MAAW,CAAC;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;YACzC,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,OAAe,EAAE,QAAgB;QAC3C,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;QAC9B,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,4BAA4B,QAAQ,UAAU,OAAO,kBAAkB,CAAC;YAEvF,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,YAAY,CAAC;gBAClC,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,8BAA8B;gBACtC,MAAM;aACP,CAAC,CAAC;YAEH,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;YAC1C,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,OAAe,IAAI;QAC7B,kBAAkB;QAClB,OAAO,CAAC,GAAG,CAAC,qCAAqC,IAAI,EAAE,CAAC,CAAC;QACzD,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1C,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAC1E,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,MAAM,CAAC,MAAM,eAAe,CAAC,CAAC;IAChE,CAAC;IAED,uCAAuC;IACvC,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,IAAI,KAAK;QACP,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IAED,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;CACF"}

package/dist/Knowledge.d.ts

@@ -0,0 +1,139 @@
+import { Tool } from "ai";
+export interface VectorModelConfig {
+    model: any;
+    dimensions?: number;
+}
+export interface ChromaConfig {
+    url?: string;
+    collection?: string;
+    persistPath?: string;
+}
+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 declare class Knowledge {
+    private knowledgeBase;
+    private categories;
+    private options;
+    private chromaClient?;
+    private collection?;
+    private embeddingModel?;
+    private embeddingFunction?;
+    constructor(options?: KnowledgeOptions);
+    /**
+     * 创建Knowledge实例
+     */
+    static create(options?: KnowledgeOptions): Knowledge;
+    /**
+     * 初始化向量数据库
+     */
+    private initializeVectorDB;
+    /**
+     * 配置向量模型
+     */
+    model(config: VectorModelConfig): Knowledge;
+    /**
+     * 保存数据到本地文件
+     */
+    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 cosineSimilarity;
+    /**
+     * 为知识条目生成向量
+     */
+    generateEmbedding(text: string): Promise<number[]>;
+    /**
+     * 添加知识条目并生成向量
+     */
+    addKnowledgeWithVector(item: Omit<KnowledgeItem, "id" | "createdAt" | "updatedAt" | "vector">): Promise<KnowledgeItem>;
+    /**
+     * 根据分类获取知识
+     */
+    getKnowledgeByCategory(category: string): KnowledgeItem[];
+    /**
+     * 获取所有分类
+     */
+    getCategories(): string[];
+    /**
+     * 删除知识条目
+     */
+    removeKnowledge(id: string): boolean;
+    /**
+     * 清空知识库
+     */
+    clearKnowledge(): void;
+    /**
+     * 获取统计信息
+     */
+    getStats(): {
+        totalKnowledge: number;
+        totalCategories: number;
+        knowledgeByCategory: Record<string, number>;
+    };
+    /**
+     * 导出数据
+     */
+    export(): any;
+    /**
+     * 导入数据
+     */
+    import(data: any): void;
+    /**
+     * 生成唯一ID
+     */
+    private generateId;
+}
+//# sourceMappingURL=Knowledge.d.ts.map

package/dist/Knowledge.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"Knowledge.d.ts","sourceRoot":"","sources":["../src/Knowledge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,IAAI,EAAE,MAAM,IAAI,CAAC;AAKvC,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,SAAS;IACpB,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,SAAS;IAIpD;;OAEG;YACW,kBAAkB;IA6ChC;;OAEG;IACH,KAAK,CAAC,MAAM,EAAE,iBAAiB,GAAG,SAAS;IAW3C;;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"}

package/dist/Knowledge.js

@@ -0,0 +1,431 @@
+import { embed, tool } from "ai";
+import z from "zod";
+import * as fs from "fs/promises";
+export class Knowledge {
+    knowledgeBase = new Map();
+    categories = new Set();
+    options;
+    chromaClient;
+    collection;
+    embeddingModel; // ai-sdk模型实例
+    embeddingFunction;
+    constructor(options = {}) {
+        this.options = {
+            autoLoad: true,
+            categories: ["general", "programming", "api", "documentation"],
+            vectorDimension: 1536, // OpenAI embedding dimension
+            similarityThreshold: 0.7,
+            chroma: {
+                collection: "downcity_knowledge",
+                persistPath: "./chroma_db",
+            },
+            localFile: "./knowledge.json",
+            ...options,
+        };
+        // 初始化默认分类
+        if (this.options.categories) {
+            this.options.categories.forEach((cat) => this.categories.add(cat));
+        }
+        // 初始化向量数据库和加载本地数据
+        this.initializeVectorDB();
+        if (this.options.autoLoad) {
+            this.loadFromFile();
+        }
+    }
+    /**
+     * 创建Knowledge实例
+     */
+    static create(options) {
+        return new Knowledge(options);
+    }
+    /**
+     * 初始化向量数据库
+     */
+    async initializeVectorDB() {
+        try {
+            // 动态导入chromadb
+            const { ChromaClient } = await import("chromadb");
+            if (this.options.chroma?.persistPath) {
+                this.chromaClient = new ChromaClient({
+                    path: this.options.chroma.persistPath,
+                });
+            }
+            else {
+                this.chromaClient = new ChromaClient();
+            }
+            // 获取或创建集合
+            const collectionName = this.options.chroma?.collection || "downcity_knowledge";
+            try {
+                const getParams = { name: collectionName };
+                if (this.embeddingFunction) {
+                    getParams.embeddingFunction = this.embeddingFunction;
+                }
+                this.collection = await this.chromaClient.getCollection(getParams);
+            }
+            catch {
+                // 集合不存在，创建新集合
+                const createParams = {
+                    name: collectionName,
+                    metadata: { description: "DownCity Knowledge Base" },
+                };
+                if (this.embeddingFunction) {
+                    createParams.embeddingFunction = this.embeddingFunction;
+                }
+                this.collection = await this.chromaClient.createCollection(createParams);
+            }
+            // 初始化embedding模型
+            if (this.options.vectorModel?.model) {
+                this.embeddingModel = this.options.vectorModel.model;
+            }
+        }
+        catch (error) {
+            console.warn("Failed to initialize vector database:", error);
+        }
+    }
+    /**
+     * 配置向量模型
+     */
+    model(config) {
+        this.options.vectorModel = { ...this.options.vectorModel, ...config };
+        // 设置embedding模型
+        if (config.model) {
+            this.embeddingModel = config.model;
+        }
+        return this;
+    }
+    /**
+     * 保存数据到本地文件
+     */
+    async saveToFile() {
+        if (!this.options.localFile)
+            return;
+        try {
+            const data = this.export();
+            await fs.writeFile(this.options.localFile, JSON.stringify(data, null, 2));
+        }
+        catch (error) {
+            console.warn("Failed to save to local file:", error);
+        }
+    }
+    /**
+     * 从本地文件加载数据
+     */
+    async loadFromFile() {
+        if (!this.options.localFile)
+            return;
+        try {
+            const data = await fs.readFile(this.options.localFile, "utf-8");
+            this.import(JSON.parse(data));
+        }
+        catch (error) {
+            // 文件不存在时忽略
+        }
+    }
+    /**
+     * 获取知识库操作工具
+     * 将知识库的操作方法包装成工具供Hero调用
+     */
+    tools() {
+        return {
+            search_knowledge: tool({
+                description: "在知识库中搜索相关信息",
+                inputSchema: z.object({
+                    query: z.string().describe("The search query"),
+                    category: z.string().optional().describe("The category to filter"),
+                }),
+                execute: async (params) => {
+                    const results = await this.searchKnowledge(params.query, params.category);
+                    return {
+                        results: results.slice(0, 5), // 限制返回前5个结果
+                        total: results.length,
+                    };
+                },
+            }),
+            getKnowledgeCategories: tool({
+                description: "获取知识库中所有可用的分类",
+                inputSchema: z.object({}),
+                execute: async () => {
+                    return {
+                        categories: this.getCategories(),
+                    };
+                },
+            }),
+            get_knowledge_stats: tool({
+                description: "获取知识库统计信息",
+                inputSchema: z.object({}),
+                execute: async () => {
+                    return this.getStats();
+                },
+            }),
+        };
+    }
+    /**
+     * 添加知识条目
+     */
+    async addKnowledge(item) {
+        const id = this.generateId();
+        const now = new Date();
+        const knowledgeItem = {
+            id,
+            ...item,
+            createdAt: now,
+            updatedAt: now,
+        };
+        // 生成向量
+        if (!knowledgeItem.vector) {
+            knowledgeItem.vector = await this.generateEmbedding(`${item.title} ${item.content}`);
+        }
+        // 存储到本地缓存
+        this.knowledgeBase.set(id, knowledgeItem);
+        this.categories.add(item.category);
+        // 存储到Chroma数据库
+        if (this.collection) {
+            try {
+                await this.collection.add({
+                    ids: [id],
+                    embeddings: [knowledgeItem.vector],
+                    metadatas: [
+                        {
+                            title: item.title,
+                            category: item.category,
+                            tags: item.tags.join(","),
+                            createdAt: now.toISOString(),
+                            updatedAt: now.toISOString(),
+                        },
+                    ],
+                    documents: [item.content],
+                });
+            }
+            catch (error) {
+                console.warn("Failed to add to Chroma database:", error);
+            }
+        }
+        // 保存到本地文件
+        await this.saveToFile();
+        return knowledgeItem;
+    }
+    /**
+     * 更新知识条目
+     */
+    updateKnowledge(id, updates) {
+        const existing = this.knowledgeBase.get(id);
+        if (!existing)
+            return undefined;
+        const updated = {
+            ...existing,
+            ...updates,
+            updatedAt: new Date(),
+        };
+        this.knowledgeBase.set(id, updated);
+        if (updates.category) {
+            this.categories.add(updates.category);
+        }
+        return updated;
+    }
+    /**
+     * 获取知识条目
+     */
+    getKnowledge(id) {
+        return this.knowledgeBase.get(id);
+    }
+    /**
+     * 搜索知识库（支持向量搜索和文本匹配）
+     */
+    async searchKnowledge(query, category, useVector = true) {
+        // 优先使用向量搜索
+        if (useVector && this.collection) {
+            try {
+                const queryVector = await this.generateEmbedding(query);
+                const chromaResults = await this.collection.query({
+                    queryEmbeddings: [queryVector],
+                    nResults: 10,
+                    where: category ? { category } : undefined,
+                });
+                // 将Chroma结果转换为KnowledgeItem
+                const results = [];
+                if (chromaResults.ids && chromaResults.ids[0]) {
+                    for (let i = 0; i < chromaResults.ids[0].length; i++) {
+                        const id = chromaResults.ids[0][i];
+                        const item = this.knowledgeBase.get(id);
+                        if (item) {
+                            results.push(item);
+                        }
+                    }
+                }
+                if (results.length > 0) {
+                    return results;
+                }
+            }
+            catch (error) {
+                console.warn("Vector search failed, falling back to text search:", error);
+            }
+        }
+        // 降级到文本搜索
+        return this.textSearchKnowledge(query, category);
+    }
+    /**
+     * 文本搜索知识库
+     */
+    textSearchKnowledge(query, category) {
+        const results = [];
+        const lowerQuery = query.toLowerCase();
+        for (const item of this.knowledgeBase.values()) {
+            // 分类过滤
+            if (category && item.category !== category)
+                continue;
+            // 文本匹配
+            const matchesTitle = item.title.toLowerCase().includes(lowerQuery);
+            const matchesContent = item.content.toLowerCase().includes(lowerQuery);
+            const matchesTags = item.tags.some((tag) => tag.toLowerCase().includes(lowerQuery));
+            if (matchesTitle || matchesContent || matchesTags) {
+                results.push(item);
+            }
+        }
+        // 按更新时间排序
+        return results.sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime());
+    }
+    /**
+     * 向量相似度搜索
+     */
+    searchByVector(queryVector, category, limit = 10) {
+        const results = [];
+        for (const item of this.knowledgeBase.values()) {
+            // 分类过滤
+            if (category && item.category !== category)
+                continue;
+            // 跳过没有向量的条目
+            if (!item.vector)
+                continue;
+            // 计算余弦相似度
+            const similarity = this.cosineSimilarity(queryVector, item.vector);
+            // 只保留超过阈值的结果
+            if (similarity >= (this.options.similarityThreshold || 0.7)) {
+                results.push({ ...item, similarity });
+            }
+        }
+        // 按相似度排序并限制结果数量
+        return results.sort((a, b) => b.similarity - a.similarity).slice(0, limit);
+    }
+    /**
+     * 计算余弦相似度
+     */
+    cosineSimilarity(vecA, vecB) {
+        if (vecA.length !== vecB.length) {
+            throw new Error("向量维度不匹配");
+        }
+        let dotProduct = 0;
+        let normA = 0;
+        let normB = 0;
+        for (let i = 0; i < vecA.length; i++) {
+            dotProduct += vecA[i] * vecB[i];
+            normA += vecA[i] * vecA[i];
+            normB += vecB[i] * vecB[i];
+        }
+        normA = Math.sqrt(normA);
+        normB = Math.sqrt(normB);
+        if (normA === 0 || normB === 0) {
+            return 0;
+        }
+        return dotProduct / (normA * normB);
+    }
+    /**
+     * 为知识条目生成向量
+     */
+    async generateEmbedding(text) {
+        if (this.embeddingModel) {
+            const { embedding } = await embed({
+                model: this.embeddingModel,
+                value: text,
+            });
+            return embedding;
+        }
+        throw new Error("Embedding model not set");
+    }
+    /**
+     * 添加知识条目并生成向量
+     */
+    async addKnowledgeWithVector(item) {
+        const vector = await this.generateEmbedding(`${item.title} ${item.content}`);
+        return this.addKnowledge({ ...item, vector });
+    }
+    /**
+     * 根据分类获取知识
+     */
+    getKnowledgeByCategory(category) {
+        return Array.from(this.knowledgeBase.values())
+            .filter((item) => item.category === category)
+            .sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime());
+    }
+    /**
+     * 获取所有分类
+     */
+    getCategories() {
+        return Array.from(this.categories);
+    }
+    /**
+     * 删除知识条目
+     */
+    removeKnowledge(id) {
+        return this.knowledgeBase.delete(id);
+    }
+    /**
+     * 清空知识库
+     */
+    clearKnowledge() {
+        this.knowledgeBase.clear();
+    }
+    /**
+     * 获取统计信息
+     */
+    getStats() {
+        const knowledgeByCategory = {};
+        for (const item of this.knowledgeBase.values()) {
+            knowledgeByCategory[item.category] =
+                (knowledgeByCategory[item.category] || 0) + 1;
+        }
+        return {
+            totalKnowledge: this.knowledgeBase.size,
+            totalCategories: this.categories.size,
+            knowledgeByCategory,
+        };
+    }
+    /**
+     * 导出数据
+     */
+    export() {
+        return {
+            knowledgeBase: Array.from(this.knowledgeBase.entries()),
+            categories: Array.from(this.categories),
+            exportedAt: new Date().toISOString(),
+        };
+    }
+    /**
+     * 导入数据
+     */
+    import(data) {
+        if (data.knowledgeBase && Array.isArray(data.knowledgeBase)) {
+            this.knowledgeBase.clear();
+            for (const [id, item] of data.knowledgeBase) {
+                // 确保日期对象正确转换
+                if (item.createdAt && typeof item.createdAt === "string") {
+                    item.createdAt = new Date(item.createdAt);
+                }
+                if (item.updatedAt && typeof item.updatedAt === "string") {
+                    item.updatedAt = new Date(item.updatedAt);
+                }
+                this.knowledgeBase.set(id, item);
+            }
+        }
+        if (data.categories && Array.isArray(data.categories)) {
+            this.categories.clear();
+            data.categories.forEach((cat) => this.categories.add(cat));
+        }
+    }
+    /**
+     * 生成唯一ID
+     */
+    generateId() {
+        return `knowledge_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
+    }
+}
+//# sourceMappingURL=Knowledge.js.map

package/dist/Knowledge.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Knowledge.js","sourceRoot":"","sources":["../src/Knowledge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,IAAI,EAAQ,MAAM,IAAI,CAAC;AACvC,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAmClC,MAAM,OAAO,SAAS;IACZ,aAAa,GAA+B,IAAI,GAAG,EAAE,CAAC;IACtD,UAAU,GAAgB,IAAI,GAAG,EAAE,CAAC;IACpC,OAAO,CAAmB;IAC1B,YAAY,CAAgB;IAC5B,UAAU,CAAc;IACxB,cAAc,CAAO,CAAC,aAAa;IACnC,iBAAiB,CAA2B;IAEpD,YAAY,UAA4B,EAAE;QACxC,IAAI,CAAC,OAAO,GAAG;YACb,QAAQ,EAAE,IAAI;YACd,UAAU,EAAE,CAAC,SAAS,EAAE,aAAa,EAAE,KAAK,EAAE,eAAe,CAAC;YAC9D,eAAe,EAAE,IAAI,EAAE,6BAA6B;YACpD,mBAAmB,EAAE,GAAG;YACxB,MAAM,EAAE;gBACN,UAAU,EAAE,oBAAoB;gBAChC,WAAW,EAAE,aAAa;aAC3B;YACD,SAAS,EAAE,kBAAkB;YAC7B,GAAG,OAAO;SACX,CAAC;QAEF,UAAU;QACV,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;YAC5B,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;QACrE,CAAC;QAED,kBAAkB;QAClB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,MAAM,CAAC,OAA0B;QACtC,OAAO,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB;QAC9B,IAAI,CAAC;YACH,eAAe;YACf,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;YAElD,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,WAAW,EAAE,CAAC;gBACrC,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAAC;oBACnC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW;iBACtC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;YACzC,CAAC;YAED,UAAU;YACV,MAAM,cAAc,GAClB,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,IAAI,oBAAoB,CAAC;YAC1D,IAAI,CAAC;gBACH,MAAM,SAAS,GAAQ,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;gBAChD,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;oBAC3B,SAAS,CAAC,iBAAiB,GAAG,IAAI,CAAC,iBAAiB,CAAC;gBACvD,CAAC;gBACD,IAAI,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YACrE,CAAC;YAAC,MAAM,CAAC;gBACP,cAAc;gBACd,MAAM,YAAY,GAAQ;oBACxB,IAAI,EAAE,cAAc;oBACpB,QAAQ,EAAE,EAAE,WAAW,EAAE,yBAAyB,EAAE;iBACrD,CAAC;gBACF,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;oBAC3B,YAAY,CAAC,iBAAiB,GAAG,IAAI,CAAC,iBAAiB,CAAC;gBAC1D,CAAC;gBACD,IAAI,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,gBAAgB,CACxD,YAAY,CACb,CAAC;YACJ,CAAC;YAED,iBAAiB;YACjB,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC;gBACpC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC;YACvD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,uCAAuC,EAAE,KAAK,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAyB;QAC7B,IAAI,CAAC,OAAO,CAAC,WAAW,GAAG,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,GAAG,MAAM,EAAE,CAAC;QAEtE,gBAAgB;QAChB,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,KAAK,CAAC;QACrC,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,UAAU;QACtB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS;YAAE,OAAO;QAEpC,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;YAC3B,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC5E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,YAAY;QACxB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS;YAAE,OAAO;QAEpC,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YAChE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,WAAW;QACb,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK;QACH,OAAO;YACL,gBAAgB,EAAE,IAAI,CAAC;gBACrB,WAAW,EAAE,aAAa;gBAC1B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;oBACpB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;oBAC9C,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;iBACnE,CAAC;gBACF,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;oBACxB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,CACxC,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,QAAQ,CAChB,CAAC;oBACF,OAAO;wBACL,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,YAAY;wBAC1C,KAAK,EAAE,OAAO,CAAC,MAAM;qBACtB,CAAC;gBACJ,CAAC;aACF,CAAC;YACF,sBAAsB,EAAE,IAAI,CAAC;gBAC3B,WAAW,EAAE,eAAe;gBAC5B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;gBACzB,OAAO,EAAE,KAAK,IAAI,EAAE;oBAClB,OAAO;wBACL,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE;qBACjC,CAAC;gBACJ,CAAC;aACF,CAAC;YACF,mBAAmB,EAAE,IAAI,CAAC;gBACxB,WAAW,EAAE,WAAW;gBACxB,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;gBACzB,OAAO,EAAE,KAAK,IAAI,EAAE;oBAClB,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACzB,CAAC;aACF,CAAC;SACH,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAChB,IAA2D;QAE3D,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,aAAa,GAAkB;YACnC,EAAE;YACF,GAAG,IAAI;YACP,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;SACf,CAAC;QAEF,OAAO;QACP,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;YAC1B,aAAa,CAAC,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CACjD,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,OAAO,EAAE,CAChC,CAAC;QACJ,CAAC;QAED,UAAU;QACV,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC;QAC1C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAEnC,eAAe;QACf,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;oBACxB,GAAG,EAAE,CAAC,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC,aAAa,CAAC,MAAM,CAAC;oBAClC,SAAS,EAAE;wBACT;4BACE,KAAK,EAAE,IAAI,CAAC,KAAK;4BACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;4BACvB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;4BACzB,SAAS,EAAE,GAAG,CAAC,WAAW,EAAE;4BAC5B,SAAS,EAAE,GAAG,CAAC,WAAW,EAAE;yBAC7B;qBACF;oBACD,SAAS,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;iBAC1B,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,IAAI,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;QAED,UAAU;QACV,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAExB,OAAO,aAAa,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,eAAe,CACb,EAAU,EACV,OAAyD;QAEzD,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,QAAQ;YAAE,OAAO,SAAS,CAAC;QAEhC,MAAM,OAAO,GAAkB;YAC7B,GAAG,QAAQ;YACX,GAAG,OAAO;YACV,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACpC,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACxC,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,EAAU;QACrB,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CACnB,KAAa,EACb,QAAiB,EACjB,YAAqB,IAAI;QAEzB,WAAW;QACX,IAAI,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACjC,IAAI,CAAC;gBACH,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;gBACxD,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;oBAChD,eAAe,EAAE,CAAC,WAAW,CAAC;oBAC9B,QAAQ,EAAE,EAAE;oBACZ,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS;iBAC3C,CAAC,CAAC;gBAEH,4BAA4B;gBAC5B,MAAM,OAAO,GAAoB,EAAE,CAAC;gBACpC,IAAI,aAAa,CAAC,GAAG,IAAI,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;wBACrD,MAAM,EAAE,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;wBACnC,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;wBACxC,IAAI,IAAI,EAAE,CAAC;4BACT,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBACrB,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACvB,OAAO,OAAO,CAAC;gBACjB,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,IAAI,CACV,oDAAoD,EACpD,KAAK,CACN,CAAC;YACJ,CAAC;QACH,CAAC;QAED,UAAU;QACV,OAAO,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACK,mBAAmB,CACzB,KAAa,EACb,QAAiB;QAEjB,MAAM,OAAO,GAAoB,EAAE,CAAC;QACpC,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;QAEvC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/C,OAAO;YACP,IAAI,QAAQ,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ;gBAAE,SAAS;YAErD,OAAO;YACP,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YACnE,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YACvE,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CACzC,GAAG,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,CACvC,CAAC;YAEF,IAAI,YAAY,IAAI,cAAc,IAAI,WAAW,EAAE,CAAC;gBAClD,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QAED,UAAU;QACV,OAAO,OAAO,CAAC,IAAI,CACjB,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,CACxD,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,cAAc,CACZ,WAAqB,EACrB,QAAiB,EACjB,QAAgB,EAAE;QAElB,MAAM,OAAO,GAAkD,EAAE,CAAC;QAElE,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/C,OAAO;YACP,IAAI,QAAQ,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ;gBAAE,SAAS;YAErD,YAAY;YACZ,IAAI,CAAC,IAAI,CAAC,MAAM;gBAAE,SAAS;YAE3B,UAAU;YACV,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YAEnE,aAAa;YACb,IAAI,UAAU,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC5D,OAAO,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QAED,gBAAgB;QAChB,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAC7E,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,IAAc,EAAE,IAAc;QACrD,IAAI,IAAI,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;QAC7B,CAAC;QAED,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,UAAU,IAAI,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YAChC,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC;QAED,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEzB,IAAI,KAAK,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,CAAC,CAAC;QACX,CAAC;QAED,OAAO,UAAU,GAAG,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,iBAAiB,CAAC,IAAY;QAClC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,KAAK,CAAC;gBAChC,KAAK,EAAE,IAAI,CAAC,cAAc;gBAC1B,KAAK,EAAE,IAAI;aACZ,CAAC,CAAC;YACH,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,sBAAsB,CAC1B,IAAsE;QAEtE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CACzC,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,OAAO,EAAE,CAChC,CAAC;QACF,OAAO,IAAI,CAAC,YAAY,CAAC,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;IAChD,CAAC;IAED;;OAEG;IACH,sBAAsB,CAAC,QAAgB;QACrC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;aAC3C,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC;aAC5C,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC;IACnE,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,EAAU;QACxB,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,QAAQ;QAKN,MAAM,mBAAmB,GAA2B,EAAE,CAAC;QAEvD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/C,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAChC,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QAClD,CAAC;QAED,OAAO;YACL,cAAc,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI;YACvC,eAAe,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI;YACrC,mBAAmB;SACpB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO;YACL,aAAa,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;YACvD,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;YACvC,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACrC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,IAAS;QACd,IAAI,IAAI,CAAC,aAAa,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC;YAC5D,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;YAC3B,KAAK,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;gBAC5C,aAAa;gBACb,IAAI,IAAI,CAAC,SAAS,IAAI,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;oBACzD,IAAI,CAAC,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC5C,CAAC;gBACD,IAAI,IAAI,CAAC,SAAS,IAAI,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;oBACzD,IAAI,CAAC,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC5C,CAAC;gBACD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YACtD,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACxB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,GAAW,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;IAED;;OAEG;IACK,UAAU;QAChB,OAAO,aAAa,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IAC9E,CAAC;CACF"}

package/dist/Memory.d.ts

@@ -0,0 +1,61 @@
+import { Session } from "./Hero.js";
+export interface MemoryOptions {
+    maxSessions?: number;
+    persistToFile?: boolean;
+    filePath?: string;
+}
+export declare class Memory {
+    private sessions;
+    private lastSessionId?;
+    private options;
+    constructor(options?: MemoryOptions);
+    /**
+     * 创建新的会话
+     */
+    newSession(id?: string): Session;
+    /**
+     * 获取最后一个会话，如果不存在则创建新会话
+     */
+    lastSession(): Session;
+    /**
+     * 根据ID获取会话
+     */
+    getSession(id: string): Session | undefined;
+    /**
+     * 获取所有会话
+     */
+    getAllSessions(): Session[];
+    /**
+     * 删除会话
+     */
+    deleteSession(id: string): boolean;
+    /**
+     * 清空所有会话
+     */
+    clear(): void;
+    /**
+     * 获取会话统计信息
+     */
+    getStats(): {
+        totalSessions: number;
+        totalMessages: number;
+        lastActivity: Date | null;
+    };
+    /**
+     * 生成唯一的会话ID
+     */
+    private generateSessionId;
+    /**
+     * 清理旧会话
+     */
+    private cleanupOldSessions;
+    /**
+     * 导出会话数据
+     */
+    export(): any;
+    /**
+     * 导入会话数据
+     */
+    import(data: any): void;
+}
+//# sourceMappingURL=Memory.d.ts.map

package/dist/Memory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"Memory.d.ts","sourceRoot":"","sources":["../src/Memory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEpC,MAAM,WAAW,aAAa;IAC5B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,qBAAa,MAAM;IACjB,OAAO,CAAC,QAAQ,CAAmC;IACnD,OAAO,CAAC,aAAa,CAAC,CAAS;IAC/B,OAAO,CAAC,OAAO,CAAgB;gBAEnB,OAAO,GAAE,aAAkB;IAQvC;;OAEG;IACH,UAAU,CAAC,EAAE,CAAC,EAAE,MAAM,GAAG,OAAO;IAoBhC;;OAEG;IACH,WAAW,IAAI,OAAO;IAOtB;;OAEG;IACH,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,GAAG,SAAS;IAI3C;;OAEG;IACH,cAAc,IAAI,OAAO,EAAE;IAK3B;;OAEG;IACH,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAUlC;;OAEG;IACH,KAAK,IAAI,IAAI;IAKb;;OAEG;IACH,QAAQ,IAAI;QACV,aAAa,EAAE,MAAM,CAAC;QACtB,aAAa,EAAE,MAAM,CAAC;QACtB,YAAY,EAAE,IAAI,GAAG,IAAI,CAAC;KAC3B;IAmBD;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAIzB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAY1B;;OAEG;IACH,MAAM,IAAI,GAAG;IAQb;;OAEG;IACH,MAAM,CAAC,IAAI,EAAE,GAAG,GAAG,IAAI;CAwBxB"}

package/dist/Memory.js

@@ -0,0 +1,148 @@
+export class Memory {
+    sessions = new Map();
+    lastSessionId;
+    options;
+    constructor(options = {}) {
+        this.options = {
+            maxSessions: 100,
+            persistToFile: false,
+            ...options
+        };
+    }
+    /**
+     * 创建新的会话
+     */
+    newSession(id) {
+        const sessionId = id || this.generateSessionId();
+        const session = {
+            id: sessionId,
+            messages: [],
+            createdAt: new Date(),
+            updatedAt: new Date()
+        };
+        this.sessions.set(sessionId, session);
+        this.lastSessionId = sessionId;
+        // 如果会话数量超过限制，删除最旧的会话
+        if (this.sessions.size > (this.options.maxSessions || 100)) {
+            this.cleanupOldSessions();
+        }
+        return session;
+    }
+    /**
+     * 获取最后一个会话，如果不存在则创建新会话
+     */
+    lastSession() {
+        if (this.lastSessionId && this.sessions.has(this.lastSessionId)) {
+            return this.sessions.get(this.lastSessionId);
+        }
+        return this.newSession();
+    }
+    /**
+     * 根据ID获取会话
+     */
+    getSession(id) {
+        return this.sessions.get(id);
+    }
+    /**
+     * 获取所有会话
+     */
+    getAllSessions() {
+        return Array.from(this.sessions.values())
+            .sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime());
+    }
+    /**
+     * 删除会话
+     */
+    deleteSession(id) {
+        const deleted = this.sessions.delete(id);
+        if (deleted && this.lastSessionId === id) {
+            // 如果删除的是最后一个会话，更新lastSessionId
+            const allSessions = this.getAllSessions();
+            this.lastSessionId = allSessions.length > 0 ? allSessions[0].id : undefined;
+        }
+        return deleted;
+    }
+    /**
+     * 清空所有会话
+     */
+    clear() {
+        this.sessions.clear();
+        this.lastSessionId = undefined;
+    }
+    /**
+     * 获取会话统计信息
+     */
+    getStats() {
+        const totalSessions = this.sessions.size;
+        let totalMessages = 0;
+        let lastActivity = null;
+        for (const session of this.sessions.values()) {
+            totalMessages += session.messages.length;
+            if (!lastActivity || session.updatedAt > lastActivity) {
+                lastActivity = session.updatedAt;
+            }
+        }
+        return {
+            totalSessions,
+            totalMessages,
+            lastActivity
+        };
+    }
+    /**
+     * 生成唯一的会话ID
+     */
+    generateSessionId() {
+        return `session_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
+    }
+    /**
+     * 清理旧会话
+     */
+    cleanupOldSessions() {
+        const sessions = this.getAllSessions();
+        const maxSessions = this.options.maxSessions || 100;
+        if (sessions.length > maxSessions) {
+            const sessionsToDelete = sessions.slice(maxSessions);
+            for (const session of sessionsToDelete) {
+                this.sessions.delete(session.id);
+            }
+        }
+    }
+    /**
+     * 导出会话数据
+     */
+    export() {
+        return {
+            sessions: Array.from(this.sessions.entries()),
+            lastSessionId: this.lastSessionId,
+            exportedAt: new Date().toISOString()
+        };
+    }
+    /**
+     * 导入会话数据
+     */
+    import(data) {
+        if (data.sessions && Array.isArray(data.sessions)) {
+            this.sessions.clear();
+            for (const [id, session] of data.sessions) {
+                // 确保日期对象正确转换
+                if (session.createdAt && typeof session.createdAt === 'string') {
+                    session.createdAt = new Date(session.createdAt);
+                }
+                if (session.updatedAt && typeof session.updatedAt === 'string') {
+                    session.updatedAt = new Date(session.updatedAt);
+                }
+                // 转换消息中的时间戳
+                if (session.messages) {
+                    session.messages.forEach((msg) => {
+                        if (msg.timestamp && typeof msg.timestamp === 'string') {
+                            msg.timestamp = new Date(msg.timestamp);
+                        }
+                    });
+                }
+                this.sessions.set(id, session);
+            }
+            this.lastSessionId = data.lastSessionId;
+        }
+    }
+}
+//# sourceMappingURL=Memory.js.map

package/dist/Memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Memory.js","sourceRoot":"","sources":["../src/Memory.ts"],"names":[],"mappings":"AAQA,MAAM,OAAO,MAAM;IACT,QAAQ,GAAyB,IAAI,GAAG,EAAE,CAAC;IAC3C,aAAa,CAAU;IACvB,OAAO,CAAgB;IAE/B,YAAY,UAAyB,EAAE;QACrC,IAAI,CAAC,OAAO,GAAG;YACb,WAAW,EAAE,GAAG;YAChB,aAAa,EAAE,KAAK;YACpB,GAAG,OAAO;SACX,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,EAAW;QACpB,MAAM,SAAS,GAAG,EAAE,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACjD,MAAM,OAAO,GAAY;YACvB,EAAE,EAAE,SAAS;YACb,QAAQ,EAAE,EAAE;YACZ,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QACtC,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;QAE/B,qBAAqB;QACrB,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,IAAI,GAAG,CAAC,EAAE,CAAC;YAC3D,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC5B,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,WAAW;QACT,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC;YAChE,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAE,CAAC;QAChD,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,EAAE,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,EAAU;QACnB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;aACtC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC;IACnE,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,EAAU;QACtB,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACzC,IAAI,OAAO,IAAI,IAAI,CAAC,aAAa,KAAK,EAAE,EAAE,CAAC;YACzC,+BAA+B;YAC/B,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;YAC1C,IAAI,CAAC,aAAa,GAAG,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAC9E,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,QAAQ;QAKN,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QACzC,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,IAAI,YAAY,GAAgB,IAAI,CAAC;QAErC,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,aAAa,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;YACzC,IAAI,CAAC,YAAY,IAAI,OAAO,CAAC,SAAS,GAAG,YAAY,EAAE,CAAC;gBACtD,YAAY,GAAG,OAAO,CAAC,SAAS,CAAC;YACnC,CAAC;QACH,CAAC;QAED,OAAO;YACL,aAAa;YACb,aAAa;YACb,YAAY;SACb,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,iBAAiB;QACvB,OAAO,WAAW,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IAC5E,CAAC;IAED;;OAEG;IACK,kBAAkB;QACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QACvC,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,IAAI,GAAG,CAAC;QAEpD,IAAI,QAAQ,CAAC,MAAM,GAAG,WAAW,EAAE,CAAC;YAClC,MAAM,gBAAgB,GAAG,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YACrD,KAAK,MAAM,OAAO,IAAI,gBAAgB,EAAE,CAAC;gBACvC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO;YACL,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC7C,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACrC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,IAAS;QACd,IAAI,IAAI,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YAClD,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;YACtB,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC1C,aAAa;gBACb,IAAI,OAAO,CAAC,SAAS,IAAI,OAAO,OAAO,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;oBAC/D,OAAO,CAAC,SAAS,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAClD,CAAC;gBACD,IAAI,OAAO,CAAC,SAAS,IAAI,OAAO,OAAO,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;oBAC/D,OAAO,CAAC,SAAS,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAClD,CAAC;gBACD,YAAY;gBACZ,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;oBACrB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,GAAQ,EAAE,EAAE;wBACpC,IAAI,GAAG,CAAC,SAAS,IAAI,OAAO,GAAG,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;4BACvD,GAAG,CAAC,SAAS,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;wBAC1C,CAAC;oBACH,CAAC,CAAC,CAAC;gBACL,CAAC;gBACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACjC,CAAC;YACD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;QAC1C,CAAC;IACH,CAAC;CACF"}

package/dist/index.d.ts

@@ -0,0 +1,7 @@
+export { Hero } from "./Hero.js";
+export { Memory } from "./Memory.js";
+export { Knowledge } from "./Knowledge.js";
+export type { Session } from "./Hero.js";
+export type { MemoryOptions } from "./Memory.js";
+export type { KnowledgeOptions, KnowledgeItem } from "./Knowledge.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAG3C,YAAY,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACzC,YAAY,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AACjD,YAAY,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC"}

package/dist/index.js

@@ -0,0 +1,5 @@
+// 导出主要的类
+export { Hero } from "./Hero.js";
+export { Memory } from "./Memory.js";
+export { Knowledge } from "./Knowledge.js";
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,SAAS;AACT,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC"}

package/package.json

@@ -0,0 +1,51 @@
+{
+  "name": "downcity",
+  "version": "1.0.0",
+  "description": "A npm package for downcity agent framework",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "docs:dev": "vitepress dev docs",
+    "docs:build": "vitepress build docs",
+    "docs:preview": "vitepress preview docs",
+    "prepublishOnly": "npm run build",
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "keywords": [
+    "agent framework",
+    "downcity",
+    "agent",
+    "typescript"
+  ],
+  "author": "wangenius",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/wangenius/downcity.git"
+  },
+  "bugs": {
+    "url": "https://github.com/wangenius/downcity/issues"
+  },
+  "homepage": "https://github.com/wangenius/downcity#readme",
+  "dependencies": {
+    "@ai-sdk/openai": "^2.0.22",
+    "ai": "^5.0.24",
+    "chromadb": "^1.8.1",
+    "zod": "^4.1.3"
+  },
+  "devDependencies": {
+    "@types/node": "^20.0.0",
+    "typescript": "^5.0.0"
+  },
+  "engines": {
+    "node": ">=16.0.0"
+  }
+}