Skip to content

Commit 7ace8d7

Browse files
committed
Fixes prompt/action name and settings
1 parent 0b01634 commit 7ace8d7

File tree

10 files changed

+41
-28
lines changed

10 files changed

+41
-28
lines changed

package.json

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4152,7 +4152,7 @@
41524152
"preview"
41534153
]
41544154
},
4155-
"gitlens.ai.generateCloudPatchMessage.customInstructions": {
4155+
"gitlens.ai.generateCreateCloudPatch.customInstructions": {
41564156
"type": "string",
41574157
"default": null,
41584158
"markdownDescription": "Specifies custom instructions to provide to the AI provider when generating a cloud patch title and description",
@@ -4162,7 +4162,7 @@
41624162
"preview"
41634163
]
41644164
},
4165-
"gitlens.ai.generateCodeSuggestMessage.customInstructions": {
4165+
"gitlens.ai.generateCreateCodeSuggest.customInstructions": {
41664166
"type": "string",
41674167
"default": null,
41684168
"markdownDescription": "Specifies custom instructions to provide to the AI provider when generating a code suggest title and description",
@@ -4171,6 +4171,16 @@
41714171
"tags": [
41724172
"preview"
41734173
]
4174+
},
4175+
"gitlens.ai.generateCreatePullRequest.customInstructions": {
4176+
"type": "string",
4177+
"default": null,
4178+
"markdownDescription": "Specifies custom instructions to provide to the AI provider when generating a pull request title and description",
4179+
"scope": "window",
4180+
"order": 500,
4181+
"tags": [
4182+
"preview"
4183+
]
41744184
}
41754185
}
41764186
},

src/commands/createPullRequestOnRemote.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ export class CreatePullRequestOnRemoteCommand extends GlCommandBase {
9898
);
9999
return result?.parsed;
100100
} catch (e) {
101-
void window.showErrorMessage(`Unable to generate pull request message: ${e}`);
101+
void window.showErrorMessage(`Unable to generate pull request details: ${e}`);
102102
return undefined;
103103
}
104104
},

src/config.ts

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -223,10 +223,13 @@ interface AIConfig {
223223
readonly generateStashMessage: {
224224
readonly customInstructions: string;
225225
};
226-
readonly generateCloudPatchMessage: {
226+
readonly generateCreateCloudPatch: {
227227
readonly customInstructions: string;
228228
};
229-
readonly generateCodeSuggestMessage: {
229+
readonly generateCreateCodeSuggest: {
230+
readonly customInstructions: string;
231+
};
232+
readonly generateCreatePullRequest: {
230233
readonly customInstructions: string;
231234
};
232235
readonly gitkraken: {

src/features.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,16 +40,17 @@ export type ProFeatures =
4040
| 'startWork'
4141
| 'associateIssueWithBranch'
4242
| ProAIFeatures;
43-
export type ProAIFeatures = 'generateStashMessage' | 'explainCommit' | 'cloudPatchGenerateTitleAndDescription';
43+
export type ProAIFeatures = 'explainCommit' | 'generateCreateDraft' | 'generateStashMessage';
4444

4545
export type AdvancedFeatures = AdvancedAIFeatures;
46-
export type AdvancedAIFeatures = 'generateChangelog';
46+
export type AdvancedAIFeatures = 'generateChangelog' | 'generateCreatePullRequest';
4747

4848
export type AIFeatures = ProAIFeatures | AdvancedAIFeatures;
4949

5050
export function isAdvancedFeature(feature: PlusFeatures): feature is AdvancedFeatures {
5151
switch (feature) {
5252
case 'generateChangelog':
53+
case 'generateCreatePullRequest':
5354
return true;
5455
default:
5556
return false;

src/git/gitProviderService.ts

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -779,10 +779,11 @@ export class GitProviderService implements Disposable {
779779
feature === 'launchpad' ||
780780
feature === 'startWork' ||
781781
feature === 'associateIssueWithBranch' ||
782-
feature === 'generateStashMessage' ||
783782
feature === 'explainCommit' ||
784-
feature === 'cloudPatchGenerateTitleAndDescription' ||
785-
feature === 'generateChangelog'
783+
feature === 'generateChangelog' ||
784+
feature === 'generateCreateDraft' ||
785+
feature === 'generateCreatePullRequest' ||
786+
feature === 'generateStashMessage'
786787
) {
787788
return { allowed: false, subscription: { current: subscription, required: SubscriptionPlanId.Pro } };
788789
}

src/plus/ai/aiProviderService.ts

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -530,7 +530,7 @@ export class AIProviderService implements Disposable {
530530
progress?: ProgressOptions;
531531
},
532532
): Promise<AISummarizeResult | undefined> {
533-
if (!(await this.ensureFeatureAccess('generateChangelog', source))) {
533+
if (!(await this.ensureFeatureAccess('generateCreatePullRequest', source))) {
534534
return undefined;
535535
}
536536

@@ -547,12 +547,12 @@ export class AIProviderService implements Disposable {
547547
}
548548

549549
const result = await this.sendRequest(
550-
'generate-pullRequestMessage',
550+
'generate-create-pullRequest',
551551
() => ({
552552
diff: diff?.contents ?? '',
553553
data: commits.sort((a, b) => a[1] - b[1]).map(c => c[0]),
554554
context: options?.context ?? '',
555-
instructions: configuration.get('ai.generatePullRequestMessage.customInstructions') ?? '',
555+
instructions: configuration.get('ai.generateCreatePullRequest.customInstructions') ?? '',
556556
}),
557557
m => `Generating pull request details with ${m.name}...`,
558558
source,
@@ -582,7 +582,7 @@ export class AIProviderService implements Disposable {
582582
codeSuggestion?: boolean;
583583
},
584584
): Promise<AISummarizeResult | undefined> {
585-
if (!(await this.ensureFeatureAccess('cloudPatchGenerateTitleAndDescription', sourceContext))) {
585+
if (!(await this.ensureFeatureAccess('generateCreateDraft', sourceContext))) {
586586
return undefined;
587587
}
588588

@@ -598,8 +598,8 @@ export class AIProviderService implements Disposable {
598598
context: options?.context ?? '',
599599
instructions:
600600
(options?.codeSuggestion
601-
? configuration.get('ai.generateCodeSuggestMessage.customInstructions')
602-
: configuration.get('ai.generateCloudPatchMessage.customInstructions')) ?? '',
601+
? configuration.get('ai.generateCreateCodeSuggest.customInstructions')
602+
: configuration.get('ai.generateCreateCloudPatch.customInstructions')) ?? '',
603603
}),
604604
m =>
605605
`Generating ${options?.codeSuggestion ? 'code suggestion' : 'cloud patch'} description with ${

src/plus/ai/models/model.ts

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@ export interface AIModelDescriptor<Provider extends AIProviders = AIProviders, M
2626

2727
export type AIActionType =
2828
| 'generate-commitMessage'
29-
| 'generate-pullRequestMessage'
3029
| 'generate-stashMessage'
3130
| 'generate-changelog'
3231
| `generate-create-${'cloudPatch' | 'codeSuggestion' | 'pullRequest'}`

src/plus/ai/models/promptTemplates.ts

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@ export interface PromptTemplate {
99

1010
export type PromptTemplateContext<T extends AIActionType> = T extends
1111
| 'generate-commitMessage'
12-
| 'generate-pullRequestMessage'
1312
| 'generate-stashMessage'
1413
| 'generate-create-cloudPatch'
1514
| 'generate-create-codeSuggestion'

src/plus/ai/utils/-webview/ai.utils.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,16 +25,16 @@ export function getActionName(action: AIActionType): string {
2525
switch (action) {
2626
case 'generate-commitMessage':
2727
return 'Generate Commit Message';
28-
case 'generate-pullRequestMessage':
29-
return 'Generate Pull Request Details (Preview)';
3028
case 'generate-stashMessage':
3129
return 'Generate Stash Message';
3230
case 'generate-changelog':
33-
return 'Generate Changelog';
31+
return 'Generate Changelog (Preview)';
3432
case 'generate-create-cloudPatch':
3533
return 'Create Cloud Patch Details';
3634
case 'generate-create-codeSuggestion':
3735
return 'Create Code Suggestion Details';
36+
case 'generate-create-pullRequest':
37+
return 'Create Pull Request Details (Preview)';
3838
case 'explain-changes':
3939
return 'Explain Changes';
4040
default:

src/plus/ai/utils/-webview/prompt.utils.ts

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -20,12 +20,6 @@ export function getLocalPromptTemplate<T extends AIActionType>(action: T, _model
2020
template: generateCommitMessageUserPrompt,
2121
variables: ['diff', 'context', 'instructions'],
2222
};
23-
case 'generate-pullRequestMessage':
24-
return {
25-
name: 'Generate Pull Request Details (Preview)',
26-
template: generatePullRequestMessageUserPrompt,
27-
variables: ['diff', 'data', 'context', 'instructions'],
28-
};
2923
case 'generate-stashMessage':
3024
return {
3125
name: 'Generate Stash Message',
@@ -34,7 +28,7 @@ export function getLocalPromptTemplate<T extends AIActionType>(action: T, _model
3428
};
3529
case 'generate-changelog':
3630
return {
37-
name: 'Generate Changelog',
31+
name: 'Generate Changelog (Preview)',
3832
template: generateChangelogUserPrompt,
3933
variables: ['data', 'instructions'],
4034
};
@@ -50,6 +44,12 @@ export function getLocalPromptTemplate<T extends AIActionType>(action: T, _model
5044
template: generateCodeSuggestMessageUserPrompt,
5145
variables: ['diff', 'context', 'instructions'],
5246
};
47+
case 'generate-create-pullRequest':
48+
return {
49+
name: 'Generate Pull Request Details (Preview)',
50+
template: generatePullRequestMessageUserPrompt,
51+
variables: ['diff', 'data', 'context', 'instructions'],
52+
};
5353
case 'explain-changes':
5454
return {
5555
name: 'Explain Changes',

0 commit comments

Comments
 (0)