apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-18 03:45:12 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix LLM output reporting (#31)

Browse source 
* Fix LLM output reporting

* Bump version
This commit is contained in:
cybermaggedon 2024-08-23 13:44:42 +01:00 • committed by GitHub
parent 1e92b2048a
commit 380bddeb90
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
20 changed files with 216 additions and 216 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -122,7 +122,7 @@ class Processor(ConsumerProducer):
resp = response.replace("```json", "")
resp = response.replace("```", "")
r = TextCompletionResponse(response=resp)
r = TextCompletionResponse(response=resp, error=None)
self.producer.send(r, properties={"id": id})
except TooManyRequests:

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

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

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

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

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

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