test: added integration tests for inter-feature interactions like RAG + Agents, function calling/MCP servers, etc.
This commit is contained in:
@@ -1,20 +1,3 @@
|
||||
//! Immutable, process-wide application configuration.
|
||||
//!
|
||||
//! `AppConfig` contains the settings loaded from `config.yaml` that are
|
||||
//! global to the Loki process: LLM provider configs, UI preferences,
|
||||
//! tool and MCP settings, RAG defaults, etc.
|
||||
//!
|
||||
//! `AppConfig` mirrors the field shape of [`Config`](super::Config) (the
|
||||
//! serde POJO loaded from YAML) but is the runtime-resolved form: env
|
||||
//! var overrides applied, wrap validated, default document loaders
|
||||
//! installed, user agent resolved, default model picked. Build it via
|
||||
//! [`AppConfig::from_config`].
|
||||
//!
|
||||
//! Runtime-only state (current role, session, agent, supervisor, etc.)
|
||||
//! lives on [`RequestContext`](super::request_context::RequestContext).
|
||||
//! Process-wide services (vault, MCP registry, function registry) live
|
||||
//! on [`AppState`](super::app_state::AppState).
|
||||
|
||||
use crate::client::{ClientConfig, list_models};
|
||||
use crate::render::{MarkdownRender, RenderOptions};
|
||||
use crate::utils::{IS_STDOUT_TERMINAL, NO_COLOR, decode_bin, get_env_name};
|
||||
|
||||
@@ -1,19 +1,3 @@
|
||||
//! Shared global services for a running Loki process.
|
||||
//!
|
||||
//! `AppState` holds the services that are genuinely process-wide and
|
||||
//! immutable during request handling: the frozen [`AppConfig`], the
|
||||
//! credential [`Vault`](GlobalVault), the [`McpFactory`](super::mcp_factory::McpFactory)
|
||||
//! for MCP subprocess sharing, the [`RagCache`](super::rag_cache::RagCache)
|
||||
//! for shared RAG instances, the global MCP registry, and the base
|
||||
//! [`Functions`] declarations seeded into per-request `ToolScope`s. It
|
||||
//! is wrapped in `Arc` and shared across every [`RequestContext`] that
|
||||
//! a frontend (CLI, REPL, API) creates.
|
||||
//!
|
||||
//! Built via [`AppState::init`] from an `Arc<AppConfig>` plus
|
||||
//! startup context (log path, MCP-start flag, abort signal). The
|
||||
//! `init` call is the single place that wires the vault, MCP
|
||||
//! registry, and global functions together.
|
||||
|
||||
use super::mcp_factory::{McpFactory, McpServerKey};
|
||||
use super::rag_cache::RagCache;
|
||||
use crate::config::AppConfig;
|
||||
|
||||
+103
-4
@@ -1,13 +1,13 @@
|
||||
use super::*;
|
||||
|
||||
use crate::client::{
|
||||
init_client, patch_messages, ChatCompletionsData, Client, ImageUrl, Message,
|
||||
MessageContent, MessageContentPart, MessageContentToolCalls, MessageRole, Model,
|
||||
ChatCompletionsData, Client, ImageUrl, Message, MessageContent, MessageContentPart,
|
||||
MessageContentToolCalls, MessageRole, Model, init_client, patch_messages,
|
||||
};
|
||||
use crate::function::ToolResult;
|
||||
use crate::utils::{base64_encode, is_loader_protocol, sha256, AbortSignal};
|
||||
use crate::utils::{AbortSignal, base64_encode, is_loader_protocol, sha256};
|
||||
|
||||
use anyhow::{bail, Context, Result};
|
||||
use anyhow::{Context, Result, bail};
|
||||
use indexmap::IndexSet;
|
||||
use std::{collections::HashMap, fs::File, io::Read, sync::Arc};
|
||||
use unicode_width::{UnicodeWidthChar, UnicodeWidthStr};
|
||||
@@ -584,7 +584,9 @@ mod tests {
|
||||
use super::*;
|
||||
use crate::config::request_context::RequestContext;
|
||||
use crate::config::{AppState, WorkingMode};
|
||||
use std::fs;
|
||||
use std::sync::Arc;
|
||||
use std::time::SystemTime;
|
||||
|
||||
fn default_app_state() -> Arc<AppState> {
|
||||
Arc::new(AppState::test_default())
|
||||
@@ -865,4 +867,101 @@ mod tests {
|
||||
let result = resolve_data_url(&data_urls, "https://example.com/image.png".to_string());
|
||||
assert_eq!(result, "https://example.com/image.png");
|
||||
}
|
||||
|
||||
fn run_async<F: Future>(f: F) -> F::Output {
|
||||
tokio::runtime::Builder::new_current_thread()
|
||||
.enable_all()
|
||||
.build()
|
||||
.unwrap()
|
||||
.block_on(f)
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn from_files_loads_single_text_file() {
|
||||
let dir = env::temp_dir().join(format!(
|
||||
"loki-input-test-{}",
|
||||
SystemTime::now()
|
||||
.duration_since(std::time::UNIX_EPOCH)
|
||||
.unwrap()
|
||||
.as_nanos()
|
||||
));
|
||||
create_dir_all(&dir).unwrap();
|
||||
let file_path = dir.join("test.txt");
|
||||
fs::write(&file_path, "file content here").unwrap();
|
||||
|
||||
let ctx = create_test_ctx();
|
||||
let input = run_async(Input::from_files(
|
||||
&ctx,
|
||||
"question",
|
||||
vec![file_path.to_string_lossy().to_string()],
|
||||
None,
|
||||
))
|
||||
.unwrap();
|
||||
|
||||
assert!(input.text().contains("file content here"));
|
||||
assert!(input.text().contains("question"));
|
||||
let _ = fs::remove_dir_all(&dir);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn from_files_loads_multiple_files() {
|
||||
let dir = env::temp_dir().join(format!(
|
||||
"loki-input-test-multi-{}",
|
||||
SystemTime::now()
|
||||
.duration_since(std::time::UNIX_EPOCH)
|
||||
.unwrap()
|
||||
.as_nanos()
|
||||
));
|
||||
create_dir_all(&dir).unwrap();
|
||||
fs::write(dir.join("a.txt"), "content A").unwrap();
|
||||
fs::write(dir.join("b.txt"), "content B").unwrap();
|
||||
|
||||
let ctx = create_test_ctx();
|
||||
let input = run_async(Input::from_files(
|
||||
&ctx,
|
||||
"question",
|
||||
vec![
|
||||
dir.join("a.txt").to_string_lossy().to_string(),
|
||||
dir.join("b.txt").to_string_lossy().to_string(),
|
||||
],
|
||||
None,
|
||||
))
|
||||
.unwrap();
|
||||
|
||||
assert!(input.text().contains("content A"));
|
||||
assert!(input.text().contains("content B"));
|
||||
let _ = fs::remove_dir_all(&dir);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn from_files_with_no_paths_just_text() {
|
||||
let ctx = create_test_ctx();
|
||||
let input = run_async(Input::from_files(&ctx, "just text", vec![], None)).unwrap();
|
||||
assert_eq!(input.text(), "just text");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn from_files_with_external_command() {
|
||||
let ctx = create_test_ctx();
|
||||
let input = run_async(Input::from_files(
|
||||
&ctx,
|
||||
"question",
|
||||
vec!["`echo hello from cmd`".to_string()],
|
||||
None,
|
||||
))
|
||||
.unwrap();
|
||||
assert!(input.text().contains("hello from cmd"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn from_files_nonexistent_file_errors() {
|
||||
let ctx = create_test_ctx();
|
||||
let result = run_async(Input::from_files(
|
||||
&ctx,
|
||||
"question",
|
||||
vec!["/nonexistent/path/xyz.txt".to_string()],
|
||||
None,
|
||||
));
|
||||
assert!(result.is_err());
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,36 +1,3 @@
|
||||
//! Per-process factory for MCP subprocess handles.
|
||||
//!
|
||||
//! `McpFactory` lives on [`AppState`](super::AppState) and is the
|
||||
//! single entrypoint that scopes use to obtain `Arc<ConnectedServer>`
|
||||
//! handles for MCP tool servers. Multiple scopes requesting the same
|
||||
//! server can (eventually) share a single subprocess via `Arc`
|
||||
//! reference counting.
|
||||
//!
|
||||
//! # Phase 1 Step 6.5 scope
|
||||
//!
|
||||
//! This file introduces the factory scaffolding with a trivial
|
||||
//! implementation:
|
||||
//!
|
||||
//! * `active` — `Mutex<HashMap<McpServerKey, Weak<ConnectedServer>>>`
|
||||
//! for future Arc-based sharing across scopes
|
||||
//! * `acquire` — unimplemented stub for now; will be filled in when
|
||||
//! Step 8 rewrites `use_role` / `use_session` / `use_agent` to
|
||||
//! actually build `ToolScope`s
|
||||
//!
|
||||
//! The full design (idle pool, reaper task, per-server TTL, health
|
||||
//! checks, graceful shutdown) lands in **Phase 5** per
|
||||
//! `docs/PHASE-5-IMPLEMENTATION-PLAN.md`. Phase 1 Step 6.5 ships just
|
||||
//! enough for the type to exist on `AppState` and participate in
|
||||
//! construction / test round-trips.
|
||||
//!
|
||||
//! The key type `McpServerKey` hashes the server name plus its full
|
||||
//! transport config (command/args/env for stdio; url/headers for
|
||||
//! http/sse) so that two scopes requesting an identically-configured
|
||||
//! server share an `Arc`, while two scopes requesting differently-
|
||||
//! configured servers (e.g., different API tokens) get independent
|
||||
//! connections. This is the sharing-vs-isolation property described
|
||||
//! in `docs/REST-API-ARCHITECTURE.md` section 5.
|
||||
|
||||
use crate::mcp::{ConnectedServer, JsonField, McpServer, McpTransportType, spawn_mcp_server};
|
||||
|
||||
use anyhow::Result;
|
||||
|
||||
@@ -1,22 +1,3 @@
|
||||
//! Static path and filesystem-lookup helpers that used to live as
|
||||
//! associated functions on [`Config`](super::Config).
|
||||
//!
|
||||
//! None of these functions depend on any `Config` instance data — they
|
||||
//! compute paths from environment variables, XDG directories, or the
|
||||
//! crate constant for the config root. Moving them here is Phase 1
|
||||
//! Step 2 of the REST API refactor: the `Config` struct is shedding
|
||||
//! anything that doesn't actually need per-instance state so the
|
||||
//! eventual split into `AppConfig` + `RequestContext` has a clean
|
||||
//! division line.
|
||||
//!
|
||||
//! # Compatibility shim during migration
|
||||
//!
|
||||
//! The existing associated functions on `Config` (e.g.,
|
||||
//! `Config::config_dir()`) are kept as `#[deprecated]` forwarders that
|
||||
//! call into this module. Callers are migrated module-by-module; when
|
||||
//! the last caller is updated, the forwarders are deleted in a later
|
||||
//! sub-step of Step 2.
|
||||
|
||||
use super::role::Role;
|
||||
use super::{
|
||||
AGENTS_DIR_NAME, BASH_PROMPT_UTILS_FILE_NAME, CONFIG_FILE_NAME, ENV_FILE_NAME,
|
||||
|
||||
@@ -1,29 +1,3 @@
|
||||
//! Per-process RAG instance cache with weak-reference sharing.
|
||||
//!
|
||||
//! `RagCache` lives on [`AppState`](super::AppState) and serves both
|
||||
//! standalone RAGs (attached via `.rag <name>`) and agent-owned RAGs
|
||||
//! (loaded from an agent's `documents:` field). The cache keys with
|
||||
//! [`RagKey`] so that agent RAGs and standalone RAGs occupy distinct
|
||||
//! namespaces even if they share a name.
|
||||
//!
|
||||
//! Entries are held as `Weak<Rag>` so the cache never keeps a RAG
|
||||
//! alive on its own — once all active scopes drop their `Arc<Rag>`,
|
||||
//! the cache entry becomes unupgradable and the next `load()` falls
|
||||
//! through to a fresh disk read.
|
||||
//!
|
||||
//! # Phase 1 Step 6.5 scope
|
||||
//!
|
||||
//! This file introduces the type scaffolding. Actual cache population
|
||||
//! (i.e., routing `use_rag`, `use_agent`, and sub-agent spawning
|
||||
//! through the cache) is deferred to Step 8 when the entry points get
|
||||
//! rewritten. During the bridge window, `Config.rag` keeps serving
|
||||
//! today's callers via direct `Rag::load` / `Rag::init` calls and
|
||||
//! `RagCache` sits on `AppState` as an unused-but-ready service.
|
||||
//!
|
||||
//! See `docs/REST-API-ARCHITECTURE.md` section 5 ("RAG Cache") for
|
||||
//! the full design including concurrent first-load serialization and
|
||||
//! invalidation semantics.
|
||||
|
||||
use crate::rag::Rag;
|
||||
|
||||
use anyhow::Result;
|
||||
|
||||
+253
-29
@@ -1,25 +1,3 @@
|
||||
//! Per-request mutable state for a single Loki interaction.
|
||||
//!
|
||||
//! `RequestContext` owns the runtime state that was previously stored
|
||||
//! on `Config` as `#[serde(skip)]` fields: the active role, session,
|
||||
//! agent, RAG, supervisor state, inbox/escalation queues, the
|
||||
//! conversation's "last message" cursor, and the per-scope
|
||||
//! [`ToolScope`](super::tool_scope::ToolScope) carrying functions and
|
||||
//! live MCP handles.
|
||||
//!
|
||||
//! Each frontend constructs and owns a `RequestContext`:
|
||||
//!
|
||||
//! * **CLI** — one `RequestContext` per invocation, dropped at exit.
|
||||
//! * **REPL** — one long-lived `RequestContext` mutated across turns.
|
||||
//! * **API** — one `RequestContext` per HTTP request, hydrated from a
|
||||
//! persisted session and written back at the end.
|
||||
//!
|
||||
//! `RequestContext` is built via [`RequestContext::bootstrap`] (CLI/REPL
|
||||
//! entry point) or [`RequestContext::new`] (test/child-agent helper).
|
||||
//! It holds an `Arc<AppState>` for shared, immutable services
|
||||
//! (config, vault, MCP factory, RAG cache, MCP registry, base
|
||||
//! functions).
|
||||
|
||||
use super::MessageContentToolCalls;
|
||||
use super::rag_cache::{RagCache, RagKey};
|
||||
use super::session::Session;
|
||||
@@ -2373,10 +2351,10 @@ mod tests {
|
||||
use crate::utils;
|
||||
use crate::utils::get_env_name;
|
||||
use crate::vault::Vault;
|
||||
use serial_test::serial;
|
||||
use std::env;
|
||||
use std::fs::{create_dir_all, remove_dir_all, write};
|
||||
use std::path::PathBuf;
|
||||
use std::sync::Arc;
|
||||
use std::time::{SystemTime, UNIX_EPOCH};
|
||||
|
||||
struct TestConfigDirGuard {
|
||||
@@ -2535,6 +2513,7 @@ mod tests {
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn exit_agent_clears_all_agent_state() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let mut ctx = create_test_ctx();
|
||||
@@ -2604,8 +2583,10 @@ mod tests {
|
||||
}
|
||||
|
||||
fn app_state_with_mcp_config(mcp_server_support: bool, server_names: &[&str]) -> Arc<AppState> {
|
||||
let mut app_config = AppConfig::default();
|
||||
app_config.mcp_server_support = mcp_server_support;
|
||||
let app_config = AppConfig {
|
||||
mcp_server_support,
|
||||
..AppConfig::default()
|
||||
};
|
||||
|
||||
let mcp_config = if server_names.is_empty() {
|
||||
None
|
||||
@@ -2651,6 +2632,7 @@ mod tests {
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn rebuild_tool_scope_mcp_disabled_skips_servers() {
|
||||
let app_state = app_state_with_mcp_config(false, &["github", "slack"]);
|
||||
let mut ctx = RequestContext::new(app_state, WorkingMode::Cmd);
|
||||
@@ -2663,6 +2645,7 @@ mod tests {
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn rebuild_tool_scope_no_enabled_servers_yields_empty_runtime() {
|
||||
let app_state = app_state_with_mcp_config(true, &["github"]);
|
||||
let mut ctx = RequestContext::new(app_state, WorkingMode::Cmd);
|
||||
@@ -2675,6 +2658,7 @@ mod tests {
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn rebuild_tool_scope_no_mcp_config_yields_empty_runtime() {
|
||||
let app_state = app_state_with_mcp_config(true, &[]);
|
||||
let mut ctx = RequestContext::new(app_state, WorkingMode::Cmd);
|
||||
@@ -2687,6 +2671,7 @@ mod tests {
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn rebuild_tool_scope_preserves_tool_tracker() {
|
||||
let app_state = app_state_with_mcp_config(false, &[]);
|
||||
let mut ctx = RequestContext::new(app_state, WorkingMode::Cmd);
|
||||
@@ -2713,6 +2698,7 @@ mod tests {
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn rebuild_tool_scope_repl_mode_appends_user_interaction_functions() {
|
||||
let app_state = app_state_with_mcp_config(false, &[]);
|
||||
let mut ctx = RequestContext::new(app_state, WorkingMode::Repl);
|
||||
@@ -2735,6 +2721,7 @@ mod tests {
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn rebuild_tool_scope_cmd_mode_no_user_interaction_functions() {
|
||||
let app_state = app_state_with_mcp_config(false, &[]);
|
||||
let mut ctx = RequestContext::new(app_state, WorkingMode::Cmd);
|
||||
@@ -2766,8 +2753,10 @@ mod tests {
|
||||
#[test]
|
||||
fn select_functions_returns_none_when_function_calling_disabled() {
|
||||
let app_state = {
|
||||
let mut config = AppConfig::default();
|
||||
config.function_calling_support = false;
|
||||
let config = AppConfig {
|
||||
function_calling_support: false,
|
||||
..AppConfig::default()
|
||||
};
|
||||
Arc::new(AppState {
|
||||
config: Arc::new(config),
|
||||
vault: Arc::new(Vault::default()),
|
||||
@@ -2818,8 +2807,10 @@ mod tests {
|
||||
#[test]
|
||||
fn select_enabled_mcp_servers_returns_empty_when_mcp_disabled() {
|
||||
let app_state = {
|
||||
let mut config = AppConfig::default();
|
||||
config.mcp_server_support = false;
|
||||
let config = AppConfig {
|
||||
mcp_server_support: false,
|
||||
..AppConfig::default()
|
||||
};
|
||||
Arc::new(AppState {
|
||||
config: Arc::new(config),
|
||||
vault: Arc::new(Vault::default()),
|
||||
@@ -2929,6 +2920,7 @@ mod tests {
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn use_role_obj_errors_when_agent_active() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let mut ctx = create_test_ctx();
|
||||
@@ -3065,4 +3057,236 @@ mod tests {
|
||||
ctx.session = Some(Session::default());
|
||||
assert!(!ctx.is_compressing_session());
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn retrieve_role_from_markdown_file() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let roles_dir = paths::roles_dir();
|
||||
create_dir_all(&roles_dir).unwrap();
|
||||
write(
|
||||
roles_dir.join("pirate.md"),
|
||||
"You are a pirate. Speak only in pirate language.",
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
let ctx = create_test_ctx();
|
||||
let role = ctx.retrieve_role(&ctx.app.config, "pirate").unwrap();
|
||||
assert_eq!(role.name(), "pirate");
|
||||
assert!(role.prompt().contains("pirate"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn retrieve_role_builtin_exists() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let ctx = create_test_ctx();
|
||||
let names = paths::list_roles(true);
|
||||
if !names.is_empty() {
|
||||
let role = ctx.retrieve_role(&ctx.app.config, &names[0]);
|
||||
assert!(role.is_ok());
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn retrieve_role_nonexistent_errors() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let ctx = create_test_ctx();
|
||||
let result = ctx.retrieve_role(&ctx.app.config, "definitely_not_a_real_role_xyz");
|
||||
assert!(result.is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn retrieve_role_no_model_id_inherits_current_model() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let roles_dir = paths::roles_dir();
|
||||
create_dir_all(&roles_dir).unwrap();
|
||||
write(roles_dir.join("simple.md"), "You are helpful.").unwrap();
|
||||
|
||||
let ctx = create_test_ctx();
|
||||
let role = ctx.retrieve_role(&ctx.app.config, "simple").unwrap();
|
||||
assert_eq!(role.model().id(), ctx.current_model().id());
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn list_roles_finds_markdown_files() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let roles_dir = paths::roles_dir();
|
||||
create_dir_all(&roles_dir).unwrap();
|
||||
write(roles_dir.join("alpha.md"), "Alpha role").unwrap();
|
||||
write(roles_dir.join("beta.md"), "Beta role").unwrap();
|
||||
write(roles_dir.join("not_a_role.txt"), "ignored").unwrap();
|
||||
|
||||
let names = paths::list_roles(false);
|
||||
assert!(names.contains(&"alpha".to_string()));
|
||||
assert!(names.contains(&"beta".to_string()));
|
||||
assert!(!names.contains(&"not_a_role".to_string()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn list_roles_empty_dir() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let roles_dir = paths::roles_dir();
|
||||
create_dir_all(&roles_dir).unwrap();
|
||||
let names = paths::list_roles(false);
|
||||
assert!(names.is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn session_new_from_ctx_captures_state() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let ctx = create_test_ctx();
|
||||
let session = Session::new_from_ctx(&ctx, &ctx.app.config, "test-session");
|
||||
assert_eq!(session.name(), "test-session");
|
||||
assert!(session.is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn session_save_creates_file() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let ctx = create_test_ctx();
|
||||
let mut session = Session::new_from_ctx(&ctx, &ctx.app.config, "save-test");
|
||||
let session_path = ctx.session_file("save-test");
|
||||
ensure_parent_exists(&session_path).unwrap();
|
||||
|
||||
session.save("save-test", &session_path, false).unwrap();
|
||||
assert!(session_path.exists());
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn use_session_errors_when_already_in_session() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let mut ctx = create_test_ctx();
|
||||
ctx.session = Some(Session::default());
|
||||
|
||||
let app = ctx.app.config.clone();
|
||||
let abort = utils::create_abort_signal();
|
||||
let result = run_async(ctx.use_session(&app, Some("new"), abort));
|
||||
assert!(result.is_err());
|
||||
assert!(
|
||||
result
|
||||
.unwrap_err()
|
||||
.to_string()
|
||||
.contains("Already in a session")
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn use_session_creates_temp_session() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let sessions_dir = paths::local_path("sessions");
|
||||
create_dir_all(&sessions_dir).unwrap();
|
||||
|
||||
let mut ctx = create_test_ctx();
|
||||
let app = ctx.app.config.clone();
|
||||
let abort = utils::create_abort_signal();
|
||||
run_async(ctx.use_session(&app, None, abort)).unwrap();
|
||||
|
||||
assert!(ctx.session.is_some());
|
||||
assert_eq!(ctx.session.as_ref().unwrap().name(), TEMP_SESSION_NAME);
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn use_session_creates_named_session() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let sessions_dir = paths::local_path("sessions");
|
||||
create_dir_all(&sessions_dir).unwrap();
|
||||
|
||||
let mut ctx = create_test_ctx();
|
||||
let app = ctx.app.config.clone();
|
||||
let abort = utils::create_abort_signal();
|
||||
run_async(ctx.use_session(&app, Some("my-session"), abort)).unwrap();
|
||||
|
||||
assert!(ctx.session.is_some());
|
||||
assert_eq!(ctx.session.as_ref().unwrap().name(), "my-session");
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn exit_session_roundtrip() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let sessions_dir = paths::local_path("sessions");
|
||||
create_dir_all(&sessions_dir).unwrap();
|
||||
|
||||
let mut ctx = create_test_ctx();
|
||||
let app = ctx.app.config.clone();
|
||||
let abort = utils::create_abort_signal();
|
||||
run_async(ctx.use_session(&app, Some("roundtrip"), abort.clone())).unwrap();
|
||||
assert!(ctx.session.is_some());
|
||||
|
||||
ctx.exit_session().unwrap();
|
||||
assert!(ctx.session.is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn use_role_obj_and_exit_role_full_cycle() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let mut ctx = create_test_ctx();
|
||||
|
||||
ctx.use_role_obj(Role::new("test-role", "test prompt"))
|
||||
.unwrap();
|
||||
assert!(ctx.role.is_some());
|
||||
assert_eq!(ctx.role.as_ref().unwrap().name(), "test-role");
|
||||
|
||||
let _ = ctx.exit_role();
|
||||
assert!(ctx.role.is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn use_role_obj_twice_replaces_role() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let mut ctx = create_test_ctx();
|
||||
|
||||
ctx.use_role_obj(Role::new("first", "prompt 1")).unwrap();
|
||||
assert_eq!(ctx.role.as_ref().unwrap().name(), "first");
|
||||
|
||||
ctx.use_role_obj(Role::new("second", "prompt 2")).unwrap();
|
||||
assert_eq!(ctx.role.as_ref().unwrap().name(), "second");
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn list_macros_finds_yaml_files() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let macros_dir = paths::macros_dir();
|
||||
create_dir_all(¯os_dir).unwrap();
|
||||
write(macros_dir.join("greet.yaml"), "steps:\n - \".help\"").unwrap();
|
||||
write(macros_dir.join("build.yaml"), "steps:\n - \".help\"").unwrap();
|
||||
|
||||
let names = paths::list_macros();
|
||||
assert!(names.contains(&"greet".to_string()));
|
||||
assert!(names.contains(&"build".to_string()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn list_rags_finds_yaml_files() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let rags_dir = paths::rags_dir();
|
||||
create_dir_all(&rags_dir).unwrap();
|
||||
write(rags_dir.join("docs.yaml"), "embedding_model: test").unwrap();
|
||||
|
||||
let names = paths::list_rags();
|
||||
assert!(names.contains(&"docs".to_string()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn list_rags_empty_dir() {
|
||||
let _guard = TestConfigDirGuard::new();
|
||||
let rags_dir = paths::rags_dir();
|
||||
create_dir_all(&rags_dir).unwrap();
|
||||
assert!(paths::list_rags().is_empty());
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,23 +1,3 @@
|
||||
//! Per-scope tool runtime: resolved functions + live MCP handles +
|
||||
//! call tracker.
|
||||
//!
|
||||
//! `ToolScope` is the unit of tool availability for a single request.
|
||||
//! Every active `RoleLike` (role, session, agent) conceptually owns one.
|
||||
//! The contents are:
|
||||
//!
|
||||
//! * `functions` — the `Functions` declarations visible to the LLM for
|
||||
//! this scope (global tools + role/session/agent filters applied)
|
||||
//! * `mcp_runtime` — live MCP subprocess handles for the servers this
|
||||
//! scope has enabled, keyed by server name
|
||||
//! * `tool_tracker` — per-scope tool call history for auto-continuation
|
||||
//! and looping detection
|
||||
//!
|
||||
//! `ToolScope` lives on [`RequestContext`](super::request_context::RequestContext)
|
||||
//! and is built/replaced as the active scope changes (role swap,
|
||||
//! session swap, agent enter/exit). The base `functions` are seeded
|
||||
//! from [`AppState`](super::app_state::AppState) and per-scope filters
|
||||
//! narrow the visible set.
|
||||
|
||||
use crate::function::{Functions, ToolCallTracker};
|
||||
use crate::mcp::{CatalogItem, ConnectedServer, McpRegistry};
|
||||
|
||||
|
||||
+5
-1
@@ -1280,7 +1280,11 @@ mod tests {
|
||||
];
|
||||
for ext in known {
|
||||
let seps = get_separators(ext);
|
||||
assert_ne!(seps, DEFAULT_SEPARATORS.to_vec(), "Extension '{ext}' should have language-specific separators");
|
||||
assert_ne!(
|
||||
seps,
|
||||
DEFAULT_SEPARATORS.to_vec(),
|
||||
"Extension '{ext}' should have language-specific separators"
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -171,7 +171,6 @@ impl RecursiveCharacterTextSplitter {
|
||||
}
|
||||
}
|
||||
|
||||
// Now that we have the separator, split the text
|
||||
let splits = split_on_separator(text, &separator, keep_separator);
|
||||
|
||||
// Now go merging things, recursively splitting longer texts.
|
||||
|
||||
Reference in New Issue
Block a user