apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-17 17:21:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix replace strings for JSON

Browse source
This commit is contained in:
Cyber MacGeddon 2024-09-05 20:38:21 +01:00
parent 3445759598
commit a50efe901e
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trustgraph/model/text_completion/azure/llm.py
View file

@ -135,7 +135,7 @@ class Processor(ConsumerProducer):
print("Send response...", flush=True) print("Send response...", flush=True)
resp = response.replace("```json", "") resp = response.replace("```json", "")
resp = response.replace("```", "") resp = resp.replace("```", "")
r = TextCompletionResponse(response=resp, error=None) r = TextCompletionResponse(response=resp, error=None)
self.producer.send(r, properties={"id": id}) self.producer.send(r, properties={"id": id})

2
trustgraph/model/text_completion/ollama/llm.py
View file

@ -88,7 +88,7 @@ class Processor(ConsumerProducer):
print("Send response...", flush=True) print("Send response...", flush=True)
resp = response.replace("```json", "") resp = response.replace("```json", "")
resp = response.replace("```", "") resp = resp.replace("```", "")
r = TextCompletionResponse(response=resp, error=None) r = TextCompletionResponse(response=resp, error=None)