Update media_type enum

app/api/schemas.py

@@ -10,8 +10,8 @@ class ChatMessage(BaseModel):
 
 
 class MediaType(str, Enum):
-    MOVIE = "movies"
-    TV = "tvs"
+    MOVIE = "movie"
+    TV = "tv"
 
 class DeviceInfo(BaseModel):
     device_type: Optional[str] = None

app/retrieval/media_retriever.py

@@ -65,12 +65,12 @@ class MediaRetriever:
     def embed_sparse(self, query: str, media_type: str) -> Dict:
         bm25_model = (
             self.bm25_models["movie"]
-            if media_type.lower() == "movies"
+            if media_type.lower() == "movie"
             else self.bm25_models["tv"]
         )
         bm25_vocab = (
             self.bm25_vocabs["movie"]
-            if media_type.lower() == "movies"
+            if media_type.lower() == "movie"
             else self.bm25_vocabs["tv"]
         )
 
@@ -99,7 +99,7 @@ class MediaRetriever:
         self,
         dense_vector: List[float],
         sparse_vector: Dict,
-        media_type: str = "movies",
+        media_type: str = "movie",
         genres=None,
         providers=None,
         year_range=None,
@@ -171,7 +171,7 @@ class MediaRetriever:
     def _query_dense(self, vector, media_type, qdrant_filter):
         collection = (
             self.movie_collection_name
-            if media_type == "movies"
+            if media_type == "movie"
             else self.tv_collection_name
         )
         return self.client.query_points(
@@ -187,7 +187,7 @@ class MediaRetriever:
     def _query_sparse(self, vector, media_type, qdrant_filter):
         collection = (
             self.movie_collection_name
-            if media_type == "movies"
+            if media_type == "movie"
             else self.tv_collection_name
         )
         return self.client.query_points(

app/services/chatbot.py

@@ -13,7 +13,7 @@ def build_chat_fn(retriever, intent_classifier):
     def chat(
         question,
         history,
-        media_type="movies",
+        media_type="movie",
         genres=None,
         providers=None,
         year_range=None,

app/services/usage_logger.py

@@ -25,7 +25,6 @@ def log_query_and_results(
     }
     
     try:
-        # 1. Insert usage log
         query_resp = httpx.post(
             f"{SUPABASE_URL}/rest/v1/query_logs",
             headers=headers,
@@ -35,7 +34,6 @@ def log_query_and_results(
         if query_resp.status_code not in (200, 201, 204):
             print("⚠️ Failed to log usage:", query_resp.text)
 
-        # 2. Insert result logs in batch
         if result_entries:
             result_resp = httpx.post(
                 f"{SUPABASE_URL}/rest/v1/result_logs",