langfuse-client-base 0.17.0

/*
 * server
 *
 * ## Real-time data  The only real-time path to get tracing data is to ingest via OpenTelemetry (`POST /api/public/otel/v1/traces`) and retrieve via Observations API v2 (`GET /api/public/v2/observations`) and Metrics API v2 (`GET /api/public/v2/metrics`). Other public API write and read endpoints can delay data by about 10 minutes. Direct OpenTelemetry exporters must send the `x-langfuse-ingestion-version: 4` header; current Python and JS SDKs already do. See the [Public API docs](https://langfuse.com/docs/api-and-data-platform/features/public-api) and [real-time ingestion](https://langfuse.com/integrations/native/opentelemetry.md#real-time-ingestion).  ## Authentication  Authenticate with the API using [Basic Auth](https://en.wikipedia.org/wiki/Basic_access_authentication), get API keys in the project settings:  - username: Langfuse Public Key - password: Langfuse Secret Key  ## Exports  - OpenAPI spec: https://cloud.langfuse.com/generated/api/openapi.yml - Self-hosted deployments: interactive API reference at `/api/docs` and OpenAPI spec at `/api/openapi.yaml`
 *
 * The version of the OpenAPI document:
 *
 * Generated by: https://openapi-generator.tech
 */

use crate::models;
use serde::{Deserialize, Serialize};

#[derive(Clone, Debug, PartialEq, Serialize)]
#[serde(untagged)]
pub enum IngestionEvent {
    TraceEvent(models::IngestionEventTraceEvent),
    ScoreEvent(models::IngestionEventScoreEvent),
    CreateSpanEvent(models::IngestionEventCreateSpanEvent),
    UpdateSpanEvent(models::IngestionEventUpdateSpanEvent),
    CreateGenerationEvent(models::IngestionEventCreateGenerationEvent),
    UpdateGenerationEvent(models::IngestionEventUpdateGenerationEvent),
    CreateEventEvent(models::IngestionEventCreateEventEvent),
    CreateObservationEvent(models::IngestionEventCreateObservationEvent),
    UpdateObservationEvent(models::IngestionEventUpdateObservationEvent),
}

impl<'de> Deserialize<'de> for IngestionEvent {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        let value = serde_json::Value::deserialize(deserializer)?;
        Self::dispatch(value).map_err(serde::de::Error::custom)
    }
}

impl IngestionEvent {
    /// Decodes the variant that `type` selects, so an error is that variant's.
    #[inline]
    fn dispatch(value: serde_json::Value) -> Result<Self, String> {
        use models::union_dispatch::{self as dispatch, Tag};
        const UNION: &str = "IngestionEvent";
        const TAG: &str = "type";
        match dispatch::tag(&value, UNION, TAG)? {
            Tag::Value("trace-create") => {
                dispatch::decode(value, UNION, "TraceEvent").map(Self::TraceEvent)
            }
            Tag::Value("score-create") => {
                dispatch::decode(value, UNION, "ScoreEvent").map(Self::ScoreEvent)
            }
            Tag::Value("span-create") => {
                dispatch::decode(value, UNION, "CreateSpanEvent").map(Self::CreateSpanEvent)
            }
            Tag::Value("span-update") => {
                dispatch::decode(value, UNION, "UpdateSpanEvent").map(Self::UpdateSpanEvent)
            }
            Tag::Value("generation-create") => {
                dispatch::decode(value, UNION, "CreateGenerationEvent")
                    .map(Self::CreateGenerationEvent)
            }
            Tag::Value("generation-update") => {
                dispatch::decode(value, UNION, "UpdateGenerationEvent")
                    .map(Self::UpdateGenerationEvent)
            }
            Tag::Value("event-create") => {
                dispatch::decode(value, UNION, "CreateEventEvent").map(Self::CreateEventEvent)
            }
            Tag::Value("observation-create") => {
                dispatch::decode(value, UNION, "CreateObservationEvent")
                    .map(Self::CreateObservationEvent)
            }
            Tag::Value("observation-update") => {
                dispatch::decode(value, UNION, "UpdateObservationEvent")
                    .map(Self::UpdateObservationEvent)
            }
            Tag::Value(other) => Err(dispatch::unknown_tag(UNION, TAG, other)),
            Tag::Missing => Err(dispatch::missing_tag(UNION, TAG)),
            Tag::NotObject => Err(dispatch::not_an_object(&value, UNION, TAG)),
        }
    }
}

impl std::fmt::Display for IngestionEvent {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::TraceEvent(value) => match serde_json::to_string(value) {
                Ok(s) => write!(f, "{}", s),
                Err(_) => Err(std::fmt::Error),
            },
            Self::ScoreEvent(value) => match serde_json::to_string(value) {
                Ok(s) => write!(f, "{}", s),
                Err(_) => Err(std::fmt::Error),
            },
            Self::CreateSpanEvent(value) => match serde_json::to_string(value) {
                Ok(s) => write!(f, "{}", s),
                Err(_) => Err(std::fmt::Error),
            },
            Self::UpdateSpanEvent(value) => match serde_json::to_string(value) {
                Ok(s) => write!(f, "{}", s),
                Err(_) => Err(std::fmt::Error),
            },
            Self::CreateGenerationEvent(value) => match serde_json::to_string(value) {
                Ok(s) => write!(f, "{}", s),
                Err(_) => Err(std::fmt::Error),
            },
            Self::UpdateGenerationEvent(value) => match serde_json::to_string(value) {
                Ok(s) => write!(f, "{}", s),
                Err(_) => Err(std::fmt::Error),
            },
            Self::CreateEventEvent(value) => match serde_json::to_string(value) {
                Ok(s) => write!(f, "{}", s),
                Err(_) => Err(std::fmt::Error),
            },
            Self::CreateObservationEvent(value) => match serde_json::to_string(value) {
                Ok(s) => write!(f, "{}", s),
                Err(_) => Err(std::fmt::Error),
            },
            Self::UpdateObservationEvent(value) => match serde_json::to_string(value) {
                Ok(s) => write!(f, "{}", s),
                Err(_) => Err(std::fmt::Error),
            },
        }
    }
}