/ mlflow / claude_code / __init__.py
__init__.py
 1  """Claude Code integration for MLflow.
 2  
 3  This module provides automatic tracing of Claude Code conversations to MLflow.
 4  
 5  Usage:
 6      mlflow autolog claude [directory] [options]
 7  
 8  After setup, use the regular 'claude' command and traces will be automatically captured.
 9  
10  To enable tracing for the Claude Agent SDK, use `mlflow.anthropic.autolog()`.
11  
12  Example:
13  
14  ```python
15  import mlflow.anthropic
16  from claude_agent_sdk import ClaudeSDKClient
17  
18  mlflow.anthropic.autolog()
19  
20  async with ClaudeSDKClient() as client:
21      await client.query("What is the capital of France?")
22  
23      async for message in client.receive_response():
24          print(message)
25  ```
26  """