Move data classes into domain package

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"

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())

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())

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: