rain-engine-core 0.1.0

Provider-neutral event kernel for RainEngine
Documentation 
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

#![allow(unused_imports)]

use crate::{AgentStateSnapshot, AgentTrigger, KernelEvent};
use async_trait::async_trait;

pub use crate::blob::{BlobStore, BlobStoreError, InMemoryBlobStore};
pub use crate::coordination::{
    CoordinationClaim, CoordinationError, CoordinationStore, InMemoryCoordinationStore,
};
pub use crate::engine::{NativeSkill, SkillExecutionError, SkillExecutor, WasmSkillExecutor};
pub use crate::llm::{LlmProvider, MockLlmProvider, ProviderError, ProviderErrorKind};
pub use crate::memory::{InMemoryMemoryStore, MemoryError, MemoryStore, MemoryStoreExt};
pub use crate::retrieval::{
    RetrievalError, RetrievalStore, RetrievedItem, RetrievedItemKind, WorkingSet,
};

#[derive(Debug, Clone, PartialEq, Default)]
pub struct PlannerOutput {
    pub events: Vec<KernelEvent>,
    pub proposed_plan: Option<crate::ExecutionPlan>,
}

#[async_trait]
pub trait Planner: Send + Sync {
    async fn plan(&self, state: &AgentStateSnapshot, trigger: &AgentTrigger) -> PlannerOutput;
}

#[async_trait]
pub trait SkillStore: Send + Sync {
    async fn store_skill(
        &self,
        manifest: crate::SkillManifest,
        wasm_bytes: Vec<u8>,
    ) -> Result<(), String>;
    async fn list_skills(&self) -> Result<Vec<(crate::SkillManifest, Vec<u8>)>, String>;
    async fn remove_skill(&self, name: &str) -> Result<(), String>;
}

#[async_trait]
pub trait StateProjectionCache: Send + Sync {
    async fn get_projection(
        &self,
        session_id: &str,
    ) -> Result<Option<crate::SessionSnapshot>, String>;
    async fn set_projection(
        &self,
        session_id: &str,
        snapshot: crate::SessionSnapshot,
    ) -> Result<(), String>;
    async fn invalidate(&self, session_id: &str) -> Result<(), String>;
}