From cf4636b2f478129e9787102094f08a4d35c45f27 Mon Sep 17 00:00:00 2001 From: Morgan McGuire Date: Wed, 22 Jan 2025 18:41:46 +0000 Subject: [PATCH] fix EmbeddingCall --- src/wandbot/models/embedding.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/wandbot/models/embedding.py b/src/wandbot/models/embedding.py index 0e4e12f..28429ba 100644 --- 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) \ No newline at end of file + 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) \ No newline at end of file