Merge PR #488 and #489

apps/aether-gateway/src/control/route/ai.rs

@@ -104,6 +104,17 @@ pub(super) fn classify_ai_public_route(
                 "gemini:video",
                 true,
             ))
+        } else if normalized_path.ends_with(":embedContent")
+            || normalized_path.ends_with(":batchEmbedContents")
+        {
+            Some(classified_with_request_auth_channel(
+                "ai_public",
+                "gemini",
+                "embedding",
+                "api_key",
+                "gemini:embedding",
+                true,
+            ))
         } else if is_gemini_cli_request(headers) {
             Some(classified_with_request_auth_channel(
                 "ai_public",

apps/aether-gateway/src/control/route/mod.rs

@@ -229,6 +229,8 @@ pub(super) fn is_gemini_models_route(path: &str) -> bool {
     (path.starts_with("/v1/models/") || path.starts_with("/v1beta/models/"))
         && (path.contains(":generateContent")
             || path.contains(":streamGenerateContent")
+            || path.contains(":embedContent")
+            || path.contains(":batchEmbedContents")
             || path.contains(":predictLongRunning"))
 }