Session Structure (#42)

* feat : Session class structure added

* fix : autopep8

memor/__init__.py

@@ -4,6 +4,7 @@
 from .template import PromptTemplate, PresetPromptTemplate
 from .prompt import Prompt, Role
 from .response import Response
+from .session import Session
 from .errors import MemorRenderError, MemorValidationError
 
 __version__ = MEMOR_VERSION

memor/session.py

@@ -0,0 +1,119 @@
+# -*- coding: utf-8 -*-
+"""Session class."""
+
+
+class Session:
+    """Session class."""
+
+    def __init__(
+            self,
+            instruction=None,
+            prompts=[],
+            file_path=None):
+        """Session object initiator."""
+        pass
+
+    def __eq__(self, other_session):
+        """Check sessions equality."""
+        pass
+
+    def __str__(self):
+        """Return string representation of Session."""
+        pass
+
+    def __repr__(self):
+        """Return string representation of Session."""
+        pass
+
+    def __copy__(self):
+        """
+        Return a copy of the Session object.
+
+        :return: a copy of Session object
+        """
+        _class = self.__class__
+        result = _class.__new__(_class)
+        result.__dict__.update(self.__dict__)
+        return result
+
+    def copy(self):
+        """
+        Return a copy of the Session object.
+
+        :return: a copy of Session object
+        """
+        return self.__copy__()
+
+    def add_prompt(self, prompt, index=None):
+        """Add a prompt to the session object."""
+        pass
+
+    def remove_prompt(self, index):
+        """Remove a prompt from the session object."""
+        pass
+
+    def enable_prompt(self, index):
+        """
+        Enable a prompt.
+
+        :param index: index
+        :type index: int
+        :return: None
+        """
+        pass
+
+    def disable_prompt(self, index):
+        """
+        Disable a prompt.
+
+        :param index: index
+        :type index: int
+        :return: None
+        """
+        pass
+
+    def update_prompts(self, prompts):
+        """Update the session prompts."""
+        pass
+
+    def update_instruction(self, instruction):
+        """Update the session instruction."""
+        pass
+
+    def save(self, file_path, save_template=True):
+        """Save method."""
+        pass
+
+    def load(self, file_path):
+        """Load method."""
+        pass
+
+    def from_json(self, json_doc):
+        """
+        Load attributes from the JSON document.
+
+        :param json_doc: JSON document
+        :type json_doc: str
+        :return: None
+        """
+        pass
+
+    def to_json(self):
+        """
+        Convert the session to a JSON object.
+
+        :return: JSON object
+        """
+        pass
+
+    def to_dict(self):
+        """
+        Convert the session to a dictionary.
+
+        :return: dict
+        """
+        pass
+
+    def render(self):
+        """Render method."""
+        pass

tests/test_prompt_template.py

@@ -12,7 +12,7 @@ def test_title1():
         content="Act as a {language} developer and respond to this question:\n{prompt_message}",
         custom_map={
             "language": "Python"})
-    assert template.title == None
+    assert template.title is None
 
 
 def test_title2():
@@ -30,7 +30,7 @@ def test_title3():
         custom_map={
             "language": "Python"},
         title=None)
-    assert template.title == None
+    assert template.title is None
 
 
 def test_content1():