vectordb-cli 1.6.0

A CLI tool for semantic code search.
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

use thiserror::Error;
#[cfg(feature = "server")]
use tonic::Status;
use std::io;
use std::net;
use qdrant_client::QdrantError;
#[cfg(feature = "ort")]
use ort;

/// Server-specific error type
#[derive(Error, Debug)]
pub enum ServerError {
    #[error("gRPC service error: {0}")]
    ServiceError(String),
    
    #[error("IO error: {0}")]
    Io(#[from] io::Error),
    
    #[error("Configuration error: {0}")]
    Configuration(String),
    
    #[error("Authentication failed: {0}")]
    Authentication(String),
    
    #[error("Authorization failed: {0}")]
    Authorization(String),
    
    #[error("Qdrant client error: {0}")]
    QdrantError(String),
    
    #[error("Internal server error: {0}")]
    Internal(String),
    
    #[error("Invalid request: {0}")]
    InvalidRequest(String),
    
    #[error("Repository error: {0}")]
    RepositoryError(String),
    
    #[error("Embedding error: {0}")]
    EmbeddingError(String),
    
    #[error("Reflection error: {0}")]
    Reflection(String),
    
    #[error("Transport error: {0}")]
    Transport(tonic::transport::Error),
}

/// Server-specific result type
pub type Result<T> = std::result::Result<T, ServerError>;

#[cfg(feature = "server")]
impl From<ServerError> for Status {
    fn from(err: ServerError) -> Self {
        match err {
            ServerError::ServiceError(msg) => Status::internal(msg),
            ServerError::Io(e) => Status::internal(format!("I/O error: {}", e)),
            ServerError::Configuration(msg) => Status::failed_precondition(msg),
            ServerError::Authentication(msg) => Status::unauthenticated(msg),
            ServerError::Authorization(msg) => Status::permission_denied(msg),
            ServerError::QdrantError(msg) => Status::internal(format!("Database error: {}", msg)),
            ServerError::Internal(msg) => Status::internal(msg),
            ServerError::InvalidRequest(msg) => Status::invalid_argument(msg),
            ServerError::RepositoryError(msg) => Status::failed_precondition(msg),
            ServerError::EmbeddingError(msg) => Status::internal(format!("Embedding error: {}", msg)),
            ServerError::Reflection(msg) => Status::internal(msg),
            ServerError::Transport(e) => Status::internal(format!("Transport error: {}", e)),
        }
    }
}

/// Convert generic errors to ServerError
impl From<anyhow::Error> for ServerError {
    fn from(err: anyhow::Error) -> Self {
        ServerError::Internal(format!("{}", err))
    }
}

/// Convert Qdrant client errors to ServerError
impl From<qdrant_client::QdrantError> for ServerError {
    fn from(err: qdrant_client::QdrantError) -> Self {
        ServerError::QdrantError(format!("{}", err))
    }
}

#[cfg(feature = "server")]
impl From<tonic::transport::Error> for ServerError {
    fn from(err: tonic::transport::Error) -> Self {
        ServerError::Transport(err)
    }
}

/// Convert Rust vector database errors to ServerError
impl From<crate::vectordb::error::VectorDBError> for ServerError {
    fn from(err: crate::vectordb::error::VectorDBError) -> Self {
        match err {
            crate::vectordb::error::VectorDBError::QdrantError(e) => 
                ServerError::QdrantError(format!("{}", e)),
            crate::vectordb::error::VectorDBError::EmbeddingError(e) => 
                ServerError::EmbeddingError(format!("{}", e)),
            _ => ServerError::Internal(format!("{}", err)),
        }
    }
}

#[cfg(feature = "server")]
impl From<tonic_reflection::server::Error> for ServerError {
    fn from(err: tonic_reflection::server::Error) -> Self {
        ServerError::Reflection(err.to_string()) // Or a more specific variant if desired
    }
}

// Add From trait implementation for AddrParseError
impl From<net::AddrParseError> for ServerError {
    fn from(err: net::AddrParseError) -> Self {
        ServerError::Configuration(format!("Invalid socket address: {}", err))
    }
}