Bläddra i källkod

feat: 新增校验字段

huangzhichao 2 veckor sedan
förälder
incheckning
b768bde30a
2 ändrade filer med 4 tillägg och 1 borttagningar
  1. 3 1
      app/providers/evaluation_provider.py
  2. 1 0
      app/schemas/base.py

+ 3 - 1
app/providers/evaluation_provider.py

@@ -118,6 +118,7 @@ class EvaluationProvider:
         # Safely parse tool call arguments (if any)
         content = True
         reason = ""
+        corrected_msg = ""
         try:
             tool_calls = getattr(msg, "tool_calls", None) or []
             if tool_calls:
@@ -128,6 +129,7 @@ class EvaluationProvider:
                     if isinstance(args, dict):
                         content = bool(args.get("pass", True))
                         reason = str(args.get("reason", "")).strip()
+                        corrected_msg = str(args.get("corrected_copy", "")).strip()
         except Exception as e:
             logger.error("parse tool call failed: %s", e, exc_info=True)
             return DataResponse(code=1, data=None, msg=f"parse tool call failed: {e}")
@@ -135,4 +137,4 @@ class EvaluationProvider:
         print("✅ PASS:\n", content)
         print("✅ REASON:\n", reason)
 
-        return DataResponse(code=0, data=CopywritingEvaluationPayload(content=content, reason=reason), msg="success")
+        return DataResponse(code=0, data=CopywritingEvaluationPayload(content=content, reason=reason, corrected_msg=corrected_msg), msg="success")

+ 1 - 0
app/schemas/base.py

@@ -34,6 +34,7 @@ class TextToSpeechPayload(BaseModel):
 class CopywritingEvaluationPayload(BaseModel):
     content: bool
     reason: str
+    corrected_msg: str
 
 class DataResponse(BaseModel):
     code: int