qdrant-client 1.17.0

Rust client for Qdrant Vector Search Engine
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

use tonic_build::Builder;

fn timestamp(f: impl AsRef<std::path::Path>) -> std::time::SystemTime {
    std::fs::metadata(f).unwrap().modified().unwrap()
}

const GRPC_OUTPUT_FILE: &str = "src/qdrant.rs";

#[test]
fn protos() {
    let out_time = timestamp(GRPC_OUTPUT_FILE);
    let mut protos = std::fs::read_dir("proto").unwrap();
    // Make sure the proto files are not dirty
    let protos_files_synced = !protos.any(|d| timestamp(d.unwrap().path()) > out_time);
    let protos_test_synced = timestamp("tests/protos.rs") <= out_time;
    if protos_files_synced && protos_test_synced {
        println!("protobuf files not changed. Exiting early!");
        return;
    }

    tonic_build::configure()
        .configure_deprecations()
        .out_dir("src/") // saves generated structures at this location
        .compile_protos(
            &["proto/qdrant.proto"], // proto entry point
            &["proto"],              // specify the root location to search proto dependencies
        )
        .unwrap();

    // Re-export all custom builder here so they are all located in the same module in the end-user
    // API.
    let custom_reexports = [
        "pub use crate::manual_builder::*;",
        "pub use crate::builder_types::*;",
        "pub use crate::qdrant_client::builders::*;",
        "pub use crate::builders::*;",
    ];
    append_to_file(GRPC_OUTPUT_FILE, &custom_reexports.join("\n"));

    // Vendor gRPC types used in our objects
    append_to_file(GRPC_OUTPUT_FILE, "pub use prost_types::Timestamp;");

    eprintln!("protos_files_synced:{protos_files_synced}, protos_test_synced:{protos_test_synced}");
    eprintln!("proto definitions may be changed. Stubs recompiled. Please commit the changes.")
}

/// Extension to [`Builder`] to configure builder attributes.
trait BuilderExt {
    fn configure_deprecations(self) -> Self;
}

impl BuilderExt for Builder {
    fn configure_deprecations(self) -> Self {
        // Clear deprecated field for VectorOutput.data

        self.field_attribute(
            "PointsUpdateOperation.operation.delete_deprecated",
            "#[deprecated(since = \"1.7.0\", note = \"use `DeletePoints` instead\")]",
        )
        .field_attribute(
            "PointsUpdateOperation.operation.clear_payload_deprecated",
            "#[deprecated(since = \"1.7.0\", note = \"use `ClearPayload` instead\")]",
        )
        .field_attribute(
            "Vector.data",
            "#[doc = \"\n\nDeprecated since 1.16.0, use [`vector`](crate::qdrant::Vector::vector) field instead.\"]",
        )
        .field_attribute(
            "Vector.indices",
            "#[doc = \"\n\nDeprecated since 1.16.0, use [`vector`](crate::qdrant::Vector::vector) field instead.\"]",
        )
        .field_attribute(
            "Vector.vectors_count",
            "#[doc = \"\n\nDeprecated since 1.16.0, use [`vector`](crate::qdrant::Vector::vector) field instead.\"]",
        )
        .field_attribute(
            "VectorOutput.data",
            "#[doc = \"\n\nDeprecated since 1.16.0, use [`into_vector`](crate::qdrant::VectorOutput::into_vector) method instead.\"]",
        )
        .field_attribute(
            "VectorOutput.indices",
            "#[doc = \"\n\nDeprecated since 1.16.0, use [`into_vector`](crate::qdrant::VectorOutput::into_vector) method instead.\"]",
        )
        .field_attribute(
            "VectorOutput.vectors_count",
            "#[doc = \"\n\nDeprecated since 1.16.0, use [`into_vector`](crate::qdrant::VectorOutput::into_vector) method instead.\"]",
        )
    }
}

fn append_to_file(path: &str, line: &str) {
    use std::fs::OpenOptions;
    use std::io::prelude::*;
    writeln!(
        OpenOptions::new().append(true).open(path).unwrap(),
        "{line}",
    )
    .unwrap()
}