diff --git a/trustgraph-bedrock/trustgraph/model/text_completion/bedrock/llm.py b/trustgraph-bedrock/trustgraph/model/text_completion/bedrock/llm.py index 75868b56..9b8818a2 100755 --- a/trustgraph-bedrock/trustgraph/model/text_completion/bedrock/llm.py +++ b/trustgraph-bedrock/trustgraph/model/text_completion/bedrock/llm.py @@ -302,7 +302,7 @@ class Processor(ConsumerProducer): model=str(self.model), ) - self.send(r, properties={"id": id}) + await self.send(r, properties={"id": id}) print("Done.", flush=True) @@ -333,6 +333,8 @@ class Processor(ConsumerProducer): model=None, ) + await self.send(r, properties={"id": id}) + self.consumer.acknowledge(msg) @staticmethod