Coverage for src / local_deep_research / database / models / logs.py: 92%

1""" 

2Logging models for storing application logs and journal information. 

3""" 

4 

5from sqlalchemy import ( 

6 Column, 

7 ForeignKey, 

8 Integer, 

9 Sequence, 

10 String, 

11 Text, 

12) 

13from sqlalchemy_utc import UtcDateTime, utcnow 

14 

15from .base import Base 

16 

17 

18class ResearchLog(Base): 

19 """ 

20 Logging table for all research operations. 

21 

22 All logging from research operations, including debug messages, 

23 errors, and milestones are stored here. 

24 """ 

25 

26 __tablename__ = "app_logs" 

27 

28 id = Column( 

29 Integer, Sequence("reseach_log_id_seq"), primary_key=True, index=True 

30 ) 

31 

32 timestamp = Column(UtcDateTime, server_default=utcnow(), nullable=False) 

33 message = Column(Text, nullable=False) 

34 # Module that the log message came from. 

35 module = Column(Text, nullable=False) 

36 # Function that the log message came from. 

37 function = Column(Text, nullable=False) 

38 # Line number that the log message came from. 

39 line_no = Column(Integer, nullable=False) 

40 # Log level. 

41 level = Column(String(32), nullable=False) 

42 research_id = Column( 

43 String(36), # UUID as string 

44 ForeignKey("research_history.id", ondelete="CASCADE"), 

45 nullable=True, 

46 index=True, 

47 ) 

48 

49 def __repr__(self): 

50 return f"<ResearchLog({self.level}: '{self.message[:50]}...')>" 

51 

52 

53class Journal(Base): 

54 """ 

55 Information about academic journals for quality scoring. 

56 """ 

57 

58 __tablename__ = "journals" 

59 

60 id = Column( 

61 Integer, Sequence("journal_id_seq"), primary_key=True, index=True 

62 ) 

63 

64 # Name of the journal 

65 name = Column(String(255), nullable=False, unique=True, index=True) 

66 # Quality score of the journal 

67 quality = Column(Integer, nullable=True) 

68 # Model that was used to generate the quality score. 

69 quality_model = Column(String(255), nullable=True, index=True) 

70 # Time at which the quality was last analyzed. 

71 quality_analysis_time = Column(Integer, nullable=False) 

72 

73 def __repr__(self): 

74 return f"<Journal(name='{self.name}', quality={self.quality})>"