feat: upgrade core agent workflow and system utilities
This commit is contained in:
+1
-27
@@ -1,5 +1,5 @@
|
||||
import * as vscode from 'vscode';
|
||||
import { MultiAgentStageConfig, getConfig } from '../config';
|
||||
import { getConfig } from '../config';
|
||||
|
||||
export abstract class BaseAgent {
|
||||
constructor(protected readonly modelName: string) {}
|
||||
@@ -145,29 +145,3 @@ Your goal is to produce a state-of-the-art final report that wows the user.
|
||||
return this.callLLM(this.persona, wrappedInput, signal);
|
||||
}
|
||||
}
|
||||
|
||||
export class ConfigurableWorkflowAgent extends BaseAgent {
|
||||
constructor(
|
||||
modelName: string,
|
||||
private readonly stage: MultiAgentStageConfig
|
||||
) {
|
||||
super(modelName);
|
||||
}
|
||||
|
||||
async execute(input: string, context?: string, signal?: AbortSignal): Promise<string> {
|
||||
const wrappedInput = [
|
||||
`### Stage: ${this.stage.name}`,
|
||||
'',
|
||||
'### Available Context',
|
||||
context || 'No specific context available.',
|
||||
'',
|
||||
'### Stage Input',
|
||||
input,
|
||||
'',
|
||||
'### Mission',
|
||||
'Complete this stage according to your role instructions. Be concise, concrete, and preserve details needed by later stages.'
|
||||
].join('\n');
|
||||
|
||||
return this.callLLM(this.stage.prompt, wrappedInput, signal);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user