@t0ken.ai/memoryx-sdk 1.2.2 → 1.3.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -4,6 +4,7 @@ export declare class APIClient {
4
4
  private apiBaseUrl;
5
5
  private projectId;
6
6
  private userId;
7
+ private readonly machineFingerprint;
7
8
  constructor(config?: Partial<StoredConfig>);
8
9
  getApiKey(): string | null;
9
10
  getProjectId(): string;
@@ -13,8 +14,18 @@ export declare class APIClient {
13
14
  setApiKey(key: string): void;
14
15
  setProjectId(id: string): void;
15
16
  setUserId(id: string): void;
17
+ /**
18
+ * 获取机器指纹(只读,返回构造时生成的值)
19
+ * 外部无法修改机器指纹
20
+ */
16
21
  getMachineFingerprint(): string;
17
- autoRegister(agentType?: string, agentName?: string): Promise<AccountInfo>;
22
+ /**
23
+ * 私有方法:生成机器指纹
24
+ * 调用 storage.ts 中统一的指纹生成函数
25
+ * 包含: MAC地址、平台、架构、主机名、CPU型号、内存、用户名
26
+ */
27
+ private generateMachineFingerprint;
28
+ autoRegister(agentType?: string, agentName?: string, agentId?: string): Promise<AccountInfo>;
18
29
  sendMemories(memories: Memory[]): Promise<AddResponse>;
19
30
  sendConversation(conversationId: string, messages: Message[]): Promise<AddResponse>;
20
31
  search(query: string, limit?: number): Promise<SearchResponse>;
@@ -0,0 +1 @@
1
+ {"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/client.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,WAAW,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAKhH,qBAAa,SAAS;IACpB,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,SAAS,CAAqB;IACtC,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAS;gBAEhC,MAAM,CAAC,EAAE,OAAO,CAAC,YAAY,CAAC;IAS1C,SAAS,IAAI,MAAM,GAAG,IAAI;IAI1B,YAAY,IAAI,MAAM;IAItB,SAAS,IAAI,MAAM,GAAG,IAAI;IAI1B,aAAa,IAAI,MAAM;IAIvB,SAAS,IAAI,YAAY;IASzB,SAAS,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAI5B,YAAY,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAI9B,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAI3B;;;OAGG;IACH,qBAAqB,IAAI,MAAM;IAI/B;;;;OAIG;IACH,OAAO,CAAC,0BAA0B;IAI5B,YAAY,CAAC,SAAS,GAAE,MAAqB,EAAE,SAAS,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAkC1G,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC;IAuCtD,gBAAgB,CAAC,cAAc,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC;IAuCnF,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,cAAc,CAAC;IA2ClE,IAAI,CAAC,KAAK,GAAE,MAAW,EAAE,MAAM,GAAE,MAAU,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,IAAI,EAAE,YAAY,EAAE,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAsChH,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;IAqBvD,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,GAAG,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAqBzG,QAAQ,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAoB/C"}
@@ -34,8 +34,8 @@ var __importStar = (this && this.__importStar) || (function () {
34
34
  })();
35
35
  Object.defineProperty(exports, "__esModule", { value: true });
36
36
  exports.APIClient = void 0;
37
- const crypto = __importStar(require("crypto"));
38
37
  const os = __importStar(require("os"));
38
+ const storage_1 = require("./storage");
39
39
  const DEFAULT_API_BASE = "https://t0ken.ai/api";
40
40
  class APIClient {
41
41
  constructor(config) {
@@ -46,6 +46,8 @@ class APIClient {
46
46
  this.apiBaseUrl = config?.apiBaseUrl || DEFAULT_API_BASE;
47
47
  this.projectId = config?.projectId || "default";
48
48
  this.userId = config?.userId || null;
49
+ // 构造时立即生成机器指纹,外部无法修改
50
+ this.machineFingerprint = this.generateMachineFingerprint();
49
51
  }
50
52
  getApiKey() {
51
53
  return this.apiKey;
@@ -76,23 +78,29 @@ class APIClient {
76
78
  setUserId(id) {
77
79
  this.userId = id;
78
80
  }
81
+ /**
82
+ * 获取机器指纹(只读,返回构造时生成的值)
83
+ * 外部无法修改机器指纹
84
+ */
79
85
  getMachineFingerprint() {
80
- const components = [
81
- os.hostname(),
82
- os.platform(),
83
- os.arch(),
84
- os.cpus()[0]?.model || "unknown",
85
- os.totalmem()
86
- ];
87
- const raw = components.join("|");
88
- return crypto.createHash("sha256").update(raw).digest("hex").slice(0, 32);
86
+ return this.machineFingerprint;
89
87
  }
90
- async autoRegister(agentType = "nodejs_sdk", agentName) {
88
+ /**
89
+ * 私有方法:生成机器指纹
90
+ * 调用 storage.ts 中统一的指纹生成函数
91
+ * 包含: MAC地址、平台、架构、主机名、CPU型号、内存、用户名
92
+ */
93
+ generateMachineFingerprint() {
94
+ return (0, storage_1.generateMachineFingerprint)();
95
+ }
96
+ async autoRegister(agentType = "nodejs_sdk", agentName, agentId) {
91
97
  const fingerprint = this.getMachineFingerprint();
98
+ const finalAgentId = agentId || `${agentType}-${fingerprint.slice(0, 8)}`;
92
99
  const response = await fetch(`${this.apiBaseUrl}/agents/auto-register`, {
93
100
  method: "POST",
94
101
  headers: { "Content-Type": "application/json" },
95
102
  body: JSON.stringify({
103
+ agent_id: finalAgentId,
96
104
  machine_fingerprint: fingerprint,
97
105
  agent_type: agentType,
98
106
  agent_name: agentName || os.hostname(),
@@ -0,0 +1 @@
1
+ {"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/client.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACA,uCAAyB;AAEzB,uCAAyF;AAEzF,MAAM,gBAAgB,GAAG,sBAAsB,CAAC;AAEhD,MAAa,SAAS;IAOpB,YAAY,MAA8B;QANlC,WAAM,GAAkB,IAAI,CAAC;QAE7B,cAAS,GAAW,SAAS,CAAC;QAC9B,WAAM,GAAkB,IAAI,CAAC;QAInC,IAAI,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,IAAI,IAAI,CAAC;QACrC,IAAI,CAAC,UAAU,GAAG,MAAM,EAAE,UAAU,IAAI,gBAAgB,CAAC;QACzD,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,SAAS,CAAC;QAChD,IAAI,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,IAAI,IAAI,CAAC;QACrC,qBAAqB;QACrB,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,0BAA0B,EAAE,CAAC;IAC9D,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,SAAS;QACP,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,GAAW;QACnB,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC;IACpB,CAAC;IAED,YAAY,CAAC,EAAU;QACrB,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;IACtB,CAAC;IAED,SAAS,CAAC,EAAU;QAClB,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;IACnB,CAAC;IAED;;;OAGG;IACH,qBAAqB;QACnB,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAED;;;;OAIG;IACK,0BAA0B;QAChC,OAAO,IAAA,oCAA8B,GAAE,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,YAAoB,YAAY,EAAE,SAAkB,EAAE,OAAgB;QACvF,MAAM,WAAW,GAAG,IAAI,CAAC,qBAAqB,EAAE,CAAC;QACjD,MAAM,YAAY,GAAG,OAAO,IAAI,GAAG,SAAS,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;QAE1E,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,UAAU,uBAAuB,EAAE;YACtE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,QAAQ,EAAE,YAAY;gBACtB,mBAAmB,EAAE,WAAW;gBAChC,UAAU,EAAE,SAAS;gBACrB,UAAU,EAAE,SAAS,IAAI,EAAE,CAAC,QAAQ,EAAE;gBACtC,QAAQ,EAAE,EAAE,CAAC,QAAQ,EAAE;gBACvB,gBAAgB,EAAE,EAAE,CAAC,OAAO,EAAE;aAC/B,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAC3F,CAAC;QAED,MAAM,IAAI,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACxC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC;QAE5B,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC;SACpC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,QAAkB;QACnC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,KAAK,CAAC;YACpC,CAAC,CAAC,GAAG,IAAI,CAAC,UAAU,cAAc;YAClC,CAAC,CAAC,GAAG,IAAI,CAAC,UAAU,oBAAoB,CAAC;QAE3C,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,KAAK,CAAC;YAChC,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,IAAI,EAAE,EAAE;YACpG,CAAC,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC;QAE7C,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE;YACrC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,IAAI,CAAC,MAAM;aACzB;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAC3F,CAAC;QAED,MAAM,IAAI,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACxC,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,cAAsB,EAAE,QAAmB;QAChE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,UAAU,yBAAyB,EAAE;YACxE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,IAAI,CAAC,MAAM;aACzB;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,eAAe,EAAE,cAAc;gBAC/B,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oBAC3B,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,OAAO,EAAE,CAAC,CAAC,OAAO;oBAClB,SAAS,EAAE,CAAC,CAAC,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE;oBACpC,MAAM,EAAE,CAAC,CAAC,MAAM,IAAI,CAAC;iBACtB,CAAC,CAAC;aACJ,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,6BAA6B,QAAQ,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAC/F,CAAC;QAED,MAAM,IAAI,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACxC,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,QAAgB,EAAE;QAC5C,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,UAAU,qBAAqB,EAAE;YACpE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,IAAI,CAAC,MAAM;aACzB;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK;gBACL,UAAU,EAAE,IAAI,CAAC,SAAS;gBAC1B,KAAK;aACN,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,kBAAkB,QAAQ,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACpF,CAAC;QAED,MAAM,IAAI,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAExC,OAAO;YACL,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;gBACvC,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,OAAO,EAAE,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,OAAO;gBAC9B,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,OAAO;gBAC/B,KAAK,EAAE,CAAC,CAAC,KAAK,IAAI,GAAG;aACtB,CAAC,CAAC;YACH,gBAAgB,EAAE,CAAC,IAAI,CAAC,gBAAgB,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;gBAC/D,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,OAAO,EAAE,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,OAAO;gBAC9B,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,OAAO;gBAC/B,KAAK,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC;aACpB,CAAC,CAAC;YACH,eAAe,EAAE,IAAI,CAAC,eAAe;SACtC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE,EAAE,SAAiB,CAAC;QAC/C,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC;YACjC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;YACpB,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC;YACtB,UAAU,EAAE,IAAI,CAAC,SAAS;SAC3B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,UAAU,qBAAqB,MAAM,EAAE,EAAE;YAC5E,MAAM,EAAE,KAAK;YACb,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,IAAI,CAAC,MAAM;aACzB;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,gBAAgB,QAAQ,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAClF,CAAC;QAED,MAAM,IAAI,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAExC,OAAO;YACL,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;gBACvC,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,OAAO,EAAE,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,OAAO;gBAC9B,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,OAAO;gBAC/B,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YACH,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,CAAC;SACvB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAgB;QAC3B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,UAAU,gBAAgB,QAAQ,EAAE,EAAE;YACzE,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,IAAI,CAAC,MAAM;aACzB;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,kBAAkB,QAAQ,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACpF,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAc;QAChC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,UAAU,qBAAqB,MAAM,EAAE,EAAE;YAC5E,MAAM,EAAE,KAAK;YACb,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,IAAI,CAAC,MAAM;aACzB;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,2BAA2B,QAAQ,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAC7F,CAAC;QAED,OAAO,QAAQ,CAAC,IAAI,EAAgF,CAAC;IACvG,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,UAAU,WAAW,EAAE;YAC1D,MAAM,EAAE,KAAK;YACb,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,IAAI,CAAC,MAAM;aACzB;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACvF,CAAC;QAED,OAAO,QAAQ,CAAC,IAAI,EAAkC,CAAC;IACzD,CAAC;CACF;AArUD,8BAqUC"}
@@ -1,16 +1,20 @@
1
- import { MemoryXSDKOptions, FlushStrategy, QueueStats, Memory, Message, SearchResponse, AccountInfo, AccountInfoResponse, PresetMode } from "./types";
1
+ import { MemoryXSDKOptions, FlushStrategy, QueueStats, Memory, Message, SearchResponse, AccountInfo, AccountInfoResponse, PresetMode, AgentType, AgentInfo } from "./types";
2
2
  import { countTokens, setDebug as setQueueDebug } from "./queue";
3
3
  export { countTokens, setQueueDebug as setDebug };
4
- export type { FlushStrategy, QueueStats, Memory, Message, SearchResponse, AccountInfo, MemoryXSDKOptions, PresetMode };
4
+ export type { FlushStrategy, QueueStats, Memory, Message, SearchResponse, AccountInfo, MemoryXSDKOptions, PresetMode, AgentType, AgentInfo };
5
5
  export declare class MemoryXSDK {
6
6
  private client;
7
7
  private queue;
8
8
  private initialized;
9
9
  private options;
10
+ private agentInfo;
10
11
  constructor(options?: MemoryXSDKOptions);
11
12
  init(): Promise<void>;
12
13
  private saveConfig;
13
14
  autoRegister(agentType?: string, agentName?: string): Promise<AccountInfo>;
15
+ getAgentId(): string | null;
16
+ getAgentType(): AgentType | null;
17
+ getAgentInfo(): AgentInfo | null;
14
18
  addMemory(content: string, metadata?: Record<string, any>): Promise<number>;
15
19
  search(query: string, limit?: number): Promise<SearchResponse>;
16
20
  list(limit?: number, offset?: number): Promise<{
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,mBAAmB,EAAW,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAErL,OAAO,EAAgB,WAAW,EAAE,QAAQ,IAAI,aAAa,EAAE,MAAM,SAAS,CAAC;AAG/E,OAAO,EAAE,WAAW,EAAE,aAAa,IAAI,QAAQ,EAAE,CAAC;AAClD,YAAY,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,iBAAiB,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;AAE7I,qBAAa,UAAU;IACrB,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,KAAK,CAAe;IAC5B,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,OAAO,CAAoB;IACnC,OAAO,CAAC,SAAS,CAA0B;gBAE/B,OAAO,GAAE,iBAAsB;IAoBrC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;YA4Cb,UAAU;IAIlB,YAAY,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAiChF,UAAU,IAAI,MAAM,GAAG,IAAI;IAI3B,YAAY,IAAI,SAAS,GAAG,IAAI;IAIhC,YAAY,IAAI,SAAS,GAAG,IAAI;IAK1B,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAK/E,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,cAAc,CAAC;IAKlE,IAAI,CAAC,KAAK,GAAE,MAAW,EAAE,MAAM,GAAE,MAAU,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,IAAI,EAAE,GAAG,EAAE,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAKvG,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;IAKvD,QAAQ,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAMxC,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,WAAW,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAK9E,oBAAoB,IAAI,IAAI;IAKtB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAKtB,aAAa,IAAI,OAAO,CAAC,UAAU,CAAC;IAIpC,oBAAoB,IAAI,OAAO,CAAC,MAAM,CAAC;IAIvC,0BAA0B,IAAI,OAAO,CAAC,MAAM,CAAC;IAInD,WAAW,CAAC,QAAQ,EAAE,aAAa,GAAG,IAAI;IAK1C,OAAO,IAAI,IAAI;IAKf,SAAS,IAAI,MAAM,GAAG,IAAI;IAI1B,YAAY,IAAI,MAAM;IAItB,SAAS,IAAI,MAAM,GAAG,IAAI;IAI1B;;;OAGG;IACG,cAAc,IAAI,OAAO,CAAC,mBAAmB,CAAC;CAqBrD;AAED,wBAAgB,aAAa,CAAC,OAAO,GAAE,iBAAsB,GAAG,UAAU,CAEzE;AAED,eAAO,MAAM,MAAM,mCAAU,CAAC"}
@@ -11,6 +11,7 @@ const storage_1 = require("./storage");
11
11
  class MemoryXSDK {
12
12
  constructor(options = {}) {
13
13
  this.initialized = false;
14
+ this.agentInfo = null;
14
15
  this.options = options;
15
16
  if (options.storageDir) {
16
17
  (0, storage_1.setStorageDir)(options.storageDir);
@@ -28,18 +29,35 @@ class MemoryXSDK {
28
29
  async init() {
29
30
  if (this.initialized)
30
31
  return;
32
+ // 1. 尝试加载已有的 Agent 信息
33
+ this.agentInfo = await storage_1.SQLiteStorage.loadAgentInfo();
34
+ if (this.agentInfo) {
35
+ // 已有 Agent 信息,使用存储的 api_key
36
+ this.client.setApiKey(this.agentInfo.apiKey);
37
+ this.queue.setAgentId(this.agentInfo.agentId);
38
+ }
39
+ else {
40
+ // 没有 Agent 信息,需要初始化
41
+ const agentType = (this.options.agentType || 'nodejs_sdk');
42
+ const machineFp = (0, storage_1.generateMachineFingerprint)();
43
+ const agentId = (0, storage_1.generateAgentId)(agentType, machineFp);
44
+ this.agentInfo = {
45
+ agentId,
46
+ agentType,
47
+ apiKey: '',
48
+ machineFp,
49
+ createdAt: Date.now()
50
+ };
51
+ // 保存初始 agent 信息(不含 api_key)
52
+ await storage_1.SQLiteStorage.saveAgentInfo(this.agentInfo);
53
+ this.queue.setAgentId(agentId);
54
+ }
55
+ // 2. 兼容旧 config
31
56
  const stored = await storage_1.SQLiteStorage.loadConfig();
32
- if (stored) {
33
- if (!this.options.apiKey && stored.apiKey) {
34
- this.client.setApiKey(stored.apiKey);
35
- }
36
- if (!this.options.projectId && stored.projectId) {
37
- this.client.setProjectId(stored.projectId);
38
- }
39
- if (stored.userId) {
40
- this.client.setUserId(stored.userId);
41
- }
57
+ if (stored && stored.apiKey && !this.client.getApiKey()) {
58
+ this.client.setApiKey(stored.apiKey);
42
59
  }
60
+ // 3. 如果没有 api_key,尝试自动注册
43
61
  if (!this.client.getApiKey() && this.options.autoRegister !== false) {
44
62
  await this.autoRegister();
45
63
  }
@@ -50,10 +68,41 @@ class MemoryXSDK {
50
68
  await storage_1.SQLiteStorage.saveConfig(this.client.getConfig());
51
69
  }
52
70
  async autoRegister(agentType, agentName) {
53
- const info = await this.client.autoRegister(agentType || this.options.agentType || "nodejs_sdk", agentName);
71
+ const type = (agentType || this.options.agentType || "nodejs_sdk");
72
+ // 如果 agentInfo 不存在或类型变化,重新生成
73
+ if (!this.agentInfo || this.agentInfo.agentType !== type) {
74
+ const machineFp = (0, storage_1.generateMachineFingerprint)();
75
+ const agentId = (0, storage_1.generateAgentId)(type, machineFp);
76
+ this.agentInfo = {
77
+ agentId,
78
+ agentType: type,
79
+ apiKey: '',
80
+ machineFp,
81
+ createdAt: Date.now()
82
+ };
83
+ }
84
+ // 调用服务端注册 API,传递 agentId
85
+ const info = await this.client.autoRegister(type, agentName, this.agentInfo.agentId);
86
+ // 更新 agentInfo 的 apiKey
87
+ if (this.agentInfo && info.api_key) {
88
+ this.agentInfo.apiKey = info.api_key;
89
+ await storage_1.SQLiteStorage.saveAgentInfo(this.agentInfo);
90
+ this.client.setApiKey(info.api_key);
91
+ }
92
+ // 兼容旧 config
54
93
  await this.saveConfig();
55
94
  return info;
56
95
  }
96
+ // Agent 信息获取
97
+ getAgentId() {
98
+ return this.agentInfo?.agentId || null;
99
+ }
100
+ getAgentType() {
101
+ return this.agentInfo?.agentType || null;
102
+ }
103
+ getAgentInfo() {
104
+ return this.agentInfo;
105
+ }
57
106
  // Memory operations
58
107
  async addMemory(content, metadata = {}) {
59
108
  await this.init();
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;AAuOA,sCAEC;AAzOD,mCAAqL;AACrL,qCAAqC;AACrC,mCAA+E;AAGtE,4FAHc,mBAAW,OAGd;AAAmB,yFAHS,gBAAa,OAGd;AAF/C,uCAAsG;AAKtG,MAAa,UAAU;IAOrB,YAAY,UAA6B,EAAE;QAJnC,gBAAW,GAAY,KAAK,CAAC;QAE7B,cAAS,GAAqB,IAAI,CAAC;QAGzC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,IAAA,uBAAa,EAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACpC,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,kBAAS,CAAC;YAC1B,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,IAAI;YAC9B,UAAU,EAAE,OAAO,CAAC,MAAM;YAC1B,SAAS,EAAE,OAAO,CAAC,SAAS;SAC7B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM;YAC7B,CAAC,CAAC,eAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YACzB,CAAC,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;QAE3B,IAAI,CAAC,KAAK,GAAG,IAAI,oBAAY,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO;QAE7B,sBAAsB;QACtB,IAAI,CAAC,SAAS,GAAG,MAAM,uBAAa,CAAC,aAAa,EAAE,CAAC;QAErD,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,4BAA4B;YAC5B,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAC7C,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAChD,CAAC;aAAM,CAAC;YACN,oBAAoB;YACpB,MAAM,SAAS,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,IAAI,YAAY,CAAc,CAAC;YACxE,MAAM,SAAS,GAAG,IAAA,oCAA0B,GAAE,CAAC;YAC/C,MAAM,OAAO,GAAG,IAAA,yBAAe,EAAC,SAAS,EAAE,SAAS,CAAC,CAAC;YAEtD,IAAI,CAAC,SAAS,GAAG;gBACf,OAAO;gBACP,SAAS;gBACT,MAAM,EAAE,EAAE;gBACV,SAAS;gBACT,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;YAEF,4BAA4B;YAC5B,MAAM,uBAAa,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClD,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACjC,CAAC;QAED,gBAAgB;QAChB,MAAM,MAAM,GAAG,MAAM,uBAAa,CAAC,UAAU,EAAE,CAAC;QAChD,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC;YACxD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,yBAAyB;QACzB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK,KAAK,EAAE,CAAC;YACpE,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC5B,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;QACzB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,MAAM,uBAAa,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,SAAkB,EAAE,SAAkB;QACvD,MAAM,IAAI,GAAG,CAAC,SAAS,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,IAAI,YAAY,CAAc,CAAC;QAEhF,6BAA6B;QAC7B,IAAI,CAAC,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC,SAAS,KAAK,IAAI,EAAE,CAAC;YACzD,MAAM,SAAS,GAAG,IAAA,oCAA0B,GAAE,CAAC;YAC/C,MAAM,OAAO,GAAG,IAAA,yBAAe,EAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YAEjD,IAAI,CAAC,SAAS,GAAG;gBACf,OAAO;gBACP,SAAS,EAAE,IAAI;gBACf,MAAM,EAAE,EAAE;gBACV,SAAS;gBACT,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;QACJ,CAAC;QAED,yBAAyB;QACzB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAErF,wBAAwB;QACxB,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACnC,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;YACrC,MAAM,uBAAa,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtC,CAAC;QAED,aAAa;QACb,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,aAAa;IACb,UAAU;QACR,OAAO,IAAI,CAAC,SAAS,EAAE,OAAO,IAAI,IAAI,CAAC;IACzC,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,EAAE,SAAS,IAAI,IAAI,CAAC;IAC3C,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,oBAAoB;IACpB,KAAK,CAAC,SAAS,CAAC,OAAe,EAAE,WAAgC,EAAE;QACjE,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,QAAgB,EAAE;QAC5C,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE,EAAE,SAAiB,CAAC;QAC/C,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAgB;QAC3B,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;IAChC,CAAC;IAED,0BAA0B;IAC1B,KAAK,CAAC,UAAU,CAAC,IAA0B,EAAE,OAAe;QAC1D,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC9C,CAAC;IAED,oBAAoB;QAClB,IAAI,CAAC,KAAK,CAAC,oBAAoB,EAAE,CAAC;IACpC,CAAC;IAED,mBAAmB;IACnB,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,OAAO,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,oBAAoB;QACxB,OAAO,IAAI,CAAC,KAAK,CAAC,oBAAoB,EAAE,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,0BAA0B;QAC9B,OAAO,IAAI,CAAC,KAAK,CAAC,0BAA0B,EAAE,CAAC;IACjD,CAAC;IAED,WAAW,CAAC,QAAuB;QACjC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IAED,UAAU;IACV,OAAO;QACL,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;IAC1B,CAAC;IAED,UAAU;IACV,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;IACjC,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;IACpC,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;IACjC,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,cAAc;QAClB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAElB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QACvC,IAAI,KAAsC,CAAC;QAE3C,IAAI,CAAC;YACH,KAAK,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACvC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,+CAA+C;QACjD,CAAC;QAED,OAAO;YACL,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,KAAK;SACN,CAAC;IACJ,CAAC;CACF;AA7ND,gCA6NC;AAED,SAAgB,aAAa,CAAC,UAA6B,EAAE;IAC3D,OAAO,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC;AACjC,CAAC;AAEY,QAAA,MAAM,GAAG,eAAO,CAAC"}
@@ -9,11 +9,14 @@ export declare class QueueManager {
9
9
  private isFlushing;
10
10
  private maxRetry;
11
11
  private baseRetryDelayMs;
12
+ private agentId;
12
13
  private getRetryDelay;
13
14
  private currentConversationId;
14
15
  private conversationCreatedAt;
15
16
  private lastActivityAt;
16
17
  constructor(client: APIClient, strategy?: FlushStrategy);
18
+ setAgentId(agentId: string): void;
19
+ getAgentId(): string;
17
20
  private generateConversationId;
18
21
  getStrategy(): FlushStrategy;
19
22
  setStrategy(strategy: FlushStrategy): void;
@@ -0,0 +1 @@
1
+ {"version":3,"file":"queue.d.ts","sourceRoot":"","sources":["../../src/queue.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,UAAU,EAA4B,MAAM,SAAS,CAAC;AAE9E,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAMrC,wBAAgB,QAAQ,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CAE/C;AAsBD,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAMhD;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAgB;IAChC,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,UAAU,CAA+B;IACjD,OAAO,CAAC,UAAU,CAAkB;IACpC,OAAO,CAAC,QAAQ,CAAa;IAC7B,OAAO,CAAC,gBAAgB,CAAgB;IACxC,OAAO,CAAC,OAAO,CAAc;IAG7B,OAAO,CAAC,aAAa;IAQrB,OAAO,CAAC,qBAAqB,CAAc;IAC3C,OAAO,CAAC,qBAAqB,CAAa;IAC1C,OAAO,CAAC,cAAc,CAAa;gBAEvB,MAAM,EAAE,SAAS,EAAE,QAAQ,GAAE,aAAkB;IAU3D,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAKjC,UAAU,IAAI,MAAM;IAIpB,OAAO,CAAC,sBAAsB;IAI9B,WAAW,IAAI,aAAa;IAI5B,WAAW,CAAC,QAAQ,EAAE,aAAa,GAAG,IAAI;IAK1C,WAAW,IAAI,IAAI;IAkCnB,UAAU,IAAI,IAAI;IAQZ,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAoB/E,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,WAAW,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAyC9E,oBAAoB,IAAI,IAAI;IAYtB,aAAa,IAAI,OAAO,CAAC,UAAU,CAAC;IAqB1C,WAAW,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO;IA2BjC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;YAwBd,gBAAgB;YAsChB,sBAAsB;IAwD9B,oBAAoB,IAAI,OAAO,CAAC,MAAM,CAAC;IAIvC,0BAA0B,IAAI,OAAO,CAAC,MAAM,CAAC;CAGpD"}
@@ -50,6 +50,7 @@ class QueueManager {
50
50
  this.isFlushing = false;
51
51
  this.maxRetry = 5;
52
52
  this.baseRetryDelayMs = 1000; // 1秒基础延迟
53
+ this.agentId = "";
53
54
  this.currentConversationId = "";
54
55
  this.conversationCreatedAt = 0;
55
56
  this.lastActivityAt = 0;
@@ -60,6 +61,13 @@ class QueueManager {
60
61
  this.lastActivityAt = Date.now();
61
62
  log('INFO', 'QueueManager initialized', { strategy });
62
63
  }
64
+ setAgentId(agentId) {
65
+ this.agentId = agentId;
66
+ log('INFO', 'AgentId set', { agentId });
67
+ }
68
+ getAgentId() {
69
+ return this.agentId;
70
+ }
63
71
  generateConversationId() {
64
72
  return `conv_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
65
73
  }
@@ -107,8 +115,12 @@ class QueueManager {
107
115
  }
108
116
  }
109
117
  async addMemory(content, metadata = {}) {
110
- log('DEBUG', 'Adding memory', { content: content.substring(0, 50) + '...', metadata });
111
- const id = await storage_1.SQLiteStorage.addToMemoryQueue(content, metadata);
118
+ log('DEBUG', 'Adding memory', { content: content.substring(0, 50) + '...', metadata, agentId: this.agentId });
119
+ if (!this.agentId) {
120
+ log('ERROR', 'Cannot add memory: agentId not set');
121
+ throw new Error('AgentId not set. Call setAgentId() first.');
122
+ }
123
+ const id = await storage_1.SQLiteStorage.addToMemoryQueue(this.agentId, content, metadata);
112
124
  log('INFO', `Memory added to queue`, { id, content: content.substring(0, 30) + '...' });
113
125
  const queueLength = await storage_1.SQLiteStorage.getMemoryQueueLength();
114
126
  if (this.strategy.batchSize && queueLength >= this.strategy.batchSize) {
@@ -118,8 +130,12 @@ class QueueManager {
118
130
  return id;
119
131
  }
120
132
  async addMessage(role, content) {
121
- log('DEBUG', 'Adding message', { role, conversationId: this.currentConversationId, content: content.substring(0, 50) + '...' });
122
- const id = await storage_1.SQLiteStorage.addToConversationQueue(this.currentConversationId, this.conversationCreatedAt, role, content);
133
+ log('DEBUG', 'Adding message', { role, conversationId: this.currentConversationId, content: content.substring(0, 50) + '...', agentId: this.agentId });
134
+ if (!this.agentId) {
135
+ log('ERROR', 'Cannot add message: agentId not set');
136
+ throw new Error('AgentId not set. Call setAgentId() first.');
137
+ }
138
+ const id = await storage_1.SQLiteStorage.addToConversationQueue(this.agentId, this.currentConversationId, this.conversationCreatedAt, role, content);
123
139
  this.lastActivityAt = Date.now();
124
140
  const stats = await this.getQueueStats();
125
141
  log('INFO', `Message added`, {
@@ -153,14 +169,10 @@ class QueueManager {
153
169
  }
154
170
  async getQueueStats() {
155
171
  const stats = await storage_1.SQLiteStorage.getConversationStats(this.currentConversationId);
156
- const db = await (0, storage_1.getDb)();
157
- const rows = await db.all(`
158
- SELECT content FROM conversation_queue
159
- WHERE conversation_id = ?
160
- `, this.currentConversationId);
172
+ const messages = await storage_1.SQLiteStorage.getConversationMessages(this.currentConversationId);
161
173
  let totalTokens = 0;
162
- for (const row of rows) {
163
- totalTokens += countTokens(row.content);
174
+ for (const msg of messages) {
175
+ totalTokens += countTokens(msg.content);
164
176
  }
165
177
  const idleTime = this.lastActivityAt > 0 ? Date.now() - this.lastActivityAt : 0;
166
178
  return {
@@ -0,0 +1 @@
1
+ {"version":3,"file":"queue.js","sourceRoot":"","sources":["../../src/queue.ts"],"names":[],"mappings":";;;AAQA,4BAEC;AAsBD,kCAMC;AArCD,uCAA0C;AAE1C,6CAA0C;AAE1C,IAAI,SAAS,GAA0C,IAAI,CAAC;AAC5D,IAAI,KAAK,GAAG,KAAK,CAAC;AAElB,SAAgB,QAAQ,CAAC,OAAgB;IACvC,KAAK,GAAG,OAAO,CAAC;AAClB,CAAC;AAED,SAAS,GAAG,CAAC,KAA0C,EAAE,OAAe,EAAE,IAAU;IAClF,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC3C,MAAM,MAAM,GAAG,IAAI,SAAS,cAAc,KAAK,GAAG,CAAC;IAEnD,IAAI,KAAK,KAAK,OAAO,IAAI,CAAC,KAAK;QAAE,OAAO;IAExC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAChG,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC/B,CAAC;AACH,CAAC;AAED,SAAS,YAAY;IACnB,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,SAAS,GAAG,IAAA,yBAAW,EAAC,aAAa,CAAC,CAAC;IACzC,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAgB,WAAW,CAAC,IAAY;IACtC,IAAI,CAAC;QACH,OAAO,YAAY,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;IAC5C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACpC,CAAC;AACH,CAAC;AAED,MAAa,YAAY;IASvB,WAAW;IACH,aAAa,CAAC,UAAkB;QACtC,mCAAmC;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,EAAE,KAAK,CAAC,CAAC;QAC/E,gBAAgB;QAChB,MAAM,MAAM,GAAG,KAAK,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC;QACnD,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,MAAM,CAAC,CAAC;IACpC,CAAC;IAMD,YAAY,MAAiB,EAAE,WAA0B,EAAE;QAnBnD,eAAU,GAA0B,IAAI,CAAC;QACzC,eAAU,GAAY,KAAK,CAAC;QAC5B,aAAQ,GAAW,CAAC,CAAC;QACrB,qBAAgB,GAAW,IAAI,CAAC,CAAC,SAAS;QAC1C,YAAO,GAAW,EAAE,CAAC;QAWrB,0BAAqB,GAAW,EAAE,CAAC;QACnC,0BAAqB,GAAW,CAAC,CAAC;QAClC,mBAAc,GAAW,CAAC,CAAC;QAGjC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC3D,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;QAC3D,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEjC,GAAG,CAAC,MAAM,EAAE,0BAA0B,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;IACxD,CAAC;IAED,UAAU,CAAC,OAAe;QACxB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,GAAG,CAAC,MAAM,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAEO,sBAAsB;QAC5B,OAAO,QAAQ,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IACzE,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,WAAW,CAAC,QAAuB;QACjC,GAAG,CAAC,MAAM,EAAE,kBAAkB,EAAE,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,WAAW;QACT,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QAC9D,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,GAAG,EAAE,EAAE,KAAK,CAAC,CAAC;QAExD,GAAG,CAAC,MAAM,EAAE,2BAA2B,EAAE;YACvC,WAAW;YACX,aAAa;YACb,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS;SACnC,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,GAAG,WAAW,CAAC,KAAK,IAAI,EAAE;YACvC,IAAI,IAAI,CAAC,cAAc,KAAK,CAAC;gBAAE,OAAO;YAEtC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC;YAElD,IAAI,QAAQ,IAAI,WAAW,EAAE,CAAC;gBAC5B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;gBACzC,IAAI,KAAK,CAAC,YAAY,GAAG,CAAC,EAAE,CAAC;oBAC3B,GAAG,CAAC,MAAM,EAAE,wCAAwC,EAAE;wBACpD,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,GAAG;wBAC3C,YAAY,EAAE,KAAK,CAAC,YAAY;wBAChC,WAAW,EAAE,KAAK,CAAC,WAAW;qBAC/B,CAAC,CAAC;oBACH,IAAI,CAAC,oBAAoB,EAAE,CAAC;oBAC5B,IAAI,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;QACH,CAAC,EAAE,aAAa,CAAC,CAAC;IACpB,CAAC;IAED,UAAU;QACR,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,GAAG,CAAC,MAAM,EAAE,sBAAsB,CAAC,CAAC;YACpC,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC/B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACzB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,OAAe,EAAE,WAAgC,EAAE;QACjE,GAAG,CAAC,OAAO,EAAE,eAAe,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAE9G,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,GAAG,CAAC,OAAO,EAAE,oCAAoC,CAAC,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC/D,CAAC;QAED,MAAM,EAAE,GAAG,MAAM,uBAAa,CAAC,gBAAgB,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;QACjF,GAAG,CAAC,MAAM,EAAE,uBAAuB,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,CAAC;QAExF,MAAM,WAAW,GAAG,MAAM,uBAAa,CAAC,oBAAoB,EAAE,CAAC;QAC/D,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,IAAI,WAAW,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;YACtE,GAAG,CAAC,MAAM,EAAE,uBAAuB,WAAW,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,qBAAqB,CAAC,CAAC;YAChG,YAAY,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,IAA0B,EAAE,OAAe;QAC1D,GAAG,CAAC,OAAO,EAAE,gBAAgB,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,CAAC,qBAAqB,EAAE,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAEvJ,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,GAAG,CAAC,OAAO,EAAE,qCAAqC,CAAC,CAAC;YACpD,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC/D,CAAC;QAED,MAAM,EAAE,GAAG,MAAM,uBAAa,CAAC,sBAAsB,CACnD,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,qBAAqB,EAC1B,IAAI,CAAC,qBAAqB,EAC1B,IAAI,EACJ,OAAO,CACR,CAAC;QAEF,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEjC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QACzC,GAAG,CAAC,MAAM,EAAE,eAAe,EAAE;YAC3B,EAAE;YACF,IAAI;YACJ,cAAc,EAAE,IAAI,CAAC,qBAAqB;YAC1C,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,WAAW,EAAE,KAAK,CAAC,WAAW;SAC/B,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC;YAC5B,GAAG,CAAC,MAAM,EAAE,6BAA6B,EAAE;gBACzC,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC,CAAC;YACH,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5B,YAAY,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,oBAAoB;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,qBAAqB,CAAC;QACzC,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC3D,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;QAC3D,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEjC,GAAG,CAAC,MAAM,EAAE,0BAA0B,EAAE;YACtC,iBAAiB,EAAE,KAAK;YACxB,iBAAiB,EAAE,IAAI,CAAC,qBAAqB;SAC9C,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,MAAM,KAAK,GAAG,MAAM,uBAAa,CAAC,oBAAoB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAEnF,MAAM,QAAQ,GAAG,MAAM,uBAAa,CAAC,uBAAuB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAEzF,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,WAAW,IAAI,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;QAEhF,OAAO;YACL,YAAY,EAAE,KAAK,CAAC,KAAK;YACzB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,WAAW;YACX,gBAAgB,EAAE,QAAQ;YAC1B,cAAc,EAAE,IAAI,CAAC,qBAAqB;SAC3C,CAAC;IACJ,CAAC;IAED,WAAW,CAAC,KAAiB;QAC3B,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;YAClD,IAAI,MAAM,EAAE,CAAC;gBACX,GAAG,CAAC,OAAO,EAAE,8BAA8B,EAAE,KAAK,CAAC,CAAC;YACtD,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YACjE,GAAG,CAAC,OAAO,EAAE,6BAA6B,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAClF,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,IAAI,KAAK,CAAC,YAAY,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;YAC7E,GAAG,CAAC,OAAO,EAAE,uBAAuB,KAAK,CAAC,YAAY,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC;YACrF,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,IAAI,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;YAC5E,GAAG,CAAC,OAAO,EAAE,uBAAuB,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC;YACpF,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,GAAG,CAAC,OAAO,EAAE,qCAAqC,CAAC,CAAC;YACpD,OAAO;QACT,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC;YAC7B,GAAG,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;YAC3C,OAAO;QACT,CAAC;QAED,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;QAE9B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC9B,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACpC,GAAG,CAAC,MAAM,EAAE,8BAA8B,CAAC,CAAC;QAC9C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,GAAG,CAAC,OAAO,EAAE,cAAc,EAAG,CAAW,CAAC,OAAO,CAAC,CAAC;QACrD,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QAC1B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB;QAC5B,mBAAmB;QACnB,MAAM,WAAW,GAAG,MAAM,uBAAa,CAAC,iCAAiC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzF,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;YAC/B,GAAG,CAAC,OAAO,EAAE,eAAe,IAAI,CAAC,EAAE,wBAAwB,IAAI,CAAC,WAAW,gCAAgC,CAAC,CAAC;YAC7G,MAAM,uBAAa,CAAC,2BAA2B,CAAC,IAAI,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QACrE,CAAC;QACD,MAAM,uBAAa,CAAC,sBAAsB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE1D,MAAM,KAAK,GAAG,MAAM,uBAAa,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAC3D,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,GAAG,CAAC,OAAO,EAAE,uBAAuB,CAAC,CAAC;YACtC,OAAO;QACT,CAAC;QAED,GAAG,CAAC,MAAM,EAAE,YAAY,KAAK,CAAC,MAAM,eAAe,CAAC,CAAC;QAErD,MAAM,QAAQ,GAAa,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC5C,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC;SACpC,CAAC,CAAC,CAAC;QAEJ,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEvC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,uBAAa,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;YAC3C,GAAG,CAAC,MAAM,EAAE,qBAAqB,QAAQ,CAAC,MAAM,WAAW,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,uBAAa,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;YAC9C,GAAG,CAAC,MAAM,EAAE,wDAAwD,GAAG,CAAC,MAAM,QAAQ,EAAE;gBACtF,KAAK,EAAG,CAAW,CAAC,OAAO;gBAC3B,WAAW,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,WAAW,GAAG,CAAC,EAAE,CAAC,CAAC;aACtE,CAAC,CAAC;YACH,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,sBAAsB;QAClC,MAAM,uBAAa,CAAC,qBAAqB,EAAE,CAAC;QAE5C,MAAM,QAAQ,GAAG,MAAM,uBAAa,CAAC,mBAAmB,EAAE,CAAC;QAC3D,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,GAAG,CAAC,OAAO,EAAE,2BAA2B,CAAC,CAAC;YAC1C,OAAO;QACT,CAAC;QAED,MAAM,EAAE,cAAc,EAAE,GAAG,QAAQ,CAAC;QAEpC,MAAM,QAAQ,GAAG,MAAM,uBAAa,CAAC,uBAAuB,CAAC,cAAc,CAAC,CAAC;QAC7E,IAAI,QAAQ,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC9B,GAAG,CAAC,OAAO,EAAE,gBAAgB,cAAc,wBAAwB,QAAQ,gCAAgC,CAAC,CAAC;YAC7G,MAAM,uBAAa,CAAC,qBAAqB,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;YAC1E,MAAM,uBAAa,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;YACvD,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,uBAAa,CAAC,uBAAuB,CAAC,cAAc,CAAC,CAAC;QAC7E,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,GAAG,CAAC,OAAO,EAAE,gBAAgB,cAAc,4BAA4B,CAAC,CAAC;YACzE,MAAM,uBAAa,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;YACvD,OAAO;QACT,CAAC;QAED,GAAG,CAAC,MAAM,EAAE,yBAAyB,cAAc,EAAE,EAAE;YACrD,YAAY,EAAE,QAAQ,CAAC,MAAM;YAC7B,QAAQ;YACR,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,CAAC;SAC7F,CAAC,CAAC;QAEH,MAAM,iBAAiB,GAAc,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACtD,IAAI,EAAE,CAAC,CAAC,IAA4B;YACpC,OAAO,EAAE,CAAC,CAAC,OAAO;YAClB,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC;SAC/B,CAAC,CAAC,CAAC;QAEJ,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,cAAc,EAAE,iBAAiB,CAAC,CAAC;YACtE,MAAM,uBAAa,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;YACvD,GAAG,CAAC,MAAM,EAAE,kCAAkC,cAAc,SAAS,QAAQ,CAAC,MAAM,WAAW,CAAC,CAAC;QACnG,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,uBAAa,CAAC,0BAA0B,CAAC,cAAc,CAAC,CAAC;YAC/D,MAAM,aAAa,GAAG,QAAQ,GAAG,CAAC,CAAC;YACnC,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;YAClD,GAAG,CAAC,MAAM,EAAE,+BAA+B,cAAc,WAAW,aAAa,EAAE,EAAE;gBACnF,KAAK,EAAG,CAAW,CAAC,OAAO;gBAC3B,gBAAgB,EAAE,OAAO;aAC1B,CAAC,CAAC;YACH,2BAA2B;YAC3B,iCAAiC;QACnC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,oBAAoB;QACxB,OAAO,uBAAa,CAAC,oBAAoB,EAAE,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,0BAA0B;QAC9B,OAAO,uBAAa,CAAC,0BAA0B,EAAE,CAAC;IACpD,CAAC;CACF;AAtVD,oCAsVC"}
@@ -1,13 +1,21 @@
1
- import { Database as SqliteDatabase } from "sqlite";
1
+ import { Database as SqlJsDatabase } from "sql.js";
2
+ import { AgentInfo, AgentType } from "./types";
2
3
  export declare function setStorageDir(dir: string): void;
3
4
  export declare function getStorageDir(): string;
4
- export declare function getDb(): Promise<SqliteDatabase>;
5
+ export declare function getDb(): Promise<SqlJsDatabase>;
6
+ export declare function generateMachineFingerprint(): string;
7
+ export declare function generateAgentId(agentType: AgentType, machineFp: string): string;
5
8
  export declare class SQLiteStorage {
9
+ static loadAgentInfo(): Promise<AgentInfo | null>;
10
+ static saveAgentInfo(info: AgentInfo): Promise<void>;
11
+ static getAgentId(): Promise<string | null>;
12
+ static getApiKey(): Promise<string | null>;
6
13
  static loadConfig(): Promise<Record<string, any> | null>;
7
14
  static saveConfig(config: Record<string, any>): Promise<void>;
8
- static addToMemoryQueue(content: string, metadata?: Record<string, any>): Promise<number>;
15
+ static addToMemoryQueue(agentId: string, content: string, metadata?: Record<string, any>): Promise<number>;
9
16
  static getMemoryQueueItems(limit?: number): Promise<Array<{
10
17
  id: number;
18
+ agent_id: string;
11
19
  content: string;
12
20
  metadata: string;
13
21
  timestamp: number;
@@ -24,7 +32,7 @@ export declare class SQLiteStorage {
24
32
  retry_count: number;
25
33
  }>>;
26
34
  static moveMemoryToDeadLetterQueue(id: number, queueType?: string, errorMessage?: string): Promise<void>;
27
- static addToConversationQueue(conversationId: string, conversationCreatedAt: number, role: string, content: string): Promise<number>;
35
+ static addToConversationQueue(agentId: string, conversationId: string, conversationCreatedAt: number, role: string, content: string): Promise<number>;
28
36
  static getConversationStats(conversationId: string): Promise<{
29
37
  count: number;
30
38
  rounds: number;
@@ -0,0 +1 @@
1
+ {"version":3,"file":"storage.d.ts","sourceRoot":"","sources":["../../src/storage.ts"],"names":[],"mappings":"AAKA,OAAkB,EAAE,QAAQ,IAAI,aAAa,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAO/C,wBAAgB,aAAa,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI,CAE/C;AAED,wBAAgB,aAAa,IAAI,MAAM,CAGtC;AAUD,wBAAsB,KAAK,IAAI,OAAO,CAAC,aAAa,CAAC,CA+FpD;AAqCD,wBAAgB,0BAA0B,IAAI,MAAM,CA+BnD;AAED,wBAAgB,eAAe,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAE/E;AAED,qBAAa,aAAa;WAGX,aAAa,IAAI,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;WAuB1C,aAAa,CAAC,IAAI,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;WAgB7C,UAAU,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;WAKpC,SAAS,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;WAOnC,UAAU,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAAC;WAajD,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;WAWtD,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,GAAG,OAAO,CAAC,MAAM,CAAC;WAYvG,mBAAmB,CAAC,KAAK,GAAE,MAAY,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;WAcrK,iBAAiB,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;WAQ/C,oBAAoB,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;WAQlD,sBAAsB,CAAC,QAAQ,GAAE,MAAU,EAAE,aAAa,GAAE,MAAyB,GAAG,OAAO,CAAC,IAAI,CAAC;WAOrG,oBAAoB,IAAI,OAAO,CAAC,MAAM,CAAC;WAMvC,iCAAiC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;WAY3I,2BAA2B,CAAC,EAAE,EAAE,MAAM,EAAE,SAAS,GAAE,MAAiB,EAAE,YAAY,GAAE,MAAW,GAAG,OAAO,CAAC,IAAI,CAAC;WAkB/G,sBAAsB,CAAC,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,qBAAqB,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;WAY9I,oBAAoB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC;WAuBxF,uBAAuB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;WAatJ,mBAAmB,IAAI,OAAO,CAAC;QAAE,cAAc,EAAE,MAAM,CAAC;QAAC,qBAAqB,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;WAgBhG,kBAAkB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;WAMzD,0BAA0B,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;WAMjE,uBAAuB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;WAMhE,qBAAqB,CAAC,aAAa,GAAE,MAAyB,GAAG,OAAO,CAAC,IAAI,CAAC;WAO9E,0BAA0B,IAAI,OAAO,CAAC,MAAM,CAAC;WAM7C,mBAAmB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;WAS5D,qBAAqB,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,GAAE,MAAuB,EAAE,YAAY,GAAE,MAAW,GAAG,OAAO,CAAC,IAAI,CAAC;WAiB3H,uBAAuB,CAAC,KAAK,GAAE,MAAY,GAAG,OAAO,CAAC,KAAK,CAAC;QACvE,EAAE,EAAE,MAAM,CAAC;QACX,UAAU,EAAE,MAAM,CAAC;QACnB,eAAe,EAAE,MAAM,CAAC;QACxB,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,aAAa,EAAE,MAAM,CAAC;QACtB,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC,CAAC;WAeU,uBAAuB,IAAI,OAAO,CAAC,MAAM,CAAC;WAM1C,oBAAoB,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;WAM/C,oBAAoB,CAAC,aAAa,GAAE,MAA0B,GAAG,OAAO,CAAC,IAAI,CAAC;CAM5F"}