simSearch/src/config.py

from .prompt import SALES_SYSTEM_PROMPT, CUSTOMER_SYSTEM_PROMPT
# Conversation Configuration
# conversation_config = {}

from dataclasses import dataclass

@dataclass
class ConversationConfig:
    date: str
    data_path: str
    sales_system_prompt: str
    customer_system_prompt: str
    sales_first: bool = True

    @classmethod
    def make(cls, sales_system_prompt: str = SALES_SYSTEM_PROMPT, customer_system_prompt: str = CUSTOMER_SYSTEM_PROMPT, sales_first: bool = True):
        # rule for naming: use date & time to name the conversation
        # e.g. 2021-06-06-12-00-00
        import datetime
        date = datetime.datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
        data_path = f'data/conversation/conversation_{date}.json'
        return cls(date, data_path, sales_system_prompt, customer_system_prompt, sales_first)

# conversation_config = ConversationConfig.make()
# print('Conversation Configuration:', conversation_config)