ros/claude-task-master
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: include tagInfo in AI service responses for MCP tools

Browse Source 
- Update all core functions that call AI services to extract and return tagInfo
- Update all direct functions to include tagInfo in MCP response data
- Fixes issue where add_task, expand_task, and other AI commands were not including current tag and available tags information
- tagInfo includes currentTag from state.json and availableTags list
- Ensures tagged task lists system information is properly propagated through the full chain: AI service -> core function -> direct function -> MCP client
This commit is contained in:
Eyal Toledano
2025-06-12 00:46:55 -04:00
parent 83d6405b17
commit 3888ef41d4
16 changed files with 39 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mcp-server/src/core/direct-functions/add-task.js
View File

@@ -95,6 +95,7 @@ export async function addTaskDirect(args, log, context = {}) {
let manualTaskData = null;
let newTaskId;
let telemetryData;
let tagInfo;
if (isManualCreation) {
// Create manual task data object
@@ -129,6 +130,7 @@ export async function addTaskDirect(args, log, context = {}) {
);
newTaskId = result.newTaskId;
telemetryData = result.telemetryData;
tagInfo = result.tagInfo;
} else {
// AI-driven task creation
log.info(
@@ -154,6 +156,7 @@ export async function addTaskDirect(args, log, context = {}) {
);
newTaskId = result.newTaskId;
telemetryData = result.telemetryData;
tagInfo = result.tagInfo;
}
// Restore normal logging
@@ -164,7 +167,8 @@ export async function addTaskDirect(args, log, context = {}) {
data: {
taskId: newTaskId,
message: `Successfully added new task #${newTaskId}`,
telemetryData: telemetryData
telemetryData: telemetryData,
tagInfo: tagInfo
}
};
} catch (error) {

3
mcp-server/src/core/direct-functions/analyze-task-complexity.js
View File

@@ -196,7 +196,8 @@ export async function analyzeTaskComplexityDirect(args, log, context = {}) {
lowComplexityTasks
},
fullReport: coreResult.report,
telemetryData: coreResult.telemetryData
telemetryData: coreResult.telemetryData,
tagInfo: coreResult.tagInfo
}
};
} catch (parseError) {

3
mcp-server/src/core/direct-functions/expand-task.js
View File

@@ -225,7 +225,8 @@ export async function expandTaskDirect(args, log, context = {}) {
task: coreResult.task,
subtasksAdded,
hasExistingSubtasks,
telemetryData: coreResult.telemetryData
telemetryData: coreResult.telemetryData,
tagInfo: coreResult.tagInfo
}
};
} catch (error) {

3
mcp-server/src/core/direct-functions/parse-prd.js
View File

@@ -170,7 +170,8 @@ export async function parsePRDDirect(args, log, context = {}) {
data: {
message: successMsg,
outputPath: result.tasksPath,
telemetryData: result.telemetryData
telemetryData: result.telemetryData,
tagInfo: result.tagInfo
}
};
} else {

3
mcp-server/src/core/direct-functions/research.js
View File

@@ -140,7 +140,8 @@ export async function researchDirect(args, log, context = {}) {
userPromptTokens: result.userPromptTokens,
totalInputTokens: result.totalInputTokens,
detailLevel: result.detailLevel,
telemetryData: result.telemetryData
telemetryData: result.telemetryData,
tagInfo: result.tagInfo
}
};
} catch (error) {

3
mcp-server/src/core/direct-functions/update-subtask-by-id.js
View File

@@ -139,7 +139,8 @@ export async function updateSubtaskByIdDirect(args, log, context = {}) {
subtask: coreResult.updatedSubtask,
tasksPath,
useResearch,
telemetryData: coreResult.telemetryData
telemetryData: coreResult.telemetryData,
tagInfo: coreResult.tagInfo
}
};
} catch (error) {

8
mcp-server/src/core/direct-functions/update-task-by-id.js
View File

@@ -76,7 +76,7 @@ export async function updateTaskByIdDirect(args, log, context = {}) {
} else {
// Parse as integer for main task IDs
taskId = parseInt(id, 10);
if (isNaN(taskId)) {
if (Number.isNaN(taskId)) {
const errorMessage = `Invalid task ID: ${id}. Task ID must be a positive integer or subtask ID (e.g., "5.2").`;
logWrapper.error(errorMessage);
return {
@@ -132,7 +132,8 @@ export async function updateTaskByIdDirect(args, log, context = {}) {
message: message,
taskId: taskId,
updated: false,
telemetryData: coreResult?.telemetryData
telemetryData: coreResult?.telemetryData,
tagInfo: coreResult?.tagInfo
}
};
}
@@ -149,7 +150,8 @@ export async function updateTaskByIdDirect(args, log, context = {}) {
useResearch: useResearch,
updated: true,
updatedTask: coreResult.updatedTask,
telemetryData: coreResult.telemetryData
telemetryData: coreResult.telemetryData,
tagInfo: coreResult.tagInfo
}
};
} catch (error) {

3
mcp-server/src/core/direct-functions/update-tasks.js
View File

@@ -90,7 +90,8 @@ export async function updateTasksDirect(args, log, context = {}) {
message: `Successfully updated ${result.updatedTasks.length} tasks.`,
tasksPath: tasksJsonPath,
updatedCount: result.updatedTasks.length,
telemetryData: result.telemetryData
telemetryData: result.telemetryData,
tagInfo: result.tagInfo
}
};
} else {

3
scripts/modules/task-manager/add-task.js
View File

@@ -624,7 +624,8 @@ async function addTask(
);
return {
newTaskId: newTaskId,
telemetryData: aiServiceResponse ? aiServiceResponse.telemetryData : null
telemetryData: aiServiceResponse ? aiServiceResponse.telemetryData : null,
tagInfo: aiServiceResponse ? aiServiceResponse.tagInfo : null
};
} catch (error) {
// Stop any loading indicator on error

3
scripts/modules/task-manager/analyze-task-complexity.js
View File

@@ -644,7 +644,8 @@ async function analyzeTaskComplexity(options, context = {}) {
return {
report: report,
telemetryData: aiServiceResponse?.telemetryData
telemetryData: aiServiceResponse?.telemetryData,
tagInfo: aiServiceResponse?.tagInfo
};
} catch (aiError) {
if (loadingIndicator) stopLoadingIndicator(loadingIndicator);

3
scripts/modules/task-manager/expand-task.js
View File

@@ -686,7 +686,8 @@ async function expandTask(
// Return the updated task object AND telemetry data
return {
task,
telemetryData: aiServiceResponse?.telemetryData
telemetryData: aiServiceResponse?.telemetryData,
tagInfo: aiServiceResponse?.tagInfo
};
} catch (error) {
// Catches errors from file reading, parsing, AI call etc.

3
scripts/modules/task-manager/parse-prd.js
View File

@@ -359,7 +359,8 @@ Guidelines:
return {
success: true,
tasksPath,
telemetryData: aiServiceResponse?.telemetryData
telemetryData: aiServiceResponse?.telemetryData,
tagInfo: aiServiceResponse?.tagInfo
};
} catch (error) {
report(`Error parsing PRD: ${error.message}`, 'error');

4
scripts/modules/task-manager/research.js
View File

@@ -244,6 +244,7 @@ async function performResearch(
const researchResult = aiResult.mainResult;
const telemetryData = aiResult.telemetryData;
const tagInfo = aiResult.tagInfo;
// Format and display results
if (outputFormat === 'text') {
@@ -285,7 +286,8 @@ async function performResearch(
userPromptTokens,
totalInputTokens,
detailLevel,
telemetryData
telemetryData,
tagInfo
};
} catch (error) {
logFn.error(`Research query failed: ${error.message}`);

3
scripts/modules/task-manager/update-subtask-by-id.js
View File

@@ -366,7 +366,8 @@ Output Requirements:
return {
updatedSubtask: updatedSubtask,
telemetryData: aiServiceResponse.telemetryData
telemetryData: aiServiceResponse.telemetryData,
tagInfo: aiServiceResponse.tagInfo
};
} catch (error) {
if (outputFormat === 'text' && loadingIndicator) {

3
scripts/modules/task-manager/update-task-by-id.js
View File

@@ -535,7 +535,8 @@ The changes described in the prompt should be thoughtfully applied to make the t
// --- Return Success with Telemetry ---
return {
updatedTask: updatedTask, // Return the updated task object
telemetryData: aiServiceResponse.telemetryData // <<< ADD telemetryData
telemetryData: aiServiceResponse.telemetryData, // <<< ADD telemetryData
tagInfo: aiServiceResponse.tagInfo
};
} catch (error) {
// Catch errors from generateTextService

3
scripts/modules/task-manager/update-tasks.js
View File

@@ -486,7 +486,8 @@ The changes described in the prompt should be applied to ALL tasks in the list.`
return {
success: true,
updatedTasks: parsedUpdatedTasks,
telemetryData: aiServiceResponse.telemetryData
telemetryData: aiServiceResponse.telemetryData,
tagInfo: aiServiceResponse.tagInfo
};
} catch (error) {
if (loadingIndicator) stopLoadingIndicator(loadingIndicator);