Fix rate limit handler, incomplete

trustgraph-bedrock/trustgraph/model/text_completion/bedrock/llm.py

@@ -258,27 +258,17 @@ class Processor(ConsumerProducer):
 
             print("Done.", flush=True)
 
-        except self.bedrock.exceptions.ThrottlingException:
+        except self.bedrock.exceptions.ThrottlingException as e:
 
-            print("Send rate limit response...", flush=True)
+            print("Hit rate limit:", e, flush=True)
 
-            r = TextCompletionResponse(
+            # Leave rate limit retries to the base handler
-                error=Error(
+            raise TooManyRequests()
-                    type = "rate-limit",
-                    message = str(e),
-                ),
-                response=None,
-                in_token=None,
-                out_token=None,
-                model=None,
-            )
-
-            self.producer.send(r, properties={"id": id})
-
-            self.consumer.acknowledge(msg)
 
         except Exception as e:
 
+            # Apart from rate limits, treat all exceptions as unrecoverable
+
             print(type(e))
             print(f"Exception: {e}")