Update bitloops-platform-embeddings to version 0.1.5, modify gateway URL handling to support optional explicit configuration, and add tests for URL resolution logic.

Author: danias

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "bitloops-platform-embeddings"
-version = "0.1.4"
+version = "0.1.5"
 edition.workspace = true
 rust-version.workspace = true
 license.workspace = true

crates/bitloops-platform-embeddings/Cargo.toml

@@ -7,13 +7,16 @@ use serde_json::{Value, json};
 
 const READY_PROTOCOL_VERSION: u32 = 1;
 const CAPABILITIES: &[&str] = &["embed", "ping", "health", "shutdown"];
+const DEFAULT_PLATFORM_EMBEDDINGS_GATEWAY_URL: &str = "https://platform.bitloops.net/v1/embeddings";
+const PLATFORM_GATEWAY_URL_ENV: &str = "BITLOOPS_PLATFORM_GATEWAY_URL";
+const DEFAULT_PLATFORM_API_KEY_ENV: &str = "BITLOOPS_PLATFORM_GATEWAY_TOKEN";
 
 #[derive(Debug, Parser)]
 #[command(name = "bitloops-platform-embeddings")]
 pub struct Cli {
     #[arg(long)]
-    pub gateway_url: String,
-    #[arg(long)]
+    pub gateway_url: Option<String>,
+    #[arg(long, default_value = DEFAULT_PLATFORM_API_KEY_ENV)]
     pub api_key_env: String,
     #[command(subcommand)]
     command: Commands,
@@ -37,14 +40,15 @@ pub fn run(cli: Cli) -> Result<()> {
 
     match cli.command {
         Commands::Daemon(args) => {
+            let gateway_url = resolve_gateway_url(cli.gateway_url.as_deref())?;
             let api_key = std::env::var(&cli.api_key_env).with_context(|| {
                 format!(
                     "reading embeddings API token from environment variable `{}`",
                     cli.api_key_env
                 )
             })?;
             let client = GatewayEmbeddingsClient::new()?;
-            let daemon = EmbeddingsDaemon::new(cli.gateway_url, api_key, args.model, client);
+            let daemon = EmbeddingsDaemon::new(gateway_url, api_key, args.model, client);
             daemon.run_with_stdio(std::io::stdin().lock(), std::io::stdout())
         }
     }
@@ -60,6 +64,22 @@ fn init_tracing() {
         .try_init();
 }
 
+fn resolve_gateway_url(explicit: Option<&str>) -> Result<String> {
+    if let Some(gateway_url) = explicit.map(str::trim).filter(|value| !value.is_empty()) {
+        return Ok(gateway_url.to_string());
+    }
+
+    if let Some(base_url) = std::env::var(PLATFORM_GATEWAY_URL_ENV)
+        .ok()
+        .map(|value| value.trim().to_string())
+        .filter(|value| !value.is_empty())
+    {
+        return Ok(format!("{}/v1/embeddings", base_url.trim_end_matches('/')));
+    }
+
+    Ok(DEFAULT_PLATFORM_EMBEDDINGS_GATEWAY_URL.to_string())
+}
+
 #[derive(Debug, Clone, PartialEq)]
 struct EmbeddingsFailure {
     message: String,
@@ -606,4 +626,32 @@ mod tests {
             vec![vec![1.0, 2.0], vec![3.0, 4.0]]
         );
     }
+
+    #[test]
+    fn resolve_gateway_url_prefers_explicit_flag() {
+        let url = resolve_gateway_url(Some("https://override.example/v1/embeddings"))
+            .expect("explicit gateway url");
+        assert_eq!(url, "https://override.example/v1/embeddings");
+    }
+
+    #[test]
+    fn resolve_gateway_url_derives_from_platform_gateway_env() {
+        unsafe {
+            std::env::set_var(PLATFORM_GATEWAY_URL_ENV, "https://platform.example");
+        }
+        let url = resolve_gateway_url(None).expect("gateway url from env");
+        unsafe {
+            std::env::remove_var(PLATFORM_GATEWAY_URL_ENV);
+        }
+        assert_eq!(url, "https://platform.example/v1/embeddings");
+    }
+
+    #[test]
+    fn resolve_gateway_url_defaults_to_production_gateway() {
+        unsafe {
+            std::env::remove_var(PLATFORM_GATEWAY_URL_ENV);
+        }
+        let url = resolve_gateway_url(None).expect("default gateway url");
+        assert_eq!(url, DEFAULT_PLATFORM_EMBEDDINGS_GATEWAY_URL);
+    }
 }