Merge branch 'feature/GPT' of https://git.mamahaha.work/sangge/BackDoorBuster into feature/GPT
Some checks are pending
Python application test / build (pull_request) Waiting to run

This commit is contained in:
2024-04-28 15:49:24 +08:00
3 changed files with 28 additions and 1 deletions

View File

@@ -5,8 +5,12 @@ import openai
def detect_gpt(filename: str):
content = read_file_content(filename)
return detectGPT(content)
def detectGPT(content: str):
client = openai.OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
text = read_file_content(filename)
text = content
# client = openai.OpenAI(api_key="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") #测试用key
response = client.chat.completions.create(
messages=[