agent-orcha 0.0.2 → 0.0.4

package/dist/lib/agents/types.d.ts

@@ -1,15 +1,15 @@
 import { z } from 'zod';
 export declare const ToolReferenceSchema: z.ZodUnion<[z.ZodString, z.ZodObject<{
     name: z.ZodString;
-    source: z.ZodEnum<["mcp", "vector", "builtin", "custom"]>;
+    source: z.ZodEnum<["mcp", "knowledge", "builtin", "custom"]>;
     config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
 }, "strip", z.ZodTypeAny, {
     name: string;
-    source: "custom" | "mcp" | "vector" | "builtin";
+    source: "custom" | "mcp" | "knowledge" | "builtin";
     config?: Record<string, unknown> | undefined;
 }, {
     name: string;
-    source: "custom" | "mcp" | "vector" | "builtin";
+    source: "custom" | "mcp" | "knowledge" | "builtin";
     config?: Record<string, unknown> | undefined;
 }>]>;
 export declare const OutputConfigSchema: z.ZodObject<{
@@ -48,15 +48,15 @@ export declare const AgentDefinitionSchema: z.ZodObject<{
     }>;
     tools: z.ZodDefault<z.ZodArray<z.ZodUnion<[z.ZodString, z.ZodObject<{
         name: z.ZodString;
-        source: z.ZodEnum<["mcp", "vector", "builtin", "custom"]>;
+        source: z.ZodEnum<["mcp", "knowledge", "builtin", "custom"]>;
         config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
     }, "strip", z.ZodTypeAny, {
         name: string;
-        source: "custom" | "mcp" | "vector" | "builtin";
+        source: "custom" | "mcp" | "knowledge" | "builtin";
         config?: Record<string, unknown> | undefined;
     }, {
         name: string;
-        source: "custom" | "mcp" | "vector" | "builtin";
+        source: "custom" | "mcp" | "knowledge" | "builtin";
         config?: Record<string, unknown> | undefined;
     }>]>, "many">>;
     output: z.ZodOptional<z.ZodObject<{
@@ -84,7 +84,7 @@ export declare const AgentDefinitionSchema: z.ZodObject<{
     };
     tools: (string | {
         name: string;
-        source: "custom" | "mcp" | "vector" | "builtin";
+        source: "custom" | "mcp" | "knowledge" | "builtin";
         config?: Record<string, unknown> | undefined;
     })[];
     output?: {
@@ -106,7 +106,7 @@ export declare const AgentDefinitionSchema: z.ZodObject<{
     } | undefined;
     tools?: (string | {
         name: string;
-        source: "custom" | "mcp" | "vector" | "builtin";
+        source: "custom" | "mcp" | "knowledge" | "builtin";
         config?: Record<string, unknown> | undefined;
     })[] | undefined;
     output?: {
@@ -130,11 +130,18 @@ export interface AgentResult {
         tokensUsed?: number;
         toolCalls?: ToolCallRecord[];
         duration: number;
+        sessionId?: string;
+        messagesInSession?: number;
+        structuredOutputValid?: boolean;
     };
 }
+export interface AgentInvokeOptions {
+    input: Record<string, unknown>;
+    sessionId?: string;
+}
 export interface AgentInstance {
     definition: AgentDefinition;
-    invoke: (input: Record<string, unknown>) => Promise<AgentResult>;
-    stream: (input: Record<string, unknown>) => AsyncGenerator<string, void, unknown>;
+    invoke: (input: Record<string, unknown> | AgentInvokeOptions) => Promise<AgentResult>;
+    stream: (input: Record<string, unknown> | AgentInvokeOptions) => AsyncGenerator<string | Record<string, unknown>, void, unknown>;
 }
 //# sourceMappingURL=types.d.ts.map

package/dist/lib/agents/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../lib/agents/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,eAAO,MAAM,mBAAmB;;;;;;;;;;;;IAO9B,CAAC;AAEH,eAAO,MAAM,kBAAkB;;;;;;;;;EAG7B,CAAC;AAEH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAYhC,CAAC;AAEH,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAChE,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAC9D,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AAEpE,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,EAAE,OAAO,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,WAAW;IAC1B,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACzC,QAAQ,EAAE;QACR,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,SAAS,CAAC,EAAE,cAAc,EAAE,CAAC;QAC7B,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;CACH;AAED,MAAM,WAAW,aAAa;IAC5B,UAAU,EAAE,eAAe,CAAC;IAC5B,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC,WAAW,CAAC,CAAC;IACjE,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;CACnF"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../lib/agents/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,eAAO,MAAM,mBAAmB;;;;;;;;;;;;IAO9B,CAAC;AAEH,eAAO,MAAM,kBAAkB;;;;;;;;;EAG7B,CAAC;AAEH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAYhC,CAAC;AAEH,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAChE,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAC9D,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AAEpE,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,EAAE,OAAO,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,WAAW;IAC1B,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACzC,QAAQ,EAAE;QACR,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,SAAS,CAAC,EAAE,cAAc,EAAE,CAAC;QAC7B,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAC3B,qBAAqB,CAAC,EAAE,OAAO,CAAC;KACjC,CAAC;CACH;AAED,MAAM,WAAW,kBAAkB;IACjC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC/B,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,aAAa;IAC5B,UAAU,EAAE,eAAe,CAAC;IAC5B,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,kBAAkB,KAAK,OAAO,CAAC,WAAW,CAAC,CAAC;IACtF,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,kBAAkB,KAAK,cAAc,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;CAClI"}

package/dist/lib/agents/types.js

@@ -4,7 +4,7 @@ export const ToolReferenceSchema = z.union([
     z.string(),
     z.object({
         name: z.string(),
-        source: z.enum(['mcp', 'vector', 'builtin', 'custom']),
+        source: z.enum(['mcp', 'knowledge', 'builtin', 'custom']),
         config: z.record(z.unknown()).optional(),
     }),
 ]);

package/dist/lib/agents/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../lib/agents/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAEpD,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC;IACzC,CAAC,CAAC,MAAM,EAAE;IACV,CAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QACtD,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;KACzC,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;IAC9D,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;CACzC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5C,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;IACpD,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC;IAC9D,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;IACpC,GAAG,EAAE,iBAAiB,CAAC,OAAO,CAAC,SAAS,CAAC;IACzC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QACf,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;QAC1D,cAAc,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;KAChD,CAAC;IACF,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;IAC/C,MAAM,EAAE,kBAAkB,CAAC,QAAQ,EAAE;IACrC,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;CAC3C,CAAC,CAAC"}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../lib/agents/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAEpD,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC;IACzC,CAAC,CAAC,MAAM,EAAE;IACV,CAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QACzD,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;KACzC,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;IAC9D,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;CACzC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5C,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;IACpD,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC;IAC9D,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;IACpC,GAAG,EAAE,iBAAiB,CAAC,OAAO,CAAC,SAAS,CAAC;IACzC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QACf,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;QAC1D,cAAc,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;KAChD,CAAC;IACF,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;IAC/C,MAAM,EAAE,kBAAkB,CAAC,QAAQ,EAAE;IACrC,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;CAC3C,CAAC,CAAC"}

package/dist/lib/index.d.ts

@@ -1,16 +1,18 @@
 export { Orchestrator } from './orchestrator.js';
 export type { OrchestratorConfig } from './orchestrator.js';
-export { AgentLoader, AgentExecutor } from './agents/index.js';
-export type { AgentDefinition, AgentInstance, AgentResult, ToolReference, OutputConfig, ToolCallRecord, } from './agents/index.js';
-export { WorkflowLoader, WorkflowExecutor } from './workflows/index.js';
-export type { WorkflowDefinition, WorkflowStep, WorkflowResult, WorkflowContext, StepResult, InputMapping, ParallelSteps, } from './workflows/index.js';
-export { VectorStoreFactory, VectorStoreManager } from './vectors/index.js';
-export type { VectorConfig, VectorStoreInstance, SearchResult, DocumentInput, } from './vectors/index.js';
+export { AgentLoader, AgentExecutor, StructuredOutputWrapper } from './agents/index.js';
+export type { AgentDefinition, AgentInstance, AgentResult, AgentInvokeOptions, ToolReference, OutputConfig, ToolCallRecord, } from './agents/index.js';
+export { WorkflowLoader, WorkflowExecutor, LangGraphExecutor, InterruptManager, } from './workflows/index.js';
+export type { WorkflowDefinition, WorkflowStep, WorkflowResult, WorkflowContext, StepResult, InputMapping, ParallelSteps, LangGraphWorkflowDefinition, StepBasedWorkflowDefinition, GraphToolConfig, GraphAgentConfig, GraphConfig, WorkflowInterrupt, InterruptState, } from './workflows/index.js';
+export { KnowledgeStoreFactory, KnowledgeStoreManager, GraphRagFactory } from './knowledge/index.js';
+export type { KnowledgeConfig, VectorKnowledgeConfig, GraphRagKnowledgeConfig, KnowledgeStoreInstance, SearchResult, DocumentInput, GraphNode, GraphEdge, Community, GraphStore, GraphConfig as GraphRagGraphConfig, GraphSearchConfig, } from './knowledge/index.js';
 export { LLMFactory } from './llm/index.js';
 export type { ModelConfig, AgentLLMRef } from './llm/index.js';
 export { MCPClientManager } from './mcp/index.js';
 export type { MCPConfig, MCPServerConfig } from './mcp/index.js';
-export { ToolRegistry, createVectorSearchTool } from './tools/index.js';
+export { ToolRegistry, ToolDiscovery, AgentToolWrapper, createKnowledgeSearchTool, createAskUserTool, } from './tools/index.js';
 export { FunctionLoader } from './functions/index.js';
 export type { FunctionMetadata, LoadedFunction } from './functions/index.js';
+export { ConversationStore } from './memory/index.js';
+export type { ConversationStoreConfig, ConversationSession } from './memory/index.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/lib/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../lib/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,YAAY,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAG5D,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAC/D,YAAY,EACV,eAAe,EACf,aAAa,EACb,WAAW,EACX,aAAa,EACb,YAAY,EACZ,cAAc,GACf,MAAM,mBAAmB,CAAC;AAG3B,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxE,YAAY,EACV,kBAAkB,EAClB,YAAY,EACZ,cAAc,EACd,eAAe,EACf,UAAU,EACV,YAAY,EACZ,aAAa,GACd,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAC5E,YAAY,EACV,YAAY,EACZ,mBAAmB,EACnB,YAAY,EACZ,aAAa,GACd,MAAM,oBAAoB,CAAC;AAG5B,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,YAAY,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAG/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAGjE,OAAO,EAAE,YAAY,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAGxE,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,YAAY,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../lib/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,YAAY,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAG5D,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AACxF,YAAY,EACV,eAAe,EACf,aAAa,EACb,WAAW,EACX,kBAAkB,EAClB,aAAa,EACb,YAAY,EACZ,cAAc,GACf,MAAM,mBAAmB,CAAC;AAG3B,OAAO,EACL,cAAc,EACd,gBAAgB,EAChB,iBAAiB,EACjB,gBAAgB,GACjB,MAAM,sBAAsB,CAAC;AAC9B,YAAY,EACV,kBAAkB,EAClB,YAAY,EACZ,cAAc,EACd,eAAe,EACf,UAAU,EACV,YAAY,EACZ,aAAa,EACb,2BAA2B,EAC3B,2BAA2B,EAC3B,eAAe,EACf,gBAAgB,EAChB,WAAW,EACX,iBAAiB,EACjB,cAAc,GACf,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EAAE,qBAAqB,EAAE,qBAAqB,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACrG,YAAY,EACV,eAAe,EACf,qBAAqB,EACrB,uBAAuB,EACvB,sBAAsB,EACtB,YAAY,EACZ,aAAa,EACb,SAAS,EACT,SAAS,EACT,SAAS,EACT,UAAU,EACV,WAAW,IAAI,mBAAmB,EAClC,iBAAiB,GAClB,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,YAAY,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAG/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAGjE,OAAO,EACL,YAAY,EACZ,aAAa,EACb,gBAAgB,EAChB,yBAAyB,EACzB,iBAAiB,GAClB,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,YAAY,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAG7E,OAAO,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AACtD,YAAY,EAAE,uBAAuB,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/lib/index.js

@@ -1,17 +1,19 @@
 // Main orchestrator
 export { Orchestrator } from './orchestrator.js';
 // Agents
-export { AgentLoader, AgentExecutor } from './agents/index.js';
+export { AgentLoader, AgentExecutor, StructuredOutputWrapper } from './agents/index.js';
 // Workflows
-export { WorkflowLoader, WorkflowExecutor } from './workflows/index.js';
-// Vectors
-export { VectorStoreFactory, VectorStoreManager } from './vectors/index.js';
+export { WorkflowLoader, WorkflowExecutor, LangGraphExecutor, InterruptManager, } from './workflows/index.js';
+// Knowledge
+export { KnowledgeStoreFactory, KnowledgeStoreManager, GraphRagFactory } from './knowledge/index.js';
 // LLM
 export { LLMFactory } from './llm/index.js';
 // MCP
 export { MCPClientManager } from './mcp/index.js';
 // Tools
-export { ToolRegistry, createVectorSearchTool } from './tools/index.js';
+export { ToolRegistry, ToolDiscovery, AgentToolWrapper, createKnowledgeSearchTool, createAskUserTool, } from './tools/index.js';
 // Functions
 export { FunctionLoader } from './functions/index.js';
+// Memory
+export { ConversationStore } from './memory/index.js';
 //# sourceMappingURL=index.js.map

package/dist/lib/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../lib/index.ts"],"names":[],"mappings":"AAAA,oBAAoB;AACpB,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAGjD,SAAS;AACT,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAU/D,YAAY;AACZ,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAWxE,UAAU;AACV,OAAO,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAQ5E,MAAM;AACN,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAG5C,MAAM;AACN,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAGlD,QAAQ;AACR,OAAO,EAAE,YAAY,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAExE,YAAY;AACZ,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../lib/index.ts"],"names":[],"mappings":"AAAA,oBAAoB;AACpB,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAGjD,SAAS;AACT,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AAWxF,YAAY;AACZ,OAAO,EACL,cAAc,EACd,gBAAgB,EAChB,iBAAiB,EACjB,gBAAgB,GACjB,MAAM,sBAAsB,CAAC;AAkB9B,YAAY;AACZ,OAAO,EAAE,qBAAqB,EAAE,qBAAqB,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAgBrG,MAAM;AACN,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAG5C,MAAM;AACN,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAGlD,QAAQ;AACR,OAAO,EACL,YAAY,EACZ,aAAa,EACb,gBAAgB,EAChB,yBAAyB,EACzB,iBAAiB,GAClB,MAAM,kBAAkB,CAAC;AAE1B,YAAY;AACZ,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,SAAS;AACT,OAAO,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/lib/knowledge/graph-rag/community-detector.d.ts

@@ -0,0 +1,16 @@
+import type { MemoryGraphStore } from './memory-graph-store.js';
+import type { Community, GraphCommunitiesConfig } from './types.js';
+/**
+ * Louvain community detection on the in-memory graph.
+ * Groups nodes into communities based on graph structure (edge density).
+ */
+export declare class CommunityDetector {
+    private config;
+    constructor(config: GraphCommunitiesConfig);
+    /**
+     * Detect communities in the graph store.
+     * Only works with MemoryGraphStore (requires graphology instance).
+     */
+    detect(store: MemoryGraphStore): Promise<Community[]>;
+}
+//# sourceMappingURL=community-detector.d.ts.map

package/dist/lib/knowledge/graph-rag/community-detector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"community-detector.d.ts","sourceRoot":"","sources":["../../../../lib/knowledge/graph-rag/community-detector.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,KAAK,EAAE,SAAS,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAKpE;;;GAGG;AACH,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,MAAM,CAAyB;gBAE3B,MAAM,EAAE,sBAAsB;IAI1C;;;OAGG;IACG,MAAM,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;CA4E5D"}

package/dist/lib/knowledge/graph-rag/community-detector.js

@@ -0,0 +1,81 @@
+import { createLogger } from '../../logger.js';
+const logger = createLogger('CommunityDetector');
+/**
+ * Louvain community detection on the in-memory graph.
+ * Groups nodes into communities based on graph structure (edge density).
+ */
+export class CommunityDetector {
+    config;
+    constructor(config) {
+        this.config = config;
+    }
+    /**
+     * Detect communities in the graph store.
+     * Only works with MemoryGraphStore (requires graphology instance).
+     */
+    async detect(store) {
+        const graph = await store.getGraphologyInstance();
+        if (graph.order === 0) {
+            logger.warn('Graph is empty, no communities to detect');
+            return [];
+        }
+        // Dynamic imports to handle ESM/CJS interop
+        const graphology = await import('graphology');
+        const GraphClass = graphology.default ?? graphology;
+        const louvainModule = await import('graphology-communities-louvain');
+        const louvain = louvainModule.default ?? louvainModule;
+        // graphology-communities-louvain requires an undirected graph for community detection.
+        // Create a temporary undirected copy.
+        const undirected = new GraphClass({ type: 'undirected' });
+        graph.forEachNode((nodeId, attrs) => {
+            undirected.addNode(nodeId, attrs);
+        });
+        graph.forEachEdge((_edgeKey, attrs, source, target) => {
+            if (source === target)
+                return; // skip self-loops
+            if (!undirected.hasEdge(source, target)) {
+                undirected.addEdge(source, target, { weight: attrs.weight ?? 1 });
+            }
+            else {
+                // Merge edge weights
+                const existing = undirected.getEdgeAttributes(source, target);
+                undirected.setEdgeAttribute(source, target, 'weight', existing.weight + (attrs.weight ?? 1));
+            }
+        });
+        if (undirected.size === 0) {
+            logger.warn('No edges in graph, each node becomes its own community');
+            const communities = [];
+            let i = 0;
+            graph.forEachNode((nodeId) => {
+                communities.push({ id: `community-${i}`, nodeIds: [nodeId] });
+                i++;
+            });
+            return communities.filter((c) => c.nodeIds.length >= this.config.minSize);
+        }
+        logger.info(`Running Louvain on ${undirected.order} nodes, ${undirected.size} edges (resolution: ${this.config.resolution})`);
+        // Run Louvain algorithm
+        const assignments = louvain(undirected, {
+            resolution: this.config.resolution,
+        });
+        // Group nodes by community
+        const communityMap = new Map();
+        for (const [nodeId, communityId] of Object.entries(assignments)) {
+            const existing = communityMap.get(communityId) ?? [];
+            existing.push(nodeId);
+            communityMap.set(communityId, existing);
+        }
+        // Filter by minimum size and create Community objects
+        const communities = [];
+        for (const [id, nodeIds] of communityMap.entries()) {
+            if (nodeIds.length >= this.config.minSize) {
+                communities.push({
+                    id: `community-${id}`,
+                    nodeIds,
+                });
+            }
+        }
+        logger.info(`Detected ${communities.length} communities (min size: ${this.config.minSize})`);
+        return communities;
+    }
+}
+//# sourceMappingURL=community-detector.js.map

package/dist/lib/knowledge/graph-rag/community-detector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"community-detector.js","sourceRoot":"","sources":["../../../../lib/knowledge/graph-rag/community-detector.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,MAAM,GAAG,YAAY,CAAC,mBAAmB,CAAC,CAAC;AAEjD;;;GAGG;AACH,MAAM,OAAO,iBAAiB;IACpB,MAAM,CAAyB;IAEvC,YAAY,MAA8B;QACxC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,MAAM,CAAC,KAAuB;QAClC,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,qBAAqB,EAAE,CAAC;QAElD,IAAI,KAAK,CAAC,KAAK,KAAK,CAAC,EAAE,CAAC;YACtB,MAAM,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;YACxD,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,4CAA4C;QAC5C,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,CAAC;QAC9C,MAAM,UAAU,GAAG,UAAU,CAAC,OAAO,IAAI,UAAU,CAAC;QACpD,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,gCAAgC,CAAC,CAAC;QACrE,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,IAAI,aAAa,CAAC;QAEvD,uFAAuF;QACvF,sCAAsC;QACtC,MAAM,UAAU,GAAkB,IAAK,UAAkB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;QAElF,KAAK,CAAC,WAAW,CAAC,CAAC,MAAc,EAAE,KAAiB,EAAE,EAAE;YACtD,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,KAAK,CAAC,WAAW,CAAC,CAAC,QAAgB,EAAE,KAAiB,EAAE,MAAc,EAAE,MAAc,EAAE,EAAE;YACxF,IAAI,MAAM,KAAK,MAAM;gBAAE,OAAO,CAAC,kBAAkB;YACjD,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC;gBACxC,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,MAAM,EAAG,KAAiC,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC;YACjG,CAAC;iBAAM,CAAC;gBACN,qBAAqB;gBACrB,MAAM,QAAQ,GAAG,UAAU,CAAC,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAuB,CAAC;gBACpF,UAAU,CAAC,gBAAgB,CACzB,MAAM,EAAE,MAAM,EAAE,QAAQ,EACxB,QAAQ,CAAC,MAAM,GAAG,CAAG,KAAiC,CAAC,MAAiB,IAAI,CAAC,CAAC,CAC/E,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,UAAU,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;YACtE,MAAM,WAAW,GAAgB,EAAE,CAAC;YACpC,IAAI,CAAC,GAAG,CAAC,CAAC;YACV,KAAK,CAAC,WAAW,CAAC,CAAC,MAAc,EAAE,EAAE;gBACnC,WAAW,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,aAAa,CAAC,EAAE,EAAE,OAAO,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAC9D,CAAC,EAAE,CAAC;YACN,CAAC,CAAC,CAAC;YACH,OAAO,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC5E,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,sBAAsB,UAAU,CAAC,KAAK,WAAW,UAAU,CAAC,IAAI,uBAAuB,IAAI,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC;QAE9H,wBAAwB;QACxB,MAAM,WAAW,GAAI,OAAe,CAAC,UAAU,EAAE;YAC/C,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU;SACnC,CAA2B,CAAC;QAE7B,2BAA2B;QAC3B,MAAM,YAAY,GAAG,IAAI,GAAG,EAAoB,CAAC;QACjD,KAAK,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;YAChE,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC;YACrD,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtB,YAAY,CAAC,GAAG,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QAC1C,CAAC;QAED,sDAAsD;QACtD,MAAM,WAAW,GAAgB,EAAE,CAAC;QACpC,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;YACnD,IAAI,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAC1C,WAAW,CAAC,IAAI,CAAC;oBACf,EAAE,EAAE,aAAa,EAAE,EAAE;oBACrB,OAAO;iBACR,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,YAAY,WAAW,CAAC,MAAM,2BAA2B,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,CAAC,CAAC;QAC7F,OAAO,WAAW,CAAC;IACrB,CAAC;CACF"}

package/dist/lib/knowledge/graph-rag/community-summarizer.d.ts

@@ -0,0 +1,17 @@
+import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import type { Community, GraphStore } from './types.js';
+/**
+ * Generates natural language summaries for graph communities using an LLM.
+ * Each community summary describes the key entities and relationships within it.
+ */
+export declare class CommunitySummarizer {
+    private llm;
+    constructor(llm: BaseChatModel);
+    /**
+     * Generate summaries for all communities.
+     */
+    summarize(communities: Community[], store: GraphStore): Promise<Community[]>;
+    private summarizeSingle;
+    private buildCommunityContext;
+}
+//# sourceMappingURL=community-summarizer.d.ts.map

package/dist/lib/knowledge/graph-rag/community-summarizer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"community-summarizer.d.ts","sourceRoot":"","sources":["../../../../lib/knowledge/graph-rag/community-summarizer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6CAA6C,CAAC;AAEjF,OAAO,KAAK,EAAE,SAAS,EAAwB,UAAU,EAAE,MAAM,YAAY,CAAC;AAK9E;;;GAGG;AACH,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,GAAG,CAAgB;gBAEf,GAAG,EAAE,aAAa;IAI9B;;OAEG;IACG,SAAS,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YAuBpE,eAAe;IAqD7B,OAAO,CAAC,qBAAqB;CAQ9B"}

package/dist/lib/knowledge/graph-rag/community-summarizer.js

@@ -0,0 +1,87 @@
+import { HumanMessage, SystemMessage } from '@langchain/core/messages';
+import { createLogger } from '../../logger.js';
+const logger = createLogger('CommunitySummarizer');
+/**
+ * Generates natural language summaries for graph communities using an LLM.
+ * Each community summary describes the key entities and relationships within it.
+ */
+export class CommunitySummarizer {
+    llm;
+    constructor(llm) {
+        this.llm = llm;
+    }
+    /**
+     * Generate summaries for all communities.
+     */
+    async summarize(communities, store) {
+        const summarized = [];
+        for (let i = 0; i < communities.length; i++) {
+            const community = communities[i];
+            logger.info(`Summarizing community ${i + 1}/${communities.length} (${community.nodeIds.length} nodes)`);
+            try {
+                const { title, summary } = await this.summarizeSingle(community, store);
+                summarized.push({ ...community, title, summary });
+            }
+            catch (error) {
+                logger.error(`Failed to summarize community ${community.id}: ${error instanceof Error ? error.message : String(error)}`);
+                summarized.push({
+                    ...community,
+                    title: `Community ${community.id}`,
+                    summary: `Community with ${community.nodeIds.length} entities.`,
+                });
+            }
+        }
+        return summarized;
+    }
+    async summarizeSingle(community, store) {
+        // Gather all nodes and their relationships within the community
+        const nodes = [];
+        for (const nodeId of community.nodeIds) {
+            const node = await store.getNode(nodeId);
+            if (node)
+                nodes.push(node);
+        }
+        const allEdges = await store.getAllEdges();
+        const communityNodeIds = new Set(community.nodeIds);
+        const internalEdges = allEdges.filter((e) => communityNodeIds.has(e.sourceId) && communityNodeIds.has(e.targetId));
+        const context = this.buildCommunityContext(nodes, internalEdges);
+        const response = await this.llm.invoke([
+            new SystemMessage(`You are a knowledge graph analyst. Given a set of entities and relationships from a community in a knowledge graph, provide:
+1. A short title (5-10 words) that captures the main theme
+2. A comprehensive summary (2-4 sentences) that describes the key entities, their relationships, and the overall theme
+
+Respond ONLY with valid JSON:
+{
+  "title": "Short descriptive title",
+  "summary": "Comprehensive summary of the community..."
+}`),
+            new HumanMessage(`Analyze this community:\n\n${context}`),
+        ]);
+        const responseText = typeof response.content === 'string'
+            ? response.content
+            : Array.isArray(response.content)
+                ? response.content.map((c) => (typeof c === 'string' ? c : 'text' in c ? c.text : '')).join('')
+                : '';
+        try {
+            const jsonMatch = responseText.match(/```(?:json)?\s*([\s\S]*?)```/) ?? [null, responseText];
+            const parsed = JSON.parse((jsonMatch[1] ?? responseText).trim());
+            return {
+                title: String(parsed.title ?? `Community ${community.id}`),
+                summary: String(parsed.summary ?? ''),
+            };
+        }
+        catch {
+            logger.warn(`Failed to parse community summary, using raw response`);
+            return {
+                title: `Community ${community.id}`,
+                summary: responseText.substring(0, 500),
+            };
+        }
+    }
+    buildCommunityContext(nodes, edges) {
+        const entityLines = nodes.map((n) => `- ${n.name} (${n.type}): ${n.description}`).join('\n');
+        const relLines = edges.map((e) => `- ${e.sourceId} -[${e.type}]-> ${e.targetId}: ${e.description}`).join('\n');
+        return `Entities:\n${entityLines}\n\nRelationships:\n${relLines}`;
+    }
+}
+//# sourceMappingURL=community-summarizer.js.map

package/dist/lib/knowledge/graph-rag/community-summarizer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"community-summarizer.js","sourceRoot":"","sources":["../../../../lib/knowledge/graph-rag/community-summarizer.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAEvE,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,MAAM,GAAG,YAAY,CAAC,qBAAqB,CAAC,CAAC;AAEnD;;;GAGG;AACH,MAAM,OAAO,mBAAmB;IACtB,GAAG,CAAgB;IAE3B,YAAY,GAAkB;QAC5B,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,WAAwB,EAAE,KAAiB;QACzD,MAAM,UAAU,GAAgB,EAAE,CAAC;QAEnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,MAAM,SAAS,GAAG,WAAW,CAAC,CAAC,CAAE,CAAC;YAClC,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,GAAG,CAAC,IAAI,WAAW,CAAC,MAAM,KAAK,SAAS,CAAC,OAAO,CAAC,MAAM,SAAS,CAAC,CAAC;YAExG,IAAI,CAAC;gBACH,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;gBACxE,UAAU,CAAC,IAAI,CAAC,EAAE,GAAG,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;YACpD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,KAAK,CAAC,iCAAiC,SAAS,CAAC,EAAE,KAAK,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBACzH,UAAU,CAAC,IAAI,CAAC;oBACd,GAAG,SAAS;oBACZ,KAAK,EAAE,aAAa,SAAS,CAAC,EAAE,EAAE;oBAClC,OAAO,EAAE,kBAAkB,SAAS,CAAC,OAAO,CAAC,MAAM,YAAY;iBAChE,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,SAAoB,EAAE,KAAiB;QACnE,gEAAgE;QAChE,MAAM,KAAK,GAAgB,EAAE,CAAC;QAC9B,KAAK,MAAM,MAAM,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;YACvC,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACzC,IAAI,IAAI;gBAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7B,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;QAC3C,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACpD,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CACnC,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAC5E,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;QAEjE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC;YACrC,IAAI,aAAa,CACf;;;;;;;;EAQN,CACK;YACD,IAAI,YAAY,CAAC,8BAA8B,OAAO,EAAE,CAAC;SAC1D,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,OAAO,QAAQ,CAAC,OAAO,KAAK,QAAQ;YACvD,CAAC,CAAC,QAAQ,CAAC,OAAO;YAClB,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAC/B,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/F,CAAC,CAAC,EAAE,CAAC;QAET,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,8BAA8B,CAAC,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YAC7F,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YACjE,OAAO;gBACL,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,KAAK,IAAI,aAAa,SAAS,CAAC,EAAE,EAAE,CAAC;gBAC1D,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;aACtC,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,CAAC,IAAI,CAAC,uDAAuD,CAAC,CAAC;YACrE,OAAO;gBACL,KAAK,EAAE,aAAa,SAAS,CAAC,EAAE,EAAE;gBAClC,OAAO,EAAE,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;aACxC,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,qBAAqB,CAAC,KAAkB,EAAE,KAAkB;QAClE,MAAM,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7F,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CACxB,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,QAAQ,MAAM,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,WAAW,EAAE,CACxE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEb,OAAO,cAAc,WAAW,uBAAuB,QAAQ,EAAE,CAAC;IACpE,CAAC;CACF"}

package/dist/lib/knowledge/graph-rag/entity-extractor.d.ts

@@ -0,0 +1,36 @@
+import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import type { EntityTypeConfig, RelationshipTypeConfig, ExtractionResult } from './types.js';
+interface EntityExtractorOptions {
+    llm: BaseChatModel;
+    entityTypes?: EntityTypeConfig[];
+    relationshipTypes?: RelationshipTypeConfig[];
+}
+/**
+ * LLM-based entity and relationship extraction from text chunks.
+ */
+export declare class EntityExtractor {
+    private llm;
+    private entityTypes?;
+    private relationshipTypes?;
+    constructor(options: EntityExtractorOptions);
+    /**
+     * Extract entities and relationships from a set of text chunks.
+     * Processes chunks sequentially to avoid rate limiting.
+     */
+    extractFromChunks(chunks: Array<{
+        id: string;
+        content: string;
+    }>): Promise<ExtractionResult>;
+    private extractFromSingleChunk;
+    private buildSystemPrompt;
+    private buildUserPrompt;
+    private parseExtractionResponse;
+    /**
+     * Deduplicate entities by normalized key (lowercase name + type).
+     * Merges descriptions and source chunk IDs.
+     * Updates relationship references to use canonical entity names.
+     */
+    private deduplicateEntities;
+}
+export {};
+//# sourceMappingURL=entity-extractor.d.ts.map

package/dist/lib/knowledge/graph-rag/entity-extractor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"entity-extractor.d.ts","sourceRoot":"","sources":["../../../../lib/knowledge/graph-rag/entity-extractor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6CAA6C,CAAC;AAEjF,OAAO,KAAK,EAAE,gBAAgB,EAAE,sBAAsB,EAA0C,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAKrI,UAAU,sBAAsB;IAC9B,GAAG,EAAE,aAAa,CAAC;IACnB,WAAW,CAAC,EAAE,gBAAgB,EAAE,CAAC;IACjC,iBAAiB,CAAC,EAAE,sBAAsB,EAAE,CAAC;CAC9C;AAED;;GAEG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,GAAG,CAAgB;IAC3B,OAAO,CAAC,WAAW,CAAC,CAAqB;IACzC,OAAO,CAAC,iBAAiB,CAAC,CAA2B;gBAEzC,OAAO,EAAE,sBAAsB;IAM3C;;;OAGG;IACG,iBAAiB,CAAC,MAAM,EAAE,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC;YA0BpF,sBAAsB;IAkBpC,OAAO,CAAC,iBAAiB;IA2CzB,OAAO,CAAC,eAAe;IAIvB,OAAO,CAAC,uBAAuB;IAgC/B;;;;OAIG;IACH,OAAO,CAAC,mBAAmB;CAmE5B"}

package/dist/lib/knowledge/graph-rag/entity-extractor.js

@@ -0,0 +1,192 @@
+import { HumanMessage, SystemMessage } from '@langchain/core/messages';
+import { createLogger } from '../../logger.js';
+const logger = createLogger('EntityExtractor');
+/**
+ * LLM-based entity and relationship extraction from text chunks.
+ */
+export class EntityExtractor {
+    llm;
+    entityTypes;
+    relationshipTypes;
+    constructor(options) {
+        this.llm = options.llm;
+        this.entityTypes = options.entityTypes;
+        this.relationshipTypes = options.relationshipTypes;
+    }
+    /**
+     * Extract entities and relationships from a set of text chunks.
+     * Processes chunks sequentially to avoid rate limiting.
+     */
+    async extractFromChunks(chunks) {
+        const allEntities = [];
+        const allRelationships = [];
+        for (let i = 0; i < chunks.length; i++) {
+            const chunk = chunks[i];
+            logger.info(`Extracting from chunk ${i + 1}/${chunks.length} (${chunk.content.length} chars)`);
+            try {
+                const result = await this.extractFromSingleChunk(chunk.id, chunk.content);
+                allEntities.push(...result.entities);
+                allRelationships.push(...result.relationships);
+            }
+            catch (error) {
+                logger.error(`Failed to extract from chunk ${i + 1}: ${error instanceof Error ? error.message : String(error)}`);
+            }
+        }
+        logger.info(`Raw extraction: ${allEntities.length} entities, ${allRelationships.length} relationships`);
+        // Deduplicate entities by normalized name + type
+        const deduplicated = this.deduplicateEntities(allEntities, allRelationships);
+        logger.info(`After dedup: ${deduplicated.entities.length} entities, ${deduplicated.relationships.length} relationships`);
+        return deduplicated;
+    }
+    async extractFromSingleChunk(chunkId, content) {
+        const systemPrompt = this.buildSystemPrompt();
+        const userPrompt = this.buildUserPrompt(content);
+        const response = await this.llm.invoke([
+            new SystemMessage(systemPrompt),
+            new HumanMessage(userPrompt),
+        ]);
+        const responseText = typeof response.content === 'string'
+            ? response.content
+            : Array.isArray(response.content)
+                ? response.content.map((c) => (typeof c === 'string' ? c : 'text' in c ? c.text : '')).join('')
+                : '';
+        return this.parseExtractionResponse(responseText, chunkId);
+    }
+    buildSystemPrompt() {
+        const entityTypesSection = this.entityTypes && this.entityTypes.length > 0
+            ? `\n\nEntity types to extract:\n${this.entityTypes.map((t) => `- ${t.name}: ${t.description}`).join('\n')}`
+            : '\n\nExtract all notable entities (people, organizations, concepts, locations, events, objects, etc.).';
+        const relationshipTypesSection = this.relationshipTypes && this.relationshipTypes.length > 0
+            ? `\n\nRelationship types to extract:\n${this.relationshipTypes.map((t) => `- ${t.name}: ${t.description}`).join('\n')}`
+            : '\n\nExtract all meaningful relationships between entities.';
+        return `You are an entity and relationship extraction system. Given a text document, extract structured entities and relationships.
+${entityTypesSection}
+${relationshipTypesSection}
+
+Respond ONLY with valid JSON in this exact format:
+{
+  "entities": [
+    {
+      "name": "Entity Name",
+      "type": "EntityType",
+      "description": "Brief description of this entity based on the text"
+    }
+  ],
+  "relationships": [
+    {
+      "sourceName": "Source Entity Name",
+      "sourceType": "SourceEntityType",
+      "targetName": "Target Entity Name",
+      "targetType": "TargetEntityType",
+      "type": "RELATIONSHIP_TYPE",
+      "description": "Brief description of this relationship",
+      "weight": 1.0
+    }
+  ]
+}
+
+Rules:
+- Entity names should be normalized (proper case, no extra whitespace)
+- Use consistent naming for the same entity across extractions
+- Relationship weight should be 0.0 to 1.0 (1.0 = very strong relationship)
+- Every entity in a relationship must also appear in the entities array
+- Be thorough but precise - extract real entities, not generic concepts`;
+    }
+    buildUserPrompt(content) {
+        return `Extract entities and relationships from the following text:\n\n---\n${content}\n---`;
+    }
+    parseExtractionResponse(responseText, chunkId) {
+        try {
+            // Extract JSON from response (handle markdown code blocks)
+            const jsonMatch = responseText.match(/```(?:json)?\s*([\s\S]*?)```/) ?? [null, responseText];
+            const jsonStr = (jsonMatch[1] ?? responseText).trim();
+            const parsed = JSON.parse(jsonStr);
+            const entities = (parsed.entities ?? []).map((e) => ({
+                name: String(e.name ?? '').trim(),
+                type: String(e.type ?? 'Unknown').trim(),
+                description: String(e.description ?? '').trim(),
+                properties: { sourceChunkId: chunkId },
+            })).filter((e) => e.name.length > 0);
+            const relationships = (parsed.relationships ?? []).map((r) => ({
+                sourceName: String(r.sourceName ?? '').trim(),
+                sourceType: String(r.sourceType ?? 'Unknown').trim(),
+                targetName: String(r.targetName ?? '').trim(),
+                targetType: String(r.targetType ?? 'Unknown').trim(),
+                type: String(r.type ?? 'RELATES_TO').trim(),
+                description: String(r.description ?? '').trim(),
+                weight: typeof r.weight === 'number' ? Math.max(0, Math.min(1, r.weight)) : 1.0,
+            })).filter((r) => r.sourceName.length > 0 && r.targetName.length > 0);
+            return { entities, relationships };
+        }
+        catch (error) {
+            logger.warn(`Failed to parse extraction response for chunk ${chunkId}: ${error instanceof Error ? error.message : String(error)}`);
+            logger.debug(`Response was: ${responseText.substring(0, 200)}...`);
+            return { entities: [], relationships: [] };
+        }
+    }
+    /**
+     * Deduplicate entities by normalized key (lowercase name + type).
+     * Merges descriptions and source chunk IDs.
+     * Updates relationship references to use canonical entity names.
+     */
+    deduplicateEntities(entities, relationships) {
+        const entityMap = new Map();
+        const nameNormMap = new Map(); // normalizedKey -> canonical name
+        for (const entity of entities) {
+            const key = `${entity.name.toLowerCase()}::${entity.type.toLowerCase()}`;
+            const existing = entityMap.get(key);
+            if (existing) {
+                // Merge: keep longer description, merge source chunks
+                if (entity.description.length > existing.description.length) {
+                    existing.description = entity.description;
+                }
+                const existingChunks = existing.properties.sourceChunkIds ?? [];
+                const newChunkId = entity.properties.sourceChunkId;
+                if (newChunkId && !existingChunks.includes(newChunkId)) {
+                    existingChunks.push(newChunkId);
+                }
+                existing.properties.sourceChunkIds = existingChunks;
+            }
+            else {
+                const chunkId = entity.properties.sourceChunkId;
+                entityMap.set(key, {
+                    ...entity,
+                    properties: {
+                        sourceChunkIds: chunkId ? [chunkId] : [],
+                    },
+                });
+                nameNormMap.set(key, entity.name);
+            }
+        }
+        // Deduplicate relationships
+        const relMap = new Map();
+        for (const rel of relationships) {
+            const sourceKey = `${rel.sourceName.toLowerCase()}::${rel.sourceType.toLowerCase()}`;
+            const targetKey = `${rel.targetName.toLowerCase()}::${rel.targetType.toLowerCase()}`;
+            // Update names to canonical versions
+            const canonicalSource = nameNormMap.get(sourceKey) ?? rel.sourceName;
+            const canonicalTarget = nameNormMap.get(targetKey) ?? rel.targetName;
+            const relKey = `${sourceKey}->${rel.type.toLowerCase()}->${targetKey}`;
+            const existing = relMap.get(relKey);
+            if (existing) {
+                // Merge: average weights, keep longer description
+                existing.weight = (existing.weight + rel.weight) / 2;
+                if (rel.description.length > existing.description.length) {
+                    existing.description = rel.description;
+                }
+            }
+            else {
+                relMap.set(relKey, {
+                    ...rel,
+                    sourceName: canonicalSource,
+                    targetName: canonicalTarget,
+                });
+            }
+        }
+        return {
+            entities: Array.from(entityMap.values()),
+            relationships: Array.from(relMap.values()),
+        };
+    }
+}
+//# sourceMappingURL=entity-extractor.js.map