Feature: Enable programmatically pass in api_key besides reading from env

Author: aryasoni98

python/cocoindex/llm.py

@@ -44,8 +44,4 @@ class LlmSpec:
     model: str
     address: str | None = None
     api_key: str | None = None
-    api_config: (
-        VertexAiConfig
-        | OpenAiConfig
-        | None
-    ) = None
+    api_config: VertexAiConfig | OpenAiConfig | None = None

src/ops/functions/embed_text.rs

@@ -94,41 +94,34 @@ impl SimpleFunctionFactoryBase for Factory {
             .required()?;
 
         // Create API config based on api_key parameter if provided
-        let api_config = if let Some(api_key) = &spec.api_key {
-            Some(match spec.api_type {
-                LlmApiType::OpenAi => {
-                    LlmApiConfig::OpenAi(super::super::super::llm::OpenAiConfig {
+        let api_config = if let Some(_api_key) = &spec.api_key {
+            match spec.api_type {
+                LlmApiType::OpenAi => Some(LlmApiConfig::OpenAi(
+                    super::super::super::llm::OpenAiConfig {
                         org_id: None,
                         project_id: None,
-                    })
+                    },
+                )),
+                LlmApiType::Anthropic
+                | LlmApiType::Gemini
+                | LlmApiType::Voyage
+                | LlmApiType::LiteLlm
+                | LlmApiType::OpenRouter
+                | LlmApiType::Vllm => {
+                    // These API types don't require a config, just an API key
+                    None
                 }
-                LlmApiType::Anthropic => {
-                    LlmApiConfig::Anthropic(super::super::super::llm::AnthropicConfig {})
-                }
-                LlmApiType::Gemini => {
-                    LlmApiConfig::Gemini(super::super::super::llm::GeminiConfig {})
-                }
-                LlmApiType::Voyage => {
-                    LlmApiConfig::Voyage(super::super::super::llm::VoyageConfig {})
-                }
-                LlmApiType::LiteLlm => {
-                    LlmApiConfig::LiteLlm(super::super::super::llm::LiteLlmConfig {})
-                }
-                LlmApiType::OpenRouter => {
-                    LlmApiConfig::OpenRouter(super::super::super::llm::OpenRouterConfig {})
-                }
-                LlmApiType::Vllm => LlmApiConfig::Vllm(super::super::super::llm::VllmConfig {}),
                 _ => {
                     if let Some(config) = spec.api_config.clone() {
-                        config
+                        Some(config)
                     } else {
                         api_bail!(
                             "API key parameter is not supported for API type {:?}",
                             spec.api_type
                         )
                     }
                 }
-            })
+            }
         } else {
             spec.api_config.clone()
         };