@polka-codes/cli 0.10.23 → 0.10.25
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/bin.cjs +143854 -0
- package/dist/bin.d.ts +2 -0
- package/dist/bin.d.ts.map +1 -0
- package/dist/index.d.ts +1 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +9201 -9195
- package/dist/program.d.ts +3 -0
- package/dist/program.d.ts.map +1 -0
- package/package.json +6 -7
- package/cli.mjs +0 -2
- package/dist/ApiProviderConfig.js +0 -57
- package/dist/ApiProviderConfig.js.map +0 -1
- package/dist/ApiProviderConfig.test.js +0 -278
- package/dist/ApiProviderConfig.test.js.map +0 -1
- package/dist/agent/advanced-discovery.js +0 -604
- package/dist/agent/advanced-discovery.js.map +0 -1
- package/dist/agent/config.js +0 -139
- package/dist/agent/config.js.map +0 -1
- package/dist/agent/config.test.js +0 -140
- package/dist/agent/config.test.js.map +0 -1
- package/dist/agent/constants.js +0 -172
- package/dist/agent/constants.js.map +0 -1
- package/dist/agent/constants.test.js +0 -132
- package/dist/agent/constants.test.js.map +0 -1
- package/dist/agent/debug-logger.js +0 -281
- package/dist/agent/debug-logger.js.map +0 -1
- package/dist/agent/debug-logger.test.js +0 -294
- package/dist/agent/debug-logger.test.js.map +0 -1
- package/dist/agent/error-handling.js +0 -114
- package/dist/agent/error-handling.js.map +0 -1
- package/dist/agent/error-handling.test.js +0 -191
- package/dist/agent/error-handling.test.js.map +0 -1
- package/dist/agent/errors.js +0 -301
- package/dist/agent/errors.js.map +0 -1
- package/dist/agent/executor.js +0 -206
- package/dist/agent/executor.js.map +0 -1
- package/dist/agent/executor.test.js +0 -228
- package/dist/agent/executor.test.js.map +0 -1
- package/dist/agent/goal-decomposer.js +0 -281
- package/dist/agent/goal-decomposer.js.map +0 -1
- package/dist/agent/goal-decomposer.test.js +0 -225
- package/dist/agent/goal-decomposer.test.js.map +0 -1
- package/dist/agent/health-monitor.js +0 -89
- package/dist/agent/health-monitor.js.map +0 -1
- package/dist/agent/improvement-loop.js +0 -177
- package/dist/agent/improvement-loop.js.map +0 -1
- package/dist/agent/index.js +0 -38
- package/dist/agent/index.js.map +0 -1
- package/dist/agent/metrics.js +0 -158
- package/dist/agent/metrics.js.map +0 -1
- package/dist/agent/metrics.test.js +0 -311
- package/dist/agent/metrics.test.js.map +0 -1
- package/dist/agent/orchestrator.js +0 -438
- package/dist/agent/orchestrator.js.map +0 -1
- package/dist/agent/planner.js +0 -199
- package/dist/agent/planner.js.map +0 -1
- package/dist/agent/planner.test.js +0 -135
- package/dist/agent/planner.test.js.map +0 -1
- package/dist/agent/progress.js +0 -298
- package/dist/agent/progress.js.map +0 -1
- package/dist/agent/progress.test.js +0 -255
- package/dist/agent/progress.test.js.map +0 -1
- package/dist/agent/resource-monitor.js +0 -114
- package/dist/agent/resource-monitor.js.map +0 -1
- package/dist/agent/safety/approval.js +0 -178
- package/dist/agent/safety/approval.js.map +0 -1
- package/dist/agent/safety/approval.test.js +0 -142
- package/dist/agent/safety/approval.test.js.map +0 -1
- package/dist/agent/safety/checks.js +0 -152
- package/dist/agent/safety/checks.js.map +0 -1
- package/dist/agent/safety/checks.test.js +0 -89
- package/dist/agent/safety/checks.test.js.map +0 -1
- package/dist/agent/safety/interrupt.js +0 -92
- package/dist/agent/safety/interrupt.js.map +0 -1
- package/dist/agent/safety/interrupt.test.js +0 -53
- package/dist/agent/safety/interrupt.test.js.map +0 -1
- package/dist/agent/session.js +0 -117
- package/dist/agent/session.js.map +0 -1
- package/dist/agent/session.test.js +0 -213
- package/dist/agent/session.test.js.map +0 -1
- package/dist/agent/state-manager.js +0 -287
- package/dist/agent/state-manager.js.map +0 -1
- package/dist/agent/task-discovery.js +0 -433
- package/dist/agent/task-discovery.js.map +0 -1
- package/dist/agent/task-discovery.test.js +0 -40
- package/dist/agent/task-discovery.test.js.map +0 -1
- package/dist/agent/task-history.js +0 -128
- package/dist/agent/task-history.js.map +0 -1
- package/dist/agent/task-prioritizer.js +0 -286
- package/dist/agent/task-prioritizer.js.map +0 -1
- package/dist/agent/test-fixtures.js +0 -112
- package/dist/agent/test-fixtures.js.map +0 -1
- package/dist/agent/types.js +0 -15
- package/dist/agent/types.js.map +0 -1
- package/dist/agent/workflow-adapter.js +0 -219
- package/dist/agent/workflow-adapter.js.map +0 -1
- package/dist/agent/workflow-adapter.test.js +0 -11
- package/dist/agent/workflow-adapter.test.js.map +0 -1
- package/dist/agent/working-dir-discovery.js +0 -54
- package/dist/agent/working-dir-discovery.js.map +0 -1
- package/dist/agent/working-space.js +0 -484
- package/dist/agent/working-space.js.map +0 -1
- package/dist/agent/working-space.test.js +0 -436
- package/dist/agent/working-space.test.js.map +0 -1
- package/dist/api.js +0 -279
- package/dist/api.js.map +0 -1
- package/dist/builtin-commands.js +0 -20
- package/dist/builtin-commands.js.map +0 -1
- package/dist/commandSummarizer.js +0 -96
- package/dist/commandSummarizer.js.map +0 -1
- package/dist/commands/__tests__/command-detection.integration.test.js +0 -238
- package/dist/commands/__tests__/command-detection.integration.test.js.map +0 -1
- package/dist/commands/__tests__/script-generator.integration.test.js +0 -271
- package/dist/commands/__tests__/script-generator.integration.test.js.map +0 -1
- package/dist/commands/agent.js +0 -218
- package/dist/commands/agent.js.map +0 -1
- package/dist/commands/code.js +0 -102
- package/dist/commands/code.js.map +0 -1
- package/dist/commands/command.constants.js +0 -13
- package/dist/commands/command.constants.js.map +0 -1
- package/dist/commands/commit.js +0 -17
- package/dist/commands/commit.js.map +0 -1
- package/dist/commands/fix.js +0 -17
- package/dist/commands/fix.js.map +0 -1
- package/dist/commands/init.js +0 -501
- package/dist/commands/init.js.map +0 -1
- package/dist/commands/mcp-server.js +0 -122
- package/dist/commands/mcp-server.js.map +0 -1
- package/dist/commands/memory.js +0 -410
- package/dist/commands/memory.js.map +0 -1
- package/dist/commands/memory.test.js +0 -453
- package/dist/commands/memory.test.js.map +0 -1
- package/dist/commands/meta.js +0 -142
- package/dist/commands/meta.js.map +0 -1
- package/dist/commands/plan.js +0 -39
- package/dist/commands/plan.js.map +0 -1
- package/dist/commands/pr.js +0 -14
- package/dist/commands/pr.js.map +0 -1
- package/dist/commands/review-json.test.js +0 -33
- package/dist/commands/review-json.test.js.map +0 -1
- package/dist/commands/review.js +0 -137
- package/dist/commands/review.js.map +0 -1
- package/dist/commands/review.usage.test.js +0 -41
- package/dist/commands/review.usage.test.js.map +0 -1
- package/dist/commands/run.js +0 -107
- package/dist/commands/run.js.map +0 -1
- package/dist/commands/skills.js +0 -175
- package/dist/commands/skills.js.map +0 -1
- package/dist/commands/task.js +0 -34
- package/dist/commands/task.js.map +0 -1
- package/dist/commands/workflow.js +0 -117
- package/dist/commands/workflow.js.map +0 -1
- package/dist/configPrompt.js +0 -39
- package/dist/configPrompt.js.map +0 -1
- package/dist/env.js +0 -22
- package/dist/env.js.map +0 -1
- package/dist/errors.js +0 -132
- package/dist/errors.js.map +0 -1
- package/dist/errors.test.js +0 -21
- package/dist/errors.test.js.map +0 -1
- package/dist/file-attachments.js +0 -77
- package/dist/file-attachments.js.map +0 -1
- package/dist/file-attachments.test.js +0 -214
- package/dist/file-attachments.test.js.map +0 -1
- package/dist/getModel.js +0 -229
- package/dist/getModel.js.map +0 -1
- package/dist/getModel.test.js +0 -67
- package/dist/getModel.test.js.map +0 -1
- package/dist/getProviderOptions.js +0 -58
- package/dist/getProviderOptions.js.map +0 -1
- package/dist/git-operations.js +0 -363
- package/dist/git-operations.js.map +0 -1
- package/dist/git-operations.test.js +0 -81
- package/dist/git-operations.test.js.map +0 -1
- package/dist/index.js.map +0 -1
- package/dist/logger.js +0 -41
- package/dist/logger.js.map +0 -1
- package/dist/mcp/client.js +0 -11
- package/dist/mcp/client.js.map +0 -1
- package/dist/mcp/error-scenarios.test.js +0 -206
- package/dist/mcp/error-scenarios.test.js.map +0 -1
- package/dist/mcp/errors.js +0 -63
- package/dist/mcp/errors.js.map +0 -1
- package/dist/mcp/index.js +0 -9
- package/dist/mcp/index.js.map +0 -1
- package/dist/mcp/manager.js +0 -231
- package/dist/mcp/manager.js.map +0 -1
- package/dist/mcp/manager.test.js +0 -40
- package/dist/mcp/manager.test.js.map +0 -1
- package/dist/mcp/sdk-client.js +0 -166
- package/dist/mcp/sdk-client.js.map +0 -1
- package/dist/mcp/shared-types.js +0 -3
- package/dist/mcp/shared-types.js.map +0 -1
- package/dist/mcp/tools-integration.test.js +0 -198
- package/dist/mcp/tools-integration.test.js.map +0 -1
- package/dist/mcp/tools.js +0 -60
- package/dist/mcp/tools.js.map +0 -1
- package/dist/mcp/transport.js +0 -257
- package/dist/mcp/transport.js.map +0 -1
- package/dist/mcp/types.js +0 -3
- package/dist/mcp/types.js.map +0 -1
- package/dist/mcp-server/index.js +0 -5
- package/dist/mcp-server/index.js.map +0 -1
- package/dist/mcp-server/plan-formatting.test.js +0 -74
- package/dist/mcp-server/plan-formatting.test.js.map +0 -1
- package/dist/mcp-server/sdk-server.js +0 -102
- package/dist/mcp-server/sdk-server.js.map +0 -1
- package/dist/mcp-server/sdk-server.test.js +0 -363
- package/dist/mcp-server/sdk-server.test.js.map +0 -1
- package/dist/mcp-server/tools.js +0 -785
- package/dist/mcp-server/tools.js.map +0 -1
- package/dist/mcp-server/types.js +0 -3
- package/dist/mcp-server/types.js.map +0 -1
- package/dist/options.js +0 -85
- package/dist/options.js.map +0 -1
- package/dist/options.test.js +0 -177
- package/dist/options.test.js.map +0 -1
- package/dist/prices.js +0 -61
- package/dist/prices.js.map +0 -1
- package/dist/prices.test.js +0 -148
- package/dist/prices.test.js.map +0 -1
- package/dist/runWorkflow.js +0 -313
- package/dist/runWorkflow.js.map +0 -1
- package/dist/script/__tests__/execution.integration.test.js +0 -274
- package/dist/script/__tests__/execution.integration.test.js.map +0 -1
- package/dist/script/__tests__/runner.test.js +0 -176
- package/dist/script/__tests__/runner.test.js.map +0 -1
- package/dist/script/__tests__/validator.test.js +0 -180
- package/dist/script/__tests__/validator.test.js.map +0 -1
- package/dist/script/executor.js +0 -127
- package/dist/script/executor.js.map +0 -1
- package/dist/script/index.js +0 -4
- package/dist/script/index.js.map +0 -1
- package/dist/script/runner.js +0 -258
- package/dist/script/runner.js.map +0 -1
- package/dist/skillIntegration.js +0 -46
- package/dist/skillIntegration.js.map +0 -1
- package/dist/skillIntegration.test.js +0 -128
- package/dist/skillIntegration.test.js.map +0 -1
- package/dist/test/utils.js +0 -200
- package/dist/test/utils.js.map +0 -1
- package/dist/test/workflow-fixtures.js +0 -120
- package/dist/test/workflow-fixtures.js.map +0 -1
- package/dist/tool-implementations.js +0 -521
- package/dist/tool-implementations.js.map +0 -1
- package/dist/tool-implementations.skill-tools.test.js +0 -106
- package/dist/tool-implementations.skill-tools.test.js.map +0 -1
- package/dist/tools/getTodoItem.js +0 -33
- package/dist/tools/getTodoItem.js.map +0 -1
- package/dist/tools/gitDiff.js +0 -108
- package/dist/tools/gitDiff.js.map +0 -1
- package/dist/tools/index.js +0 -8
- package/dist/tools/index.js.map +0 -1
- package/dist/tools/listMemoryTopics.js +0 -24
- package/dist/tools/listMemoryTopics.js.map +0 -1
- package/dist/tools/listTodoItems.js +0 -35
- package/dist/tools/listTodoItems.js.map +0 -1
- package/dist/tools/listTodoItems.test.js +0 -89
- package/dist/tools/listTodoItems.test.js.map +0 -1
- package/dist/tools/readMemory.js +0 -33
- package/dist/tools/readMemory.js.map +0 -1
- package/dist/tools/updateMemory.js +0 -62
- package/dist/tools/updateMemory.js.map +0 -1
- package/dist/tools/updateMemory.test.js +0 -109
- package/dist/tools/updateMemory.test.js.map +0 -1
- package/dist/tools/updateTodoItem.js +0 -31
- package/dist/tools/updateTodoItem.js.map +0 -1
- package/dist/tools/utils/diffLineNumbers.js +0 -178
- package/dist/tools/utils/diffLineNumbers.js.map +0 -1
- package/dist/utils/cacheControl.js +0 -59
- package/dist/utils/cacheControl.js.map +0 -1
- package/dist/utils/cacheControl.test.js +0 -128
- package/dist/utils/cacheControl.test.js.map +0 -1
- package/dist/utils/command.js +0 -50
- package/dist/utils/command.js.map +0 -1
- package/dist/utils/shell.js +0 -56
- package/dist/utils/shell.js.map +0 -1
- package/dist/utils/userInput.js +0 -47
- package/dist/utils/userInput.js.map +0 -1
- package/dist/workflow-tools.js +0 -21
- package/dist/workflow-tools.js.map +0 -1
- package/dist/workflows/agent-builder.js +0 -90
- package/dist/workflows/agent-builder.js.map +0 -1
- package/dist/workflows/agent-builder.test.js +0 -115
- package/dist/workflows/agent-builder.test.js.map +0 -1
- package/dist/workflows/code.workflow.js +0 -145
- package/dist/workflows/code.workflow.js.map +0 -1
- package/dist/workflows/commit.workflow.js +0 -111
- package/dist/workflows/commit.workflow.js.map +0 -1
- package/dist/workflows/commit.workflow.test.js +0 -141
- package/dist/workflows/commit.workflow.test.js.map +0 -1
- package/dist/workflows/fix.workflow.js +0 -172
- package/dist/workflows/fix.workflow.js.map +0 -1
- package/dist/workflows/fix.workflow.test.js +0 -137
- package/dist/workflows/fix.workflow.test.js.map +0 -1
- package/dist/workflows/git-file-tools.js +0 -408
- package/dist/workflows/git-file-tools.js.map +0 -1
- package/dist/workflows/index.js +0 -12
- package/dist/workflows/index.js.map +0 -1
- package/dist/workflows/init-interactive.workflow.js +0 -198
- package/dist/workflows/init-interactive.workflow.js.map +0 -1
- package/dist/workflows/init.workflow.js +0 -41
- package/dist/workflows/init.workflow.js.map +0 -1
- package/dist/workflows/meta.workflow.js +0 -107
- package/dist/workflows/meta.workflow.js.map +0 -1
- package/dist/workflows/plan.workflow.js +0 -275
- package/dist/workflows/plan.workflow.js.map +0 -1
- package/dist/workflows/plan.workflow.test.js +0 -419
- package/dist/workflows/plan.workflow.test.js.map +0 -1
- package/dist/workflows/pr.workflow.js +0 -54
- package/dist/workflows/pr.workflow.js.map +0 -1
- package/dist/workflows/pr.workflow.test.js +0 -98
- package/dist/workflows/pr.workflow.test.js.map +0 -1
- package/dist/workflows/prompts/coder.js +0 -85
- package/dist/workflows/prompts/coder.js.map +0 -1
- package/dist/workflows/prompts/commit.js +0 -16
- package/dist/workflows/prompts/commit.js.map +0 -1
- package/dist/workflows/prompts/fix.js +0 -44
- package/dist/workflows/prompts/fix.js.map +0 -1
- package/dist/workflows/prompts/index.js +0 -10
- package/dist/workflows/prompts/index.js.map +0 -1
- package/dist/workflows/prompts/init.js +0 -48
- package/dist/workflows/prompts/init.js.map +0 -1
- package/dist/workflows/prompts/meta.js +0 -17
- package/dist/workflows/prompts/meta.js.map +0 -1
- package/dist/workflows/prompts/plan.js +0 -212
- package/dist/workflows/prompts/plan.js.map +0 -1
- package/dist/workflows/prompts/pr.js +0 -15
- package/dist/workflows/prompts/pr.js.map +0 -1
- package/dist/workflows/prompts/review.js +0 -145
- package/dist/workflows/prompts/review.js.map +0 -1
- package/dist/workflows/prompts/shared.js +0 -93
- package/dist/workflows/prompts/shared.js.map +0 -1
- package/dist/workflows/review.workflow.js +0 -357
- package/dist/workflows/review.workflow.js.map +0 -1
- package/dist/workflows/task.workflow.js +0 -47
- package/dist/workflows/task.workflow.js.map +0 -1
- package/dist/workflows/testing/helper.js +0 -41
- package/dist/workflows/testing/helper.js.map +0 -1
- package/dist/workflows/workflow.utils.js +0 -351
- package/dist/workflows/workflow.utils.js.map +0 -1
- package/dist/workflows/workflow.utils.test.js +0 -45
- package/dist/workflows/workflow.utils.test.js.map +0 -1
|
@@ -1,120 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Workflow Testing Fixtures
|
|
3
|
-
*
|
|
4
|
-
* Provides reusable fixtures for workflow testing with reduced mock usage
|
|
5
|
-
* and better type safety. Addresses P0 mock overuse issues from test review plan.
|
|
6
|
-
*/
|
|
7
|
-
import { mock } from 'bun:test';
|
|
8
|
-
/**
|
|
9
|
-
* Create a workflow test context with properly typed mocks
|
|
10
|
-
*
|
|
11
|
-
* This is a minimal fixture - only mocks what cannot be tested with real implementations.
|
|
12
|
-
* Following the test review plan's guidance:
|
|
13
|
-
* - Prefer real implementations over mocks
|
|
14
|
-
* - Use typed mocks, not 'any'
|
|
15
|
-
* - Maximum 3-5 mocks per test (enforced by this fixture's API)
|
|
16
|
-
*/
|
|
17
|
-
export function createWorkflowTestContext() {
|
|
18
|
-
// Only mock the tools that are actually needed for workflow testing
|
|
19
|
-
const tools = {
|
|
20
|
-
executeCommand: mock(),
|
|
21
|
-
input: mock(),
|
|
22
|
-
generateText: mock(),
|
|
23
|
-
taskEvent: mock(),
|
|
24
|
-
getMemoryContext: mock().mockResolvedValue(''),
|
|
25
|
-
updateMemory: mock(),
|
|
26
|
-
printChangeFile: mock(),
|
|
27
|
-
createCommit: mock(),
|
|
28
|
-
confirm: mock(),
|
|
29
|
-
select: mock(),
|
|
30
|
-
readFile: mock(),
|
|
31
|
-
writeToFile: mock(),
|
|
32
|
-
createPullRequest: mock(),
|
|
33
|
-
invokeTool: mock(),
|
|
34
|
-
};
|
|
35
|
-
const step = mock(async (_name, arg2, arg3) => {
|
|
36
|
-
const fn = typeof arg2 === 'function' ? arg2 : arg3;
|
|
37
|
-
if (typeof fn === 'function') {
|
|
38
|
-
return fn();
|
|
39
|
-
}
|
|
40
|
-
return undefined;
|
|
41
|
-
});
|
|
42
|
-
const logger = {
|
|
43
|
-
info: mock(() => { }),
|
|
44
|
-
error: mock(() => { }),
|
|
45
|
-
warn: mock(() => { }),
|
|
46
|
-
debug: mock(() => { }),
|
|
47
|
-
};
|
|
48
|
-
const context = {
|
|
49
|
-
tools,
|
|
50
|
-
step,
|
|
51
|
-
logger,
|
|
52
|
-
};
|
|
53
|
-
return { context, tools, step, logger };
|
|
54
|
-
}
|
|
55
|
-
/**
|
|
56
|
-
* Helper to set up successful command execution
|
|
57
|
-
* @param tools - Tools object from createWorkflowTestContext
|
|
58
|
-
* @param stdout - Standard output to return
|
|
59
|
-
* @param stderr - Standard error to return (default: '')
|
|
60
|
-
* @param exitCode - Exit code to return (default: 0)
|
|
61
|
-
*/
|
|
62
|
-
export function mockSuccessfulCommand(tools, stdout, stderr = '', exitCode = 0) {
|
|
63
|
-
tools.executeCommand.mockResolvedValue({ exitCode, stdout, stderr });
|
|
64
|
-
}
|
|
65
|
-
/**
|
|
66
|
-
* Helper to set up failed command execution
|
|
67
|
-
* @param tools - Tools object from createWorkflowTestContext
|
|
68
|
-
* @param stdout - Standard output to return (default: 'FAIL')
|
|
69
|
-
* @param stderr - Standard error to return (default: 'Error')
|
|
70
|
-
*/
|
|
71
|
-
export function mockFailedCommand(tools, stdout = 'FAIL', stderr = 'Error') {
|
|
72
|
-
tools.executeCommand.mockResolvedValue({ exitCode: 1, stdout, stderr });
|
|
73
|
-
}
|
|
74
|
-
/**
|
|
75
|
-
* Helper to set up agent response
|
|
76
|
-
* @param tools - Tools object from createWorkflowTestContext
|
|
77
|
-
* @param summary - Summary text from agent
|
|
78
|
-
* @param bailReason - Optional bail reason
|
|
79
|
-
*/
|
|
80
|
-
export function mockAgentResponse(tools, summary, bailReason) {
|
|
81
|
-
const content = JSON.stringify({ summary, bailReason: bailReason ?? null });
|
|
82
|
-
tools.generateText.mockResolvedValue([
|
|
83
|
-
{
|
|
84
|
-
role: 'assistant',
|
|
85
|
-
content: `\`\`\`json\n${content}\n\`\`\``,
|
|
86
|
-
},
|
|
87
|
-
]);
|
|
88
|
-
}
|
|
89
|
-
/**
|
|
90
|
-
* Helper to set up multiple command attempts (e.g., for retry logic)
|
|
91
|
-
* @param tools - Tools object from createWorkflowTestContext
|
|
92
|
-
* @param attempts - Array of exit codes for each attempt
|
|
93
|
-
*/
|
|
94
|
-
export function mockCommandAttempts(tools, attempts) {
|
|
95
|
-
for (const exitCode of attempts) {
|
|
96
|
-
tools.executeCommand.mockResolvedValueOnce({
|
|
97
|
-
exitCode,
|
|
98
|
-
stdout: exitCode === 0 ? 'PASS' : 'FAIL',
|
|
99
|
-
stderr: exitCode === 0 ? '' : 'Error',
|
|
100
|
-
});
|
|
101
|
-
}
|
|
102
|
-
}
|
|
103
|
-
/**
|
|
104
|
-
* Helper to verify exact number of tool calls (deprecated - use expect() directly)
|
|
105
|
-
* @param tools - Tools object from createWorkflowTestContext
|
|
106
|
-
* @param expectations - Object mapping tool names to expected call counts
|
|
107
|
-
*/
|
|
108
|
-
// Note: Not used - use expect(tool).toHaveBeenCalledTimes(count) directly in tests
|
|
109
|
-
/*
|
|
110
|
-
export function expectToolCalls(
|
|
111
|
-
tools: WorkflowTestContext['tools'],
|
|
112
|
-
expectations: Partial<Record<keyof WorkflowTestContext['tools'], number>>,
|
|
113
|
-
): void {
|
|
114
|
-
for (const [toolName, expectedCount] of Object.entries(expectations)) {
|
|
115
|
-
const tool = tools[toolName as keyof typeof tools]
|
|
116
|
-
expect(tool).toHaveBeenCalledTimes(expectedCount)
|
|
117
|
-
}
|
|
118
|
-
}
|
|
119
|
-
*/
|
|
120
|
-
//# sourceMappingURL=workflow-fixtures.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"workflow-fixtures.js","sourceRoot":"","sources":["../../src/test/workflow-fixtures.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,IAAI,EAAE,MAAM,UAAU,CAAA;AAmC/B;;;;;;;;GAQG;AACH,MAAM,UAAU,yBAAyB;IACvC,oEAAoE;IACpE,MAAM,KAAK,GAAG;QACZ,cAAc,EAAE,IAAI,EAAuE;QAC3F,KAAK,EAAE,IAAI,EAAyB;QACpC,YAAY,EAAE,IAAI,EAA2D;QAC7E,SAAS,EAAE,IAAI,EAAuB;QACtC,gBAAgB,EAAE,IAAI,EAAyB,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACrE,YAAY,EAAE,IAAI,EAAuB;QACzC,eAAe,EAAE,IAAI,EAA+D;QACpF,YAAY,EAAE,IAAI,EAAsC;QACxD,OAAO,EAAE,IAAI,EAA0B;QACvC,MAAM,EAAE,IAAI,EAAyB;QACrC,QAAQ,EAAE,IAAI,EAAgC;QAC9C,WAAW,EAAE,IAAI,EAAuB;QACxC,iBAAiB,EAAE,IAAI,EAAyD;QAChF,UAAU,EAAE,IAAI,EAAsB;KACvC,CAAA;IAED,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,EAAE,KAAa,EAAE,IAAa,EAAE,IAAa,EAAE,EAAE;QACtE,MAAM,EAAE,GAAG,OAAO,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAA;QACnD,IAAI,OAAO,EAAE,KAAK,UAAU,EAAE,CAAC;YAC7B,OAAO,EAAE,EAAE,CAAA;QACb,CAAC;QACD,OAAO,SAAS,CAAA;IAClB,CAAC,CAAC,CAAA;IAEF,MAAM,MAAM,GAAG;QACb,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;QACpB,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;QACrB,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;QACpB,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;KACtB,CAAA;IAED,MAAM,OAAO,GAAG;QACd,KAAK;QACL,IAAI;QACJ,MAAM;KAC4C,CAAA;IAEpD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,CAAA;AACzC,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,qBAAqB,CACnC,KAAmC,EACnC,MAAc,EACd,SAAiB,EAAE,EACnB,WAAmB,CAAC;IAEpB,KAAK,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAA;AACtE,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,iBAAiB,CAAC,KAAmC,EAAE,SAAiB,MAAM,EAAE,SAAiB,OAAO;IACtH,KAAK,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAA;AACzE,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,iBAAiB,CAAC,KAAmC,EAAE,OAAsB,EAAE,UAAmB;IAChH,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,IAAI,IAAI,EAAE,CAAC,CAAA;IAC3E,KAAK,CAAC,YAAY,CAAC,iBAAiB,CAAC;QACnC;YACE,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,eAAe,OAAO,UAAU;SAC1C;KACF,CAAC,CAAA;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CAAC,KAAmC,EAAE,QAAkB;IACzF,KAAK,MAAM,QAAQ,IAAI,QAAQ,EAAE,CAAC;QAChC,KAAK,CAAC,cAAc,CAAC,qBAAqB,CAAC;YACzC,QAAQ;YACR,MAAM,EAAE,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM;YACxC,MAAM,EAAE,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO;SACtC,CAAC,CAAA;IACJ,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,mFAAmF;AACnF;;;;;;;;;;EAUE"}
|
|
@@ -1,521 +0,0 @@
|
|
|
1
|
-
// generated by polka.codes
|
|
2
|
-
import { spawn, spawnSync } from 'node:child_process';
|
|
3
|
-
import { confirm as inquirerConfirm, select as inquirerSelect } from '@inquirer/prompts';
|
|
4
|
-
import { agentWorkflow, askFollowupQuestion, computeRateLimitBackoffSeconds, listSkills as coreListSkills, loadSkill as coreLoadSkill, readSkillFile as coreReadSkillFile, executeCommand as executeCommandTool, fetchUrl, fromJsonModelMessage, listFiles as listFilesTool, readBinaryFile, readFile as readFileTool, removeFile, renameFile, replaceInFile, SOURCE_ICONS, search, searchFiles, TaskEventKind, toJsonModelMessage, writeToFile as writeToFileTool, } from '@polka-codes/core';
|
|
5
|
-
import { streamText } from 'ai';
|
|
6
|
-
import { createProviderErrorFromStatus, MaxRetriesExceededError, ProviderTimeoutError, QuotaExceededError, UserCancelledError, } from './errors';
|
|
7
|
-
import { McpError } from './mcp/errors';
|
|
8
|
-
import { createSkillContext, generateSkillsSystemPrompt } from './skillIntegration';
|
|
9
|
-
import { getTodoItem as getTodoItemTool, gitDiff, listMemoryTopics as listMemoryTopicsTool, listTodoItems as listTodoItemsTool, readMemory as readMemoryTool, updateMemory as updateMemoryTool, updateTodoItem as updateTodoItemTool, } from './tools';
|
|
10
|
-
import { applyCacheControl } from './utils/cacheControl';
|
|
11
|
-
import { getUserInput } from './utils/userInput';
|
|
12
|
-
import { getLocalChanges } from './workflows/workflow.utils';
|
|
13
|
-
const allTools = [
|
|
14
|
-
askFollowupQuestion,
|
|
15
|
-
executeCommandTool,
|
|
16
|
-
fetchUrl,
|
|
17
|
-
listFilesTool,
|
|
18
|
-
readBinaryFile,
|
|
19
|
-
readFileTool,
|
|
20
|
-
readMemoryTool,
|
|
21
|
-
listMemoryTopicsTool,
|
|
22
|
-
updateMemoryTool,
|
|
23
|
-
removeFile,
|
|
24
|
-
renameFile,
|
|
25
|
-
replaceInFile,
|
|
26
|
-
search,
|
|
27
|
-
searchFiles,
|
|
28
|
-
writeToFileTool,
|
|
29
|
-
gitDiff,
|
|
30
|
-
getTodoItemTool,
|
|
31
|
-
listTodoItemsTool,
|
|
32
|
-
updateTodoItemTool,
|
|
33
|
-
];
|
|
34
|
-
export const toolHandlers = new Map(allTools.map((t) => [t.name, t]));
|
|
35
|
-
async function createPullRequest(input, _context) {
|
|
36
|
-
spawnSync('gh', ['pr', 'create', '--title', input.title, '--body', input.description], {
|
|
37
|
-
stdio: 'inherit',
|
|
38
|
-
});
|
|
39
|
-
return { title: input.title, description: input.description };
|
|
40
|
-
}
|
|
41
|
-
async function createCommit(input, _context) {
|
|
42
|
-
const result = spawnSync('git', ['commit', '-m', input.message], {
|
|
43
|
-
stdio: 'inherit',
|
|
44
|
-
});
|
|
45
|
-
if (result.status !== 0) {
|
|
46
|
-
throw new Error('Commit failed');
|
|
47
|
-
}
|
|
48
|
-
return { message: input.message };
|
|
49
|
-
}
|
|
50
|
-
async function printChangeFile(_input, context) {
|
|
51
|
-
const { stagedFiles, unstagedFiles } = getLocalChanges();
|
|
52
|
-
if (stagedFiles.length === 0 && unstagedFiles.length === 0) {
|
|
53
|
-
context.workflowContext.logger.info('No changes to commit.');
|
|
54
|
-
}
|
|
55
|
-
else {
|
|
56
|
-
if (stagedFiles.length > 0) {
|
|
57
|
-
context.workflowContext.logger.info('Staged files:');
|
|
58
|
-
for (const file of stagedFiles) {
|
|
59
|
-
context.workflowContext.logger.info(`- ${file.status}: ${file.path}`);
|
|
60
|
-
}
|
|
61
|
-
}
|
|
62
|
-
if (unstagedFiles.length > 0) {
|
|
63
|
-
context.workflowContext.logger.info('\nUnstaged files:');
|
|
64
|
-
for (const file of unstagedFiles) {
|
|
65
|
-
context.workflowContext.logger.info(`- ${file.status}: ${file.path}`);
|
|
66
|
-
}
|
|
67
|
-
}
|
|
68
|
-
}
|
|
69
|
-
return { stagedFiles, unstagedFiles };
|
|
70
|
-
}
|
|
71
|
-
async function confirm(input, context) {
|
|
72
|
-
if (context.yes) {
|
|
73
|
-
return true;
|
|
74
|
-
}
|
|
75
|
-
// to allow ora to fully stop the spinner so inquirer can takeover the cli window
|
|
76
|
-
await new Promise((resolve) => setTimeout(resolve, 50));
|
|
77
|
-
process.stderr.write('\u0007');
|
|
78
|
-
return await inquirerConfirm({ message: input.message });
|
|
79
|
-
}
|
|
80
|
-
async function input(input, context) {
|
|
81
|
-
if (context.yes) {
|
|
82
|
-
return input.default ?? '';
|
|
83
|
-
}
|
|
84
|
-
// to allow ora to fully stop the spinner so inquirer can takeover the cli window
|
|
85
|
-
await new Promise((resolve) => setTimeout(resolve, 50));
|
|
86
|
-
process.stderr.write('\u0007');
|
|
87
|
-
const result = await getUserInput(input.message, {
|
|
88
|
-
default: input.default,
|
|
89
|
-
});
|
|
90
|
-
if (result === undefined) {
|
|
91
|
-
throw new UserCancelledError();
|
|
92
|
-
}
|
|
93
|
-
return result;
|
|
94
|
-
}
|
|
95
|
-
async function select(input, context) {
|
|
96
|
-
if (context.yes) {
|
|
97
|
-
return input.choices[0].value;
|
|
98
|
-
}
|
|
99
|
-
// to allow ora to fully stop the spinner so inquirer can takeover the cli window
|
|
100
|
-
await new Promise((resolve) => setTimeout(resolve, 50));
|
|
101
|
-
process.stderr.write('\u0007');
|
|
102
|
-
return await inquirerSelect({ message: input.message, choices: input.choices });
|
|
103
|
-
}
|
|
104
|
-
async function executeCommand(input) {
|
|
105
|
-
return new Promise((resolve, reject) => {
|
|
106
|
-
// SECURITY: When shell: true, use args as separate parameters to prevent command injection
|
|
107
|
-
// If args are provided, we use them as separate spawn arguments even with shell: true
|
|
108
|
-
// This is safer than concatenating command + args into a single string
|
|
109
|
-
const child = input.shell === true
|
|
110
|
-
? input.args && input.args.length > 0
|
|
111
|
-
? // Use shell: true with command and args as separate parameters for safety
|
|
112
|
-
spawn(input.command, input.args, { shell: true, stdio: 'pipe' })
|
|
113
|
-
: // No args, just the command
|
|
114
|
-
spawn(input.command, { shell: true, stdio: 'pipe' })
|
|
115
|
-
: // shell: false or not specified - always use separate args
|
|
116
|
-
spawn(input.command, input.args, {
|
|
117
|
-
shell: false,
|
|
118
|
-
stdio: 'pipe',
|
|
119
|
-
});
|
|
120
|
-
let stdout = '';
|
|
121
|
-
let stderr = '';
|
|
122
|
-
let stdoutEnded = !child.stdout;
|
|
123
|
-
let stderrEnded = !child.stderr;
|
|
124
|
-
let closeEventFired = false;
|
|
125
|
-
let exitCode = null;
|
|
126
|
-
const checkAndResolve = () => {
|
|
127
|
-
if (stdoutEnded && stderrEnded && closeEventFired) {
|
|
128
|
-
resolve({ exitCode: exitCode ?? -1, stdout, stderr });
|
|
129
|
-
}
|
|
130
|
-
};
|
|
131
|
-
if (child.stdout) {
|
|
132
|
-
child.stdout.setEncoding('utf8');
|
|
133
|
-
child.stdout.on('data', (data) => {
|
|
134
|
-
if (input.pipe) {
|
|
135
|
-
process.stdout.write(data);
|
|
136
|
-
}
|
|
137
|
-
stdout += data;
|
|
138
|
-
});
|
|
139
|
-
child.stdout.on('end', () => {
|
|
140
|
-
stdoutEnded = true;
|
|
141
|
-
checkAndResolve();
|
|
142
|
-
});
|
|
143
|
-
}
|
|
144
|
-
if (child.stderr) {
|
|
145
|
-
child.stderr.setEncoding('utf8');
|
|
146
|
-
child.stderr.on('data', (data) => {
|
|
147
|
-
if (input.pipe) {
|
|
148
|
-
process.stderr.write(data);
|
|
149
|
-
}
|
|
150
|
-
stderr += data;
|
|
151
|
-
});
|
|
152
|
-
child.stderr.on('end', () => {
|
|
153
|
-
stderrEnded = true;
|
|
154
|
-
checkAndResolve();
|
|
155
|
-
});
|
|
156
|
-
}
|
|
157
|
-
child.on('close', (code) => {
|
|
158
|
-
exitCode = code;
|
|
159
|
-
closeEventFired = true;
|
|
160
|
-
checkAndResolve();
|
|
161
|
-
});
|
|
162
|
-
child.on('error', (err) => {
|
|
163
|
-
reject(err);
|
|
164
|
-
});
|
|
165
|
-
});
|
|
166
|
-
}
|
|
167
|
-
async function generateText(input, context) {
|
|
168
|
-
const { model, agentCallback } = context;
|
|
169
|
-
if (!model) {
|
|
170
|
-
throw new Error('Model not found in context');
|
|
171
|
-
}
|
|
172
|
-
// Check usage limits
|
|
173
|
-
const limitResult = context.parameters.usageMeter.isLimitExceeded();
|
|
174
|
-
if (limitResult.result) {
|
|
175
|
-
agentCallback?.({
|
|
176
|
-
kind: TaskEventKind.UsageExceeded,
|
|
177
|
-
});
|
|
178
|
-
throw new QuotaExceededError(model.provider, model.modelId, context.parameters.usageMeter.usage.cost, limitResult.maxCost);
|
|
179
|
-
}
|
|
180
|
-
const { retryCount = 5, requestTimeoutSeconds = 90 } = context.parameters;
|
|
181
|
-
// Convert messages and apply cache control
|
|
182
|
-
const messages = applyCacheControl(input.messages.map(fromJsonModelMessage), model.provider, model.modelId);
|
|
183
|
-
// Repetition detection constants
|
|
184
|
-
const REPETITION_DETECTION_WINDOW = 20;
|
|
185
|
-
const REPETITION_CHECK_HALF = 10;
|
|
186
|
-
const REPETITION_MIN_LENGTH = 20;
|
|
187
|
-
let lastError;
|
|
188
|
-
for (let i = 0; i < retryCount; i++) {
|
|
189
|
-
const abortController = new AbortController();
|
|
190
|
-
let timeout = setTimeout(() => abortController.abort(), requestTimeoutSeconds * 1000);
|
|
191
|
-
const lastOutputs = [];
|
|
192
|
-
let repetitionDetected = false;
|
|
193
|
-
const usageMeterOnFinishHandler = context.parameters.usageMeter.onFinishHandler(model);
|
|
194
|
-
try {
|
|
195
|
-
const stream = streamText({
|
|
196
|
-
model,
|
|
197
|
-
temperature: 0,
|
|
198
|
-
messages,
|
|
199
|
-
tools: input.tools,
|
|
200
|
-
async onChunk({ chunk }) {
|
|
201
|
-
clearTimeout(timeout);
|
|
202
|
-
timeout = setTimeout(() => abortController.abort(), requestTimeoutSeconds * 1000);
|
|
203
|
-
switch (chunk.type) {
|
|
204
|
-
case 'text-delta':
|
|
205
|
-
lastOutputs.push(chunk.text);
|
|
206
|
-
if (lastOutputs.length > REPETITION_DETECTION_WINDOW) {
|
|
207
|
-
lastOutputs.shift();
|
|
208
|
-
}
|
|
209
|
-
if (lastOutputs.length === REPETITION_DETECTION_WINDOW) {
|
|
210
|
-
const firstHalf = lastOutputs.slice(0, REPETITION_CHECK_HALF).join('');
|
|
211
|
-
const secondHalf = lastOutputs.slice(REPETITION_CHECK_HALF).join('');
|
|
212
|
-
if (firstHalf === secondHalf) {
|
|
213
|
-
if (firstHalf.length > REPETITION_MIN_LENGTH) {
|
|
214
|
-
repetitionDetected = true;
|
|
215
|
-
abortController.abort();
|
|
216
|
-
}
|
|
217
|
-
}
|
|
218
|
-
}
|
|
219
|
-
agentCallback?.({
|
|
220
|
-
kind: TaskEventKind.Text,
|
|
221
|
-
newText: chunk.text,
|
|
222
|
-
});
|
|
223
|
-
break;
|
|
224
|
-
case 'reasoning-delta':
|
|
225
|
-
agentCallback?.({
|
|
226
|
-
kind: TaskEventKind.Reasoning,
|
|
227
|
-
newText: chunk.text,
|
|
228
|
-
});
|
|
229
|
-
break;
|
|
230
|
-
}
|
|
231
|
-
},
|
|
232
|
-
onFinish(result) {
|
|
233
|
-
usageMeterOnFinishHandler?.(result);
|
|
234
|
-
},
|
|
235
|
-
providerOptions: context.parameters.providerOptions,
|
|
236
|
-
abortSignal: abortController.signal,
|
|
237
|
-
});
|
|
238
|
-
await stream.consumeStream({
|
|
239
|
-
onError: (error) => {
|
|
240
|
-
context.workflowContext.logger.error('Error in stream:', error);
|
|
241
|
-
lastError = error instanceof Error ? error : new Error(String(error));
|
|
242
|
-
},
|
|
243
|
-
});
|
|
244
|
-
const resp = await stream.response;
|
|
245
|
-
return resp.messages.map(toJsonModelMessage);
|
|
246
|
-
}
|
|
247
|
-
catch (error) {
|
|
248
|
-
// Handle AbortError (timeouts and repetition)
|
|
249
|
-
if (error instanceof Error && error.name === 'AbortError') {
|
|
250
|
-
if (repetitionDetected) {
|
|
251
|
-
context.workflowContext.logger.warn('Repetition detected, retrying...');
|
|
252
|
-
lastError = new Error('Repetition detected in model output');
|
|
253
|
-
continue;
|
|
254
|
-
}
|
|
255
|
-
// This is a timeout
|
|
256
|
-
context.workflowContext.logger.warn(`Request timed out after ${requestTimeoutSeconds} seconds, retrying...`);
|
|
257
|
-
lastError = new ProviderTimeoutError(model.provider, model.modelId, requestTimeoutSeconds);
|
|
258
|
-
continue;
|
|
259
|
-
}
|
|
260
|
-
// Handle HTTP errors
|
|
261
|
-
if (error && typeof error === 'object' && 'response' in error) {
|
|
262
|
-
const errorWithResponse = error;
|
|
263
|
-
const response = errorWithResponse.response;
|
|
264
|
-
const statusCode = response.status;
|
|
265
|
-
// Create appropriate error based on status code
|
|
266
|
-
const providerError = createProviderErrorFromStatus(model.provider, model.modelId, statusCode, error instanceof Error ? error : undefined);
|
|
267
|
-
// Only retry if error is retryable
|
|
268
|
-
if (providerError.retryable && (statusCode === 429 || statusCode >= 500)) {
|
|
269
|
-
context.workflowContext.logger.warn(`${providerError.message} (attempt ${i + 1}/${retryCount})`);
|
|
270
|
-
lastError = providerError;
|
|
271
|
-
const backoff = computeRateLimitBackoffSeconds(i);
|
|
272
|
-
context.workflowContext.logger.debug(`Waiting ${backoff}s before retry...`);
|
|
273
|
-
await new Promise((resolve) => setTimeout(resolve, backoff * 1000));
|
|
274
|
-
continue;
|
|
275
|
-
}
|
|
276
|
-
// Non-retryable HTTP error - throw immediately
|
|
277
|
-
throw providerError;
|
|
278
|
-
}
|
|
279
|
-
// Handle other errors
|
|
280
|
-
if (error instanceof Error) {
|
|
281
|
-
// Check for network errors
|
|
282
|
-
if (error.message.includes('ECONNREFUSED') || error.message.includes('ENOTFOUND') || error.message.includes('ETIMEDOUT')) {
|
|
283
|
-
context.workflowContext.logger.warn(`Network error: ${error.message}, retrying...`);
|
|
284
|
-
lastError = error;
|
|
285
|
-
continue;
|
|
286
|
-
}
|
|
287
|
-
// Unknown error - save and continue to retry
|
|
288
|
-
context.workflowContext.logger.warn(`Unexpected error: ${error.message}, retrying...`);
|
|
289
|
-
lastError = error;
|
|
290
|
-
continue;
|
|
291
|
-
}
|
|
292
|
-
// Non-Error throwable
|
|
293
|
-
lastError = new Error(String(error));
|
|
294
|
-
}
|
|
295
|
-
finally {
|
|
296
|
-
clearTimeout(timeout);
|
|
297
|
-
}
|
|
298
|
-
}
|
|
299
|
-
// All retries exhausted
|
|
300
|
-
throw new MaxRetriesExceededError(model.provider, model.modelId, retryCount, lastError || new Error('Unknown error'));
|
|
301
|
-
}
|
|
302
|
-
async function invokeTool(input, context) {
|
|
303
|
-
const tool = toolHandlers.get(input.toolName);
|
|
304
|
-
if (!tool) {
|
|
305
|
-
return {
|
|
306
|
-
success: false,
|
|
307
|
-
message: {
|
|
308
|
-
type: 'error-text',
|
|
309
|
-
value: `Tool not found: ${input.toolName}`,
|
|
310
|
-
},
|
|
311
|
-
};
|
|
312
|
-
}
|
|
313
|
-
try {
|
|
314
|
-
const result = await tool.handler(context.toolProvider, input.input);
|
|
315
|
-
return result;
|
|
316
|
-
}
|
|
317
|
-
catch (error) {
|
|
318
|
-
return {
|
|
319
|
-
success: false,
|
|
320
|
-
message: {
|
|
321
|
-
type: 'error-text',
|
|
322
|
-
value: error instanceof Error ? error.message : String(error),
|
|
323
|
-
},
|
|
324
|
-
};
|
|
325
|
-
}
|
|
326
|
-
}
|
|
327
|
-
async function taskEvent(input, context) {
|
|
328
|
-
await context.agentCallback?.(input);
|
|
329
|
-
}
|
|
330
|
-
async function getMemoryContext(_input, context) {
|
|
331
|
-
const provider = context.toolProvider;
|
|
332
|
-
const topics = await provider.listMemoryTopics();
|
|
333
|
-
const defaultContent = await provider.readMemory();
|
|
334
|
-
const contextParts = [];
|
|
335
|
-
if (topics?.length > 0) {
|
|
336
|
-
const topicList = topics.map((topic) => `- ${topic}`).join('\n');
|
|
337
|
-
contextParts.push(`<memory_topics>\n${topicList}\n</memory_topics>`);
|
|
338
|
-
}
|
|
339
|
-
if (defaultContent) {
|
|
340
|
-
contextParts.push(`<memory topic=":default:">\n${defaultContent}\n</memory>`);
|
|
341
|
-
}
|
|
342
|
-
return contextParts.join('\n');
|
|
343
|
-
}
|
|
344
|
-
async function listTodoItems(input, context) {
|
|
345
|
-
const provider = context.toolProvider;
|
|
346
|
-
return provider.listTodoItems(input.id, input.status);
|
|
347
|
-
}
|
|
348
|
-
async function getTodoItem(input, context) {
|
|
349
|
-
const provider = context.toolProvider;
|
|
350
|
-
return provider.getTodoItem(input.id);
|
|
351
|
-
}
|
|
352
|
-
async function updateTodoItem(input, context) {
|
|
353
|
-
const provider = context.toolProvider;
|
|
354
|
-
return provider.updateTodoItem(input);
|
|
355
|
-
}
|
|
356
|
-
async function runAgent(input, context) {
|
|
357
|
-
return await agentWorkflow(input, context.workflowContext);
|
|
358
|
-
}
|
|
359
|
-
/**
|
|
360
|
-
* Wrapper for skill tool operations that handles context validation and error handling
|
|
361
|
-
*/
|
|
362
|
-
async function withSkillContext(context, fn) {
|
|
363
|
-
if (!context.parameters.skillContext) {
|
|
364
|
-
return {
|
|
365
|
-
success: false,
|
|
366
|
-
message: { type: 'error-text', value: 'Skill context not initialized' },
|
|
367
|
-
};
|
|
368
|
-
}
|
|
369
|
-
try {
|
|
370
|
-
return await fn(context.parameters.skillContext);
|
|
371
|
-
}
|
|
372
|
-
catch (error) {
|
|
373
|
-
return {
|
|
374
|
-
success: false,
|
|
375
|
-
message: {
|
|
376
|
-
type: 'error-text',
|
|
377
|
-
value: error instanceof Error ? error.message : String(error),
|
|
378
|
-
},
|
|
379
|
-
};
|
|
380
|
-
}
|
|
381
|
-
}
|
|
382
|
-
async function loadSkill(input, context) {
|
|
383
|
-
return withSkillContext(context, async (skillContext) => {
|
|
384
|
-
const result = await coreLoadSkill(input, skillContext);
|
|
385
|
-
if (!result.success || !result.skill) {
|
|
386
|
-
return {
|
|
387
|
-
success: false,
|
|
388
|
-
message: { type: 'error-text', value: result.error ?? 'Failed to load skill' },
|
|
389
|
-
};
|
|
390
|
-
}
|
|
391
|
-
return {
|
|
392
|
-
success: true,
|
|
393
|
-
message: {
|
|
394
|
-
type: 'text',
|
|
395
|
-
value: `Loaded skill '${result.skill.name}':\n\n${result.skill.content}\n\nAvailable files: ${result.skill.availableFiles.join(', ')}${result.warnings && result.warnings.length > 0 ? `\n\nWarnings:\n${result.warnings.join('\n')}` : ''}`,
|
|
396
|
-
},
|
|
397
|
-
};
|
|
398
|
-
});
|
|
399
|
-
}
|
|
400
|
-
async function listSkills(input, context) {
|
|
401
|
-
return withSkillContext(context, async (skillContext) => {
|
|
402
|
-
const result = await coreListSkills(input, skillContext);
|
|
403
|
-
const skillsList = result.skills
|
|
404
|
-
.map((skill) => {
|
|
405
|
-
const sourceIcon = SOURCE_ICONS[skill.source];
|
|
406
|
-
return `${sourceIcon} **${skill.name}**: ${skill.description}`;
|
|
407
|
-
})
|
|
408
|
-
.join('\n');
|
|
409
|
-
return {
|
|
410
|
-
success: true,
|
|
411
|
-
message: {
|
|
412
|
-
type: 'text',
|
|
413
|
-
value: `Found ${result.total} skill${result.total === 1 ? '' : 's'}:\n\n${skillsList}`,
|
|
414
|
-
},
|
|
415
|
-
};
|
|
416
|
-
});
|
|
417
|
-
}
|
|
418
|
-
async function readSkillFile(input, context) {
|
|
419
|
-
return withSkillContext(context, async (skillContext) => {
|
|
420
|
-
const result = await coreReadSkillFile(input, skillContext);
|
|
421
|
-
if (!result.success || !result.content) {
|
|
422
|
-
return {
|
|
423
|
-
success: false,
|
|
424
|
-
message: { type: 'error-text', value: result.error ?? 'Failed to read skill file' },
|
|
425
|
-
};
|
|
426
|
-
}
|
|
427
|
-
return {
|
|
428
|
-
success: true,
|
|
429
|
-
message: { type: 'text', value: result.content },
|
|
430
|
-
};
|
|
431
|
-
});
|
|
432
|
-
}
|
|
433
|
-
const localToolHandlers = {
|
|
434
|
-
runAgent,
|
|
435
|
-
createPullRequest,
|
|
436
|
-
createCommit,
|
|
437
|
-
printChangeFile,
|
|
438
|
-
confirm,
|
|
439
|
-
input,
|
|
440
|
-
select,
|
|
441
|
-
// Note: writeToFile and readFile removed to use core implementations with safety features
|
|
442
|
-
executeCommand,
|
|
443
|
-
generateText,
|
|
444
|
-
invokeTool,
|
|
445
|
-
taskEvent,
|
|
446
|
-
getMemoryContext,
|
|
447
|
-
// Note: readMemory, listMemoryTopics, updateMemory removed to use proper tool implementations
|
|
448
|
-
// from tools/ directory that return proper XML formatted responses
|
|
449
|
-
listTodoItems,
|
|
450
|
-
getTodoItem,
|
|
451
|
-
updateTodoItem,
|
|
452
|
-
loadSkill,
|
|
453
|
-
listSkills,
|
|
454
|
-
readSkillFile,
|
|
455
|
-
};
|
|
456
|
-
// Export local tool names for Proxy existence checks
|
|
457
|
-
export const localToolNames = Object.keys(localToolHandlers);
|
|
458
|
-
export async function toolCall(toolCall, context) {
|
|
459
|
-
// Check localToolHandlers first
|
|
460
|
-
const handler = localToolHandlers[toolCall.tool];
|
|
461
|
-
if (handler) {
|
|
462
|
-
// Runtime tool selection means TypeScript can't infer the correct input type
|
|
463
|
-
// We use 'as never' to bypass the type check since the actual type is validated at runtime
|
|
464
|
-
return handler(toolCall.input, context);
|
|
465
|
-
}
|
|
466
|
-
// Check MCP tools
|
|
467
|
-
if (context.parameters.mcpManager?.hasTool(toolCall.tool)) {
|
|
468
|
-
const input = typeof toolCall.input === 'object' && toolCall.input !== null && !Array.isArray(toolCall.input) ? toolCall.input : {};
|
|
469
|
-
try {
|
|
470
|
-
const result = await context.parameters.mcpManager.callTool(toolCall.tool, input);
|
|
471
|
-
// Wrap result in ToolResponse format
|
|
472
|
-
const value = typeof result === 'string' ? result : result == null ? '' : JSON.stringify(result, null, 2);
|
|
473
|
-
return {
|
|
474
|
-
success: true,
|
|
475
|
-
message: {
|
|
476
|
-
type: 'text',
|
|
477
|
-
value,
|
|
478
|
-
},
|
|
479
|
-
};
|
|
480
|
-
}
|
|
481
|
-
catch (error) {
|
|
482
|
-
// McpError should bubble up to the workflow level for proper handling
|
|
483
|
-
if (error instanceof McpError) {
|
|
484
|
-
throw error;
|
|
485
|
-
}
|
|
486
|
-
// Other errors are returned as ToolResponse so the agent can see them and recover
|
|
487
|
-
return {
|
|
488
|
-
success: false,
|
|
489
|
-
message: {
|
|
490
|
-
type: 'error-text',
|
|
491
|
-
value: `Error: ${error instanceof Error ? error.message : String(error)}`,
|
|
492
|
-
},
|
|
493
|
-
};
|
|
494
|
-
}
|
|
495
|
-
}
|
|
496
|
-
// Check toolHandlers Map (for core/registered tools)
|
|
497
|
-
const toolHandler = toolHandlers.get(toolCall.tool);
|
|
498
|
-
if (toolHandler) {
|
|
499
|
-
// The tool handler expects specific input types based on the tool
|
|
500
|
-
// We cast to satisfy TypeScript since the exact type varies by tool
|
|
501
|
-
return toolHandler.handler(context.toolProvider, toolCall.input);
|
|
502
|
-
}
|
|
503
|
-
throw new Error(`Unknown tool: ${toolCall.tool}`);
|
|
504
|
-
}
|
|
505
|
-
/**
|
|
506
|
-
* Create skill context for agent execution
|
|
507
|
-
*/
|
|
508
|
-
export async function initializeSkillContext(cwd) {
|
|
509
|
-
return await createSkillContext(cwd);
|
|
510
|
-
}
|
|
511
|
-
/**
|
|
512
|
-
* Generate system prompt with available skills
|
|
513
|
-
*/
|
|
514
|
-
export function generateSystemPromptWithSkills(basePrompt, skillContext) {
|
|
515
|
-
if (!skillContext || skillContext.availableSkills.length === 0) {
|
|
516
|
-
return basePrompt;
|
|
517
|
-
}
|
|
518
|
-
const skillsPrompt = generateSkillsSystemPrompt(skillContext.availableSkills);
|
|
519
|
-
return basePrompt + skillsPrompt;
|
|
520
|
-
}
|
|
521
|
-
//# sourceMappingURL=tool-implementations.js.map
|