def Predict(self, request, context):
with status.context(context):
print("request {0}".format(request))
projects = self.cloudml.projects()
body = {
'instances': [
{
'inputs': [request.input],
},
],
}
cmle_request = projects.predict(
name='projects/{0}/models/{1}/versions/{2}'.format(
self.project, self.model_name, self.model_version),
body=body)
response = cmle_example_pb2.PredictResponse()
try:
json_response = cmle_request.execute()
except errors.HttpError as e:
logging.error(
"There was a problem issuing the CMLE predict request: %s",
e)
context.code(code_pb2.UNAVAILABLE)
context.details(
'There was a problem issuing the CMLE predict request {}'.
format(e))
return response
response.output = json_response["predictions"][0]["outputs"][0]
# TODO(jlewi): Need to issue an actual predict request.
return response
def DeletePredictionObjects(self, request, context):
''' Endpoint for pod deletion.
Arguments:
request: request object.
context: the request context.
Return:
Empty Object
'''
with status.context(context):
pods = map(self._map_object, request.objects)
self.dao.delete_pod(pods)
return struct_pb2.Value()
def CreatePredictionObjects(self, request, context):
''' Endpoint for pod creation.
Arguments:
request: request object.
context: the request context.
Return:
RequestResponse
'''
with status.context(context):
pods = map(self._map_object, request.objects)
self.dao.create_pod(pods)
return ai_service_pb2.RequestResponse(message="ok")
def DeleteBook(self, request, context):
with status.context(context):
self._store.delete_book(request.shelf, request.book)
return struct_pb2.Value()
def GetBook(self, request, context):
with status.context(context):
return self._store.get_book(request.shelf, request.book)
def CreateBook(self, request, context):
with status.context(context):
return self._store.create_book(request.shelf, request.book)
def ListBooks(self, request, context):
with status.context(context):
response = bookstore_pb2.ListBooksResponse()
response.books.extend(self._store.list_books(request.shelf))
return response
def DeleteShelf(self, request, context):
with status.context(context):
self._store.delete_shelf(request.shelf)
return struct_pb2.Value()
def GetShelf(self, request, context):
with status.context(context):
return self._store.get_shelf(request.shelf)
def CreateShelf(self, request, context):
with status.context(context):
shelf, _ = self._store.create_shelf(request.shelf)
return shelf
def ListShelves(self, unused_request, context):
with status.context(context):
response = bookstore_pb2.ListShelvesResponse()
response.shelves.extend(self._store.list_shelf())
return response
