SmartBlogger/state.py at main · codeamt/SmartBlogger · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
from typing import Optional, List, Dict, Any, Set
from pydantic import BaseModel, Field


class EnhancedBlogState(BaseModel):
    # Inputs
    source_code: Optional[str] = None
    uploaded_files: Optional[List[str]] = None
    research_focus: Optional[str] = None

    # Writing style preferences
    tone: Optional[str] = "Professional"
    target_audience: Optional[str] = "Developers"
    writing_style: Optional[List[str]] = Field(default_factory=list)
    custom_questions: Optional[List[str]] = Field(default_factory=list)
    post_type: Optional[str] = None
    content_constraints: Optional[Dict] = None
    style_preferences: Optional[Dict] = None
    success_criteria: Optional[str] = None
    seo_targets: Optional[Dict] = None
    seo_constraints: Optional[Dict] = None

    # Content pipeline
    documents: Optional[List[str]] = None
    document_summaries: Optional[List[str]] = None
    content_summary: Optional[str] = None
    research_queries: Optional[List[str]] = None
    research_context: Optional[Dict] = None
    research_plan: Optional[Dict] = None
    research_sources: Optional[List[str]] = Field(default_factory=list)
    research_status: Optional[str] = "pending"

    # Drafting
    sections: Optional[List[Dict]] = None
    current_section: Optional[Dict] = None
    section_drafts: Dict[str, str] = Field(default_factory=dict)
    citations: Dict[str, List[Dict]] = Field(default_factory=dict)
    next_action: str = "process_inputs"

    # Resources
    token_usage: Dict[str, int] = Field(default_factory=dict)
    free_tier_credits: int = 100
    content_fingerprints: Set[str] = Field(default_factory=set)

    # Plagiarism
    plagiarism_checks: Dict[str, Dict] = Field(default_factory=dict)
    revision_history: Dict[str, List[str]] = Field(default_factory=dict)
    plagiarism_feedback: Optional[str] = None
    needs_rewrite: bool = False

    def update(self, **updates) -> 'EnhancedBlogState':
        """Update state and return new instance"""
        return self.copy(update=updates)

    def dict(self, **kwargs) -> Dict[str, Any]:
        """Override dict to include all fields"""
        return super().dict(**kwargs)