o fgK+@sLddlmZmZmZmZmZmZddlmZm Z ddl m Z ddl m Z ddlZddlZddlmZmZddlmZddlmZdd lmZdd lTddlmZdd lmZmZm Z m!Z!ddl"m#Z#eZ$eZ%e e&d Z'e(e&d dZ)e(e&ddZ*eZ+e$,dededededededee%fde(dee(de-de.dee-de(de fddZ/e$,dedededededee&dedededededededee%fde(dee(de-de.d ee(d!ee(d"ee-d#ee-dee-dee(d$ee(d%ee(d&ee(de fd'd(Z0d)d*Z1dS)+)Form APIRouterDepends HTTPExceptionstatusRequest) HTTPBearerHTTPAuthorizationCredentials) JSONResponse) load_dotenvN)ListOptional)OpenAI)text) HybridSearch)*)UserFolder Documents ChatHistoryCOLLECTION_NAMEDB_PATH/ STORAGE_PATHz/text.querydoc_id folder_id hybrid_weighttop_klimits credentialsc s8t}|j}t||} z|ttj| jktj |k } |r=|t t j |k } | s:t ddWS| j} | sEt ddWStt| j| j} | | }| j||||d}d}d}g}|rtt|dddd|dd f}|ddD],}t|d}|d d kr|dkr|d }|d d kr|dkr|d}tdd|}q~ddd|D}|D]?}|d dd|d <|d dd|d <|d dd|d <|d dd|d <|tt|ddd|d dqnd}d}tdd|iWSty}z tddt|dd}~ww)NDocument not found.Folder does not exist.alpharrmetadata page_numbercontenttypeimagepathtablehtml \\t(?![hd])  cs|]}|dVqdSr*N.0resultr5r5-/var/www/html/answerous_api/routers/search.py \zsearch_text..  z )page_numr* chunk_dataSomething went wrong:  status_codedetail) dbase SessionLocalr!authget_current_userrrfilteruser_idrfirstrr errorResponsesanitized_namercollection_name user_codeget_document_uuidsearchintjsonloadsresubjoinreplaceappendsuccessReponse Exceptionrstr)rrrrrr r!dbtokenuserfolderdocumentfilenamehs document_uuidsearch_resultsimg_pathr/r@citationr8r'passageer5r5r9 search_textsp ( * rkz/answerg? OPENAI_MODELr)llmprompt temperaturetop_pemailr'query_responsecst}| j}t||}|}d|krdkr nnd}d|kr*dkr.nnd}z|ttj |j ktj |k }|r| rit | } t| dd}|ttj|ktd|d }n |ttj|k }|s~tdd WS|j}nd }|std d WS| r|ttj| ktj|jk }|std d WSt||||j||||||||j | \}}}}}|\}}td |||d WSWdSty}z tddt|dd}~ww)Nrr)'"zJSON_CONTAINS(doc_metadata, 'z')r"r#r?r$zUser does not exist.) full_resultr>r*rArBrC)rFrGr!rHrIstriprrrJrKrrLrTrUr]rYrrrrMrNrrqorg_id answer_queryr[r\r)rrrrrmrnrorprr rqr'rrr!r^r_r`rajson_filter_strrbrc chat_useranswerrgr/rhrir>r*rjr5r5r9 search_answer|sh (  (. r}c  Cs@tt|j|} | |}| j||| | d}d}d}|rntt|dddd|ddf}|ddD],}t|d}|ddkrM|dkrM|d }|dd krb|dkrb|d }t d d |}q6d dd|D}nd}d}|svd}d|d|d|d|d }t d}d|dg}| rt | trt| } | r|| ztjjj||d}|jdjj}Wnty}z d|}WYd}~nd}~wwt}z)t| ||||tt|dddd||d}|||||Wnty}z|td|WYd}~nd}~ww|||||fS)Nr%rr'r(r)r*r+r,r-r.r/r0r1r2csr3r4r5r6r5r5r9r:r;zanswer_query..r?aU You are a bot-assistant that answers the above question. When answering this question, use the following rules: - always answer in English language; - use ONLY the information from the sources below; - answer briefly in just a few sentences, strictly in accordance with the sources, and do not make any assumptions; - reference the source if you use it in the answer, e.g. [#1] or [#2][#4]; - if there is no information on the question in the sources: say that you can't find the answer and ask the user to try to reformulate the question. zThe question is z?===z=== The sources are z And again, the question is z? rlr`)roler*)modelmessageszError occurred: )rKrrerrfrh query_promptrr) rrOrPrQrRrSrTrUrVrWrXosgetenv isinstancer]extend openai_clientchat completionscreatechoicesmessager*r\rFrGraddcommitrefreshrollbackprint)rpdf_namer` folder_namerrmrnrorprr chat_user_idquery_responsesrdrerfrgr/rhr8r'rir gpt_model message_dataresponse chat_responserjr^ chat_historyr5r5r9rys~ *     ry)2fastapirrrrrrfastapi.securityrr fastapi.responsesr dotenvr rrTtypingr r openair sqlalchemyrdependencies.HybridSearchrdependencies.helperdb_config.databasedatabaserFdb_config.modelsrrrrdb_config.authrHroutersecurityrrOr] DB_FOLDER UPLOAD_FOLDERrpostrSfloatrkr}ryr5r5r5r9s          `       H