lc-core 0.16.0

use async_trait::async_trait;

/// 输出解析器的统一错误类型
#[derive(Debug, Clone, thiserror::Error)]
#[non_exhaustive]
pub enum OutputParserError {
    /// 解析失败：输入格式不符合预期
    #[error("Parse error: {0}")]
    ParseError(String),
    /// JSON 格式错误
    #[error("JSON error: {0}")]
    JsonError(String),
    /// 类型转换错误
    #[error("Type error: {0}")]
    TypeError(String),
    /// 自定义错误
    #[error("{0}")]
    Custom(String),
}

impl From<serde_json::Error> for OutputParserError {
    fn from(e: serde_json::Error) -> Self {
        OutputParserError::JsonError(e.to_string())
    }
}

/// 输出解析器的结果类型
pub type OutputParserResult<T> = Result<T, OutputParserError>;

/// 输出解析器的核心 trait
///
/// 所有输出解析器必须实现此 trait。
/// 与 `Runnable` 不同，`parse` 不接收 config 参数，
/// 适合在 Runnable 内部调用。
#[async_trait]
pub trait BaseOutputParser<Output: Send + Sync + 'static>: Send + Sync {
    /// 将原始 LLM 输出文本解析为目标类型
    async fn parse(&self, text: &str) -> OutputParserResult<Output>;

    /// 带重试的解析（默认实现：真正重试 `max_retries` 次）
    ///
    /// 对同一份文本反复调用 [`parse`](Self::parse)，最多尝试
    /// `max_retries + 1` 次。重试同一份文本只对非确定性解析（例如内部
    /// 依赖网络/外部服务的解析器）有意义；确定性解析器首次失败后必然
    /// 重复失败，最终返回最后一次错误。需要基于失败原因修正输入的
    /// 解析器应覆写此方法。
    async fn parse_with_retry(&self, text: &str, max_retries: usize) -> OutputParserResult<Output> {
        let mut last_err = None;
        for _ in 0..=max_retries {
            match self.parse(text).await {
                Ok(output) => return Ok(output),
                Err(e) => last_err = Some(e),
            }
        }
        Err(last_err.unwrap_or_else(|| {
            OutputParserError::ParseError("parse_with_retry made no parse attempts".to_string())
        }))
    }

    /// 获取格式指令（用于提示 LLM 按指定格式输出）
    fn get_format_instructions(&self) -> String {
        String::new()
    }
}