Move Context-impls to relevant modules

src/chat.rs

@@ -1,9 +1,10 @@
 use std::collections::HashMap;
 
 use derive_builder::Builder;
+use reqwest::Client;
 use serde::{Serialize, Deserialize};
 
-use crate::completion::{Sequence, Usage};
+use crate::{completion::{Sequence, Usage}, context::{API_URL, Context}};
 
 #[derive(Debug, Clone)]
 pub enum Role {
@@ -107,4 +108,10 @@ pub struct ChatCompletionResponse {
     pub model: String,
     pub choices: Vec<ChatCompletion>,
     pub usage: Usage
+}
+
+impl Context {
+    pub async fn create_chat_completion(&self, chat_completion_request: ChatHistory) -> anyhow::Result<ChatCompletionResponse> {
+        Ok(self.with_auth(Client::builder().build()?.post(&format!("{API_URL}/v1/chat/completions"))).json(&chat_completion_request).send().await?.json::<ChatCompletionResponse>().await?)
+    }
 }

src/completion.rs

@@ -1,8 +1,11 @@
 use std::collections::HashMap;
 
 use derive_builder::Builder;
+use reqwest::Client;
 use serde::{Serialize, Deserialize};
 
+use crate::context::{API_URL, Context};
+
 #[derive(Debug, Clone)]
 pub enum Sequence {
     String(String),
@@ -121,4 +124,10 @@ pub struct CompletionResponse {
     pub model: String,
     pub choices: Vec<Choice>,
     pub usage: Usage,
+}
+
+impl Context {
+    pub async fn create_completion(&self, completion_request: CompletionRequest) -> anyhow::Result<CompletionResponse> {
+        Ok(self.with_auth(Client::builder().build()?.post(&format!("{API_URL}/v1/completions")).json(&completion_request)).send().await?.json::<CompletionResponse>().await?)
+    }
 }

src/context.rs

@@ -1,6 +1,4 @@
-use reqwest::{Client, RequestBuilder};
-
-use crate::{model::{Model, ModelList}, completion::{CompletionRequest, CompletionResponse}, chat::{ChatCompletionResponse, ChatHistory}, edits::{EditRequest, EditResponse}};
+use reqwest::RequestBuilder;
 
 pub struct Context {
     api_key: String,
@@ -33,28 +31,4 @@ impl Context {
             }
         ).bearer_auth(&self.api_key)
     }
-
-    pub async fn get_models(&self) -> anyhow::Result<Vec<Model>> {
-        Ok(self.with_auth(Client::builder().build()?.get(&format!("{API_URL}/v1/models"))).send().await?.json::<ModelList>().await?.data)
-    }
-
-    pub async fn get_model(&self, model_id: &str) -> anyhow::Result<Model> {
-        Ok(self.with_auth(Client::builder().build()?.get(&format!("{API_URL}/v1/models/{model_id}", model_id = model_id))).send().await?.json::<Model>().await?)
-    }
-
-    pub async fn create_completion(&self, completion_request: CompletionRequest) -> anyhow::Result<CompletionResponse> {
-        Ok(self.with_auth(Client::builder().build()?.post(&format!("{API_URL}/v1/completions")).json(&completion_request)).send().await?.json::<CompletionResponse>().await?)
-    }
-
-    pub async fn create_chat_completion(&self, chat_completion_request: ChatHistory) -> anyhow::Result<ChatCompletionResponse> {
-        Ok(self.with_auth(Client::builder().build()?.post(&format!("{API_URL}/v1/chat/completions")).json(&chat_completion_request)).send().await?.json::<ChatCompletionResponse>().await?)
-    }
-
-    pub async fn create_edit(&self, edit_request: EditRequest) -> anyhow::Result<crate::edits::EditResponse> {
-        Ok(self.with_auth(Client::builder().build()?.post(&format!("{API_URL}/v1/edits")).json(&edit_request)).send().await?.json::<EditResponse>().await?)
-    }
-
-    pub async fn create_image(&self, image_request: crate::image::ImageRequest) -> anyhow::Result<crate::image::ImageResponse> {
-        Ok(self.with_auth(Client::builder().build()?.post(&format!("{API_URL}/v1/images/generations")).json(&image_request)).send().await?.json::<crate::image::ImageResponse>().await?)
-    }
 }

src/edits.rs

@@ -1,7 +1,8 @@
 use derive_builder::Builder;
+use reqwest::Client;
 use serde::{Serialize, Deserialize};
 
-use crate::completion::Usage;
+use crate::{completion::Usage, context::{API_URL, Context}};
 
 #[derive(Debug, Serialize, Builder)]
 pub struct EditRequest {
@@ -35,4 +36,10 @@ pub struct EditResponse {
     pub created: u64,
     pub choices: Vec<Edit>,
     pub usage: Usage
+}
+
+impl Context {
+    pub async fn create_edit(&self, edit_request: EditRequest) -> anyhow::Result<crate::edits::EditResponse> {
+        Ok(self.with_auth(Client::builder().build()?.post(&format!("{API_URL}/v1/edits")).json(&edit_request)).send().await?.json::<EditResponse>().await?)
+    }
 }

src/image.rs

@@ -1,6 +1,9 @@
 use derive_builder::Builder;
+use reqwest::Client;
 use serde::{Serialize, Deserialize};
 
+use crate::context::{API_URL, Context};
+
 #[derive(Debug, Clone)]
 pub enum ResponseFormat {
     URL,
@@ -93,4 +96,10 @@ impl<'de> Deserialize<'de> for Image {
 pub struct ImageResponse {
     pub created: u64,
     pub data: Vec<Image>,
+}
+
+impl Context {
+    pub async fn create_image(&self, image_request: crate::image::ImageRequest) -> anyhow::Result<crate::image::ImageResponse> {
+        Ok(self.with_auth(Client::builder().build()?.post(&format!("{API_URL}/v1/images/generations")).json(&image_request)).send().await?.json::<crate::image::ImageResponse>().await?)
+    }
 }

src/model.rs

@@ -1,5 +1,8 @@
+use reqwest::Client;
 use serde::Deserialize;
 
+use crate::context::{API_URL, Context};
+
 #[derive(Debug, Deserialize)]
 pub struct Permission {
     pub id: String,
@@ -29,4 +32,14 @@ pub struct Model {
 #[derive(Debug, Deserialize)]
 pub(crate) struct ModelList {
     pub data: Vec<Model>,
+}
+
+impl Context {
+    pub async fn get_models(&self) -> anyhow::Result<Vec<Model>> {
+        Ok(self.with_auth(Client::builder().build()?.get(&format!("{API_URL}/v1/models"))).send().await?.json::<ModelList>().await?.data)
+    }
+
+    pub async fn get_model(&self, model_id: &str) -> anyhow::Result<Model> {
+        Ok(self.with_auth(Client::builder().build()?.get(&format!("{API_URL}/v1/models/{model_id}", model_id = model_id))).send().await?.json::<Model>().await?)
+    }
 }