From 623f094e5b7596ef859c58b56448696c4da36e11 Mon Sep 17 00:00:00 2001
From: Rock Chin <1010553892@qq.com>
Date: Thu, 2 Mar 2023 16:41:03 +0800
Subject: [PATCH] =?UTF-8?q?doc:=20=E6=B7=BB=E5=8A=A0=E6=B3=A8=E9=87=8A?=
 =?UTF-8?q?=EF=BC=9B=E5=AE=8C=E5=96=84=E6=A0=BC=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/openai/modelmgr.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pkg/openai/modelmgr.py b/pkg/openai/modelmgr.py
index 62540e6b..185feea5 100644
--- a/pkg/openai/modelmgr.py
+++ b/pkg/openai/modelmgr.py
@@ -17,6 +17,7 @@ IMAGE_MODELS = {
 
 }
 
+
 class Model():
 
     can_chat = False
@@ -45,8 +46,10 @@ class Model():
     
     def get_response(self):
         return self.ret
+    
 
 class ChatCompletionModel(Model):
+    """ChatCompletion接口实现"""
     def __init__(self, model_name, user_name):
         request_fun = openai.ChatCompletion.create
         self.can_chat = True
@@ -59,8 +62,10 @@ class ChatCompletionModel(Model):
 
     def get_content(self):
         return self.message
+    
 
 class CompletionModel(Model):
+    """Completion接口实现"""
     def __init__(self, model_name, user_name):
         request_fun = openai.Completion.create
         super().__init__(model_name, user_name, request_fun)
@@ -82,6 +87,7 @@ class CompletionModel(Model):
     
     def get_text(self):
         return self.message
+    
 
 def OpenaiModel(model_name:str, user_name='user'):
     if model_name in CHAT_COMPLETION_MODELS: