ChatGPT Python插件开发指南：增强聊天交互的能力

import openai

class MyChatPlugin:
    def __init__(self):
        self.model = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=[
                {"role": "system", "content": "You are a helpful assistant."},
                {"role": "user", "content": "Who won the world series in 2020?"},
                {"role": "assistant", "content": "The Los Angeles Dodgers won the World Series in 2020."},
                {"role": "user", "content": "Where was it played?"}
            ]
        )

    def generate_response(self, user_message):
        self.model.messages.append({"role": "user", "content": user_message})
        response = self.model.choices[0].message.get('content')
        self.model.messages.append({"role": "assistant", "content": response})
        return response

def main():
    plugin = MyChatPlugin()
    print("Welcome to ChatGPT!")
    while True:
        user_message = input("User: ")
        response = plugin.generate_response(user_message)
        print("ChatGPT: ", response)

if __name__ == "__main__":
    main()