From c6d0221dc391a6b0b6f5efd4beb533e76eb51337 Mon Sep 17 00:00:00 2001
From: Stephen Witkowski <stephen.witkowski@66degrees.com>
Date: Fri, 29 Mar 2024 14:42:14 -0400
Subject: [PATCH] Add Google embedding model configuration
---
semantic_router/utils/defaults.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/semantic_router/utils/defaults.py b/semantic_router/utils/defaults.py
index 7b5a3b5f..3c9cbb2d 100644
--- a/semantic_router/utils/defaults.py
+++ b/semantic_router/utils/defaults.py
@@ -26,3 +26,8 @@ class EncoderDefault(Enum):
"AZURE_OPENAI_DEPLOYMENT_NAME", "text-embedding-ada-002"
),
}
+ GOOGLE = {
+ "embedding_model": os.getenv(
+ "GOOGLE_EMBEDDING_MODEL", "textembedding-gecko@003"
+ ),
+ }
--
GitLab