Merge branch 'develop' into 423-model-prompts

Author: sahilds1

server/api/views/uploadFile/test_title.py

@@ -59,25 +59,20 @@ def test_falls_back_to_chatgpt_if_no_title_found(self, mock_openAI):
         doc.metadata = {"title": None}
         doc[0].get_text.return_value = []
 
-        mock_response = MagicMock()
-        mock_response.choices = [MagicMock()]
-        mock_response.choices[0].message.content = "A Study Regarding The Efficacy of Drugs"
-        mock_openAI.return_value = mock_response
+        mock_openAI.return_value = "A Study Regarding The Efficacy of Drugs"
 
-        title.generate_title(doc)
+        result = title.generate_title(doc)
 
         self.assertTrue(mock_openAI.called)
+        self.assertEqual(result, "A Study Regarding The Efficacy of Drugs")
 
     @patch("api.views.uploadFile.title.openAIServices.openAI")
     def test_strips_quotes_from_openai_title(self, mock_openAI):
         doc = MagicMock()
         doc.metadata = {"title": None}
         doc[0].get_text.return_value = []
 
-        mock_response = MagicMock()
-        mock_response.choices = [MagicMock()]
-        mock_response.choices[0].message.content = '"Updated CANMAT/ISBD Guidelines for Treating Mixed Features in Bipolar Disorder"'
-        mock_openAI.return_value = mock_response
+        mock_openAI.return_value = '"Updated CANMAT/ISBD Guidelines for Treating Mixed Features in Bipolar Disorder"'
 
         result = title.generate_title(doc)
 
@@ -89,10 +84,7 @@ def test_truncates_long_openai_title(self, mock_openAI):
         doc.metadata = {"title": None}
         doc[0].get_text.return_value = []
 
-        mock_response = MagicMock()
-        mock_response.choices = [MagicMock()]
-        mock_response.choices[0].message.content = "A" * 300
-        mock_openAI.return_value = mock_response
+        mock_openAI.return_value = "A" * 300
 
         result = title.generate_title(doc)
 

server/api/views/uploadFile/title.py

@@ -56,8 +56,7 @@ def summarize_pdf(pdf: fitz.Document) -> str:
         raise Exception("Failed to read the first page of the PDF file")
 
     # UploadFile model title is limited to 255 chars.
-    response = openAIServices.openAI(
-        first_page_content, UPLOAD_FILE_TITLE_PROMPT, model='gpt-4o', temp=0.0)
-    title = response.choices[0].message.content.strip().strip('"').strip("'")
+    response = openAIServices.openAI(first_page_content, UPLOAD_FILE_TITLE_PROMPT, model='gpt-4o', temp=0.0)
+    title = response.strip().strip('"').strip("'")
     # Truncate to fit UploadFile model's max_length=255 title field as a final safeguard
     return title[:255]