fix : minor edits

openscilab · Feb 4, 2025 · 1511f53 · 1511f53
1 parent 4e6843a
commit 1511f53
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/memor/prompt.py b/memor/prompt.py
@@ -267,14 +267,14 @@ def from_json(self, json_doc):
             self._message = loaded_obj["message"]
             responses = []
             for response in loaded_obj["responses"]:
-                response_obj = Response(message="")
+                response_obj = Response()
                 response_obj.from_json(response)
                 responses.append(response_obj)
             self._responses = responses
             self._role = Role(loaded_obj["role"])
             self._template = PresetPromptTemplate.DEFAULT.value
             if "template" in loaded_obj:
-                template_obj = CustomPromptTemplate(content="")
+                template_obj = CustomPromptTemplate()
                 template_obj.from_json(loaded_obj["template"])
                 self._template = template_obj
             self._memor_version = loaded_obj["memor_version"]

diff --git a/tests/test_response.py b/tests/test_response.py
@@ -70,7 +70,7 @@ def test_date():
 def test_json():
     response1 = Response(message="I am fine.", model="GPT-4", temperature=0.5, role=Role.USER, score=0.8)
     response1_json = response1.to_json()
-    response2 = Response(message="")
+    response2 = Response()
     response2.from_json(response1_json)
     assert response1 == response2