Coverage for src/app/services/save_conversation_title_service.py: 100%

1from usecases.save_conversation_title_useCase import SaveConversationTitleUseCase 

2from ports.save_conversation_title_port import SaveConversationTitlePort 

3from models.conversation_model import ConversationModel 

4 

5class SaveConversationTitleService(SaveConversationTitleUseCase): 

6 """ 

7 Service class to save conversation titles. 

8 """ 

9 def __init__(self, save_conversation_title_port: SaveConversationTitlePort): 

10 self.save_conversation_title_port = save_conversation_title_port 

11 

12 

13 def save_conversation_title(self, conversation: ConversationModel) -> int: 

14 """ 

15 Save a conversation title by its ID. 

16 Args: 

17 conversation (ConversationModel): The conversation model containing the ID and title. 

18 Returns: 

19 int: The ID of the updated conversation. 

20 """ 

21 try: 

22 return self.save_conversation_title_port.save_conversation_title(conversation) 

23 except Exception as e: 

24 raise e 

25