File tree Expand file tree Collapse file tree 1 file changed +12
-2
lines changed
aws-distro-opentelemetry-node-autoinstrumentation/test/patches/aws/services Expand file tree Collapse file tree 1 file changed +12
-2
lines changed Original file line number Diff line number Diff line change @@ -277,8 +277,18 @@ describe('BedrockRuntime', () => {
277277
278278 describe ( 'InvokeModel' , ( ) => {
279279 it ( 'adds modelId to span' , async ( ) => {
280- const dummyModelId : string = 'ABCDEFGH' ;
281- const dummyBody : string = 'HGFEDCBA' ;
280+ const dummyModelId : string = 'amazon.titan-text-express-v1' ;
281+ const prompt = "Complete this text. It was the best of times it was the worst..." ;
282+ const nativeRequest = {
283+ inputText : prompt ,
284+ textGenerationConfig : {
285+ maxTokenCount : 4096 ,
286+ stopSequences : [ ] ,
287+ temperature : 0 ,
288+ topP : 1 ,
289+ } ,
290+ } ;
291+ const dummyBody : string = JSON . stringify ( nativeRequest ) ;
282292
283293 nock ( `https://bedrock-runtime.${ region } .amazonaws.com` ) . post ( `/model/${ dummyModelId } /invoke` ) . reply ( 200 , {
284294 modelId : dummyModelId ,
You can’t perform that action at this time.
0 commit comments