More smoothing, as they say

server/src/http/anthropic.rs

@@ -1,18 +1,40 @@
 use crate::http::context::{AppContext, OpenApiResult, bad_request};
-use crate::http::payloads::ApiKeyPayload;
-use crate::llm;
-use poem_openapi::payload::Json;
+use crate::llm::chat;
+use poem_openapi::{Object, OpenApi, payload::Json};
+use serde::Deserialize;
+use std::sync::Arc;
 
-pub async fn get_anthropic_api_key_exists(ctx: &AppContext) -> OpenApiResult<Json<bool>> {
-    let exists =
-        llm::chat::get_anthropic_api_key_exists(ctx.store.as_ref()).map_err(bad_request)?;
-    Ok(Json(exists))
+#[derive(Deserialize, Object)]
+struct ApiKeyPayload {
+    api_key: String,
 }
 
-pub async fn set_anthropic_api_key(
-    payload: Json<ApiKeyPayload>,
-    ctx: &AppContext,
-) -> OpenApiResult<Json<bool>> {
-    llm::chat::set_anthropic_api_key(ctx.store.as_ref(), payload.0.api_key).map_err(bad_request)?;
-    Ok(Json(true))
+pub struct AnthropicApi {
+    ctx: Arc<AppContext>,
+}
+
+impl AnthropicApi {
+    pub fn new(ctx: Arc<AppContext>) -> Self {
+        Self { ctx }
+    }
+}
+
+#[OpenApi]
+impl AnthropicApi {
+    #[oai(path = "/anthropic/key/exists", method = "get")]
+    async fn get_anthropic_api_key_exists(&self) -> OpenApiResult<Json<bool>> {
+        let exists =
+            chat::get_anthropic_api_key_exists(self.ctx.store.as_ref()).map_err(bad_request)?;
+        Ok(Json(exists))
+    }
+
+    #[oai(path = "/anthropic/key", method = "post")]
+    async fn set_anthropic_api_key(
+        &self,
+        payload: Json<ApiKeyPayload>,
+    ) -> OpenApiResult<Json<bool>> {
+        chat::set_anthropic_api_key(self.ctx.store.as_ref(), payload.0.api_key)
+            .map_err(bad_request)?;
+        Ok(Json(true))
+    }
 }