Python SDK - CoreWeave Docs

The CoreWeave Forge Python SDK (coreweave) records agent conversations, turns, LLM calls, tool calls, and sub-agents as OpenTelemetry spans and exports them to Agent Lens. This reference covers version 0.1.0b0 of the SDK.

Install

The SDK requires Python 3.10 or later.

pip install coreweave

To trace a provider SDK automatically, install the matching extra:

pip install "coreweave[openai]"
pip install "coreweave[anthropic]"
pip install "coreweave[google-genai]"

Import and initialize

Import the tracing module, then call init with your project in entity/project form. init reads your API key from the WANDB_API_KEY environment variable unless you pass api_key.

from coreweave.forge.agentlens import tracing

tracing.init("[ENTITY]/[PROJECT]")

Reference pages

Setup

Page	Covers
Setup and shutdown	`init`, `force_flush`, `shutdown`, `TracingSession`

Spans

Page	Covers
Conversation	`Conversation`
Turn	`Turn`
LLM	`LLM`
Tool	`Tool`
SubAgent	`SubAgent`
Span functions	`start_conversation`, `start_turn`, `start_llm`, `start_tool`, `start_subagent`, `end_conversation`, `end_turn`, `end_llm`, `get_current_conversation`, `get_current_turn`, `get_current_llm`

Batch logging

Page	Covers
Batch logging	`log_turn`, `log_conversation`, `LogResult`

Messages

Page	Covers
Message types	`Message`, `MessagePart`, `TextPart`, `ReasoningPart`, `ToolCallPart`, `ToolCallResponsePart`, `BlobPart`, `UriPart`, `FilePart`, `Usage`, `Reasoning`, `MediaAttachment`

Provider integrations

Page	Covers
Integrations	`autopatch`, `disable_autopatch`, `patch_openai`, `unpatch_openai`, `patch_anthropic`, `unpatch_anthropic`, `patch_google_genai`, `unpatch_google_genai`, `unpatch_all`

​Install

​Import and initialize

​Reference pages

​Setup

​Spans

​Batch logging

​Messages

​Provider integrations