💎 [Feature] ConversationStyle: Use Enum for more intuitive, and set d…

…efault styles

conversations/__init__.py

@@ -1,3 +1,4 @@
+from .conversation_style import ConversationStyle
 from .conversation_connector import ConversationConnector
 from .conversation_creator import ConversationCreator
 from .conversation_session import ConversationSession

conversations/conversation_connector.py

@@ -8,6 +8,7 @@
     MessageParser,
     OpenaiStreamOutputer,
 )
+from conversations import ConversationStyle
 from utils.logger import logger
 from utils.enver import enver
 
@@ -26,14 +27,18 @@ class ConversationConnector:
 
     def __init__(
         self,
-        conversation_style: str = "precise",
+        conversation_style: ConversationStyle = "precise",
         sec_access_token: str = "",
         client_id: str = "",
         conversation_id: str = "",
         invocation_id: int = 0,
         cookies={},
     ):
-        self.conversation_style = conversation_style
+        if conversation_style.lower() not in ConversationStyle.__members__:
+            self.conversation_style = ConversationStyle.PRECISE.value
+        else:
+            self.conversation_style = conversation_style.lower()
+
         self.sec_access_token = sec_access_token
         self.client_id = client_id
         self.conversation_id = conversation_id

conversations/conversation_style.py

@@ -0,0 +1,10 @@
+from enum import Enum
+
+
+class ConversationStyle(Enum):
+    PRECISE: str = "precise"
+    BALANCED: str = "balanced"
+    CREATIVE: str = "creative"
+    PRECISE_OFFLINE: str = "precise-offline"
+    BALANCED_OFFLINE: str = "balanced-offline"
+    CREATIVE_OFFLINE: str = "creative-offline"

networks/chathub_request_payload_constructor.py

@@ -1,5 +1,6 @@
 import random
 import uuid
+from conversations import ConversationStyle
 
 
 class ChathubRequestPayloadConstructor:
@@ -9,7 +10,7 @@ def __init__(
         client_id: str,
         conversation_id: str,
         invocation_id: int = 0,
-        conversation_style: str = "precise",
+        conversation_style: ConversationStyle = "precise",
     ):
         self.prompt = prompt
         self.client_id = client_id