mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-05-26 15:55:16 +02:00
a70ae9793a
* Altered API to incorporate Flow IDs, refactored for dynamic start/stop of flows * Gateway: Split endpoint / dispatcher for maintainability
29 lines
912 B
Python
29 lines
912 B
Python
|
|
from .. schema import LookupRequest, LookupResponse
|
|
from .. schema import internet_search_request_queue
|
|
from .. schema import internet_search_response_queue
|
|
|
|
from . endpoint import ServiceEndpoint
|
|
from . requestor import ServiceRequestor
|
|
|
|
class InternetSearchRequestor(ServiceRequestor):
|
|
def __init__(self, pulsar_client, timeout, auth):
|
|
|
|
super(InternetSearchRequestor, self).__init__(
|
|
pulsar_client=pulsar_client,
|
|
request_queue=internet_search_request_queue,
|
|
response_queue=internet_search_response_queue,
|
|
request_schema=LookupRequest,
|
|
response_schema=LookupResponse,
|
|
timeout=timeout,
|
|
)
|
|
|
|
def to_request(self, body):
|
|
return LookupRequest(
|
|
term=body["term"],
|
|
kind=body.get("kind", None),
|
|
)
|
|
|
|
def from_response(self, message):
|
|
return { "text": message.text }, True
|
|
|