Skip to content

Commit 47d3c44

Browse files
committed
feat: add token methods to StreamingClientServiceBase
1 parent 96022cc commit 47d3c44

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

server/aws-lsp-codewhisperer/src/shared/streamingClientService.ts

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,16 @@ export abstract class StreamingClientServiceBase {
4949
abortController?: AbortController
5050
): Promise<SendMessageCommandOutput>
5151

52+
abstract generateAssistantResponse(
53+
request: GenerateAssistantResponseCommandInputCodeWhispererStreaming,
54+
abortController?: AbortController
55+
): Promise<GenerateAssistantResponseCommandOutputCodeWhispererStreaming>
56+
57+
abstract exportResultArchive(
58+
request: ExportResultArchiveCommandInputCodeWhispererStreaming,
59+
abortController?: AbortController
60+
): Promise<ExportResultArchiveCommandOutputCodeWhispererStreaming>
61+
5262
public abortInflightRequests() {
5363
this.inflightRequests.forEach(abortController => {
5464
abortController.abort()
@@ -209,4 +219,17 @@ export class StreamingClientServiceIAM extends StreamingClientServiceBase {
209219

210220
return response
211221
}
222+
223+
public async generateAssistantResponse(
224+
request: GenerateAssistantResponseCommandInputCodeWhispererStreaming,
225+
abortController?: AbortController
226+
): Promise<GenerateAssistantResponseCommandOutputCodeWhispererStreaming> {
227+
throw new Error('Method not implemented.')
228+
}
229+
public async exportResultArchive(
230+
request: ExportResultArchiveCommandInputCodeWhispererStreaming,
231+
abortController?: AbortController
232+
): Promise<ExportResultArchiveCommandOutputCodeWhispererStreaming> {
233+
throw new Error('Method not implemented.')
234+
}
212235
}

0 commit comments

Comments
 (0)