9418ab9815
* feat: add provider trace backend abstraction for multi-backend telemetry Introduces a pluggable backend system for provider traces: - Base class with async/sync create and read interfaces - PostgreSQL backend (existing behavior) - ClickHouse backend (via OTEL instrumentation) - Socket backend (writes to Unix socket for crouton sidecar) - Factory for instantiating backends from config Refactors TelemetryManager to use backends with support for: - Multi-backend writes (concurrent via asyncio.gather) - Primary backend for reads (first in config list) - Graceful error handling per backend Config: LETTA_TELEMETRY_PROVIDER_TRACE_BACKEND (comma-separated) Example: "postgres,socket" for dual-write to Postgres and crouton 🐙 Generated with [Letta Code](https://letta.com) Co-Authored-By: Letta <noreply@letta.com> * feat: add protocol version to socket backend records Adds PROTOCOL_VERSION constant to socket backend: - Included in every telemetry record sent to crouton - Must match ProtocolVersion in apps/crouton/main.go - Enables crouton to detect and reject incompatible messages 🐙 Generated with [Letta Code](https://letta.com) Co-Authored-By: Letta <noreply@letta.com> * fix: remove organization_id from ProviderTraceCreate calls The organization_id is now handled via the actor parameter in the telemetry manager, not through ProviderTraceCreate schema. This fixes validation errors after changing ProviderTraceCreate to inherit from BaseProviderTrace which forbids extra fields. 🐙 Generated with [Letta Code](https://letta.com) Co-Authored-By: Letta <noreply@letta.com> * consolidate provider trace * add clickhouse-connect to fix bug on main lmao * auto generated sdk changes, and deployment details, and clikchouse prefix bug and added fields to runs trace return api * auto generated sdk changes, and deployment details, and clikchouse prefix bug and added fields to runs trace return api * consolidate provider trace * consolidate provider trace bug fix --------- Co-authored-by: Letta <noreply@letta.com>
46 lines
1.8 KiB
Python
46 lines
1.8 KiB
Python
"""PostgreSQL provider trace backend."""
|
|
|
|
from letta.helpers.json_helpers import json_dumps, json_loads
|
|
from letta.orm.provider_trace import ProviderTrace as ProviderTraceModel
|
|
from letta.schemas.provider_trace import ProviderTrace
|
|
from letta.schemas.user import User
|
|
from letta.server.db import db_registry
|
|
from letta.services.provider_trace_backends.base import ProviderTraceBackendClient
|
|
|
|
|
|
class PostgresProviderTraceBackend(ProviderTraceBackendClient):
|
|
"""Store provider traces in PostgreSQL."""
|
|
|
|
async def create_async(
|
|
self,
|
|
actor: User,
|
|
provider_trace: ProviderTrace,
|
|
) -> ProviderTrace:
|
|
async with db_registry.async_session() as session:
|
|
provider_trace_model = ProviderTraceModel(**provider_trace.model_dump())
|
|
provider_trace_model.organization_id = actor.organization_id
|
|
|
|
if provider_trace.request_json:
|
|
request_json_str = json_dumps(provider_trace.request_json)
|
|
provider_trace_model.request_json = json_loads(request_json_str)
|
|
|
|
if provider_trace.response_json:
|
|
response_json_str = json_dumps(provider_trace.response_json)
|
|
provider_trace_model.response_json = json_loads(response_json_str)
|
|
|
|
await provider_trace_model.create_async(session, actor=actor, no_commit=True, no_refresh=True)
|
|
return provider_trace_model.to_pydantic()
|
|
|
|
async def get_by_step_id_async(
|
|
self,
|
|
step_id: str,
|
|
actor: User,
|
|
) -> ProviderTrace | None:
|
|
async with db_registry.async_session() as session:
|
|
provider_trace_model = await ProviderTraceModel.read_async(
|
|
db_session=session,
|
|
step_id=step_id,
|
|
actor=actor,
|
|
)
|
|
return provider_trace_model.to_pydantic() if provider_trace_model else None
|