bigrag 0.1.0

Rust client for bigRAG — a self-hostable RAG 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

use crate::client::BigRag;
use crate::core::urlencode;
use crate::error::BigRagError;
use crate::types::vectors::{DeleteResponse, UpsertResponse, VectorEntry};

/// Vectors resource — manage raw vectors directly.
pub struct Vectors<'a> {
    pub(crate) client: &'a BigRag,
}

impl Vectors<'_> {
    /// Upsert vectors into a collection.
    pub async fn upsert(
        &self,
        collection: &str,
        vectors: Vec<VectorEntry>,
    ) -> Result<UpsertResponse, BigRagError> {
        let path = format!("/v1/collections/{}/vectors/upsert", urlencode(collection));
        let body = serde_json::json!({ "vectors": vectors });
        self.client.transport.post(&path, &body).await
    }

    /// Delete vectors by ID from a collection.
    pub async fn delete(
        &self,
        collection: &str,
        ids: &[&str],
    ) -> Result<DeleteResponse, BigRagError> {
        let path = format!("/v1/collections/{}/vectors/delete", urlencode(collection));
        let body = serde_json::json!({ "ids": ids });
        self.client.transport.post(&path, &body).await
    }
}