This commit is contained in:
rafaelmmiller 2025-04-17 23:37:45 -07:00
commit 23ef26655c

View File

@ -16,12 +16,20 @@ from typing import Any, Dict, Optional, List, Union, Callable, Literal, TypeVar,
import json
from datetime import datetime
import re
import warnings
import requests
import pydantic
import websockets
import aiohttp
import asyncio
from pydantic import Field
# Suppress Pydantic warnings about attribute shadowing
warnings.filterwarnings("ignore", message="Field name \"json\" in \"FirecrawlDocument\" shadows an attribute in parent \"BaseModel\"")
warnings.filterwarnings("ignore", message="Field name \"json\" in \"ChangeTrackingData\" shadows an attribute in parent \"BaseModel\"")
warnings.filterwarnings("ignore", message="Field name \"schema\" in \"ExtractConfig\" shadows an attribute in parent \"BaseModel\"")
warnings.filterwarnings("ignore", message="Field name \"schema\" in \"ExtractParams\" shadows an attribute in parent \"BaseModel\"")
def get_version():
try:
@ -380,8 +388,7 @@ class ChangeTrackingData(pydantic.BaseModel):
diff: Optional[Dict[str, Any]] = None
json: Optional[Any] = None
class FirecrawlApp:
class SearchResponse(pydantic.BaseModel):
class SearchResponse(pydantic.BaseModel):
"""
Response from the search operation.
"""
@ -390,7 +397,7 @@ class FirecrawlApp:
warning: Optional[str] = None
error: Optional[str] = None
class ExtractParams(pydantic.BaseModel):
class ExtractParams(pydantic.BaseModel):
"""
Parameters for the extract operation.
"""
@ -404,10 +411,7 @@ class FirecrawlApp:
show_sources: Optional[bool] = False
agent: Optional[Dict[str, Any]] = None
class ExtractResponse(pydantic.BaseModel):
class ExtractResponse(pydantic.BaseModel):
"""
Response from the extract operation.
"""
@ -415,6 +419,7 @@ class FirecrawlApp:
data: Optional[Any] = None
error: Optional[str] = None
class FirecrawlApp:
def __init__(self, api_key: Optional[str] = None, api_url: Optional[str] = None) -> None:
"""
Initialize the FirecrawlApp instance with API key, API URL.