From d39d3d239e9762d8a6d97c015e6fbaa94d3559c6 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Wed, 6 Mar 2024 16:49:27 -0800 Subject: [PATCH] fix formatting --- vision_agent/lmm/lmm.py | 24 ++++++++++++++++++------ 1 file changed, 18 insertions(+), 6 deletions(-) diff --git a/vision_agent/lmm/lmm.py b/vision_agent/lmm/lmm.py index c1e402f7..488048fc 100644 --- a/vision_agent/lmm/lmm.py +++ b/vision_agent/lmm/lmm.py @@ -112,9 +112,13 @@ def generate_classifier(self, prompt: str) -> ImageTool: ) try: - prompt = json.loads(cast(str, response.choices[0].message.content))["prompt"] + prompt = json.loads(cast(str, response.choices[0].message.content))[ + "prompt" + ] except json.JSONDecodeError: - _LOGGER.error(f"Failed to decode response: {response.choices[0].message.content}") + _LOGGER.error( + f"Failed to decode response: {response.choices[0].message.content}" + ) raise ValueError("Failed to decode response") return CLIP(prompt) @@ -131,9 +135,13 @@ def generate_detector(self, prompt: str) -> ImageTool: ) try: - prompt = json.loads(cast(str, response.choices[0].message.content))["prompt"] + prompt = json.loads(cast(str, response.choices[0].message.content))[ + "prompt" + ] except json.JSONDecodeError: - _LOGGER.error(f"Failed to decode response: {response.choices[0].message.content}") + _LOGGER.error( + f"Failed to decode response: {response.choices[0].message.content}" + ) raise ValueError("Failed to decode response") return GroundingDINO(prompt) @@ -150,9 +158,13 @@ def generate_segmentor(self, prompt: str) -> ImageTool: ) try: - prompt = json.loads(cast(str, response.choices[0].message.content))["prompt"] + prompt = json.loads(cast(str, response.choices[0].message.content))[ + "prompt" + ] except json.JSONDecodeError: - _LOGGER.error(f"Failed to decode response: {response.choices[0].message.content}") + _LOGGER.error( + f"Failed to decode response: {response.choices[0].message.content}" + ) raise ValueError("Failed to decode response") return GroundingSAM(prompt)