return content_category from the server

nwaughachukwuma · Nov 12, 2024 · c745abf · c745abf
1 parent c368006
commit c745abf
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/api/src/utils/audiocast_utils.py b/api/src/utils/audiocast_utils.py
@@ -15,6 +15,7 @@ class GenerateAudioCastResponse(BaseModel):
     script: str
     source_content: str
     chats: List[SessionChatItem]
+    category: ContentCategory
     title: str | None
     created_at: str | None
 

diff --git a/api/src/utils/generate_audiocast.py b/api/src/utils/generate_audiocast.py
@@ -93,4 +93,5 @@ def _run_on_background():
         created_at=datetime.now().strftime("%Y-%m-%d %H:%M"),
         chats=session_data.chats,
         title=title,
+        category=session_data.category,
     )
diff --git a/api/src/utils/get_audiocast.py b/api/src/utils/get_audiocast.py
@@ -41,4 +41,5 @@ def get_audiocast(session_id: str):
         created_at=created_at,
         chats=session_data.chats,
         title=title,
+        category=session_data.category,
     )