jCdZddlZddlZddlmZddlmZmZmZm Z m Z m Z ddl Z ddl Z ddlmZddlmZddlmZddlmZdd lmZdd lmZmZmZmZmZmZm Z dd l!m"Z"m#Z#dd l$m%Z%eZ& d&d e e e fdZ' d&de efdZ(e# d'de e)de e e)de ede ede ee)e)fde ee)efde ee)efde ee)efde e e*e jVfde e)defdZ,e# d'de e)de e e)de ede ede ee)e)fde ee)efde ee)efde ee)efde e e*e jVfde e)de eeeeefffdZ-e# d(de)de e)e e)fd e ed!e e.d"e ed#e e/de ee)efde ee)efde ee)efde e e*e jVfde e)defd$Z0e# d(de)de e)e e)fd e ed!e e.d"e ed#e e/de ee)efde ee)efde ee)efde e e*e jVfde e)de eeeeefffd%Z1y))z@ LiteLLM SDK Functions for Creating and Searching Vector Stores N)partial)Any CoroutineDictListOptionalUnion)request_timeout)get_llm_provider)Logging)BaseLLMHTTPHandler)GenericLiteLLMParams)&VectorStoreCreateOptionalRequestParamsVectorStoreCreateResponseVectorStoreFileCountsVectorStoreResultContent&VectorStoreSearchOptionalRequestParamsVectorStoreSearchResponseVectorStoreSearchResult)ProviderConfigManagerclient)VectorStoreRequestUtils mock_resultscV|tdtddgg}tdd|S) z%Mock response for vector store searchgffffff?z5This is a sample search result from the vector store.text)rtype)scorecontentz vector_store.search_results.pagez sample query)object search_querydata)rrrrs s/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/litellm/vector_stores/main.py!mock_vector_store_search_responser$$sG #,T#    %1#   mock_responsecR|$tdddddtdddddddddd }|S) z%Mock response for vector store createN vs_mock123 vector_storeiEezMock Vector Storer) in_progress completedfailed cancelledtotalr+) idr created_atnamebytes file_countsstatus expires_after expires_atlast_active_atmetadata)rrr&s r#!mock_vector_store_create_responser:<sS1!!$- # ( r%r1file_idsr5chunking_strategyr8 extra_headers extra_query extra_bodytimeoutcustom_llm_providerreturnc Kt} tj} d| d<| d} ttf|||||||||| d | } t j }t|j| }| jd|d{}tj|r |d{}|S|}|S7)7 #t$r}tjd| || | d}~wwxYww)z' Async: Create a vector store. TacreateNopenai) r1r;r5r<r8r=r>r?r@rAmodelrAoriginal_exceptioncompletion_kwargs extra_kwargs) localsasyncioget_event_looprcreate contextvars copy_contextrunrun_in_executor iscoroutine Exceptionlitellmexception_type)r1r;r5r<r8r=r>r?r@rAkwargs local_varsloopfuncctxfunc_with_context init_responseresponsees r#rDrDXs(J( %%' y  &"*    '/'#! 3    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY CA7B3B/B3$B1%B3*C+B3.C/B31B33 C<CCCc  t} | jd} | jdd} | jdddu}tdi| }|jr9t |jt rttdi|jS| d} t| dd \}} }}tjtj| | }|td | | j| t!j"| }| j%dd |i|d| i| t&j)||| || |||xst*|| jd }|S#t,$r}tj.d| || | d}~wwxYw)a Create a vector store. Args: name: The name of the vector store. file_ids: A list of File IDs that the vector store should use. expires_after: The expiration policy for the vector store. chunking_strategy: The chunking strategy used to chunk the file(s). metadata: Set of 16 key-value pairs that can be attached to an object. Returns: VectorStoreCreateResponse containing the created vector store details. litellm_logging_objlitellm_call_idNrDFTr9rErGrAlitellm_paramsproviderapi_typez)Vector store create is not supported for r1rGoptional_paramsrdrAr) #vector_store_create_optional_paramsvector_store_provider_configrArd logging_objr=r?r@ _is_asyncrrF)rKgetpoprr& isinstancedictr:rr r!get_provider_vector_stores_configrU LlmProviders ValueErrorupdater0get_requested_vector_store_create_optional_paramupdate_environment_variablesbase_llm_http_handlervector_store_create_handlerr rTrV)r1r;r5r<r8r=r>r?r@rArWrXrarbrmrdrg_rkrjr^r_s r#rNrNs<JR 17% $/ +%q! " C C --.AB!  % ( /;r?r@rArF) rKrLrMrsearchrOrPrQrRrSrTrUrV)r|r}r~rrrr=r>r?r@rArWrXrYrZr[r\r]r^r_s r#rr s*J) %%' y  &"*   +++''#! 3   &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY CA8B4B0B4%B2&B4+C,B4/C0B42B44 C=CCCc  Vt} | jd} | jdd}| jdddu}| jd|}tj3|1 tjj |}| j |tdd|i| }|jrTt|jttfr4d}t|jtr |j}t|S| d } d | vrt| dd \}} }}nd}| } tj tj"| | }|t%d | | j | t'j(| |}| j+|||d|||d|j-d| t.j1||||| || ||| xst2|| jd }|S#t$rYdwxYw#t$r}tj4d| || | d}~wwxYw)a[ Search a vector store for relevant chunks based on a query and file attributes filter. Args: vector_store_id: The ID of the vector store to search. query: A query string or array for the search. filters: Optional filter to apply based on file attributes. max_num_results: Maximum number of results to return (1-50, default 10). ranking_options: Optional ranking options for search. rewrite_query: Whether to rewrite the natural language query for vector search. Returns: VectorStoreSearchResponse containing the search results. rarbNrFTr|r"rE/rcrez)Vector store search is not supported for )rk)r|r})rbr|) exclude_nonerhr) r|r}#vector_store_search_optional_paramsrkrArdrlr=r?r@rmrrFrn)rKrorprUvector_store_registry get_credentials_for_vector_storervrTrr&rqstrlistr$r rrsrtrur0get_requested_vector_store_search_optional_paramrx model_dumpryvector_store_search_handlerr rV)r|r}r~rrrr=r>r?r@rArWrXrarbrmvector_store_id_for_credentialsregistry_credentialsrdrrgr{rkrr^r_s r#rrOs@Jm 17 !4 9 )DD+0S)E 3)+'!.::h'E  g  h  $$ 3 (    sDAH(0G0A,HC*H0 G=9H<G==H H( H##H()N) NNNNNNNNNN) NNNNNNNNN)2__doc__rLrO functoolsrtypingrrrrrr httpxrUlitellm.constantsr 1litellm.litellm_core_utils.get_llm_provider_logicr *litellm.litellm_core_utils.litellm_loggingr LiteLLMLoggingObj*litellm.llms.custom_httpx.llm_http_handlerr litellm.types.routerrlitellm.types.vector_storesrrrrrrr litellm.utilsrrlitellm.vector_stores.utilsrryr$r:rfloatTimeoutrDrNintboolrrrnr%r#rs>> -NSI58?+, =A4 7892:>568$($((,)-/3,0+/59)-< 3-< tCy!< D><  ~ < tCH~& < DcN+< $sCx.)< c3h(< eE5==01 2< "#< < < ~$($((,)-/3,0+/59)-p 3-p tCy!p D>p  ~ p tCH~& p DcN+p $sCx.)p c3h(p eE5==01 2p "#p  $iS:S0S&T TUp p f#%)&*$(/3,0+/59)-? ? d3i ? d^? c] ? d^ ? D> ? DcN+? $sCx.)? c3h(? eE5==01 2? "#? !? ? D#%)&*$(/3,0+/59)-M M d3i M d^M c] M d^ M D> M DcN+M $sCx.)M c3h(M eE5==01 2M "#M  $iS:S0S&T TU!M M r%