mattrguerra.comHouston area
Matt Guerra
  1. 01AI & Automation
  2. 02Photography
  3. 03Design
  4. 04About
  5. 05Notes
  6. 06Contact
Back to the systems

mattrguerra/cortexmainPrivate on GitHub, published here in full

api.rs

Rust6.8 KB236 lines

crates/cortex-core/src/api.rs

  1. //! The shapes that cross from Rust to the interface (commands and events).
  3. use serde::{Deserialize, Serialize};
  4. use ts_rs::TS;
  6. use crate::pricing::ModelInfo;
  7. use crate::storage::{Conversation, Message, UsageRow};
  8. use crate::types::Usage;
  10. /// Events on a reply's stream, in order: deltas, maybe usage, then `done` or `error`.
  11. #[derive(Debug, Clone, PartialEq, Serialize, Deserialize, TS)]
  12. #[serde(tag = "type", rename_all = "snake_case")]
  13. #[ts(export)]
  14. pub enum StreamEvent {
  15. Delta {
  16. text: String,
  17. },
  18. Reasoning {
  19. text: String,
  20. },
  21. Usage {
  22. usage: Usage,
  23. },
  24. /// The finished (or stopped) reply, as saved.
  25. Done {
  26. message: Message,
  27. },
  28. /// The reply failed; `message` is the saved row, with any partial text.
  29. Error {
  30. kind: String,
  31. error: String,
  32. message: Message,
  33. },
  34. }
  36. #[derive(Debug, Clone, Serialize, Deserialize, TS)]
  37. #[ts(export)]
  38. pub struct SendRequest {
  39. pub conversation_id: String,
  40. /// Empty when regenerating the last reply.
  41. pub text: String,
  42. pub attachment_ids: Vec<String>,
  43. }
  45. #[derive(Debug, Clone, Serialize, Deserialize, TS)]
  46. #[ts(export)]
  47. pub struct SendAck {
  48. pub stream_id: String,
  49. pub user_message: Option<Message>,
  50. pub assistant_message: Message,
  51. }
  53. #[derive(Debug, Clone, Serialize, Deserialize, TS)]
  54. #[ts(export)]
  55. pub struct ConversationView {
  56. pub conversation: Conversation,
  57. pub messages: Vec<Message>,
  58. }
  60. #[derive(Debug, Clone, Serialize, Deserialize, TS)]
  61. #[ts(export)]
  62. pub struct ModelsResponse {
  63. pub models: Vec<ModelInfo>,
  64. /// `live`, `snapshot` or `local`
  65. pub source: String,
  66. /// When the list was fetched (ms), or the snapshot's date.
  67. #[ts(type = "number | null")]
  68. pub fetched_at: Option<i64>,
  69. pub snapshot_date: Option<String>,
  70. /// Why a refresh failed, if it did (the list above is then the fallback).
  71. pub warning: Option<String>,
  72. }
  74. #[derive(Debug, Clone, Serialize, Deserialize, TS)]
  75. #[ts(export)]
  76. pub struct Estimate {
  77. #[ts(type = "number")]
  78. pub prompt_tokens: i64,
  79. /// The prompt's cost at the model's input price.
  80. #[ts(type = "number")]
  81. pub prompt_cost_nanousd: i64,
  82. #[ts(type = "number | null")]
  83. pub context_length: Option<i64>,
  84. }
  86. #[derive(Debug, Clone, Serialize, Deserialize, TS)]
  87. #[ts(export)]
  88. pub struct SpendSummary {
  89. #[ts(type = "number")]
  90. pub month_nanousd: i64,
  91. pub month_estimated: bool,
  92. pub budget_usd: Option<f64>,
  93. /// 0 below 80% of the budget, 1 from 80%, 2 at or over it.
  94. pub budget_level: u8,
  95. }
  97. #[derive(Debug, Clone, Serialize, Deserialize, TS)]
  98. #[ts(export)]
  99. pub struct UsageReport {
  100. pub by_model: Vec<UsageRow>,
  101. pub by_day: Vec<UsageRow>,
  102. #[ts(type = "number")]
  103. pub since_ms: i64,
  104. }
  106. #[derive(Debug, Clone, Serialize, Deserialize, TS)]
  107. #[ts(export)]
  108. pub struct AppInfo {
  109. pub version: String,
  110. /// `macos`, `linux` or `windows`
  111. pub platform: String,
  112. /// True when the window has native vibrancy behind it.
  113. pub vibrancy: bool,
  114. pub data_dir: String,
  115. }
  117. /// What happened when a key was saved. A key the provider refuses is not saved
  118. /// at all; that comes back as an error.
  119. #[derive(Debug, Clone, Serialize, Deserialize, TS)]
  120. #[ts(export)]
  121. pub struct KeySaved {
  122. /// False when the key was stored without being checked; `detail` says why.
  123. pub accepted: bool,
  124. /// Why the key could not be checked (the provider was unreachable, say).
  125. pub detail: Option<String>,
  126. }
  128. #[derive(Debug, Clone, PartialEq, Serialize, Deserialize, TS)]
  129. #[serde(default)]
  130. #[ts(export)]
  131. pub struct Settings {
  132. /// `system`, `light` or `dark`
  133. pub theme: String,
  134. pub reduce_transparency: bool,
  135. /// macOS window material: `sidebar`, `window`, `fullscreen` or `hud`.
  136. pub material: String,
  137. /// How strongly panels tint the material, 0 to 100 percent.
  138. pub glass_tint: u8,
  139. pub shortcut: String,
  140. pub default_provider_id: String,
  141. pub default_model_id: String,
  142. pub quick_provider_id: String,
  143. pub quick_model_id: String,
  144. pub monthly_budget_usd: Option<f64>,
  145. pub confirm_over_budget: bool,
  146. pub vault_dir: Option<String>,
  147. /// `wiki` for `![[x.png]]`, `markdown` for `![x](attachments/x.png)`
  148. pub image_links: String,
  149. pub favorite_models: Vec<String>,
  150. /// Local model ids the person has marked as able to see images.
  151. pub vision_models: Vec<String>,
  152. pub send_on_enter: bool,
  153. }
  155. impl Default for Settings {
  156. fn default() -> Self {
  157. Self {
  158. theme: "system".into(),
  159. reduce_transparency: false,
  160. material: "sidebar".into(),
  161. glass_tint: 100,
  162. shortcut: "CommandOrControl+Shift+Space".into(),
  163. default_provider_id: crate::storage::DEEPINFRA_ID.into(),
  164. default_model_id: "Qwen/Qwen3-235B-A22B-Instruct-2507".into(),
  165. quick_provider_id: crate::storage::DEEPINFRA_ID.into(),
  166. quick_model_id: "google/gemma-3-12b-it".into(),
  167. monthly_budget_usd: Some(10.0),
  168. confirm_over_budget: false,
  169. vault_dir: None,
  170. image_links: "wiki".into(),
  171. favorite_models: vec![],
  172. vision_models: vec![],
  173. send_on_enter: true,
  174. }
  175. }
  176. }
  178. #[derive(Debug, Clone, Serialize, Deserialize, TS)]
  179. #[ts(export)]
  180. pub struct SaveSettingsResult {
  181. pub settings: Settings,
  182. /// Set when the new shortcut could not be registered.
  183. pub shortcut_error: Option<String>,
  184. }
  186. #[derive(Debug, Clone, Serialize, Deserialize, TS)]
  187. #[ts(export)]
  188. pub struct ExportResult {
  189. pub path: String,
  190. #[ts(type = "number")]
  191. pub files_copied: i64,
  192. }
  194. pub fn budget_level(spent_nanousd: i64, budget_usd: Option<f64>) -> u8 {
  195. match budget_usd {
  196. Some(b) if b > 0.0 => {
  197. let spent = crate::cost::to_usd(spent_nanousd);
  198. if spent >= b {
  199. 2
  200. } else if spent >= b * 0.8 {
  201. 1
  202. } else {
  203. 0
  204. }
  205. }
  206. _ => 0,
  207. }
  208. }
  210. #[cfg(test)]
  211. mod tests {
  212. use super::*;
  214. #[test]
  215. fn budget_levels() {
  216. assert_eq!(budget_level(0, Some(10.0)), 0);
  217. assert_eq!(budget_level(7_999_999_999, Some(10.0)), 0);
  218. assert_eq!(budget_level(8_000_000_000, Some(10.0)), 1);
  219. assert_eq!(budget_level(10_000_000_000, Some(10.0)), 2);
  220. assert_eq!(budget_level(99_000_000_000, None), 0);
  221. assert_eq!(budget_level(1, Some(0.0)), 0);
  222. }
  224. #[test]
  225. fn settings_tolerate_missing_fields() {
  226. let s: Settings = serde_json::from_str(r#"{"theme":"dark"}"#).unwrap();
  227. assert_eq!(s.theme, "dark");
  228. assert_eq!(s.shortcut, Settings::default().shortcut);
  229. }
  231. #[test]
  232. fn stream_event_wire_shape() {
  233. let v = serde_json::to_value(StreamEvent::Delta { text: "hi".into() }).unwrap();
  234. assert_eq!(v, serde_json::json!({"type": "delta", "text": "hi"}));
  235. }
  236. }