참고소스 수정본

참고/firecrawl-main/apps/python-sdk/firecrawl/v2/methods/aio/parse.py

@@ -0,0 +1,63 @@
+import asyncio
+from functools import partial
+import json
+from typing import Optional, Dict, Any, Tuple
+
+from ...types import Document, ParseOptions
+from ...utils.normalize import normalize_document_input
+from ...utils.error_handler import handle_response_error
+from ...utils.http_client_async import AsyncHttpClient
+from ..parse import (
+    ParseFileInput,
+    _prepare_file_payload,
+    _prepare_parse_options_payload,
+)
+
+async def _prepare_parse_request(
+    file: ParseFileInput,
+    options: Optional[ParseOptions] = None,
+    *,
+    filename: Optional[str] = None,
+    content_type: Optional[str] = None,
+) -> Tuple[Dict[str, Any], Dict[str, Tuple[str, bytes, str]]]:
+    request_data = _prepare_parse_options_payload(options)
+    multipart_fields = {"options": json.dumps(request_data)}
+    loop = asyncio.get_running_loop()
+    multipart_files = await loop.run_in_executor(
+        None,
+        partial(
+            _prepare_file_payload,
+            file,
+            filename=filename,
+            content_type=content_type,
+        ),
+    )
+    return multipart_fields, multipart_files
+
+
+async def parse(
+    client: AsyncHttpClient,
+    file: ParseFileInput,
+    options: Optional[ParseOptions] = None,
+    *,
+    filename: Optional[str] = None,
+    content_type: Optional[str] = None,
+) -> Document:
+    fields, files = await _prepare_parse_request(
+        file,
+        options,
+        filename=filename,
+        content_type=content_type,
+    )
+
+    response = await client.post_multipart("/v2/parse", data=fields, files=files)
+    if response.status_code >= 400:
+        handle_response_error(response, "parse")
+
+    body = response.json()
+    if not body.get("success"):
+        raise Exception(body.get("error", "Unknown error occurred"))
+
+    document_data = body.get("data", {})
+    normalized = normalize_document_input(document_data)
+    return Document(**normalized)