@centerseedwu/naru-agent 0.2.0
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/README.md +392 -0
- package/dist/agent.d.ts +39 -0
- package/dist/agent.d.ts.map +1 -0
- package/dist/agent.js +578 -0
- package/dist/agent.js.map +1 -0
- package/dist/compression/base.d.ts +12 -0
- package/dist/compression/base.d.ts.map +1 -0
- package/dist/compression/base.js +2 -0
- package/dist/compression/base.js.map +1 -0
- package/dist/compression/compressor.d.ts +20 -0
- package/dist/compression/compressor.d.ts.map +1 -0
- package/dist/compression/compressor.js +52 -0
- package/dist/compression/compressor.js.map +1 -0
- package/dist/compression/in-memory-store.d.ts +8 -0
- package/dist/compression/in-memory-store.d.ts.map +1 -0
- package/dist/compression/in-memory-store.js +13 -0
- package/dist/compression/in-memory-store.js.map +1 -0
- package/dist/decision/index.d.ts +7 -0
- package/dist/decision/index.d.ts.map +1 -0
- package/dist/decision/index.js +4 -0
- package/dist/decision/index.js.map +1 -0
- package/dist/decision/llm-structured-classifier.d.ts +18 -0
- package/dist/decision/llm-structured-classifier.d.ts.map +1 -0
- package/dist/decision/llm-structured-classifier.js +44 -0
- package/dist/decision/llm-structured-classifier.js.map +1 -0
- package/dist/decision/tool-planner.d.ts +14 -0
- package/dist/decision/tool-planner.d.ts.map +1 -0
- package/dist/decision/tool-planner.js +35 -0
- package/dist/decision/tool-planner.js.map +1 -0
- package/dist/decision/types.d.ts +53 -0
- package/dist/decision/types.d.ts.map +1 -0
- package/dist/decision/types.js +12 -0
- package/dist/decision/types.js.map +1 -0
- package/dist/event-bus.d.ts +10 -0
- package/dist/event-bus.d.ts.map +1 -0
- package/dist/event-bus.js +34 -0
- package/dist/event-bus.js.map +1 -0
- package/dist/guardrails/base.d.ts +10 -0
- package/dist/guardrails/base.d.ts.map +1 -0
- package/dist/guardrails/base.js +2 -0
- package/dist/guardrails/base.js.map +1 -0
- package/dist/guardrails/keyword.d.ts +15 -0
- package/dist/guardrails/keyword.d.ts.map +1 -0
- package/dist/guardrails/keyword.js +37 -0
- package/dist/guardrails/keyword.js.map +1 -0
- package/dist/index.d.ts +51 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +39 -0
- package/dist/index.js.map +1 -0
- package/dist/intent/base.d.ts +19 -0
- package/dist/intent/base.d.ts.map +1 -0
- package/dist/intent/base.js +2 -0
- package/dist/intent/base.js.map +1 -0
- package/dist/intent/llm-classifier.d.ts +22 -0
- package/dist/intent/llm-classifier.d.ts.map +1 -0
- package/dist/intent/llm-classifier.js +49 -0
- package/dist/intent/llm-classifier.js.map +1 -0
- package/dist/intent/tool-calling-classifier.d.ts +16 -0
- package/dist/intent/tool-calling-classifier.d.ts.map +1 -0
- package/dist/intent/tool-calling-classifier.js +56 -0
- package/dist/intent/tool-calling-classifier.js.map +1 -0
- package/dist/knowledge/base.d.ts +14 -0
- package/dist/knowledge/base.d.ts.map +1 -0
- package/dist/knowledge/base.js +10 -0
- package/dist/knowledge/base.js.map +1 -0
- package/dist/knowledge/chroma-store.d.ts +25 -0
- package/dist/knowledge/chroma-store.d.ts.map +1 -0
- package/dist/knowledge/chroma-store.js +70 -0
- package/dist/knowledge/chroma-store.js.map +1 -0
- package/dist/knowledge/contextualizer.d.ts +17 -0
- package/dist/knowledge/contextualizer.d.ts.map +1 -0
- package/dist/knowledge/contextualizer.js +47 -0
- package/dist/knowledge/contextualizer.js.map +1 -0
- package/dist/knowledge/graph-store.d.ts +22 -0
- package/dist/knowledge/graph-store.d.ts.map +1 -0
- package/dist/knowledge/graph-store.js +116 -0
- package/dist/knowledge/graph-store.js.map +1 -0
- package/dist/knowledge/hybrid-store.d.ts +16 -0
- package/dist/knowledge/hybrid-store.d.ts.map +1 -0
- package/dist/knowledge/hybrid-store.js +39 -0
- package/dist/knowledge/hybrid-store.js.map +1 -0
- package/dist/knowledge/in-memory-store.d.ts +27 -0
- package/dist/knowledge/in-memory-store.d.ts.map +1 -0
- package/dist/knowledge/in-memory-store.js +54 -0
- package/dist/knowledge/in-memory-store.js.map +1 -0
- package/dist/knowledge/pgvector-store.d.ts +27 -0
- package/dist/knowledge/pgvector-store.d.ts.map +1 -0
- package/dist/knowledge/pgvector-store.js +68 -0
- package/dist/knowledge/pgvector-store.js.map +1 -0
- package/dist/memory/base.d.ts +21 -0
- package/dist/memory/base.d.ts.map +1 -0
- package/dist/memory/base.js +9 -0
- package/dist/memory/base.js.map +1 -0
- package/dist/memory/in-memory-store.d.ts +10 -0
- package/dist/memory/in-memory-store.d.ts.map +1 -0
- package/dist/memory/in-memory-store.js +35 -0
- package/dist/memory/in-memory-store.js.map +1 -0
- package/dist/memory/manager.d.ts +29 -0
- package/dist/memory/manager.d.ts.map +1 -0
- package/dist/memory/manager.js +121 -0
- package/dist/memory/manager.js.map +1 -0
- package/dist/memory/mem0-manager.d.ts +32 -0
- package/dist/memory/mem0-manager.d.ts.map +1 -0
- package/dist/memory/mem0-manager.js +77 -0
- package/dist/memory/mem0-manager.js.map +1 -0
- package/dist/memory/pgvector-store.d.ts +22 -0
- package/dist/memory/pgvector-store.d.ts.map +1 -0
- package/dist/memory/pgvector-store.js +89 -0
- package/dist/memory/pgvector-store.js.map +1 -0
- package/dist/orchestration/channel.d.ts +18 -0
- package/dist/orchestration/channel.d.ts.map +1 -0
- package/dist/orchestration/channel.js +2 -0
- package/dist/orchestration/channel.js.map +1 -0
- package/dist/orchestration/executor.d.ts +19 -0
- package/dist/orchestration/executor.d.ts.map +1 -0
- package/dist/orchestration/executor.js +2 -0
- package/dist/orchestration/executor.js.map +1 -0
- package/dist/orchestration/index.d.ts +13 -0
- package/dist/orchestration/index.d.ts.map +1 -0
- package/dist/orchestration/index.js +9 -0
- package/dist/orchestration/index.js.map +1 -0
- package/dist/orchestration/intent.d.ts +49 -0
- package/dist/orchestration/intent.d.ts.map +1 -0
- package/dist/orchestration/intent.js +57 -0
- package/dist/orchestration/intent.js.map +1 -0
- package/dist/orchestration/orchestrator.d.ts +59 -0
- package/dist/orchestration/orchestrator.d.ts.map +1 -0
- package/dist/orchestration/orchestrator.js +216 -0
- package/dist/orchestration/orchestrator.js.map +1 -0
- package/dist/orchestration/pending.d.ts +28 -0
- package/dist/orchestration/pending.d.ts.map +1 -0
- package/dist/orchestration/pending.js +32 -0
- package/dist/orchestration/pending.js.map +1 -0
- package/dist/orchestration/result.d.ts +24 -0
- package/dist/orchestration/result.d.ts.map +1 -0
- package/dist/orchestration/result.js +2 -0
- package/dist/orchestration/result.js.map +1 -0
- package/dist/orchestration/session-state.d.ts +24 -0
- package/dist/orchestration/session-state.d.ts.map +1 -0
- package/dist/orchestration/session-state.js +19 -0
- package/dist/orchestration/session-state.js.map +1 -0
- package/dist/orchestration/trace.d.ts +20 -0
- package/dist/orchestration/trace.d.ts.map +1 -0
- package/dist/orchestration/trace.js +2 -0
- package/dist/orchestration/trace.js.map +1 -0
- package/dist/session/base.d.ts +7 -0
- package/dist/session/base.d.ts.map +1 -0
- package/dist/session/base.js +2 -0
- package/dist/session/base.js.map +1 -0
- package/dist/session/in-memory-store.d.ts +9 -0
- package/dist/session/in-memory-store.d.ts.map +1 -0
- package/dist/session/in-memory-store.js +13 -0
- package/dist/session/in-memory-store.js.map +1 -0
- package/dist/session/redis-store.d.ts +16 -0
- package/dist/session/redis-store.d.ts.map +1 -0
- package/dist/session/redis-store.js +29 -0
- package/dist/session/redis-store.js.map +1 -0
- package/dist/skills/base.d.ts +38 -0
- package/dist/skills/base.d.ts.map +1 -0
- package/dist/skills/base.js +24 -0
- package/dist/skills/base.js.map +1 -0
- package/dist/skills/registry.d.ts +11 -0
- package/dist/skills/registry.d.ts.map +1 -0
- package/dist/skills/registry.js +37 -0
- package/dist/skills/registry.js.map +1 -0
- package/dist/skills/selectors.d.ts +23 -0
- package/dist/skills/selectors.d.ts.map +1 -0
- package/dist/skills/selectors.js +52 -0
- package/dist/skills/selectors.js.map +1 -0
- package/dist/tools/base.d.ts +34 -0
- package/dist/tools/base.d.ts.map +1 -0
- package/dist/tools/base.js +22 -0
- package/dist/tools/base.js.map +1 -0
- package/dist/tools/vercel-adapter.d.ts +17 -0
- package/dist/tools/vercel-adapter.d.ts.map +1 -0
- package/dist/tools/vercel-adapter.js +66 -0
- package/dist/tools/vercel-adapter.js.map +1 -0
- package/dist/tracing/collector.d.ts +16 -0
- package/dist/tracing/collector.d.ts.map +1 -0
- package/dist/tracing/collector.js +92 -0
- package/dist/tracing/collector.js.map +1 -0
- package/dist/tracing/exporters/jsonl.d.ts +13 -0
- package/dist/tracing/exporters/jsonl.d.ts.map +1 -0
- package/dist/tracing/exporters/jsonl.js +14 -0
- package/dist/tracing/exporters/jsonl.js.map +1 -0
- package/dist/tracing/trace.d.ts +34 -0
- package/dist/tracing/trace.d.ts.map +1 -0
- package/dist/tracing/trace.js +28 -0
- package/dist/tracing/trace.js.map +1 -0
- package/dist/types.d.ts +77 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +7 -0
- package/dist/types.js.map +1 -0
- package/dist/utils/math.d.ts +3 -0
- package/dist/utils/math.d.ts.map +1 -0
- package/dist/utils/math.js +14 -0
- package/dist/utils/math.js.map +1 -0
- package/dist-cjs/agent.js +581 -0
- package/dist-cjs/compression/base.js +2 -0
- package/dist-cjs/compression/compressor.js +55 -0
- package/dist-cjs/compression/in-memory-store.js +16 -0
- package/dist-cjs/decision/index.js +9 -0
- package/dist-cjs/decision/llm-structured-classifier.js +47 -0
- package/dist-cjs/decision/tool-planner.js +38 -0
- package/dist-cjs/decision/types.js +15 -0
- package/dist-cjs/event-bus.js +37 -0
- package/dist-cjs/guardrails/base.js +2 -0
- package/dist-cjs/guardrails/keyword.js +40 -0
- package/dist-cjs/index.js +84 -0
- package/dist-cjs/intent/base.js +2 -0
- package/dist-cjs/intent/llm-classifier.js +52 -0
- package/dist-cjs/intent/tool-calling-classifier.js +59 -0
- package/dist-cjs/knowledge/base.js +12 -0
- package/dist-cjs/knowledge/chroma-store.js +106 -0
- package/dist-cjs/knowledge/contextualizer.js +50 -0
- package/dist-cjs/knowledge/graph-store.js +152 -0
- package/dist-cjs/knowledge/hybrid-store.js +42 -0
- package/dist-cjs/knowledge/in-memory-store.js +57 -0
- package/dist-cjs/knowledge/pgvector-store.js +71 -0
- package/dist-cjs/memory/base.js +11 -0
- package/dist-cjs/memory/in-memory-store.js +38 -0
- package/dist-cjs/memory/manager.js +124 -0
- package/dist-cjs/memory/mem0-manager.js +113 -0
- package/dist-cjs/memory/pgvector-store.js +92 -0
- package/dist-cjs/orchestration/channel.js +2 -0
- package/dist-cjs/orchestration/executor.js +2 -0
- package/dist-cjs/orchestration/index.js +17 -0
- package/dist-cjs/orchestration/intent.js +61 -0
- package/dist-cjs/orchestration/orchestrator.js +219 -0
- package/dist-cjs/orchestration/pending.js +36 -0
- package/dist-cjs/orchestration/result.js +2 -0
- package/dist-cjs/orchestration/session-state.js +22 -0
- package/dist-cjs/orchestration/trace.js +2 -0
- package/dist-cjs/session/base.js +2 -0
- package/dist-cjs/session/in-memory-store.js +16 -0
- package/dist-cjs/session/redis-store.js +32 -0
- package/dist-cjs/skills/base.js +27 -0
- package/dist-cjs/skills/registry.js +40 -0
- package/dist-cjs/skills/selectors.js +56 -0
- package/dist-cjs/tools/base.js +24 -0
- package/dist-cjs/tools/vercel-adapter.js +69 -0
- package/dist-cjs/tracing/collector.js +95 -0
- package/dist-cjs/tracing/exporters/jsonl.js +17 -0
- package/dist-cjs/tracing/trace.js +31 -0
- package/dist-cjs/types.js +9 -0
- package/dist-cjs/utils/math.js +17 -0
- package/package.json +70 -0
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
import { tool as vercelTool } from "ai";
|
|
2
|
+
/**
|
|
3
|
+
* Convert naru BaseTool[] to Vercel AI SDK ToolSet record.
|
|
4
|
+
*
|
|
5
|
+
* @example
|
|
6
|
+
* const tools = toVercelTools([weatherTool, searchTool]);
|
|
7
|
+
* const result = await generateText({ model, tools, prompt: "..." });
|
|
8
|
+
*/
|
|
9
|
+
export function toVercelTools(tools) {
|
|
10
|
+
const result = {};
|
|
11
|
+
for (const t of tools) {
|
|
12
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
13
|
+
result[t.name] = vercelTool({
|
|
14
|
+
description: t.description,
|
|
15
|
+
parameters: t.parameters,
|
|
16
|
+
execute: async (params) => {
|
|
17
|
+
const maxAttempts = (t.retries ?? 0) + 1;
|
|
18
|
+
for (let attempt = 1; attempt <= maxAttempts; attempt++) {
|
|
19
|
+
try {
|
|
20
|
+
const exec = t.execute(params);
|
|
21
|
+
if (t.timeout != null) {
|
|
22
|
+
let timerId;
|
|
23
|
+
const timeout = new Promise((_, reject) => {
|
|
24
|
+
timerId = setTimeout(() => reject(new Error(`Timed out after ${t.timeout}ms`)), t.timeout);
|
|
25
|
+
});
|
|
26
|
+
try {
|
|
27
|
+
return await Promise.race([exec, timeout]);
|
|
28
|
+
}
|
|
29
|
+
finally {
|
|
30
|
+
clearTimeout(timerId);
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
return await exec;
|
|
34
|
+
}
|
|
35
|
+
catch (err) {
|
|
36
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
37
|
+
if (attempt >= maxAttempts) {
|
|
38
|
+
return `[Tool Error] ${t.name}: ${msg}`;
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
// unreachable, but satisfies TS
|
|
43
|
+
return `[Tool Error] ${t.name}: unknown failure`;
|
|
44
|
+
},
|
|
45
|
+
});
|
|
46
|
+
}
|
|
47
|
+
return result;
|
|
48
|
+
}
|
|
49
|
+
/**
|
|
50
|
+
* Same as toVercelTools but with a no-op execute.
|
|
51
|
+
* Useful for tool planning where you need LLM tool selection
|
|
52
|
+
* without actually executing tools.
|
|
53
|
+
*/
|
|
54
|
+
export function toVercelToolsNoop(tools) {
|
|
55
|
+
const result = {};
|
|
56
|
+
for (const t of tools) {
|
|
57
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
58
|
+
result[t.name] = vercelTool({
|
|
59
|
+
description: t.description,
|
|
60
|
+
parameters: t.parameters,
|
|
61
|
+
execute: async () => "",
|
|
62
|
+
});
|
|
63
|
+
}
|
|
64
|
+
return result;
|
|
65
|
+
}
|
|
66
|
+
//# sourceMappingURL=vercel-adapter.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"vercel-adapter.js","sourceRoot":"","sources":["../../src/tools/vercel-adapter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,IAAI,UAAU,EAAgB,MAAM,IAAI,CAAC;AAGtD;;;;;;GAMG;AACH,MAAM,UAAU,aAAa,CAC3B,KAAiB;IAEjB,MAAM,MAAM,GAA4B,EAAE,CAAC;IAC3C,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;QACtB,8DAA8D;QAC9D,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAI,UAAkB,CAAC;YACnC,WAAW,EAAE,CAAC,CAAC,WAAW;YAC1B,UAAU,EAAE,CAAC,CAAC,UAAU;YACxB,OAAO,EAAE,KAAK,EAAE,MAA+B,EAAE,EAAE;gBACjD,MAAM,WAAW,GAAG,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;gBACzC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;oBACxD,IAAI,CAAC;wBACH,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;wBAC/B,IAAI,CAAC,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC;4BACtB,IAAI,OAAsC,CAAC;4BAC3C,MAAM,OAAO,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gCAC/C,OAAO,GAAG,UAAU,CAClB,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,EACzD,CAAC,CAAC,OAAO,CACV,CAAC;4BACJ,CAAC,CAAC,CAAC;4BACH,IAAI,CAAC;gCACH,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC;4BAC7C,CAAC;oCAAS,CAAC;gCACT,YAAY,CAAC,OAAQ,CAAC,CAAC;4BACzB,CAAC;wBACH,CAAC;wBACD,OAAO,MAAM,IAAI,CAAC;oBACpB,CAAC;oBAAC,OAAO,GAAY,EAAE,CAAC;wBACtB,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;wBAC7D,IAAI,OAAO,IAAI,WAAW,EAAE,CAAC;4BAC3B,OAAO,gBAAgB,CAAC,CAAC,IAAI,KAAK,GAAG,EAAE,CAAC;wBAC1C,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,gCAAgC;gBAChC,OAAO,gBAAgB,CAAC,CAAC,IAAI,mBAAmB,CAAC;YACnD,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IACD,OAAO,MAAiB,CAAC;AAC3B,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,iBAAiB,CAAC,KAAiB;IACjD,MAAM,MAAM,GAA4B,EAAE,CAAC;IAC3C,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;QACtB,8DAA8D;QAC9D,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAI,UAAkB,CAAC;YACnC,WAAW,EAAE,CAAC,CAAC,WAAW;YAC1B,UAAU,EAAE,CAAC,CAAC,UAAU;YACxB,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;SACxB,CAAC,CAAC;IACL,CAAC;IACD,OAAO,MAAiB,CAAC;AAC3B,CAAC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { EventBus } from "../event-bus.js";
|
|
2
|
+
import type { NaruResult } from "../types.js";
|
|
3
|
+
import { type Trace } from "./trace.js";
|
|
4
|
+
export declare class TraceCollector {
|
|
5
|
+
private eventBus;
|
|
6
|
+
private currentTrace;
|
|
7
|
+
private activeSpans;
|
|
8
|
+
constructor(eventBus: EventBus);
|
|
9
|
+
private setupListeners;
|
|
10
|
+
startTrace(message: string, userId?: string, sessionId?: string): string;
|
|
11
|
+
startSpan(name: string, attributes?: Record<string, unknown>): string;
|
|
12
|
+
endSpan(name: string, error?: string): void;
|
|
13
|
+
private addSpanEvent;
|
|
14
|
+
endTrace(result: NaruResult): Trace | null;
|
|
15
|
+
}
|
|
16
|
+
//# sourceMappingURL=collector.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"collector.d.ts","sourceRoot":"","sources":["../../src/tracing/collector.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,EAA2B,KAAK,KAAK,EAAa,MAAM,YAAY,CAAC;AAE5E,qBAAa,cAAc;IACzB,OAAO,CAAC,QAAQ,CAAW;IAC3B,OAAO,CAAC,YAAY,CAAsB;IAC1C,OAAO,CAAC,WAAW,CAA2B;gBAElC,QAAQ,EAAE,QAAQ;IAK9B,OAAO,CAAC,cAAc;IAkBtB,UAAU,CACR,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,MAAM,GACjB,MAAM;IAWT,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;IAarE,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI;IAa3C,OAAO,CAAC,YAAY;IAYpB,QAAQ,CAAC,MAAM,EAAE,UAAU,GAAG,KAAK,GAAG,IAAI;CAe3C"}
|
|
@@ -0,0 +1,92 @@
|
|
|
1
|
+
import { v4 as uuidv4 } from "uuid";
|
|
2
|
+
import { createTrace, createSpan } from "./trace.js";
|
|
3
|
+
export class TraceCollector {
|
|
4
|
+
eventBus;
|
|
5
|
+
currentTrace = null;
|
|
6
|
+
activeSpans = new Map();
|
|
7
|
+
constructor(eventBus) {
|
|
8
|
+
this.eventBus = eventBus;
|
|
9
|
+
this.setupListeners();
|
|
10
|
+
}
|
|
11
|
+
setupListeners() {
|
|
12
|
+
this.eventBus.on("memory_retrieved", (data) => {
|
|
13
|
+
this.addSpanEvent("memory_retrieval", data);
|
|
14
|
+
});
|
|
15
|
+
this.eventBus.on("intent_classified", (data) => {
|
|
16
|
+
this.addSpanEvent("intent_classification", data);
|
|
17
|
+
});
|
|
18
|
+
this.eventBus.on("knowledge_retrieved", (data) => {
|
|
19
|
+
this.addSpanEvent("knowledge_retrieval", data);
|
|
20
|
+
});
|
|
21
|
+
this.eventBus.on("before_llm_call", () => {
|
|
22
|
+
this.startSpan("llm_call");
|
|
23
|
+
});
|
|
24
|
+
this.eventBus.on("after_llm_call", () => {
|
|
25
|
+
this.endSpan("llm_call");
|
|
26
|
+
});
|
|
27
|
+
}
|
|
28
|
+
startTrace(message, userId, sessionId) {
|
|
29
|
+
const traceId = uuidv4();
|
|
30
|
+
this.currentTrace = createTrace({
|
|
31
|
+
traceId,
|
|
32
|
+
threadId: sessionId ?? null,
|
|
33
|
+
userId: userId ?? null,
|
|
34
|
+
input: message,
|
|
35
|
+
});
|
|
36
|
+
return traceId;
|
|
37
|
+
}
|
|
38
|
+
startSpan(name, attributes) {
|
|
39
|
+
if (!this.currentTrace)
|
|
40
|
+
return "";
|
|
41
|
+
const spanId = uuidv4();
|
|
42
|
+
const span = createSpan({
|
|
43
|
+
spanId,
|
|
44
|
+
traceId: this.currentTrace.traceId,
|
|
45
|
+
name,
|
|
46
|
+
attributes,
|
|
47
|
+
});
|
|
48
|
+
this.activeSpans.set(name, span);
|
|
49
|
+
return spanId;
|
|
50
|
+
}
|
|
51
|
+
endSpan(name, error) {
|
|
52
|
+
const span = this.activeSpans.get(name);
|
|
53
|
+
if (span) {
|
|
54
|
+
span.endTime = Date.now();
|
|
55
|
+
if (error) {
|
|
56
|
+
span.status = "ERROR";
|
|
57
|
+
span.errorMessage = error;
|
|
58
|
+
}
|
|
59
|
+
this.currentTrace?.spans.push(span);
|
|
60
|
+
this.activeSpans.delete(name);
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
addSpanEvent(name, data) {
|
|
64
|
+
if (!this.currentTrace)
|
|
65
|
+
return;
|
|
66
|
+
const span = createSpan({
|
|
67
|
+
spanId: uuidv4(),
|
|
68
|
+
traceId: this.currentTrace.traceId,
|
|
69
|
+
name,
|
|
70
|
+
endTime: Date.now(),
|
|
71
|
+
attributes: data ?? {},
|
|
72
|
+
});
|
|
73
|
+
this.currentTrace.spans.push(span);
|
|
74
|
+
}
|
|
75
|
+
endTrace(result) {
|
|
76
|
+
if (!this.currentTrace)
|
|
77
|
+
return null;
|
|
78
|
+
this.currentTrace.endTime = Date.now();
|
|
79
|
+
this.currentTrace.output = result.content;
|
|
80
|
+
this.currentTrace.blocked = result.blocked;
|
|
81
|
+
this.currentTrace.usage = result.usage;
|
|
82
|
+
this.currentTrace.toolCalls = result.toolCalls;
|
|
83
|
+
if (result.intent) {
|
|
84
|
+
this.currentTrace.intent = result.intent;
|
|
85
|
+
}
|
|
86
|
+
const trace = this.currentTrace;
|
|
87
|
+
this.currentTrace = null;
|
|
88
|
+
this.activeSpans.clear();
|
|
89
|
+
return trace;
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
//# sourceMappingURL=collector.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"collector.js","sourceRoot":"","sources":["../../src/tracing/collector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM,CAAC;AAGpC,OAAO,EAAE,WAAW,EAAE,UAAU,EAAyB,MAAM,YAAY,CAAC;AAE5E,MAAM,OAAO,cAAc;IACjB,QAAQ,CAAW;IACnB,YAAY,GAAiB,IAAI,CAAC;IAClC,WAAW,GAAG,IAAI,GAAG,EAAgB,CAAC;IAE9C,YAAY,QAAkB;QAC5B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,cAAc,EAAE,CAAC;IACxB,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC,IAAI,EAAE,EAAE;YAC5C,IAAI,CAAC,YAAY,CAAC,kBAAkB,EAAE,IAA+B,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,IAAI,EAAE,EAAE;YAC7C,IAAI,CAAC,YAAY,CAAC,uBAAuB,EAAE,IAA+B,CAAC,CAAC;QAC9E,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,qBAAqB,EAAE,CAAC,IAAI,EAAE,EAAE;YAC/C,IAAI,CAAC,YAAY,CAAC,qBAAqB,EAAE,IAA+B,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,iBAAiB,EAAE,GAAG,EAAE;YACvC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,gBAAgB,EAAE,GAAG,EAAE;YACtC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CACR,OAAe,EACf,MAAe,EACf,SAAkB;QAElB,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC;QACzB,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;YAC9B,OAAO;YACP,QAAQ,EAAE,SAAS,IAAI,IAAI;YAC3B,MAAM,EAAE,MAAM,IAAI,IAAI;YACtB,KAAK,EAAE,OAAO;SACf,CAAC,CAAC;QACH,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,UAAoC;QAC1D,IAAI,CAAC,IAAI,CAAC,YAAY;YAAE,OAAO,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,UAAU,CAAC;YACtB,MAAM;YACN,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO;YAClC,IAAI;YACJ,UAAU;SACX,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACjC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,OAAO,CAAC,IAAY,EAAE,KAAc;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC1B,IAAI,KAAK,EAAE,CAAC;gBACV,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC;gBACtB,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;YAC5B,CAAC;YACD,IAAI,CAAC,YAAY,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,IAAY,EAAE,IAA8B;QAC/D,IAAI,CAAC,IAAI,CAAC,YAAY;YAAE,OAAO;QAC/B,MAAM,IAAI,GAAG,UAAU,CAAC;YACtB,MAAM,EAAE,MAAM,EAAE;YAChB,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO;YAClC,IAAI;YACJ,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;YACnB,UAAU,EAAE,IAAI,IAAI,EAAE;SACvB,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC;IAED,QAAQ,CAAC,MAAkB;QACzB,IAAI,CAAC,IAAI,CAAC,YAAY;YAAE,OAAO,IAAI,CAAC;QACpC,IAAI,CAAC,YAAY,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvC,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC;QAC1C,IAAI,CAAC,YAAY,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC3C,IAAI,CAAC,YAAY,CAAC,KAAK,GAAG,MAAM,CAAC,KAA0C,CAAC;QAC5E,IAAI,CAAC,YAAY,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAC/C,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,MAAM,CAAC,MAA4C,CAAC;QACjF,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC;QAChC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC;IACf,CAAC;CACF"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { Trace } from "../trace.js";
|
|
2
|
+
export interface BaseTraceExporter {
|
|
3
|
+
export(trace: Trace): Promise<void>;
|
|
4
|
+
}
|
|
5
|
+
/**
|
|
6
|
+
* Exports traces as JSONL (one JSON object per line).
|
|
7
|
+
*/
|
|
8
|
+
export declare class JSONLTraceExporter implements BaseTraceExporter {
|
|
9
|
+
private filePath;
|
|
10
|
+
constructor(filePath: string);
|
|
11
|
+
export(trace: Trace): Promise<void>;
|
|
12
|
+
}
|
|
13
|
+
//# sourceMappingURL=jsonl.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"jsonl.d.ts","sourceRoot":"","sources":["../../../src/tracing/exporters/jsonl.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAEzC,MAAM,WAAW,iBAAiB;IAChC,MAAM,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACrC;AAED;;GAEG;AACH,qBAAa,kBAAmB,YAAW,iBAAiB;IAC1D,OAAO,CAAC,QAAQ,CAAS;gBAEb,QAAQ,EAAE,MAAM;IAItB,MAAM,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC;CAG1C"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import { appendFile } from "node:fs/promises";
|
|
2
|
+
/**
|
|
3
|
+
* Exports traces as JSONL (one JSON object per line).
|
|
4
|
+
*/
|
|
5
|
+
export class JSONLTraceExporter {
|
|
6
|
+
filePath;
|
|
7
|
+
constructor(filePath) {
|
|
8
|
+
this.filePath = filePath;
|
|
9
|
+
}
|
|
10
|
+
async export(trace) {
|
|
11
|
+
await appendFile(this.filePath, JSON.stringify(trace) + "\n", "utf-8");
|
|
12
|
+
}
|
|
13
|
+
}
|
|
14
|
+
//# sourceMappingURL=jsonl.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"jsonl.js","sourceRoot":"","sources":["../../../src/tracing/exporters/jsonl.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAO9C;;GAEG;AACH,MAAM,OAAO,kBAAkB;IACrB,QAAQ,CAAS;IAEzB,YAAY,QAAgB;QAC1B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAY;QACvB,MAAM,UAAU,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;IACzE,CAAC;CACF"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
export interface Span {
|
|
2
|
+
spanId: string;
|
|
3
|
+
traceId: string;
|
|
4
|
+
name: string;
|
|
5
|
+
startTime: number;
|
|
6
|
+
endTime: number | null;
|
|
7
|
+
attributes: Record<string, unknown>;
|
|
8
|
+
status: "OK" | "ERROR";
|
|
9
|
+
errorMessage: string | null;
|
|
10
|
+
}
|
|
11
|
+
export interface Trace {
|
|
12
|
+
traceId: string;
|
|
13
|
+
threadId: string | null;
|
|
14
|
+
userId: string | null;
|
|
15
|
+
startTime: number;
|
|
16
|
+
endTime: number | null;
|
|
17
|
+
input: string;
|
|
18
|
+
output: string;
|
|
19
|
+
blocked: boolean;
|
|
20
|
+
usage: Record<string, number>;
|
|
21
|
+
intent: Record<string, unknown> | null;
|
|
22
|
+
toolCalls: string[];
|
|
23
|
+
spans: Span[];
|
|
24
|
+
metadata: Record<string, unknown>;
|
|
25
|
+
}
|
|
26
|
+
export declare function createTrace(partial: Partial<Trace> & {
|
|
27
|
+
traceId: string;
|
|
28
|
+
}): Trace;
|
|
29
|
+
export declare function createSpan(partial: Partial<Span> & {
|
|
30
|
+
spanId: string;
|
|
31
|
+
traceId: string;
|
|
32
|
+
name: string;
|
|
33
|
+
}): Span;
|
|
34
|
+
//# sourceMappingURL=trace.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"trace.d.ts","sourceRoot":"","sources":["../../src/tracing/trace.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,IAAI;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACpC,MAAM,EAAE,IAAI,GAAG,OAAO,CAAC;IACvB,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;CAC7B;AAED,MAAM,WAAW,KAAK;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC9B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IACvC,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,wBAAgB,WAAW,CAAC,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,GAAG;IAAE,OAAO,EAAE,MAAM,CAAA;CAAE,GAAG,KAAK,CAgBhF;AAED,wBAAgB,UAAU,CACxB,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACzE,IAAI,CASN"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
export function createTrace(partial) {
|
|
2
|
+
return {
|
|
3
|
+
threadId: null,
|
|
4
|
+
userId: null,
|
|
5
|
+
startTime: Date.now(),
|
|
6
|
+
endTime: null,
|
|
7
|
+
input: "",
|
|
8
|
+
output: "",
|
|
9
|
+
blocked: false,
|
|
10
|
+
usage: {},
|
|
11
|
+
intent: null,
|
|
12
|
+
toolCalls: [],
|
|
13
|
+
spans: [],
|
|
14
|
+
metadata: {},
|
|
15
|
+
...partial,
|
|
16
|
+
};
|
|
17
|
+
}
|
|
18
|
+
export function createSpan(partial) {
|
|
19
|
+
return {
|
|
20
|
+
startTime: Date.now(),
|
|
21
|
+
endTime: null,
|
|
22
|
+
attributes: {},
|
|
23
|
+
status: "OK",
|
|
24
|
+
errorMessage: null,
|
|
25
|
+
...partial,
|
|
26
|
+
};
|
|
27
|
+
}
|
|
28
|
+
//# sourceMappingURL=trace.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"trace.js","sourceRoot":"","sources":["../../src/tracing/trace.ts"],"names":[],"mappings":"AA2BA,MAAM,UAAU,WAAW,CAAC,OAA6C;IACvE,OAAO;QACL,QAAQ,EAAE,IAAI;QACd,MAAM,EAAE,IAAI;QACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,OAAO,EAAE,IAAI;QACb,KAAK,EAAE,EAAE;QACT,MAAM,EAAE,EAAE;QACV,OAAO,EAAE,KAAK;QACd,KAAK,EAAE,EAAE;QACT,MAAM,EAAE,IAAI;QACZ,SAAS,EAAE,EAAE;QACb,KAAK,EAAE,EAAE;QACT,QAAQ,EAAE,EAAE;QACZ,GAAG,OAAO;KACX,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,UAAU,CACxB,OAA0E;IAE1E,OAAO;QACL,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,OAAO,EAAE,IAAI;QACb,UAAU,EAAE,EAAE;QACd,MAAM,EAAE,IAAI;QACZ,YAAY,EAAE,IAAI;QAClB,GAAG,OAAO;KACX,CAAC;AACJ,CAAC"}
|
package/dist/types.d.ts
ADDED
|
@@ -0,0 +1,77 @@
|
|
|
1
|
+
import type { LanguageModel, ModelMessage, ToolChoice } from "ai";
|
|
2
|
+
import type { BaseTool } from "./tools/base.js";
|
|
3
|
+
export interface TokenUsage {
|
|
4
|
+
promptTokens: number;
|
|
5
|
+
completionTokens: number;
|
|
6
|
+
totalTokens: number;
|
|
7
|
+
cacheReadTokens?: number;
|
|
8
|
+
cacheWriteTokens?: number;
|
|
9
|
+
}
|
|
10
|
+
/** Map Vercel AI SDK usage shape → naru TokenUsage. */
|
|
11
|
+
export declare function normalizeUsage(usage?: {
|
|
12
|
+
inputTokens?: number;
|
|
13
|
+
outputTokens?: number;
|
|
14
|
+
totalTokens?: number;
|
|
15
|
+
}): TokenUsage;
|
|
16
|
+
export interface NaruResult {
|
|
17
|
+
content: string;
|
|
18
|
+
blocked: boolean;
|
|
19
|
+
usage: TokenUsage;
|
|
20
|
+
intent: IntentResult | null;
|
|
21
|
+
toolCalls: string[];
|
|
22
|
+
timings: Record<string, number>;
|
|
23
|
+
sessionId: string | null;
|
|
24
|
+
traceId: string | null;
|
|
25
|
+
trace: import("./tracing/trace.js").Trace | null;
|
|
26
|
+
}
|
|
27
|
+
export interface IntentResult {
|
|
28
|
+
needsKnowledge: boolean;
|
|
29
|
+
needsTools: boolean;
|
|
30
|
+
raw: string;
|
|
31
|
+
}
|
|
32
|
+
export interface NaruAgentConfig {
|
|
33
|
+
model: LanguageModel;
|
|
34
|
+
name?: string;
|
|
35
|
+
instructions?: string[];
|
|
36
|
+
tools?: import("./tools/base.js").BaseTool[];
|
|
37
|
+
alwaysTools?: import("./tools/base.js").BaseTool[];
|
|
38
|
+
knowledgeStore?: import("./knowledge/base.js").BaseKnowledgeStore;
|
|
39
|
+
knowledgeTopK?: number;
|
|
40
|
+
knowledgeMinScore?: number;
|
|
41
|
+
intentClassifier?: import("./intent/base.js").BaseIntentClassifier;
|
|
42
|
+
toolCallingClassifier?: import("./intent/base.js").BaseToolCallingClassifier;
|
|
43
|
+
memoryManager?: import("./memory/manager.js").MemoryManager;
|
|
44
|
+
guardrails?: import("./guardrails/base.js").BaseGuardrail[];
|
|
45
|
+
toolCallLimit?: number;
|
|
46
|
+
temperature?: number;
|
|
47
|
+
prefetchTimeout?: number;
|
|
48
|
+
toolChoice?: ToolChoice<Record<string, unknown>>;
|
|
49
|
+
toolChoiceResolver?: (context: {
|
|
50
|
+
message: string;
|
|
51
|
+
userId?: string;
|
|
52
|
+
sessionId?: string;
|
|
53
|
+
intent: IntentResult | null;
|
|
54
|
+
activeTools: BaseTool[];
|
|
55
|
+
}) => ToolChoice<Record<string, unknown>> | undefined;
|
|
56
|
+
sessionStore?: import("./session/base.js").BaseSessionStore;
|
|
57
|
+
numHistoryMessages?: number;
|
|
58
|
+
contextCompression?: boolean;
|
|
59
|
+
summaryStore?: import("./compression/base.js").BaseSummaryStore;
|
|
60
|
+
summaryModel?: LanguageModel;
|
|
61
|
+
compressionKeepLastRounds?: number;
|
|
62
|
+
compressionThresholdRounds?: number;
|
|
63
|
+
eventBus?: import("./event-bus.js").EventBus;
|
|
64
|
+
prefetchHooks?: Array<(message: string, userId?: string) => Promise<string>>;
|
|
65
|
+
traceExporters?: import("./tracing/exporters/jsonl.js").BaseTraceExporter[];
|
|
66
|
+
skills?: import("./skills/base.js").BaseSkill[];
|
|
67
|
+
skillSelector?: import("./skills/selectors.js").BaseSkillSelector;
|
|
68
|
+
maxActiveSkills?: number;
|
|
69
|
+
promptCaching?: boolean;
|
|
70
|
+
}
|
|
71
|
+
export interface ChatOptions {
|
|
72
|
+
userId?: string;
|
|
73
|
+
sessionId?: string;
|
|
74
|
+
}
|
|
75
|
+
export type EmbedFn = (texts: string[]) => Promise<number[][]>;
|
|
76
|
+
export { ModelMessage };
|
|
77
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAClE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAEhD,MAAM,WAAW,UAAU;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED,uDAAuD;AACvD,wBAAgB,cAAc,CAC5B,KAAK,CAAC,EAAE;IAAE,WAAW,CAAC,EAAE,MAAM,CAAC;IAAC,YAAY,CAAC,EAAE,MAAM,CAAC;IAAC,WAAW,CAAC,EAAE,MAAM,CAAA;CAAE,GAC5E,UAAU,CAIZ;AAED,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,EAAE,UAAU,CAAC;IAClB,MAAM,EAAE,YAAY,GAAG,IAAI,CAAC;IAC5B,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,KAAK,EAAE,OAAO,oBAAoB,EAAE,KAAK,GAAG,IAAI,CAAC;CAClD;AAED,MAAM,WAAW,YAAY;IAC3B,cAAc,EAAE,OAAO,CAAC;IACxB,UAAU,EAAE,OAAO,CAAC;IACpB,GAAG,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,aAAa,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IAGxB,KAAK,CAAC,EAAE,OAAO,iBAAiB,EAAE,QAAQ,EAAE,CAAC;IAC7C,WAAW,CAAC,EAAE,OAAO,iBAAiB,EAAE,QAAQ,EAAE,CAAC;IACnD,cAAc,CAAC,EAAE,OAAO,qBAAqB,EAAE,kBAAkB,CAAC;IAClE,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAG3B,gBAAgB,CAAC,EAAE,OAAO,kBAAkB,EAAE,oBAAoB,CAAC;IACnE,qBAAqB,CAAC,EAAE,OAAO,kBAAkB,EAAE,yBAAyB,CAAC;IAG7E,aAAa,CAAC,EAAE,OAAO,qBAAqB,EAAE,aAAa,CAAC;IAG5D,UAAU,CAAC,EAAE,OAAO,sBAAsB,EAAE,aAAa,EAAE,CAAC;IAG5D,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,CAAC,EAAE,UAAU,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IACjD,kBAAkB,CAAC,EAAE,CAAC,OAAO,EAAE;QAC7B,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,MAAM,EAAE,YAAY,GAAG,IAAI,CAAC;QAC5B,WAAW,EAAE,QAAQ,EAAE,CAAC;KACzB,KAAK,UAAU,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,GAAG,SAAS,CAAC;IAGtD,YAAY,CAAC,EAAE,OAAO,mBAAmB,EAAE,gBAAgB,CAAC;IAC5D,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAG5B,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,YAAY,CAAC,EAAE,OAAO,uBAAuB,EAAE,gBAAgB,CAAC;IAChE,YAAY,CAAC,EAAE,aAAa,CAAC;IAC7B,yBAAyB,CAAC,EAAE,MAAM,CAAC;IACnC,0BAA0B,CAAC,EAAE,MAAM,CAAC;IAGpC,QAAQ,CAAC,EAAE,OAAO,gBAAgB,EAAE,QAAQ,CAAC;IAC7C,aAAa,CAAC,EAAE,KAAK,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7E,cAAc,CAAC,EAAE,OAAO,8BAA8B,EAAE,iBAAiB,EAAE,CAAC;IAG5E,MAAM,CAAC,EAAE,OAAO,kBAAkB,EAAE,SAAS,EAAE,CAAC;IAChD,aAAa,CAAC,EAAE,OAAO,uBAAuB,EAAE,iBAAiB,CAAC;IAClE,eAAe,CAAC,EAAE,MAAM,CAAC;IAGzB,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB;AAED,MAAM,WAAW,WAAW;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,MAAM,OAAO,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;AAE/D,OAAO,EAAE,YAAY,EAAE,CAAC"}
|
package/dist/types.js
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
/** Map Vercel AI SDK usage shape → naru TokenUsage. */
|
|
2
|
+
export function normalizeUsage(usage) {
|
|
3
|
+
const p = usage?.inputTokens ?? 0;
|
|
4
|
+
const c = usage?.outputTokens ?? 0;
|
|
5
|
+
return { promptTokens: p, completionTokens: c, totalTokens: usage?.totalTokens ?? p + c };
|
|
6
|
+
}
|
|
7
|
+
//# sourceMappingURL=types.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAWA,uDAAuD;AACvD,MAAM,UAAU,cAAc,CAC5B,KAA6E;IAE7E,MAAM,CAAC,GAAG,KAAK,EAAE,WAAW,IAAI,CAAC,CAAC;IAClC,MAAM,CAAC,GAAG,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC;IACnC,OAAO,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;AAC5F,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"math.d.ts","sourceRoot":"","sources":["../../src/utils/math.ts"],"names":[],"mappings":"AAAA,wBAAgB,gBAAgB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,MAAM,CASjE;AAED,wBAAgB,eAAe,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,MAAM,CAE3D"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
export function cosineSimilarity(a, b) {
|
|
2
|
+
let dot = 0, normA = 0, normB = 0;
|
|
3
|
+
for (let i = 0; i < a.length; i++) {
|
|
4
|
+
dot += a[i] * b[i];
|
|
5
|
+
normA += a[i] * a[i];
|
|
6
|
+
normB += b[i] * b[i];
|
|
7
|
+
}
|
|
8
|
+
const denom = Math.sqrt(normA) * Math.sqrt(normB);
|
|
9
|
+
return denom === 0 ? 0 : dot / denom;
|
|
10
|
+
}
|
|
11
|
+
export function toVectorLiteral(embedding) {
|
|
12
|
+
return `[${embedding.join(",")}]`;
|
|
13
|
+
}
|
|
14
|
+
//# sourceMappingURL=math.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"math.js","sourceRoot":"","sources":["../../src/utils/math.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,gBAAgB,CAAC,CAAW,EAAE,CAAW;IACvD,IAAI,GAAG,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC;IAClC,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,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,KAAK,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,SAAmB;IACjD,OAAO,IAAI,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;AACpC,CAAC"}
|