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

Fixed prompt-generic

Browse source
This commit is contained in:
JackColquitt 2024-09-11 08:40:28 -07:00
parent 81a368737d
commit 8f5f0973c2
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
trustgraph/model/prompt/generic/service.py
View file

@ -6,6 +6,7 @@ import json
import re
from .... schema import Definition, Relationship, Triple
from .... schema import Topic
from .... schema import PromptRequest, PromptResponse, Error
from .... schema import TextCompletionRequest, TextCompletionResponse
from .... schema import text_completion_request_queue
@ -202,7 +203,7 @@ class Processor(ConsumerProducer):
if d is None: continue
output.append(
Definition(
Topic(
name=e, definition=d
)
)

3
trustgraph/model/prompt/template/service.py
View file

@ -7,6 +7,7 @@ import json
import re
from .... schema import Definition, Relationship, Triple
from .... schema import Topic
from .... schema import PromptRequest, PromptResponse, Error
from .... schema import TextCompletionRequest, TextCompletionResponse
from .... schema import text_completion_request_queue
@ -216,7 +217,7 @@ class Processor(ConsumerProducer):
if d is None: continue
output.append(
Definition(
Topic(
name=e, definition=d
)
)