feat: cleanup display of free endpoint (#1860)

2024-10-09 17:47:20 -07:00
parent 828aae8d6e
commit 117134810c
3 changed files with 6 additions and 2 deletions
--- a/letta/embeddings.py
+++ b/letta/embeddings.py
@@ -91,6 +91,9 @@ class EmbeddingEndpoint:
            raise ValueError(
                f"Embeddings endpoint was provided an invalid URL (set to: '{base_url}'). Make sure embedding_endpoint is set correctly in your Letta config."
            )
+        # TODO: find a neater solution - re-mapping for letta endpoint
+        if model == "letta-free":
+            model = "BAAI/bge-large-en-v1.5"
        self.model_name = model
        self._user = user
        self._base_url = base_url