2
2
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
3
3
* SPDX-License-Identifier: Apache-2.0
4
4
*/
5
- import { GenerateAssistantResponseRequest , SupplementaryWebLink , Reference } from '@amzn/codewhisperer-streaming'
5
+ import {
6
+ GenerateAssistantResponseRequest ,
7
+ SupplementaryWebLink ,
8
+ Reference ,
9
+ UserIntent ,
10
+ } from '@amzn/codewhisperer-streaming'
6
11
7
12
import {
8
13
GenerateResourceRequestMessage ,
@@ -19,7 +24,7 @@ const TIMEOUT = 30_000
19
24
20
25
export async function generateResourceHandler ( request : GenerateResourceRequestMessage , context : WebviewContext ) {
21
26
try {
22
- const { chatResponse, references, metadata, isSuccess } = await generateResource ( request . prompt )
27
+ const { chatResponse, references, metadata, isSuccess } = await generateResource ( request . cfnType )
23
28
24
29
const responseMessage : GenerateResourceResponseMessage = {
25
30
command : Command . GENERATE_RESOURCE ,
@@ -49,7 +54,7 @@ export async function generateResourceHandler(request: GenerateResourceRequestMe
49
54
}
50
55
}
51
56
52
- async function generateResource ( prompt : string ) {
57
+ async function generateResource ( cfnType : string ) {
53
58
let startTime = globals . clock . Date . now ( )
54
59
55
60
try {
@@ -61,7 +66,8 @@ async function generateResource(prompt: string) {
61
66
conversationState : {
62
67
currentMessage : {
63
68
userInputMessage : {
64
- content : prompt ,
69
+ content : cfnType ,
70
+ userIntent : UserIntent . GENERATE_CLOUDFORMATION_TEMPLATE ,
65
71
} ,
66
72
} ,
67
73
chatTriggerType : 'MANUAL' ,
@@ -99,7 +105,6 @@ async function generateResource(prompt: string) {
99
105
throw new Error ( 'Invalid model response' )
100
106
}
101
107
}
102
-
103
108
if ( value ?. messageMetadataEvent ?. conversationId ) {
104
109
conversationId = value . messageMetadataEvent . conversationId
105
110
}
@@ -134,7 +139,7 @@ async function generateResource(prompt: string) {
134
139
135
140
getLogger ( ) . debug (
136
141
`CW Chat Debug message:
137
- prompt = "${ prompt } ",
142
+ cfnType = "${ cfnType } ",
138
143
conversationId = ${ conversationId } ,
139
144
metadata = \n${ JSON . stringify ( metadata , undefined , 2 ) } ,
140
145
supplementaryWebLinks = \n${ JSON . stringify ( supplementaryWebLinks , undefined , 2 ) } ,
0 commit comments