pass key instead of rt

src/cli/generator/generator.rs

@@ -166,7 +166,7 @@ impl Generator {
 
         if infer_type_names {
             let mut llm_gen = InferTypeName::default();
-            let suggested_names = llm_gen.generate(config.config(), &self.runtime).await?;
+            let suggested_names = llm_gen.generate(config.config(), self.runtime.env.get("TAILCALL_SECRET")).await?;
             let cfg = RenameTypes::new(suggested_names.iter())
                 .transform(config.config().to_owned())
                 .to_result()?;

src/cli/llm/adapter.rs

@@ -1,12 +1,12 @@
+use std::borrow::Cow;
 use crate::core::runtime::TargetRuntime;
 
 pub struct Adapter {}
 
 impl Adapter {
-    pub fn config(rt: &TargetRuntime) -> genai::adapter::AdapterConfig {
-        let env_key = rt.env.get("TAILCALL_SECRET");
+    pub fn config(key: Option<Cow<str>>) -> genai::adapter::AdapterConfig {
         let mut config = genai::adapter::AdapterConfig::default();
-        if let Some(key) = env_key {
+        if let Some(key) = key {
             config = config.with_auth_env_name(key);
         }
         config

src/cli/llm/infer_type_name.rs

@@ -1,3 +1,4 @@
+use std::borrow::Cow;
 use std::collections::HashMap;
 
 use genai::chat::{ChatMessage, ChatRequest, ChatResponse};
@@ -77,10 +78,10 @@ impl InferTypeName {
     pub async fn generate(
         &mut self,
         config: &Config,
-        runtime: &TargetRuntime,
+        auth_key: Option<Cow<str>>,
     ) -> Result<HashMap<String, String>> {
         let wizard: Wizard<Question, Answer> =
-            Wizard::new(Model::GROQ.llama38192(), Adapter::config(runtime));
+            Wizard::new(Model::GROQ.llama38192(), Adapter::config(auth_key));
 
         let mut new_name_mappings: HashMap<String, String> = HashMap::new();