openai-interface 0.14.0

//! Create a vector store via `POST /vector_stores`.
//!
//! > ![warn] This module is untested!
//! > If you encounter any issues, please report them on the repository.

use serde::{Deserialize, Serialize};
use url::Url;

use crate::{
    errors::OapiError,
    rest::post::{Post, PostNoStream},
    vector_stores::VectorStore,
};

crate::request_body! {
    /// Creates a vector store.
    #[derive(Debug, Clone, Serialize, Deserialize, Default)]
    pub struct CreateVectorStoreRequest {
        /// The name of the vector store.
        #[serde(skip_serializing_if = "Option::is_none")]
        pub name: Option<String>,
        /// The IDs of files to add to the store right away.
        #[serde(skip_serializing_if = "Option::is_none")]
        pub file_ids: Option<Vec<String>>,
        /// The expiration policy of the store, as raw JSON
        /// (`{"anchor": "last_active_at", "days": N}`).
        #[serde(skip_serializing_if = "Option::is_none")]
        pub expires_after: Option<serde_json::Value>,
        /// Arbitrary key-value metadata (up to 16 pairs) attached to the
        /// store.
        #[serde(skip_serializing_if = "Option::is_none")]
        pub metadata: Option<std::collections::HashMap<String, String>>,
    }
}

impl Post for CreateVectorStoreRequest {
    fn is_streaming(&self) -> bool {
        false
    }

    /// 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");
        Ok(url.to_string())
    }
}

impl PostNoStream for CreateVectorStoreRequest {
    type Response = VectorStore;
}