Publish harness and TUI open-source
initial sync from the monorepo
This commit is contained in:
commit
c68e39f604
2734 changed files with 1437016 additions and 0 deletions
15
crates/codegen/xai-prompt-queue/Cargo.toml
Normal file
15
crates/codegen/xai-prompt-queue/Cargo.toml
Normal file
|
|
@ -0,0 +1,15 @@
|
|||
[package]
|
||||
edition.workspace = true
|
||||
license = "Apache-2.0"
|
||||
name = "xai-prompt-queue"
|
||||
version = "0.1.0"
|
||||
description = "Shared prompt-queue wire types for xai-grok-shell and xai-grok-pager"
|
||||
|
||||
[dependencies]
|
||||
serde = { workspace = true, features = ["derive"] }
|
||||
|
||||
[dev-dependencies]
|
||||
serde_json = { workspace = true }
|
||||
|
||||
[lints]
|
||||
workspace = true
|
||||
5
crates/codegen/xai-prompt-queue/src/lib.rs
Normal file
5
crates/codegen/xai-prompt-queue/src/lib.rs
Normal file
|
|
@ -0,0 +1,5 @@
|
|||
//! Shared prompt-queue wire types for xai-grok-shell and xai-grok-pager.
|
||||
|
||||
mod types;
|
||||
|
||||
pub use types::{QueueChanged, QueueEntryMeta, QueueEntryWire};
|
||||
171
crates/codegen/xai-prompt-queue/src/types.rs
Normal file
171
crates/codegen/xai-prompt-queue/src/types.rs
Normal file
|
|
@ -0,0 +1,171 @@
|
|||
use serde::{Deserialize, Serialize};
|
||||
|
||||
/// Per-item queue metadata the session actor attaches to user-originated inputs; synthetic
|
||||
/// inputs (auto-wake, nudges) carry none and never appear in the visible queue. Held in
|
||||
/// actor state, never serialized itself.
|
||||
#[derive(Clone, Debug, PartialEq, Eq)]
|
||||
pub struct QueueEntryMeta {
|
||||
/// Stable id, reusing the prompt's unique `prompt_id`.
|
||||
pub id: String,
|
||||
/// Monotonic, bumped on each in-place edit; an edit against a stale version is a no-op.
|
||||
pub version: u64,
|
||||
/// Enqueuing client identifier (attribution); never overwritten by edits.
|
||||
pub owner: Option<String>,
|
||||
/// Most recent editor's client identifier, replaced on every in-place edit.
|
||||
pub last_editor: Option<String>,
|
||||
/// Display kind label; client-cosmetic kinds resolve to their send-intent before enqueue.
|
||||
pub kind: String,
|
||||
/// Plain prompt text for the shared queue display.
|
||||
pub text: String,
|
||||
}
|
||||
|
||||
/// One queue row on the wire.
|
||||
#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
pub struct QueueEntryWire {
|
||||
pub id: String,
|
||||
#[serde(default)]
|
||||
pub version: u64,
|
||||
/// Omitted from the wire when `None`.
|
||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||
pub owner: Option<String>,
|
||||
/// Mirrors [`QueueEntryMeta::last_editor`]; omitted from the wire when `None`.
|
||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||
pub last_editor: Option<String>,
|
||||
#[serde(default)]
|
||||
pub kind: String,
|
||||
#[serde(default)]
|
||||
pub text: String,
|
||||
/// 0-based position among queued, not-yet-running prompts.
|
||||
#[serde(default)]
|
||||
pub position: usize,
|
||||
}
|
||||
|
||||
/// Broadcast payload for the `x.ai/queue/changed` notification.
|
||||
#[derive(Clone, Debug, Default, Serialize, Deserialize, PartialEq, Eq)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
pub struct QueueChanged {
|
||||
/// The session this queue belongs to; drives per-session fan-out routing.
|
||||
pub session_id: String,
|
||||
#[serde(default)]
|
||||
pub entries: Vec<QueueEntryWire>,
|
||||
/// The prompt the actor is currently draining, `None` when no turn runs. The correlation
|
||||
/// signal a subscriber uses to adopt `current_prompt_id` for notification routing.
|
||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||
pub running_prompt_id: Option<String>,
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn queue_changed_full_round_trip() {
|
||||
let original = QueueChanged {
|
||||
session_id: "sess-42".into(),
|
||||
entries: vec![
|
||||
QueueEntryWire {
|
||||
id: "p1".into(),
|
||||
version: 3,
|
||||
owner: Some("alice".into()),
|
||||
last_editor: Some("bob".into()),
|
||||
kind: "prompt".into(),
|
||||
text: "fix the bug".into(),
|
||||
position: 0,
|
||||
},
|
||||
QueueEntryWire {
|
||||
id: "p2".into(),
|
||||
version: 0,
|
||||
owner: None,
|
||||
last_editor: None,
|
||||
kind: "bash".into(),
|
||||
text: "ls -la".into(),
|
||||
position: 1,
|
||||
},
|
||||
],
|
||||
running_prompt_id: Some("p0".into()),
|
||||
};
|
||||
let json = serde_json::to_value(&original).unwrap();
|
||||
assert_eq!(json["sessionId"], "sess-42");
|
||||
assert_eq!(json["entries"][0]["lastEditor"], "bob");
|
||||
assert_eq!(json["runningPromptId"], "p0");
|
||||
assert!(json["entries"][1].get("owner").is_none());
|
||||
assert!(json["entries"][1].get("lastEditor").is_none());
|
||||
let round: QueueChanged = serde_json::from_value(json).unwrap();
|
||||
assert_eq!(round, original);
|
||||
}
|
||||
|
||||
/// Pins the exact wire JSON; a key rename here breaks deployed clients.
|
||||
#[test]
|
||||
fn queue_changed_golden_wire_json() {
|
||||
let payload = QueueChanged {
|
||||
session_id: "s1".into(),
|
||||
entries: vec![QueueEntryWire {
|
||||
id: "p1".into(),
|
||||
version: 2,
|
||||
owner: Some("alice".into()),
|
||||
last_editor: Some("bob".into()),
|
||||
kind: "prompt".into(),
|
||||
text: "hi".into(),
|
||||
position: 0,
|
||||
}],
|
||||
running_prompt_id: Some("p0".into()),
|
||||
};
|
||||
let expected = serde_json::json!({
|
||||
"sessionId": "s1",
|
||||
"entries": [{
|
||||
"id": "p1",
|
||||
"version": 2,
|
||||
"owner": "alice",
|
||||
"lastEditor": "bob",
|
||||
"kind": "prompt",
|
||||
"text": "hi",
|
||||
"position": 0
|
||||
}],
|
||||
"runningPromptId": "p0"
|
||||
});
|
||||
assert_eq!(serde_json::to_value(&payload).unwrap(), expected);
|
||||
}
|
||||
|
||||
/// A broadcast without sessionId must fail to parse, not apply under the wrong key.
|
||||
#[test]
|
||||
fn queue_changed_requires_session_id() {
|
||||
let missing = serde_json::json!({ "entries": [] });
|
||||
assert!(serde_json::from_value::<QueueChanged>(missing).is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn sparse_payload_deserializes_with_defaults() {
|
||||
let sparse = serde_json::json!({
|
||||
"sessionId": "s1",
|
||||
"entries": [{"id": "p1"}]
|
||||
});
|
||||
let parsed: QueueChanged = serde_json::from_value(sparse).unwrap();
|
||||
assert_eq!(parsed.entries[0].version, 0);
|
||||
assert_eq!(parsed.entries[0].kind, "");
|
||||
assert_eq!(parsed.entries[0].text, "");
|
||||
assert_eq!(parsed.entries[0].position, 0);
|
||||
assert!(parsed.entries[0].owner.is_none());
|
||||
assert!(parsed.running_prompt_id.is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn extra_unknown_fields_ignored() {
|
||||
let json = serde_json::json!({
|
||||
"sessionId": "s1",
|
||||
"entries": [],
|
||||
"runningPromptId": null,
|
||||
"futureField": "should be ignored"
|
||||
});
|
||||
let parsed: QueueChanged = serde_json::from_value(json).unwrap();
|
||||
assert_eq!(parsed.session_id, "s1");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn queue_changed_derives_default() {
|
||||
let d = QueueChanged::default();
|
||||
assert_eq!(d.session_id, "");
|
||||
assert!(d.entries.is_empty());
|
||||
assert!(d.running_prompt_id.is_none());
|
||||
}
|
||||
}
|
||||
Loading…
Reference in a new issue