o hi@srddlZddlmZddlmZddlmZddlm Z ddl m Z m Z ddl Z e eZd dd Zdd d ZdS)N)SentenceTransformer)settings)get_object_or_404)FileChatbotc Csztt||d}tjttjdd}td}d|j}d | }| |gd}g} t } zq|j |d} | j|g|d d gd } td | tt| d d| d d} | jd dd| d|D]8\}}|d}ztjj|d}| |jj|d|dWqhtjytd|d| |YqhwWn)ty}ztd|dt|ddt|iWYd}~WSd}~ww| dd| DdWSty}ztdt|dt|iWYd}~Sd}~ww)a Retrieve RAG entries from ChromaDB for a given chatbot and query. Args: chatbot_id (int): ID of the chatbot. user (User): Authenticated user making the request. query (str): User input query. n_results (int): Number of RAG results to retrieve (default: 3). Returns: dict: Contains 'results' (list of RAG entries), 'warnings' (list of warnings), and 'error' (if any). )iduser chroma_db)pathzall-MiniLM-L6-v2chatbot_ r)name metadatas distances)query_embeddings n_resultsincludezChroma query results: cSs|dS)Nr)xrr0/var/www/ai-chatbot-integration/chatbot/utils.py4sz&retrieve_rag_entries..)keyNfile_id)r text) file_namerdistance File with ID not foundzFailed to query collection z: errorzNo RAG data available: cSsg|]}d|dqS)rrr).0fidrrr Isz(retrieve_rag_entries..resultswarningszRAG retrieval failed: )rrchromadbPersistentClientstrr MEDIA_ROOTrr joinsplitencodesetget_collectionqueryloggerdebuglistzipsortgetrobjectsappendfiler DoesNotExistwarningadd Exception exception) chatbot_idr r0rchatbot chroma_clientembeddercollection_namequery_embedding rag_results missing_files collection query_results result_pairsmetadatarrr9errrretrieve_rag_entries s^        rLTc Cs&t||||}d|vr |S|d|dd}|rzSd}|d7}t|ddD]\}} ||d| d d | d d d | dd7}q&|d|7}tjtjdtd} | |} | j|d<t d|dW|St y} zt dt | ddt | iWYd} ~ Sd} ~ ww|S)aC Retrieve RAG entries and optionally generate an AI response. Args: chatbot_id (int): ID of the chatbot. user (User): Authenticated user making the request. message (str): User input message/query. n_results (int): Number of RAG results to retrieve (default: 3). include_ai (bool): Whether to query the AI model for a response (default: True). Returns: dict: Contains 'results' (RAG entries), 'response' (AI response if include_ai=True), 'warnings' (list of warnings), and 'error' (if any). r r%r&r$aGeneral guidelines: use the inclusive, inspiring, global local blend for tone, be conscise. ask to rewrite the question if it is not obvious in answer. if you detect that user message is in another language, answer in that language. zRelevant document excerpts: rz. From rz (Similarity: rz.4fz): rz zUser message: )api_keyzgemini-1.5-flashresponsez AI response: zFailed to query AI model: z Failed to get response from AI: N)rL enumerategenai configurerGEMINI_API_KEYGenerativeModelgenerate_contentrr1r2r=r>r)) r?r messager include_airag_data response_datacontextidxresultmodelrNrKrrrretrieve_ai_responsePs20   r])r)rT)r'sentence_transformersrgoogle.generativeai generativeairP django.confrdjango.shortcutsrmodelsrrlogging getLogger__name__r1rLr]rrrrs      E