
    hh                     F    d dl mZ d dlmZ d dlmZ  G d de          ZdS )    )Iterator)Document)
BaseLoaderc                   >    e Zd ZdZdededefdZdee         fdZ	dS )	OpenCityDataLoaderzLoad from `Open City`.city_id
dataset_idlimitc                 0    || _         || _        || _        dS )as  Initialize with dataset_id.
        Example: https://dev.socrata.com/foundry/data.sfgov.org/vw6y-z8j6
        e.g., city_id = data.sfgov.org
        e.g., dataset_id = vw6y-z8j6

        Args:
            city_id: The Open City city identifier.
            dataset_id: The Open City dataset identifier.
            limit: The maximum number of documents to load.
        N)r   r	   r
   )selfr   r	   r
   s       p/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/langchain_community/document_loaders/open_city_data.py__init__zOpenCityDataLoader.__init__   s     $


    returnc              #      K   ddl m}  || j        d          }|                    | j        | j                  }|D ]4}t          t          |          d| j        dz   | j        z   i          V  5dS )zLazy load records.r   )SocrataN)r
   source_)page_contentmetadata)sodapyr   r   getr	   r
   r   str)r   r   clientresultsrecords        r   	lazy_loadzOpenCityDataLoader.lazy_load   s       	#"""""t,,**T_DJ*?? 	 	F [[dlS04?B      	 	r   N)
__name__
__module____qualname____doc__r   intr   r   r   r    r   r   r   r      sc            S    8H-      r   r   N)typingr   langchain_core.documentsr   )langchain_community.document_loaders.baser   r   r#   r   r   <module>r'      so          - - - - - - @ @ @ @ @ @         r   