diff --git a/trustgraph-flow/trustgraph/model/text_completion/cohere/llm.py b/trustgraph-flow/trustgraph/model/text_completion/cohere/llm.py index d6b2b971..8e583040 100755 --- a/trustgraph-flow/trustgraph/model/text_completion/cohere/llm.py +++ b/trustgraph-flow/trustgraph/model/text_completion/cohere/llm.py @@ -70,6 +70,8 @@ class Processor(LlmService): model = self.model ) + return resp + # FIXME: Wrong exception, don't know what this LLM throws # for a rate limit except cohere.TooManyRequestsError: diff --git a/trustgraph-flow/trustgraph/model/text_completion/mistral/llm.py b/trustgraph-flow/trustgraph/model/text_completion/mistral/llm.py index 93eccb35..0c5c1430 100755 --- a/trustgraph-flow/trustgraph/model/text_completion/mistral/llm.py +++ b/trustgraph-flow/trustgraph/model/text_completion/mistral/llm.py @@ -86,6 +86,8 @@ class Processor(LlmService): model = self.model ) + return resp + # FIXME: Wrong exception. The MistralAI library has retry logic # so retry-able errors are retried transparently. It means we # don't get rate limit events.