jim/morphik-core
1
0
Fork 0
mirror of https://github.com/james-m-jordan/morphik-core.git synced 2025-05-09 19:32:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

slice azure embeddings as well

Browse Source
This commit is contained in:
Arnav Agrawal 2025-04-08 20:06:18 -07:00
parent 7bdc09828a
commit 0857a7f57e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
core/embedding/litellm_embedding.py
View File

@ -54,7 +54,7 @@ class LiteLLMEmbeddingModel(BaseEmbeddingModel):
try:
model_params = {"model": self.model_config["model_name"]}
if self.model_config["model_name"] == "text-embedding-3-large":
if self.model_config["model_name"] in ["text-embedding-3-large", "azure/text-embedding-3-large"]:
model_params["dimensions"] = PGVECTOR_MAX_DIMENSIONS
# Add all model-specific parameters from the config