@metad/contracts 0.0.1 → 3.6.0-beta.0

package/src/ai/xpert-workflow.model.ts

@@ -0,0 +1,449 @@
+import { I18nObject, letterStartSUID } from "../types"
+import { ICopilotModel } from "./copilot-model.model"
+import { TKBRecallParams } from "./knowledgebase.model"
+import { ApiAuthType, TErrorHandling, TXpertRefParameter } from "./types"
+import { TStateVariable, TXpertParameter } from "./xpert.model"
+
+export type TWorkflowNodeMeta = {
+  name: string
+  label: I18nObject
+  icon: {
+    svg?: string
+    remixicon?: string
+    color?: string
+  }
+  configSchema: any
+}
+
+export type TWorkflowTriggerMeta = TWorkflowNodeMeta
+
+export enum WorkflowNodeTypeEnum {
+  /**
+   * Trigger
+   */
+  TRIGGER = 'trigger',
+  /**
+   * State Variable Assigner
+   */
+  ASSIGNER = 'assigner',
+  /**
+   * Router
+   */
+  IF_ELSE = 'if-else',
+  LIST_OPERATOR = 'list-operator',
+  VARIABLE_AGGREGATOR = 'variable-aggregator',
+  // SPLITTER = 'splitter',
+  ITERATING = 'iterating',
+  ANSWER = 'answer',
+  CODE = 'code',
+  HTTP = 'http',
+  KNOWLEDGE = 'knowledge',
+  SUBFLOW = 'subflow',
+  TEMPLATE = 'template',
+  CLASSIFIER = 'classifier',
+  TOOL = 'tool',
+  AGENT_TOOL = 'agent-tool',
+  NOTE = 'note',
+  /**
+   * Task node, distribute tasks to sub-agents
+   */
+  TASK = 'task',
+
+  // Knowledge Pipeline nodes
+  SOURCE = 'source',
+  PROCESSOR = 'processor',
+  CHUNKER = 'chunker',
+  UNDERSTANDING = 'understanding',
+  KNOWLEDGE_BASE = 'knowledgebase',
+}
+
+export interface IWorkflowNode {
+  id: string
+  key: string
+  title?: string
+  description?: string | I18nObject
+  type: WorkflowNodeTypeEnum
+}
+
+export enum VariableOperationEnum {
+  APPEND = 'append',
+  EXTEND = 'extend',
+  OVERWRITE = 'overwrite',
+  CLEAR = 'clear'
+}
+
+export type TVariableAssigner = {
+  id: string
+  inputType: 'variable' | 'constant' | 'message'
+  /**
+   * value from variable
+   */
+  value?: string
+  /**
+   * value write to state's variable
+   */
+  variableSelector: string
+  /**
+   * How to write value to variable
+   */
+  operation: VariableOperationEnum
+
+  /**
+   * The message template to write to messages variable
+   */
+  messages?: {role: string; content: string}[]
+}
+
+export interface IWFNAssigner extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.ASSIGNER
+  assigners: TVariableAssigner[]
+}
+
+export interface IWFNIfElse extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.IF_ELSE
+  cases: TWFCase[]
+}
+
+/**
+ * The list operator can filter and extract attributes such as file format type, file name, and size,
+ * passing different format files to corresponding processing nodes to achieve precise control over different file processing flows.
+ */
+export interface IWFNListOperator extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.LIST_OPERATOR
+  // Input Variable *
+  input: string
+  // Variable type of items in the input array
+  itemVarType: string
+  // Filter condition
+  filterBy?: {
+    enabled: boolean
+    conditions: TWFCaseCondition[]
+    logicalOperator: WorkflowLogicalOperator
+  }
+  // Extract the N item
+  extractBy?: {
+    enabled: boolean
+    index: number
+  }
+  // Sort items by variable
+  sortBy?: {
+    enabled: boolean
+    variable: string
+    descending?: boolean
+  }
+  // Top N items
+  topN?: {
+    enabled: boolean
+    count: number
+  }
+}
+
+/**
+ * Aggregate variables from multiple branches into a single variable to achieve unified configuration for downstream nodes.
+ */
+export interface IWFNVariableAggregator extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.VARIABLE_AGGREGATOR
+  // Input Variables *
+  inputs: string[]
+  // Output Variable type
+  outputType: string
+  // Aggregation method
+  // method?: 'concat' | 'merge' | 'custom'
+}
+
+// export interface IWFNSplitter extends IWorkflowNode {
+//   type: WorkflowNodeTypeEnum.SPLITTER
+// }
+
+export interface IWFNIterating extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.ITERATING
+  /**
+   * Variable name of input array in state
+   */
+  inputVariable: string
+  inputParams?: TXpertRefParameter[]
+  outputParams?: TXpertRefParameter[]
+  /**
+   * Execute in parallel, otherwise execute sequentially
+   */
+  parallel?: boolean
+  /**
+   * Maximum number of parallel task
+   */
+  maximum?: number
+
+  /**
+   * - terminate: terminate on error
+   * - ignore: ignore error and continue
+   * - remove: remove error output
+   */
+  errorMode?: 'terminate' | 'ignore' | 'remove'
+}
+
+/**
+ * The parameter name that represents the entire current element in array
+ */
+export const IteratingItemParameterName = '$item'
+export const IteratingIndexParameterName = '$index'
+
+export interface IWFNAnswer extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.ANSWER
+  promptTemplate: string
+  mute?: boolean
+}
+
+export enum WorkflowLogicalOperator {
+  AND = 'and',
+  OR = 'or'
+}
+
+export enum WorkflowComparisonOperator {
+  CONTAINS = 'contains',
+  NOT_CONTAINS = 'not-contains',
+  EQUAL = 'equal',
+  NOT_EQUAL = 'not-equal',
+  GT = 'gt',
+  LT = 'lt',
+  GE = 'ge',
+  LE = 'le',
+  STARTS_WITH = 'starts-with',
+  ENDS_WITH = 'ends-with',
+  EMPTY = 'empty',
+  NOT_EMPTY = 'not-empty',
+  IS_TRUE = 'is-true',
+  IS_FALSE = 'is-false',
+  LIKE = 'like',
+  NOT_LIKE = 'not-like'
+}
+
+/**
+ * Comparison condition for if-else and list-operator nodes
+ */
+export type TWFCaseCondition = {
+  id: string
+  comparisonOperator: WorkflowComparisonOperator
+  value?: string
+  varType?: string
+  variableSelector?: string
+}
+
+export type TWFCase = {
+  caseId: string
+  conditions: TWFCaseCondition[]
+  logicalOperator: WorkflowLogicalOperator
+}
+
+export type TWorkflowVarGroup = {
+  // /**
+  //  * @deprecated use group
+  //  */
+  // agent?: Partial<IXpertAgent>
+  group?: {
+    name: string
+    description: string | I18nObject
+  }
+  variables: TStateVariable[]
+}
+
+export type TWorkflowRetry = {
+  enabled?: boolean
+  stopAfterAttempt?: number
+  retryInterval?: number // second
+}
+
+export interface IWFNCode extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.CODE
+  language: 'python' | 'javascript'
+  code: string
+  inputs: {name: string; variable?: string}[]
+  outputs: TXpertParameter[]
+  /**
+   * Retry on failure
+   */
+  retry?: TWorkflowRetry
+  /**
+   * Error handling
+   */
+  errorHandling?: TErrorHandling
+}
+
+// Http workflow node
+export type BodyType = 'none' | 'form-data' | 'x-www-form-urlencoded' | 'json' | 'raw' | 'binary';
+interface HttpParam {
+  key: string;
+  value: string;
+}
+interface HttpHeader {
+  name: string;
+  value: string;
+}
+
+export type TWorkflowAuthorization = {
+  auth_type?: ApiAuthType
+  api_key_type?: '' | 'bearer' | 'custom'
+  api_key_header?: string
+  api_key_value?: string
+  username?: string
+  password?: string
+}
+
+export interface IWFNHttp extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.HTTP
+  authorization?: TWorkflowAuthorization
+  method: 'get' | 'post' | 'put' | 'delete' | 'patch' | 'head'
+  url: string
+  headers: HttpHeader[]
+  params: HttpParam[]
+  body: {
+    type?: BodyType
+    body?: string; // 如果是 JSON 类型
+    encodedForm?: HttpParam[]
+  }
+  connectionTimeout?: number
+  readTimeout?: number
+  writeTimeout?: number
+  /**
+   * Retry on failure
+   */
+  retry?: TWorkflowRetry
+  /**
+   * Error handling
+   */
+  errorHandling?: TErrorHandling
+}
+
+export interface IWFNKnowledgeRetrieval extends IWorkflowNode {
+  queryVariable: string
+  knowledgebases: string[]
+  recall?: TKBRecallParams
+}
+
+export interface IWFNSubflow extends IWorkflowNode {
+  inputParams?: TXpertRefParameter[]
+  outputParams?: TXpertRefParameter[]
+}
+
+export interface IWFNTemplate extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.TEMPLATE,
+  inputParams?: TXpertRefParameter[]
+  code: string
+  
+  /**
+   * Error handling
+   */
+  errorHandling?: TErrorHandling
+}
+
+export interface IWFNClassifier extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.CLASSIFIER
+  copilotModel: ICopilotModel
+  inputVariables: string[]
+  classes: {
+    description?: string
+  }[]
+  instruction?: string
+}
+
+export interface IWFNClassifier extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.CLASSIFIER
+  copilotModel: ICopilotModel
+  inputVariables: string[]
+  classes: {
+    description?: string
+  }[]
+  instruction?: string
+}
+
+export interface IWFNTemplate extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.TEMPLATE,
+  inputParams?: TXpertRefParameter[]
+  code: string
+  
+  /**
+   * Error handling
+   */
+  errorHandling?: TErrorHandling
+}
+
+export interface IWFNTool extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.TOOL,
+  toolsetId: string
+  toolName: string
+  /**
+   * @deprecated Use parameters instead
+   */
+  parameterVariable: string
+  parameters?: any
+  omitBlankValues?: boolean
+  
+  /**
+   * Error handling
+   */
+  errorHandling?: TErrorHandling
+}
+
+export interface IWFNNote extends IWorkflowNode {
+  content: string
+}
+
+export interface IWFNAgentTool extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.AGENT_TOOL,
+  // Tool name
+  toolName: string
+  // Tool description
+  toolDescription?: string
+  // Tool schema
+  toolParameters?: TXpertParameter[]
+
+  /**
+   * End point tool
+   */
+  isEnd?: boolean
+
+  /**
+   * @deprecated Mixed sub-processes are not suitable for catching exceptions unless they are implemented using a separate subgraph
+   * Error handling
+   */
+  errorHandling?: TErrorHandling
+}
+
+export interface IWFNTrigger extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.TRIGGER,
+  from: 'chat' | 'integration' | 'scheduler'
+  parameters?: TXpertParameter[]
+  config?: any
+}
+
+export interface IWFNTask extends IWorkflowNode {
+  type: WorkflowNodeTypeEnum.TASK,
+  descriptionPrefix?: string
+  descriptionSuffix?: string
+}
+
+export function genListOperatorKey() {
+  return letterStartSUID('ListOperator_')
+}
+
+export function genVariableAggregatorKey() {
+  return letterStartSUID('VariableAggregator_')
+}
+
+export function isAgentKey(key: string) {
+  return key?.toLowerCase().startsWith('agent_')
+}
+
+export function isRouterKey(key: string) {
+  return key?.toLowerCase().startsWith('router_')
+}
+
+export function isIteratingKey(key: string) {
+  return key?.toLowerCase().startsWith('iterating_')
+}
+
+export function isWorkflowKey(key: string) {
+  return isRouterKey(key) || isIteratingKey(key)
+}
+
+export function workflowNodeIdentifier(node: IWorkflowNode) {
+  return node.title || node.key
+}

package/src/ai/xpert-workspace.model.ts

@@ -0,0 +1,42 @@
+import { IBasePerTenantAndOrganizationEntityModel } from '../base-entity.model'
+import { IUser } from '../user.model'
+import { IEnvironment } from './environment.model'
+import { IXpert } from './xpert.model'
+
+/**
+ * Expert Workspace
+ */
+export interface IXpertWorkspace extends IBasePerTenantAndOrganizationEntityModel {
+  name: string
+  description?: string
+  status: TXpertWorkspaceStatus
+  settings?: TXpertWorkspaceSettings
+
+  // Many to one
+  ownerId: string
+  owner?: IUser
+
+  // One to many
+  xperts?: IXpert[]
+  environments?: IEnvironment[]
+
+  members?: IUser[]
+}
+
+export type TXpertWorkspaceSettings = {
+  //
+}
+export type TXpertWorkspaceStatus = 'active' | 'deprecated' | 'archived'
+
+export interface IBasePerWorkspaceEntityModel extends IBasePerTenantAndOrganizationEntityModel {
+  workspaceId?: string
+  workspace?: IXpertWorkspace
+  /**
+   * Publish date of latest
+   */
+  publishAt?: Date
+  /**
+   * Soft deleted
+   */
+  deletedAt?: Date
+}