diff --git a/prepline_emails/api/app.py b/prepline_emails/api/app.py index 6bc68dc0c..3a69df118 100644 --- a/prepline_emails/api/app.py +++ b/prepline_emails/api/app.py @@ -14,7 +14,11 @@ limiter = Limiter(key_func=get_remote_address) -app = FastAPI() +app = FastAPI( + title="Unstructured Pipeline API", + description="""""", + version="1.0.0", +) app.state.limiter = limiter app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler) diff --git a/prepline_emails/api/email.py b/prepline_emails/api/email.py index 68631d74b..5c4ec22d4 100644 --- a/prepline_emails/api/email.py +++ b/prepline_emails/api/email.py @@ -5,12 +5,18 @@ import os from typing import List, Union - from fastapi import status, FastAPI, File, Form, Request, UploadFile, APIRouter from slowapi.errors import RateLimitExceeded from slowapi import Limiter, _rate_limit_exceeded_handler from slowapi.util import get_remote_address from fastapi.responses import PlainTextResponse +import json +from fastapi.responses import StreamingResponse +from starlette.types import Send +from base64 import b64encode +from typing import Optional, Mapping, Iterator, Tuple +import secrets + limiter = Limiter(key_func=get_remote_address) app = FastAPI() @@ -26,14 +32,6 @@ def pipeline_api(text): pass -import json -from fastapi.responses import StreamingResponse -from starlette.types import Send -from base64 import b64encode -from typing import Optional, Mapping, Iterator, Tuple -import secrets - - class MultipartMixedResponse(StreamingResponse): CRLF = b"\r\n" diff --git a/requirements/base.txt b/requirements/base.txt index df98a1425..1d760b1d1 100644 --- a/requirements/base.txt +++ b/requirements/base.txt @@ -203,7 +203,7 @@ typing-extensions==4.3.0 # pydantic unstructured==0.3.5 # via -r base.in -unstructured-api-tools==0.4.6 +unstructured-api-tools==0.4.7 # via -r base.in urllib3==1.26.13 # via requests