edgequake-sdk 0.3.0

Rust SDK for the EdgeQuake 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
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

//! Entities resource.
//!
//! WHY: All entity endpoints live under /api/v1/graph/entities in the real API.
//! Verified against edgequake/crates/edgequake-api/src/routes.rs.

use crate::client::EdgeQuakeClient;
use crate::error::Result;
use crate::types::graph::*;

pub struct EntitiesResource<'a> {
    pub(crate) client: &'a EdgeQuakeClient,
}

impl<'a> EntitiesResource<'a> {
    /// `GET /api/v1/graph/entities` — paginated list.
    pub async fn list(&self) -> Result<EntityListResponse> {
        self.client.get("/api/v1/graph/entities").await
    }

    /// `GET /api/v1/graph/entities/{name}` — entity detail with relationships + statistics.
    pub async fn get(&self, name: &str) -> Result<EntityDetailResponse> {
        self.client
            .get(&format!(
                "/api/v1/graph/entities/{}",
                urlencoding::encode(name)
            ))
            .await
    }

    /// `POST /api/v1/graph/entities` — create a new entity.
    pub async fn create(&self, req: &CreateEntityRequest) -> Result<CreateEntityResponse> {
        self.client
            .post("/api/v1/graph/entities", Some(req))
            .await
    }

    /// `DELETE /api/v1/graph/entities/{name}?confirm=true`
    pub async fn delete(&self, name: &str) -> Result<()> {
        self.client
            .delete_no_content(&format!(
                "/api/v1/graph/entities/{}?confirm=true",
                urlencoding::encode(name)
            ))
            .await
    }

    /// `GET /api/v1/graph/entities/exists?entity_name=NAME` — check existence.
    pub async fn exists(&self, name: &str) -> Result<EntityExistsResponse> {
        self.client
            .get(&format!(
                "/api/v1/graph/entities/exists?entity_name={}",
                urlencoding::encode(name)
            ))
            .await
    }

    /// `POST /api/v1/graph/entities/merge` — merge two entities.
    pub async fn merge(&self, source: &str, target: &str) -> Result<MergeEntitiesResponse> {
        let body = MergeEntitiesRequest {
            source_entity: source.to_string(),
            target_entity: target.to_string(),
        };
        self.client
            .post("/api/v1/graph/entities/merge", Some(&body))
            .await
    }

    /// `GET /api/v1/graph/entities/{name}/neighborhood`
    pub async fn neighborhood(&self, name: &str) -> Result<NeighborhoodResponse> {
        self.client
            .get(&format!(
                "/api/v1/graph/entities/{}/neighborhood",
                urlencoding::encode(name)
            ))
            .await
    }

    /// `POST /api/v1/graph/degrees/batch`
    pub async fn degrees(&self, names: &[String]) -> Result<DegreesBatchResponse> {
        self.client
            .post("/api/v1/graph/degrees/batch", Some(&names))
            .await
    }

    /// `PUT /api/v1/graph/entities/{name}` — Update entity.
    pub async fn update(
        &self,
        name: &str,
        body: &serde_json::Value,
    ) -> Result<serde_json::Value> {
        self.client
            .put(
                &format!(
                    "/api/v1/graph/entities/{}",
                    urlencoding::encode(name)
                ),
                Some(body),
            )
            .await
    }
}