@xpert-ai/plugin-sdk 0.0.1

package/src/lib/vectorstore/strategy.registry.ts

@@ -0,0 +1,17 @@
+import { Injectable } from '@nestjs/common';
+import { DiscoveryService, Reflector } from '@nestjs/core';
+import { BaseStrategyRegistry } from '../strategy';
+import { VECTOR_STORE_STRATEGY } from './strategy.decorator';
+import { IVectorStoreStrategy } from './strategy.interface';
+
+@Injectable()
+export class VectorStoreRegistry<TConfig = any>
+  extends BaseStrategyRegistry<IVectorStoreStrategy<TConfig>> 
+{
+  constructor(
+    discoveryService: DiscoveryService,
+    reflector: Reflector
+  ) {
+    super(VECTOR_STORE_STRATEGY, discoveryService, reflector);
+  }
+}

package/src/lib/workflow/index.ts

@@ -0,0 +1,2 @@
+export * from './trigger/index'
+export * from './node/index'

package/src/lib/workflow/node/index.ts

@@ -0,0 +1,3 @@
+export * from './strategy.decorator'
+export * from './strategy.interface'
+export * from './strategy.registry'

package/src/lib/workflow/node/strategy.decorator.ts

@@ -0,0 +1,9 @@
+import { SetMetadata } from '@nestjs/common';
+
+export const WORKFLOW_NODE_STRATEGY = 'WORKFLOW_NODE_STRATEGY';
+
+/**
+ * Decorator to mark a provider as a Workflow Node Strategy
+ */
+export const WorkflowNodeStrategy = (provider: string) =>
+  SetMetadata(WORKFLOW_NODE_STRATEGY, provider);

package/src/lib/workflow/node/strategy.interface.ts

@@ -0,0 +1,49 @@
+import { Runnable } from '@langchain/core/runnables'
+import { BaseChannel } from '@langchain/langgraph'
+import {
+  IEnvironment,
+  IWorkflowNode,
+  TWorkflowNodeMeta,
+  TXpertGraph,
+  TXpertParameter,
+  TXpertTeamNode
+} from '@metad/contracts'
+
+export type TWorkflowNodeParams<TConfig = any> = {
+  xpertId: string
+  agentKey?: string
+  node: TXpertTeamNode
+  config: TConfig
+}
+
+/**
+ * Workflow Node Strategy interface
+ */
+export interface IWorkflowNodeStrategy<TConfig = any, TResult = any> {
+  /**
+   * Metadata describing the node (type, label, description, config schema, etc.)
+   */
+  meta: TWorkflowNodeMeta
+
+  /**
+   * Create the node subgraph
+   */
+  create(payload: {
+    graph: TXpertGraph
+    node: TXpertTeamNode & { type: 'workflow' }
+    xpertId: string
+    environment: IEnvironment
+    isDraft: boolean
+  }): {
+    name?: string
+    graph: Runnable
+    ends: string[]
+    channel: {
+      name: string
+      annotation: BaseChannel
+    }
+    navigator?: (state, config) => Promise<any>
+  }
+
+  outputVariables(entity: IWorkflowNode): TXpertParameter[]
+}

package/src/lib/workflow/node/strategy.registry.ts

@@ -0,0 +1,18 @@
+import { Injectable } from '@nestjs/common';
+import { DiscoveryService, Reflector } from '@nestjs/core';
+import { BaseStrategyRegistry } from '../../strategy';
+import { WORKFLOW_NODE_STRATEGY } from './strategy.decorator';
+import { IWorkflowNodeStrategy } from './strategy.interface';
+
+/**
+ * Registry for Workflow Node Strategies
+ */
+@Injectable()
+export class WorkflowNodeRegistry<TConfig = any, TResult = any> extends BaseStrategyRegistry<IWorkflowNodeStrategy<TConfig, TResult>> {
+  constructor(
+    discoveryService: DiscoveryService,
+    reflector: Reflector
+  ) {
+    super(WORKFLOW_NODE_STRATEGY, discoveryService, reflector);
+  }
+}

package/src/lib/workflow/trigger/index.ts

@@ -0,0 +1,3 @@
+export * from './strategy.decorator'
+export * from './strategy.interface'
+export * from './strategy.registry'

package/src/lib/workflow/trigger/strategy.decorator.ts

@@ -0,0 +1,6 @@
+import { SetMetadata } from '@nestjs/common';
+
+export const WORKFLOW_TRIGGER_STRATEGY = 'WORKFLOW_TRIGGER_STRATEGY';
+
+export const WorkflowTriggerStrategy = (provider: string) =>
+  SetMetadata(WORKFLOW_TRIGGER_STRATEGY, provider);

package/src/lib/workflow/trigger/strategy.interface.ts

@@ -0,0 +1,27 @@
+import { TWorkflowTriggerMeta, TXpertTeamNode } from '@metad/contracts'
+
+export type TWorkflowTriggerParams<T> = {
+  xpertId: string
+  agentKey?: string
+  node?: TXpertTeamNode
+  config: T
+}
+
+export interface IWorkflowTriggerStrategy<T> {
+  meta: TWorkflowTriggerMeta
+
+  validate(payload: TWorkflowTriggerParams<T>): Promise<any[]>
+
+  /**
+   * Initialize the trigger when publish xpert workflow
+   * 
+   * @param payload 
+   * @param callback 
+   */
+  publish(payload: TWorkflowTriggerParams<T>, callback: (payload: any) => void): Promise<any> | void
+
+  /**
+   * Stop the trigger
+   */
+  stop(payload: TWorkflowTriggerParams<T>): void;
+}

package/src/lib/workflow/trigger/strategy.registry.ts

@@ -0,0 +1,17 @@
+import { Injectable } from '@nestjs/common'
+import { DiscoveryService, Reflector } from '@nestjs/core'
+import { BaseStrategyRegistry } from '../../strategy'
+import { WORKFLOW_TRIGGER_STRATEGY } from './strategy.decorator'
+import { IWorkflowTriggerStrategy } from './strategy.interface'
+
+@Injectable()
+export class WorkflowTriggerRegistry<T = any> extends BaseStrategyRegistry<IWorkflowTriggerStrategy<T>> {
+
+	constructor(
+		discoveryService: DiscoveryService,
+		reflector: Reflector
+	) {
+		super(WORKFLOW_TRIGGER_STRATEGY, discoveryService, reflector);
+	}
+
+}

package/tsconfig.json

@@ -0,0 +1,22 @@
+{
+  "extends": "../../tsconfig.base.json",
+  "compilerOptions": {
+    "module": "commonjs",
+    "forceConsistentCasingInFileNames": true,
+    "strict": true,
+    "noImplicitOverride": true,
+    "noPropertyAccessFromIndexSignature": true,
+    "noImplicitReturns": true,
+    "noFallthroughCasesInSwitch": true
+  },
+  "files": [],
+  "include": [],
+  "references": [
+    {
+      "path": "./tsconfig.lib.json"
+    },
+    {
+      "path": "./tsconfig.spec.json"
+    }
+  ]
+}

package/tsconfig.lib.json

@@ -0,0 +1,10 @@
+{
+  "extends": "./tsconfig.json",
+  "compilerOptions": {
+    "outDir": "../../dist/out-tsc",
+    "declaration": true,
+    "types": ["node"]
+  },
+  "include": ["src/**/*.ts"],
+  "exclude": ["jest.config.ts", "src/**/*.spec.ts", "src/**/*.test.ts"]
+}

package/tsconfig.spec.json

@@ -0,0 +1,9 @@
+{
+  "extends": "./tsconfig.json",
+  "compilerOptions": {
+    "outDir": "../../dist/out-tsc",
+    "module": "commonjs",
+    "types": ["jest", "node"]
+  },
+  "include": ["jest.config.ts", "src/**/*.test.ts", "src/**/*.spec.ts", "src/**/*.d.ts"]
+}