7|hddlmZmZmZmZmZmZddlmZddl m Z deee efde fdZ Gdde ZGd d e Zy ) )AnyCallableIteratorListOptionalTuple)Document) BaseLoaderdocsreturncPdj|Dcgc]}|d c}Scc}w)z#Default joiner for content columns. )join)r docs m/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/document_loaders/rocksetdb.pydefault_joinerrs# 99-c!f- ..-s #c,eZdZdZdedeffd ZxZS)ColumnNotFoundErrorzColumn not found error. missing_keyqueryc0t|d|d|y)NzColumn "z" not selected in query: )super__init__)selfrr __class__s rrzColumnNotFoundError.__init__s 8K=0J5'RS)__name__ __module__ __qualname____doc__strr __classcell__)rs@rrr s"!TCTTTrrcreZdZdZdefdededeedeeede ee eefgeff dZ d e e fd Zy) RocksetLoaderaLoad from a `Rockset` database. To use, you should have the `rockset` python package installed. Example: .. code-block:: python # This code will load 3 records from the "langchain_demo" # collection as Documents, with the `text` column used as # the content from langchain_community.document_loaders import RocksetLoader from rockset import RocksetClient, Regions, models loader = RocksetLoader( RocksetClient(Regions.usw2a1, ""), models.QueryRequestSql( query="select * from langchain_demo limit 3" ), ["text"] ) ) Nclientr content_keys metadata_keyscontent_columns_joinerc ddlm}m}ddlm}t ||stdt|t ||stdt|||_ ||_ ||_ ||_ ||_ ||_||_ |jj!dy#t $r t dwxYw#t"$rYywxYw) aKInitialize with Rockset client. Args: client: Rockset client object. query: Rockset query object. content_keys: The collection columns to be written into the `page_content` of the Documents. metadata_keys: The collection columns to be written into the `metadata` of the Documents. By default, this is all the keys in the document. content_columns_joiner: Method that joins content_keys and its values into a string. It's method that takes in a List[Tuple[str, Any]]], representing a list of tuples of (column name, column value). By default, this is a method that joins each column value with a new line. This method is only relevant if there are multiple content_keys. r)QueryPaginator RocksetClient)QueryRequestSqlz]Could not import rockset client python package. Please install it with `pip install rockset`.z;client should be an instance of rockset.RocksetClient, got zBquery should be an instance of rockset.model.QueryRequestSql, got langchainN)rocksetr+r,rockset.modelsr- ImportError isinstance ValueErrortyper&rr'r)r( paginator request_modelset_applicationAttributeError) rr&rr'r(r)r+r,r-s rrzRocksetLoader.__init__-s.  = 6&-0F|n&  %1E{m%    (&<#*',  KK ' ' 47 @  8   sB$B<$B9< CCr c #K|jjj|jj}|D]h} t |j |j Dcgc] }|||f c}|j|jDcic]}||| c}n|jycc}wcc}w#t$r(}t|jd|jd}~wwxYww)N)sql) page_contentmetadatar) r&Queriesrresultsr r)r'r(KeyErrorrargs)r query_resultsrcoles r lazy_loadzRocksetLoader.lazy_loadgs ++11 2 ' ! C !%!#C!!C&&C))rrr r!rrrr"rrrrrr rDrrr%r%s}:.2IW 8883i 8  S * 8 !)$uS#X*?)@#)E F 8t8H-rr%N)typingrrrrrrlangchain_core.documentsr )langchain_community.document_loaders.baser r"r Exceptionrr%rErrrJsMAA-@/eCHo./3/ T)TfJfr