Spaces:
Sleeping
Sleeping
| from pydantic import BaseModel | |
| class AgentConfig: | |
| """Configuration settings for the advanced research agent.""" | |
| # Model settings | |
| WRITER_MODEL = "gemini-2.5-flash-lite-preview-06-17" | |
| PLANNER_MODEL = "gemini-2.5-flash-lite-preview-06-17" | |
| # Research settings - Deep search parameters | |
| INITIAL_SEARCH_RESULTS = 8 # For initial planning phase | |
| DEEP_DIVE_SEARCH_RESULTS = 12 # Per query in each section | |
| MAX_QUERIES_PER_SECTION = 4 # Limit queries to avoid rate limits | |
| # RAG settings - Document processing | |
| CHUNKS_TO_RETRIEVE = 50 # Initial retrieval before re-ranking | |
| CHUNKS_TO_USE_FOR_WRITING = 15 # Final chunks after re-ranking | |
| CHUNK_SIZE = 1000 # Size of text chunks | |
| CHUNK_OVERLAP = 200 # Overlap between chunks | |
| # LLM settings - Temperature controls | |
| WRITER_TEMPERATURE = 0.4 # Balanced creativity for writing | |
| PLANNER_TEMPERATURE = 0.2 # Low for structured planning | |
| VERIFICATION_TEMPERATURE = 0.1 # Very low for fact-checking | |
| # Quality settings | |
| ENABLE_VERIFICATION = True # Enable fact-checking step | |
| ENABLE_CITATIONS = True # Include source citations | |
| MIN_SOURCES_PER_SECTION = 3 # Minimum sources per section | |
| # Rate limiting | |
| SEARCH_DELAY = 0.5 # Seconds between searches | |
| SECTION_DELAY = 1.0 # Seconds between sections |