
    3fi                     >    d dl mZ d dlmZ d dlmZ  G d de      Zy)    )Iterator)Document)
BaseLoaderc                   6    e Zd ZdZdededefdZdee   fdZ	y)	OpenCityDataLoaderzLoad from `Open City`.city_id
dataset_idlimitc                 .    || _         || _        || _        y)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       q/var/www/auto_recruiter/arenv/lib/python3.12/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                  | j
                        }|D ]6  }t        t        |      d| j                  dz   | j                  z   i       8 yw)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   st      	#t,**T__DJJ*? 	F [dllS04??B 	s   A=A?N)
__name__
__module____qualname____doc__r   intr   r   r   r    r   r   r   r      s0       S 8H- r   r   N)typingr   langchain_core.documentsr   )langchain_community.document_loaders.baser   r   r#   r   r   <module>r'      s     - @ r   