
    dh}2                     Z    S SK r S SKJrJrJr  S SKJr  S SKJr  S SK	J
r
   " S S\5      rg)    N)IteratorLiteralOptional)
BaseLoader)Document)get_from_envc                       \ rS rSrSrS\S\4S jrS\S\4S jrSSSSS	.S
\S\	\   S\	\   S\
S   S\	\   4
S jjrS\\   4S jrSrg)FireCrawlLoader	   a	  
FireCrawlLoader document loader integration

Setup:
    Install ``firecrawl-py``,``langchain_community`` and set environment variable ``FIRECRAWL_API_KEY``.

    .. code-block:: bash

        pip install -U firecrawl-py langchain_community
        export FIRECRAWL_API_KEY="your-api-key"

Instantiate:
    .. code-block:: python

        from langchain_community.document_loaders import FireCrawlLoader

        loader = FireCrawlLoader(
            url = "https://firecrawl.dev",
            mode = "crawl"
            # other params = ...
        )

Lazy load:
    .. code-block:: python

        docs = []
        docs_lazy = loader.lazy_load()

        # async variant:
        # docs_lazy = await loader.alazy_load()

        for doc in docs_lazy:
            docs.append(doc)
        print(docs[0].page_content[:100])
        print(docs[0].metadata)

    .. code-block:: python

        Introducing [Smart Crawl!](https://www.firecrawl.dev/smart-crawl)
         Join the waitlist to turn any web
        {'ogUrl': 'https://www.firecrawl.dev/', 'title': 'Home - Firecrawl', 'robots': 'follow, index', 'ogImage': 'https://www.firecrawl.dev/og.png?123', 'ogTitle': 'Firecrawl', 'sitemap': {'lastmod': '2024-08-12T00:28:16.681Z', 'changefreq': 'weekly'}, 'keywords': 'Firecrawl,Markdown,Data,Mendable,Langchain', 'sourceURL': 'https://www.firecrawl.dev/', 'ogSiteName': 'Firecrawl', 'description': 'Firecrawl crawls and converts any website into clean markdown.', 'ogDescription': 'Turn any website into LLM-ready data.', 'pageStatusCode': 200, 'ogLocaleAlternate': []}

Async load:
    .. code-block:: python

        docs = await loader.aload()
        print(docs[0].page_content[:100])
        print(docs[0].metadata)

    .. code-block:: python

        Introducing [Smart Crawl!](https://www.firecrawl.dev/smart-crawl)
         Join the waitlist to turn any web
        {'ogUrl': 'https://www.firecrawl.dev/', 'title': 'Home - Firecrawl', 'robots': 'follow, index', 'ogImage': 'https://www.firecrawl.dev/og.png?123', 'ogTitle': 'Firecrawl', 'sitemap': {'lastmod': '2024-08-12T00:28:16.681Z', 'changefreq': 'weekly'}, 'keywords': 'Firecrawl,Markdown,Data,Mendable,Langchain', 'sourceURL': 'https://www.firecrawl.dev/', 'ogSiteName': 'Firecrawl', 'description': 'Firecrawl crawls and converts any website into clean markdown.', 'ogDescription': 'Turn any website into LLM-ready data.', 'pageStatusCode': 200, 'ogLocaleAlternate': []}

paramsreturnc                    Sn/ SQnU H  nUR                  U5      (       d  M  Sn  O   U(       a  [        R                  " S[        5        SU;   a  US   SL a  US   US'   US	 SU;   a  US   SL a  US   US'   US	 S	U;   a  US	   SL a  US	   US
'   US		 SU;   a  US   SL a  US   US'   US	 SU;   a2  [	        US   [
        5      (       a  U R                  US   5      US'   US	 U$ )NF)includesexcludesallowBackwardCrawlingallowExternalContentLinkspageOptionsTBDeprecated parameters detected. See Firecrawl v1 docs for updates.r   includePathsr   excludePathsr   allowBackwardLinksr   allowExternalLinksr   scrapeOptions)getwarningswarnDeprecationWarning
isinstancedictlegacy_scrape_options_adapter)selfr   use_legacy_optionslegacy_keyskeys        f/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/document_loaders/firecrawl.pylegacy_crawler_options_adapter.FireCrawlLoader.legacy_crawler_options_adapterC   sH   "
 Czz#%)" 
 MMT" V#*%--3J-?F>*:&V#*%--3J-?F>*:&&&012d:39:Q3RF/023*f456$>39:U3VF/067&f]3T::.2.P.P}-/F?+ =)    c                 `   SnS/nSU;   a  SUS   ;   a  US   S   S:X  d  US   S   S:X  d  US   S   S:X  a  SnS	US   ;   a1  US   S	   (       a  US   S	   US
'   OUS   R                  S	S5      US
'   SUS   ;   a  US   S   (       a  US   S   US'   SUS   ;   a  US   S   (       a  US   S   US
'   US	 / SQnU H  nUR                  U5      (       d  M  Sn  O   U(       Ga;  [        R                  " S[        5        SU;   a  US   SL a  UR	                  S5        US	 SU;   a  US   SL a  UR                  S5        US	 SU;   a  US   SL a  UR                  S5        US	 SU;   a  US   SL a  UR                  S5        US	 SU;   a  US   SL a  UR                  S5        US	 SU;   a  US   SL a  UR                  S5        US	 SU;   a  US   SL a  UR                  S5        US	 SU;   a  US   SL a  US   US'   US	 SU;   a  US   SL a  US   US '   US	 S!U;  a  X1S!'   U$ )"NFmarkdownextractorOptionsmodezllm-extractionzllm-extraction-from-raw-htmlzllm-extraction-from-markdownTextractionPromptpromptz-Extract page information based on the schema.extractionSchemaschema
userPrompt)	includeMarkdownincludeHtmlincludeRawHtmlincludeExtractincludeLinks
screenshotfullPageScreenshotonlyIncludeTags
removeTagsr   r2   r3   htmlr4   rawHtmlr5   extractr6   linksr7   r8   zscreenshot@fullPager9   includeTagsr:   excludeTagsformats)r   r   r   r   removeappend)r!   r   r"   rA   scrape_keysr$   s         r%   r    -FireCrawlLoader.legacy_scrape_options_adapters   s   ",' 233-.v6:JJ01&95601&956 *.&)V4F-GG!"456HI/56H/I 20F8, 066H/I/M/M 2 O0F8,
 *V4F-GG!"456HI/56H/I 20F8, $v.@'AA!"45lC/56H/I,/WF8,12

 Czz#%)" 
 MMT" !F*+,5NN:.,-&-(D0NN6*=)6)*+t3NN9-+,6)*+t3NN9-+,'.)T1NN7+>*v%,'4/NN<0<(#v-./47NN#89/0 F*+,4,23D,EF=),-v%,'4/,2<,@F=)<(F" '9r(   Ncrawl)api_keyapi_urlr,   r   urlrG   rH   r,   )rF   scrapemapr=   searchc                
    SSK Jn  US;  a  [        SU S35      eU(       d  [        S5      eU=(       d    [	        SS	5      nU" X#S
9U l         Xl        X@l        U=(       d    0 U l        g! [         a    [        S5      ef = f)aQ  Initialize with API key and url.

Args:
    url: The url to be crawled.
    api_key: The Firecrawl API key. If not specified will be read from env var
        FIRECRAWL_API_KEY. Get an API key
    api_url: The Firecrawl API URL. If not specified will be read from env var
        FIRECRAWL_API_URL or defaults to https://api.firecrawl.dev.
    mode: The mode to run the loader in. Default is "crawl".
         Options include "scrape" (single url),
         "crawl" (all accessible sub pages),
         "map" (returns list of links that are semantically related).
         "extract" (extracts structured data from a page).
         "search" (search for data across the web).
    params: The parameters to pass to the Firecrawl API.
        Examples include crawlerOptions.
        For more details, visit: https://github.com/mendableai/firecrawl-py
r   )FirecrawlAppzD`firecrawl` package not found, please run `pip install firecrawl-py`)rF   rJ   rL   rK   r=   rL   Invalid mode 'zT'.
                Allowed: 'crawl', 'scrape', 'search', 'map', 'extract', 'search'.zUrl must be providedrG   FIRECRAWL_API_KEY)rG   rH   N)	firecrawlrN   ImportError
ValueErrorr   rI   r,   r   )r!   rI   rG   rH   r,   r   rN   s          r%   __init__FireCrawlLoader.__init__   s    8	.
 PP"4& )R U 
 344I\)5HI%gG	l#  	V 	s   A, ,Bc              #   $  #    U R                   S:X  a@  U R                  R                  U R                  U R	                  U R
                  5      S9/nGOU R                   S:X  am  U R                  (       d  [        S5      eU R                  R                  U R                  U R                  U R
                  5      S9nUR                  S/ 5      nGO(U R                   S:X  aK  U R                  (       d  [        S5      eU R                  R                  U R                  U R
                  S9nOU R                   S:X  aV  U R                  (       d  [        S	5      e[        U R                  R                  U R                  /U R
                  S95      /nOgU R                   S
:X  a>  U R                  R                  U R
                  R                  S5      U R
                  S9nO[        SU R                    S35      eU H  nU R                   S:X  d  U R                   S:X  a  Un0 nOTUR                  S5      =(       d*    UR                  S5      =(       d    UR                  SS5      nUR                  S0 5      nU(       d  M  [        UUS9v   M     g 7f)NrJ   )r   rF   zURL is required for crawl modedatarK   zURL is required for map moder=   z URL is required for extract moderL   query)rX   r   rO   zJ'.
                Allowed: 'crawl', 'scrape', 'map', 'extract', 'search'.r*   r;   r<    metadata)page_contentrZ   )r,   rQ   
scrape_urlrI   r    r   rS   	crawl_urlr&   r   map_urlstrr=   rL   r   )r!   firecrawl_docscrawl_responsedocr[   rZ   s         r%   	lazy_loadFireCrawlLoader.lazy_load  s&    99 ))HHT%G%G%T * N
 YY'!88 !ABB!^^55!D!DT[[!Q 6 N ,//;NYY%88 !?@@!^^33DHHT[[3QNYY)#88 !CDDDNN**DHH:dkk*JKN YY("!^^22kkoog.{{ 3 N
 "499+ .H K  "CyyE!TYY)%;" GGJ'T3776?TcggiQS>T  77:r2)!  "s   JJ)rQ   r,   r   rI   )__name__
__module____qualname____firstlineno____doc__r   r&   r    r_   r   r   rT   r   r   rc   __static_attributes__ r(   r%   r
   r
   	   s    7r.T .d .`jD jT j` "&!%GN!%/#/# #	/#
 #/# CD/# /#b08H- 0r(   r
   )r   typingr   r   r   langchain_core.document_loadersr   langchain_core.documentsr   langchain_core.utilsr   r
   rk   r(   r%   <module>rp      s%     . . 6 - -wj wr(   