API: import openai API_KEY = "your_api_key_here" openai.api_key = API_KEY def generate_text(prompt, model="text-davinci-002"):
时间: 2024-05-13 07:14:23 浏览: 188
这是一个使用 OpenAI API 生成文本的 Python 代码示例。你需要替换 `API_KEY` 为你自己的 OpenAI API 密钥。
```python
import openai
API_KEY = "your_api_key_here"
openai.api_key = API_KEY
def generate_text(prompt, model="text-davinci-002"):
response = openai.Completion.create(
engine=model,
prompt=prompt,
max_tokens=1024,
n=1,
stop=None,
temperature=0.7,
)
if response.choices:
return response.choices[0].text.strip()
else:
return ""
```
该函数接受两个参数:`prompt` 和 `model`。`prompt` 是一个字符串,表示你想生成的文本的起始文本,`model` 是一个字符串,表示你想使用的 OpenAI 模型的名称。在本例中,我们使用了 `text-davinci-002` 这个模型,这是 OpenAI 提供的最强大的文本生成模型之一。
该函数使用 `openai.Completion.create` 方法与 OpenAI API 进行通信,以生成文本。在本例中,我们使用了一些默认参数,但你可以根据你的需求调整这些参数,例如 `max_tokens` 表示生成文本的最大长度,`temperature` 表示生成文本的创造性程度。
函数返回一个字符串,表示生成的文本。
阅读全文