mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-28 19:58:11 +08:00
Refactor: reformat all code for lefthook using ruff and gofmt (#16585)
This commit is contained in:
Wang Qi
@@ -12,31 +12,31 @@ from .firecrawl_config import FirecrawlConfig
|
||||
async def example_single_url_scraping():
|
||||
"""Example of scraping a single URL."""
|
||||
print("=== Single URL Scraping Example ===")
|
||||
|
||||
|
||||
# Configuration
|
||||
config = {
|
||||
"api_key": "fc-your-api-key-here", # Replace with your actual API key
|
||||
"api_url": "https://api.firecrawl.dev",
|
||||
"max_retries": 3,
|
||||
"timeout": 30,
|
||||
"rate_limit_delay": 1.0
|
||||
"rate_limit_delay": 1.0,
|
||||
}
|
||||
|
||||
|
||||
# Create integration
|
||||
integration = create_firecrawl_integration(config)
|
||||
|
||||
|
||||
# Test connection
|
||||
connection_test = await integration.test_connection()
|
||||
print(f"Connection test: {connection_test}")
|
||||
|
||||
|
||||
if not connection_test["success"]:
|
||||
print("Connection failed, please check your API key")
|
||||
return
|
||||
|
||||
|
||||
# Scrape a single URL
|
||||
urls = ["https://httpbin.org/json"]
|
||||
documents = await integration.scrape_and_import(urls)
|
||||
|
||||
|
||||
for doc in documents:
|
||||
print(f"Title: {doc.title}")
|
||||
print(f"URL: {doc.source_url}")
|
||||
@@ -49,37 +49,31 @@ async def example_single_url_scraping():
|
||||
async def example_website_crawling():
|
||||
"""Example of crawling an entire website."""
|
||||
print("=== Website Crawling Example ===")
|
||||
|
||||
|
||||
# Configuration
|
||||
config = {
|
||||
"api_key": "fc-your-api-key-here", # Replace with your actual API key
|
||||
"api_url": "https://api.firecrawl.dev",
|
||||
"max_retries": 3,
|
||||
"timeout": 30,
|
||||
"rate_limit_delay": 1.0
|
||||
"rate_limit_delay": 1.0,
|
||||
}
|
||||
|
||||
|
||||
# Create integration
|
||||
integration = create_firecrawl_integration(config)
|
||||
|
||||
|
||||
# Crawl a website
|
||||
start_url = "https://httpbin.org"
|
||||
documents = await integration.crawl_and_import(
|
||||
start_url=start_url,
|
||||
limit=5, # Limit to 5 pages for demo
|
||||
scrape_options={
|
||||
"formats": ["markdown", "html"],
|
||||
"extractOptions": {
|
||||
"extractMainContent": True,
|
||||
"excludeTags": ["nav", "footer", "header"]
|
||||
}
|
||||
}
|
||||
scrape_options={"formats": ["markdown", "html"], "extractOptions": {"extractMainContent": True, "excludeTags": ["nav", "footer", "header"]}},
|
||||
)
|
||||
|
||||
|
||||
print(f"Crawled {len(documents)} pages from {start_url}")
|
||||
|
||||
|
||||
for i, doc in enumerate(documents):
|
||||
print(f"Page {i+1}: {doc.title}")
|
||||
print(f"Page {i + 1}: {doc.title}")
|
||||
print(f"URL: {doc.source_url}")
|
||||
print(f"Content length: {len(doc.content)}")
|
||||
print("-" * 30)
|
||||
@@ -88,42 +82,31 @@ async def example_website_crawling():
|
||||
async def example_batch_processing():
|
||||
"""Example of batch processing multiple URLs."""
|
||||
print("=== Batch Processing Example ===")
|
||||
|
||||
|
||||
# Configuration
|
||||
config = {
|
||||
"api_key": "fc-your-api-key-here", # Replace with your actual API key
|
||||
"api_url": "https://api.firecrawl.dev",
|
||||
"max_retries": 3,
|
||||
"timeout": 30,
|
||||
"rate_limit_delay": 1.0
|
||||
"rate_limit_delay": 1.0,
|
||||
}
|
||||
|
||||
|
||||
# Create integration
|
||||
integration = create_firecrawl_integration(config)
|
||||
|
||||
|
||||
# Batch scrape multiple URLs
|
||||
urls = [
|
||||
"https://httpbin.org/json",
|
||||
"https://httpbin.org/html",
|
||||
"https://httpbin.org/xml"
|
||||
]
|
||||
|
||||
documents = await integration.scrape_and_import(
|
||||
urls=urls,
|
||||
formats=["markdown", "html"],
|
||||
extract_options={
|
||||
"extractMainContent": True,
|
||||
"excludeTags": ["nav", "footer", "header"]
|
||||
}
|
||||
)
|
||||
|
||||
urls = ["https://httpbin.org/json", "https://httpbin.org/html", "https://httpbin.org/xml"]
|
||||
|
||||
documents = await integration.scrape_and_import(urls=urls, formats=["markdown", "html"], extract_options={"extractMainContent": True, "excludeTags": ["nav", "footer", "header"]})
|
||||
|
||||
print(f"Processed {len(documents)} URLs")
|
||||
|
||||
|
||||
for doc in documents:
|
||||
print(f"Title: {doc.title}")
|
||||
print(f"URL: {doc.source_url}")
|
||||
print(f"Content length: {len(doc.content)}")
|
||||
|
||||
|
||||
# Example of chunking for RAG processing
|
||||
chunks = integration.processor.chunk_content(doc, chunk_size=500, chunk_overlap=100)
|
||||
print(f"Number of chunks: {len(chunks)}")
|
||||
@@ -133,38 +116,34 @@ async def example_batch_processing():
|
||||
async def example_content_processing():
|
||||
"""Example of content processing and chunking."""
|
||||
print("=== Content Processing Example ===")
|
||||
|
||||
|
||||
# Configuration
|
||||
config = {
|
||||
"api_key": "fc-your-api-key-here", # Replace with your actual API key
|
||||
"api_url": "https://api.firecrawl.dev",
|
||||
"max_retries": 3,
|
||||
"timeout": 30,
|
||||
"rate_limit_delay": 1.0
|
||||
"rate_limit_delay": 1.0,
|
||||
}
|
||||
|
||||
|
||||
# Create integration
|
||||
integration = create_firecrawl_integration(config)
|
||||
|
||||
|
||||
# Scrape content
|
||||
urls = ["https://httpbin.org/html"]
|
||||
documents = await integration.scrape_and_import(urls)
|
||||
|
||||
|
||||
for doc in documents:
|
||||
print(f"Original document: {doc.title}")
|
||||
print(f"Content length: {len(doc.content)}")
|
||||
|
||||
|
||||
# Chunk the content
|
||||
chunks = integration.processor.chunk_content(
|
||||
doc,
|
||||
chunk_size=1000,
|
||||
chunk_overlap=200
|
||||
)
|
||||
|
||||
chunks = integration.processor.chunk_content(doc, chunk_size=1000, chunk_overlap=200)
|
||||
|
||||
print(f"Number of chunks: {len(chunks)}")
|
||||
|
||||
|
||||
for i, chunk in enumerate(chunks):
|
||||
print(f"Chunk {i+1}:")
|
||||
print(f"Chunk {i + 1}:")
|
||||
print(f" ID: {chunk['id']}")
|
||||
print(f" Content length: {len(chunk['content'])}")
|
||||
print(f" Metadata: {chunk['metadata']}")
|
||||
@@ -174,23 +153,17 @@ async def example_content_processing():
|
||||
async def example_error_handling():
|
||||
"""Example of error handling."""
|
||||
print("=== Error Handling Example ===")
|
||||
|
||||
|
||||
# Configuration with invalid API key
|
||||
config = {
|
||||
"api_key": "invalid-key",
|
||||
"api_url": "https://api.firecrawl.dev",
|
||||
"max_retries": 3,
|
||||
"timeout": 30,
|
||||
"rate_limit_delay": 1.0
|
||||
}
|
||||
|
||||
config = {"api_key": "invalid-key", "api_url": "https://api.firecrawl.dev", "max_retries": 3, "timeout": 30, "rate_limit_delay": 1.0}
|
||||
|
||||
# Create integration
|
||||
integration = create_firecrawl_integration(config)
|
||||
|
||||
|
||||
# Test connection (should fail)
|
||||
connection_test = await integration.test_connection()
|
||||
print(f"Connection test with invalid key: {connection_test}")
|
||||
|
||||
|
||||
# Try to scrape (should fail gracefully)
|
||||
try:
|
||||
urls = ["https://httpbin.org/json"]
|
||||
@@ -203,33 +176,27 @@ async def example_error_handling():
|
||||
async def example_configuration_validation():
|
||||
"""Example of configuration validation."""
|
||||
print("=== Configuration Validation Example ===")
|
||||
|
||||
|
||||
# Test various configurations
|
||||
test_configs = [
|
||||
{
|
||||
"api_key": "fc-valid-key",
|
||||
"api_url": "https://api.firecrawl.dev",
|
||||
"max_retries": 3,
|
||||
"timeout": 30,
|
||||
"rate_limit_delay": 1.0
|
||||
},
|
||||
{"api_key": "fc-valid-key", "api_url": "https://api.firecrawl.dev", "max_retries": 3, "timeout": 30, "rate_limit_delay": 1.0},
|
||||
{
|
||||
"api_key": "invalid-key", # Invalid format
|
||||
"api_url": "https://api.firecrawl.dev"
|
||||
"api_url": "https://api.firecrawl.dev",
|
||||
},
|
||||
{
|
||||
"api_key": "fc-valid-key",
|
||||
"api_url": "invalid-url", # Invalid URL
|
||||
"max_retries": 15, # Too high
|
||||
"timeout": 500, # Too high
|
||||
"rate_limit_delay": 15.0 # Too high
|
||||
}
|
||||
"rate_limit_delay": 15.0, # Too high
|
||||
},
|
||||
]
|
||||
|
||||
|
||||
for i, config in enumerate(test_configs):
|
||||
print(f"Test configuration {i+1}:")
|
||||
print(f"Test configuration {i + 1}:")
|
||||
errors = RAGFlowFirecrawlIntegration(FirecrawlConfig.from_dict(config)).validate_config(config)
|
||||
|
||||
|
||||
if errors:
|
||||
print(" Errors found:")
|
||||
for field, error in errors.items():
|
||||
@@ -243,17 +210,17 @@ async def main():
|
||||
"""Run all examples."""
|
||||
# Set up logging
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
|
||||
|
||||
print("Firecrawl RAGFlow Integration Examples")
|
||||
print("=" * 50)
|
||||
|
||||
|
||||
# Run examples
|
||||
await example_configuration_validation()
|
||||
await example_single_url_scraping()
|
||||
await example_batch_processing()
|
||||
await example_content_processing()
|
||||
await example_error_handling()
|
||||
|
||||
|
||||
print("Examples completed!")
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user