Implement deep load function for OpenAIPrompt in OpenAIChat

devchat-ai · May 22, 2023 · d294743 · d294743
1 parent ebb7987
commit d294743
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/devchat/openai/openai_chat.py b/devchat/openai/openai_chat.py
@@ -3,6 +3,7 @@
 import openai
 from devchat.chat import Chat
 from devchat.utils import get_git_user_info
+from .openai_message import OpenAIMessage
 from .openai_prompt import OpenAIPrompt
 
 
@@ -49,6 +50,12 @@ def init_prompt(self, request: str) -> OpenAIPrompt:
         return prompt
 
     def load_prompt(self, data: dict) -> OpenAIPrompt:
+        data['_new_messages'] = {
+            k: [OpenAIMessage(**m) for m in v] if isinstance(v, list) else OpenAIMessage(**v)
+            for k, v in data['_new_messages'].items()
+        }
+        data['_history_messages'] = {k: [OpenAIMessage(**m) for m in v]
+                                     for k, v in data['_history_messages'].items()}
         return OpenAIPrompt(**data)
 
     def complete_response(self, prompt: OpenAIPrompt) -> str:

diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@ def read_file(file_path):
 
 setup(
     name="devchat",
-    version="0.1.7",
+    version="0.1.8",
     packages=find_packages(),
     install_requires=requirements,
     entry_points={