mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 00:16:23 +02:00
34eb083836
* Plugin architecture for messaging fabric * Schemas use a technology neutral expression * Schemas strictness has uncovered some incorrect schema use which is fixed
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
|
|
from ... schema import TextCompletionRequest, TextCompletionResponse
|
|
from ... messaging import TranslatorRegistry
|
|
|
|
from . requestor import ServiceRequestor
|
|
|
|
class TextCompletionRequestor(ServiceRequestor):
|
|
def __init__(
|
|
self, backend, request_queue, response_queue, timeout,
|
|
consumer, subscriber,
|
|
):
|
|
|
|
super(TextCompletionRequestor, self).__init__(
|
|
backend=backend,
|
|
request_queue=request_queue,
|
|
response_queue=response_queue,
|
|
request_schema=TextCompletionRequest,
|
|
response_schema=TextCompletionResponse,
|
|
subscription = subscriber,
|
|
consumer_name = consumer,
|
|
timeout=timeout,
|
|
)
|
|
|
|
self.request_translator = TranslatorRegistry.get_request_translator("text-completion")
|
|
self.response_translator = TranslatorRegistry.get_response_translator("text-completion")
|
|
|
|
def to_request(self, body):
|
|
return self.request_translator.to_pulsar(body)
|
|
|
|
def from_response(self, message):
|
|
return self.response_translator.from_response_with_completion(message)
|
|
|