This commit is contained in:
Aidan Do 2024-12-21 22:29:08 +11:00
parent d913fbeafe
commit a230008d08

View file

@ -130,7 +130,9 @@ class GroqInferenceAdapter(Inference, ModelRegistryHelper, NeedsRequestProviderD
except groq.BadRequestError as e:
if e.body.get("error", {}).get("code") == "tool_use_failed":
# For smaller models, Groq may fail to call a tool even when the request is well formed
raise ValueError("Groq failed to call a tool", e.body.get("error", {}))
raise ValueError(
"Groq failed to call a tool", e.body.get("error", {})
) from e
else:
raise e