pensieve-server 0.1.0

HTTP + gRPC query API, auth stub, health, observability.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

//! Inline agent surface — `POST /v1/agent/ask` + `GET /v1/agent/runs/:run_id`.
//!
//! Wires an ADK-Rust `Runner` backed by Ollama with four inline tools
//! (`list_databases`, `describe_table`, `run_sql`, `sample_rows`) and
//! streams SSE frames back to the caller. Each completed run is persisted
//! into the `agent_runs` table.
//!
//! See `docs/superpowers/specs/2026-04-21-nl-query-agent-and-vectors-design.md`
//! for the broader design context.

pub mod artifact_graph_sync;
pub mod cc_curate;
pub mod ci_correlate;
pub mod datasource_tools;
pub mod dreaming;
pub mod dreaming_local;
pub mod dreaming_skill;
pub mod engine;
pub mod file_promote;
pub mod file_tools;
pub mod identity;
pub mod local;
pub mod memory;

#[cfg(test)]
mod cc_curate_unit_tests;
pub mod memory_conflict;
pub mod memory_extract;
pub mod memory_gate;
pub mod memory_policy;
pub mod memory_queue_store;
pub mod memory_resolve;
pub mod memory_retrieve;
pub mod memory_review;
pub mod memory_settings;
pub mod memory_tools;
pub mod memory_usage_store;
pub mod memory_validity_gate;
pub mod routes;
pub mod runner;
pub mod search_tools;
pub mod sessions;
pub mod skill_delivery;
pub mod skills;
pub mod state;
pub mod tools;
pub mod ui_stream;

pub use ci_correlate::CiCorrelator;
pub use file_promote::FilePromoter;
pub use file_tools::{
    tool_contribute_file, tool_describe_file, tool_file_neighbors, tool_recall_file,
};
pub use memory::MemoryConsolidator;
pub use memory_tools::{
    tool_flush_memory, tool_ingest_entity, tool_link_memory_to_entity, tool_list_memories,
    tool_list_memory_usage, tool_memory_compare, tool_memory_judge, tool_memory_search,
    tool_memory_session_summary, tool_recall_memory, tool_reinforce_memory, tool_save_memories,
    tool_save_memory, tool_update_memory_importance, tool_update_memory_status,
};
pub use search_tools::{tool_graph_search, tool_search};
pub use state::AgentState;
pub use tools::{
    execute_sql, tool_describe_table, tool_explore_schema, tool_find_references_to,
    tool_graph_analytics, tool_graph_traverse, tool_list_databases, tool_retrieve_artifact,
    tool_run_kql, tool_run_sql, tool_sample_rows, ConsumerPublisher, ConsumerSink,
    LocalConsumerPublisher, SharedToolCtx,
};

/// Build the `/v1/agent/*` sub-router. Caller typically mounts it under
/// `/v1/agent` and layers the same auth middleware as `/v1/query`.
pub fn router(state: AgentState) -> axum::Router {
    routes::router(state)
}