Coverage for src/local_deep_research/research_library/notes/services/__init__.py: 100%

1"""Notes services.""" 

2 

3from typing import Optional 

4 

5from ....database.models import SourceType 

6from .note_ai_service import NoteAIService 

7from .note_service import NoteService 

8 

9 

10def get_note_source_type_id(session) -> Optional[str]: 

11 """Get the source type ID for notes. Returns None if not found.""" 

12 source_type = session.query(SourceType).filter_by(name="note").first() 

13 return source_type.id if source_type else None 

14 

15 

16__all__ = ["NoteService", "NoteAIService", "get_note_source_type_id"]