o
    1i                     @   sJ   d Z ddlZddlZddlmZ ddlmZ dd Zedkr#e  dS dS )zCBackfill transcript data from {bid}_sarvamresponse into RAG tables.    N)Config)
RAGHandlerc                  C   s   t jdd} | jdddd | jdtdd	d
 | jdddd | jdddd |  }ttj}|j|j	|j
|j|jd}ttj|dd d S )Nz&Backfill transcripts into MCube AI RAG)descriptionz--bidTzBusiness ID)requiredhelpz--limiti  zMax calls to process)typedefaultr   z--presales-only
store_truezOnly ingest presales-like calls)actionr   z--overwrite-existingz%Recreate docs/chunks already ingested)bidpresales_onlylimitoverwrite_existing   )indent)argparseArgumentParseradd_argumentint
parse_argsr   r   __dict__backfill_transcriptsr   r   r   r   printjsondumps)parserargsragresult r   backfill_rag_transcripts.pymain
   s   
r!   __main__)	__doc__r   r   configr   rag_handlerr   r!   __name__r   r   r   r    <module>   s   
