opc-agent 4.2.13 → 5.0.0-rc.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.
- package/.opc/memory.db +0 -0
- package/.opc/schedules.json +784 -0
- package/.opc/voice-tmp/tts-1776835848670.mp3 +0 -0
- package/.opc/voice-tmp/tts-1776835873696.mp3 +0 -0
- package/README.md +125 -141
- package/README.zh-CN.md +75 -160
- package/TASK.md +34 -0
- package/data/brain.db/deepbrain.sqlite +0 -0
- package/dist/channels/api.d.ts +14 -0
- package/dist/channels/api.d.ts.map +1 -0
- package/dist/channels/api.js +58 -0
- package/dist/channels/api.js.map +1 -0
- package/dist/channels/voice.d.ts +21 -0
- package/dist/channels/voice.d.ts.map +1 -1
- package/dist/channels/voice.js +31 -1
- package/dist/channels/voice.js.map +1 -1
- package/dist/cli/chat.d.ts.map +1 -1
- package/dist/cli/chat.js +104 -0
- package/dist/cli/chat.js.map +1 -1
- package/dist/cli/setup.d.ts +1 -1
- package/dist/cli/setup.d.ts.map +1 -1
- package/dist/cli/setup.js +263 -33
- package/dist/cli/setup.js.map +1 -1
- package/dist/cli.js +201 -116
- package/dist/cli.js.map +1 -1
- package/dist/core/agent-loop.d.ts +3 -0
- package/dist/core/agent-loop.d.ts.map +1 -0
- package/dist/core/agent-loop.js +51 -0
- package/dist/core/agent-loop.js.map +1 -0
- package/dist/core/context-assembler.d.ts +12 -0
- package/dist/core/context-assembler.d.ts.map +1 -0
- package/dist/core/context-assembler.js +81 -0
- package/dist/core/context-assembler.js.map +1 -0
- package/dist/core/guardrails.d.ts +16 -0
- package/dist/core/guardrails.d.ts.map +1 -0
- package/dist/core/guardrails.js +62 -0
- package/dist/core/guardrails.js.map +1 -0
- package/dist/core/index.d.ts +1 -0
- package/dist/core/index.d.ts.map +1 -0
- package/dist/core/index.js +3 -0
- package/dist/core/index.js.map +1 -0
- package/dist/core/iteration-budget.d.ts +12 -0
- package/dist/core/iteration-budget.d.ts.map +1 -0
- package/dist/core/iteration-budget.js +26 -0
- package/dist/core/iteration-budget.js.map +1 -0
- package/dist/core/runtime.js +1 -1
- package/dist/core/runtime.js.map +1 -1
- package/dist/core/types.d.ts +412 -0
- package/dist/core/types.d.ts.map +1 -1
- package/dist/deepbrain/embedding.d.ts +7 -0
- package/dist/deepbrain/embedding.d.ts.map +1 -0
- package/dist/deepbrain/embedding.js +108 -0
- package/dist/deepbrain/embedding.js.map +1 -0
- package/dist/deepbrain/index.d.ts +8 -0
- package/dist/deepbrain/index.d.ts.map +1 -0
- package/dist/deepbrain/index.js +19 -0
- package/dist/deepbrain/index.js.map +1 -0
- package/dist/deepbrain/migrate.d.ts +2 -0
- package/dist/deepbrain/migrate.d.ts.map +1 -0
- package/dist/deepbrain/migrate.js +139 -0
- package/dist/deepbrain/migrate.js.map +1 -0
- package/dist/deepbrain/provider.d.ts +13 -0
- package/dist/deepbrain/provider.d.ts.map +1 -0
- package/dist/deepbrain/provider.js +85 -0
- package/dist/deepbrain/provider.js.map +1 -0
- package/dist/deepbrain/recall.d.ts +9 -0
- package/dist/deepbrain/recall.d.ts.map +1 -0
- package/dist/deepbrain/recall.js +48 -0
- package/dist/deepbrain/recall.js.map +1 -0
- package/dist/deepbrain/store.d.ts +36 -0
- package/dist/deepbrain/store.d.ts.map +1 -0
- package/dist/deepbrain/store.js +342 -0
- package/dist/deepbrain/store.js.map +1 -0
- package/dist/deepbrain/workspace-files.d.ts +5 -0
- package/dist/deepbrain/workspace-files.d.ts.map +1 -0
- package/dist/deepbrain/workspace-files.js +159 -0
- package/dist/deepbrain/workspace-files.js.map +1 -0
- package/dist/evolution/index.d.ts +1 -0
- package/dist/evolution/index.d.ts.map +1 -0
- package/dist/evolution/index.js +3 -0
- package/dist/evolution/index.js.map +1 -0
- package/dist/evolution/l1-experience.d.ts +11 -0
- package/dist/evolution/l1-experience.d.ts.map +1 -0
- package/dist/evolution/l1-experience.js +185 -0
- package/dist/evolution/l1-experience.js.map +1 -0
- package/dist/evolution/l2-consolidation.d.ts +4 -0
- package/dist/evolution/l2-consolidation.d.ts.map +1 -0
- package/dist/evolution/l2-consolidation.js +106 -0
- package/dist/evolution/l2-consolidation.js.map +1 -0
- package/dist/evolution/l2-memskill.d.ts +12 -0
- package/dist/evolution/l2-memskill.d.ts.map +1 -0
- package/dist/evolution/l2-memskill.js +57 -0
- package/dist/evolution/l2-memskill.js.map +1 -0
- package/dist/evolution/l3-skill-discover.d.ts +4 -0
- package/dist/evolution/l3-skill-discover.d.ts.map +1 -0
- package/dist/evolution/l3-skill-discover.js +139 -0
- package/dist/evolution/l3-skill-discover.js.map +1 -0
- package/dist/evolution/l3-skill-verify.d.ts +12 -0
- package/dist/evolution/l3-skill-verify.d.ts.map +1 -0
- package/dist/evolution/l3-skill-verify.js +122 -0
- package/dist/evolution/l3-skill-verify.js.map +1 -0
- package/dist/evolution/l4-desensitize.d.ts +7 -0
- package/dist/evolution/l4-desensitize.d.ts.map +1 -0
- package/dist/evolution/l4-desensitize.js +30 -0
- package/dist/evolution/l4-desensitize.js.map +1 -0
- package/dist/evolution/l4-group-evolve.d.ts +8 -0
- package/dist/evolution/l4-group-evolve.d.ts.map +1 -0
- package/dist/evolution/l4-group-evolve.js +15 -0
- package/dist/evolution/l4-group-evolve.js.map +1 -0
- package/dist/evolution/maturity-scorer.d.ts +11 -0
- package/dist/evolution/maturity-scorer.d.ts.map +1 -0
- package/dist/evolution/maturity-scorer.js +21 -0
- package/dist/evolution/maturity-scorer.js.map +1 -0
- package/dist/index.d.ts +7 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +33 -5
- package/dist/index.js.map +1 -1
- package/dist/providers/agentkits.d.ts +20 -0
- package/dist/providers/agentkits.d.ts.map +1 -0
- package/dist/providers/agentkits.js +173 -0
- package/dist/providers/agentkits.js.map +1 -0
- package/dist/providers/model-provider.d.ts +16 -0
- package/dist/providers/model-provider.d.ts.map +1 -0
- package/dist/providers/model-provider.js +13 -0
- package/dist/providers/model-provider.js.map +1 -0
- package/dist/providers/model-recommender.d.ts +15 -0
- package/dist/providers/model-recommender.d.ts.map +1 -0
- package/dist/providers/model-recommender.js +71 -0
- package/dist/providers/model-recommender.js.map +1 -0
- package/dist/providers/ollama.d.ts +22 -0
- package/dist/providers/ollama.d.ts.map +1 -0
- package/dist/providers/ollama.js +176 -0
- package/dist/providers/ollama.js.map +1 -0
- package/dist/providers/openai-compat.d.ts +23 -0
- package/dist/providers/openai-compat.d.ts.map +1 -0
- package/dist/providers/openai-compat.js +184 -0
- package/dist/providers/openai-compat.js.map +1 -0
- package/dist/providers/router.d.ts +11 -0
- package/dist/providers/router.d.ts.map +1 -0
- package/dist/providers/router.js +48 -0
- package/dist/providers/router.js.map +1 -0
- package/dist/scheduler/cron.d.ts +19 -0
- package/dist/scheduler/cron.d.ts.map +1 -0
- package/dist/scheduler/cron.js +64 -0
- package/dist/scheduler/cron.js.map +1 -0
- package/dist/schema/oad.d.ts +72 -72
- package/dist/skills/loader.d.ts +16 -0
- package/dist/skills/loader.d.ts.map +1 -0
- package/dist/skills/loader.js +114 -0
- package/dist/skills/loader.js.map +1 -0
- package/dist/skills/matcher.d.ts +18 -0
- package/dist/skills/matcher.d.ts.map +1 -0
- package/dist/skills/matcher.js +70 -0
- package/dist/skills/matcher.js.map +1 -0
- package/dist/studio/agent-pool.d.ts +17 -0
- package/dist/studio/agent-pool.d.ts.map +1 -0
- package/dist/studio/agent-pool.js +35 -0
- package/dist/studio/agent-pool.js.map +1 -0
- package/dist/studio/assistant-tools.d.ts +4 -0
- package/dist/studio/assistant-tools.d.ts.map +1 -0
- package/dist/studio/assistant-tools.js +36 -0
- package/dist/studio/assistant-tools.js.map +1 -0
- package/dist/studio/index.d.ts +1 -0
- package/dist/studio/index.d.ts.map +1 -0
- package/dist/studio/index.js +3 -0
- package/dist/studio/index.js.map +1 -0
- package/dist/studio/server.d.ts +3 -0
- package/dist/studio/server.d.ts.map +1 -1
- package/dist/studio/server.js +156 -24
- package/dist/studio/server.js.map +1 -1
- package/dist/templates/index.d.ts +1 -0
- package/dist/templates/index.d.ts.map +1 -0
- package/dist/templates/index.js +3 -0
- package/dist/templates/index.js.map +1 -0
- package/dist/templates/roles/index.d.ts +4 -0
- package/dist/templates/roles/index.d.ts.map +1 -0
- package/dist/templates/roles/index.js +46 -0
- package/dist/templates/roles/index.js.map +1 -0
- package/dist/templates/template-provider.d.ts +16 -0
- package/dist/templates/template-provider.d.ts.map +1 -0
- package/dist/templates/template-provider.js +60 -0
- package/dist/templates/template-provider.js.map +1 -0
- package/dist/tools/builtin/definitions.d.ts +7 -0
- package/dist/tools/builtin/definitions.d.ts.map +1 -0
- package/dist/tools/builtin/definitions.js +60 -0
- package/dist/tools/builtin/definitions.js.map +1 -0
- package/dist/tools/execute-code.d.ts +20 -0
- package/dist/tools/execute-code.d.ts.map +1 -0
- package/dist/tools/execute-code.js +92 -0
- package/dist/tools/execute-code.js.map +1 -0
- package/dist/tools/hooks.d.ts +47 -0
- package/dist/tools/hooks.d.ts.map +1 -0
- package/dist/tools/hooks.js +69 -0
- package/dist/tools/hooks.js.map +1 -0
- package/dist/tools/index.d.ts +9 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +16 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/permission.d.ts +20 -0
- package/dist/tools/permission.d.ts.map +1 -0
- package/dist/tools/permission.js +35 -0
- package/dist/tools/permission.js.map +1 -0
- package/dist/tools/registry.d.ts +25 -0
- package/dist/tools/registry.d.ts.map +1 -0
- package/dist/tools/registry.js +42 -0
- package/dist/tools/registry.js.map +1 -0
- package/package.json +1 -1
package/dist/core/types.d.ts
CHANGED
|
@@ -1,5 +1,13 @@
|
|
|
1
1
|
import { EventEmitter } from 'events';
|
|
2
2
|
export type AgentState = 'init' | 'ready' | 'running' | 'stopped' | 'error';
|
|
3
|
+
export type SkillLevel = 'builtin' | 'auto' | 'user';
|
|
4
|
+
export type PermissionLevel = 'allow' | 'ask' | 'deny';
|
|
5
|
+
export type DeepBrainLayer = 'l1' | 'l2' | 'l3' | 'l4';
|
|
6
|
+
export type WorkstationLayer = 'workstation' | 'job' | 'industry';
|
|
7
|
+
export type KnowledgeSource = 'l1' | 'l2' | 'l3' | 'l4' | 'user' | 'seed';
|
|
8
|
+
export type EmbeddingProvider = 'ollama' | 'openai' | 'local' | 'none';
|
|
9
|
+
export type BudgetState = 'ok' | 'warn' | 'critical' | 'stop';
|
|
10
|
+
export type ContextLevel = 'simple' | 'recall' | 'complex';
|
|
3
11
|
export interface Message {
|
|
4
12
|
id: string;
|
|
5
13
|
role: 'user' | 'assistant' | 'system';
|
|
@@ -55,4 +63,408 @@ export interface IAgent extends EventEmitter {
|
|
|
55
63
|
registerSkill(skill: ISkill): void;
|
|
56
64
|
bindChannel(channel: IChannel): void;
|
|
57
65
|
}
|
|
66
|
+
export interface UserMessage {
|
|
67
|
+
role: 'user';
|
|
68
|
+
content: string;
|
|
69
|
+
name?: string;
|
|
70
|
+
sessionId?: string;
|
|
71
|
+
timestamp?: number;
|
|
72
|
+
}
|
|
73
|
+
export interface AssistantMessage {
|
|
74
|
+
role: 'assistant';
|
|
75
|
+
content: string;
|
|
76
|
+
toolCalls?: ToolCall[];
|
|
77
|
+
timestamp?: number;
|
|
78
|
+
}
|
|
79
|
+
export interface ToolMessage {
|
|
80
|
+
role: 'tool';
|
|
81
|
+
toolCallId: string;
|
|
82
|
+
content: string;
|
|
83
|
+
isError?: boolean;
|
|
84
|
+
}
|
|
85
|
+
export interface SystemMessage {
|
|
86
|
+
role: 'system';
|
|
87
|
+
content: string;
|
|
88
|
+
}
|
|
89
|
+
export interface ToolCall {
|
|
90
|
+
id: string;
|
|
91
|
+
name: string;
|
|
92
|
+
arguments: string;
|
|
93
|
+
}
|
|
94
|
+
export interface ToolResult {
|
|
95
|
+
callId: string;
|
|
96
|
+
content: string;
|
|
97
|
+
isError?: boolean;
|
|
98
|
+
}
|
|
99
|
+
export type ChatMessage = UserMessage | AssistantMessage | ToolMessage | SystemMessage;
|
|
100
|
+
export interface EgoConfig {
|
|
101
|
+
identity: {
|
|
102
|
+
name: string;
|
|
103
|
+
creature: string;
|
|
104
|
+
emoji: string;
|
|
105
|
+
};
|
|
106
|
+
role: string;
|
|
107
|
+
principles: string[];
|
|
108
|
+
evolutionGoals: string[];
|
|
109
|
+
egoContext: string;
|
|
110
|
+
}
|
|
111
|
+
export interface DeepBrainConfig {
|
|
112
|
+
dbPath: string;
|
|
113
|
+
layers: WorkstationLayer[];
|
|
114
|
+
embeddingProvider: 'ollama' | 'none';
|
|
115
|
+
embeddingModel: string;
|
|
116
|
+
maxEntries?: number;
|
|
117
|
+
autoMigrate?: boolean;
|
|
118
|
+
}
|
|
119
|
+
export interface ModelConfig {
|
|
120
|
+
provider: 'agentkits' | 'openai' | 'anthropic' | 'ollama' | string;
|
|
121
|
+
strategy: 'experience' | 'cost' | 'free';
|
|
122
|
+
apiKey?: string;
|
|
123
|
+
override?: {
|
|
124
|
+
provider: string;
|
|
125
|
+
apiKey: string;
|
|
126
|
+
model: string;
|
|
127
|
+
};
|
|
128
|
+
local: {
|
|
129
|
+
provider: 'ollama';
|
|
130
|
+
model: 'auto' | string;
|
|
131
|
+
embeddingModel: 'auto' | string;
|
|
132
|
+
};
|
|
133
|
+
fallback: 'ollama' | 'none';
|
|
134
|
+
}
|
|
135
|
+
export interface EvolutionConfig {
|
|
136
|
+
strategy: 'experience' | 'cost' | 'free';
|
|
137
|
+
l1: {
|
|
138
|
+
enabled: boolean;
|
|
139
|
+
};
|
|
140
|
+
l2: {
|
|
141
|
+
enabled: boolean;
|
|
142
|
+
trigger: 'idle_2h' | 'messages_50' | 'manual';
|
|
143
|
+
};
|
|
144
|
+
l3: {
|
|
145
|
+
enabled: boolean;
|
|
146
|
+
autoDiscover: boolean;
|
|
147
|
+
verifier: boolean;
|
|
148
|
+
};
|
|
149
|
+
l4: {
|
|
150
|
+
enabled: boolean;
|
|
151
|
+
upwardFlow: boolean;
|
|
152
|
+
desensitize: boolean;
|
|
153
|
+
};
|
|
154
|
+
}
|
|
155
|
+
export interface SkillConfig {
|
|
156
|
+
name: string;
|
|
157
|
+
description: string;
|
|
158
|
+
path: string;
|
|
159
|
+
level: SkillLevel;
|
|
160
|
+
enabled?: boolean;
|
|
161
|
+
parameters?: Record<string, unknown>;
|
|
162
|
+
}
|
|
163
|
+
export interface McpServerConfig {
|
|
164
|
+
name: string;
|
|
165
|
+
command: string;
|
|
166
|
+
args: string[];
|
|
167
|
+
env: Record<string, string>;
|
|
168
|
+
transport?: 'stdio' | 'sse';
|
|
169
|
+
url?: string;
|
|
170
|
+
}
|
|
171
|
+
export interface ToolConfig {
|
|
172
|
+
builtin: string[];
|
|
173
|
+
mcp: McpServerConfig[];
|
|
174
|
+
permissions: Record<string, PermissionLevel>;
|
|
175
|
+
timeout?: number;
|
|
176
|
+
maxConcurrent?: number;
|
|
177
|
+
}
|
|
178
|
+
export interface WebConfig {
|
|
179
|
+
enabled: boolean;
|
|
180
|
+
port?: number;
|
|
181
|
+
host?: string;
|
|
182
|
+
cors?: boolean | string[];
|
|
183
|
+
auth?: boolean;
|
|
184
|
+
wsPath?: string;
|
|
185
|
+
}
|
|
186
|
+
/** @deprecated Use WebConfig */
|
|
187
|
+
export type WebChannelConfig = WebConfig;
|
|
188
|
+
export interface TelegramConfig {
|
|
189
|
+
enabled: boolean;
|
|
190
|
+
token: string;
|
|
191
|
+
webhookUrl?: string;
|
|
192
|
+
allowedUsers?: string[];
|
|
193
|
+
}
|
|
194
|
+
/** @deprecated Use TelegramConfig */
|
|
195
|
+
export type TelegramChannelConfig = TelegramConfig;
|
|
196
|
+
export interface ApiConfig {
|
|
197
|
+
enabled: boolean;
|
|
198
|
+
port?: number;
|
|
199
|
+
basePath?: string;
|
|
200
|
+
auth?: 'bearer' | 'apikey' | 'none';
|
|
201
|
+
apiKeys?: string[];
|
|
202
|
+
}
|
|
203
|
+
/** @deprecated Use ApiConfig */
|
|
204
|
+
export type ApiChannelConfig = ApiConfig;
|
|
205
|
+
export interface VoiceConfig {
|
|
206
|
+
enabled: boolean;
|
|
207
|
+
provider?: 'whisper' | 'azure' | 'google';
|
|
208
|
+
language?: string;
|
|
209
|
+
wakeWord?: string;
|
|
210
|
+
}
|
|
211
|
+
/** @deprecated Use VoiceConfig */
|
|
212
|
+
export type VoiceChannelConfig = VoiceConfig;
|
|
213
|
+
export interface ChannelConfig {
|
|
214
|
+
web?: WebConfig;
|
|
215
|
+
telegram?: TelegramConfig;
|
|
216
|
+
api?: ApiConfig;
|
|
217
|
+
voice?: VoiceConfig;
|
|
218
|
+
}
|
|
219
|
+
export interface GuardrailRule {
|
|
220
|
+
id: string;
|
|
221
|
+
name: string;
|
|
222
|
+
pattern: string | RegExp;
|
|
223
|
+
action: 'block' | 'warn' | 'redact';
|
|
224
|
+
message?: string;
|
|
225
|
+
}
|
|
226
|
+
export interface InputFilter {
|
|
227
|
+
id: string;
|
|
228
|
+
fn: (input: string) => string | null;
|
|
229
|
+
description?: string;
|
|
230
|
+
}
|
|
231
|
+
export interface OutputFilter {
|
|
232
|
+
id: string;
|
|
233
|
+
fn: (output: string) => string | null;
|
|
234
|
+
description?: string;
|
|
235
|
+
}
|
|
236
|
+
export interface GuardrailConfig {
|
|
237
|
+
inputFilters: InputFilter[];
|
|
238
|
+
outputFilters: OutputFilter[];
|
|
239
|
+
blockedPatterns: GuardrailRule[];
|
|
240
|
+
maxInputLength?: number;
|
|
241
|
+
maxOutputLength?: number;
|
|
242
|
+
}
|
|
243
|
+
export interface AgentConfig {
|
|
244
|
+
ego: EgoConfig;
|
|
245
|
+
deepbrain: DeepBrainConfig;
|
|
246
|
+
model: ModelConfig;
|
|
247
|
+
evolution: EvolutionConfig;
|
|
248
|
+
skills: SkillConfig[];
|
|
249
|
+
tools: ToolConfig;
|
|
250
|
+
channels: ChannelConfig;
|
|
251
|
+
guardrails: GuardrailConfig;
|
|
252
|
+
}
|
|
253
|
+
export interface KnowledgeEntry {
|
|
254
|
+
id: string;
|
|
255
|
+
content: string;
|
|
256
|
+
source: KnowledgeSource;
|
|
257
|
+
layer: WorkstationLayer;
|
|
258
|
+
tags: string[];
|
|
259
|
+
embedding: number[] | null;
|
|
260
|
+
maturityScore: number;
|
|
261
|
+
useCount: number;
|
|
262
|
+
lastUsed: string;
|
|
263
|
+
createdAt: string;
|
|
264
|
+
updatedAt: string;
|
|
265
|
+
}
|
|
266
|
+
export interface Experience {
|
|
267
|
+
id: string;
|
|
268
|
+
sessionId: string;
|
|
269
|
+
summary: string;
|
|
270
|
+
lessons: string[];
|
|
271
|
+
errorPatterns: string[];
|
|
272
|
+
createdAt: string;
|
|
273
|
+
}
|
|
274
|
+
export interface SkillRecord {
|
|
275
|
+
id: string;
|
|
276
|
+
name: string;
|
|
277
|
+
description: string;
|
|
278
|
+
filePath: string;
|
|
279
|
+
status: 'candidate' | 'active' | 'retired';
|
|
280
|
+
useCount: number;
|
|
281
|
+
createdAt: string;
|
|
282
|
+
}
|
|
283
|
+
export interface EvolutionLog {
|
|
284
|
+
id: string;
|
|
285
|
+
layer: 'l1' | 'l2' | 'l3' | 'l4';
|
|
286
|
+
action: string;
|
|
287
|
+
details: Record<string, any>;
|
|
288
|
+
modelUsed: string;
|
|
289
|
+
createdAt: string;
|
|
290
|
+
}
|
|
291
|
+
export interface ChatRequest {
|
|
292
|
+
model?: string;
|
|
293
|
+
messages: Message[];
|
|
294
|
+
systemPrompt?: string;
|
|
295
|
+
temperature?: number;
|
|
296
|
+
maxTokens?: number;
|
|
297
|
+
tools?: ToolDefinition[];
|
|
298
|
+
stream?: boolean;
|
|
299
|
+
}
|
|
300
|
+
export interface ChatResponse {
|
|
301
|
+
id: string;
|
|
302
|
+
model: string;
|
|
303
|
+
message: AssistantMessage;
|
|
304
|
+
usage: TokenUsage;
|
|
305
|
+
finishReason: 'stop' | 'tool_calls' | 'length' | 'error';
|
|
306
|
+
}
|
|
307
|
+
export interface TokenUsage {
|
|
308
|
+
promptTokens: number;
|
|
309
|
+
completionTokens: number;
|
|
310
|
+
totalTokens: number;
|
|
311
|
+
}
|
|
312
|
+
export interface StreamChunk {
|
|
313
|
+
id: string;
|
|
314
|
+
delta: string;
|
|
315
|
+
toolCallDelta?: Partial<ToolCall>;
|
|
316
|
+
finishReason?: 'stop' | 'tool_calls' | 'length' | 'error';
|
|
317
|
+
done: boolean;
|
|
318
|
+
}
|
|
319
|
+
export interface EmbedRequest {
|
|
320
|
+
model?: string;
|
|
321
|
+
input: string | string[];
|
|
322
|
+
}
|
|
323
|
+
export interface EmbedResponse {
|
|
324
|
+
model: string;
|
|
325
|
+
embeddings: number[][];
|
|
326
|
+
usage?: TokenUsage;
|
|
327
|
+
}
|
|
328
|
+
export interface ModelInfo {
|
|
329
|
+
id: string;
|
|
330
|
+
provider: string;
|
|
331
|
+
contextLength: number;
|
|
332
|
+
supportedFeatures: ('tools' | 'vision' | 'streaming' | 'embedding')[];
|
|
333
|
+
costPerMToken?: {
|
|
334
|
+
input: number;
|
|
335
|
+
output: number;
|
|
336
|
+
};
|
|
337
|
+
isLocal: boolean;
|
|
338
|
+
}
|
|
339
|
+
export interface ModelProvider {
|
|
340
|
+
chat(req: ChatRequest): Promise<ChatResponse>;
|
|
341
|
+
chatStream(req: ChatRequest): AsyncIterable<StreamChunk>;
|
|
342
|
+
embed(req: EmbedRequest): Promise<EmbedResponse>;
|
|
343
|
+
info(): Promise<ModelInfo[]>;
|
|
344
|
+
}
|
|
345
|
+
export interface RecallQuery {
|
|
346
|
+
query: string;
|
|
347
|
+
layer?: DeepBrainLayer | DeepBrainLayer[] | WorkstationLayer | WorkstationLayer[];
|
|
348
|
+
topK?: number;
|
|
349
|
+
minScore?: number;
|
|
350
|
+
tags?: string[];
|
|
351
|
+
includeEmbedding?: boolean;
|
|
352
|
+
}
|
|
353
|
+
export interface RecallResult {
|
|
354
|
+
entries: KnowledgeEntry[];
|
|
355
|
+
query: string;
|
|
356
|
+
elapsedMs: number;
|
|
357
|
+
}
|
|
358
|
+
export interface StoreResult {
|
|
359
|
+
id: string;
|
|
360
|
+
layer: DeepBrainLayer | WorkstationLayer;
|
|
361
|
+
success: boolean;
|
|
362
|
+
deduplicated?: boolean;
|
|
363
|
+
}
|
|
364
|
+
export interface DeepBrainStats {
|
|
365
|
+
totalEntries: number;
|
|
366
|
+
entriesByLayer: Record<string, number>;
|
|
367
|
+
avgMaturityScore: number;
|
|
368
|
+
lastEvolution?: number;
|
|
369
|
+
}
|
|
370
|
+
export interface DeepBrainProvider {
|
|
371
|
+
store(entry: Omit<KnowledgeEntry, 'id' | 'createdAt' | 'updatedAt'>): Promise<StoreResult>;
|
|
372
|
+
recall(query: RecallQuery): Promise<RecallResult>;
|
|
373
|
+
evolve(layer: DeepBrainLayer, config?: Partial<EvolutionConfig>): Promise<EvolutionLog>;
|
|
374
|
+
getStats(): Promise<DeepBrainStats>;
|
|
375
|
+
}
|
|
376
|
+
export interface TemplateProvider {
|
|
377
|
+
listIndustries(): Promise<string[]>;
|
|
378
|
+
listRoles(industry: string): Promise<string[]>;
|
|
379
|
+
getTemplate(role: string): Promise<OADSpec | null>;
|
|
380
|
+
}
|
|
381
|
+
export interface ToolDefinition {
|
|
382
|
+
name: string;
|
|
383
|
+
description: string;
|
|
384
|
+
parameters: Record<string, any>;
|
|
385
|
+
handler?: Function;
|
|
386
|
+
}
|
|
387
|
+
export interface ToolHook {
|
|
388
|
+
beforeTool?: (name: string, args: Record<string, unknown>) => Promise<void | Record<string, unknown>>;
|
|
389
|
+
afterTool?: (name: string, result: ToolResult) => Promise<void | ToolResult>;
|
|
390
|
+
}
|
|
391
|
+
export interface SkillDefinition {
|
|
392
|
+
name: string;
|
|
393
|
+
description: string;
|
|
394
|
+
triggers: string[];
|
|
395
|
+
content: string;
|
|
396
|
+
references: string[];
|
|
397
|
+
}
|
|
398
|
+
export interface SkillMatch {
|
|
399
|
+
skill: SkillDefinition;
|
|
400
|
+
score: number;
|
|
401
|
+
level: 'L0' | 'L1' | 'L2';
|
|
402
|
+
}
|
|
403
|
+
export interface Chunk {
|
|
404
|
+
type: 'text' | 'tool_call' | 'tool_result' | 'error';
|
|
405
|
+
content: string;
|
|
406
|
+
}
|
|
407
|
+
export interface IterationBudget {
|
|
408
|
+
maxIterations: number;
|
|
409
|
+
maxTokens: number;
|
|
410
|
+
timeoutMs?: number;
|
|
411
|
+
allowOverrun?: boolean;
|
|
412
|
+
}
|
|
413
|
+
export interface BudgetSnapshot {
|
|
414
|
+
iterationsUsed: number;
|
|
415
|
+
iterationsMax: number;
|
|
416
|
+
tokensUsed: number;
|
|
417
|
+
tokensMax: number;
|
|
418
|
+
elapsed: number;
|
|
419
|
+
deadline?: number;
|
|
420
|
+
}
|
|
421
|
+
export interface AgentLoopOptions {
|
|
422
|
+
maxIterations: number;
|
|
423
|
+
config: AgentConfig;
|
|
424
|
+
budgetStatus?: BudgetState;
|
|
425
|
+
stream?: boolean;
|
|
426
|
+
sessionId?: string;
|
|
427
|
+
}
|
|
428
|
+
export interface ContextAssemblerOptions {
|
|
429
|
+
level: ContextLevel;
|
|
430
|
+
maxMessages?: number;
|
|
431
|
+
includeSummary?: boolean;
|
|
432
|
+
includeMemory?: boolean;
|
|
433
|
+
systemPromptOverride?: string;
|
|
434
|
+
}
|
|
435
|
+
export interface AutoDreamAction {
|
|
436
|
+
type: 'merge' | 'trim' | 'reorganize';
|
|
437
|
+
ids: string[];
|
|
438
|
+
newContent?: string;
|
|
439
|
+
}
|
|
440
|
+
export interface MaturityScore {
|
|
441
|
+
frequency: number;
|
|
442
|
+
accuracy: number;
|
|
443
|
+
recency: number;
|
|
444
|
+
coverage: number;
|
|
445
|
+
total: number;
|
|
446
|
+
}
|
|
447
|
+
export interface DesensitizeResult {
|
|
448
|
+
original: string;
|
|
449
|
+
desensitized: string;
|
|
450
|
+
removedEntities: string[];
|
|
451
|
+
}
|
|
452
|
+
export interface OADMetadata {
|
|
453
|
+
name: string;
|
|
454
|
+
version: string;
|
|
455
|
+
description?: string;
|
|
456
|
+
author?: string;
|
|
457
|
+
created: number;
|
|
458
|
+
updated: number;
|
|
459
|
+
tags?: string[];
|
|
460
|
+
}
|
|
461
|
+
export interface OADSpec {
|
|
462
|
+
metadata: OADMetadata;
|
|
463
|
+
ego: EgoConfig;
|
|
464
|
+
model: ModelConfig;
|
|
465
|
+
skills: SkillConfig[];
|
|
466
|
+
tools: ToolConfig;
|
|
467
|
+
channels: ChannelConfig;
|
|
468
|
+
evolution?: EvolutionConfig;
|
|
469
|
+
}
|
|
58
470
|
//# sourceMappingURL=types.d.ts.map
|
package/dist/core/types.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/core/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAItC,MAAM,MAAM,UAAU,GAAG,MAAM,GAAG,OAAO,GAAG,SAAS,GAAG,SAAS,GAAG,OAAO,CAAC;AAE5E,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,YAAY;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,MAAM,EAAE,WAAW,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,MAAM;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,OAAO,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;CACxE;AAED,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,SAAS,CAAC,OAAO,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC;CAClE;AAED,MAAM,WAAW,WAAW;IAC1B,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACnC,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAChD,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IACvD,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/D,KAAK,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1C;AAED,MAAM,WAAW,WAAW;IAC1B,cAAc,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,UAAU,KAAK,IAAI,CAAC;IAC3D,YAAY,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IACzC,aAAa,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IAC1C,eAAe,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,KAAK,IAAI,CAAC;IAC9D,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CACjC;AAED,MAAM,WAAW,MAAO,SAAQ,YAAY;IAC1C,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAC;IAC3B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,aAAa,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAClD,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACnC,WAAW,CAAC,OAAO,EAAE,QAAQ,GAAG,IAAI,CAAC;CACtC"}
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/core/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAItC,MAAM,MAAM,UAAU,GAAG,MAAM,GAAG,OAAO,GAAG,SAAS,GAAG,SAAS,GAAG,OAAO,CAAC;AAC5E,MAAM,MAAM,UAAU,GAAG,SAAS,GAAG,MAAM,GAAG,MAAM,CAAC;AACrD,MAAM,MAAM,eAAe,GAAG,OAAO,GAAG,KAAK,GAAG,MAAM,CAAC;AACvD,MAAM,MAAM,cAAc,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;AACvD,MAAM,MAAM,gBAAgB,GAAG,aAAa,GAAG,KAAK,GAAG,UAAU,CAAC;AAClE,MAAM,MAAM,eAAe,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,GAAG,MAAM,GAAG,MAAM,CAAC;AAC1E,MAAM,MAAM,iBAAiB,GAAG,QAAQ,GAAG,QAAQ,GAAG,OAAO,GAAG,MAAM,CAAC;AACvE,MAAM,MAAM,WAAW,GAAG,IAAI,GAAG,MAAM,GAAG,UAAU,GAAG,MAAM,CAAC;AAC9D,MAAM,MAAM,YAAY,GAAG,QAAQ,GAAG,QAAQ,GAAG,SAAS,CAAC;AAI3D,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,YAAY;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,MAAM,EAAE,WAAW,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,MAAM;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,OAAO,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;CACxE;AAED,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,SAAS,CAAC,OAAO,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC;CAClE;AAED,MAAM,WAAW,WAAW;IAC1B,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACnC,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAChD,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IACvD,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/D,KAAK,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1C;AAED,MAAM,WAAW,WAAW;IAC1B,cAAc,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,UAAU,KAAK,IAAI,CAAC;IAC3D,YAAY,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IACzC,aAAa,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IAC1C,eAAe,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,KAAK,IAAI,CAAC;IAC9D,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CACjC;AAED,MAAM,WAAW,MAAO,SAAQ,YAAY;IAC1C,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAC;IAC3B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,aAAa,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAClD,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACnC,WAAW,CAAC,OAAO,EAAE,QAAQ,GAAG,IAAI,CAAC;CACtC;AAID,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,WAAW,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,QAAQ,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,UAAU;IACzB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,MAAM,WAAW,GAAG,WAAW,GAAG,gBAAgB,GAAG,WAAW,GAAG,aAAa,CAAC;AAIvF,MAAM,WAAW,SAAS;IACxB,QAAQ,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAC5D,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,gBAAgB,EAAE,CAAC;IAC3B,iBAAiB,EAAE,QAAQ,GAAG,MAAM,CAAC;IACrC,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;AAED,MAAM,WAAW,WAAW;IAC1B,QAAQ,EAAE,WAAW,GAAG,QAAQ,GAAG,WAAW,GAAG,QAAQ,GAAG,MAAM,CAAC;IACnE,QAAQ,EAAE,YAAY,GAAG,MAAM,GAAG,MAAM,CAAC;IACzC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAC/D,KAAK,EAAE;QAAE,QAAQ,EAAE,QAAQ,CAAC;QAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAAC;QAAC,cAAc,EAAE,MAAM,GAAG,MAAM,CAAA;KAAE,CAAC;IACvF,QAAQ,EAAE,QAAQ,GAAG,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,YAAY,GAAG,MAAM,GAAG,MAAM,CAAC;IACzC,EAAE,EAAE;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;IACzB,EAAE,EAAE;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,OAAO,EAAE,SAAS,GAAG,aAAa,GAAG,QAAQ,CAAA;KAAE,CAAC;IACxE,EAAE,EAAE;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,YAAY,EAAE,OAAO,CAAC;QAAC,QAAQ,EAAE,OAAO,CAAA;KAAE,CAAC;IACnE,EAAE,EAAE;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,UAAU,EAAE,OAAO,CAAC;QAAC,WAAW,EAAE,OAAO,CAAA;KAAE,CAAC;CACrE;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,UAAU,CAAC;IAClB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5B,SAAS,CAAC,EAAE,OAAO,GAAG,KAAK,CAAC;IAC5B,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,GAAG,EAAE,eAAe,EAAE,CAAC;IACvB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;IAC7C,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,SAAS;IACxB,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,OAAO,GAAG,MAAM,EAAE,CAAC;IAC1B,IAAI,CAAC,EAAE,OAAO,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,gCAAgC;AAChC,MAAM,MAAM,gBAAgB,GAAG,SAAS,CAAC;AAEzC,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;CACzB;AAED,qCAAqC;AACrC,MAAM,MAAM,qBAAqB,GAAG,cAAc,CAAC;AAEnD,MAAM,WAAW,SAAS;IACxB,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,QAAQ,GAAG,QAAQ,GAAG,MAAM,CAAC;IACpC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;CACpB;AAED,gCAAgC;AAChC,MAAM,MAAM,gBAAgB,GAAG,SAAS,CAAC;AAEzC,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,CAAC,EAAE,SAAS,GAAG,OAAO,GAAG,QAAQ,CAAC;IAC1C,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,kCAAkC;AAClC,MAAM,MAAM,kBAAkB,GAAG,WAAW,CAAC;AAE7C,MAAM,WAAW,aAAa;IAC5B,GAAG,CAAC,EAAE,SAAS,CAAC;IAChB,QAAQ,CAAC,EAAE,cAAc,CAAC;IAC1B,GAAG,CAAC,EAAE,SAAS,CAAC;IAChB,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,GAAG,MAAM,CAAC;IACzB,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,QAAQ,CAAC;IACpC,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,EAAE,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,MAAM,GAAG,IAAI,CAAC;IACrC,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,EAAE,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,MAAM,GAAG,IAAI,CAAC;IACtC,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,eAAe;IAC9B,YAAY,EAAE,WAAW,EAAE,CAAC;IAC5B,aAAa,EAAE,YAAY,EAAE,CAAC;IAC9B,eAAe,EAAE,aAAa,EAAE,CAAC;IACjC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,WAAW;IAC1B,GAAG,EAAE,SAAS,CAAC;IACf,SAAS,EAAE,eAAe,CAAC;IAC3B,KAAK,EAAE,WAAW,CAAC;IACnB,SAAS,EAAE,eAAe,CAAC;IAC3B,MAAM,EAAE,WAAW,EAAE,CAAC;IACtB,KAAK,EAAE,UAAU,CAAC;IAClB,QAAQ,EAAE,aAAa,CAAC;IACxB,UAAU,EAAE,eAAe,CAAC;CAC7B;AAID,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,eAAe,CAAC;IACxB,KAAK,EAAE,gBAAgB,CAAC;IACxB,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,SAAS,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IAC3B,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,aAAa,EAAE,MAAM,EAAE,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,WAAW,GAAG,QAAQ,GAAG,SAAS,CAAC;IAC3C,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,IAAI,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAID,MAAM,WAAW,WAAW;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,cAAc,EAAE,CAAC;IACzB,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,gBAAgB,CAAC;IAC1B,KAAK,EAAE,UAAU,CAAC;IAClB,YAAY,EAAE,MAAM,GAAG,YAAY,GAAG,QAAQ,GAAG,OAAO,CAAC;CAC1D;AAED,MAAM,WAAW,UAAU;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;IAClC,YAAY,CAAC,EAAE,MAAM,GAAG,YAAY,GAAG,QAAQ,GAAG,OAAO,CAAC;IAC1D,IAAI,EAAE,OAAO,CAAC;CACf;AAED,MAAM,WAAW,YAAY;IAC3B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;CAC1B;AAED,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,EAAE,EAAE,CAAC;IACvB,KAAK,CAAC,EAAE,UAAU,CAAC;CACpB;AAED,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,aAAa,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,CAAC,OAAO,GAAG,QAAQ,GAAG,WAAW,GAAG,WAAW,CAAC,EAAE,CAAC;IACtE,aAAa,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC;IAClD,OAAO,EAAE,OAAO,CAAC;CAClB;AAID,MAAM,WAAW,aAAa;IAC5B,IAAI,CAAC,GAAG,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IAC9C,UAAU,CAAC,GAAG,EAAE,WAAW,GAAG,aAAa,CAAC,WAAW,CAAC,CAAC;IACzD,KAAK,CAAC,GAAG,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;IACjD,IAAI,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;CAC9B;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,cAAc,GAAG,cAAc,EAAE,GAAG,gBAAgB,GAAG,gBAAgB,EAAE,CAAC;IAClF,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,YAAY;IAC3B,OAAO,EAAE,cAAc,EAAE,CAAC;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,cAAc,GAAG,gBAAgB,CAAC;IACzC,OAAO,EAAE,OAAO,CAAC;IACjB,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC7B,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IAC3F,MAAM,CAAC,KAAK,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IAClD,MAAM,CAAC,KAAK,EAAE,cAAc,EAAE,MAAM,CAAC,EAAE,OAAO,CAAC,eAAe,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IACxF,QAAQ,IAAI,OAAO,CAAC,cAAc,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,gBAAgB;IAC/B,cAAc,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IACpC,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC/C,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC;CACpD;AAID,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAChC,OAAO,CAAC,EAAE,QAAQ,CAAC;CACpB;AAED,MAAM,WAAW,QAAQ;IACvB,UAAU,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC,IAAI,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IACtG,SAAS,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,KAAK,OAAO,CAAC,IAAI,GAAG,UAAU,CAAC,CAAC;CAC9E;AAID,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,EAAE,CAAC;CACtB;AAED,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,eAAe,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;CAC3B;AAID,MAAM,WAAW,KAAK;IACpB,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,aAAa,GAAG,OAAO,CAAC;IACrD,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,eAAe;IAC9B,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC7B,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,gBAAgB;IAC/B,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,WAAW,CAAC;IACpB,YAAY,CAAC,EAAE,WAAW,CAAC;IAC3B,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,uBAAuB;IACtC,KAAK,EAAE,YAAY,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B;AAID,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,OAAO,GAAG,MAAM,GAAG,YAAY,CAAC;IACtC,GAAG,EAAE,MAAM,EAAE,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,EAAE,CAAC;CAC3B;AAID,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB;AAED,MAAM,WAAW,OAAO;IACtB,QAAQ,EAAE,WAAW,CAAC;IACtB,GAAG,EAAE,SAAS,CAAC;IACf,KAAK,EAAE,WAAW,CAAC;IACnB,MAAM,EAAE,WAAW,EAAE,CAAC;IACtB,KAAK,EAAE,UAAU,CAAC;IAClB,QAAQ,EAAE,aAAa,CAAC;IACxB,SAAS,CAAC,EAAE,eAAe,CAAC;CAC7B"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embedding.d.ts","sourceRoot":"","sources":["../../src/deepbrain/embedding.ts"],"names":[],"mappings":"AAmDA,qBAAa,QAAQ;IACnB,OAAO,CAAC,KAAK,CAA4B;gBAE7B,KAAK,CAAC,EAAE,MAAM;IAIpB,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC;CA0BpD;AAED,wBAAgB,gBAAgB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,MAAM,CAUjE"}
|
|
@@ -0,0 +1,108 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.Embedder = void 0;
|
|
4
|
+
exports.cosineSimilarity = cosineSimilarity;
|
|
5
|
+
const OLLAMA_BASE = 'http://localhost:11434';
|
|
6
|
+
const PREFERRED_MODELS = ['nomic-embed-text', 'bge-m3'];
|
|
7
|
+
const LRU_MAX = 100;
|
|
8
|
+
class LruCache {
|
|
9
|
+
maxSize;
|
|
10
|
+
map = new Map();
|
|
11
|
+
constructor(maxSize) {
|
|
12
|
+
this.maxSize = maxSize;
|
|
13
|
+
}
|
|
14
|
+
get(key) {
|
|
15
|
+
const val = this.map.get(key);
|
|
16
|
+
if (val === undefined)
|
|
17
|
+
return null;
|
|
18
|
+
this.map.delete(key);
|
|
19
|
+
this.map.set(key, val);
|
|
20
|
+
return val;
|
|
21
|
+
}
|
|
22
|
+
set(key, value) {
|
|
23
|
+
if (this.map.has(key))
|
|
24
|
+
this.map.delete(key);
|
|
25
|
+
if (this.map.size >= this.maxSize) {
|
|
26
|
+
const oldest = this.map.keys().next().value;
|
|
27
|
+
if (oldest !== undefined)
|
|
28
|
+
this.map.delete(oldest);
|
|
29
|
+
}
|
|
30
|
+
this.map.set(key, value);
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
const cache = new LruCache(LRU_MAX);
|
|
34
|
+
let detectedModel = undefined; // undefined = not yet checked
|
|
35
|
+
async function detectModel() {
|
|
36
|
+
if (detectedModel !== undefined)
|
|
37
|
+
return detectedModel;
|
|
38
|
+
try {
|
|
39
|
+
const res = await fetch(`${OLLAMA_BASE}/api/tags`);
|
|
40
|
+
if (!res.ok) {
|
|
41
|
+
detectedModel = null;
|
|
42
|
+
return null;
|
|
43
|
+
}
|
|
44
|
+
const data = await res.json();
|
|
45
|
+
const names = (data.models ?? []).map((m) => m.name);
|
|
46
|
+
for (const preferred of PREFERRED_MODELS) {
|
|
47
|
+
if (names.some((n) => n === preferred || n.startsWith(preferred + ':'))) {
|
|
48
|
+
detectedModel = preferred;
|
|
49
|
+
return detectedModel;
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
detectedModel = names[0] ?? null;
|
|
53
|
+
return detectedModel;
|
|
54
|
+
}
|
|
55
|
+
catch {
|
|
56
|
+
detectedModel = null;
|
|
57
|
+
return null;
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
class Embedder {
|
|
61
|
+
model;
|
|
62
|
+
constructor(model) {
|
|
63
|
+
this.model = model ?? undefined;
|
|
64
|
+
}
|
|
65
|
+
async embed(text) {
|
|
66
|
+
if (this.model === undefined) {
|
|
67
|
+
this.model = await detectModel();
|
|
68
|
+
}
|
|
69
|
+
if (!this.model)
|
|
70
|
+
return null;
|
|
71
|
+
const key = `${this.model}\x00${text}`;
|
|
72
|
+
const cached = cache.get(key);
|
|
73
|
+
if (cached)
|
|
74
|
+
return cached;
|
|
75
|
+
try {
|
|
76
|
+
const res = await fetch(`${OLLAMA_BASE}/api/embed`, {
|
|
77
|
+
method: 'POST',
|
|
78
|
+
headers: { 'Content-Type': 'application/json' },
|
|
79
|
+
body: JSON.stringify({ model: this.model, input: text }),
|
|
80
|
+
});
|
|
81
|
+
if (!res.ok)
|
|
82
|
+
return null;
|
|
83
|
+
const data = await res.json();
|
|
84
|
+
const vec = data.embeddings?.[0];
|
|
85
|
+
if (!vec || vec.length === 0)
|
|
86
|
+
return null;
|
|
87
|
+
cache.set(key, vec);
|
|
88
|
+
return vec;
|
|
89
|
+
}
|
|
90
|
+
catch {
|
|
91
|
+
return null;
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
exports.Embedder = Embedder;
|
|
96
|
+
function cosineSimilarity(a, b) {
|
|
97
|
+
if (a.length !== b.length || a.length === 0)
|
|
98
|
+
return 0;
|
|
99
|
+
let dot = 0, magA = 0, magB = 0;
|
|
100
|
+
for (let i = 0; i < a.length; i++) {
|
|
101
|
+
dot += a[i] * b[i];
|
|
102
|
+
magA += a[i] * a[i];
|
|
103
|
+
magB += b[i] * b[i];
|
|
104
|
+
}
|
|
105
|
+
const denom = Math.sqrt(magA) * Math.sqrt(magB);
|
|
106
|
+
return denom === 0 ? 0 : dot / denom;
|
|
107
|
+
}
|
|
108
|
+
//# sourceMappingURL=embedding.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embedding.js","sourceRoot":"","sources":["../../src/deepbrain/embedding.ts"],"names":[],"mappings":";;;AAsFA,4CAUC;AAhGD,MAAM,WAAW,GAAG,wBAAwB,CAAC;AAC7C,MAAM,gBAAgB,GAAG,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAC;AACxD,MAAM,OAAO,GAAG,GAAG,CAAC;AAEpB,MAAM,QAAQ;IAGQ;IAFZ,GAAG,GAAG,IAAI,GAAG,EAAoB,CAAC;IAE1C,YAAoB,OAAe;QAAf,YAAO,GAAP,OAAO,CAAQ;IAAG,CAAC;IAEvC,GAAG,CAAC,GAAW;QACb,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC9B,IAAI,GAAG,KAAK,SAAS;YAAE,OAAO,IAAI,CAAC;QACnC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACrB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACvB,OAAO,GAAG,CAAC;IACb,CAAC;IAED,GAAG,CAAC,GAAW,EAAE,KAAe;QAC9B,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC5C,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;YAC5C,IAAI,MAAM,KAAK,SAAS;gBAAE,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IAC3B,CAAC;CACF;AAED,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC;AACpC,IAAI,aAAa,GAA8B,SAAS,CAAC,CAAC,8BAA8B;AAExF,KAAK,UAAU,WAAW;IACxB,IAAI,aAAa,KAAK,SAAS;QAAE,OAAO,aAAa,CAAC;IACtD,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,WAAW,WAAW,CAAC,CAAC;QACnD,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YAAC,aAAa,GAAG,IAAI,CAAC;YAAC,OAAO,IAAI,CAAC;QAAC,CAAC;QACnD,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAA0C,CAAC;QACtE,MAAM,KAAK,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACrD,KAAK,MAAM,SAAS,IAAI,gBAAgB,EAAE,CAAC;YACzC,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,IAAI,CAAC,CAAC,UAAU,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC;gBACxE,aAAa,GAAG,SAAS,CAAC;gBAC1B,OAAO,aAAa,CAAC;YACvB,CAAC;QACH,CAAC;QACD,aAAa,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC;QACjC,OAAO,aAAa,CAAC;IACvB,CAAC;IAAC,MAAM,CAAC;QACP,aAAa,GAAG,IAAI,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAa,QAAQ;IACX,KAAK,CAA4B;IAEzC,YAAY,KAAc;QACxB,IAAI,CAAC,KAAK,GAAG,KAAK,IAAI,SAAS,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,KAAK,GAAG,MAAM,WAAW,EAAE,CAAC;QACnC,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAE7B,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,KAAK,OAAO,IAAI,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC9B,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC;QAE1B,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,WAAW,YAAY,EAAE;gBAClD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;aACzD,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,EAAE;gBAAE,OAAO,IAAI,CAAC;YACzB,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAiC,CAAC;YAC7D,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;YACjC,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC;YAC1C,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YACpB,OAAO,GAAG,CAAC;QACb,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;CACF;AAjCD,4BAiCC;AAED,SAAgB,gBAAgB,CAAC,CAAW,EAAE,CAAW;IACvD,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IACtD,IAAI,GAAG,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC;IAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACnB,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACpB,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAChD,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,KAAK,CAAC;AACvC,CAAC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
export { DeepBrain } from './provider';
|
|
2
|
+
export { BrainStore } from './store';
|
|
3
|
+
export type { BrainStoreOptions, FtsResult } from './store';
|
|
4
|
+
export { Embedder, cosineSimilarity } from './embedding';
|
|
5
|
+
export { recall } from './recall';
|
|
6
|
+
export { readEgo, readDeepBrain, writeDeepBrainSummary } from './workspace-files';
|
|
7
|
+
export { migrate } from './migrate';
|
|
8
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/deepbrain/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AACvC,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,YAAY,EAAE,iBAAiB,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAC5D,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,qBAAqB,EAAE,MAAM,mBAAmB,CAAC;AAClF,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.migrate = exports.writeDeepBrainSummary = exports.readDeepBrain = exports.readEgo = exports.recall = exports.cosineSimilarity = exports.Embedder = exports.BrainStore = exports.DeepBrain = void 0;
|
|
4
|
+
var provider_1 = require("./provider");
|
|
5
|
+
Object.defineProperty(exports, "DeepBrain", { enumerable: true, get: function () { return provider_1.DeepBrain; } });
|
|
6
|
+
var store_1 = require("./store");
|
|
7
|
+
Object.defineProperty(exports, "BrainStore", { enumerable: true, get: function () { return store_1.BrainStore; } });
|
|
8
|
+
var embedding_1 = require("./embedding");
|
|
9
|
+
Object.defineProperty(exports, "Embedder", { enumerable: true, get: function () { return embedding_1.Embedder; } });
|
|
10
|
+
Object.defineProperty(exports, "cosineSimilarity", { enumerable: true, get: function () { return embedding_1.cosineSimilarity; } });
|
|
11
|
+
var recall_1 = require("./recall");
|
|
12
|
+
Object.defineProperty(exports, "recall", { enumerable: true, get: function () { return recall_1.recall; } });
|
|
13
|
+
var workspace_files_1 = require("./workspace-files");
|
|
14
|
+
Object.defineProperty(exports, "readEgo", { enumerable: true, get: function () { return workspace_files_1.readEgo; } });
|
|
15
|
+
Object.defineProperty(exports, "readDeepBrain", { enumerable: true, get: function () { return workspace_files_1.readDeepBrain; } });
|
|
16
|
+
Object.defineProperty(exports, "writeDeepBrainSummary", { enumerable: true, get: function () { return workspace_files_1.writeDeepBrainSummary; } });
|
|
17
|
+
var migrate_1 = require("./migrate");
|
|
18
|
+
Object.defineProperty(exports, "migrate", { enumerable: true, get: function () { return migrate_1.migrate; } });
|
|
19
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/deepbrain/index.ts"],"names":[],"mappings":";;;AAAA,uCAAuC;AAA9B,qGAAA,SAAS,OAAA;AAClB,iCAAqC;AAA5B,mGAAA,UAAU,OAAA;AAEnB,yCAAyD;AAAhD,qGAAA,QAAQ,OAAA;AAAE,6GAAA,gBAAgB,OAAA;AACnC,mCAAkC;AAAzB,gGAAA,MAAM,OAAA;AACf,qDAAkF;AAAzE,0GAAA,OAAO,OAAA;AAAE,gHAAA,aAAa,OAAA;AAAE,wHAAA,qBAAqB,OAAA;AACtD,qCAAoC;AAA3B,kGAAA,OAAO,OAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"migrate.d.ts","sourceRoot":"","sources":["../../src/deepbrain/migrate.ts"],"names":[],"mappings":"AAiCA,wBAAsB,OAAO,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CA8DxD"}
|