use std::collections::BTreeMap;
use std::path::PathBuf;
use schemars::JsonSchema;
use serde::{Deserialize, Serialize};
use serde_json::Value;
use ts_rs::TS;
use super::item::AppItem;
use super::turn::AppTurn;
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadStartParams {
#[serde(default = "default_agent_key")]
pub agent_key: String,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub cwd: Option<PathBuf>,
#[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
pub metadata: BTreeMap<String, Value>,
}
impl Default for ThreadStartParams {
fn default() -> Self {
Self {
agent_key: default_agent_key(),
cwd: None,
metadata: BTreeMap::new(),
}
}
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadResumeParams {
pub thread_id: String,
#[serde(default = "default_subscribe")]
pub subscribe: bool,
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadReadParams {
pub thread_id: String,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub after_item_id: Option<String>,
}
#[derive(Debug, Clone, Default, PartialEq, Eq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadListParams {
#[serde(default)]
pub include_archived: bool,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub archived: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub offset: Option<usize>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub limit: Option<usize>,
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadArchiveParams {
pub thread_id: String,
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadUnsubscribeParams {
pub thread_id: String,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadStartResponse {
pub thread_id: String,
pub agent_key: String,
#[serde(default)]
pub cwd: Option<PathBuf>,
pub status: ThreadStatus,
}
impl ThreadStartResponse {
pub fn from_thread(thread: &AppThread) -> Self {
Self {
thread_id: thread.thread_id.clone(),
agent_key: thread.agent_key.clone(),
cwd: thread.cwd.clone(),
status: thread.status,
}
}
}
pub type ThreadStartedParams = ThreadStartResponse;
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadResumeResponse {
pub thread: AppThread,
pub turns: Vec<AppTurn>,
pub items: Vec<AppItem>,
}
pub type ThreadReadResponse = ThreadResumeResponse;
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadListResponse {
pub threads: Vec<AppThread>,
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadArchiveResponse {
pub thread_id: String,
pub archived: bool,
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadUnsubscribeResponse {
pub thread_id: String,
pub subscribed: bool,
pub closed: bool,
}
pub type ThreadArchivedParams = ThreadArchiveResponse;
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadClosedParams {
pub thread_id: String,
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct ThreadStatusChangedParams {
pub thread_id: String,
pub status: ThreadStatus,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub struct AppThread {
pub thread_id: String,
pub agent_key: String,
#[serde(default)]
pub cwd: Option<PathBuf>,
pub created_at: f64,
pub updated_at: f64,
#[serde(default)]
pub archived_at: Option<f64>,
pub status: ThreadStatus,
#[serde(default)]
pub metadata: BTreeMap<String, Value>,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize, JsonSchema, TS)]
#[serde(rename_all = "camelCase")]
pub enum ThreadStatus {
Idle,
Running,
Archived,
Closed,
}
fn default_agent_key() -> String {
"default".to_string()
}
fn default_subscribe() -> bool {
true
}