154 lines
4.6 KiB
Rust
154 lines
4.6 KiB
Rust
pub mod ws_adapter;
|
|
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
pub struct SessionSummary {
|
|
pub session_id: String,
|
|
pub title: String,
|
|
pub channel_name: String,
|
|
pub chat_id: String,
|
|
pub message_count: i64,
|
|
pub last_active_at: i64,
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
pub archived_at: Option<i64>,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
#[serde(tag = "type")]
|
|
pub enum WsInbound {
|
|
#[serde(rename = "user_input")]
|
|
UserInput {
|
|
content: String,
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
channel: Option<String>,
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
chat_id: Option<String>,
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
sender_id: Option<String>,
|
|
},
|
|
#[serde(rename = "clear_history")]
|
|
ClearHistory {
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
chat_id: Option<String>,
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
session_id: Option<String>,
|
|
},
|
|
#[serde(rename = "create_session")]
|
|
CreateSession {
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
title: Option<String>,
|
|
},
|
|
#[serde(rename = "list_sessions")]
|
|
ListSessions {
|
|
#[serde(default)]
|
|
include_archived: bool,
|
|
},
|
|
#[serde(rename = "load_session")]
|
|
LoadSession { session_id: String },
|
|
#[serde(rename = "rename_session")]
|
|
RenameSession {
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
session_id: Option<String>,
|
|
title: String,
|
|
},
|
|
#[serde(rename = "archive_session")]
|
|
ArchiveSession {
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
session_id: Option<String>,
|
|
},
|
|
#[serde(rename = "delete_session")]
|
|
DeleteSession {
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
session_id: Option<String>,
|
|
},
|
|
#[serde(rename = "save_session")]
|
|
SaveSession {
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
filepath: Option<String>,
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
session_id: Option<String>,
|
|
},
|
|
#[serde(rename = "ping")]
|
|
Ping,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
#[serde(tag = "type")]
|
|
pub enum WsOutbound {
|
|
#[serde(rename = "assistant_response")]
|
|
AssistantResponse {
|
|
id: String,
|
|
content: String,
|
|
role: String,
|
|
},
|
|
#[serde(rename = "tool_call")]
|
|
ToolCall {
|
|
id: String,
|
|
tool_call_id: String,
|
|
tool_name: String,
|
|
arguments: serde_json::Value,
|
|
content: String,
|
|
role: String,
|
|
},
|
|
#[serde(rename = "tool_result")]
|
|
ToolResult {
|
|
id: String,
|
|
tool_call_id: String,
|
|
tool_name: String,
|
|
content: String,
|
|
role: String,
|
|
},
|
|
#[serde(rename = "tool_pending")]
|
|
ToolPending {
|
|
id: String,
|
|
tool_call_id: String,
|
|
tool_name: String,
|
|
content: String,
|
|
role: String,
|
|
resume_hint: String,
|
|
},
|
|
#[serde(rename = "error")]
|
|
Error { code: String, message: String },
|
|
#[serde(rename = "session_established")]
|
|
SessionEstablished { session_id: String },
|
|
#[serde(rename = "session_created")]
|
|
SessionCreated { session_id: String, title: String },
|
|
#[serde(rename = "session_list")]
|
|
SessionList {
|
|
sessions: Vec<SessionSummary>,
|
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
current_session_id: Option<String>,
|
|
},
|
|
#[serde(rename = "session_loaded")]
|
|
SessionLoaded {
|
|
session_id: String,
|
|
title: String,
|
|
message_count: i64,
|
|
},
|
|
#[serde(rename = "session_renamed")]
|
|
SessionRenamed { session_id: String, title: String },
|
|
#[serde(rename = "session_archived")]
|
|
SessionArchived { session_id: String },
|
|
#[serde(rename = "session_deleted")]
|
|
SessionDeleted { session_id: String },
|
|
#[serde(rename = "history_cleared")]
|
|
HistoryCleared { session_id: String },
|
|
#[serde(rename = "session_saved")]
|
|
SessionSaved { session_id: String, filepath: String },
|
|
#[serde(rename = "pong")]
|
|
Pong,
|
|
}
|
|
|
|
pub fn parse_inbound(raw: &str) -> Result<WsInbound, serde_json::Error> {
|
|
serde_json::from_str(raw)
|
|
}
|
|
|
|
pub fn serialize_inbound(msg: &WsInbound) -> Result<String, serde_json::Error> {
|
|
serde_json::to_string(msg)
|
|
}
|
|
|
|
pub fn serialize_outbound(msg: &WsOutbound) -> Result<String, serde_json::Error> {
|
|
serde_json::to_string(msg)
|
|
}
|