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 {