diff --git a/app/helpers/send_classify_request.py b/app/helpers/send_classify_request.py index b1a0c5c..e58b00f 100644 --- a/app/helpers/send_classify_request.py +++ b/app/helpers/send_classify_request.py @@ -1,5 +1,5 @@ from openai import AsyncOpenAI -from models import EmailData +from app.models import EmailData openai_client = AsyncOpenAI( base_url="http://ollama.internal.henryhosted.com:9292/v1", diff --git a/app/main.py b/app/main.py index 4b3b7b4..1570288 100644 --- a/app/main.py +++ b/app/main.py @@ -1,6 +1,6 @@ import uvicorn from fastapi import FastAPI -from routers.classify_email import router as classify_email_router +from app.routers.classify_email import router as classify_email_router app = FastAPI() @@ -8,4 +8,4 @@ app.include_router(classify_email_router) if __name__ == "__main__": - uvicorn.run("main:app", host="0.0.0.0", port=7999, reload=True) \ No newline at end of file + uvicorn.run("app.main:app", host="0.0.0.0", port=7999, reload=True) \ No newline at end of file diff --git a/app/routers/classify_email.py b/app/routers/classify_email.py index c76c465..81dac45 100644 --- a/app/routers/classify_email.py +++ b/app/routers/classify_email.py @@ -1,10 +1,10 @@ from fastapi import APIRouter from pydantic import BaseModel -from helpers.extract_latest_message import extract_latest_message -from helpers.clean_email_html import clean_email_html -from helpers.remove_disclaimer import remove_disclaimer -from helpers.send_classify_request import send_classify_request -from models import EmailData +from app.helpers.extract_latest_message import extract_latest_message +from app.helpers.clean_email_html import clean_email_html +from app.helpers.remove_disclaimer import remove_disclaimer +from app.helpers.send_classify_request import send_classify_request +from app.models import EmailData class ClassifyRequest(BaseModel):