
    PhD                     f    d dl Z d dlZd dlmZmZmZmZ d dlmZ d dl	m
Z
 d dlmZ  G d de      Zy)    N)AnyIteratorLiteralOptional)
BaseLoader)Document)get_from_envc                   x    e Zd ZdZ	 ddddddddee   dee   dee   dee   d	ed
   dee   fdZde	e
   fdZy)FireCrawlLoadera!
  
    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': []}

    Ncrawl)queryapi_keyapi_urlmodeparamsurlr   r   r   r   )r   scrapemapextractsearchr   c                   	 ddl m} |dvrt        d| d      |dv r|st        d      |d	k(  r |s|r|j	                  d
      st        d      |xs t        dd      }|xs t        j                  d      xs d} |||      | _         |xs d| _        || _	        |xs i | _
        ||| j                  d
<   yy# t        $ r t        d      w xY w)a  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`)r   r   r   r   r   r   Invalid mode 'zT'.
                Allowed: 'crawl', 'scrape', 'search', 'map', 'extract', 'search'.)r   r   r   r   z2Url must be provided for modes other than 'search'r   r   z&Query must be provided for search moder   FIRECRAWL_API_KEYFIRECRAWL_API_URLzhttps://api.firecrawl.dev)r   r    N)	firecrawlr   ImportError
ValueErrorgetr	   osgetenvr   r   r   )	selfr   r   r   r   r   r   r   resolved_api_urls	            h/var/www/html/saasai/venv/lib/python3.12/site-packages/langchain_community/document_loaders/firecrawl.py__init__zFireCrawlLoader.__init__F   s
   :	.
 PP"4& )R U 
 88QRR8Uv&**W:MEFFI\)5HI Tryy!45T9T 	 &g?OP9"	l#(DKK  3  	V 	s   B= =Creturnc           	   #   n  K   g }| j                   dk(  reh d}| j                  j                         D ci c]  \  }}||v s|| }}}d|d<    | j                  j                  | j
                  fi |g}n| j                   dk(  r| j
                  st        d      h d}| j                  j                         D ci c]  \  }}||v s|| }}}d|d<    | j                  j                  | j
                  fi |}t        |t              r1|j                  dg       }t        |t              rt        |      ng }n6t        |dg       }t        |t              rt        |      ng }n
| j                   d	k(  r| j
                  st        d
      h d}| j                  j                         D ci c]  \  }}||v s|| }}}d|d<    | j                  j                  | j
                  fi |}t        |t              r0|j                  d      }	t        |	t              rt        |	      ng }nAt        |d      r+t        |d      }	t        |	t              rt        |	      ng }n
t        |t              }
|
rt        |      ng }n| j                   dk(  r| j
                  st        d      h d}| j                  j                         D ci c]  \  }}||v s|| }}}d|d<   t         | j                  j                   | j
                  gfi |      g}nT| j                   dk(  r+h d}| j                  j                         D ci c]  \  }}||v s|| }}}d|d<    | j                  j"                  d,d| j                  j                  d      i|}t        |t              rt        |      }ng }g }t        |t              r3|j                  d      |j                  d      |j                  d      g}n&t        |dd       t        |dd       t        |dd       g}d|d   fd|d   fd|d   ffD ]  \  }}|s
|D ]  }t        |t              st        |dd       n|j                  d      }t        |t              st        |dd       n|j                  d      }t        |t              st        |dd       n|j                  d      }|xs
 |xs |xs d}||t        |t              st        |dd       n|j                  d      |dj                         D ci c]
  \  }}||| }}}|j%                  ||d         |}nt        d | j                    d!      |D ]  }| j                   d	k(  rHt        |t              r|}i }nPt        |t              r|j                  d      xs |j                  d"      xs d}t        |t              r|nt        |xs d      }|j                  d      |j                  d      d#j                         D ci c]
  \  }}||| }}}nt        |d      st        |d      rjt        |dd      xs t        |d"d      }t        |t              r|nt        |xs d      }i }t        |dd       }t        |dd       }|||d<   |4||d<   n-t        |      }i }n| j                   dk(  rt        |      }i }n | j                   dk(  r}t        |t              r^|j                  d$      xs d}t        |t              r|nt        |xs d      }|j                  d%i       }t        |t              r|ni }nt        |      }i }ntt        |t              rl|j                  d$      xs% |j                  d&      xs |j                  d'd      }t        |t              r|nt        |xs d      }|j                  d%i       }n]t        |d$d       xs t        |d&d       xs t        |d'd      }t        |t              r|nt        |xs d      }t        |d%i       xs i }t        |t              st        |d(      r&t'        |j(                        r|j)                         }nYt+        j,                  |      rt+        j.                  |      }n.t        |d)      rt        t1        |            }nd*t        |      i}|s~t3        ||+        y c c}}w c c}}w c c}}w c c}}w c c}}w c c}}w c c}}w w)-Nr   >   proxymobileactionsformatsheadersmax_ageparserstimeoutlocationuse_mockwait_for	block_ads	fast_modeexclude_tagsinclude_tagsstore_in_cacheonly_main_contentremove_base64_imagesskip_tls_verification	langchainintegrationr   zURL is required for crawl mode>   delaylimitpromptr0   webhookexclude_pathsinclude_pathspoll_intervalignore_sitemapscrape_optionsmax_concurrencyallow_subdomainscrawl_entire_domainmax_discovery_depthzero_data_retentionallow_external_linksignore_query_parametersdatar   zURL is required for map mode>   r?   r   sitemapr0   r1   include_subdomainslinksr   z URL is required for extract mode>   agentr@   schemar0   show_sourcesrD   system_promptrF   enable_web_searchignore_invalid_urlsrL   r   >   tbsr?   sourcesr0   r1   
categoriesrF   rW   r   webnewsimagesr         r   titledescriptionr   category)r   r`   rb   type)markdownmetadatar   zJ'.
                Allowed: 'crawl', 'scrape', 'map', 'extract', 'search'.href)r`   ra   rd   re   htmlrawHtml
model_dump__dict__value)page_contentre    )r   r   itemsr   r   r   r   r   
isinstancedictr    listgetattrr   hasattrstrr   r   appendcallableri   dataclassesis_dataclassasdictvarsr   )r#   firecrawl_docsallowedkvkwargscrawl_responserN   map_responserQ   is_listsearch_dataresults
containerskindrn   itemurl_val	title_valdesc_valcontent_valmetadata_valdocrl   metapage_content_valuer`   ra   markdown_valuemetadata_objcontent_values                                  r%   	lazy_loadzFireCrawlLoader.lazy_load   s	    $&99 G* (,{{'8'8':K':tq!a7lad':FK$/F=!3dnn33DHHGGHNYY'!88 !ABBG& (,{{'8'8':K':tq!a7lad':FK$/F=!1T^^11$((EfEN.$/%))&"5/9$/Ed2~vr:/9$/Ed2YY%88 !?@@G (,{{'8'8':K':tq!a7lad':FK$/F=!-4>>--dhhA&AL ,-$((10:5$0GeRw/g60:5$0GeR$\487>l!3BYY)#88 !CDDG (,{{'8'8':K':tq!a7lad':FK$/F=!!"8$.."8"8$(("Nv"NOPNYY("	G (,{{'8'8':K':tq!a7lad':FK$/F=!/$..// kkoog.28K +t,!%k!2 13
k40#.#/#1"J  UD9VT:Xt<"J JqM*Z]+z!}-$KD%
 !  % $.dD#9 $D%6!%%   $.dD#9 $D'48!%'!2 " $.dD#9 $D->!%-!8 !
 '/&L)&Lw&L" (/)2'1$'= -4D*d,K%)XXj%9(,) $eg)&()&1  !} qD)& % (  )4,O; !&$N ")"499+ .H K  "CyyE!c3'(+L+-DT*),)P3776?)Pb& &&8#> + !3!9r: ! &)WWW%5+.77=+A%  %'%"%"DAq = 1%"   S%(GC,A)0eR)@ *GVRE&
 &&8#> + !3!9r: !
 D#C$7E")#}d"CK((-W"..9]+#&s8LDi'"3xh&c4(%(WWZ%8%>BN &nc: ' !526 !
 $'77:r#:L+5lD+I<rD#&s8LDc4(
+XswwvX#'')UWBX "
 &mS9 & !4"5 !
 77:r2D  Z6 7"357"3	26 " &mS9 & !4"5 !
 #3
B7=2D "$-t\2x7P#0$11$7*11$7 z2#DJ/ 'T3) { " L2 L* L< L L`(Ds}   3d5ddA9d5 ddC$d56ddDd5d#d(A-d5d#"d#'G	d50d)
?C2d51d/ K5d5)N)__name__
__module____qualname____doc__r   rt   r   rp   r&   r   r   r   rm       r%   r   r   
   s    7z "9)  $!%!%GN!%9)c]9) }	9)
 #9) #9) CD9) 9)vX8H- Xr   r   )rw   r!   typingr   r   r   r   langchain_core.document_loadersr   langchain_core.documentsr   langchain_core.utilsr	   r   rm   r   r%   <module>r      s(     	 3 3 6 - -Oj Or   