diff --git a/trustgraph/model/text_completion/cohere/llm.py b/trustgraph/model/text_completion/cohere/llm.py index 75ef8ae4..4c64e8b6 100755 --- a/trustgraph/model/text_completion/cohere/llm.py +++ b/trustgraph/model/text_completion/cohere/llm.py @@ -91,8 +91,8 @@ class Processor(ConsumerProducer): ) resp = output.text - inputtokens = output.meta.billed_units.input_tokens - outputtokens = output.meta.billed_units.output_tokens + inputtokens = int(output.meta.billed_units.input_tokens) + outputtokens = int(output.meta.billed_units.output_tokens) print(resp, flush=True) print(f"Input Tokens: {inputtokens}", flush=True)