feat: support gpt-3.5 model

2026-03-04 07:47:58 +08:00 · 2023-03-02 13:45:23 +08:00
parent ec0a67dd85
commit 65bcf73dac
4 changed files with 195 additions and 2 deletions
--- a/model/model_factory.py
+++ b/model/model_factory.py
@@ -12,8 +12,14 @@ def create_bot(model_type):
    """

    if model_type == const.OPEN_AI:
-        # OpenAI 官方对话模型API
+        # OpenAI 官方对话模型API (gpt-3.0)
        from model.openai.open_ai_model import OpenAIModel
        return OpenAIModel()

+    elif model_type == const.CHATGPT:
+        # ChatGPT API (gpt-3.5-turbo)
+        from model.chatgpt.chatgpt_model import ChatGPTModel
+        return ChatGPTModel()
+
    raise RuntimeError
+