o
    suBi                     @   s>  d dl Z d dlmZ e dZdZdZdZe de d Z	e j
e j
e j
eZe j
ed	d
Ze j
edZe j
edZe j
edZe j
eddZeeeefD ]	Z
e je
dd qZdZdZdZdZdZdZdZdZg dZddddZg dZ e j
edZ!e j
ede	 d Z"d!e#d"e#fd#d$Z$dS )%    N)datetimeOPENAI_API_KEYztext-embedding-3-smallzgpt-4o-miniz	zydus-poc-z%Y%m%d-%H%M%Sz	Zydus-POCAnnualReportsInputOutput	ArtifactsindexesfaissT)exist_oki      (      <      )zRegulatory & CompliancezQuality / Audit / InspectionzOperational / ManufacturingzSupply ChainzESG / SustainabilityzFinancial RiskszLegal / LitigationzCyber / Data / Privacy         )HighMediumLow)znon-compliance
regulatoryzaudit finding
inspectionzfindings.jsonrun_z.logcategoryreturnc                 C   s.   |   dddd}tjtd| dS )N _/retrieval_bundle_z.json)lowerreplaceospathjoinARTIFACTS_DIR)r   	safe_name r'   @/var/www/html/fyndo/python/python_agents/rag_suite/rag/config.pyretrieval_bundle_pathG   s
   
r)   )%r"   r   getenvr   EMBED_MODEL	LLM_MODEL	TENANT_IDnowstrftimeRUN_IDr#   dirnameabspath__file__BASE_DIRr$   POC_ROOT	INPUT_DIR
OUTPUT_DIRr%   	INDEX_DIRmakedirs
CHUNK_SIZECHUNK_OVERLAPENABLE_PAGE_NUMBERSFALLBACK_TO_CHUNK_ORDERTOP_K_PER_CATEGORY	MIN_TOP_K	MAX_TOP_KMAX_FINDINGS_PER_CATEGORY
CATEGORIESSEVERITY_SCORESKEYWORD_BOOSTSFINDINGS_FILELOG_FILEstrr)   r'   r'   r'   r(   <module>   s@    
