azure_search_rs 0.2.0

Azure AI Search client library for Rust
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

use azure_search_rs::client::AzureSearchClient;

use serde::{Deserialize, Serialize};

#[derive(Debug, Serialize, Deserialize)]
struct Hotel {
    id: String,
    description_vector: Vec<f32>,
    description: String,
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Create client instance
    let service_name = std::env::var("AZURE_SEARCH_SERVICE_NAME")?;
    let endpoint = format!("https://{}.search.windows.net", service_name);
    let index_name = std::env::var("AZURE_SEARCH_INDEX_NAME")?;
    let api_key = std::env::var("AZURE_SEARCH_API_KEY")?;

    let client = AzureSearchClient::new(&endpoint, &api_key, "2024-07-01", None)?;

    // Prepare test documents
    let documents = vec![
        Hotel {
            id: "1".to_string(),
            description_vector: vec![0.1, 0.2, 0.3, 0.4],
            description: "luxury hotel".to_string(),
        },
        Hotel {
            id: "2".to_string(),
            description_vector: vec![0.5, 0.6, 0.7, 0.8],
            description: "budget hotel".to_string(),
        },
    ];

    // Upload documents
    client.put_documents(&index_name, documents).await?;

    println!("Documents uploaded successfully");

    //get documents
    let document: Hotel = client
        .get_document(&index_name, "1")
        .await?;

    println!("Document retrieved: {:?}", document);
    Ok(())
}