能封装起来吗
时间: 2023-07-25 10:15:42 浏览: 30
当然可以封装起来。我们可以将 ChitGPT 封装成一个 Python 类,让用户可以直接调用实例来与 AI 进行对话。以下是一个简单的示例代码:
```python
import openai
import json
class ChitGPT:
def __init__(self, api_key, model_id):
self.api_key = api_key
self.model_id = model_id
self.session = openai.Session(api_key=self.api_key)
def ask(self, question, context=None):
prompt = f"{question.strip()}\\nAI:"
if context:
prompt = f"{context.strip()}\\n{prompt}"
response = self.session.complete(
engine="text-davinci-002",
prompt=prompt,
max_tokens=1024,
n=1,
stop=None,
temperature=0.5,
)
answer = response.choices[0].text.strip().split("AI:")[1]
return answer
```
在上述代码中,我们使用了 CSDN 开发的 Python SDK,通过 API Key 和模型 ID 构建了一个 `openai.Session` 实例。`ask` 方法接受用户的问题和上下文,将其组装成 GPT-3 的输入格式,并调用 `complete` 方法来获取 AI 的回答。最后,我们从 API 返回的结果中提取出 AI 的回答,并返回给用户。