fix get_embedding in class AzureSim, passing self.emb_call instead of…

… the client
landing-ai · Sep 19, 2024 · 97b4062 · 97b4062
1 parent cc8be2e
commit 97b4062
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):