o pDi@sddlmZmZddlmZddlmZmZddlm Z ddl m Z edZ dZ ee dZe je ed d Zed d d ZdZeddZdZdZe dZdedefddZdS)) defaultdictdeque) load_dotenv)OpenAIEmbeddings ChatOpenAI)FAISS)ChatPromptTemplatezvector_store/tracwater_faissztext-embedding-3-large)modelT)allow_dangerous_deserializationz gpt-4.1-minig?)r temperaturecCs ttdS)N)maxlen)r MAX_TURNSrr(/home/anubhav/Desktop/Tracwater/query.py&s ra You are a domain-specific expert assistant representing TracWater. You must answer all user questions using ONLY the information provided in the context below and speak strictly in the first person plural (e.g., "we", "our", "us") when referring to TracWater, its products, or services. Conversation History: {history} Context: {context} Mandatory Rules: - ALWAYS refer to TracWater using first-person language ("we", "our", "us"). - NEVER use third-person references such as "they", "the company", or "TracWater provides". - Do NOT use any external knowledge, assumptions, or hallucinations. - Base your response strictly on the provided context. - If the required information is not present in the context, respond EXACTLY with: "Sorry, no information for this query is available in TracWater sources. Kindly rephrase or ask another question." - Keep responses clear, professional, and concise. User Question: {question} q session_idc Cstj|td}dd|D}|s|ddSdd|D}ddd |D}d d d t|D}ttj|p9d ||d }|j }t| ||d||dS)zC Perform session-aware RAG query with relevance threshold. )kcSs g|] \}}|tkr||fqSr)SIMILARITY_THRESHOLD).0docscorerrr os z query_search..z2Information not available in the provided sources.)queryanswercSsg|]\}}|qSrr)rr_rrrr}sz css(|]}d|jdd|jVqdS)z [Source: source_urlz] N)metadata page_content)rdrrr   zquery_search.. css(|]}d|dd|dVqdS)zUser: rz Assistant: aNr)rturnrrrr#r$None)historycontextquestion)rr&) vectordbsimilarity_search_with_scoreTOP_KjoinSESSION_MEMORYllminvokePROMPTformatcontentappend) rrresultsfiltereddocsr*r)responserrrr query_searchgs<   r;N) collectionsrrdotenvrlangchain_openairr langchain_community.vectorstoresrlangchain_core.promptsr VECTOR_DIREMBEDDING_MODEL embeddings load_localr,r1rr0rr. from_templater3strr;rrrrs0