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)