From c9ea03fcb3b8416fc8a011624187055185c48432 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Sun, 25 Aug 2024 20:11:07 -0700 Subject: [PATCH] fix type error --- 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 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