Add ability to specify query params for post requests with chat completions

Author: skeptrunedev

openai_dive/src/v1/api.rs

@@ -172,9 +172,11 @@ impl Client {
         &self,
         path: &str,
         parameters: &T,
+        query_params: impl Into<Option<&HashMap<String, String>>>,
     ) -> Result<ResponseWrapper<String>, APIError> {
         let result = self
             .build_request(Method::POST, path, "application/json")
+            .query(&query_params.into())
             .json(&parameters)
             .send()
             .await;

openai_dive/src/v1/endpoints/administration/invites.rs

@@ -55,7 +55,7 @@ impl Invites<'_> {
         let response = self
             .administration
             .client
-            .post("/organization/invites", &parameters)
+            .post("/organization/invites", &parameters, None)
             .await?;
 
         let response: Invite = format_response(response.data)?;

openai_dive/src/v1/endpoints/administration/project_rate_limits.rs

@@ -58,6 +58,7 @@ impl ProjectRateLimits<'_> {
                     project_id, rate_limit_id
                 ),
                 &parameters,
+                None,
             )
             .await?;
 

openai_dive/src/v1/endpoints/administration/project_service_accounts.rs

@@ -73,6 +73,7 @@ impl ProjectServiceAccounts<'_> {
             .post(
                 &format!("/organization/projects/{}/service_accounts", project_id),
                 &parameters,
+                None,
             )
             .await?;
 

openai_dive/src/v1/endpoints/administration/project_users.rs

@@ -70,6 +70,7 @@ impl ProjectUsers<'_> {
             .post(
                 &format!("/organization/projects/{}/users", project_id),
                 &parameters,
+                None,
             )
             .await?;
 
@@ -91,6 +92,7 @@ impl ProjectUsers<'_> {
             .post(
                 &format!("/organization/projects/{}/users/{}", project_id, user_id),
                 &parameters,
+                None,
             )
             .await?;
 

openai_dive/src/v1/endpoints/administration/projects.rs

@@ -55,7 +55,7 @@ impl Projects<'_> {
         let response = self
             .administration
             .client
-            .post("/organization/projects", &parameters)
+            .post("/organization/projects", &parameters, None)
             .await?;
 
         let response: Project = format_response(response.data)?;
@@ -75,6 +75,7 @@ impl Projects<'_> {
             .post(
                 &format!("/organization/projects/{}", project_id),
                 &parameters,
+                None,
             )
             .await?;
 
@@ -91,6 +92,7 @@ impl Projects<'_> {
             .post(
                 &format!("/organization/projects/{}/archive", project_id),
                 &(),
+                None,
             )
             .await?;
 

openai_dive/src/v1/endpoints/administration/users.rs

@@ -59,7 +59,11 @@ impl Users<'_> {
         let response = self
             .administration
             .client
-            .post(&format!("/organization/users/{}", user_id), &parameters)
+            .post(
+                &format!("/organization/users/{}", user_id),
+                &parameters,
+                None,
+            )
             .await?;
 
         let response: User = format_response(response.data)?;

openai_dive/src/v1/endpoints/batch.rs

@@ -22,7 +22,7 @@ impl Client {
 impl Batches<'_> {
     /// Creates and executes a batch from an uploaded file of requests
     pub async fn create(&self, parameters: CreateBatchParameters) -> Result<Batch, APIError> {
-        let response = self.client.post("/batches", &parameters).await?;
+        let response = self.client.post("/batches", &parameters, None).await?;
 
         let response: Batch = format_response(response.data)?;
 
@@ -42,7 +42,7 @@ impl Batches<'_> {
     pub async fn cancel(&self, id: &str) -> Result<Batch, APIError> {
         let response = self
             .client
-            .post(&format!("/batches/{id}/cancel"), &())
+            .post(&format!("/batches/{id}/cancel"), &(), None)
             .await?;
 
         let response: Batch = format_response(response.data)?;

openai_dive/src/v1/endpoints/chat.rs

@@ -40,7 +40,14 @@ impl Chat<'_> {
         &self,
         parameters: ChatCompletionParameters,
     ) -> Result<ResponseWrapper<ChatCompletionResponse>, APIError> {
-        let response = self.client.post("/chat/completions", &parameters).await?;
+        let response = self
+            .client
+            .post(
+                "/chat/completions",
+                &parameters,
+                parameters.query_params.as_ref(),
+            )
+            .await?;
 
         let data: ChatCompletionResponse = format_response(response.data)?;
 

openai_dive/src/v1/endpoints/embeddings.rs

@@ -31,7 +31,7 @@ impl Embeddings<'_> {
         &self,
         parameters: EmbeddingParameters,
     ) -> Result<ResponseWrapper<EmbeddingResponse>, APIError> {
-        let response = self.client.post("/embeddings", &parameters).await?;
+        let response = self.client.post("/embeddings", &parameters, None).await?;
 
         let data: EmbeddingResponse = format_response(response.data)?;