dac strip io from bedrock (#6154)

* first run at bedrock

* inline tag building

Author: jsimpher

packages/datadog-plugin-aws-sdk/src/services/bedrockruntime/tracing.js

@@ -1,7 +1,7 @@
 'use strict'
 
 const BaseAwsSdkPlugin = require('../../base')
-const { parseModelId, extractRequestParams, extractTextAndResponseReason } = require('./utils')
+const { parseModelId } = require('./utils')
 
 const enabledOperations = new Set(['invokeModel'])
 
@@ -20,44 +20,12 @@ class BedrockRuntime extends BaseAwsSdkPlugin {
   generateTags (params, operation, response) {
     const { modelProvider, modelName } = parseModelId(params.modelId)
 
-    const requestParams = extractRequestParams(params, modelProvider)
-    const textAndResponseReason = extractTextAndResponseReason(response, modelProvider, modelName)
-
-    const tags = buildTagsFromParams(requestParams, textAndResponseReason, modelProvider, modelName, operation)
-
-    return tags
-  }
-}
-
-function buildTagsFromParams (requestParams, textAndResponseReason, modelProvider, modelName, operation) {
-  const tags = {}
-
-  // add request tags
-  tags['resource.name'] = operation
-  tags['aws.bedrock.request.model'] = modelName
-  tags['aws.bedrock.request.model_provider'] = modelProvider.toLowerCase()
-  tags['aws.bedrock.request.prompt'] = requestParams.prompt
-  tags['aws.bedrock.request.temperature'] = requestParams.temperature
-  tags['aws.bedrock.request.top_p'] = requestParams.topP
-  tags['aws.bedrock.request.top_k'] = requestParams.topK
-  tags['aws.bedrock.request.max_tokens'] = requestParams.maxTokens
-  tags['aws.bedrock.request.stop_sequences'] = requestParams.stopSequences
-  tags['aws.bedrock.request.input_type'] = requestParams.inputType
-  tags['aws.bedrock.request.truncate'] = requestParams.truncate
-  tags['aws.bedrock.request.stream'] = requestParams.stream
-  tags['aws.bedrock.request.n'] = requestParams.n
-
-  // add response tags
-  if (modelName.includes('embed')) {
-    tags['aws.bedrock.response.embedding_length'] = textAndResponseReason.message.length
-  }
-  if (textAndResponseReason.choiceId) {
-    tags['aws.bedrock.response.choices.id'] = textAndResponseReason.choiceId
+    return {
+      'resource.name': operation,
+      'aws.bedrock.request.model': modelName,
+      'aws.bedrock.request.model_provider': modelProvider.toLowerCase()
+    }
   }
-  tags['aws.bedrock.response.choices.text'] = textAndResponseReason.message
-  tags['aws.bedrock.response.choices.finish_reason'] = textAndResponseReason.finishReason
-
-  return tags
 }
 
 module.exports = BedrockRuntime

packages/datadog-plugin-aws-sdk/test/bedrockruntime.spec.js

@@ -3,7 +3,7 @@
 const agent = require('../../dd-trace/test/plugins/agent')
 const nock = require('nock')
 const { setup } = require('./spec_helpers')
-const { models, modelConfig } = require('./fixtures/bedrockruntime')
+const { models } = require('./fixtures/bedrockruntime')
 
 const serviceName = 'bedrock-service-name-test'
 
@@ -64,12 +64,6 @@ describe('Plugin', () => {
                 'aws.operation': 'invokeModel',
                 'aws.bedrock.request.model': model.modelId.split('.')[1],
                 'aws.bedrock.request.model_provider': model.provider.toLowerCase(),
-                'aws.bedrock.request.prompt': model.userPrompt
-              })
-              expect(span.metrics).to.include({
-                'aws.bedrock.request.temperature': modelConfig.temperature,
-                'aws.bedrock.request.top_p': modelConfig.topP,
-                'aws.bedrock.request.max_tokens': modelConfig.maxTokens
               })
             }).then(done).catch(done)