From 77541330bdeeef8ff190f1cc01214814b2bc01ef Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Mon, 12 Aug 2024 15:34:56 +0100 Subject: [PATCH] Fix OpenAI API usage --- trustgraph/model/text_completion/openai/llm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/trustgraph/model/text_completion/openai/llm.py b/trustgraph/model/text_completion/openai/llm.py index 93f45b54..44226bbd 100755 --- a/trustgraph/model/text_completion/openai/llm.py +++ b/trustgraph/model/text_completion/openai/llm.py @@ -80,10 +80,10 @@ class Processor(ConsumerProducer): } ) - print(resp, flush=True) + print(resp.choices[0].message.content, flush=True) print("Send response...", flush=True) - r = TextCompletionResponse(response=resp) + r = TextCompletionResponse(response=resp.choices[0].message.content) self.send(r, properties={"id": id}) print("Done.", flush=True)