From a50efe901e2ac1d1b8e40c7fa9a4b0ea7e266ec3 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Thu, 5 Sep 2024 20:38:21 +0100 Subject: [PATCH] Fix replace strings for JSON --- trustgraph/model/text_completion/azure/llm.py | 2 +- trustgraph/model/text_completion/ollama/llm.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/trustgraph/model/text_completion/azure/llm.py b/trustgraph/model/text_completion/azure/llm.py index bbbd749b..44c83452 100755 --- a/trustgraph/model/text_completion/azure/llm.py +++ b/trustgraph/model/text_completion/azure/llm.py @@ -135,7 +135,7 @@ class Processor(ConsumerProducer): print("Send response...", flush=True) resp = response.replace("```json", "") - resp = response.replace("```", "") + resp = resp.replace("```", "") r = TextCompletionResponse(response=resp, error=None) self.producer.send(r, properties={"id": id}) diff --git a/trustgraph/model/text_completion/ollama/llm.py b/trustgraph/model/text_completion/ollama/llm.py index 86bab3b7..e7cdbaf4 100755 --- a/trustgraph/model/text_completion/ollama/llm.py +++ b/trustgraph/model/text_completion/ollama/llm.py @@ -88,7 +88,7 @@ class Processor(ConsumerProducer): print("Send response...", flush=True) resp = response.replace("```json", "") - resp = response.replace("```", "") + resp = resp.replace("```", "") r = TextCompletionResponse(response=resp, error=None)