from fastapi import UploadFile
from hayhooks.server.logger import log
from hayhooks.server.utils.base_pipeline_wrapper import BasePipelineWrapper
class PipelineWrapper(BasePipelineWrapper):
def setup(self):
self.pipeline = None
def run_api(self, test_param: str, files: list[UploadFile] | None = None) -> str:
log.info("----- Received files: {}", files)
if files and len(files) > 0:
filenames = [f.filename for f in files if f.filename is not None]
file_contents = [f.file.read() for f in files]
log.info("----- File contents: {}", file_contents)
return f"Received files: {', '.join(filenames)} with param {test_param}"
return f"No files received, param: {test_param}"
