chore(assets): scripted screenshot pipeline
This commit is contained in:
@@ -1,3 +1,6 @@
|
|||||||
|
[alias]
|
||||||
|
xtask = "run -p xtask --"
|
||||||
|
|
||||||
[target.x86_64-unknown-linux-musl]
|
[target.x86_64-unknown-linux-musl]
|
||||||
linker = "x86_64-linux-gnu-gcc"
|
linker = "x86_64-linux-gnu-gcc"
|
||||||
rustflags = ["-C", "target-feature=+crt-static", "-C", "link-arg=-lgcc"]
|
rustflags = ["-C", "target-feature=+crt-static", "-C", "link-arg=-lgcc"]
|
||||||
|
|||||||
1
.gitignore
vendored
1
.gitignore
vendored
@@ -3,6 +3,7 @@
|
|||||||
# will have compiled files and executables
|
# will have compiled files and executables
|
||||||
debug/
|
debug/
|
||||||
target/
|
target/
|
||||||
|
images/generated/
|
||||||
dev/
|
dev/
|
||||||
.agents/
|
.agents/
|
||||||
.env
|
.env
|
||||||
|
|||||||
@@ -160,6 +160,12 @@ For more detailed information, please refer to the following documents:
|
|||||||
- **Experimental providers staging area**: [crates/providers/experimental/README.md](crates/providers/experimental/README.md) records the placeholder crates (OpenAI, Anthropic, Gemini) and their current status.
|
- **Experimental providers staging area**: [crates/providers/experimental/README.md](crates/providers/experimental/README.md) records the placeholder crates (OpenAI, Anthropic, Gemini) and their current status.
|
||||||
- **[docs/platform-support.md](docs/platform-support.md)**: Current OS support matrix and cross-check instructions.
|
- **[docs/platform-support.md](docs/platform-support.md)**: Current OS support matrix and cross-check instructions.
|
||||||
|
|
||||||
|
## Developer Tasks
|
||||||
|
|
||||||
|
- `cargo xtask screenshots` regenerates deterministic ANSI dumps (and, when
|
||||||
|
`chafa` is available, PNG renders) for the documentation gallery. Use
|
||||||
|
`--no-png` to skip the PNG step or `--output <dir>` to redirect the output.
|
||||||
|
|
||||||
## Configuration
|
## Configuration
|
||||||
|
|
||||||
OWLEN stores its configuration in the standard platform-specific config directory:
|
OWLEN stores its configuration in the standard platform-specific config directory:
|
||||||
|
|||||||
@@ -176,3 +176,16 @@ When modifying TUI UX or keybindings:
|
|||||||
|
|
||||||
Keeping these steps in sync ensures Owlen’s keyboard-first UX remains
|
Keeping these steps in sync ensures Owlen’s keyboard-first UX remains
|
||||||
predictable, accessible, and discoverable.
|
predictable, accessible, and discoverable.
|
||||||
|
|
||||||
|
## 8. Screenshot Pipeline
|
||||||
|
|
||||||
|
Use the scripted pipeline to regenerate gallery assets and documentation
|
||||||
|
illustrations:
|
||||||
|
|
||||||
|
- `cargo xtask screenshots` emits ANSI dumps under `images/generated` and, by
|
||||||
|
default, converts them to PNG with `chafa`. Pass `--no-png` to skip PNG
|
||||||
|
rendering or `--chafa /path/to/chafa` to override the binary. Each scene is
|
||||||
|
deterministic—no network calls—and mirrors the regression snapshots.
|
||||||
|
- The pipeline reuses the same stub provider harness as the snapshot tests, so
|
||||||
|
new scenes should be added in tandem with `chat_snapshots.rs` to keep visual
|
||||||
|
regression coverage and documentation imagery aligned.
|
||||||
|
|||||||
@@ -7,3 +7,12 @@ publish = false
|
|||||||
[dependencies]
|
[dependencies]
|
||||||
anyhow = { workspace = true }
|
anyhow = { workspace = true }
|
||||||
clap = { workspace = true, features = ["derive"] }
|
clap = { workspace = true, features = ["derive"] }
|
||||||
|
crossterm = { workspace = true }
|
||||||
|
async-trait = { workspace = true }
|
||||||
|
futures-util = { workspace = true }
|
||||||
|
owlen-core = { path = "../crates/owlen-core" }
|
||||||
|
owlen-tui = { path = "../crates/owlen-tui" }
|
||||||
|
ratatui = { workspace = true }
|
||||||
|
serde_json = { workspace = true }
|
||||||
|
tempfile = { workspace = true }
|
||||||
|
tokio = { workspace = true }
|
||||||
|
|||||||
@@ -4,6 +4,8 @@ use std::process::Command;
|
|||||||
use anyhow::{Context, Result, bail};
|
use anyhow::{Context, Result, bail};
|
||||||
use clap::{Parser, Subcommand};
|
use clap::{Parser, Subcommand};
|
||||||
|
|
||||||
|
mod screenshots;
|
||||||
|
|
||||||
#[derive(Parser)]
|
#[derive(Parser)]
|
||||||
#[command(author, version, about = "Owlen developer tasks", long_about = None)]
|
#[command(author, version, about = "Owlen developer tasks", long_about = None)]
|
||||||
struct Xtask {
|
struct Xtask {
|
||||||
@@ -36,6 +38,23 @@ enum Task {
|
|||||||
#[arg(long, value_name = "PATH", help = "Override the repo map output path")]
|
#[arg(long, value_name = "PATH", help = "Override the repo map output path")]
|
||||||
output: Option<PathBuf>,
|
output: Option<PathBuf>,
|
||||||
},
|
},
|
||||||
|
/// Generate deterministic TUI screenshots and optional PNG renders.
|
||||||
|
Screenshots {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
value_name = "PATH",
|
||||||
|
help = "Output directory (defaults to images/generated)"
|
||||||
|
)]
|
||||||
|
output: Option<PathBuf>,
|
||||||
|
#[arg(long, help = "Skip PNG conversion and only emit ANSI dumps")]
|
||||||
|
no_png: bool,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
value_name = "PATH",
|
||||||
|
help = "Path to chafa binary (default: chafa in PATH)"
|
||||||
|
)]
|
||||||
|
chafa: Option<PathBuf>,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
fn main() -> Result<()> {
|
fn main() -> Result<()> {
|
||||||
@@ -49,6 +68,11 @@ fn main() -> Result<()> {
|
|||||||
Task::DevRun { args } => dev_run(args),
|
Task::DevRun { args } => dev_run(args),
|
||||||
Task::ReleaseCheck => release_check(),
|
Task::ReleaseCheck => release_check(),
|
||||||
Task::GenRepoMap { output } => gen_repo_map(output),
|
Task::GenRepoMap { output } => gen_repo_map(output),
|
||||||
|
Task::Screenshots {
|
||||||
|
output,
|
||||||
|
no_png,
|
||||||
|
chafa,
|
||||||
|
} => screenshots::run(output, chafa, no_png),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -154,7 +178,7 @@ fn run_cargo(args: Vec<String>) -> Result<()> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn workspace_root() -> PathBuf {
|
pub(crate) fn workspace_root() -> PathBuf {
|
||||||
Path::new(env!("CARGO_MANIFEST_DIR"))
|
Path::new(env!("CARGO_MANIFEST_DIR"))
|
||||||
.parent()
|
.parent()
|
||||||
.expect("xtask has a parent directory")
|
.expect("xtask has a parent directory")
|
||||||
|
|||||||
444
xtask/src/screenshots.rs
Normal file
444
xtask/src/screenshots.rs
Normal file
@@ -0,0 +1,444 @@
|
|||||||
|
use std::env;
|
||||||
|
use std::fs;
|
||||||
|
use std::path::{Path, PathBuf};
|
||||||
|
use std::process::Command;
|
||||||
|
|
||||||
|
use anyhow::{Context, Result, bail};
|
||||||
|
use async_trait::async_trait;
|
||||||
|
use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
|
||||||
|
use futures_util::FutureExt;
|
||||||
|
use futures_util::future::BoxFuture;
|
||||||
|
use owlen_core::{
|
||||||
|
Config, McpMode, Mode, Provider,
|
||||||
|
session::SessionController,
|
||||||
|
storage::StorageManager,
|
||||||
|
types::{Message, ModelInfo, ToolCall},
|
||||||
|
ui::{NoOpUiController, UiController},
|
||||||
|
};
|
||||||
|
use owlen_tui::ChatApp;
|
||||||
|
use owlen_tui::events::Event;
|
||||||
|
use owlen_tui::ui::render_chat;
|
||||||
|
use ratatui::{Terminal, backend::TestBackend};
|
||||||
|
use serde_json::json;
|
||||||
|
use tokio::runtime::Runtime;
|
||||||
|
use tokio::sync::mpsc;
|
||||||
|
|
||||||
|
use crate::workspace_root;
|
||||||
|
|
||||||
|
#[derive(Clone, Copy)]
|
||||||
|
struct SceneSpec {
|
||||||
|
name: &'static str,
|
||||||
|
width: u16,
|
||||||
|
height: u16,
|
||||||
|
builder: fn() -> BoxFuture<'static, Result<ChatApp>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn run(output: Option<PathBuf>, chafa: Option<PathBuf>, no_png: bool) -> Result<()> {
|
||||||
|
let output_dir = output.unwrap_or_else(|| workspace_root().join("images/generated"));
|
||||||
|
fs::create_dir_all(&output_dir)
|
||||||
|
.with_context(|| format!("failed to create output directory {}", output_dir.display()))?;
|
||||||
|
|
||||||
|
let config_home = workspace_root().join("dev/xtask-config");
|
||||||
|
fs::create_dir_all(&config_home).with_context(|| {
|
||||||
|
format!(
|
||||||
|
"failed to create config home directory {}",
|
||||||
|
config_home.display()
|
||||||
|
)
|
||||||
|
})?;
|
||||||
|
unsafe {
|
||||||
|
// Safe: we control the value and keep it within the workspace for deterministic snapshots.
|
||||||
|
env::set_var("XDG_CONFIG_HOME", &config_home);
|
||||||
|
env::set_var("XDG_DATA_HOME", &config_home);
|
||||||
|
}
|
||||||
|
|
||||||
|
let runtime = Runtime::new().context("failed to create tokio runtime")?;
|
||||||
|
|
||||||
|
for scene in scenes() {
|
||||||
|
let mut app = runtime.block_on((scene.builder)())?;
|
||||||
|
let ansi = render_snapshot(&mut app, scene.width, scene.height)?;
|
||||||
|
let ans_path = output_dir.join(format!("{}.ans", scene.name));
|
||||||
|
fs::write(&ans_path, ansi.as_bytes())
|
||||||
|
.with_context(|| format!("failed to write ANSI dump {}", ans_path.display()))?;
|
||||||
|
|
||||||
|
if !no_png
|
||||||
|
&& let Err(err) = convert_to_png(
|
||||||
|
&ans_path,
|
||||||
|
scene.width,
|
||||||
|
scene.height,
|
||||||
|
&output_dir,
|
||||||
|
scene.name,
|
||||||
|
chafa.as_deref(),
|
||||||
|
)
|
||||||
|
{
|
||||||
|
eprintln!("warning: {}", err);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
println!("Screenshots written to {}", output_dir.display());
|
||||||
|
if no_png {
|
||||||
|
println!("PNG conversion skipped (use --no-png=false or omit flag to enable)");
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn convert_to_png(
|
||||||
|
ans_path: &Path,
|
||||||
|
width: u16,
|
||||||
|
height: u16,
|
||||||
|
output_dir: &Path,
|
||||||
|
name: &str,
|
||||||
|
chafa_path: Option<&Path>,
|
||||||
|
) -> Result<()> {
|
||||||
|
let png_path = output_dir.join(format!("{}.png", name));
|
||||||
|
let mut command = if let Some(path) = chafa_path {
|
||||||
|
Command::new(path)
|
||||||
|
} else {
|
||||||
|
Command::new("chafa")
|
||||||
|
};
|
||||||
|
let status = command
|
||||||
|
.arg("--size")
|
||||||
|
.arg(format!("{}x{}", width, height))
|
||||||
|
.arg("--save")
|
||||||
|
.arg(&png_path)
|
||||||
|
.arg(ans_path)
|
||||||
|
.status()
|
||||||
|
.with_context(|| "failed to spawn chafa".to_string())?;
|
||||||
|
if !status.success() {
|
||||||
|
bail!(
|
||||||
|
"chafa exited with status {} when rendering {}",
|
||||||
|
status.code().unwrap_or_default(),
|
||||||
|
name
|
||||||
|
);
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn scenes() -> &'static [SceneSpec] {
|
||||||
|
&[
|
||||||
|
SceneSpec {
|
||||||
|
name: "chat-idle-dark",
|
||||||
|
width: 120,
|
||||||
|
height: 32,
|
||||||
|
builder: || async move { Ok(build_chat_app(|_| {}, |_| {}).await) }.boxed(),
|
||||||
|
},
|
||||||
|
SceneSpec {
|
||||||
|
name: "chat-tool-call",
|
||||||
|
width: 120,
|
||||||
|
height: 32,
|
||||||
|
builder: || {
|
||||||
|
async move {
|
||||||
|
let app = build_chat_app(
|
||||||
|
|_| {},
|
||||||
|
|session| {
|
||||||
|
let conversation = session.conversation_mut();
|
||||||
|
conversation.push_user_message("What happened in the Rust ecosystem today?");
|
||||||
|
let stream_id = conversation.start_streaming_response();
|
||||||
|
conversation
|
||||||
|
.set_stream_placeholder(stream_id, "Consulting the knowledge base…")
|
||||||
|
.expect("placeholder");
|
||||||
|
let tool_call = ToolCall {
|
||||||
|
id: "call-search-1".into(),
|
||||||
|
name: "web_search".into(),
|
||||||
|
arguments: json!({ "query": "Rust language news" }),
|
||||||
|
};
|
||||||
|
conversation
|
||||||
|
.set_tool_calls_on_message(stream_id, vec![tool_call.clone()])
|
||||||
|
.expect("tool call metadata");
|
||||||
|
conversation
|
||||||
|
.append_stream_chunk(stream_id, "Found multiple articles…", false)
|
||||||
|
.expect("stream chunk");
|
||||||
|
conversation.push_message(Message::tool(
|
||||||
|
tool_call.id.clone(),
|
||||||
|
"Rust 1.85 released with generics cleanups and faster async compilation.".into(),
|
||||||
|
));
|
||||||
|
conversation.push_message(Message::assistant(
|
||||||
|
"Summarising the latest Rust release and the async runtime updates.".into(),
|
||||||
|
));
|
||||||
|
},
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
Ok(app)
|
||||||
|
}
|
||||||
|
.boxed()
|
||||||
|
},
|
||||||
|
},
|
||||||
|
SceneSpec {
|
||||||
|
name: "command-palette",
|
||||||
|
width: 110,
|
||||||
|
height: 28,
|
||||||
|
builder: || {
|
||||||
|
async move {
|
||||||
|
let mut app = build_chat_app(|_| {}, |_| {}).await;
|
||||||
|
send_key(&mut app, KeyCode::Char(':'), KeyModifiers::NONE).await;
|
||||||
|
type_text(&mut app, "focus").await;
|
||||||
|
send_key(&mut app, KeyCode::Down, KeyModifiers::NONE).await;
|
||||||
|
Ok(app)
|
||||||
|
}
|
||||||
|
.boxed()
|
||||||
|
},
|
||||||
|
},
|
||||||
|
SceneSpec {
|
||||||
|
name: "guidance-onboarding",
|
||||||
|
width: 100,
|
||||||
|
height: 28,
|
||||||
|
builder: || {
|
||||||
|
async move {
|
||||||
|
let app = build_chat_app(
|
||||||
|
|cfg| {
|
||||||
|
cfg.ui.show_onboarding = true;
|
||||||
|
cfg.ui.guidance.coach_marks_complete = false;
|
||||||
|
},
|
||||||
|
|_| {},
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
Ok(app)
|
||||||
|
}
|
||||||
|
.boxed()
|
||||||
|
},
|
||||||
|
},
|
||||||
|
SceneSpec {
|
||||||
|
name: "guidance-cheatsheet",
|
||||||
|
width: 120,
|
||||||
|
height: 30,
|
||||||
|
builder: || {
|
||||||
|
async move {
|
||||||
|
let mut app =
|
||||||
|
build_chat_app(|cfg| cfg.ui.guidance.coach_marks_complete = true, |_| {})
|
||||||
|
.await;
|
||||||
|
send_key(&mut app, KeyCode::Char('?'), KeyModifiers::NONE).await;
|
||||||
|
Ok(app)
|
||||||
|
}
|
||||||
|
.boxed()
|
||||||
|
},
|
||||||
|
},
|
||||||
|
SceneSpec {
|
||||||
|
name: "editing-mode",
|
||||||
|
width: 120,
|
||||||
|
height: 32,
|
||||||
|
builder: || {
|
||||||
|
async move {
|
||||||
|
let mut app = build_chat_app(|_| {}, |_| {}).await;
|
||||||
|
send_key(&mut app, KeyCode::Char('i'), KeyModifiers::NONE).await;
|
||||||
|
type_text(&mut app, "Editing mode demonstration").await;
|
||||||
|
Ok(app)
|
||||||
|
}
|
||||||
|
.boxed()
|
||||||
|
},
|
||||||
|
},
|
||||||
|
SceneSpec {
|
||||||
|
name: "visual-mode",
|
||||||
|
width: 120,
|
||||||
|
height: 32,
|
||||||
|
builder: || {
|
||||||
|
async move {
|
||||||
|
let mut app = build_chat_app(
|
||||||
|
|_| {},
|
||||||
|
|session| {
|
||||||
|
let conversation = session.conversation_mut();
|
||||||
|
conversation.push_user_message(
|
||||||
|
"Render visual selection across multiple lines.",
|
||||||
|
);
|
||||||
|
conversation.push_message(Message::assistant(
|
||||||
|
"Assistant reply for visual mode highlighting.".into(),
|
||||||
|
));
|
||||||
|
},
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
send_key(&mut app, KeyCode::Char('v'), KeyModifiers::NONE).await;
|
||||||
|
send_key(&mut app, KeyCode::Char('j'), KeyModifiers::NONE).await;
|
||||||
|
send_key(&mut app, KeyCode::Char('j'), KeyModifiers::NONE).await;
|
||||||
|
Ok(app)
|
||||||
|
}
|
||||||
|
.boxed()
|
||||||
|
},
|
||||||
|
},
|
||||||
|
SceneSpec {
|
||||||
|
name: "accessibility-high-contrast",
|
||||||
|
width: 120,
|
||||||
|
height: 32,
|
||||||
|
builder: || {
|
||||||
|
async move {
|
||||||
|
let app = build_chat_app(
|
||||||
|
|cfg| {
|
||||||
|
cfg.ui.accessibility.high_contrast = true;
|
||||||
|
cfg.ui.guidance.coach_marks_complete = true;
|
||||||
|
},
|
||||||
|
|_| {},
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
Ok(app)
|
||||||
|
}
|
||||||
|
.boxed()
|
||||||
|
},
|
||||||
|
},
|
||||||
|
SceneSpec {
|
||||||
|
name: "accessibility-reduced-chrome",
|
||||||
|
width: 120,
|
||||||
|
height: 32,
|
||||||
|
builder: || {
|
||||||
|
async move {
|
||||||
|
let app = build_chat_app(
|
||||||
|
|cfg| {
|
||||||
|
cfg.ui.accessibility.reduced_chrome = true;
|
||||||
|
cfg.ui.guidance.coach_marks_complete = true;
|
||||||
|
},
|
||||||
|
|_| {},
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
Ok(app)
|
||||||
|
}
|
||||||
|
.boxed()
|
||||||
|
},
|
||||||
|
},
|
||||||
|
SceneSpec {
|
||||||
|
name: "emacs-profile",
|
||||||
|
width: 120,
|
||||||
|
height: 32,
|
||||||
|
builder: || {
|
||||||
|
async move {
|
||||||
|
let mut app = build_chat_app(
|
||||||
|
|cfg| {
|
||||||
|
cfg.ui.keymap_profile = Some("emacs".into());
|
||||||
|
cfg.ui.guidance.coach_marks_complete = true;
|
||||||
|
},
|
||||||
|
|_| {},
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
send_key(&mut app, KeyCode::Char(':'), KeyModifiers::NONE).await;
|
||||||
|
type_text(&mut app, "help").await;
|
||||||
|
Ok(app)
|
||||||
|
}
|
||||||
|
.boxed()
|
||||||
|
},
|
||||||
|
},
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
|
fn render_snapshot(app: &mut ChatApp, width: u16, height: u16) -> Result<String> {
|
||||||
|
let backend = TestBackend::new(width, height);
|
||||||
|
let mut terminal = Terminal::new(backend).context("failed to create test terminal")?;
|
||||||
|
terminal
|
||||||
|
.draw(|frame| render_chat(frame, app))
|
||||||
|
.context("failed to render chat frame")?;
|
||||||
|
let buffer = terminal.backend().buffer();
|
||||||
|
Ok(buffer_to_string(buffer))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn buffer_to_string(buffer: &ratatui::buffer::Buffer) -> String {
|
||||||
|
let mut output = String::new();
|
||||||
|
for y in 0..buffer.area.height {
|
||||||
|
for x in 0..buffer.area.width {
|
||||||
|
output.push_str(buffer[(x, y)].symbol());
|
||||||
|
}
|
||||||
|
output.push('\n');
|
||||||
|
}
|
||||||
|
output
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn send_key(app: &mut ChatApp, code: KeyCode, modifiers: KeyModifiers) {
|
||||||
|
app.handle_event(Event::Key(KeyEvent::new(code, modifiers)))
|
||||||
|
.await
|
||||||
|
.expect("send key event");
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn type_text(app: &mut ChatApp, text: &str) {
|
||||||
|
for ch in text.chars() {
|
||||||
|
send_key(app, KeyCode::Char(ch), KeyModifiers::NONE).await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn build_chat_app<C, F>(configure_config: C, configure_session: F) -> ChatApp
|
||||||
|
where
|
||||||
|
C: FnOnce(&mut Config) + Send + 'static,
|
||||||
|
F: FnOnce(&mut SessionController) + Send + 'static,
|
||||||
|
{
|
||||||
|
let temp_dir = tempfile::tempdir().expect("temp dir");
|
||||||
|
let storage =
|
||||||
|
StorageManager::with_database_path(temp_dir.path().join("owlen-tui-screenshots.db"))
|
||||||
|
.await
|
||||||
|
.expect("storage");
|
||||||
|
let storage = std::sync::Arc::new(storage);
|
||||||
|
|
||||||
|
let mut config = Config::default();
|
||||||
|
configure_config(&mut config);
|
||||||
|
config.general.default_model = Some("stub-model".into());
|
||||||
|
config.general.enable_streaming = true;
|
||||||
|
config.privacy.encrypt_local_data = false;
|
||||||
|
config.privacy.require_consent_per_session = false;
|
||||||
|
config.ui.show_timestamps = false;
|
||||||
|
config.mcp.mode = McpMode::LocalOnly;
|
||||||
|
config.mcp.allow_fallback = false;
|
||||||
|
config.mcp.warn_on_legacy = false;
|
||||||
|
|
||||||
|
let provider: std::sync::Arc<dyn Provider> = std::sync::Arc::new(StubProvider);
|
||||||
|
let ui: std::sync::Arc<dyn UiController> = std::sync::Arc::new(NoOpUiController);
|
||||||
|
let (event_tx, controller_event_rx) = mpsc::unbounded_channel();
|
||||||
|
|
||||||
|
let mut session = SessionController::new(provider, config, storage, ui, true, Some(event_tx))
|
||||||
|
.await
|
||||||
|
.expect("session controller");
|
||||||
|
|
||||||
|
session
|
||||||
|
.set_operating_mode(Mode::Chat)
|
||||||
|
.await
|
||||||
|
.expect("chat mode");
|
||||||
|
|
||||||
|
configure_session(&mut session);
|
||||||
|
|
||||||
|
let (app, mut session_rx) = ChatApp::new(session, controller_event_rx)
|
||||||
|
.await
|
||||||
|
.expect("chat app");
|
||||||
|
session_rx.close();
|
||||||
|
|
||||||
|
app
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Default)]
|
||||||
|
struct StubProvider;
|
||||||
|
|
||||||
|
#[async_trait]
|
||||||
|
impl Provider for StubProvider {
|
||||||
|
fn name(&self) -> &str {
|
||||||
|
"stub-provider"
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn list_models(&self) -> owlen_core::Result<Vec<owlen_core::types::ModelInfo>> {
|
||||||
|
Ok(vec![ModelInfo {
|
||||||
|
id: "stub-model".into(),
|
||||||
|
name: "Stub Model".into(),
|
||||||
|
description: Some("Stub model for screenshot generation".into()),
|
||||||
|
provider: self.name().into(),
|
||||||
|
context_window: Some(8_192),
|
||||||
|
capabilities: vec!["chat".into(), "tool-use".into()],
|
||||||
|
supports_tools: true,
|
||||||
|
}])
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn send_prompt(
|
||||||
|
&self,
|
||||||
|
_request: owlen_core::types::ChatRequest,
|
||||||
|
) -> owlen_core::Result<owlen_core::types::ChatResponse> {
|
||||||
|
Ok(owlen_core::types::ChatResponse {
|
||||||
|
message: Message::assistant("stub completion".into()),
|
||||||
|
usage: None,
|
||||||
|
is_streaming: false,
|
||||||
|
is_final: true,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn stream_prompt(
|
||||||
|
&self,
|
||||||
|
_request: owlen_core::types::ChatRequest,
|
||||||
|
) -> owlen_core::Result<owlen_core::ChatStream> {
|
||||||
|
Ok(Box::pin(futures_util::stream::empty()))
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn health_check(&self) -> owlen_core::Result<()> {
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn as_any(&self) -> &(dyn std::any::Any + Send + Sync) {
|
||||||
|
self
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user