mirror of
https://github.com/langgenius/dify.git
synced 2026-07-21 18:58:35 +08:00
97 lines
3.4 KiB
TypeScript
97 lines
3.4 KiB
TypeScript
import type { ActiveContext } from '@/auth/hosts'
|
|
import type { AppInfoCache } from '@/cache/app-info'
|
|
import type { HttpClient } from '@/http/types'
|
|
import type { IOStreams } from '@/sys/io/streams'
|
|
import { AppMetaClient } from '@/api/app-meta'
|
|
import { selectAppReader } from '@/api/app-reader'
|
|
import { AppRunClient } from '@/api/app-run'
|
|
import { FileUploadClient } from '@/api/file-upload'
|
|
import { pickStrategy } from '@/commands/run/app/_strategies/index'
|
|
import { BaseError, HttpClientError } from '@/errors/base'
|
|
import { ErrorCode } from '@/errors/codes'
|
|
import { processExit } from '@/sys/index'
|
|
import { FieldInfo } from '@/types/app-meta'
|
|
import { resolveFileInputs } from './file-flags.js'
|
|
import { RUN_MODES } from './handlers.js'
|
|
import { resolveInputs, TEXT_FORMATS } from './input-flags.js'
|
|
|
|
export type RunAppOptions = {
|
|
readonly appId: string
|
|
readonly message?: string
|
|
readonly inputs?: Readonly<Record<string, unknown>>
|
|
readonly inputsJson?: string
|
|
readonly inputsFile?: string
|
|
readonly files?: readonly string[]
|
|
readonly conversationId?: string
|
|
readonly workflowId?: string
|
|
readonly workspace?: string
|
|
readonly format?: string
|
|
readonly stream?: boolean
|
|
readonly think?: boolean
|
|
readonly retryOnRateLimit?: boolean
|
|
}
|
|
|
|
export type RunAppDeps = {
|
|
readonly active: ActiveContext
|
|
readonly http: HttpClient
|
|
readonly host: string
|
|
readonly io: IOStreams
|
|
readonly cache?: AppInfoCache
|
|
readonly envLookup?: (k: string) => string | undefined
|
|
readonly exit?: (code: number) => never
|
|
}
|
|
|
|
export async function runApp(opts: RunAppOptions, deps: RunAppDeps): Promise<void> {
|
|
const apps = selectAppReader(deps.active, deps.http)
|
|
const meta = new AppMetaClient({ apps, host: deps.host, cache: deps.cache })
|
|
|
|
try {
|
|
await executeRun(opts, deps, meta)
|
|
}
|
|
catch (err) {
|
|
if (err instanceof HttpClientError && err.httpStatus === 422) {
|
|
await meta.invalidate(opts.appId)
|
|
throw err.withHint('app metadata cache cleared — if the app was recently republished, run the command again')
|
|
}
|
|
throw err
|
|
}
|
|
}
|
|
|
|
async function executeRun(
|
|
opts: RunAppOptions,
|
|
deps: RunAppDeps,
|
|
meta: AppMetaClient,
|
|
): Promise<void> {
|
|
const m = await meta.get(opts.appId, [FieldInfo])
|
|
const mode = m.info?.mode ?? ''
|
|
if (mode === '')
|
|
throw new Error(`app ${opts.appId}: mode missing from app metadata`)
|
|
|
|
if (mode === RUN_MODES.Workflow && opts.message !== undefined && opts.message !== '') {
|
|
throw new BaseError({
|
|
code: ErrorCode.UsageInvalidFlag,
|
|
message: 'workflow apps do not accept a positional message',
|
|
hint: 'pass workflow inputs via --inputs \'{"key":"value"}\'',
|
|
})
|
|
}
|
|
|
|
const inputs = await resolveInputs(opts.inputsJson, opts.inputsFile, opts.inputs)
|
|
if (opts.files !== undefined && opts.files.length > 0) {
|
|
const uploadClient = new FileUploadClient(deps.http)
|
|
const fileInputs = await resolveFileInputs(
|
|
opts.appId,
|
|
opts.files,
|
|
(appId, path) => uploadClient.upload(appId, path),
|
|
)
|
|
Object.assign(inputs, fileInputs)
|
|
}
|
|
const format = opts.format ?? ''
|
|
const isText = TEXT_FORMATS.has(format)
|
|
const livePrint = opts.stream === true
|
|
const runClient = new AppRunClient(deps.http)
|
|
|
|
const exit = deps.exit ?? processExit
|
|
const ctx = { opts: { ...opts, inputs }, deps, mode, format, isText, livePrint, runClient, exit, think: opts.think ?? false }
|
|
await pickStrategy(isText, livePrint).execute(ctx)
|
|
}
|