Можно ли использовать Firestore из платформы ML Engine / AI?

Question

Я построил модель и положил ее на платформу AI.

Я хотел бы получить дополнительные данные из Firestore (который я использую для хранения документов) перед отправкой прогноза.

Это возможно?

Если нет, то как решить эту проблему? Один из способов - создать еще один микросервис, который подключается к Firebase и возвращает объект, но я бы предпочел оставить все внутри одного контейнера.

Например, с помощью настраиваемого конвейера прогнозирования у меня есть:

class MyPredictor(object):
    """An example Predictor for an AI Platform custom prediction routine."""

    def __init__(self, model):
        """Stores artifacts for prediction. Only initialized via `from_path`.
        """

        self._model = model


    def predict(self, instances, **kwargs):
        """Performs custom prediction.

        Preprocesses inputs, then performs prediction using the trained
        scikit-learn model.

        Args:
            instances: A list of prediction input instances.
            **kwargs: A dictionary of keyword args provided as additional
                fields on the predict request body.

        Returns:
            A list of outputs containing the prediction results.
        """
        # inputs = np.asarray(instances)
        # outputs = self._model.predict(inputs)
        import firebase_admin
        from firebase_admin import credentials
        from firebase_admin import firestore
        cred = credentials.ApplicationDefault()
        return cred

Но это дает мне внутреннюю ошибку при подаче прогноза на AI-Platform.