39 lines
1.4 KiB
TypeScript
39 lines
1.4 KiB
TypeScript
export type RalphLoopStrategy = "reset" | "continue"
|
|
|
|
export type ParsedRalphLoopArguments = {
|
|
prompt: string
|
|
maxIterations?: number
|
|
completionPromise?: string
|
|
strategy?: RalphLoopStrategy
|
|
}
|
|
|
|
const DEFAULT_PROMPT = "Complete the task as instructed"
|
|
|
|
export function parseRalphLoopArguments(rawArguments: string): ParsedRalphLoopArguments {
|
|
const taskMatch = rawArguments.match(/^("|')(.+?)\1/)
|
|
const promptCandidate =
|
|
taskMatch?.[2] ??
|
|
(rawArguments.startsWith("--")
|
|
? ""
|
|
: rawArguments.split(/\s+--/)[0]?.trim() ?? "")
|
|
const prompt = promptCandidate || DEFAULT_PROMPT
|
|
|
|
const maxIterationMatch = rawArguments.match(/--max-iterations=(\d+)/i)
|
|
const maxIterationsRaw = maxIterationMatch?.[1]
|
|
const completionPromiseQuoted = rawArguments.match(/--completion-promise=("|')(.+?)\1/i)
|
|
const completionPromiseUnquoted = rawArguments.match(/--completion-promise=([^\s\"']+)/i)
|
|
const completionPromise = completionPromiseQuoted?.[2] ?? completionPromiseUnquoted?.[1]
|
|
const strategyMatch = rawArguments.match(/--strategy=(reset|continue)/i)
|
|
const strategyValue = strategyMatch?.[1]?.toLowerCase()
|
|
|
|
return {
|
|
prompt,
|
|
maxIterations: maxIterationsRaw ? Number.parseInt(maxIterationsRaw, 10) : undefined,
|
|
completionPromise,
|
|
strategy:
|
|
strategyValue === "reset" || strategyValue === "continue"
|
|
? (strategyValue as RalphLoopStrategy)
|
|
: undefined,
|
|
}
|
|
}
|