80 lines
4.0 KiB
TypeScript
80 lines
4.0 KiB
TypeScript
// *****************************************************************************
|
|
// Copyright (C) 2025 EclipseSource GmbH and others.
|
|
//
|
|
// This program and the accompanying materials are made available under the
|
|
// terms of the Eclipse Public License v. 2.0 which is available at
|
|
// http://www.eclipse.org/legal/epl-2.0.
|
|
//
|
|
// This Source Code may also be made available under the following Secondary
|
|
// Licenses when the conditions for such availability set forth in the Eclipse
|
|
// Public License v. 2.0 are satisfied: GNU General Public License, version 2
|
|
// with the GNU Classpath Exception which is available at
|
|
// https://www.gnu.org/software/classpath/license.html.
|
|
//
|
|
// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
|
|
// *****************************************************************************
|
|
|
|
import { MaybePromise, nls } from '@theia/core';
|
|
import { injectable } from '@theia/core/shared/inversify';
|
|
import {
|
|
AIVariable,
|
|
ResolvedAIVariable,
|
|
AIVariableContribution,
|
|
AIVariableService,
|
|
AIVariableResolutionRequest,
|
|
AIVariableContext,
|
|
AIVariableResolverWithVariableDependencies,
|
|
AIVariableArg
|
|
} from '@theia/ai-core';
|
|
import { ChatSessionContext } from '@theia/ai-chat';
|
|
import { TASK_CONTEXT_VARIABLE } from '@theia/ai-chat/lib/browser/task-context-variable';
|
|
import { TASK_CONTEXT_SUMMARY_VARIABLE_ID } from '../common/context-variables';
|
|
|
|
export const TASK_CONTEXT_SUMMARY_VARIABLE: AIVariable = {
|
|
id: TASK_CONTEXT_SUMMARY_VARIABLE_ID,
|
|
description: nls.localize('theia/ai/core/taskContextSummary/description', 'Resolves all task context items present in the session context.'),
|
|
name: TASK_CONTEXT_SUMMARY_VARIABLE_ID,
|
|
};
|
|
|
|
@injectable()
|
|
/**
|
|
* @class provides a summary of all TaskContextVariables in the context of a given session. Oriented towards use in prompts.
|
|
*/
|
|
export class TaskContextSummaryVariableContribution implements AIVariableContribution, AIVariableResolverWithVariableDependencies {
|
|
registerVariables(service: AIVariableService): void {
|
|
service.registerResolver(TASK_CONTEXT_SUMMARY_VARIABLE, this);
|
|
}
|
|
|
|
canResolve(request: AIVariableResolutionRequest, context: AIVariableContext): MaybePromise<number> {
|
|
return request.variable.name === TASK_CONTEXT_SUMMARY_VARIABLE.name ? 50 : 0;
|
|
}
|
|
|
|
async resolve(
|
|
request: AIVariableResolutionRequest,
|
|
context: AIVariableContext,
|
|
resolveDependency?: (variable: AIVariableArg) => Promise<ResolvedAIVariable | undefined>
|
|
): Promise<ResolvedAIVariable | undefined> {
|
|
if (!resolveDependency || !ChatSessionContext.is(context) || request.variable.name !== TASK_CONTEXT_SUMMARY_VARIABLE.name) { return undefined; }
|
|
const allSummaryRequests = context.model.context.getVariables().filter(candidate => candidate.variable.id === TASK_CONTEXT_VARIABLE.id);
|
|
if (!allSummaryRequests.length) { return { ...request, value: '' }; }
|
|
const allSummaries = await Promise.all(allSummaryRequests.map(summaryRequest => resolveDependency(summaryRequest).then(resolved => resolved?.value)));
|
|
const value = `# Current Task Context
|
|
|
|
The following task context defines the task you are expected to work on. It was explicitly provided by the user and represents your primary objective.
|
|
This context is authoritative: follow it unless you identify issues (e.g., outdated assumptions, technical conflicts, or unclear steps).
|
|
The task context may contain errors or outdated assumptions. You are expected to identify and report these, not blindly execute incorrect instructions.
|
|
Note: This context is a snapshot from the start of the conversation and will not update during this run.
|
|
When deviating from the plan:
|
|
- Explain the deviation and your reasoning before proceeding
|
|
- Summarize all deviations at the end of your response, and suggest updates to the task context if the plan needs revision
|
|
|
|
---
|
|
|
|
${allSummaries.map((content, index) => `## Task ${index + 1}\n\n${content}`).join('\n\n')}`;
|
|
return {
|
|
...request,
|
|
value
|
|
};
|
|
}
|
|
}
|