From a230008d0816dcd0da88fbef57c2e003e2505546 Mon Sep 17 00:00:00 2001 From: Aidan Do Date: Sat, 21 Dec 2024 22:29:08 +1100 Subject: [PATCH] . --- llama_stack/providers/remote/inference/groq/groq.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/llama_stack/providers/remote/inference/groq/groq.py b/llama_stack/providers/remote/inference/groq/groq.py index 788cb8ce3..1bde81aaf 100644 --- a/llama_stack/providers/remote/inference/groq/groq.py +++ b/llama_stack/providers/remote/inference/groq/groq.py @@ -125,12 +125,14 @@ class GroqInferenceAdapter(Inference, ModelRegistryHelper, NeedsRequestProviderD ) ) - try: + try: response = self._get_client().chat.completions.create(**request) 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