diff --git a/vision_agent/utils/sim.py b/vision_agent/utils/sim.py index 96489134..5c89f700 100644 --- a/vision_agent/utils/sim.py +++ b/vision_agent/utils/sim.py @@ -146,7 +146,7 @@ def emb_call(text: List[str]) -> List[float]: resp = requests.post( base_url, json={"prompt": text[0], "model": model_name} ) - return resp.json()["embedding"] + return resp.json()["embedding"] # type: ignore self.emb_call = emb_call