fix:增加llm结果鲁棒性

detection/GPTdetection.py

@@ -25,7 +25,7 @@ def detectGPT(content: str):
     signal.signal(signal.SIGTERM, timeout_handler)
     signal.alarm(10)
 
-    client = openai.OpenAI(base_url="https://api.xiaoai.plus/v1",api_key=api_key)
+    client = openai.OpenAI(base_url="https://api.xiaoai.plus/v1", api_key=api_key)
     text = content
     # client = openai.OpenAI(api_key="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") #测试用key
     response = client.chat.completions.create(
@@ -60,7 +60,10 @@ def detectGPT(content: str):
 
     classified_results = {"high": [], "medium": [], "low": [], "none": []}
     for res in res_json:
+        try:
             classified_results[res["Risk"]].append(
                 (res["Line"], text.split("\n")[res["Line"] - 1].strip())
             )
+        except IndexError:
+            pass
     return classified_results