@slopus/beer 0.1.1
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/README.md +24 -0
- package/bin/beer.js +2 -0
- package/dist/apps/bootstrap.d.ts +5 -0
- package/dist/apps/bootstrap.d.ts.map +1 -0
- package/dist/apps/bootstrap.js +170 -0
- package/dist/apps/bootstrap.js.map +1 -0
- package/dist/commands/bootstrapCommand.d.ts +6 -0
- package/dist/commands/bootstrapCommand.d.ts.map +1 -0
- package/dist/commands/bootstrapCommand.js +16 -0
- package/dist/commands/bootstrapCommand.js.map +1 -0
- package/dist/main.d.ts +2 -0
- package/dist/main.d.ts.map +1 -0
- package/dist/main.js +17 -0
- package/dist/main.js.map +1 -0
- package/dist/modules/ai/aiOutputExtract.d.ts +6 -0
- package/dist/modules/ai/aiOutputExtract.d.ts.map +1 -0
- package/dist/modules/ai/aiOutputExtract.js +16 -0
- package/dist/modules/ai/aiOutputExtract.js.map +1 -0
- package/dist/modules/ai/aiOutputExtract.spec.d.ts +2 -0
- package/dist/modules/ai/aiOutputExtract.spec.d.ts.map +1 -0
- package/dist/modules/ai/aiOutputExtract.spec.js +25 -0
- package/dist/modules/ai/aiOutputExtract.spec.js.map +1 -0
- package/dist/modules/ai/aiReadmeGenerate.d.ts +18 -0
- package/dist/modules/ai/aiReadmeGenerate.d.ts.map +1 -0
- package/dist/modules/ai/aiReadmeGenerate.js +23 -0
- package/dist/modules/ai/aiReadmeGenerate.js.map +1 -0
- package/dist/modules/ai/aiReadmeGenerate.spec.d.ts +2 -0
- package/dist/modules/ai/aiReadmeGenerate.spec.d.ts.map +1 -0
- package/dist/modules/ai/aiReadmeGenerate.spec.js +27 -0
- package/dist/modules/ai/aiReadmeGenerate.spec.js.map +1 -0
- package/dist/modules/ai/generate.d.ts +28 -0
- package/dist/modules/ai/generate.d.ts.map +1 -0
- package/dist/modules/ai/generate.js +132 -0
- package/dist/modules/ai/generate.js.map +1 -0
- package/dist/modules/ai/generate.spec.d.ts +2 -0
- package/dist/modules/ai/generate.spec.d.ts.map +1 -0
- package/dist/modules/ai/generate.spec.js +53 -0
- package/dist/modules/ai/generate.spec.js.map +1 -0
- package/dist/modules/ai/generateFile.d.ts +14 -0
- package/dist/modules/ai/generateFile.d.ts.map +1 -0
- package/dist/modules/ai/generateFile.js +56 -0
- package/dist/modules/ai/generateFile.js.map +1 -0
- package/dist/modules/ai/generateFile.spec.d.ts +2 -0
- package/dist/modules/ai/generateFile.spec.d.ts.map +1 -0
- package/dist/modules/ai/generateFile.spec.js +60 -0
- package/dist/modules/ai/generateFile.spec.js.map +1 -0
- package/dist/modules/ai/generateText.d.ts +9 -0
- package/dist/modules/ai/generateText.d.ts.map +1 -0
- package/dist/modules/ai/generateText.js +12 -0
- package/dist/modules/ai/generateText.js.map +1 -0
- package/dist/modules/ai/generateText.spec.d.ts +2 -0
- package/dist/modules/ai/generateText.spec.d.ts.map +1 -0
- package/dist/modules/ai/generateText.spec.js +28 -0
- package/dist/modules/ai/generateText.spec.js.map +1 -0
- package/dist/modules/ai/providerGenerate.d.ts +29 -0
- package/dist/modules/ai/providerGenerate.d.ts.map +1 -0
- package/dist/modules/ai/providerGenerate.js +144 -0
- package/dist/modules/ai/providerGenerate.js.map +1 -0
- package/dist/modules/ai/providerGenerate.spec.d.ts +2 -0
- package/dist/modules/ai/providerGenerate.spec.d.ts.map +1 -0
- package/dist/modules/ai/providerGenerate.spec.js +85 -0
- package/dist/modules/ai/providerGenerate.spec.js.map +1 -0
- package/dist/modules/ai/providerGenerate.unit.spec.d.ts +2 -0
- package/dist/modules/ai/providerGenerate.unit.spec.d.ts.map +1 -0
- package/dist/modules/ai/providerGenerate.unit.spec.js +49 -0
- package/dist/modules/ai/providerGenerate.unit.spec.js.map +1 -0
- package/dist/modules/beer/beerDevFolderResolve.d.ts +5 -0
- package/dist/modules/beer/beerDevFolderResolve.d.ts.map +1 -0
- package/dist/modules/beer/beerDevFolderResolve.js +9 -0
- package/dist/modules/beer/beerDevFolderResolve.js.map +1 -0
- package/dist/modules/beer/beerDevFolderResolve.spec.d.ts +2 -0
- package/dist/modules/beer/beerDevFolderResolve.spec.d.ts.map +1 -0
- package/dist/modules/beer/beerDevFolderResolve.spec.js +10 -0
- package/dist/modules/beer/beerDevFolderResolve.spec.js.map +1 -0
- package/dist/modules/beer/beerOriginalPathResolve.d.ts +5 -0
- package/dist/modules/beer/beerOriginalPathResolve.d.ts.map +1 -0
- package/dist/modules/beer/beerOriginalPathResolve.js +8 -0
- package/dist/modules/beer/beerOriginalPathResolve.js.map +1 -0
- package/dist/modules/beer/beerOriginalPathResolve.spec.d.ts +2 -0
- package/dist/modules/beer/beerOriginalPathResolve.spec.d.ts.map +1 -0
- package/dist/modules/beer/beerOriginalPathResolve.spec.js +10 -0
- package/dist/modules/beer/beerOriginalPathResolve.spec.js.map +1 -0
- package/dist/modules/beer/beerSettingsPathResolve.d.ts +5 -0
- package/dist/modules/beer/beerSettingsPathResolve.d.ts.map +1 -0
- package/dist/modules/beer/beerSettingsPathResolve.js +9 -0
- package/dist/modules/beer/beerSettingsPathResolve.js.map +1 -0
- package/dist/modules/beer/beerSettingsPathResolve.spec.d.ts +2 -0
- package/dist/modules/beer/beerSettingsPathResolve.spec.d.ts.map +1 -0
- package/dist/modules/beer/beerSettingsPathResolve.spec.js +10 -0
- package/dist/modules/beer/beerSettingsPathResolve.spec.js.map +1 -0
- package/dist/modules/beer/beerSettingsRead.d.ts +6 -0
- package/dist/modules/beer/beerSettingsRead.d.ts.map +1 -0
- package/dist/modules/beer/beerSettingsRead.js +28 -0
- package/dist/modules/beer/beerSettingsRead.js.map +1 -0
- package/dist/modules/beer/beerSettingsTypes.d.ts +12 -0
- package/dist/modules/beer/beerSettingsTypes.d.ts.map +1 -0
- package/dist/modules/beer/beerSettingsTypes.js +2 -0
- package/dist/modules/beer/beerSettingsTypes.js.map +1 -0
- package/dist/modules/beer/beerSettingsWrite.d.ts +6 -0
- package/dist/modules/beer/beerSettingsWrite.d.ts.map +1 -0
- package/dist/modules/beer/beerSettingsWrite.js +10 -0
- package/dist/modules/beer/beerSettingsWrite.js.map +1 -0
- package/dist/modules/context/contextGet.d.ts +7 -0
- package/dist/modules/context/contextGet.d.ts.map +1 -0
- package/dist/modules/context/contextGet.js +12 -0
- package/dist/modules/context/contextGet.js.map +1 -0
- package/dist/modules/context/contextGetOrInitialize.d.ts +7 -0
- package/dist/modules/context/contextGetOrInitialize.d.ts.map +1 -0
- package/dist/modules/context/contextGetOrInitialize.js +12 -0
- package/dist/modules/context/contextGetOrInitialize.js.map +1 -0
- package/dist/modules/context/contextInitialize.d.ts +7 -0
- package/dist/modules/context/contextInitialize.d.ts.map +1 -0
- package/dist/modules/context/contextInitialize.js +25 -0
- package/dist/modules/context/contextInitialize.js.map +1 -0
- package/dist/modules/context/contextInitialize.spec.d.ts +2 -0
- package/dist/modules/context/contextInitialize.spec.d.ts.map +1 -0
- package/dist/modules/context/contextInitialize.spec.js +77 -0
- package/dist/modules/context/contextInitialize.spec.js.map +1 -0
- package/dist/modules/context/contextTypes.d.ts +21 -0
- package/dist/modules/context/contextTypes.d.ts.map +1 -0
- package/dist/modules/context/contextTypes.js +2 -0
- package/dist/modules/context/contextTypes.js.map +1 -0
- package/dist/modules/git/gitPush.d.ts +6 -0
- package/dist/modules/git/gitPush.d.ts.map +1 -0
- package/dist/modules/git/gitPush.js +9 -0
- package/dist/modules/git/gitPush.js.map +1 -0
- package/dist/modules/git/gitRemoteEnsure.d.ts +6 -0
- package/dist/modules/git/gitRemoteEnsure.d.ts.map +1 -0
- package/dist/modules/git/gitRemoteEnsure.js +20 -0
- package/dist/modules/git/gitRemoteEnsure.js.map +1 -0
- package/dist/modules/git/gitRepoCheckout.d.ts +6 -0
- package/dist/modules/git/gitRepoCheckout.d.ts.map +1 -0
- package/dist/modules/git/gitRepoCheckout.js +15 -0
- package/dist/modules/git/gitRepoCheckout.js.map +1 -0
- package/dist/modules/git/gitRepoCheckout.spec.d.ts +2 -0
- package/dist/modules/git/gitRepoCheckout.spec.d.ts.map +1 -0
- package/dist/modules/git/gitRepoCheckout.spec.js +33 -0
- package/dist/modules/git/gitRepoCheckout.spec.js.map +1 -0
- package/dist/modules/git/gitStageAndCommit.d.ts +6 -0
- package/dist/modules/git/gitStageAndCommit.d.ts.map +1 -0
- package/dist/modules/git/gitStageAndCommit.js +18 -0
- package/dist/modules/git/gitStageAndCommit.js.map +1 -0
- package/dist/modules/github/githubCliEnsure.d.ts +6 -0
- package/dist/modules/github/githubCliEnsure.d.ts.map +1 -0
- package/dist/modules/github/githubCliEnsure.js +13 -0
- package/dist/modules/github/githubCliEnsure.js.map +1 -0
- package/dist/modules/github/githubOwnerChoicesGet.d.ts +5 -0
- package/dist/modules/github/githubOwnerChoicesGet.d.ts.map +1 -0
- package/dist/modules/github/githubOwnerChoicesGet.js +20 -0
- package/dist/modules/github/githubOwnerChoicesGet.js.map +1 -0
- package/dist/modules/github/githubRepoCreate.d.ts +6 -0
- package/dist/modules/github/githubRepoCreate.d.ts.map +1 -0
- package/dist/modules/github/githubRepoCreate.js +17 -0
- package/dist/modules/github/githubRepoCreate.js.map +1 -0
- package/dist/modules/github/githubRepoExists.d.ts +5 -0
- package/dist/modules/github/githubRepoExists.d.ts.map +1 -0
- package/dist/modules/github/githubRepoExists.js +9 -0
- package/dist/modules/github/githubRepoExists.js.map +1 -0
- package/dist/modules/github/githubRepoNameResolve.d.ts +17 -0
- package/dist/modules/github/githubRepoNameResolve.d.ts.map +1 -0
- package/dist/modules/github/githubRepoNameResolve.js +34 -0
- package/dist/modules/github/githubRepoNameResolve.js.map +1 -0
- package/dist/modules/github/githubRepoNameResolve.spec.d.ts +2 -0
- package/dist/modules/github/githubRepoNameResolve.spec.d.ts.map +1 -0
- package/dist/modules/github/githubRepoNameResolve.spec.js +37 -0
- package/dist/modules/github/githubRepoNameResolve.spec.js.map +1 -0
- package/dist/modules/github/githubRepoParse.d.ts +7 -0
- package/dist/modules/github/githubRepoParse.d.ts.map +1 -0
- package/dist/modules/github/githubRepoParse.js +35 -0
- package/dist/modules/github/githubRepoParse.js.map +1 -0
- package/dist/modules/github/githubRepoParse.spec.d.ts +2 -0
- package/dist/modules/github/githubRepoParse.spec.d.ts.map +1 -0
- package/dist/modules/github/githubRepoParse.spec.js +17 -0
- package/dist/modules/github/githubRepoParse.spec.js.map +1 -0
- package/dist/modules/github/githubRepoStatusGet.d.ts +7 -0
- package/dist/modules/github/githubRepoStatusGet.d.ts.map +1 -0
- package/dist/modules/github/githubRepoStatusGet.js +21 -0
- package/dist/modules/github/githubRepoStatusGet.js.map +1 -0
- package/dist/modules/github/githubRepoUrlBuild.d.ts +5 -0
- package/dist/modules/github/githubRepoUrlBuild.d.ts.map +1 -0
- package/dist/modules/github/githubRepoUrlBuild.js +7 -0
- package/dist/modules/github/githubRepoUrlBuild.js.map +1 -0
- package/dist/modules/github/githubTypes.d.ts +8 -0
- package/dist/modules/github/githubTypes.d.ts.map +1 -0
- package/dist/modules/github/githubTypes.js +2 -0
- package/dist/modules/github/githubTypes.js.map +1 -0
- package/dist/modules/github/githubViewerGet.d.ts +5 -0
- package/dist/modules/github/githubViewerGet.d.ts.map +1 -0
- package/dist/modules/github/githubViewerGet.js +14 -0
- package/dist/modules/github/githubViewerGet.js.map +1 -0
- package/dist/modules/prompt/promptConfirm.d.ts +6 -0
- package/dist/modules/prompt/promptConfirm.d.ts.map +1 -0
- package/dist/modules/prompt/promptConfirm.js +23 -0
- package/dist/modules/prompt/promptConfirm.js.map +1 -0
- package/dist/modules/prompt/promptInput.d.ts +6 -0
- package/dist/modules/prompt/promptInput.d.ts.map +1 -0
- package/dist/modules/prompt/promptInput.js +14 -0
- package/dist/modules/prompt/promptInput.js.map +1 -0
- package/dist/modules/providers/providerDetect.d.ts +6 -0
- package/dist/modules/providers/providerDetect.d.ts.map +1 -0
- package/dist/modules/providers/providerDetect.js +43 -0
- package/dist/modules/providers/providerDetect.js.map +1 -0
- package/dist/modules/providers/providerDetect.spec.d.ts +2 -0
- package/dist/modules/providers/providerDetect.spec.d.ts.map +1 -0
- package/dist/modules/providers/providerDetect.spec.js +76 -0
- package/dist/modules/providers/providerDetect.spec.js.map +1 -0
- package/dist/modules/providers/providerModelSelect.d.ts +12 -0
- package/dist/modules/providers/providerModelSelect.d.ts.map +1 -0
- package/dist/modules/providers/providerModelSelect.js +134 -0
- package/dist/modules/providers/providerModelSelect.js.map +1 -0
- package/dist/modules/providers/providerModelSelect.spec.d.ts +2 -0
- package/dist/modules/providers/providerModelSelect.spec.d.ts.map +1 -0
- package/dist/modules/providers/providerModelSelect.spec.js +68 -0
- package/dist/modules/providers/providerModelSelect.spec.js.map +1 -0
- package/dist/modules/providers/providerModelsGet.d.ts +7 -0
- package/dist/modules/providers/providerModelsGet.d.ts.map +1 -0
- package/dist/modules/providers/providerModelsGet.js +100 -0
- package/dist/modules/providers/providerModelsGet.js.map +1 -0
- package/dist/modules/providers/providerModelsGet.spec.d.ts +2 -0
- package/dist/modules/providers/providerModelsGet.spec.d.ts.map +1 -0
- package/dist/modules/providers/providerModelsGet.spec.js +88 -0
- package/dist/modules/providers/providerModelsGet.spec.js.map +1 -0
- package/dist/modules/providers/providerPriorityList.d.ts +6 -0
- package/dist/modules/providers/providerPriorityList.d.ts.map +1 -0
- package/dist/modules/providers/providerPriorityList.js +16 -0
- package/dist/modules/providers/providerPriorityList.js.map +1 -0
- package/dist/modules/providers/providerPriorityList.spec.d.ts +2 -0
- package/dist/modules/providers/providerPriorityList.spec.d.ts.map +1 -0
- package/dist/modules/providers/providerPriorityList.spec.js +19 -0
- package/dist/modules/providers/providerPriorityList.spec.js.map +1 -0
- package/dist/modules/providers/providerTypes.d.ts +21 -0
- package/dist/modules/providers/providerTypes.d.ts.map +1 -0
- package/dist/modules/providers/providerTypes.js +2 -0
- package/dist/modules/providers/providerTypes.js.map +1 -0
- package/dist/modules/sandbox/sandboxInferenceFilesystemPolicy.d.ts +17 -0
- package/dist/modules/sandbox/sandboxInferenceFilesystemPolicy.d.ts.map +1 -0
- package/dist/modules/sandbox/sandboxInferenceFilesystemPolicy.js +88 -0
- package/dist/modules/sandbox/sandboxInferenceFilesystemPolicy.js.map +1 -0
- package/dist/modules/sandbox/sandboxInferenceFilesystemPolicy.spec.d.ts +2 -0
- package/dist/modules/sandbox/sandboxInferenceFilesystemPolicy.spec.d.ts.map +1 -0
- package/dist/modules/sandbox/sandboxInferenceFilesystemPolicy.spec.js +70 -0
- package/dist/modules/sandbox/sandboxInferenceFilesystemPolicy.spec.js.map +1 -0
- package/dist/modules/sandbox/sandboxInferenceGet.d.ts +12 -0
- package/dist/modules/sandbox/sandboxInferenceGet.d.ts.map +1 -0
- package/dist/modules/sandbox/sandboxInferenceGet.js +18 -0
- package/dist/modules/sandbox/sandboxInferenceGet.js.map +1 -0
- package/dist/modules/sandbox/sandboxInferenceGet.spec.d.ts +2 -0
- package/dist/modules/sandbox/sandboxInferenceGet.spec.d.ts.map +1 -0
- package/dist/modules/sandbox/sandboxInferenceGet.spec.js +65 -0
- package/dist/modules/sandbox/sandboxInferenceGet.spec.js.map +1 -0
- package/dist/modules/sandbox/sandboxInferenceTypes.d.ts +7 -0
- package/dist/modules/sandbox/sandboxInferenceTypes.d.ts.map +1 -0
- package/dist/modules/sandbox/sandboxInferenceTypes.js +2 -0
- package/dist/modules/sandbox/sandboxInferenceTypes.js.map +1 -0
- package/dist/modules/sandbox/sandboxTypes.d.ts +4 -0
- package/dist/modules/sandbox/sandboxTypes.d.ts.map +1 -0
- package/dist/modules/sandbox/sandboxTypes.js +2 -0
- package/dist/modules/sandbox/sandboxTypes.js.map +1 -0
- package/dist/modules/util/commandRun.d.ts +22 -0
- package/dist/modules/util/commandRun.d.ts.map +1 -0
- package/dist/modules/util/commandRun.js +112 -0
- package/dist/modules/util/commandRun.js.map +1 -0
- package/dist/modules/util/commandRun.spec.d.ts +2 -0
- package/dist/modules/util/commandRun.spec.d.ts.map +1 -0
- package/dist/modules/util/commandRun.spec.js +43 -0
- package/dist/modules/util/commandRun.spec.js.map +1 -0
- package/dist/modules/util/pathResolveFromInitCwd.d.ts +6 -0
- package/dist/modules/util/pathResolveFromInitCwd.d.ts.map +1 -0
- package/dist/modules/util/pathResolveFromInitCwd.js +10 -0
- package/dist/modules/util/pathResolveFromInitCwd.js.map +1 -0
- package/dist/release/releaseRun.d.ts +6 -0
- package/dist/release/releaseRun.d.ts.map +1 -0
- package/dist/release/releaseRun.js +153 -0
- package/dist/release/releaseRun.js.map +1 -0
- package/dist/release/releaseVersionIncrement.d.ts +7 -0
- package/dist/release/releaseVersionIncrement.d.ts.map +1 -0
- package/dist/release/releaseVersionIncrement.js +23 -0
- package/dist/release/releaseVersionIncrement.js.map +1 -0
- package/dist/release/releaseVersionIncrement.spec.d.ts +2 -0
- package/dist/release/releaseVersionIncrement.spec.d.ts.map +1 -0
- package/dist/release/releaseVersionIncrement.spec.js +20 -0
- package/dist/release/releaseVersionIncrement.spec.js.map +1 -0
- package/dist/release/releaseVersionPrompt.d.ts +6 -0
- package/dist/release/releaseVersionPrompt.d.ts.map +1 -0
- package/dist/release/releaseVersionPrompt.js +55 -0
- package/dist/release/releaseVersionPrompt.js.map +1 -0
- package/dist/text/all.txt +85 -0
- package/dist/text/text.d.ts +7 -0
- package/dist/text/text.d.ts.map +1 -0
- package/dist/text/text.js +34 -0
- package/dist/text/text.js.map +1 -0
- package/dist/types.d.ts +6 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/workflows/steps/generateCommitMessage.d.ts +13 -0
- package/dist/workflows/steps/generateCommitMessage.d.ts.map +1 -0
- package/dist/workflows/steps/generateCommitMessage.js +25 -0
- package/dist/workflows/steps/generateCommitMessage.js.map +1 -0
- package/dist/workflows/steps/generateCommitMessage.spec.d.ts +2 -0
- package/dist/workflows/steps/generateCommitMessage.spec.d.ts.map +1 -0
- package/dist/workflows/steps/generateCommitMessage.spec.js +28 -0
- package/dist/workflows/steps/generateCommitMessage.spec.js.map +1 -0
- package/package.json +38 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generate.spec.d.ts","sourceRoot":"","sources":["../../../sources/modules/ai/generate.spec.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
import { spawnSync } from "node:child_process";
|
|
2
|
+
import { describe, expect, it } from "vitest";
|
|
3
|
+
import { generate } from "@/modules/ai/generate.js";
|
|
4
|
+
const TEST_TIMEOUT_MS = 180_000;
|
|
5
|
+
function commandExists(command) {
|
|
6
|
+
const result = spawnSync("zsh", ["-lc", `command -v ${command}`], {
|
|
7
|
+
encoding: "utf-8"
|
|
8
|
+
});
|
|
9
|
+
return result.status === 0;
|
|
10
|
+
}
|
|
11
|
+
function errorTextResolve(error) {
|
|
12
|
+
if (error instanceof Error) {
|
|
13
|
+
return error.message;
|
|
14
|
+
}
|
|
15
|
+
return String(error);
|
|
16
|
+
}
|
|
17
|
+
function contextForProvider(providerId, command) {
|
|
18
|
+
return {
|
|
19
|
+
projectPath: "/tmp/test-project",
|
|
20
|
+
providers: [{ id: providerId, available: true, command, priority: 1 }],
|
|
21
|
+
inferText: (input) => generate(contextForProvider(providerId, command), input.prompt, {
|
|
22
|
+
providerPriority: input.providerPriority,
|
|
23
|
+
modelPriority: input.modelPriority,
|
|
24
|
+
showProgress: input.showProgress,
|
|
25
|
+
writePolicy: input.writePolicy
|
|
26
|
+
}),
|
|
27
|
+
stageAndCommit: async () => false
|
|
28
|
+
};
|
|
29
|
+
}
|
|
30
|
+
async function runRealProviderInference(providerId, command) {
|
|
31
|
+
const context = contextForProvider(providerId, command);
|
|
32
|
+
try {
|
|
33
|
+
const result = await generate(context, "Reply with a short single sentence.", {
|
|
34
|
+
providerPriority: [providerId]
|
|
35
|
+
});
|
|
36
|
+
expect(result.provider).toBe(providerId);
|
|
37
|
+
expect(result.text.trim().length).toBeGreaterThan(0);
|
|
38
|
+
}
|
|
39
|
+
catch (error) {
|
|
40
|
+
const message = errorTextResolve(error);
|
|
41
|
+
// Real provider can fail when auth/network is missing; ensure failures
|
|
42
|
+
// are not caused by unsupported CLI flags.
|
|
43
|
+
expect(message).not.toContain("unknown option '--mode'");
|
|
44
|
+
expect(message).not.toContain("unknown option '--print'");
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
describe("generate pi integration", () => {
|
|
48
|
+
it("runs real pi inference without json-mode flag parsing errors", async () => {
|
|
49
|
+
expect(commandExists("pi")).toBe(true);
|
|
50
|
+
await runRealProviderInference("pi", "pi");
|
|
51
|
+
}, TEST_TIMEOUT_MS);
|
|
52
|
+
});
|
|
53
|
+
//# sourceMappingURL=generate.spec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generate.spec.js","sourceRoot":"","sources":["../../../sources/modules/ai/generate.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AAIpD,MAAM,eAAe,GAAG,OAAO,CAAC;AAEhC,SAAS,aAAa,CAAC,OAAe;IACpC,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,cAAc,OAAO,EAAE,CAAC,EAAE;QAChE,QAAQ,EAAE,OAAO;KAClB,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC;AAC7B,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACtC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC,OAAO,CAAC;IACvB,CAAC;IACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC;AAED,SAAS,kBAAkB,CAAC,UAAsB,EAAE,OAAe;IACjE,OAAO;QACL,WAAW,EAAE,mBAAmB;QAChC,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QACtE,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE,CACnB,QAAQ,CACN,kBAAkB,CAAC,UAAU,EAAE,OAAO,CAAC,EACvC,KAAK,CAAC,MAAM,EACZ;YACE,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;YACxC,aAAa,EAAE,KAAK,CAAC,aAAa;YAClC,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,WAAW,EAAE,KAAK,CAAC,WAAW;SAC/B,CACF;QACH,cAAc,EAAE,KAAK,IAAI,EAAE,CAAC,KAAK;KAClC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,wBAAwB,CAAC,UAAsB,EAAE,OAAe;IAC7E,MAAM,OAAO,GAAG,kBAAkB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IAExD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,qCAAqC,EAAE;YAC5E,gBAAgB,EAAE,CAAC,UAAU,CAAC;SAC/B,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACvD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAExC,uEAAuE;QACvE,2CAA2C;QAC3C,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC;QACzD,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC;AAED,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,EAAE,CACA,8DAA8D,EAC9D,KAAK,IAAI,EAAE;QACT,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,wBAAwB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAC7C,CAAC,EACD,eAAe,CAChB,CAAC;AACJ,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { Context } from "@/types";
|
|
2
|
+
import { type GeneratePermissions } from "@/modules/ai/generate.js";
|
|
3
|
+
export interface GenerateFilePermissions extends GeneratePermissions {
|
|
4
|
+
retries?: number;
|
|
5
|
+
}
|
|
6
|
+
/**
|
|
7
|
+
* Generates content and asks the model to write it into a specific file path only.
|
|
8
|
+
* Retries when inference succeeds but the target file does not appear on disk.
|
|
9
|
+
*/
|
|
10
|
+
export declare function generateFile(context: Context, prompt: string, outputFilePath: string, permissions?: GenerateFilePermissions): Promise<{
|
|
11
|
+
provider?: string;
|
|
12
|
+
text: string;
|
|
13
|
+
}>;
|
|
14
|
+
//# sourceMappingURL=generateFile.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generateFile.d.ts","sourceRoot":"","sources":["../../../sources/modules/ai/generateFile.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAY,KAAK,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAE9E,MAAM,WAAW,uBAAwB,SAAQ,mBAAmB;IAClE,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;;GAGG;AACH,wBAAsB,YAAY,CAChC,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,MAAM,EACd,cAAc,EAAE,MAAM,EACtB,WAAW,GAAE,uBAA4B,GACxC,OAAO,CAAC;IAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,CAAC,CAmC9C"}
|
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
import { stat } from "node:fs/promises";
|
|
2
|
+
import path from "node:path";
|
|
3
|
+
import { generate } from "@/modules/ai/generate.js";
|
|
4
|
+
/**
|
|
5
|
+
* Generates content and asks the model to write it into a specific file path only.
|
|
6
|
+
* Retries when inference succeeds but the target file does not appear on disk.
|
|
7
|
+
*/
|
|
8
|
+
export async function generateFile(context, prompt, outputFilePath, permissions = {}) {
|
|
9
|
+
const resolvedOutputPath = pathResolveInProject(context.projectPath, outputFilePath);
|
|
10
|
+
const maxRetries = Math.max(0, permissions.retries ?? 1);
|
|
11
|
+
const basePrompt = [
|
|
12
|
+
prompt,
|
|
13
|
+
`Write the final result only to this file: ${resolvedOutputPath}.`,
|
|
14
|
+
"Do not write to any other files."
|
|
15
|
+
].join("\n\n");
|
|
16
|
+
const permissionsResolved = {
|
|
17
|
+
...permissions,
|
|
18
|
+
expectedOutput: {
|
|
19
|
+
type: "file",
|
|
20
|
+
filePath: resolvedOutputPath
|
|
21
|
+
},
|
|
22
|
+
writePolicy: {
|
|
23
|
+
mode: "write-whitelist",
|
|
24
|
+
writablePaths: [resolvedOutputPath]
|
|
25
|
+
}
|
|
26
|
+
};
|
|
27
|
+
for (let attempt = 0; attempt <= maxRetries; attempt += 1) {
|
|
28
|
+
const promptResolved = attempt === 0
|
|
29
|
+
? basePrompt
|
|
30
|
+
: [
|
|
31
|
+
basePrompt,
|
|
32
|
+
`Last time file did not appear at ${resolvedOutputPath} - create it now.`
|
|
33
|
+
].join("\n\n");
|
|
34
|
+
const result = await generate(context, promptResolved, permissionsResolved);
|
|
35
|
+
if (await fileExists(resolvedOutputPath)) {
|
|
36
|
+
return result;
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
throw new Error(`Inference did not create expected file: ${resolvedOutputPath}`);
|
|
40
|
+
}
|
|
41
|
+
function pathResolveInProject(projectPath, outputFilePath) {
|
|
42
|
+
if (path.isAbsolute(outputFilePath)) {
|
|
43
|
+
return path.resolve(outputFilePath);
|
|
44
|
+
}
|
|
45
|
+
return path.resolve(projectPath, outputFilePath);
|
|
46
|
+
}
|
|
47
|
+
async function fileExists(filePath) {
|
|
48
|
+
try {
|
|
49
|
+
const fileStat = await stat(filePath);
|
|
50
|
+
return fileStat.isFile();
|
|
51
|
+
}
|
|
52
|
+
catch {
|
|
53
|
+
return false;
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
//# sourceMappingURL=generateFile.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generateFile.js","sourceRoot":"","sources":["../../../sources/modules/ai/generateFile.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AACxC,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,QAAQ,EAA4B,MAAM,0BAA0B,CAAC;AAM9E;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAAgB,EAChB,MAAc,EACd,cAAsB,EACtB,cAAuC,EAAE;IAEzC,MAAM,kBAAkB,GAAG,oBAAoB,CAAC,OAAO,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;IACrF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;IACzD,MAAM,UAAU,GAAG;QACjB,MAAM;QACN,6CAA6C,kBAAkB,GAAG;QAClE,kCAAkC;KACnC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACf,MAAM,mBAAmB,GAAwB;QAC/C,GAAG,WAAW;QACd,cAAc,EAAE;YACd,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE,kBAAkB;SAC7B;QACD,WAAW,EAAE;YACX,IAAI,EAAE,iBAAiB;YACvB,aAAa,EAAE,CAAC,kBAAkB,CAAC;SACpC;KACF,CAAC;IAEF,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,UAAU,EAAE,OAAO,IAAI,CAAC,EAAE,CAAC;QAC1D,MAAM,cAAc,GAAG,OAAO,KAAK,CAAC;YAClC,CAAC,CAAC,UAAU;YACZ,CAAC,CAAC;gBACE,UAAU;gBACV,oCAAoC,kBAAkB,mBAAmB;aAC1E,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnB,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,cAAc,EAAE,mBAAmB,CAAC,CAAC;QAE5E,IAAI,MAAM,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;YACzC,OAAO,MAAM,CAAC;QAChB,CAAC;IACH,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,2CAA2C,kBAAkB,EAAE,CAAC,CAAC;AACnF,CAAC;AAED,SAAS,oBAAoB,CAAC,WAAmB,EAAE,cAAsB;IACvE,IAAI,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;QACpC,OAAO,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IACtC,CAAC;IACD,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;AACnD,CAAC;AAED,KAAK,UAAU,UAAU,CAAC,QAAgB;IACxC,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,OAAO,QAAQ,CAAC,MAAM,EAAE,CAAC;IAC3B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generateFile.spec.d.ts","sourceRoot":"","sources":["../../../sources/modules/ai/generateFile.spec.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,60 @@
|
|
|
1
|
+
import { mkdtemp, rm, writeFile } from "node:fs/promises";
|
|
2
|
+
import os from "node:os";
|
|
3
|
+
import path from "node:path";
|
|
4
|
+
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
5
|
+
const generateMock = vi.hoisted(() => vi.fn());
|
|
6
|
+
vi.mock("./generate.js", () => ({
|
|
7
|
+
generate: generateMock
|
|
8
|
+
}));
|
|
9
|
+
import { generateFile } from "@/modules/ai/generateFile.js";
|
|
10
|
+
describe("generateFile", () => {
|
|
11
|
+
beforeEach(() => {
|
|
12
|
+
generateMock.mockReset();
|
|
13
|
+
});
|
|
14
|
+
it("retries when file is missing and succeeds after file appears", async () => {
|
|
15
|
+
const tempDir = await mkdtemp(path.join(os.tmpdir(), "holdmybeer-generate-file-"));
|
|
16
|
+
try {
|
|
17
|
+
const outputPath = path.join(tempDir, "README.md");
|
|
18
|
+
const context = { projectPath: tempDir };
|
|
19
|
+
generateMock
|
|
20
|
+
.mockResolvedValueOnce({ provider: "pi", text: "first" })
|
|
21
|
+
.mockImplementationOnce(async () => {
|
|
22
|
+
await writeFile(outputPath, "# ok\n", "utf-8");
|
|
23
|
+
return { provider: "pi", text: "second" };
|
|
24
|
+
});
|
|
25
|
+
const result = await generateFile(context, "Create readme", outputPath);
|
|
26
|
+
expect(result).toEqual({ provider: "pi", text: "second" });
|
|
27
|
+
expect(generateMock).toHaveBeenCalledTimes(2);
|
|
28
|
+
expect(generateMock.mock.calls[0]?.[2]).toEqual({
|
|
29
|
+
expectedOutput: {
|
|
30
|
+
type: "file",
|
|
31
|
+
filePath: outputPath
|
|
32
|
+
},
|
|
33
|
+
writePolicy: {
|
|
34
|
+
mode: "write-whitelist",
|
|
35
|
+
writablePaths: [outputPath]
|
|
36
|
+
}
|
|
37
|
+
});
|
|
38
|
+
expect(generateMock.mock.calls[0]?.[1]).toContain(`Write the final result only to this file: ${outputPath}.`);
|
|
39
|
+
expect(generateMock.mock.calls[0]?.[1]).toContain("Do not write to any other files.");
|
|
40
|
+
expect(generateMock.mock.calls[1]?.[1]).toContain(`Last time file did not appear at ${outputPath} - create it now.`);
|
|
41
|
+
}
|
|
42
|
+
finally {
|
|
43
|
+
await rm(tempDir, { recursive: true, force: true });
|
|
44
|
+
}
|
|
45
|
+
});
|
|
46
|
+
it("throws when file still does not exist after retries", async () => {
|
|
47
|
+
const tempDir = await mkdtemp(path.join(os.tmpdir(), "holdmybeer-generate-file-"));
|
|
48
|
+
try {
|
|
49
|
+
const outputPath = path.join(tempDir, "README.md");
|
|
50
|
+
const context = { projectPath: tempDir };
|
|
51
|
+
generateMock.mockResolvedValue({ provider: "pi", text: "done" });
|
|
52
|
+
await expect(generateFile(context, "Create readme", outputPath, { retries: 1 })).rejects.toThrow(`Inference did not create expected file: ${outputPath}`);
|
|
53
|
+
expect(generateMock).toHaveBeenCalledTimes(2);
|
|
54
|
+
}
|
|
55
|
+
finally {
|
|
56
|
+
await rm(tempDir, { recursive: true, force: true });
|
|
57
|
+
}
|
|
58
|
+
});
|
|
59
|
+
});
|
|
60
|
+
//# sourceMappingURL=generateFile.spec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generateFile.spec.js","sourceRoot":"","sources":["../../../sources/modules/ai/generateFile.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG9D,MAAM,YAAY,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AAE/C,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9B,QAAQ,EAAE,YAAY;CACvB,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAE5D,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,UAAU,CAAC,GAAG,EAAE;QACd,YAAY,CAAC,SAAS,EAAE,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,2BAA2B,CAAC,CAAC,CAAC;QACnF,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;YACnD,MAAM,OAAO,GAAG,EAAE,WAAW,EAAE,OAAO,EAAa,CAAC;YAEpD,YAAY;iBACT,qBAAqB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;iBACxD,sBAAsB,CAAC,KAAK,IAAI,EAAE;gBACjC,MAAM,SAAS,CAAC,UAAU,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;gBAC/C,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YAC5C,CAAC,CAAC,CAAC;YAEL,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE,eAAe,EAAE,UAAU,CAAC,CAAC;YAExE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC3D,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBAC9C,cAAc,EAAE;oBACd,IAAI,EAAE,MAAM;oBACZ,QAAQ,EAAE,UAAU;iBACrB;gBACD,WAAW,EAAE;oBACX,IAAI,EAAE,iBAAiB;oBACvB,aAAa,EAAE,CAAC,UAAU,CAAC;iBAC5B;aACF,CAAC,CAAC;YACH,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAC/C,6CAA6C,UAAU,GAAG,CAC3D,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,kCAAkC,CAAC,CAAC;YACtF,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAC/C,oCAAoC,UAAU,mBAAmB,CAClE,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACtD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,2BAA2B,CAAC,CAAC,CAAC;QACnF,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;YACnD,MAAM,OAAO,GAAG,EAAE,WAAW,EAAE,OAAO,EAAa,CAAC;YAEpD,YAAY,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;YAEjE,MAAM,MAAM,CACV,YAAY,CAAC,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CACnE,CAAC,OAAO,CAAC,OAAO,CAAC,2CAA2C,UAAU,EAAE,CAAC,CAAC;YAC3E,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACtD,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { type GeneratePermissions, type GenerateResult } from "@/modules/ai/generate.js";
|
|
2
|
+
import type { Context } from "@/types";
|
|
3
|
+
export type GenerateTextPermissions = Omit<GeneratePermissions, "expectedOutput">;
|
|
4
|
+
/**
|
|
5
|
+
* Shortcut for text generation with default text expectation.
|
|
6
|
+
* Expects: prompt is inference task text and permissions optionally constrain provider/write policy.
|
|
7
|
+
*/
|
|
8
|
+
export declare function generateText(context: Context, prompt: string, permissions?: GenerateTextPermissions): Promise<GenerateResult>;
|
|
9
|
+
//# sourceMappingURL=generateText.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generateText.d.ts","sourceRoot":"","sources":["../../../sources/modules/ai/generateText.ts"],"names":[],"mappings":"AAAA,OAAO,EAAY,KAAK,mBAAmB,EAAE,KAAK,cAAc,EAAE,MAAM,0BAA0B,CAAC;AACnG,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAEvC,MAAM,MAAM,uBAAuB,GAAG,IAAI,CAAC,mBAAmB,EAAE,gBAAgB,CAAC,CAAC;AAElF;;;GAGG;AACH,wBAAsB,YAAY,CAChC,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,MAAM,EACd,WAAW,GAAE,uBAA4B,GACxC,OAAO,CAAC,cAAc,CAAC,CAKzB"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import { generate } from "@/modules/ai/generate.js";
|
|
2
|
+
/**
|
|
3
|
+
* Shortcut for text generation with default text expectation.
|
|
4
|
+
* Expects: prompt is inference task text and permissions optionally constrain provider/write policy.
|
|
5
|
+
*/
|
|
6
|
+
export async function generateText(context, prompt, permissions = {}) {
|
|
7
|
+
return generate(context, prompt, {
|
|
8
|
+
...permissions,
|
|
9
|
+
expectedOutput: { type: "text" }
|
|
10
|
+
});
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=generateText.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generateText.js","sourceRoot":"","sources":["../../../sources/modules/ai/generateText.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAiD,MAAM,0BAA0B,CAAC;AAKnG;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAAgB,EAChB,MAAc,EACd,cAAuC,EAAE;IAEzC,OAAO,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE;QAC/B,GAAG,WAAW;QACd,cAAc,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;KACjC,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generateText.spec.d.ts","sourceRoot":"","sources":["../../../sources/modules/ai/generateText.spec.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
2
|
+
const generateMock = vi.hoisted(() => vi.fn());
|
|
3
|
+
vi.mock("./generate.js", () => ({
|
|
4
|
+
generate: generateMock
|
|
5
|
+
}));
|
|
6
|
+
import { generateText } from "@/modules/ai/generateText.js";
|
|
7
|
+
describe("generateText", () => {
|
|
8
|
+
beforeEach(() => {
|
|
9
|
+
generateMock.mockReset();
|
|
10
|
+
});
|
|
11
|
+
it("forwards to generate with text expected output", async () => {
|
|
12
|
+
const context = { projectPath: "/tmp/test-project", providers: [] };
|
|
13
|
+
generateMock.mockResolvedValue({ provider: "pi", text: "ok" });
|
|
14
|
+
const result = await generateText(context, "hello", {
|
|
15
|
+
providerPriority: ["pi"],
|
|
16
|
+
showProgress: true,
|
|
17
|
+
writePolicy: { mode: "read-only" }
|
|
18
|
+
});
|
|
19
|
+
expect(result).toEqual({ provider: "pi", text: "ok" });
|
|
20
|
+
expect(generateMock).toHaveBeenCalledWith(context, "hello", {
|
|
21
|
+
providerPriority: ["pi"],
|
|
22
|
+
showProgress: true,
|
|
23
|
+
writePolicy: { mode: "read-only" },
|
|
24
|
+
expectedOutput: { type: "text" }
|
|
25
|
+
});
|
|
26
|
+
});
|
|
27
|
+
});
|
|
28
|
+
//# sourceMappingURL=generateText.spec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generateText.spec.js","sourceRoot":"","sources":["../../../sources/modules/ai/generateText.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG9D,MAAM,YAAY,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AAE/C,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9B,QAAQ,EAAE,YAAY;CACvB,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAE5D,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,UAAU,CAAC,GAAG,EAAE;QACd,YAAY,CAAC,SAAS,EAAE,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,OAAO,GAAG,EAAE,WAAW,EAAE,mBAAmB,EAAE,SAAS,EAAE,EAAE,EAAwB,CAAC;QAC1F,YAAY,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAE/D,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE,OAAO,EAAE;YAClD,gBAAgB,EAAE,CAAC,IAAI,CAAC;YACxB,YAAY,EAAE,IAAI;YAClB,WAAW,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;SACnC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,OAAO,EAAE;YAC1D,gBAAgB,EAAE,CAAC,IAAI,CAAC;YACxB,YAAY,EAAE,IAAI;YAClB,WAAW,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;YAClC,cAAc,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;SACjC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
import type { ProviderId } from "@/types";
|
|
2
|
+
import type { CommandSandbox } from "@/modules/sandbox/sandboxTypes.js";
|
|
3
|
+
import type { InferenceWritePolicy } from "@/modules/sandbox/sandboxInferenceTypes.js";
|
|
4
|
+
export interface ProviderGenerateInput {
|
|
5
|
+
providerId: ProviderId;
|
|
6
|
+
command: string;
|
|
7
|
+
model?: string;
|
|
8
|
+
prompt: string;
|
|
9
|
+
sandbox: CommandSandbox;
|
|
10
|
+
writePolicy?: InferenceWritePolicy;
|
|
11
|
+
requireOutputTags?: boolean;
|
|
12
|
+
onStdoutText?: (chunk: string) => void;
|
|
13
|
+
onStderrText?: (chunk: string) => void;
|
|
14
|
+
}
|
|
15
|
+
export interface ProviderGenerateFailure {
|
|
16
|
+
providerId: ProviderId;
|
|
17
|
+
exitCode: number;
|
|
18
|
+
stderr: string;
|
|
19
|
+
}
|
|
20
|
+
export interface ProviderGenerateResult {
|
|
21
|
+
output: string | null;
|
|
22
|
+
failure?: ProviderGenerateFailure;
|
|
23
|
+
}
|
|
24
|
+
/**
|
|
25
|
+
* Runs provider inference through CLI and extracts response wrapped in <output> tags.
|
|
26
|
+
* Expects: providerId is pi and command is executable.
|
|
27
|
+
*/
|
|
28
|
+
export declare function providerGenerate(input: ProviderGenerateInput): Promise<ProviderGenerateResult>;
|
|
29
|
+
//# sourceMappingURL=providerGenerate.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providerGenerate.d.ts","sourceRoot":"","sources":["../../../sources/modules/ai/providerGenerate.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAC1C,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mCAAmC,CAAC;AACxE,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,4CAA4C,CAAC;AAGvF,MAAM,WAAW,qBAAqB;IACpC,UAAU,EAAE,UAAU,CAAC;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,cAAc,CAAC;IACxB,WAAW,CAAC,EAAE,oBAAoB,CAAC;IACnC,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CACxC;AAED,MAAM,WAAW,uBAAuB;IACtC,UAAU,EAAE,UAAU,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,sBAAsB;IACrC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,OAAO,CAAC,EAAE,uBAAuB,CAAC;CACnC;AAID;;;GAGG;AACH,wBAAsB,gBAAgB,CACpC,KAAK,EAAE,qBAAqB,GAC3B,OAAO,CAAC,sBAAsB,CAAC,CAyEjC"}
|
|
@@ -0,0 +1,144 @@
|
|
|
1
|
+
import { aiOutputExtract } from "@/modules/ai/aiOutputExtract.js";
|
|
2
|
+
import { commandRun } from "@/modules/util/commandRun.js";
|
|
3
|
+
const OUTPUT_RETRY_PROMPT = "Last time you didnt return <output> - do this now";
|
|
4
|
+
/**
|
|
5
|
+
* Runs provider inference through CLI and extracts response wrapped in <output> tags.
|
|
6
|
+
* Expects: providerId is pi and command is executable.
|
|
7
|
+
*/
|
|
8
|
+
export async function providerGenerate(input) {
|
|
9
|
+
const requireOutputTags = input.requireOutputTags ?? true;
|
|
10
|
+
const prompts = [
|
|
11
|
+
input.prompt,
|
|
12
|
+
[input.prompt, OUTPUT_RETRY_PROMPT].join("\n\n")
|
|
13
|
+
];
|
|
14
|
+
for (let attempt = 0; attempt < prompts.length; attempt += 1) {
|
|
15
|
+
const result = await commandRun(input.command, providerArgs(input.providerId, prompts[attempt], input.model), {
|
|
16
|
+
allowFailure: true,
|
|
17
|
+
timeoutMs: 90_000,
|
|
18
|
+
sandbox: providerSandboxResolve(input),
|
|
19
|
+
onStdoutText: input.onStdoutText,
|
|
20
|
+
onStderrText: input.onStderrText
|
|
21
|
+
});
|
|
22
|
+
if (result.exitCode !== 0) {
|
|
23
|
+
return {
|
|
24
|
+
output: null,
|
|
25
|
+
failure: {
|
|
26
|
+
providerId: input.providerId,
|
|
27
|
+
exitCode: result.exitCode,
|
|
28
|
+
stderr: result.stderr.trim()
|
|
29
|
+
}
|
|
30
|
+
};
|
|
31
|
+
}
|
|
32
|
+
const outputRaw = providerOutputResolve(input.providerId, result.stdout);
|
|
33
|
+
if (!outputRaw) {
|
|
34
|
+
if (!requireOutputTags) {
|
|
35
|
+
// File-generation mode allows empty assistant text.
|
|
36
|
+
return { output: "" };
|
|
37
|
+
}
|
|
38
|
+
return {
|
|
39
|
+
output: null,
|
|
40
|
+
failure: {
|
|
41
|
+
providerId: input.providerId,
|
|
42
|
+
exitCode: 1,
|
|
43
|
+
stderr: "Provider returned no JSON assistant output."
|
|
44
|
+
}
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
if (!requireOutputTags) {
|
|
48
|
+
return { output: outputRaw };
|
|
49
|
+
}
|
|
50
|
+
const extractedOutput = aiOutputExtract(outputRaw);
|
|
51
|
+
if (extractedOutput) {
|
|
52
|
+
return { output: extractedOutput };
|
|
53
|
+
}
|
|
54
|
+
if (attempt === prompts.length - 1) {
|
|
55
|
+
return {
|
|
56
|
+
output: null,
|
|
57
|
+
failure: {
|
|
58
|
+
providerId: input.providerId,
|
|
59
|
+
exitCode: 1,
|
|
60
|
+
stderr: "Provider did not return <output> tags."
|
|
61
|
+
}
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
return {
|
|
66
|
+
output: null,
|
|
67
|
+
failure: {
|
|
68
|
+
providerId: input.providerId,
|
|
69
|
+
exitCode: 1,
|
|
70
|
+
stderr: "Provider execution failed."
|
|
71
|
+
}
|
|
72
|
+
};
|
|
73
|
+
}
|
|
74
|
+
function providerArgs(providerId, prompt, model) {
|
|
75
|
+
const args = ["--mode", "json", "--print", "--no-session"];
|
|
76
|
+
if (model && model.trim().length > 0) {
|
|
77
|
+
args.push("--model", model.trim());
|
|
78
|
+
}
|
|
79
|
+
args.push(prompt);
|
|
80
|
+
return args;
|
|
81
|
+
}
|
|
82
|
+
function providerSandboxResolve(input) {
|
|
83
|
+
return input.sandbox;
|
|
84
|
+
}
|
|
85
|
+
function providerOutputResolve(providerId, stdout) {
|
|
86
|
+
if (providerId !== "pi") {
|
|
87
|
+
return stdout.trim() || null;
|
|
88
|
+
}
|
|
89
|
+
return providerOutputResolvePiJson(stdout);
|
|
90
|
+
}
|
|
91
|
+
function providerOutputResolvePiJson(stdout) {
|
|
92
|
+
const lines = stdout
|
|
93
|
+
.split(/\r?\n/)
|
|
94
|
+
.map((line) => line.trim())
|
|
95
|
+
.filter((line) => line.length > 0);
|
|
96
|
+
let finalAssistantText;
|
|
97
|
+
for (const line of lines) {
|
|
98
|
+
const parsed = providerPiEventParse(line);
|
|
99
|
+
if (!parsed || typeof parsed !== "object") {
|
|
100
|
+
continue;
|
|
101
|
+
}
|
|
102
|
+
const text = providerPiAssistantText(parsed);
|
|
103
|
+
if (typeof text === "string") {
|
|
104
|
+
finalAssistantText = text;
|
|
105
|
+
}
|
|
106
|
+
}
|
|
107
|
+
const output = finalAssistantText?.trim();
|
|
108
|
+
return output && output.length > 0 ? output : null;
|
|
109
|
+
}
|
|
110
|
+
function providerPiEventParse(line) {
|
|
111
|
+
try {
|
|
112
|
+
return JSON.parse(line);
|
|
113
|
+
}
|
|
114
|
+
catch {
|
|
115
|
+
return null;
|
|
116
|
+
}
|
|
117
|
+
}
|
|
118
|
+
function providerPiAssistantText(event) {
|
|
119
|
+
if (!event || typeof event !== "object") {
|
|
120
|
+
return undefined;
|
|
121
|
+
}
|
|
122
|
+
const typed = event;
|
|
123
|
+
if (typed.type !== "message_end" || typed.message?.role !== "assistant") {
|
|
124
|
+
return undefined;
|
|
125
|
+
}
|
|
126
|
+
return providerPiContentText(typed.message.content);
|
|
127
|
+
}
|
|
128
|
+
function providerPiContentText(content) {
|
|
129
|
+
if (!Array.isArray(content)) {
|
|
130
|
+
return "";
|
|
131
|
+
}
|
|
132
|
+
return content
|
|
133
|
+
.map((entry) => {
|
|
134
|
+
if (!entry || typeof entry !== "object") {
|
|
135
|
+
return "";
|
|
136
|
+
}
|
|
137
|
+
const typed = entry;
|
|
138
|
+
return typed.type === "text" && typeof typed.text === "string"
|
|
139
|
+
? typed.text
|
|
140
|
+
: "";
|
|
141
|
+
})
|
|
142
|
+
.join("");
|
|
143
|
+
}
|
|
144
|
+
//# sourceMappingURL=providerGenerate.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providerGenerate.js","sourceRoot":"","sources":["../../../sources/modules/ai/providerGenerate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAIlE,OAAO,EAAE,UAAU,EAAE,MAAM,8BAA8B,CAAC;AAyB1D,MAAM,mBAAmB,GAAG,mDAAmD,CAAC;AAEhF;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,KAA4B;IAE5B,MAAM,iBAAiB,GAAG,KAAK,CAAC,iBAAiB,IAAI,IAAI,CAAC;IAC1D,MAAM,OAAO,GAAG;QACd,KAAK,CAAC,MAAM;QACZ,CAAC,KAAK,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;KACjD,CAAC;IAEF,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,MAAM,EAAE,OAAO,IAAI,CAAC,EAAE,CAAC;QAC7D,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,OAAO,EAAE,YAAY,CAAC,KAAK,CAAC,UAAU,EAAE,OAAO,CAAC,OAAO,CAAE,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE;YAC7G,YAAY,EAAE,IAAI;YAClB,SAAS,EAAE,MAAM;YACjB,OAAO,EAAE,sBAAsB,CAAC,KAAK,CAAC;YACtC,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,YAAY,EAAE,KAAK,CAAC,YAAY;SACjC,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO;gBACL,MAAM,EAAE,IAAI;gBACZ,OAAO,EAAE;oBACP,UAAU,EAAE,KAAK,CAAC,UAAU;oBAC5B,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE;iBAC7B;aACF,CAAC;QACJ,CAAC;QAED,MAAM,SAAS,GAAG,qBAAqB,CAAC,KAAK,CAAC,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QACzE,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBACvB,oDAAoD;gBACpD,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;YACxB,CAAC;YAED,OAAO;gBACL,MAAM,EAAE,IAAI;gBACZ,OAAO,EAAE;oBACP,UAAU,EAAE,KAAK,CAAC,UAAU;oBAC5B,QAAQ,EAAE,CAAC;oBACX,MAAM,EAAE,6CAA6C;iBACtD;aACF,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACvB,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QAC/B,CAAC;QAED,MAAM,eAAe,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;QACnD,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC;QACrC,CAAC;QAED,IAAI,OAAO,KAAK,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnC,OAAO;gBACL,MAAM,EAAE,IAAI;gBACZ,OAAO,EAAE;oBACP,UAAU,EAAE,KAAK,CAAC,UAAU;oBAC5B,QAAQ,EAAE,CAAC;oBACX,MAAM,EAAE,wCAAwC;iBACjD;aACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO;QACL,MAAM,EAAE,IAAI;QACZ,OAAO,EAAE;YACP,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,QAAQ,EAAE,CAAC;YACX,MAAM,EAAE,4BAA4B;SACrC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CACnB,UAAsB,EACtB,MAAc,EACd,KAAyB;IAEzB,MAAM,IAAI,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;IAC3D,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IACrC,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAClB,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,sBAAsB,CAAC,KAA4B;IAC1D,OAAO,KAAK,CAAC,OAAO,CAAC;AACvB,CAAC;AAED,SAAS,qBAAqB,CAAC,UAAsB,EAAE,MAAc;IACnE,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;QACxB,OAAO,MAAM,CAAC,IAAI,EAAE,IAAI,IAAI,CAAC;IAC/B,CAAC;IACD,OAAO,2BAA2B,CAAC,MAAM,CAAC,CAAC;AAC7C,CAAC;AAED,SAAS,2BAA2B,CAAC,MAAc;IACjD,MAAM,KAAK,GAAG,MAAM;SACjB,KAAK,CAAC,OAAO,CAAC;SACd,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;SAC1B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAErC,IAAI,kBAAsC,CAAC;IAC3C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC1C,SAAS;QACX,CAAC;QAED,MAAM,IAAI,GAAG,uBAAuB,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,kBAAkB,GAAG,IAAI,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,kBAAkB,EAAE,IAAI,EAAE,CAAC;IAC1C,OAAO,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC;AACrD,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAY;IACxC,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAAC,KAAc;IAC7C,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,KAAK,GAAG,KAGb,CAAC;IACF,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,KAAK,CAAC,OAAO,EAAE,IAAI,KAAK,WAAW,EAAE,CAAC;QACxE,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,qBAAqB,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AACtD,CAAC;AAED,SAAS,qBAAqB,CAAC,OAAgB;IAC7C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,OAAO;SACX,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QACb,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACxC,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,KAAK,GAAG,KAAyC,CAAC;QACxD,OAAO,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;YAC5D,CAAC,CAAC,KAAK,CAAC,IAAI;YACZ,CAAC,CAAC,EAAE,CAAC;IACT,CAAC,CAAC;SACD,IAAI,CAAC,EAAE,CAAC,CAAC;AACd,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providerGenerate.spec.d.ts","sourceRoot":"","sources":["../../../sources/modules/ai/providerGenerate.spec.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,85 @@
|
|
|
1
|
+
import { spawnSync } from "node:child_process";
|
|
2
|
+
import { mkdtemp, readFile, rm } from "node:fs/promises";
|
|
3
|
+
import os from "node:os";
|
|
4
|
+
import path from "node:path";
|
|
5
|
+
import { describe, expect, it } from "vitest";
|
|
6
|
+
import { providerGenerate } from "@/modules/ai/providerGenerate.js";
|
|
7
|
+
import { sandboxInferenceGet } from "@/modules/sandbox/sandboxInferenceGet.js";
|
|
8
|
+
const TEST_TIMEOUT_MS = 180_000;
|
|
9
|
+
function commandExists(command) {
|
|
10
|
+
const result = spawnSync("zsh", ["-lc", `command -v ${command}`], {
|
|
11
|
+
encoding: "utf-8"
|
|
12
|
+
});
|
|
13
|
+
return result.status === 0;
|
|
14
|
+
}
|
|
15
|
+
function assertNoPiFlagParseError(stderr) {
|
|
16
|
+
expect(stderr).not.toContain("unknown option '--mode'");
|
|
17
|
+
expect(stderr).not.toContain("unknown option '--print'");
|
|
18
|
+
}
|
|
19
|
+
async function runReadOnly(command) {
|
|
20
|
+
const sandbox = await sandboxInferenceGet({
|
|
21
|
+
writePolicy: { mode: "read-only" }
|
|
22
|
+
});
|
|
23
|
+
const result = await providerGenerate({
|
|
24
|
+
providerId: "pi",
|
|
25
|
+
command,
|
|
26
|
+
prompt: "Reply with <output>ok</output> only.",
|
|
27
|
+
sandbox,
|
|
28
|
+
writePolicy: { mode: "read-only" }
|
|
29
|
+
});
|
|
30
|
+
if (!result.output) {
|
|
31
|
+
const stderr = result.failure?.stderr ?? "";
|
|
32
|
+
assertNoPiFlagParseError(stderr);
|
|
33
|
+
expect(stderr.length).toBeGreaterThan(0);
|
|
34
|
+
return;
|
|
35
|
+
}
|
|
36
|
+
expect(result.output.trim().length).toBeGreaterThan(0);
|
|
37
|
+
}
|
|
38
|
+
async function runSingleFileAllowed(command) {
|
|
39
|
+
const tempDir = await mkdtemp(path.join(os.tmpdir(), "holdmybeer-provider-generate-"));
|
|
40
|
+
try {
|
|
41
|
+
const outputFile = path.join(tempDir, "result.txt");
|
|
42
|
+
const sandbox = await sandboxInferenceGet({
|
|
43
|
+
writePolicy: {
|
|
44
|
+
mode: "write-whitelist",
|
|
45
|
+
writablePaths: [outputFile]
|
|
46
|
+
}
|
|
47
|
+
});
|
|
48
|
+
const result = await providerGenerate({
|
|
49
|
+
providerId: "pi",
|
|
50
|
+
command,
|
|
51
|
+
prompt: [
|
|
52
|
+
`Write EXACTLY this text to ${outputFile}: hello-from-provider`,
|
|
53
|
+
"Do not write to any other file.",
|
|
54
|
+
"You may return plain confirmation text."
|
|
55
|
+
].join("\n"),
|
|
56
|
+
sandbox,
|
|
57
|
+
writePolicy: {
|
|
58
|
+
mode: "write-whitelist",
|
|
59
|
+
writablePaths: [outputFile]
|
|
60
|
+
},
|
|
61
|
+
requireOutputTags: false
|
|
62
|
+
});
|
|
63
|
+
if (result.failure) {
|
|
64
|
+
assertNoPiFlagParseError(result.failure.stderr);
|
|
65
|
+
expect(result.failure.stderr.length).toBeGreaterThan(0);
|
|
66
|
+
return;
|
|
67
|
+
}
|
|
68
|
+
const written = await readFile(outputFile, "utf-8");
|
|
69
|
+
expect(written).toContain("hello-from-provider");
|
|
70
|
+
}
|
|
71
|
+
finally {
|
|
72
|
+
await rm(tempDir, { recursive: true, force: true });
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
describe("providerGenerate pi integration", () => {
|
|
76
|
+
it("runs in read-only sandbox without mock command execution", async () => {
|
|
77
|
+
expect(commandExists("pi")).toBe(true);
|
|
78
|
+
await runReadOnly("pi");
|
|
79
|
+
}, TEST_TIMEOUT_MS);
|
|
80
|
+
it("runs in single-file write-whitelist sandbox without mock command execution", async () => {
|
|
81
|
+
expect(commandExists("pi")).toBe(true);
|
|
82
|
+
await runSingleFileAllowed("pi");
|
|
83
|
+
}, TEST_TIMEOUT_MS);
|
|
84
|
+
});
|
|
85
|
+
//# sourceMappingURL=providerGenerate.spec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providerGenerate.spec.js","sourceRoot":"","sources":["../../../sources/modules/ai/providerGenerate.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AACzD,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,EAAE,mBAAmB,EAAE,MAAM,0CAA0C,CAAC;AAE/E,MAAM,eAAe,GAAG,OAAO,CAAC;AAEhC,SAAS,aAAa,CAAC,OAAe;IACpC,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,cAAc,OAAO,EAAE,CAAC,EAAE;QAChE,QAAQ,EAAE,OAAO;KAClB,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC;AAC7B,CAAC;AAED,SAAS,wBAAwB,CAAC,MAAc;IAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC;IACxD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;AAC3D,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,OAAe;IACxC,MAAM,OAAO,GAAG,MAAM,mBAAmB,CAAC;QACxC,WAAW,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;KACnC,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC;QACpC,UAAU,EAAE,IAAI;QAChB,OAAO;QACP,MAAM,EAAE,sCAAsC;QAC9C,OAAO;QACP,WAAW,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;KACnC,CAAC,CAAC;IAEH,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,EAAE,MAAM,IAAI,EAAE,CAAC;QAC5C,wBAAwB,CAAC,MAAM,CAAC,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACzC,OAAO;IACT,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;AACzD,CAAC;AAED,KAAK,UAAU,oBAAoB,CAAC,OAAe;IACjD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,+BAA+B,CAAC,CAAC,CAAC;IACvF,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,MAAM,mBAAmB,CAAC;YACxC,WAAW,EAAE;gBACX,IAAI,EAAE,iBAAiB;gBACvB,aAAa,EAAE,CAAC,UAAU,CAAC;aAC5B;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC;YACpC,UAAU,EAAE,IAAI;YAChB,OAAO;YACP,MAAM,EAAE;gBACN,8BAA8B,UAAU,uBAAuB;gBAC/D,iCAAiC;gBACjC,yCAAyC;aAC1C,CAAC,IAAI,CAAC,IAAI,CAAC;YACZ,OAAO;YACP,WAAW,EAAE;gBACX,IAAI,EAAE,iBAAiB;gBACvB,aAAa,EAAE,CAAC,UAAU,CAAC;aAC5B;YACD,iBAAiB,EAAE,KAAK;SACzB,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,wBAAwB,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;IACnD,CAAC;YAAS,CAAC;QACT,MAAM,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACtD,CAAC;AACH,CAAC;AAED,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,EAAE,CACA,0DAA0D,EAC1D,KAAK,IAAI,EAAE;QACT,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,WAAW,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC,EACD,eAAe,CAChB,CAAC;IAEF,EAAE,CACA,4EAA4E,EAC5E,KAAK,IAAI,EAAE;QACT,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,oBAAoB,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC,EACD,eAAe,CAChB,CAAC;AACJ,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providerGenerate.unit.spec.d.ts","sourceRoot":"","sources":["../../../sources/modules/ai/providerGenerate.unit.spec.ts"],"names":[],"mappings":""}
|