From 6fcdd124707fce8225ca1d8a1f934410b8ae707f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Hi=E1=BA=BFu=20Slash?= Date: Fri, 27 Sep 2024 22:07:08 +0700 Subject: [PATCH] =?UTF-8?q?fix=20get=5Fembedding=20in=20class=20AzureSim,?= =?UTF-8?q?=20passing=20self.emb=5Fcall=20instead=20of=E2=80=A6=20(#239)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * fix get_embedding in class AzureSim, passing self.emb_call instead of the client * black reformatted sim.py --- vision_agent/utils/sim.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/vision_agent/utils/sim.py b/vision_agent/utils/sim.py index 5c89f700..8bb9b205 100644 --- a/vision_agent/utils/sim.py +++ b/vision_agent/utils/sim.py @@ -125,7 +125,9 @@ def __init__( raise ValueError("key is required if no column 'embs' is present.") if sim_key is not None: - self.df["embs"] = self.df[sim_key].apply(lambda x: get_embedding(client, x)) + self.df["embs"] = self.df[sim_key].apply( + lambda x: get_embedding(self.emb_call, x) + ) class OllamaSim(Sim):