symbi-runtime 1.10.0

Agent Runtime System for the Symbi platform
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

//! Integration tests for vector backends.
//!
//! Tests LanceDB (when feature enabled) and Qdrant (when feature enabled + server available).

use std::sync::Arc;
use symbi_runtime::context::types::{KnowledgeId, KnowledgeItem, KnowledgeSource, KnowledgeType};
#[cfg(feature = "vector-lancedb")]
use symbi_runtime::context::LanceDbConfig;
use symbi_runtime::context::{create_vector_backend, VectorBackendConfig, VectorDb};
use symbi_runtime::types::AgentId;
#[cfg(feature = "vector-lancedb")]
use tempfile::TempDir;

fn make_test_item(content: &str) -> KnowledgeItem {
    KnowledgeItem {
        id: KnowledgeId::new(),
        content: content.to_string(),
        knowledge_type: KnowledgeType::Fact,
        confidence: 0.9,
        relevance_score: 0.8,
        source: KnowledgeSource::UserProvided,
        created_at: std::time::SystemTime::now(),
    }
}

async fn run_store_and_search_suite(backend: Arc<dyn VectorDb>) {
    backend.initialize().await.unwrap();

    // Store
    let item = make_test_item("The quick brown fox");
    let id = backend
        .store_knowledge_item(&item, vec![1.0, 0.0, 0.0, 0.0])
        .await
        .unwrap();

    // Stats
    let stats = backend.get_stats().await.unwrap();
    assert!(stats.total_vectors >= 1);

    // Search
    let agent_id = AgentId::new();
    let results = backend
        .search_knowledge_base(agent_id, vec![0.9, 0.1, 0.0, 0.0], 5)
        .await
        .unwrap();
    assert!(!results.is_empty());

    // Delete
    backend.delete_knowledge_item(id).await.unwrap();

    // Health
    assert!(backend.health_check().await.unwrap());
}

#[cfg(feature = "vector-lancedb")]
#[tokio::test]
async fn test_lancedb_backend_integration() {
    let tmp = TempDir::new().unwrap();
    let config = VectorBackendConfig::LanceDb(LanceDbConfig {
        data_path: tmp.path().to_path_buf(),
        collection_name: "integration_test".to_string(),
        vector_dimension: 4,
        ..Default::default()
    });
    let backend = create_vector_backend(config).await.unwrap();
    run_store_and_search_suite(backend).await;
}

#[cfg(feature = "vector-qdrant")]
#[tokio::test]
#[ignore] // Requires running Qdrant: docker run -p 6333:6333 qdrant/qdrant
async fn test_qdrant_backend_integration() {
    use symbi_runtime::context::QdrantConfig;
    let config = VectorBackendConfig::Qdrant(QdrantConfig {
        url: "http://localhost:6333".to_string(),
        collection_name: "integration_test".to_string(),
        vector_dimension: 4,
        ..Default::default()
    });
    let backend = create_vector_backend(config).await.unwrap();
    run_store_and_search_suite(backend).await;
}