From 97b40622874a3c5c3dba5b4143ad38189e4795e2 Mon Sep 17 00:00:00 2001 From: Hieu Tran Date: Thu, 19 Sep 2024 08:32:37 +0700 Subject: [PATCH] fix get_embedding in class AzureSim, passing self.emb_call instead of the client --- vision_agent/utils/sim.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vision_agent/utils/sim.py b/vision_agent/utils/sim.py index 5c89f700..0f3a8174 100644 --- a/vision_agent/utils/sim.py +++ b/vision_agent/utils/sim.py @@ -125,7 +125,7 @@ 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):