@polka-codes/cli 0.10.23 → 0.10.24
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/index.js +1 -1
- package/package.json +3 -3
- 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,419 +0,0 @@
|
|
|
1
|
-
import { afterEach, describe, expect, mock, test } from 'bun:test';
|
|
2
|
-
import { UserCancelledError } from '../errors';
|
|
3
|
-
import { planWorkflow } from './plan.workflow';
|
|
4
|
-
import { createTestProxy } from './testing/helper';
|
|
5
|
-
describe('planWorkflow', () => {
|
|
6
|
-
let assert;
|
|
7
|
-
afterEach(() => {
|
|
8
|
-
assert?.();
|
|
9
|
-
mock.restore();
|
|
10
|
-
});
|
|
11
|
-
test('should generate a plan and save it', async () => {
|
|
12
|
-
const task = 'Create a new component';
|
|
13
|
-
const generatedPlan = '1. Create the component file.';
|
|
14
|
-
const savePath = '.plans/test.md';
|
|
15
|
-
const { context, assert: proxyAssert, logger, } = createTestProxy([
|
|
16
|
-
{
|
|
17
|
-
toolName: 'generateText',
|
|
18
|
-
args: expect.anything(),
|
|
19
|
-
returnValue: [
|
|
20
|
-
{
|
|
21
|
-
role: 'assistant',
|
|
22
|
-
content: JSON.stringify({ plan: generatedPlan }),
|
|
23
|
-
},
|
|
24
|
-
],
|
|
25
|
-
},
|
|
26
|
-
{
|
|
27
|
-
toolName: 'updateMemory',
|
|
28
|
-
args: expect.anything(),
|
|
29
|
-
returnValue: undefined,
|
|
30
|
-
},
|
|
31
|
-
{
|
|
32
|
-
toolName: 'select',
|
|
33
|
-
args: {
|
|
34
|
-
message: 'What do you want to do?',
|
|
35
|
-
choices: expect.anything(),
|
|
36
|
-
},
|
|
37
|
-
returnValue: 'save',
|
|
38
|
-
},
|
|
39
|
-
{
|
|
40
|
-
toolName: 'input',
|
|
41
|
-
args: {
|
|
42
|
-
message: 'Where do you want to save the plan?',
|
|
43
|
-
default: expect.any(String),
|
|
44
|
-
},
|
|
45
|
-
returnValue: savePath,
|
|
46
|
-
},
|
|
47
|
-
{
|
|
48
|
-
toolName: 'writeToFile',
|
|
49
|
-
args: { path: savePath, content: generatedPlan },
|
|
50
|
-
returnValue: undefined,
|
|
51
|
-
},
|
|
52
|
-
]);
|
|
53
|
-
assert = proxyAssert;
|
|
54
|
-
const result = await planWorkflow({ task, mode: 'interactive', interactive: true, additionalTools: {} }, context);
|
|
55
|
-
expect(result.plan).toBe(generatedPlan);
|
|
56
|
-
expect(logger.info).toHaveBeenCalledWith(`Plan saved to ${savePath}`);
|
|
57
|
-
});
|
|
58
|
-
test('should handle user feedback and regenerate plan', async () => {
|
|
59
|
-
const task = 'Create a new component';
|
|
60
|
-
const initialPlan = '1. Create the component file.';
|
|
61
|
-
const feedback = 'Add a test file.';
|
|
62
|
-
const regeneratedPlan = '1. Create the component file.\n2. Create a test file.';
|
|
63
|
-
const { context, assert: proxyAssert, logger, } = createTestProxy([
|
|
64
|
-
{
|
|
65
|
-
toolName: 'generateText',
|
|
66
|
-
args: expect.anything(),
|
|
67
|
-
returnValue: [
|
|
68
|
-
{
|
|
69
|
-
role: 'assistant',
|
|
70
|
-
content: `\`\`\`json
|
|
71
|
-
${JSON.stringify({ plan: initialPlan })}
|
|
72
|
-
\`\`\``,
|
|
73
|
-
},
|
|
74
|
-
],
|
|
75
|
-
},
|
|
76
|
-
{
|
|
77
|
-
toolName: 'updateMemory',
|
|
78
|
-
args: expect.anything(),
|
|
79
|
-
returnValue: undefined,
|
|
80
|
-
},
|
|
81
|
-
{
|
|
82
|
-
toolName: 'select',
|
|
83
|
-
args: {
|
|
84
|
-
message: 'What do you want to do?',
|
|
85
|
-
choices: expect.anything(),
|
|
86
|
-
},
|
|
87
|
-
returnValue: 'feedback',
|
|
88
|
-
},
|
|
89
|
-
{
|
|
90
|
-
toolName: 'input',
|
|
91
|
-
args: {
|
|
92
|
-
message: 'What changes do you want to make?',
|
|
93
|
-
},
|
|
94
|
-
returnValue: feedback,
|
|
95
|
-
},
|
|
96
|
-
{
|
|
97
|
-
toolName: 'generateText',
|
|
98
|
-
args: expect.anything(),
|
|
99
|
-
returnValue: [
|
|
100
|
-
{
|
|
101
|
-
role: 'assistant',
|
|
102
|
-
content: `\`\`\`json
|
|
103
|
-
${JSON.stringify({ plan: regeneratedPlan })}
|
|
104
|
-
\`\`\``,
|
|
105
|
-
},
|
|
106
|
-
],
|
|
107
|
-
},
|
|
108
|
-
{
|
|
109
|
-
toolName: 'updateMemory',
|
|
110
|
-
args: expect.anything(),
|
|
111
|
-
returnValue: undefined,
|
|
112
|
-
},
|
|
113
|
-
{
|
|
114
|
-
toolName: 'select',
|
|
115
|
-
args: {
|
|
116
|
-
message: 'What do you want to do?',
|
|
117
|
-
choices: expect.anything(),
|
|
118
|
-
},
|
|
119
|
-
returnValue: 'save',
|
|
120
|
-
},
|
|
121
|
-
{
|
|
122
|
-
toolName: 'input',
|
|
123
|
-
args: expect.anything(),
|
|
124
|
-
returnValue: '.plans/test.md',
|
|
125
|
-
},
|
|
126
|
-
{
|
|
127
|
-
toolName: 'writeToFile',
|
|
128
|
-
args: expect.anything(),
|
|
129
|
-
returnValue: undefined,
|
|
130
|
-
},
|
|
131
|
-
]);
|
|
132
|
-
assert = proxyAssert;
|
|
133
|
-
const result = await planWorkflow({ task, mode: 'interactive', interactive: true, additionalTools: {} }, context);
|
|
134
|
-
expect(result.plan).toBe(regeneratedPlan);
|
|
135
|
-
expect(logger.info).toHaveBeenCalledWith('Plan saved to .plans/test.md');
|
|
136
|
-
});
|
|
137
|
-
test('should exit the workflow', async () => {
|
|
138
|
-
const task = 'Create a new component';
|
|
139
|
-
const generatedPlan = '1. Create the component file.';
|
|
140
|
-
const { context, assert: proxyAssert } = createTestProxy([
|
|
141
|
-
{
|
|
142
|
-
toolName: 'generateText',
|
|
143
|
-
args: expect.anything(),
|
|
144
|
-
returnValue: [
|
|
145
|
-
{
|
|
146
|
-
role: 'assistant',
|
|
147
|
-
content: `\`\`\`json
|
|
148
|
-
${JSON.stringify({ plan: generatedPlan })}
|
|
149
|
-
\`\`\``,
|
|
150
|
-
},
|
|
151
|
-
],
|
|
152
|
-
},
|
|
153
|
-
{
|
|
154
|
-
toolName: 'updateMemory',
|
|
155
|
-
args: expect.anything(),
|
|
156
|
-
returnValue: undefined,
|
|
157
|
-
},
|
|
158
|
-
{
|
|
159
|
-
toolName: 'select',
|
|
160
|
-
args: {
|
|
161
|
-
message: 'What do you want to do?',
|
|
162
|
-
choices: expect.anything(),
|
|
163
|
-
},
|
|
164
|
-
returnValue: 'exit',
|
|
165
|
-
},
|
|
166
|
-
]);
|
|
167
|
-
assert = proxyAssert;
|
|
168
|
-
const result = await planWorkflow({ task, mode: 'interactive', interactive: true, additionalTools: {} }, context);
|
|
169
|
-
expect(result.plan).toBe(generatedPlan);
|
|
170
|
-
});
|
|
171
|
-
test('should run in non-interactive mode', async () => {
|
|
172
|
-
const task = 'Create a new component';
|
|
173
|
-
const generatedPlan = '1. Create the component file.';
|
|
174
|
-
const { context, assert: proxyAssert } = createTestProxy([
|
|
175
|
-
{
|
|
176
|
-
toolName: 'generateText',
|
|
177
|
-
args: expect.any(Object),
|
|
178
|
-
returnValue: [
|
|
179
|
-
{
|
|
180
|
-
role: 'assistant',
|
|
181
|
-
content: `\`\`\`json
|
|
182
|
-
${JSON.stringify({ plan: generatedPlan })}
|
|
183
|
-
\`\`\``,
|
|
184
|
-
},
|
|
185
|
-
],
|
|
186
|
-
},
|
|
187
|
-
{
|
|
188
|
-
toolName: 'updateMemory',
|
|
189
|
-
args: expect.anything(),
|
|
190
|
-
returnValue: undefined,
|
|
191
|
-
},
|
|
192
|
-
]);
|
|
193
|
-
assert = proxyAssert;
|
|
194
|
-
const result = await planWorkflow({ task, mode: 'noninteractive', interactive: false, additionalTools: {} }, context);
|
|
195
|
-
expect(result.plan).toBe(generatedPlan);
|
|
196
|
-
});
|
|
197
|
-
test('should run in confirm mode and be accepted', async () => {
|
|
198
|
-
const task = 'Create a new component';
|
|
199
|
-
const generatedPlan = '1. Create the component file.';
|
|
200
|
-
const { context, assert: proxyAssert } = createTestProxy([
|
|
201
|
-
{
|
|
202
|
-
toolName: 'generateText',
|
|
203
|
-
args: expect.any(Object),
|
|
204
|
-
returnValue: [
|
|
205
|
-
{
|
|
206
|
-
role: 'assistant',
|
|
207
|
-
content: `\`\`\`json
|
|
208
|
-
${JSON.stringify({ plan: generatedPlan })}
|
|
209
|
-
\`\`\``,
|
|
210
|
-
},
|
|
211
|
-
],
|
|
212
|
-
},
|
|
213
|
-
{
|
|
214
|
-
toolName: 'updateMemory',
|
|
215
|
-
args: expect.anything(),
|
|
216
|
-
returnValue: undefined,
|
|
217
|
-
},
|
|
218
|
-
{
|
|
219
|
-
toolName: 'input',
|
|
220
|
-
args: {
|
|
221
|
-
message: 'Do you approve this plan and want to proceed with implementation? (leave blank to approve, or enter feedback)',
|
|
222
|
-
},
|
|
223
|
-
returnValue: '',
|
|
224
|
-
},
|
|
225
|
-
]);
|
|
226
|
-
assert = proxyAssert;
|
|
227
|
-
const result = await planWorkflow({ task, mode: 'confirm', interactive: true, additionalTools: {} }, context);
|
|
228
|
-
expect(result.plan).toBe(generatedPlan);
|
|
229
|
-
});
|
|
230
|
-
test('should run in confirm mode and be rejected', async () => {
|
|
231
|
-
const task = 'Create a new component';
|
|
232
|
-
const generatedPlan = '1. Create the component file.';
|
|
233
|
-
const { context, assert: proxyAssert } = createTestProxy([
|
|
234
|
-
{
|
|
235
|
-
toolName: 'generateText',
|
|
236
|
-
args: expect.any(Object),
|
|
237
|
-
returnValue: [
|
|
238
|
-
{
|
|
239
|
-
role: 'assistant',
|
|
240
|
-
content: `\`\`\`json
|
|
241
|
-
${JSON.stringify({ plan: generatedPlan })}
|
|
242
|
-
\`\`\``,
|
|
243
|
-
},
|
|
244
|
-
],
|
|
245
|
-
},
|
|
246
|
-
{
|
|
247
|
-
toolName: 'updateMemory',
|
|
248
|
-
args: expect.anything(),
|
|
249
|
-
returnValue: undefined,
|
|
250
|
-
},
|
|
251
|
-
{
|
|
252
|
-
toolName: 'input',
|
|
253
|
-
args: {
|
|
254
|
-
message: 'Do you approve this plan and want to proceed with implementation? (leave blank to approve, or enter feedback)',
|
|
255
|
-
},
|
|
256
|
-
returnValue: new UserCancelledError(),
|
|
257
|
-
isError: true,
|
|
258
|
-
},
|
|
259
|
-
]);
|
|
260
|
-
assert = proxyAssert;
|
|
261
|
-
await expect(planWorkflow({ task, mode: 'confirm', interactive: true, additionalTools: {} }, context)).rejects.toThrow(UserCancelledError);
|
|
262
|
-
});
|
|
263
|
-
test('should handle agent asking a question', async () => {
|
|
264
|
-
const task = 'Create a new component';
|
|
265
|
-
const question = 'What is the name of the component?';
|
|
266
|
-
const answer = 'MyComponent';
|
|
267
|
-
const generatedPlan = `1. Create ${answer}`;
|
|
268
|
-
const { context, assert: proxyAssert } = createTestProxy([
|
|
269
|
-
{
|
|
270
|
-
toolName: 'generateText',
|
|
271
|
-
args: expect.any(Object),
|
|
272
|
-
returnValue: [
|
|
273
|
-
{
|
|
274
|
-
role: 'assistant',
|
|
275
|
-
content: `${JSON.stringify({ question: { question } })}`,
|
|
276
|
-
},
|
|
277
|
-
],
|
|
278
|
-
},
|
|
279
|
-
{
|
|
280
|
-
toolName: 'input',
|
|
281
|
-
args: { message: question, default: undefined },
|
|
282
|
-
returnValue: answer,
|
|
283
|
-
},
|
|
284
|
-
{
|
|
285
|
-
toolName: 'generateText',
|
|
286
|
-
args: expect.any(Object),
|
|
287
|
-
returnValue: [
|
|
288
|
-
{
|
|
289
|
-
role: 'assistant',
|
|
290
|
-
content: `\`\`\`json
|
|
291
|
-
${JSON.stringify({ plan: generatedPlan })}
|
|
292
|
-
\`\`\``,
|
|
293
|
-
},
|
|
294
|
-
],
|
|
295
|
-
},
|
|
296
|
-
{
|
|
297
|
-
toolName: 'updateMemory',
|
|
298
|
-
args: expect.anything(),
|
|
299
|
-
returnValue: undefined,
|
|
300
|
-
},
|
|
301
|
-
{
|
|
302
|
-
toolName: 'select',
|
|
303
|
-
args: {
|
|
304
|
-
message: 'What do you want to do?',
|
|
305
|
-
choices: expect.anything(),
|
|
306
|
-
},
|
|
307
|
-
returnValue: 'exit',
|
|
308
|
-
},
|
|
309
|
-
]);
|
|
310
|
-
assert = proxyAssert;
|
|
311
|
-
const result = await planWorkflow({ task, mode: 'interactive', interactive: true, additionalTools: {} }, context);
|
|
312
|
-
expect(result.plan).toBe(generatedPlan);
|
|
313
|
-
});
|
|
314
|
-
test('should regenerate the plan', async () => {
|
|
315
|
-
const task = 'Create a new component';
|
|
316
|
-
const initialPlan = '1. Create the component file.';
|
|
317
|
-
const regeneratedPlan = '1. Create a better component file.';
|
|
318
|
-
const { context, assert: proxyAssert } = createTestProxy([
|
|
319
|
-
{
|
|
320
|
-
toolName: 'generateText',
|
|
321
|
-
args: expect.anything(),
|
|
322
|
-
returnValue: [
|
|
323
|
-
{
|
|
324
|
-
role: 'assistant',
|
|
325
|
-
content: `\`\`\`json
|
|
326
|
-
${JSON.stringify({ plan: initialPlan })}
|
|
327
|
-
\`\`\``,
|
|
328
|
-
},
|
|
329
|
-
],
|
|
330
|
-
},
|
|
331
|
-
{
|
|
332
|
-
toolName: 'updateMemory',
|
|
333
|
-
args: expect.anything(),
|
|
334
|
-
returnValue: undefined,
|
|
335
|
-
},
|
|
336
|
-
{
|
|
337
|
-
toolName: 'select',
|
|
338
|
-
args: {
|
|
339
|
-
message: 'What do you want to do?',
|
|
340
|
-
choices: expect.anything(),
|
|
341
|
-
},
|
|
342
|
-
returnValue: 'regenerate',
|
|
343
|
-
},
|
|
344
|
-
{
|
|
345
|
-
toolName: 'generateText',
|
|
346
|
-
args: expect.anything(),
|
|
347
|
-
returnValue: [
|
|
348
|
-
{
|
|
349
|
-
role: 'assistant',
|
|
350
|
-
content: `\`\`\`json
|
|
351
|
-
${JSON.stringify({ plan: regeneratedPlan })}
|
|
352
|
-
\`\`\``,
|
|
353
|
-
},
|
|
354
|
-
],
|
|
355
|
-
},
|
|
356
|
-
{
|
|
357
|
-
toolName: 'updateMemory',
|
|
358
|
-
args: expect.anything(),
|
|
359
|
-
returnValue: undefined,
|
|
360
|
-
},
|
|
361
|
-
{
|
|
362
|
-
toolName: 'select',
|
|
363
|
-
args: {
|
|
364
|
-
message: 'What do you want to do?',
|
|
365
|
-
choices: expect.anything(),
|
|
366
|
-
},
|
|
367
|
-
returnValue: 'exit',
|
|
368
|
-
},
|
|
369
|
-
]);
|
|
370
|
-
assert = proxyAssert;
|
|
371
|
-
const result = await planWorkflow({ task, mode: 'interactive', interactive: true, additionalTools: {} }, context);
|
|
372
|
-
expect(result.plan).toBe(regeneratedPlan);
|
|
373
|
-
});
|
|
374
|
-
test('should return a plan with files', async () => {
|
|
375
|
-
const task = 'Create a new component';
|
|
376
|
-
const generatedPlan = {
|
|
377
|
-
plan: '1. Create the component file.',
|
|
378
|
-
files: ['src/index.ts'],
|
|
379
|
-
};
|
|
380
|
-
const fileContent = 'export {}';
|
|
381
|
-
const expectedCalls = [
|
|
382
|
-
{
|
|
383
|
-
toolName: 'generateText',
|
|
384
|
-
args: expect.anything(),
|
|
385
|
-
returnValue: [
|
|
386
|
-
{
|
|
387
|
-
role: 'assistant',
|
|
388
|
-
content: `\`\`\`json
|
|
389
|
-
${JSON.stringify(generatedPlan)}
|
|
390
|
-
\`\`\``,
|
|
391
|
-
},
|
|
392
|
-
],
|
|
393
|
-
},
|
|
394
|
-
{
|
|
395
|
-
toolName: 'readFile',
|
|
396
|
-
args: { path: 'src/index.ts' },
|
|
397
|
-
returnValue: fileContent,
|
|
398
|
-
},
|
|
399
|
-
{
|
|
400
|
-
toolName: 'updateMemory',
|
|
401
|
-
args: expect.anything(),
|
|
402
|
-
returnValue: undefined,
|
|
403
|
-
},
|
|
404
|
-
];
|
|
405
|
-
const { context, assert: proxyAssert } = createTestProxy(expectedCalls);
|
|
406
|
-
assert = proxyAssert;
|
|
407
|
-
const result = await planWorkflow({ task, mode: 'noninteractive', interactive: false, additionalTools: {} }, context);
|
|
408
|
-
expect(result.plan).toBe(generatedPlan.plan);
|
|
409
|
-
// Verify files structure
|
|
410
|
-
expect(result.files).toBeDefined();
|
|
411
|
-
expect(result.files).toBeInstanceOf(Array);
|
|
412
|
-
expect(result.files).toHaveLength(1);
|
|
413
|
-
expect(result.files[0]).toEqual({
|
|
414
|
-
path: 'src/index.ts',
|
|
415
|
-
content: fileContent,
|
|
416
|
-
});
|
|
417
|
-
});
|
|
418
|
-
});
|
|
419
|
-
//# sourceMappingURL=plan.workflow.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"plan.workflow.test.js","sourceRoot":"","sources":["../../src/workflows/plan.workflow.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,UAAU,CAAA;AAClE,OAAO,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAA;AAE9C,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAA;AAC9C,OAAO,EAAE,eAAe,EAAyB,MAAM,kBAAkB,CAAA;AAEzE,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,MAAkB,CAAA;IAEtB,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,EAAE,EAAE,CAAA;QACV,IAAI,CAAC,OAAO,EAAE,CAAA;IAChB,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,IAAI,GAAG,wBAAwB,CAAA;QACrC,MAAM,aAAa,GAAG,+BAA+B,CAAA;QACrD,MAAM,QAAQ,GAAG,gBAAgB,CAAA;QAEjC,MAAM,EACJ,OAAO,EACP,MAAM,EAAE,WAAW,EACnB,MAAM,GACP,GAAG,eAAe,CAAC;YAClB;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE;oBACX;wBACE,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;qBACjD;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,SAAS;aACvB;YACD;gBACE,QAAQ,EAAE,QAAQ;gBAClB,IAAI,EAAE;oBACJ,OAAO,EAAE,yBAAyB;oBAClC,OAAO,EAAE,MAAM,CAAC,QAAQ,EAAE;iBAC3B;gBACD,WAAW,EAAE,MAAM;aACpB;YACD;gBACE,QAAQ,EAAE,OAAO;gBACjB,IAAI,EAAE;oBACJ,OAAO,EAAE,qCAAqC;oBAC9C,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;iBAC5B;gBACD,WAAW,EAAE,QAAQ;aACtB;YACD;gBACE,QAAQ,EAAE,aAAa;gBACvB,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,EAAE;gBAChD,WAAW,EAAE,SAAS;aACvB;SACF,CAAC,CAAA;QACF,MAAM,GAAG,WAAW,CAAA;QAEpB,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,CAAA;QAEjH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;QACvC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,QAAQ,EAAE,CAAC,CAAA;IACvE,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,IAAI,GAAG,wBAAwB,CAAA;QACrC,MAAM,WAAW,GAAG,+BAA+B,CAAA;QACnD,MAAM,QAAQ,GAAG,kBAAkB,CAAA;QACnC,MAAM,eAAe,GAAG,uDAAuD,CAAA;QAE/E,MAAM,EACJ,OAAO,EACP,MAAM,EAAE,WAAW,EACnB,MAAM,GACP,GAAG,eAAe,CAAC;YAClB;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE;oBACX;wBACE,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE;EACnB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;OAChC;qBACI;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,SAAS;aACvB;YACD;gBACE,QAAQ,EAAE,QAAQ;gBAClB,IAAI,EAAE;oBACJ,OAAO,EAAE,yBAAyB;oBAClC,OAAO,EAAE,MAAM,CAAC,QAAQ,EAAE;iBAC3B;gBACD,WAAW,EAAE,UAAU;aACxB;YACD;gBACE,QAAQ,EAAE,OAAO;gBACjB,IAAI,EAAE;oBACJ,OAAO,EAAE,mCAAmC;iBAC7C;gBACD,WAAW,EAAE,QAAQ;aACtB;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE;oBACX;wBACE,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE;EACnB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;OACpC;qBACI;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,SAAS;aACvB;YACD;gBACE,QAAQ,EAAE,QAAQ;gBAClB,IAAI,EAAE;oBACJ,OAAO,EAAE,yBAAyB;oBAClC,OAAO,EAAE,MAAM,CAAC,QAAQ,EAAE;iBAC3B;gBACD,WAAW,EAAE,MAAM;aACpB;YACD;gBACE,QAAQ,EAAE,OAAO;gBACjB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,gBAAgB;aAC9B;YACD;gBACE,QAAQ,EAAE,aAAa;gBACvB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,SAAS;aACvB;SACF,CAAC,CAAA;QACF,MAAM,GAAG,WAAW,CAAA;QAEpB,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,CAAA;QAEjH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;QACzC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,8BAA8B,CAAC,CAAA;IAC1E,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,IAAI,GAAG,wBAAwB,CAAA;QACrC,MAAM,aAAa,GAAG,+BAA+B,CAAA;QAErD,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,CAAC;YACvD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE;oBACX;wBACE,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE;EACnB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;OAClC;qBACI;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,SAAS;aACvB;YACD;gBACE,QAAQ,EAAE,QAAQ;gBAClB,IAAI,EAAE;oBACJ,OAAO,EAAE,yBAAyB;oBAClC,OAAO,EAAE,MAAM,CAAC,QAAQ,EAAE;iBAC3B;gBACD,WAAW,EAAE,MAAM;aACpB;SACF,CAAC,CAAA;QACF,MAAM,GAAG,WAAW,CAAA;QAEpB,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,CAAA;QACjH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,IAAI,GAAG,wBAAwB,CAAA;QACrC,MAAM,aAAa,GAAG,+BAA+B,CAAA;QAErD,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,CAAC;YACvD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBACxB,WAAW,EAAE;oBACX;wBACE,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE;EACnB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;OAClC;qBACI;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,SAAS;aACvB;SACF,CAAC,CAAA;QACF,MAAM,GAAG,WAAW,CAAA;QAEpB,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,gBAAgB,EAAE,WAAW,EAAE,KAAK,EAAE,eAAe,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,CAAA;QAErH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,IAAI,GAAG,wBAAwB,CAAA;QACrC,MAAM,aAAa,GAAG,+BAA+B,CAAA;QAErD,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,CAAC;YACvD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBACxB,WAAW,EAAE;oBACX;wBACE,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE;EACnB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;OAClC;qBACI;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,SAAS;aACvB;YACD;gBACE,QAAQ,EAAE,OAAO;gBACjB,IAAI,EAAE;oBACJ,OAAO,EAAE,+GAA+G;iBACzH;gBACD,WAAW,EAAE,EAAE;aAChB;SACF,CAAC,CAAA;QACF,MAAM,GAAG,WAAW,CAAA;QAEpB,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,CAAA;QAE7G,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,IAAI,GAAG,wBAAwB,CAAA;QACrC,MAAM,aAAa,GAAG,+BAA+B,CAAA;QAErD,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,CAAC;YACvD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBACxB,WAAW,EAAE;oBACX;wBACE,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE;EACnB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;OAClC;qBACI;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,SAAS;aACvB;YACD;gBACE,QAAQ,EAAE,OAAO;gBACjB,IAAI,EAAE;oBACJ,OAAO,EAAE,+GAA+G;iBACzH;gBACD,WAAW,EAAE,IAAI,kBAAkB,EAAE;gBACrC,OAAO,EAAE,IAAI;aACd;SACF,CAAC,CAAA;QACF,MAAM,GAAG,WAAW,CAAA;QAEpB,MAAM,MAAM,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACpH,kBAAkB,CACnB,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,IAAI,GAAG,wBAAwB,CAAA;QACrC,MAAM,QAAQ,GAAG,oCAAoC,CAAA;QACrD,MAAM,MAAM,GAAG,aAAa,CAAA;QAC5B,MAAM,aAAa,GAAG,aAAa,MAAM,EAAE,CAAA;QAE3C,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,CAAC;YACvD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBACxB,WAAW,EAAE;oBACX;wBACE,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE;qBACzD;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,OAAO;gBACjB,IAAI,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE;gBAC/C,WAAW,EAAE,MAAM;aACpB;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBACxB,WAAW,EAAE;oBACX;wBACE,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE;EACnB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;OAClC;qBACI;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,SAAS;aACvB;YACD;gBACE,QAAQ,EAAE,QAAQ;gBAClB,IAAI,EAAE;oBACJ,OAAO,EAAE,yBAAyB;oBAClC,OAAO,EAAE,MAAM,CAAC,QAAQ,EAAE;iBAC3B;gBACD,WAAW,EAAE,MAAM;aACpB;SACF,CAAC,CAAA;QACF,MAAM,GAAG,WAAW,CAAA;QAEpB,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,CAAA;QAEjH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,IAAI,GAAG,wBAAwB,CAAA;QACrC,MAAM,WAAW,GAAG,+BAA+B,CAAA;QACnD,MAAM,eAAe,GAAG,oCAAoC,CAAA;QAE5D,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,CAAC;YACvD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE;oBACX;wBACE,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE;EACnB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;OAChC;qBACI;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,SAAS;aACvB;YACD;gBACE,QAAQ,EAAE,QAAQ;gBAClB,IAAI,EAAE;oBACJ,OAAO,EAAE,yBAAyB;oBAClC,OAAO,EAAE,MAAM,CAAC,QAAQ,EAAE;iBAC3B;gBACD,WAAW,EAAE,YAAY;aAC1B;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE;oBACX;wBACE,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE;EACnB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;OACpC;qBACI;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,SAAS;aACvB;YACD;gBACE,QAAQ,EAAE,QAAQ;gBAClB,IAAI,EAAE;oBACJ,OAAO,EAAE,yBAAyB;oBAClC,OAAO,EAAE,MAAM,CAAC,QAAQ,EAAE;iBAC3B;gBACD,WAAW,EAAE,MAAM;aACpB;SACF,CAAC,CAAA;QACF,MAAM,GAAG,WAAW,CAAA;QAEpB,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,CAAA;QAEjH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;IAC3C,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,IAAI,GAAG,wBAAwB,CAAA;QACrC,MAAM,aAAa,GAAG;YACpB,IAAI,EAAE,+BAA+B;YACrC,KAAK,EAAE,CAAC,cAAc,CAAC;SACxB,CAAA;QACD,MAAM,WAAW,GAAG,WAAW,CAAA;QAE/B,MAAM,aAAa,GAA8C;YAC/D;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE;oBACX;wBACE,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE;EACnB,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;OACxB;qBACI;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,UAAU;gBACpB,IAAI,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE;gBAC9B,WAAW,EAAE,WAAW;aACzB;YACD;gBACE,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gBACvB,WAAW,EAAE,SAAS;aACvB;SACF,CAAA;QAED,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,CAAC,aAAa,CAAC,CAAA;QACvE,MAAM,GAAG,WAAW,CAAA;QAEpB,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,gBAAgB,EAAE,WAAW,EAAE,KAAK,EAAE,eAAe,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,CAAA;QAErH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAA;QAC5C,yBAAyB;QACzB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAA;QAClC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAA;QAC1C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QACpC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC9B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,WAAW;SACrB,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
|
|
@@ -1,54 +0,0 @@
|
|
|
1
|
-
// generated by polka.codes
|
|
2
|
-
import { agentWorkflow } from '@polka-codes/core';
|
|
3
|
-
import { z } from 'zod';
|
|
4
|
-
import { GET_PR_DETAILS_SYSTEM_PROMPT } from './prompts';
|
|
5
|
-
import { checkGhInstalled, getDefaultBranch } from './workflow.utils';
|
|
6
|
-
const prDetailsSchema = z.object({
|
|
7
|
-
title: z.string(),
|
|
8
|
-
description: z.string(),
|
|
9
|
-
});
|
|
10
|
-
export const prWorkflow = async (input, context) => {
|
|
11
|
-
const { step, tools } = context;
|
|
12
|
-
const { diff, commits, branchName } = await step('get-git-info', async () => {
|
|
13
|
-
await checkGhInstalled(tools.executeCommand);
|
|
14
|
-
const branchResult = await tools.executeCommand({ command: 'git', args: ['rev-parse', '--abbrev-ref', 'HEAD'] });
|
|
15
|
-
const branchName = branchResult.stdout.trim();
|
|
16
|
-
const defaultBranch = await getDefaultBranch(tools.executeCommand);
|
|
17
|
-
if (!defaultBranch) {
|
|
18
|
-
throw new Error('Could not determine default branch name.');
|
|
19
|
-
}
|
|
20
|
-
const commitsResult = await tools.executeCommand({
|
|
21
|
-
command: 'git',
|
|
22
|
-
args: ['--no-pager', 'log', '--oneline', '--no-color', '--no-merges', '--no-decorate', `${defaultBranch}..HEAD`],
|
|
23
|
-
});
|
|
24
|
-
const commits = commitsResult.stdout;
|
|
25
|
-
const diffResult = await tools.executeCommand({
|
|
26
|
-
command: 'git',
|
|
27
|
-
args: ['--no-pager', 'diff', `${defaultBranch}..HEAD`],
|
|
28
|
-
});
|
|
29
|
-
const diff = diffResult.stdout;
|
|
30
|
-
return { diff, commits, branchName };
|
|
31
|
-
});
|
|
32
|
-
const messageParts = [`<branch_name>${branchName}</branch_name>`, `<commit_messages>${commits}</commit_messages>`, `<diff>${diff}</diff>`];
|
|
33
|
-
if (input.context) {
|
|
34
|
-
messageParts.push(`<user_context>${input.context}</user_context>`);
|
|
35
|
-
}
|
|
36
|
-
const agentResult = await agentWorkflow({
|
|
37
|
-
systemPrompt: GET_PR_DETAILS_SYSTEM_PROMPT,
|
|
38
|
-
userMessage: [
|
|
39
|
-
{
|
|
40
|
-
role: 'user',
|
|
41
|
-
content: messageParts.join('\n'),
|
|
42
|
-
},
|
|
43
|
-
],
|
|
44
|
-
tools: [], // No tools are needed for this agent
|
|
45
|
-
outputSchema: prDetailsSchema,
|
|
46
|
-
}, context);
|
|
47
|
-
if (agentResult.type !== 'Exit') {
|
|
48
|
-
throw new Error(`Workflow exited unexpectedly with type: ${agentResult.type}`);
|
|
49
|
-
}
|
|
50
|
-
const prDetails = agentResult.object;
|
|
51
|
-
const result = await tools.createPullRequest(prDetails);
|
|
52
|
-
return result;
|
|
53
|
-
};
|
|
54
|
-
//# sourceMappingURL=pr.workflow.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"pr.workflow.js","sourceRoot":"","sources":["../../src/workflows/pr.workflow.ts"],"names":[],"mappings":"AAAA,2BAA2B;AAE3B,OAAO,EAAE,aAAa,EAAmB,MAAM,mBAAmB,CAAA;AAClE,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAAE,4BAA4B,EAAE,MAAM,WAAW,CAAA;AACxD,OAAO,EAA0B,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAA;AAE7F,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;IACjB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;CACxB,CAAC,CAAA;AAMF,MAAM,CAAC,MAAM,UAAU,GAA6G,KAAK,EACvI,KAAK,EACL,OAAO,EACP,EAAE;IACF,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,OAAO,CAAA;IAC/B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,gBAAgB,CAAC,KAAK,CAAC,cAAc,CAAC,CAAA;QAC5C,MAAM,YAAY,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,cAAc,EAAE,MAAM,CAAC,EAAE,CAAC,CAAA;QAChH,MAAM,UAAU,GAAG,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE,CAAA;QAC7C,MAAM,aAAa,GAAG,MAAM,gBAAgB,CAAC,KAAK,CAAC,cAAc,CAAC,CAAA;QAClE,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAA;QAC7D,CAAC;QACD,MAAM,aAAa,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC;YAC/C,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,CAAC,YAAY,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,aAAa,EAAE,eAAe,EAAE,GAAG,aAAa,QAAQ,CAAC;SACjH,CAAC,CAAA;QACF,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAA;QACpC,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC;YAC5C,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,GAAG,aAAa,QAAQ,CAAC;SACvD,CAAC,CAAA;QACF,MAAM,IAAI,GAAG,UAAU,CAAC,MAAM,CAAA;QAC9B,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,CAAA;IACtC,CAAC,CAAC,CAAA;IAEF,MAAM,YAAY,GAAG,CAAC,gBAAgB,UAAU,gBAAgB,EAAE,oBAAoB,OAAO,oBAAoB,EAAE,SAAS,IAAI,SAAS,CAAC,CAAA;IAC1I,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QAClB,YAAY,CAAC,IAAI,CAAC,iBAAiB,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAA;IACpE,CAAC;IAED,MAAM,WAAW,GAAG,MAAM,aAAa,CACrC;QACE,YAAY,EAAE,4BAA4B;QAC1C,WAAW,EAAE;YACX;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;aACjC;SACF;QACD,KAAK,EAAE,EAAE,EAAE,qCAAqC;QAChD,YAAY,EAAE,eAAe;KAC9B,EACD,OAAO,CACR,CAAA;IAED,IAAI,WAAW,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CAAC,2CAA2C,WAAW,CAAC,IAAI,EAAE,CAAC,CAAA;IAChF,CAAC;IAED,MAAM,SAAS,GAAG,WAAW,CAAC,MAAyC,CAAA;IAEvE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAA;IAEvD,OAAO,MAAM,CAAA;AACf,CAAC,CAAA"}
|
|
@@ -1,98 +0,0 @@
|
|
|
1
|
-
// generated by polka.codes
|
|
2
|
-
import { expect, mock, test } from 'bun:test';
|
|
3
|
-
import { createContext } from '@polka-codes/core';
|
|
4
|
-
import { createWorkflowTestContext } from '../test/workflow-fixtures';
|
|
5
|
-
import { prWorkflow } from './pr.workflow';
|
|
6
|
-
const createMockContext = () => {
|
|
7
|
-
const { tools } = createWorkflowTestContext();
|
|
8
|
-
// Add PR-specific tools
|
|
9
|
-
tools.createPullRequest = mock(async (details) => details);
|
|
10
|
-
tools.createCommit = mock(async () => ({ message: '' }));
|
|
11
|
-
tools.printChangeFile = mock(async () => ({ stagedFiles: [], unstagedFiles: [] }));
|
|
12
|
-
tools.confirm = mock(async () => true);
|
|
13
|
-
tools.input = mock(async () => '');
|
|
14
|
-
tools.select = mock(async () => '');
|
|
15
|
-
tools.writeToFile = mock(async () => { });
|
|
16
|
-
tools.readFile = mock(async () => '');
|
|
17
|
-
tools.invokeTool = mock(async () => ({ success: true, message: { type: 'text', value: '' } }));
|
|
18
|
-
const run = (input) => {
|
|
19
|
-
const context = createContext(tools);
|
|
20
|
-
const testContext = {
|
|
21
|
-
...context,
|
|
22
|
-
step: (_name, arg2, arg3) => {
|
|
23
|
-
const fn = (typeof arg2 === 'function' ? arg2 : arg3);
|
|
24
|
-
return fn();
|
|
25
|
-
},
|
|
26
|
-
};
|
|
27
|
-
return prWorkflow(input, testContext);
|
|
28
|
-
};
|
|
29
|
-
return {
|
|
30
|
-
tools,
|
|
31
|
-
run,
|
|
32
|
-
};
|
|
33
|
-
};
|
|
34
|
-
test('should create PR with user context', async () => {
|
|
35
|
-
const { tools, run } = createMockContext();
|
|
36
|
-
// Mock the sequence of executeCommand calls
|
|
37
|
-
tools.executeCommand
|
|
38
|
-
.mockResolvedValueOnce({ exitCode: 0, stdout: 'gh version 2.40.0', stderr: '' }) // gh --version
|
|
39
|
-
.mockResolvedValueOnce({ exitCode: 0, stdout: 'feature/add-auth\n', stderr: '' }) // git rev-parse
|
|
40
|
-
.mockResolvedValueOnce({ exitCode: 1, stdout: '', stderr: '' }) // git show-ref master
|
|
41
|
-
.mockResolvedValueOnce({ exitCode: 0, stdout: '', stderr: '' }) // git show-ref main
|
|
42
|
-
.mockResolvedValueOnce({
|
|
43
|
-
exitCode: 0,
|
|
44
|
-
stdout: 'abc1234 feat: add authentication module\ndef5678 fix: resolve login bug\n',
|
|
45
|
-
stderr: '',
|
|
46
|
-
}) // git log
|
|
47
|
-
.mockResolvedValueOnce({
|
|
48
|
-
exitCode: 0,
|
|
49
|
-
stdout: `--- a/src/auth.ts
|
|
50
|
-
+++ b/src/auth.ts
|
|
51
|
-
@@ -1,3 +1,5 @@
|
|
52
|
-
+export function authenticate() {
|
|
53
|
-
+ return true;
|
|
54
|
-
}
|
|
55
|
-
export function login() {}`,
|
|
56
|
-
stderr: '',
|
|
57
|
-
}); // git diff
|
|
58
|
-
// Mock agent workflow response
|
|
59
|
-
tools.generateText.mockResolvedValue([
|
|
60
|
-
{
|
|
61
|
-
role: 'assistant',
|
|
62
|
-
content: JSON.stringify({
|
|
63
|
-
title: 'feat: add authentication feature',
|
|
64
|
-
description: 'This PR implements the authentication feature for the application.\n\n### Changes\n- Added authenticate function\n- Fixed login bug',
|
|
65
|
-
}),
|
|
66
|
-
},
|
|
67
|
-
]);
|
|
68
|
-
const result = await run({ context: 'Implementing new authentication feature' });
|
|
69
|
-
expect(result).toEqual({
|
|
70
|
-
title: 'feat: add authentication feature',
|
|
71
|
-
description: 'This PR implements the authentication feature for the application.\n\n### Changes\n- Added authenticate function\n- Fixed login bug',
|
|
72
|
-
});
|
|
73
|
-
expect(tools.executeCommand).toHaveBeenCalledTimes(6);
|
|
74
|
-
expect(tools.generateText).toHaveBeenCalledTimes(1);
|
|
75
|
-
expect(tools.createPullRequest).toHaveBeenCalledTimes(1);
|
|
76
|
-
});
|
|
77
|
-
test('should fail when GitHub CLI is not installed', async () => {
|
|
78
|
-
const { tools, run } = createMockContext();
|
|
79
|
-
tools.executeCommand.mockResolvedValueOnce({
|
|
80
|
-
exitCode: 1,
|
|
81
|
-
stdout: '',
|
|
82
|
-
stderr: 'command not found: gh',
|
|
83
|
-
});
|
|
84
|
-
await expect(run({})).rejects.toThrow('GitHub CLI (gh) is not installed. Please install it from https://cli.github.com/');
|
|
85
|
-
});
|
|
86
|
-
test('should fail when unable to determine default branch', async () => {
|
|
87
|
-
const { tools, run } = createMockContext();
|
|
88
|
-
tools.executeCommand
|
|
89
|
-
.mockResolvedValueOnce({ exitCode: 0, stdout: 'gh version 2.40.0', stderr: '' }) // gh --version
|
|
90
|
-
.mockResolvedValueOnce({ exitCode: 0, stdout: 'feature/my-branch\n', stderr: '' }) // git rev-parse
|
|
91
|
-
.mockResolvedValueOnce({ exitCode: 1, stdout: '', stderr: 'not a git repository' }) // gh repo view
|
|
92
|
-
.mockResolvedValueOnce({ exitCode: 1, stdout: '', stderr: '' }) // git show-ref master
|
|
93
|
-
.mockResolvedValueOnce({ exitCode: 1, stdout: '', stderr: '' }) // git show-ref main
|
|
94
|
-
.mockResolvedValueOnce({ exitCode: 1, stdout: '', stderr: '' }) // git show-ref develop
|
|
95
|
-
.mockResolvedValueOnce({ exitCode: 1, stdout: '', stderr: 'fatal: No remote named origin' }); // git remote show origin
|
|
96
|
-
await expect(run({})).rejects.toThrow('Could not determine default branch name.');
|
|
97
|
-
});
|
|
98
|
-
//# sourceMappingURL=pr.workflow.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"pr.workflow.test.js","sourceRoot":"","sources":["../../src/workflows/pr.workflow.test.ts"],"names":[],"mappings":"AAAA,2BAA2B;AAE3B,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,UAAU,CAAA;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAA;AACjD,OAAO,EAAE,yBAAyB,EAAE,MAAM,2BAA2B,CAAA;AACrE,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAE1C,MAAM,iBAAiB,GAAG,GAAG,EAAE;IAC7B,MAAM,EAAE,KAAK,EAAE,GAAG,yBAAyB,EAAE,CAAA;IAE7C,wBAAwB;IACxB,KAAK,CAAC,iBAAiB,GAAG,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,CAAA;IAC1D,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;IACxD,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;IAClF,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,CAAA;IACtC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC,CAAA;IAClC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC,CAAA;IACnC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,CAAA;IACxC,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC,CAAA;IACrC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;IAE9F,MAAM,GAAG,GAAG,CAAC,KAAU,EAAE,EAAE;QACzB,MAAM,OAAO,GAAG,aAAa,CAAC,KAAY,CAAC,CAAA;QAC3C,MAAM,WAAW,GAAG;YAClB,GAAG,OAAO;YACV,IAAI,EAAE,CAAC,KAAa,EAAE,IAAS,EAAE,IAAU,EAAE,EAAE;gBAC7C,MAAM,EAAE,GAAG,CAAC,OAAO,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAuB,CAAA;gBAC3E,OAAO,EAAE,EAAE,CAAA;YACb,CAAC;SACK,CAAA;QACR,OAAO,UAAU,CAAC,KAAK,EAAE,WAAW,CAAC,CAAA;IACvC,CAAC,CAAA;IAED,OAAO;QACL,KAAK;QACL,GAAG;KACJ,CAAA;AACH,CAAC,CAAA;AAED,IAAI,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;IACpD,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,iBAAiB,EAAE,CAAA;IAE1C,4CAA4C;IAC5C,KAAK,CAAC,cAAc;SACjB,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,mBAAmB,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,eAAe;SAC/F,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,oBAAoB,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,gBAAgB;SACjG,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,sBAAsB;SACrF,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,oBAAoB;SACnF,qBAAqB,CAAC;QACrB,QAAQ,EAAE,CAAC;QACX,MAAM,EAAE,2EAA2E;QACnF,MAAM,EAAE,EAAE;KACX,CAAC,CAAC,UAAU;SACZ,qBAAqB,CAAC;QACrB,QAAQ,EAAE,CAAC;QACX,MAAM,EAAE;;;;;;4BAMc;QACtB,MAAM,EAAE,EAAE;KACX,CAAC,CAAA,CAAC,WAAW;IAEhB,+BAA+B;IAC/B,KAAK,CAAC,YAAY,CAAC,iBAAiB,CAAC;QACnC;YACE,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;gBACtB,KAAK,EAAE,kCAAkC;gBACzC,WAAW,EACT,qIAAqI;aACxI,CAAC;SACH;KACF,CAAC,CAAA;IAEF,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,OAAO,EAAE,yCAAyC,EAAE,CAAC,CAAA;IAEhF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;QACrB,KAAK,EAAE,kCAAkC;QACzC,WAAW,EACT,qIAAqI;KACxI,CAAC,CAAA;IAEF,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IACrD,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IACnD,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;AAC1D,CAAC,CAAC,CAAA;AAEF,IAAI,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;IAC9D,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,iBAAiB,EAAE,CAAA;IAE1C,KAAK,CAAC,cAAc,CAAC,qBAAqB,CAAC;QACzC,QAAQ,EAAE,CAAC;QACX,MAAM,EAAE,EAAE;QACV,MAAM,EAAE,uBAAuB;KAChC,CAAC,CAAA;IAEF,MAAM,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,kFAAkF,CAAC,CAAA;AAC3H,CAAC,CAAC,CAAA;AAEF,IAAI,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;IACrE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,iBAAiB,EAAE,CAAA;IAE1C,KAAK,CAAC,cAAc;SACjB,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,mBAAmB,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,eAAe;SAC/F,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,qBAAqB,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,gBAAgB;SAClG,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,sBAAsB,EAAE,CAAC,CAAC,eAAe;SAClG,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,sBAAsB;SACrF,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,oBAAoB;SACnF,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,uBAAuB;SACtF,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,+BAA+B,EAAE,CAAC,CAAA,CAAC,yBAAyB;IAExH,MAAM,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,0CAA0C,CAAC,CAAA;AACnF,CAAC,CAAC,CAAA"}
|