tingwu_new

backend/ai_services/bailian_service.py

@@ -0,0 +1,98 @@
+import logging
+import json
+import os
+from django.conf import settings
+from openai import OpenAI
+from .models import AIEvaluation
+
+logger = logging.getLogger(__name__)
+
+class BailianService:
+    def __init__(self):
+        self.api_key = getattr(settings, 'DASHSCOPE_API_KEY', None)
+        if not self.api_key:
+             self.api_key = os.environ.get("DASHSCOPE_API_KEY")
+        
+        if self.api_key:
+            self.client = OpenAI(
+                api_key=self.api_key,
+                base_url="https://dashscope.aliyuncs.com/compatible-mode/v1"
+            )
+        else:
+            self.client = None
+            logger.warning("DASHSCOPE_API_KEY not configured.")
+
+    def evaluate_task(self, evaluation: AIEvaluation):
+        """
+        执行AI评估
+        """
+        if not self.client:
+            evaluation.status = AIEvaluation.Status.FAILED
+            evaluation.error_message = "服务未配置 (DASHSCOPE_API_KEY missing)"
+            evaluation.save()
+            return
+
+        task = evaluation.task
+        if not task.transcription:
+             evaluation.status = AIEvaluation.Status.FAILED
+             evaluation.error_message = "关联任务无逐字稿内容"
+             evaluation.save()
+             return
+
+        evaluation.status = AIEvaluation.Status.PROCESSING
+        evaluation.save()
+
+        try:
+            prompt = evaluation.prompt
+            content = task.transcription
+            
+            # 截断过长的内容以防止超出Token限制 (简单处理，取前10000字)
+            if len(content) > 10000:
+                content = content[:10000] + "...(内容过长已截断)"
+
+            # Construct messages
+            messages = [
+                {'role': 'system', 'content': 'You are a helpful assistant designed to output JSON.'},
+                {'role': 'user', 'content': f"{prompt}\n\n以下是需要评估的内容：\n{content}"}
+            ]
+
+            completion = self.client.chat.completions.create(
+                model=evaluation.model_selection,
+                messages=messages,
+                response_format={"type": "json_object"} 
+            )
+            
+            response_content = completion.choices[0].message.content
+            # Convert to dict for storage
+            raw_response = completion.model_dump()
+            
+            evaluation.raw_response = raw_response
+            
+            # Parse JSON
+            try:
+                result = json.loads(response_content)
+                evaluation.score = result.get('score')
+                evaluation.evaluation = result.get('evaluation') or result.get('comment')
+                
+                # 尝试获取推理过程（如果模型返回了）
+                evaluation.reasoning = result.get('reasoning') or result.get('analysis')
+                
+                if not evaluation.reasoning:
+                     # 如果JSON里没有，把整个JSON作为推理参考
+                     evaluation.reasoning = json.dumps(result, ensure_ascii=False, indent=2)
+
+                evaluation.status = AIEvaluation.Status.COMPLETED
+            except json.JSONDecodeError:
+                evaluation.status = AIEvaluation.Status.FAILED
+                evaluation.error_message = f"无法解析JSON响应: {response_content}"
+                evaluation.reasoning = response_content 
+            
+            evaluation.save()
+            return evaluation
+
+        except Exception as e:
+            logger.error(f"AI Evaluation failed: {e}")
+            evaluation.status = AIEvaluation.Status.FAILED
+            evaluation.error_message = str(e)
+            evaluation.save()
+            return evaluation