diff --git a/prophet/app.py b/prophet/app.py index f1a5dc7..18d45e7 100644 --- a/prophet/app.py +++ b/prophet/app.py @@ -9,7 +9,8 @@ from fastapi.middleware.cors import CORSMiddleware from fastapi.responses import HTMLResponse from fastapi_utils.tasks import repeat_every -from prophet.data import Improvement, Original +from prophet.domain.improvement import Improvement +from prophet.domain.original import Original from prophet.llm import LLMClient BEE_FEED = "https://babylonbee.com/feed" diff --git a/prophet/domain/improvement.py b/prophet/domain/improvement.py new file mode 100644 index 0000000..50c7618 --- /dev/null +++ b/prophet/domain/improvement.py @@ -0,0 +1,12 @@ +from dataclasses import dataclass +from uuid import uuid4 + +from prophet.domain.original import Original + + +@dataclass +class Improvement: # GoodJoke: Queen + original: Original + title: str + summary: str + id: str = str(uuid4()) diff --git a/prophet/data.py b/prophet/domain/original.py similarity index 85% rename from prophet/data.py rename to prophet/domain/original.py index 97e00a4..87dda00 100644 --- a/prophet/data.py +++ b/prophet/domain/original.py @@ -2,7 +2,6 @@ import hashlib import re from dataclasses import dataclass, field from datetime import datetime -from uuid import uuid4 @dataclass @@ -33,11 +32,3 @@ class Original: # BadJoke: Sting if extracted[0]: self.image_link = extracted[0] self.summary = extracted[1] - - -@dataclass -class Improvement: # GoodJoke: Queen - original: Original - title: str - summary: str - id: str = str(uuid4()) diff --git a/prophet/llm.py b/prophet/llm.py index a0b9159..473dcca 100644 --- a/prophet/llm.py +++ b/prophet/llm.py @@ -1,7 +1,7 @@ from groq import Groq -from prophet.data import Original from prophet.config import AiConfig +from prophet.domain.original import Original class LLMClient: