_build_fine_tune_user_content

educational_material.py

@@ -44,13 +44,28 @@ class EducationalMaterial:
 
     def get_fine_tuned_ai_content(self, ai_client, ai_type, original_prompt, result, fine_tune_prompt):
         system_content = self.system_content
-        user_content = self.build_fine_tune_user_content(original_prompt, result, fine_tune_prompt)
+        user_content = self._build_fine_tune_user_content(original_prompt, result, fine_tune_prompt)
         if ai_type.lower() == "openai":
             return self.send_openai_request(ai_client, system_content, user_content)
         elif ai_type.lower() == "bedrock":
             return self.send_bedrock_request(ai_client, system_content, user_content)
         else:
             raise ValueError("Unsupported AI type. Please choose 'openai' or 'redrock'.")
+    
+    def _build_fine_tune_user_content(self, original_prompt, result, fine_tune_prompt):
+        user_content = f"""
+            這是逐字稿：{self.context}
+            ---
+            這是預設的 prompt
+            {original_prompt}
+            ---
+            產生了以下的結果：
+            {result}
+            ---
+            但我不是很滿意，請根據以下的調整，產生新的結果
+            {fine_tune_prompt}
+        """
+        return user_content
 
     def send_openai_request(self, AI_Client, system_content, user_content):
         OPEN_AI_CLIENT = AI_Client
@@ -514,31 +529,4 @@ class EducationalMaterial:
         print("====User content====")
         return user_content
 
-    def build_messages(self, user_content):
-        messages = [{"role": "system", "content": self.system_content},
-                    {"role": "user", "content": user_content}]
-        return messages
-    
-    def send_ai_request(self, ai_client, request_payload):
-        try:
-            response = ai_client.chat.completions.create(**request_payload)
-            response_content = response.choices[0].message.content.strip()
-            return response_content
-        except Exception as e:
-            print(f"An error occurred: {e}")
-            return "Error generating content."
-
-    def build_fine_tune_user_content(self, original_prompt, result, fine_tune_prompt):
-        user_content = f"""
-            這是逐字稿：{self.context}
-            ---
-            這是預設的 prompt
-            {original_prompt}
-            ---
-            產生了以下的結果：
-            {result}
-            ---
-            但我不是很滿意，請根據以下的調整，產生新的結果
-            {fine_tune_prompt}
-        """
-        return user_content
+