52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
from openai import OpenAI
|
|
|
|
|
|
# =============================godgpt提问====================================
|
|
|
|
def get_god_chat_completion(prompt,model="gpt-4o"):
|
|
"""
|
|
gotAPI
|
|
:param prompt:
|
|
:return:
|
|
"""
|
|
api_key = "sk-YYHnMAyHysBx7EOLBe6303Ab5d324903905e7c1b3f6e8e58"
|
|
api_base = "https://api.wlai.vip/v1"
|
|
client = OpenAI(api_key=api_key, base_url=api_base)
|
|
|
|
completion = client.chat.completions.create(
|
|
model=model,
|
|
messages=[
|
|
{"role": "system", "content": "你是一个十万粉丝的头条爆款文章创作者。"},
|
|
{"role": "user", "content": prompt}
|
|
]
|
|
)
|
|
|
|
print(completion.choices[0].message.content)
|
|
return completion.choices[0].message.content
|
|
|
|
|
|
|
|
def get_deepseek_chat_completion(prompt, model="deepseek-reasoner"):
|
|
"""
|
|
deepseek问答API
|
|
:param prompt: 提示词
|
|
:param model: 使用的模型
|
|
:return: 返回结果
|
|
deepseek-reasoner 思考模型
|
|
"""
|
|
client = OpenAI(api_key="sk-b37c17c924044f73996a77b3ef74904c", base_url="https://api.deepseek.com")
|
|
|
|
response = client.chat.completions.create(
|
|
model=model,
|
|
messages=[
|
|
{"role": "system", "content": "你是一个十万粉丝的爆款文章创作者！"},
|
|
{"role": "user", "content": prompt},
|
|
],
|
|
stream=False
|
|
)
|
|
|
|
print(response.choices[0].message.content)
|
|
return response.choices[0].message.content
|
|
|
|
# =========================读取文档链接，并开始程序================================
|