@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.
- package/index.cjs.d.ts +1 -0
- package/index.cjs.js +45273 -5
- package/index.esm.d.ts +1 -0
- package/index.esm.js +45192 -0
- package/package.json +3 -3
- package/src/index.d.ts +12 -0
- package/src/lib/ai-model/abstract-provider.d.ts +41 -0
- package/src/lib/ai-model/ai-model-provider.decorator.d.ts +2 -0
- package/src/lib/ai-model/ai-model-provider.interface.d.ts +42 -0
- package/src/lib/ai-model/ai-model-provider.registry.d.ts +6 -0
- package/src/lib/ai-model/ai-model.d.ts +33 -0
- package/src/lib/ai-model/entities/defaults.d.ts +3 -0
- package/src/lib/ai-model/entities/index.d.ts +2 -0
- package/src/lib/ai-model/entities/model_entities.d.ts +14 -0
- package/src/lib/ai-model/errors.d.ts +5 -0
- package/src/lib/ai-model/index.d.ts +8 -0
- package/src/lib/ai-model/llm.d.ts +49 -0
- package/src/lib/ai-model/openai-compatible/completions.d.ts +23 -0
- package/src/lib/ai-model/openai-compatible/index.d.ts +3 -0
- package/src/lib/ai-model/openai-compatible/rerank.d.ts +20 -0
- package/src/lib/ai-model/openai-compatible/speech2text.d.ts +17 -0
- package/src/lib/ai-model/types/index.d.ts +11 -0
- package/src/lib/ai-model/types/model.d.ts +43 -0
- package/src/lib/ai-model/types/rerank.d.ts +20 -0
- package/src/lib/ai-model/types/speech2text.d.ts +3 -0
- package/src/lib/ai-model/types/text-embedding-model.d.ts +7 -0
- package/src/lib/ai-model/types/tts.d.ts +5 -0
- package/src/lib/core/file-system.d.ts +55 -0
- package/src/lib/core/i18n.d.ts +2 -0
- package/src/lib/core/index.d.ts +5 -0
- package/src/lib/core/permissions.d.ts +74 -0
- package/src/lib/core/schema.d.ts +22 -0
- package/src/lib/core/utils.d.ts +12 -0
- package/src/lib/integration/index.d.ts +3 -0
- package/src/lib/integration/strategy.decorator.d.ts +2 -0
- package/src/lib/integration/strategy.interface.d.ts +9 -0
- package/src/lib/integration/strategy.registry.d.ts +6 -0
- package/src/lib/logger.d.ts +2 -0
- package/src/lib/plugin-metadata.d.ts +17 -0
- package/src/lib/plugin.d.ts +8 -0
- package/src/lib/{plugin.hook.ts → plugin.hook.d.ts} +0 -4
- package/src/lib/plugin.interface.d.ts +59 -0
- package/src/lib/rag/image/index.d.ts +3 -0
- package/src/lib/rag/image/strategy.decorator.d.ts +5 -0
- package/src/lib/rag/image/strategy.interface.d.ts +31 -0
- package/src/lib/rag/image/strategy.registry.d.ts +6 -0
- package/src/lib/rag/index.d.ts +7 -0
- package/src/lib/rag/knowledge/index.d.ts +3 -0
- package/src/lib/rag/knowledge/knowledge-strategy.decorator.d.ts +2 -0
- package/src/lib/rag/knowledge/knowledge-strategy.interface.d.ts +15 -0
- package/src/lib/rag/knowledge/knowledge-strategy.registry.d.ts +6 -0
- package/src/lib/rag/retriever/index.d.ts +3 -0
- package/src/lib/rag/retriever/strategy.decorator.d.ts +5 -0
- package/src/lib/rag/retriever/strategy.interface.d.ts +30 -0
- package/src/lib/rag/retriever/strategy.registry.d.ts +6 -0
- package/src/lib/rag/source/index.d.ts +3 -0
- package/src/lib/rag/source/strategy.decorator.d.ts +5 -0
- package/src/lib/rag/source/strategy.interface.d.ts +36 -0
- package/src/lib/rag/source/strategy.registry.d.ts +6 -0
- package/src/lib/rag/textsplitter/index.d.ts +3 -0
- package/src/lib/rag/textsplitter/strategy.decorator.d.ts +2 -0
- package/src/lib/rag/textsplitter/strategy.interface.d.ts +23 -0
- package/src/lib/rag/textsplitter/strategy.registry.d.ts +6 -0
- package/src/lib/rag/transformer/index.d.ts +3 -0
- package/src/lib/rag/transformer/strategy.decorator.d.ts +5 -0
- package/src/lib/rag/transformer/strategy.interface.d.ts +26 -0
- package/src/lib/rag/transformer/strategy.registry.d.ts +6 -0
- package/src/lib/rag/types.d.ts +18 -0
- package/src/lib/strategy.d.ts +12 -0
- package/src/lib/toolset/builtin.d.ts +54 -0
- package/src/lib/toolset/index.d.ts +5 -0
- package/src/lib/toolset/strategy.decorator.d.ts +5 -0
- package/src/lib/toolset/strategy.interface.d.ts +28 -0
- package/src/lib/toolset/strategy.registry.d.ts +6 -0
- package/src/lib/toolset/toolset.d.ts +53 -0
- package/src/lib/types.d.ts +47 -0
- package/src/lib/vectorstore/index.d.ts +3 -0
- package/src/lib/vectorstore/strategy.decorator.d.ts +2 -0
- package/src/lib/vectorstore/strategy.interface.d.ts +24 -0
- package/src/lib/vectorstore/strategy.registry.d.ts +6 -0
- package/src/lib/workflow/index.d.ts +2 -0
- package/src/lib/workflow/node/index.d.ts +3 -0
- package/src/lib/workflow/node/strategy.decorator.d.ts +5 -0
- package/src/lib/workflow/node/strategy.interface.d.ts +41 -0
- package/src/lib/workflow/node/strategy.registry.d.ts +9 -0
- package/src/lib/workflow/trigger/index.d.ts +3 -0
- package/src/lib/workflow/trigger/strategy.decorator.d.ts +2 -0
- package/src/lib/workflow/trigger/strategy.interface.d.ts +22 -0
- package/src/lib/workflow/trigger/strategy.registry.d.ts +6 -0
- package/.eslintrc.json +0 -30
- package/.swcrc +0 -29
- package/jest.config.ts +0 -28
- package/project.json +0 -59
- package/src/index.ts +0 -11
- package/src/lib/core/file-system.ts +0 -113
- package/src/lib/core/i18n.ts +0 -31
- package/src/lib/core/index.ts +0 -4
- package/src/lib/core/permissions.ts +0 -97
- package/src/lib/core/schema.ts +0 -23
- package/src/lib/integration/index.ts +0 -3
- package/src/lib/integration/strategy.decorator.ts +0 -6
- package/src/lib/integration/strategy.interface.ts +0 -11
- package/src/lib/integration/strategy.registry.ts +0 -12
- package/src/lib/logger.ts +0 -18
- package/src/lib/plugin-metadata.ts +0 -20
- package/src/lib/plugin.interface.ts +0 -67
- package/src/lib/plugin.ts +0 -32
- package/src/lib/rag/image/index.ts +0 -3
- package/src/lib/rag/image/strategy.decorator.ts +0 -9
- package/src/lib/rag/image/strategy.interface.ts +0 -37
- package/src/lib/rag/image/strategy.registry.ts +0 -17
- package/src/lib/rag/index.ts +0 -7
- package/src/lib/rag/knowledge/index.ts +0 -3
- package/src/lib/rag/knowledge/knowledge-strategy.decorator.ts +0 -6
- package/src/lib/rag/knowledge/knowledge-strategy.interface.ts +0 -15
- package/src/lib/rag/knowledge/knowledge-strategy.registry.ts +0 -12
- package/src/lib/rag/retriever/index.ts +0 -3
- package/src/lib/rag/retriever/strategy.decorator.ts +0 -9
- package/src/lib/rag/retriever/strategy.interface.ts +0 -32
- package/src/lib/rag/retriever/strategy.registry.ts +0 -12
- package/src/lib/rag/source/index.ts +0 -3
- package/src/lib/rag/source/strategy.decorator.ts +0 -9
- package/src/lib/rag/source/strategy.interface.ts +0 -36
- package/src/lib/rag/source/strategy.registry.ts +0 -17
- package/src/lib/rag/textsplitter/index.ts +0 -3
- package/src/lib/rag/textsplitter/strategy.decorator.ts +0 -6
- package/src/lib/rag/textsplitter/strategy.interface.ts +0 -28
- package/src/lib/rag/textsplitter/strategy.registry.ts +0 -17
- package/src/lib/rag/transformer/index.ts +0 -3
- package/src/lib/rag/transformer/strategy.decorator.ts +0 -9
- package/src/lib/rag/transformer/strategy.interface.ts +0 -34
- package/src/lib/rag/transformer/strategy.registry.ts +0 -14
- package/src/lib/rag/types.ts +0 -76
- package/src/lib/strategy.ts +0 -37
- package/src/lib/toolset/builtin.ts +0 -111
- package/src/lib/toolset/index.ts +0 -5
- package/src/lib/toolset/strategy.decorator.ts +0 -9
- package/src/lib/toolset/strategy.interface.ts +0 -32
- package/src/lib/toolset/strategy.registry.ts +0 -17
- package/src/lib/toolset/toolset.ts +0 -76
- package/src/lib/types.ts +0 -47
- package/src/lib/vectorstore/index.ts +0 -3
- package/src/lib/vectorstore/strategy.decorator.ts +0 -6
- package/src/lib/vectorstore/strategy.interface.ts +0 -25
- package/src/lib/vectorstore/strategy.registry.ts +0 -17
- package/src/lib/workflow/index.ts +0 -2
- package/src/lib/workflow/node/index.ts +0 -3
- package/src/lib/workflow/node/strategy.decorator.ts +0 -9
- package/src/lib/workflow/node/strategy.interface.ts +0 -51
- package/src/lib/workflow/node/strategy.registry.ts +0 -18
- package/src/lib/workflow/trigger/index.ts +0 -3
- package/src/lib/workflow/trigger/strategy.decorator.ts +0 -6
- package/src/lib/workflow/trigger/strategy.interface.ts +0 -27
- package/src/lib/workflow/trigger/strategy.registry.ts +0 -17
- package/tsconfig.json +0 -22
- package/tsconfig.lib.json +0 -10
- 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,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
|
-
}
|
package/src/lib/rag/index.ts
DELETED
|
@@ -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,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,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,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,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
|
-
}
|
package/src/lib/rag/types.ts
DELETED
|
@@ -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
|
-
}
|
package/src/lib/strategy.ts
DELETED
|
@@ -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
|
-
}
|