@xpert-ai/plugin-sdk 3.6.0 → 3.6.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.
Files changed (157) hide show
  1. package/index.cjs.d.ts +1 -0
  2. package/index.cjs.js +45273 -5
  3. package/index.esm.d.ts +1 -0
  4. package/index.esm.js +45192 -0
  5. package/package.json +3 -3
  6. package/src/index.d.ts +12 -0
  7. package/src/lib/ai-model/abstract-provider.d.ts +41 -0
  8. package/src/lib/ai-model/ai-model-provider.decorator.d.ts +2 -0
  9. package/src/lib/ai-model/ai-model-provider.interface.d.ts +42 -0
  10. package/src/lib/ai-model/ai-model-provider.registry.d.ts +6 -0
  11. package/src/lib/ai-model/ai-model.d.ts +33 -0
  12. package/src/lib/ai-model/entities/defaults.d.ts +3 -0
  13. package/src/lib/ai-model/entities/index.d.ts +2 -0
  14. package/src/lib/ai-model/entities/model_entities.d.ts +14 -0
  15. package/src/lib/ai-model/errors.d.ts +5 -0
  16. package/src/lib/ai-model/index.d.ts +8 -0
  17. package/src/lib/ai-model/llm.d.ts +49 -0
  18. package/src/lib/ai-model/openai-compatible/completions.d.ts +23 -0
  19. package/src/lib/ai-model/openai-compatible/index.d.ts +3 -0
  20. package/src/lib/ai-model/openai-compatible/rerank.d.ts +20 -0
  21. package/src/lib/ai-model/openai-compatible/speech2text.d.ts +17 -0
  22. package/src/lib/ai-model/types/index.d.ts +11 -0
  23. package/src/lib/ai-model/types/model.d.ts +43 -0
  24. package/src/lib/ai-model/types/rerank.d.ts +20 -0
  25. package/src/lib/ai-model/types/speech2text.d.ts +3 -0
  26. package/src/lib/ai-model/types/text-embedding-model.d.ts +7 -0
  27. package/src/lib/ai-model/types/tts.d.ts +5 -0
  28. package/src/lib/core/file-system.d.ts +55 -0
  29. package/src/lib/core/i18n.d.ts +2 -0
  30. package/src/lib/core/index.d.ts +5 -0
  31. package/src/lib/core/permissions.d.ts +74 -0
  32. package/src/lib/core/schema.d.ts +22 -0
  33. package/src/lib/core/utils.d.ts +12 -0
  34. package/src/lib/integration/index.d.ts +3 -0
  35. package/src/lib/integration/strategy.decorator.d.ts +2 -0
  36. package/src/lib/integration/strategy.interface.d.ts +9 -0
  37. package/src/lib/integration/strategy.registry.d.ts +6 -0
  38. package/src/lib/logger.d.ts +2 -0
  39. package/src/lib/plugin-metadata.d.ts +17 -0
  40. package/src/lib/plugin.d.ts +8 -0
  41. package/src/lib/{plugin.hook.ts → plugin.hook.d.ts} +0 -4
  42. package/src/lib/plugin.interface.d.ts +59 -0
  43. package/src/lib/rag/image/index.d.ts +3 -0
  44. package/src/lib/rag/image/strategy.decorator.d.ts +5 -0
  45. package/src/lib/rag/image/strategy.interface.d.ts +31 -0
  46. package/src/lib/rag/image/strategy.registry.d.ts +6 -0
  47. package/src/lib/rag/index.d.ts +7 -0
  48. package/src/lib/rag/knowledge/index.d.ts +3 -0
  49. package/src/lib/rag/knowledge/knowledge-strategy.decorator.d.ts +2 -0
  50. package/src/lib/rag/knowledge/knowledge-strategy.interface.d.ts +15 -0
  51. package/src/lib/rag/knowledge/knowledge-strategy.registry.d.ts +6 -0
  52. package/src/lib/rag/retriever/index.d.ts +3 -0
  53. package/src/lib/rag/retriever/strategy.decorator.d.ts +5 -0
  54. package/src/lib/rag/retriever/strategy.interface.d.ts +30 -0
  55. package/src/lib/rag/retriever/strategy.registry.d.ts +6 -0
  56. package/src/lib/rag/source/index.d.ts +3 -0
  57. package/src/lib/rag/source/strategy.decorator.d.ts +5 -0
  58. package/src/lib/rag/source/strategy.interface.d.ts +36 -0
  59. package/src/lib/rag/source/strategy.registry.d.ts +6 -0
  60. package/src/lib/rag/textsplitter/index.d.ts +3 -0
  61. package/src/lib/rag/textsplitter/strategy.decorator.d.ts +2 -0
  62. package/src/lib/rag/textsplitter/strategy.interface.d.ts +23 -0
  63. package/src/lib/rag/textsplitter/strategy.registry.d.ts +6 -0
  64. package/src/lib/rag/transformer/index.d.ts +3 -0
  65. package/src/lib/rag/transformer/strategy.decorator.d.ts +5 -0
  66. package/src/lib/rag/transformer/strategy.interface.d.ts +26 -0
  67. package/src/lib/rag/transformer/strategy.registry.d.ts +6 -0
  68. package/src/lib/rag/types.d.ts +18 -0
  69. package/src/lib/strategy.d.ts +12 -0
  70. package/src/lib/toolset/builtin.d.ts +54 -0
  71. package/src/lib/toolset/index.d.ts +5 -0
  72. package/src/lib/toolset/strategy.decorator.d.ts +5 -0
  73. package/src/lib/toolset/strategy.interface.d.ts +28 -0
  74. package/src/lib/toolset/strategy.registry.d.ts +6 -0
  75. package/src/lib/toolset/toolset.d.ts +53 -0
  76. package/src/lib/types.d.ts +47 -0
  77. package/src/lib/vectorstore/index.d.ts +3 -0
  78. package/src/lib/vectorstore/strategy.decorator.d.ts +2 -0
  79. package/src/lib/vectorstore/strategy.interface.d.ts +24 -0
  80. package/src/lib/vectorstore/strategy.registry.d.ts +6 -0
  81. package/src/lib/workflow/index.d.ts +2 -0
  82. package/src/lib/workflow/node/index.d.ts +3 -0
  83. package/src/lib/workflow/node/strategy.decorator.d.ts +5 -0
  84. package/src/lib/workflow/node/strategy.interface.d.ts +41 -0
  85. package/src/lib/workflow/node/strategy.registry.d.ts +9 -0
  86. package/src/lib/workflow/trigger/index.d.ts +3 -0
  87. package/src/lib/workflow/trigger/strategy.decorator.d.ts +2 -0
  88. package/src/lib/workflow/trigger/strategy.interface.d.ts +22 -0
  89. package/src/lib/workflow/trigger/strategy.registry.d.ts +6 -0
  90. package/.eslintrc.json +0 -30
  91. package/.swcrc +0 -29
  92. package/jest.config.ts +0 -28
  93. package/project.json +0 -59
  94. package/src/index.ts +0 -11
  95. package/src/lib/core/file-system.ts +0 -113
  96. package/src/lib/core/i18n.ts +0 -31
  97. package/src/lib/core/index.ts +0 -4
  98. package/src/lib/core/permissions.ts +0 -97
  99. package/src/lib/core/schema.ts +0 -23
  100. package/src/lib/integration/index.ts +0 -3
  101. package/src/lib/integration/strategy.decorator.ts +0 -6
  102. package/src/lib/integration/strategy.interface.ts +0 -11
  103. package/src/lib/integration/strategy.registry.ts +0 -12
  104. package/src/lib/logger.ts +0 -18
  105. package/src/lib/plugin-metadata.ts +0 -20
  106. package/src/lib/plugin.interface.ts +0 -67
  107. package/src/lib/plugin.ts +0 -32
  108. package/src/lib/rag/image/index.ts +0 -3
  109. package/src/lib/rag/image/strategy.decorator.ts +0 -9
  110. package/src/lib/rag/image/strategy.interface.ts +0 -37
  111. package/src/lib/rag/image/strategy.registry.ts +0 -17
  112. package/src/lib/rag/index.ts +0 -7
  113. package/src/lib/rag/knowledge/index.ts +0 -3
  114. package/src/lib/rag/knowledge/knowledge-strategy.decorator.ts +0 -6
  115. package/src/lib/rag/knowledge/knowledge-strategy.interface.ts +0 -15
  116. package/src/lib/rag/knowledge/knowledge-strategy.registry.ts +0 -12
  117. package/src/lib/rag/retriever/index.ts +0 -3
  118. package/src/lib/rag/retriever/strategy.decorator.ts +0 -9
  119. package/src/lib/rag/retriever/strategy.interface.ts +0 -32
  120. package/src/lib/rag/retriever/strategy.registry.ts +0 -12
  121. package/src/lib/rag/source/index.ts +0 -3
  122. package/src/lib/rag/source/strategy.decorator.ts +0 -9
  123. package/src/lib/rag/source/strategy.interface.ts +0 -36
  124. package/src/lib/rag/source/strategy.registry.ts +0 -17
  125. package/src/lib/rag/textsplitter/index.ts +0 -3
  126. package/src/lib/rag/textsplitter/strategy.decorator.ts +0 -6
  127. package/src/lib/rag/textsplitter/strategy.interface.ts +0 -28
  128. package/src/lib/rag/textsplitter/strategy.registry.ts +0 -17
  129. package/src/lib/rag/transformer/index.ts +0 -3
  130. package/src/lib/rag/transformer/strategy.decorator.ts +0 -9
  131. package/src/lib/rag/transformer/strategy.interface.ts +0 -34
  132. package/src/lib/rag/transformer/strategy.registry.ts +0 -14
  133. package/src/lib/rag/types.ts +0 -76
  134. package/src/lib/strategy.ts +0 -37
  135. package/src/lib/toolset/builtin.ts +0 -111
  136. package/src/lib/toolset/index.ts +0 -5
  137. package/src/lib/toolset/strategy.decorator.ts +0 -9
  138. package/src/lib/toolset/strategy.interface.ts +0 -32
  139. package/src/lib/toolset/strategy.registry.ts +0 -17
  140. package/src/lib/toolset/toolset.ts +0 -76
  141. package/src/lib/types.ts +0 -47
  142. package/src/lib/vectorstore/index.ts +0 -3
  143. package/src/lib/vectorstore/strategy.decorator.ts +0 -6
  144. package/src/lib/vectorstore/strategy.interface.ts +0 -25
  145. package/src/lib/vectorstore/strategy.registry.ts +0 -17
  146. package/src/lib/workflow/index.ts +0 -2
  147. package/src/lib/workflow/node/index.ts +0 -3
  148. package/src/lib/workflow/node/strategy.decorator.ts +0 -9
  149. package/src/lib/workflow/node/strategy.interface.ts +0 -51
  150. package/src/lib/workflow/node/strategy.registry.ts +0 -18
  151. package/src/lib/workflow/trigger/index.ts +0 -3
  152. package/src/lib/workflow/trigger/strategy.decorator.ts +0 -6
  153. package/src/lib/workflow/trigger/strategy.interface.ts +0 -27
  154. package/src/lib/workflow/trigger/strategy.registry.ts +0 -17
  155. package/tsconfig.json +0 -22
  156. package/tsconfig.lib.json +0 -10
  157. package/tsconfig.spec.json +0 -9
@@ -1,67 +0,0 @@
1
- import { ModuleMetadata, Type } from '@nestjs/common';
2
-
3
- /**
4
- * Metadata definition for a plugin in NestJS.
5
- */
6
- export interface PluginMetadata extends ModuleMetadata {
7
- /**
8
- * List of entities injected by the plugin.
9
- */
10
- entities?: Array<Type<any>> | (() => Array<Type<any>>);
11
-
12
- /**
13
- * List of subscribers injected by the plugin.
14
- */
15
- subscribers?: Array<Type<any>> | (() => Array<Type<any>>);
16
- }
17
-
18
- /**
19
- * Interface for plugins with a bootstrap lifecycle method.
20
- */
21
- export interface IOnPluginBootstrap {
22
- /**
23
- * Called when the plugin is being initialized.
24
- * @returns A void or a Promise representing the completion of the operation.
25
- */
26
- onPluginBootstrap(): void | Promise<void>;
27
- }
28
-
29
- /**
30
- * Interface for plugins with a destroy lifecycle method.
31
- */
32
- export interface IOnPluginDestroy {
33
- /**
34
- * Called when the plugin is being destroyed.
35
- * @returns A void or a Promise representing the completion of the operation.
36
- */
37
- onPluginDestroy(): void | Promise<void>;
38
- }
39
-
40
- /**
41
- * Interface for plugins supporting various seed operations.
42
- */
43
- export interface IOnPluginSeedable {
44
- /**
45
- * Invoked when seeding basic plugin data.
46
- * @returns A void or a Promise representing the completion of the operation.
47
- */
48
- onPluginBasicSeed?(): void | Promise<void>;
49
-
50
- /**
51
- * Invoked when seeding default plugin data.
52
- * @returns A void or a Promise representing the completion of the operation.
53
- */
54
- onPluginDefaultSeed?(): void | Promise<void>;
55
-
56
- /**
57
- * Invoked when seeding random plugin data.
58
- * @returns A void or a Promise representing the completion of the operation.
59
- */
60
- onPluginRandomSeed?(): void | Promise<void>;
61
- }
62
-
63
- /**
64
- * Represents the combined lifecycle methods for a plugin.
65
- * This type combines interfaces for initializing and destroying a plugin.
66
- */
67
- export type PluginLifecycleMethods = IOnPluginBootstrap & IOnPluginDestroy & IOnPluginSeedable;
package/src/lib/plugin.ts DELETED
@@ -1,32 +0,0 @@
1
- import { Module } from '@nestjs/common';
2
- import { MODULE_METADATA } from '@nestjs/common/constants';
3
- import { pick } from 'lodash-es';
4
- import { PluginMetadata } from './plugin.interface';
5
- import { PLUGIN_METADATA } from './plugin-metadata';
6
-
7
- /**
8
- * Decorator function for extending NestJS features with additional metadata.
9
- *
10
- * @param pluginMetadata Metadata to be applied to the target class.
11
- * @returns Class decorator function.
12
- */
13
- export function XpertServerPlugin(pluginMetadata: PluginMetadata): ClassDecorator {
14
- return (targetClass) => {
15
- // Iterate over properties in PLUGIN_METADATA
16
- for (const metadataProperty of Object.values(PLUGIN_METADATA)) {
17
- const property = metadataProperty as keyof PluginMetadata;
18
-
19
- // Check if the property exists in pluginMetadata and is not undefined
20
- if (property in pluginMetadata && pluginMetadata[property] !== undefined) {
21
- // Set metadata on the target class using Reflect
22
- Reflect.defineMetadata(property, pluginMetadata[property] || [], targetClass);
23
- }
24
- }
25
-
26
- // Pick relevant metadata from pluginMetadata based on MODULE_METADATA values
27
- const metadata = pick(pluginMetadata, Object.values(MODULE_METADATA) as string[]);
28
-
29
- // Apply the Module decorator with the picked metadata
30
- Module(metadata)(targetClass);
31
- };
32
- }
@@ -1,3 +0,0 @@
1
- export * from './strategy.decorator'
2
- export * from './strategy.interface'
3
- export * from './strategy.registry'
@@ -1,9 +0,0 @@
1
- import { SetMetadata } from '@nestjs/common';
2
-
3
- export const IMAGE_UNDERSTANDING_STRATEGY = 'IMAGE_UNDERSTANDING_STRATEGY';
4
-
5
- /**
6
- * Decorator to mark a provider as an Image Understanding Strategy
7
- */
8
- export const ImageUnderstandingStrategy = (provider: string) =>
9
- SetMetadata(IMAGE_UNDERSTANDING_STRATEGY, provider);
@@ -1,37 +0,0 @@
1
- import { DocumentInterface } from '@langchain/core/documents'
2
- import { BaseChatModel } from '@langchain/core/language_models/chat_models'
3
- import { IDocumentUnderstandingProvider, IKnowledgeDocument } from '@metad/contracts'
4
- import { Permissions, XpFileSystem } from '../../core/index'
5
- import { ChunkMetadata } from '../types'
6
-
7
- export type TImageUnderstandingConfig = {
8
- stage: 'test' | 'prod'
9
- visionModel: BaseChatModel
10
- permissions?: {
11
- fileSystem?: XpFileSystem
12
- }
13
- }
14
-
15
- export type TImageUnderstandingResult = {
16
- chunks: DocumentInterface<Partial<ChunkMetadata>>[]
17
- // pages?: Document<Partial<ChunkMetadata>>[]
18
- metadata: any // Additional metadata (e.g. model name, processing time)
19
- }
20
-
21
- export interface IImageUnderstandingStrategy<TConfig extends TImageUnderstandingConfig = TImageUnderstandingConfig> {
22
- readonly permissions: Permissions
23
- /**
24
- * Metadata about this strategy
25
- */
26
- readonly meta: IDocumentUnderstandingProvider
27
-
28
- /**
29
- * Validate the configuration
30
- */
31
- validateConfig(config: TConfig): Promise<void>
32
-
33
- /**
34
- * Understand image files (e.g., OCR, VLM, Chart Parsing)
35
- */
36
- understandImages(doc: IKnowledgeDocument<ChunkMetadata>, config: TConfig): Promise<TImageUnderstandingResult>
37
- }
@@ -1,17 +0,0 @@
1
- import { Injectable } from '@nestjs/common';
2
- import { DiscoveryService, Reflector } from '@nestjs/core';
3
- import { BaseStrategyRegistry } from '../../strategy';
4
- import { IMAGE_UNDERSTANDING_STRATEGY } from './strategy.decorator';
5
- import { IImageUnderstandingStrategy, TImageUnderstandingConfig } from './strategy.interface';
6
-
7
- @Injectable()
8
- export class ImageUnderstandingRegistry<TConfig extends TImageUnderstandingConfig = TImageUnderstandingConfig>
9
- extends BaseStrategyRegistry<IImageUnderstandingStrategy<TConfig>>
10
- {
11
- constructor(
12
- discoveryService: DiscoveryService,
13
- reflector: Reflector
14
- ) {
15
- super(IMAGE_UNDERSTANDING_STRATEGY, discoveryService, reflector);
16
- }
17
- }
@@ -1,7 +0,0 @@
1
- export * from './textsplitter/index'
2
- export * from './source/index'
3
- export * from './transformer/index'
4
- export * from './retriever/index'
5
- export * from './types'
6
- export * from './image/index'
7
- export * from './knowledge/index'
@@ -1,3 +0,0 @@
1
- export * from './knowledge-strategy.decorator'
2
- export * from './knowledge-strategy.interface'
3
- export * from './knowledge-strategy.registry'
@@ -1,6 +0,0 @@
1
- import { SetMetadata } from '@nestjs/common';
2
-
3
- export const KNOWLEDGE_STRATEGY = 'KNOWLEDGE_STRATEGY';
4
-
5
- export const KnowledgeStrategyKey = (provider: string) =>
6
- SetMetadata(KNOWLEDGE_STRATEGY, provider);
@@ -1,15 +0,0 @@
1
- import { IIntegration } from '@metad/contracts'
2
- import { Document } from '@langchain/core/documents'
3
-
4
- export type TKnowledgeStrategyParams = {
5
- query: string;
6
- k: number;
7
- filter: Record<string, any>
8
- options: {
9
- knowledgebaseId: string
10
- }
11
- }
12
-
13
- export interface KnowledgeStrategy {
14
- execute(integration: IIntegration, payload: TKnowledgeStrategyParams): Promise<{ chunks: [Document, number][] }>
15
- }
@@ -1,12 +0,0 @@
1
- import { Injectable } from '@nestjs/common'
2
- import { DiscoveryService, Reflector } from '@nestjs/core'
3
- import { BaseStrategyRegistry } from '../../strategy'
4
- import { KNOWLEDGE_STRATEGY } from './knowledge-strategy.decorator'
5
- import { KnowledgeStrategy } from './knowledge-strategy.interface'
6
-
7
- @Injectable()
8
- export class KnowledgeStrategyRegistry extends BaseStrategyRegistry<KnowledgeStrategy> {
9
- constructor(discoveryService: DiscoveryService, reflector: Reflector) {
10
- super(KNOWLEDGE_STRATEGY, discoveryService, reflector)
11
- }
12
- }
@@ -1,3 +0,0 @@
1
- export * from './strategy.decorator'
2
- export * from './strategy.interface'
3
- export * from './strategy.registry'
@@ -1,9 +0,0 @@
1
- import { SetMetadata } from '@nestjs/common';
2
-
3
- export const RETRIEVER_STRATEGY = 'RETRIEVER_STRATEGY';
4
-
5
- /**
6
- * Decorator to mark a provider as a Retriever Strategy
7
- */
8
- export const RetrieverStrategy = (provider: string) =>
9
- SetMetadata(RETRIEVER_STRATEGY, provider);
@@ -1,32 +0,0 @@
1
- import { VectorStore } from '@langchain/core/vectorstores'
2
- import { I18nObject } from '@metad/contracts'
3
- import { Document } from '@langchain/core/documents'
4
-
5
- export type TRetrieverConfig = {
6
- vectorStore: VectorStore
7
- }
8
-
9
- export interface IRetrieverStrategy<TConfig extends TRetrieverConfig = TRetrieverConfig> {
10
- /**
11
- * Metadata about this retriever
12
- */
13
- readonly meta: {
14
- name: string
15
- label: I18nObject
16
- configSchema: any
17
- icon: {
18
- svg?: string
19
- color?: string
20
- }
21
- }
22
-
23
- /**
24
- * Validate the configuration
25
- */
26
- validateConfig(config: TConfig): Promise<void>;
27
-
28
- /**
29
- * Retrieve relevant documents for a given query
30
- */
31
- retrieve(query: string, options?: TConfig): Promise<{ documents: Document[] }>;
32
- }
@@ -1,12 +0,0 @@
1
- import { Injectable } from '@nestjs/common'
2
- import { DiscoveryService, Reflector } from '@nestjs/core'
3
- import { BaseStrategyRegistry } from '../../strategy'
4
- import { RETRIEVER_STRATEGY } from './strategy.decorator'
5
- import { IRetrieverStrategy } from './strategy.interface'
6
-
7
- @Injectable()
8
- export class RetrieverRegistry extends BaseStrategyRegistry<IRetrieverStrategy> {
9
- constructor(discoveryService: DiscoveryService, reflector: Reflector) {
10
- super(RETRIEVER_STRATEGY, discoveryService, reflector)
11
- }
12
- }
@@ -1,3 +0,0 @@
1
- export * from './strategy.decorator'
2
- export * from './strategy.interface'
3
- export * from './strategy.registry'
@@ -1,9 +0,0 @@
1
- import { SetMetadata } from '@nestjs/common';
2
-
3
- export const DOCUMENT_SOURCE_STRATEGY = 'DOCUMENT_SOURCE_STRATEGY';
4
-
5
- /**
6
- * Decorator to mark a provider as a Document Source Strategy
7
- */
8
- export const DocumentSourceStrategy = (provider: string) =>
9
- SetMetadata(DOCUMENT_SOURCE_STRATEGY, provider);
@@ -1,36 +0,0 @@
1
- import { IDocumentSourceProvider, IIntegration } from '@metad/contracts';
2
- import { Document } from '@langchain/core/documents';
3
- import { Permissions } from '../../core';
4
-
5
- export interface IDocumentSourceStrategy<TConfig = any> {
6
- readonly permissions: Permissions
7
- /**
8
- * Metadata about this document source
9
- */
10
- readonly meta: IDocumentSourceProvider
11
-
12
- /**
13
- * Validate the configuration
14
- */
15
- validateConfig(config: TConfig): Promise<void>;
16
-
17
- /**
18
- * Test the connection to the source
19
- *
20
- * @param config
21
- */
22
- test(config: TConfig): Promise<any>;
23
-
24
- /**
25
- * Load documents from the source
26
- */
27
- loadDocuments(config: TConfig, context?: {integration?: IIntegration}): Promise<Document[]>;
28
- /**
29
- * Load a single document by its ID
30
- *
31
- * @deprecated Planning
32
- * @param documentId
33
- * @param context
34
- */
35
- loadDocument?(document: Document, context: {integration?: IIntegration}): Promise<Document>;
36
- }
@@ -1,17 +0,0 @@
1
- import { Injectable } from '@nestjs/common';
2
- import { DiscoveryService, Reflector } from '@nestjs/core';
3
- import { BaseStrategyRegistry } from '../../strategy';
4
- import { DOCUMENT_SOURCE_STRATEGY } from './strategy.decorator';
5
- import { IDocumentSourceStrategy } from './strategy.interface';
6
-
7
- @Injectable()
8
- export class DocumentSourceRegistry<TConfig = any>
9
- extends BaseStrategyRegistry<IDocumentSourceStrategy<TConfig>>
10
- {
11
- constructor(
12
- discoveryService: DiscoveryService,
13
- reflector: Reflector
14
- ) {
15
- super(DOCUMENT_SOURCE_STRATEGY, discoveryService, reflector);
16
- }
17
- }
@@ -1,3 +0,0 @@
1
- export * from './strategy.decorator'
2
- export * from './strategy.interface'
3
- export * from './strategy.registry'
@@ -1,6 +0,0 @@
1
- import { SetMetadata } from '@nestjs/common';
2
-
3
- export const TEXT_SPLITTER_STRATEGY = 'TEXT_SPLITTER_STRATEGY';
4
-
5
- export const TextSplitterStrategy = (provider: string) =>
6
- SetMetadata(TEXT_SPLITTER_STRATEGY, provider);
@@ -1,28 +0,0 @@
1
- import { IDocumentChunkerProvider, KnowledgeStructureEnum } from '@metad/contracts'
2
- import { DocumentInterface } from '@langchain/core/documents'
3
- import { ChunkMetadata } from '../types'
4
-
5
- /**
6
- * Split text content into chunks for embedding and retrieval
7
- */
8
- export interface ITextSplitterStrategy<TConfig = any> {
9
- /**
10
- * Metadata about this splitter
11
- */
12
- readonly meta: IDocumentChunkerProvider
13
-
14
- readonly structure: KnowledgeStructureEnum
15
-
16
- /**
17
- * Validate the configuration
18
- */
19
- validateConfig(config: TConfig): Promise<void>
20
-
21
- /**
22
- * Split a text into chunks and pages (if applicable)
23
- */
24
- splitDocuments(
25
- documents: DocumentInterface[],
26
- options?: TConfig
27
- ): Promise<{ chunks: DocumentInterface<ChunkMetadata>[] }>
28
- }
@@ -1,17 +0,0 @@
1
- import { Injectable } from '@nestjs/common';
2
- import { DiscoveryService, Reflector } from '@nestjs/core';
3
- import { BaseStrategyRegistry } from '../../strategy';
4
- import { TEXT_SPLITTER_STRATEGY } from './strategy.decorator';
5
- import { ITextSplitterStrategy } from './strategy.interface';
6
-
7
- @Injectable()
8
- export class TextSplitterRegistry<TConfig = any>
9
- extends BaseStrategyRegistry<ITextSplitterStrategy<TConfig>>
10
- {
11
- constructor(
12
- discoveryService: DiscoveryService,
13
- reflector: Reflector
14
- ) {
15
- super(TEXT_SPLITTER_STRATEGY, discoveryService, reflector);
16
- }
17
- }
@@ -1,3 +0,0 @@
1
- export * from './strategy.decorator'
2
- export * from './strategy.interface'
3
- export * from './strategy.registry'
@@ -1,9 +0,0 @@
1
- import { SetMetadata } from '@nestjs/common';
2
-
3
- export const DOCUMENT_TRANSFORMER_STRATEGY = 'DOCUMENT_TRANSFORMER_STRATEGY';
4
-
5
- /**
6
- * Decorator to mark a provider as a Document Transformer Strategy
7
- */
8
- export const DocumentTransformerStrategy = (provider: string) =>
9
- SetMetadata(DOCUMENT_TRANSFORMER_STRATEGY, provider);
@@ -1,34 +0,0 @@
1
- import { IDocumentProcessorProvider, IIntegration, IKnowledgeDocument } from '@metad/contracts'
2
- import { Permissions, XpFileSystem } from '../../core/index'
3
- import { ChunkMetadata } from '../types'
4
-
5
- export type TDocumentTransformerConfig = {
6
- stage: 'test' | 'prod'
7
- tempDir?: string
8
- permissions?: {
9
- fileSystem?: XpFileSystem
10
- integration?: IIntegration
11
- }
12
- }
13
-
14
- export interface IDocumentTransformerStrategy<TConfig extends TDocumentTransformerConfig = TDocumentTransformerConfig> {
15
- /**
16
- * Metadata about this transformer
17
- */
18
- readonly meta: IDocumentProcessorProvider
19
-
20
- readonly permissions: Permissions
21
-
22
- /**
23
- * Validate the configuration
24
- */
25
- validateConfig(config: TConfig): Promise<void>
26
-
27
- /**
28
- * Transform documents (e.g., extract, OCR, normalize, enrich metadata)
29
- */
30
- transformDocuments(
31
- files: Partial<IKnowledgeDocument>[],
32
- config: TConfig
33
- ): Promise<Partial<IKnowledgeDocument<ChunkMetadata>>[]>
34
- }
@@ -1,14 +0,0 @@
1
- import { Injectable } from '@nestjs/common'
2
- import { DiscoveryService, Reflector } from '@nestjs/core'
3
- import { BaseStrategyRegistry } from '../../strategy'
4
- import { DOCUMENT_TRANSFORMER_STRATEGY } from './strategy.decorator'
5
- import { IDocumentTransformerStrategy, TDocumentTransformerConfig } from './strategy.interface'
6
-
7
- @Injectable()
8
- export class DocumentTransformerRegistry<
9
- TConfig extends TDocumentTransformerConfig = TDocumentTransformerConfig
10
- > extends BaseStrategyRegistry<IDocumentTransformerStrategy<TConfig>> {
11
- constructor(discoveryService: DiscoveryService, reflector: Reflector) {
12
- super(DOCUMENT_TRANSFORMER_STRATEGY, discoveryService, reflector)
13
- }
14
- }
@@ -1,76 +0,0 @@
1
- import { DocumentInterface } from '@langchain/core/documents'
2
- import { IDocChunkMetadata } from '@metad/contracts'
3
- import fs from 'fs'
4
- import http from 'http'
5
- import https from 'https'
6
-
7
- export { TDocumentAsset } from '@metad/contracts'
8
-
9
- export interface ChunkMetadata extends IDocChunkMetadata{
10
- startOffset?: number // Start position in the original text
11
- endOffset?: number // End position in the original text
12
- type?: 'parent' | 'child' // Chunk type
13
-
14
- [key: string]: any // Allow plugin extensions
15
- }
16
-
17
- /**
18
- * Merge parent chunks with their child chunks based on metadata (parentId and chunkId)
19
- *
20
- * @deprecated use buildChunkTreeAndFindLeaves instead
21
- */
22
- export function mergeParentChildChunks(
23
- chunks: DocumentInterface<ChunkMetadata>[], // Parent chunks
24
- children: DocumentInterface<ChunkMetadata>[] // Child chunks
25
- ): DocumentInterface<ChunkMetadata>[] {
26
- const chunkMap = new Map<string, DocumentInterface<ChunkMetadata>>()
27
- for (const chunk of chunks) {
28
- chunkMap.set(chunk.metadata.chunkId, chunk)
29
- }
30
- for (const child of children) {
31
- if (!child.metadata.parentId) {
32
- if (chunkMap.has(child.metadata.chunkId)) {
33
- console.warn(`Duplicate chunkId found: ${child.metadata.chunkId}, skipping...`)
34
- continue
35
- }
36
- chunkMap.set(child.metadata.chunkId, child)
37
- continue
38
- }
39
- const parent = chunkMap.get(child.metadata.parentId)
40
- if (parent) {
41
- if (!parent.metadata.children) {
42
- parent.metadata.children = []
43
- }
44
- parent.metadata.children.push(child)
45
- }
46
- }
47
- return Array.from(chunkMap.values())
48
- }
49
-
50
- export function isRemoteFile(path: string): boolean {
51
- return path.startsWith('http://') || path.startsWith('https://') || path.startsWith('ftp://')
52
- }
53
-
54
- export async function downloadRemoteFile(url: string, dest: string): Promise<string> {
55
- return new Promise((resolve, reject) => {
56
- const file = fs.createWriteStream(dest)
57
- const client = url.startsWith('https') ? https : http
58
- const request = client.get(url, (response) => {
59
- if (response.statusCode !== 200) {
60
- reject(new Error(`Failed to get '${url}' (${response.statusCode})`))
61
- return
62
- }
63
- response.pipe(file)
64
- })
65
- file.on('finish', () => {
66
- file.close()
67
- resolve(dest)
68
- })
69
- request.on('error', (err) => {
70
- fs.unlink(dest, () => reject(err))
71
- })
72
- file.on('error', (err) => {
73
- fs.unlink(dest, () => reject(err))
74
- })
75
- })
76
- }
@@ -1,37 +0,0 @@
1
- import { OnModuleInit } from "@nestjs/common"
2
- import { DiscoveryService, Reflector } from "@nestjs/core"
3
-
4
- export class BaseStrategyRegistry<S> implements OnModuleInit {
5
- protected strategies = new Map<string, S>()
6
-
7
- constructor(
8
- protected readonly strategyKey: string,
9
- protected discoveryService: DiscoveryService,
10
- protected reflector: Reflector
11
- ) {}
12
-
13
- onModuleInit() {
14
- const providers = this.discoveryService.getProviders()
15
- for (const wrapper of providers) {
16
- const { instance } = wrapper
17
- if (!instance) continue
18
-
19
- const type = this.reflector.get<string>(this.strategyKey, instance.constructor)
20
- if (type) {
21
- this.strategies.set(type, instance as S)
22
- }
23
- }
24
- }
25
-
26
- get(type: string): S {
27
- const strategy = this.strategies.get(type)
28
- if (!strategy) {
29
- throw new Error(`No strategy found for type ${type}`)
30
- }
31
- return strategy
32
- }
33
-
34
- list(): S[] {
35
- return Array.from(this.strategies.values())
36
- }
37
- }