AWS Bedrock Agent 的 Open Information 仪器运行时

本模块为下列人员提供自动仪器: AWS 贝德洛克代理运行时客户端 可用于与 @ opleteometry/ sdk- trace- 节点。 。 。 .

安装

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

使用量

要装入 Bedrock Agent 运行时仪表, 请在寄存器中指定它 .

import { NodeTracerProvider } = from "@opentelemetry/sdk-trace-node";
import {
  BedrockAgentInstrumentation,
} = from "@arizeai/openinference-instrumentation-bedrock-agent-runtime";
import { registerInstrumentations } = "@opentelemetry/instrumentation";

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

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

手动仪器

对于进口订单难以控制的情况,您可以手动为 Bedrock Agent Runtime 客户端提供设备:

import { BedrockAgentInstrumentation } from "@arizeai/openinference-instrumentation-bedrock-agent-runtime";
import * as bedrockAgentRuntime from "@aws-sdk/client-bedrock-agent-runtime";

const instrumentation = new BedrockAgentInstrumentation();
instrumentation.manuallyInstrument(bedrockAgentRuntime);

// Now use the Bedrock Agent Runtime client as normal
const client = new bedrockAgentRuntime.BedrockAgentRuntimeClient({
  region: "us-east-1",
});

实例

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

cd js/packages/openinference-instrumentation-bedrock-agent-runtime
pnpm install
pnpm -r build
npx -y tsx examples/run-invoke-agent.ts

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

使用自定义追踪提供者

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

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

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

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

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

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

const bedrockAgentInstrumentation = new BedrockAgentInstrumentation();
bedrockAgentInstrumentation.setTracerProvider(customTracerProvider);

兼容性

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