From 7e9f9596613b6966af34e2e5f57c321cd91e84d7 Mon Sep 17 00:00:00 2001 From: Alice Hau <110418948+ahau-square@users.noreply.github.com> Date: Fri, 30 May 2025 13:22:46 -0400 Subject: [PATCH] [chore] update tool router env var names to have goose prefix (#2722) Co-authored-by: Alice Hau --- crates/goose/src/agents/router_tool_selector.rs | 6 +++--- crates/goose/src/providers/openai.rs | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/crates/goose/src/agents/router_tool_selector.rs b/crates/goose/src/agents/router_tool_selector.rs index 62d2819885..316197ea37 100644 --- a/crates/goose/src/agents/router_tool_selector.rs +++ b/crates/goose/src/agents/router_tool_selector.rs @@ -39,13 +39,13 @@ impl VectorToolSelector { pub async fn new(provider: Arc, table_name: String) -> Result { let vector_db = ToolVectorDB::new(Some(table_name)).await?; - let embedding_provider = if env::var("EMBEDDING_MODEL_PROVIDER").is_ok() { + let embedding_provider = if env::var("GOOSE_EMBEDDING_MODEL_PROVIDER").is_ok() { // If env var is set, create a new provider for embeddings // Get embedding model and provider from environment variables - let embedding_model = env::var("EMBEDDING_MODEL") + let embedding_model = env::var("GOOSE_EMBEDDING_MODEL") .unwrap_or_else(|_| "text-embedding-3-small".to_string()); let embedding_provider_name = - env::var("EMBEDDING_MODEL_PROVIDER").unwrap_or_else(|_| "openai".to_string()); + env::var("GOOSE_EMBEDDING_MODEL_PROVIDER").unwrap_or_else(|_| "openai".to_string()); // Create the provider using the factory let model_config = ModelConfig::new(embedding_model); diff --git a/crates/goose/src/providers/openai.rs b/crates/goose/src/providers/openai.rs index 71553a4c59..fd46fbcb23 100644 --- a/crates/goose/src/providers/openai.rs +++ b/crates/goose/src/providers/openai.rs @@ -249,7 +249,7 @@ impl EmbeddingCapable for OpenAiProvider { } // Get embedding model from env var or use default - let embedding_model = std::env::var("EMBEDDING_MODEL") + let embedding_model = std::env::var("GOOSE_EMBEDDING_MODEL") .unwrap_or_else(|_| "text-embedding-3-small".to_string()); let request = EmbeddingRequest {