detection/GPTdetection.py

@@ -0,0 +1,25 @@
+import os
+
+import openai
+
+
+def detect_gpt(text):
+    client = openai.OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
+    response = client.chat.completions.create(
+        messages=[
+            {
+                "role": "system",
+                "content": "You are a Python code reviewer.Read the code below and identify any potential security vulnerabilities. Classify them by risk level (high, medium, low, none). Only report the line number and the risk level.",
+            },
+            {
+                "role": "user",
+                "content": text,
+            }
+        ],
+        model="gpt-3.5-turbo",
+    )
+    return response.choices[0].message.content
+#TODO: 解析GPT输出成标准格式
+#TODO: 强化提示词，更改模型输出文本格式
+
+print(detect_gpt("import os\nos.system('rm -rf /')"))

requirements.txt

@@ -1,2 +1,3 @@
 reportlab
 packaging
+openai