o Zh0&@sddlmZmZmZmZmZmZddlmZddl Z ddl Z ddl m Z ddlmZddlmZddlmZddlmZGd d d eZdS) )AnyIterableListOptionalTuplecast)uuid4NDocument) Embeddings) get_from_env) VectorStore)DistanceStrategyc@seZdZUdZdZeed<deZej dfdede de d ed ef d d Z e d efddZd efddZd efddZd efddZ  d1deedeeede ded eef ddZe d e fddZd2d eeeded eefd!d"Zd3deed$e d eefd%d&Z #d3d'ed$e ded eefd(d)Z #d3d'ed$e ded eeeeffd*d+Z #d3deed$e ded eefd,d-Z!e"ddd.dej fdeede deeedede d ed eded dfd/d0Z#dS)4SemaDBa`SemaDB` vector store. This vector store is a wrapper around the SemaDB database. Example: .. code-block:: python from langchain_community.vectorstores import SemaDB db = SemaDB('mycollection', 768, embeddings, DistanceStrategy.COSINE) zsemadb.p.rapidapi.comHOSTzhttps://collection_name vector_size embeddingdistance_strategyapi_keycCs,||_||_|p tdd|_||_||_dS)z#initialize the SemaDB vector store.rZSEMADB_API_KEYN)rrr r _embeddingr)selfrrrrrr^/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/vectorstores/semadb.py__init__s   zSemaDB.__init__returncCsd|jtjdS)zReturn the common headers.zapplication/json)z content-typezX-RapidAPI-KeyzX-RapidAPI-Host)rrrrrrrheaders.szSemaDB.headerscCsh|jtjkrdS|jtjkrtd|jtjkrdS|jtjkr$td|jtjkr,dStd|j)z&Return the internal distance strategy.Z euclideanz,Max inner product is not supported by SemaDBdotZcosinezUnknown distance strategy )rrEUCLIDEAN_DISTANCEZMAX_INNER_PRODUCT ValueErrorZ DOT_PRODUCTZJACCARDCOSINErrrr_get_internal_distance_strategy7s     z&SemaDB._get_internal_distance_strategycCs6|j|j|d}tjtjd||jd}|jdkS)z/Creates the corresponding collection in SemaDB.)idZ vectorSizeZdistanceMetricz /collectionsjsonr) rrr#requestspostrBASE_URLr status_code)rpayloadresponserrrcreate_collectionFs zSemaDB.create_collectioncCs(tjtjd|j|jd}|jdkS)z/Deletes the corresponding collection in SemaDB. /collections/)rr')r(deleterr*rrr+)rr-rrrdelete_collectionTs  zSemaDB.delete_collectionNtexts metadatas batch_sizekwargscKst|ts t|}|j|}t|d|jkr'tdt|dd|j|jtj krHt |}|t j j |ddd}tttt|}g}g}|durvt|||D]\} } } tt} || || | i| d| id qVnt||D]\} } tt} || || | d| id q{tdt||D]B} || | |}tjtjd |jd d |i|jd }|jdkrtd|td|j| d}t|dkrtd|q|S)zAdd texts to the vector store.rzEmbedding size mismatch z != T)ZaxisZkeepdimsNtext)r$vectormetadatar//pointspointsr%r'zHERE--zError adding points: Z failedRanges)! isinstancelistrZembed_documentslenrr!rrr"nparraylinalgnormrrfloattolistzipstrrappendranger(r)rr*rrr+printr8r&)rr3r4r5r6 embeddingsZ embed_matrixidsr<r8rr:Znew_idibatchr-Z failed_rangesrrr add_texts\sd            zSemaDB.add_textscCs|jS)zReturn the embeddings.)rrrrrrKszSemaDB.embeddingsrLcKsHd|i}tjtjd|jd||jd}|jdko#t|ddkS)a3Delete by vector ID or other criteria. Args: ids: List of ids to delete. **kwargs: Other keyword arguments that subclasses might use. Returns: Optional[bool]: True if deletion is successful, False otherwise, None if not implemented. rLr/r;r%r'Z failedPointsr) r(r0rr*rrr+r?r&)rrLr6r,r-rrrr0s z SemaDB.deletekcCs|jtjkrt|}|tj|}ttt | }||d}t j t jd|jd||jd}|jdkr?td|j|dS)zSearch points.)r9limitr/z/points/searchr%r'zError searching: r<)rrr"r@rArBrCrrrDrEr(r)rr*rrr+r!r8r&)rrrQZvecr,r-rrr_search_pointss    zSemaDB._search_pointsquerycKs|j|}|j||dS)z"Return docs most similar to query.rQ)r embed_querysimilarity_search_by_vector)rrTrQr6query_embeddingrrrsimilarity_searchs zSemaDB.similarity_searchcKs(|j|}|j||d}dd|DS)z$Run similarity search with distance.rUcSs,g|]}t|dd|dd|dfqS)r:r8Z page_contentr:Zdistancer .0prrr s z7SemaDB.similarity_search_with_score..)rrVrS)rrTrQr6rXr<rrrsimilarity_search_with_scores z#SemaDB.similarity_search_with_scorecKs|j||d}dd|DS)aReturn docs most similar to embedding vector. Args: embedding: Embedding to look up documents similar to. k: Number of Documents to return. Defaults to 4. Returns: List of Documents most similar to the query vector. rUcSs$g|]}t|dd|ddqS)r:r8rZr r[rrrr^sz6SemaDB.similarity_search_by_vector..)rS)rrrQr6r<rrrrWs z"SemaDB.similarity_search_by_vectorrc KsX|std|s td|std||||||d} | s#td| j||d| S)z9Return VectorStore initialized from texts and embeddings.z Collection name must be providedzVector size must be providedzAPI key must be provided)rrzError creating collection)r4)r!r.rO) clsr3rr4rrrrr6Zsemadbrrr from_textss" zSemaDB.from_texts)Nr2)N)rP)$__name__ __module__ __qualname____doc__rrG__annotations__r*rr intr rpropertydictrr#boolr.r1rrrrrOrKr0rDrSr rYrr_rW classmethodrarrrrrs      A$       r)typingrrrrrruuidrnumpyr@r(Zlangchain_core.documentsr Zlangchain_core.embeddingsr Zlangchain_core.utilsr Zlangchain_core.vectorstoresr Z&langchain_community.vectorstores.utilsrrrrrrs