Browse Source

feat: Add userAgent to Bedrock client for version tracking (#8663)

Co-authored-by: Roo Code <[email protected]>
roomote[bot] 2 months ago
parent
commit
6d6b836480
1 changed files with 2 additions and 0 deletions
  1. 2 0
      src/api/providers/bedrock.ts

+ 2 - 0
src/api/providers/bedrock.ts

@@ -27,6 +27,7 @@ import {
 import { ApiStream } from "../transform/stream"
 import { BaseProvider } from "./base-provider"
 import { logger } from "../../utils/logging"
+import { Package } from "../../shared/package"
 import { MultiPointStrategy } from "../transform/cache-strategy/multi-point-strategy"
 import { ModelInfo as CacheModelInfo } from "../transform/cache-strategy/types"
 import { convertToBedrockConverseMessages as sharedConverter } from "../transform/bedrock-converse-format"
@@ -218,6 +219,7 @@ export class AwsBedrockHandler extends BaseProvider implements SingleCompletionH
 		this.costModelConfig = this.getModel()
 
 		const clientConfig: BedrockRuntimeClientConfig = {
+			defaultUserAgentProvider: () => Promise.resolve([["RooCode", Package.version]]),
 			region: this.options.awsRegion,
 			// Add the endpoint configuration when specified and enabled
 			...(this.options.awsBedrockEndpoint &&