fix EmbeddingCall

wandb · Jan 22, 2025 · cf4636b · cf4636b
1 parent c8b4648
commit cf4636b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/wandbot/models/embedding.py b/src/wandbot/models/embedding.py
@@ -113,4 +113,8 @@ def __init__(self, provider: str, **kwargs):
         self.model = self.PROVIDER_MAP[provider](**kwargs)
 
     def embed(self, input: Union[str, List[str]] = None) -> List[List[float]]:
-        return self.model.embed(input)
+        return self.model.embed(input)
+
+    def __call__(self, input: Union[str, List[str]] = None) -> List[List[float]]:
+        """Required interface for Chroma's EmbeddingFunction"""
+        return self.embed(input)