openai-interface 0.13.0

//! List vector stores via `GET /vector_stores`.
//!
//! > ![warn] This module is untested!
//! > If you encounter any issues, please report them on the repository.

use url::Url;

use crate::{
    errors::OapiError,
    pagination::{PaginationOrder, PaginationQuery},
    rest::get::{Get, GetNoStream},
    vector_stores::VectorStore,
};

/// Lists vector stores.
#[derive(Debug, Clone, Default)]
pub struct ListVectorStoresRequest<'a> {
    /// The standard pagination parameters (`before`, `after`, `limit`,
    /// `order`).
    pub pagination: PaginationQuery<'a>,
    /// The metadata filter, as raw JSON (e.g. `{"key": "value"}`).
    pub filter: Option<serde_json::Value>,
    /// Additional query parameters appended verbatim to the URL.
    pub extra_query: Option<std::collections::HashMap<String, String>>,
}

impl Get for ListVectorStoresRequest<'_> {
    /// Builds the URL for the request.
    ///
    /// `base_url` should be like <https://api.openai.com/v1>
    fn build_url(&self, base_url: &str) -> Result<String, OapiError> {
        let mut url = Url::parse(base_url.trim_end_matches('/')).map_err(OapiError::UrlError)?;
        url.path_segments_mut()
            .map_err(|_| OapiError::UrlCannotBeBase(base_url.to_string()))?
            .push("vector_stores");

        let mut touched = false;
        {
            let mut pairs = url.query_pairs_mut();
            if self.pagination.any_set() {
                self.pagination.append_to(&mut pairs);
                touched = true;
            }
            if let Some(filter) = &self.filter {
                let serialized = serde_json::to_string(filter)
                    .map_err(|e| OapiError::ResponseError(e.to_string()))?;
                pairs.append_pair("filter", &serialized);
                touched = true;
            }
            if let Some(extra_query) = &self.extra_query {
                for (key, value) in extra_query {
                    pairs.append_pair(key, value);
                }
                touched = true;
            }
        }
        if !touched {
            url.set_query(None);
        }

        Ok(url.to_string())
    }
}

impl GetNoStream for ListVectorStoresRequest<'_> {
    type Response = ListVectorStoresResponse;
}

/// The response of listing vector stores.
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
pub struct ListVectorStoresResponse {
    /// The vector stores on this page.
    #[serde(default)]
    pub data: Vec<VectorStore>,
    /// Whether more vector stores exist after this page.
    #[serde(default)]
    pub has_more: Option<bool>,
    /// The ID of the first store on the page, for cursor pagination.
    #[serde(default)]
    pub first_id: Option<String>,
    /// The ID of the last store on the page, for cursor pagination.
    #[serde(default)]
    pub last_id: Option<String>,
    /// The object type (`list`), if the provider sends it.
    #[serde(default)]
    pub object: Option<String>,
}

crate::impl_from_str!(ListVectorStoresResponse);

// `PaginationOrder` is used inside `pagination.append_to`; this alias
// import keeps it reachable for callers of this module.
#[allow(unused_imports)]
use PaginationOrder as _;