from celery.schedules import crontab

beat_schedule = {
    "knowledge-reindex-weekly": {
        "task": "knowledge.tasks.reindex_all",
        "schedule": crontab(day_of_week=0, hour=2, minute=0),
        "options": {"queue": "heavy"}
    },
    "knowledge-cleanup": {
        "task": "knowledge.tasks.cleanup_orphan_chunks",
        "schedule": crontab(hour=4, minute=0),
        "options": {"queue": "default"}
    },
    "knowledge-sync-external": {
        "task": "knowledge.tasks.sync_external_sources",
        "schedule": crontab(minute=0),
        "options": {"queue": "default"}
    },
    "knowledge-office-heartbeat": {
        "task": "knowledge.tasks.office_heartbeat",
        "schedule": 60.0,
        "options": {"queue": "default"}
    },
}

from celery import shared_task
from app.utils.office_reporter import OfficeReporter

reporter = OfficeReporter(
    agent_id="knowledge_rag",
    department="knowledge",
    name="RAG процессор",
    salary_equivalent=60000,
    fte_coefficient=1.0
)

@shared_task(
    name='knowledge.tasks.index_documents',
    bind=True,
    max_retries=2
)
def index_documents(self, document_ids: list):
    """Индексация документов для RAG."""
    
    reporter.report_working(f"Индексация {len(document_ids)} документов")
    
    try:
        # ... логика индексации ...
        indexed = process_documents(document_ids)
        
        reporter.report_idle(metrics={
            "docs_indexed": get_total_docs(),
            "chunks_count": get_total_chunks(),
            "rag_queries_today": get_daily_queries()
        })
        
        return {"success": True, "indexed": len(indexed)}
        
    except Exception as e:
        reporter.report_error(f"Indexing: {e}")
        raise self.retry(exc=e)

@shared_task(name='knowledge.tasks.sync_external_sources')
def sync_external_sources():
    """Синхронизация внешних источников (Confluence, Google Docs)."""
    
    reporter.report_working("Синхронизация внешних источников")
    
    try:
        # ... логика синхронизации ...
        new_docs = fetch_external_sources()
        
        if new_docs:
            index_documents.delay([d.id for d in new_docs])
        
        reporter.report_idle(metrics={
            "docs_indexed": get_total_docs(),
            "rag_queries_today": get_daily_queries()
        })
        
        return {"success": True, "new_docs": len(new_docs)}
        
    except Exception as e:
        reporter.report_error(f"External sync: {e}")
        raise

# app/tasks/knowledge/office.py

from app.utils.office_reporter import OfficeReporter

OFFICE_REPORTER = OfficeReporter(
    agent_id="knowledge_rag",
    department="knowledge",
    name="RAG процессор",
    salary_equivalent=60000,
    fte_coefficient=1.0
)

@shared_task(name='knowledge.tasks.office_heartbeat')
def office_heartbeat():
    OFFICE_REPORTER.heartbeat()
    return {"success": True}