vecgraph-proto 0.1.1

Protobuf definitions for vecgraph's gRPC API.
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

use crate::{Edge, EdgeWithVector};
use vecgraph_core::VecGraphError;

impl From<Edge> for vecgraph_core::Edge {
    fn from(proto: Edge) -> Self {
        vecgraph_core::Edge {
            id: proto.id.into(),
            source_node_id: proto.source_node_id.into(),
            target_node_id: proto.target_node_id.into(),
            kind: proto.kind,
            content: proto.content,
            metadata: proto
                .metadata
                .and_then(|bytes| serde_json::from_slice(&bytes).ok()),
        }
    }
}

impl From<vecgraph_core::Edge> for Edge {
    fn from(core: vecgraph_core::Edge) -> Self {
        Edge {
            id: core.id.0,
            source_node_id: core.source_node_id.0,
            target_node_id: core.target_node_id.0,
            kind: core.kind,
            content: core.content,
            metadata: core
                .metadata
                .map(|m| serde_json::to_vec(&m).unwrap_or_default()),
        }
    }
}

impl TryFrom<EdgeWithVector> for vecgraph_core::EdgeWithVector {
    type Error = VecGraphError;

    fn try_from(proto: EdgeWithVector) -> Result<Self, Self::Error> {
        let edge = proto
            .edge
            .ok_or_else(|| VecGraphError::Other("EdgeWithVector missing edge field".into()))?;

        Ok(vecgraph_core::EdgeWithVector {
            edge: edge.into(),
            vector: proto.vectors,
        })
    }
}

impl TryFrom<vecgraph_core::EdgeWithVector> for EdgeWithVector {
    type Error = VecGraphError;

    fn try_from(core: vecgraph_core::EdgeWithVector) -> Result<Self, Self::Error> {
        Ok(EdgeWithVector {
            edge: Some(core.edge.into()),
            vectors: core.vector,
        })
    }
}