use serde::{Deserialize, Serialize};
use serde_json::{Map, Value};

pub use appflowy_ai_client::dto::*;
use database_entity::dto::ChatMessage;

#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct SummarizeRowParams {
  pub workspace_id: String,
  pub data: SummarizeRowData,
}

/// Represents different types of content that can be used to summarize a database row.
#[derive(Clone, Debug, Serialize, Deserialize)]
pub enum SummarizeRowData {
  /// Specifies the identity of the row within the database.
  Identity { database_id: String, row_id: String },
  /// Content of the row provided as key-value pairs.
  /// For example:
  /// ```json
  /// {
  ///  "name": "Jack",
  ///  "age": 25,
  ///  "city": "New York"
  /// }
  Content(Map<String, Value>),
}

#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct SummarizeRowResponse {
  pub text: String,
}

#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct CompleteTextParams {
  pub text: String,
  pub completion_type: Option<CompletionType>,
  pub custom_prompt: Option<CustomPrompt>,
}

impl CompleteTextParams {
  pub fn new_with_completion_type(text: String, completion_type: CompletionType) -> Self {
    Self {
      text,
      completion_type: Some(completion_type),
      custom_prompt: None,
    }
  }
}

#[derive(Debug)]
pub enum StringOrMessage {
  Left(String),
  Right(ChatMessage),
}