🐛 Fix Arkose Static

luoshuijs · Mar 27, 2024 · 94cc8dc · 94cc8dc
1 parent 698cb7c
commit 94cc8dc
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/crates/openai/src/serve/mod.rs b/crates/openai/src/serve/mod.rs
@@ -51,7 +51,7 @@ use std::str::FromStr;
 use std::sync::Arc;
 use std::time::{Duration, SystemTime, UNIX_EPOCH};
 use tower_http::trace;
-use tracing::Level;
+use tracing::{Instrument, Level};
 use tracing_subscriber::prelude::__tracing_subscriber_SubscriberExt;
 use tracing_subscriber::util::SubscriberInitExt;
 
@@ -185,7 +185,8 @@ impl Serve {
             .route("/auth/revoke_token", post(post_revoke_token))
             .route("/auth/refresh_session", post(post_refresh_session))
             .route("/auth/sess_token", post(post_sess_token))
-            .route("/auth/billing", post(post_billing));
+            .route("/auth/billing", post(post_billing))
+            .route("/v2/*path", any(arkos_static));
 
         let router = router::config(
             // Enable arkose token endpoint proxy
@@ -509,3 +510,19 @@ async fn check_wan_address() {
         }
     }
 }
+
+
+async fn arkos_static(path: Path<String>) -> Result<Response<Body>, ResponseError> {
+    let client = with_context!(api_client);
+    let data = client
+        .get(format!("{}/v2/{}", arkose::Type::GPT4.origin_url(), path.0.as_str()))
+        .send()
+        .await?;
+    let mut builder = Response::builder()
+        .status(data.status());
+    for (key, value) in data.headers().iter() {
+        builder = builder.header(key, value);
+    }
+    let content = data.bytes().await?;
+    builder.body(content.into()).map_err(ResponseError::InternalServerError)
+}