fix formatting

landing-ai · Mar 7, 2024 · d39d3d2 · d39d3d2
1 parent 27d0a5a
commit d39d3d2
Showing 1 changed file with 18 additions and 6 deletions.
diff --git 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)