AWS Bedrock JS SDK 的开源仪器

本模块为下列人员提供自动仪器: AWS Bedrock 运行时间,可以与 @ opleteometry/ sdk- trace- 节点。 。 。 .

安装

npm install --save @arizeai/openinference-instrumentation-bedrock

使用量

要装入 Bedrock 仪表, 请在仪表簿中指定它 .

const { NodeTracerProvider } = require("@opentelemetry/sdk-trace-node");
const { BedrockInstrumentation } = require("@arizeai/openinference-instrumentation-bedrock");
const { registerInstrumentations } = require("@opentelemetry/instrumentation");

const provider = new NodeTracerProvider();
provider.register();

registerInstrumentations({
  instrumentations: [new BedrockInstrumentation()],
});

实例

要运行示例, 请运行以下命令:

cd js/packages/openinference-instrumentation-bedrock
pnpm install
pnpm -r build
npx tsx scripts/validate-invoke-model.ts # or validate-converse-comprehensive.ts

关于OpenTeleometry Node.js SDK的更多信息,请参见: OpenTeleometry 节点.js SDK 文档。 。 。 .

使用自定义追踪提供者

您可以在创建 Bedrock 仪器时指定自定义的跟踪器提供者 。 当您想要使用非全局追踪器提供者或对追踪配置有更强的控制时,此功能是有用的.

import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
import { Resource } from "@opentelemetry/resources";
import { SEMRESATTRS_PROJECT_NAME } from "@arizeai/openinference-semantic-conventions";
import { BedrockInstrumentation } from "@arizeai/openinference-instrumentation-bedrock";
import { registerInstrumentations } from "@opentelemetry/instrumentation";

// Create a custom tracer provider
const customTracerProvider = new NodeTracerProvider({
  resource: new Resource({
    [SEMRESATTRS_PROJECT_NAME]: "my-bedrock-project",
  }),
});

// Pass the custom tracer provider to the instrumentation
const bedrockInstrumentation = new BedrockInstrumentation({
  tracerProvider: customTracerProvider,
});

// Register the instrumentation with the custom tracer provider
registerInstrumentations({
  instrumentations: [bedrockInstrumentation],
  tracerProvider: customTracerProvider,
});

或者,您可以在创建仪器后设置跟踪提供者:

const bedrockInstrumentation = new BedrockInstrumentation();
bedrockInstrumentation.setTracerProvider(customTracerProvider);

兼容性

@arizeai/openinference-instrumentation-bedrock 与以下版本兼容 @aws-sdk/client-bedrock-runtime 软件包 :