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)