tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

[chore] update tool router env var names to have goose prefix (#2722)

Browse Source 
Co-authored-by: Alice Hau <ahau@squareup.com>
This commit is contained in:
Alice Hau
2025-05-30 13:22:46 -04:00
committed by GitHub
parent de02f5d17a
commit 7e9f959661
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/agents/router_tool_selector.rs
+3 -3
View File
@@ -39,13 +39,13 @@ impl VectorToolSelector {
pub async fn new(provider: Arc<dyn Provider>, table_name: String) -> Result<Self> { pub async fn new(provider: Arc<dyn Provider>, table_name: String) -> Result<Self> {
let vector_db = ToolVectorDB::new(Some(table_name)).await?; 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 // If env var is set, create a new provider for embeddings
// Get embedding model and provider from environment variables // 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()); .unwrap_or_else(|_| "text-embedding-3-small".to_string());
let embedding_provider_name = 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 // Create the provider using the factory
let model_config = ModelConfig::new(embedding_model); let model_config = ModelConfig::new(embedding_model);
crates/goose/src/providers/openai.rs
+1 -1
View File
@@ -249,7 +249,7 @@ impl EmbeddingCapable for OpenAiProvider {
} }
// Get embedding model from env var or use default // 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()); .unwrap_or_else(|_| "text-embedding-3-small".to_string());
let request = EmbeddingRequest { let request = EmbeddingRequest {