update to use v0.11.0 of core

run-llama · Aug 23, 2024 · 6a29b1a · 6a29b1a
1 parent 6c700d9
commit 6a29b1a
Show file tree

Hide file tree

Showing 3 changed files with 543 additions and 615 deletions.
diff --git a/llama_parse/base.py b/llama_parse/base.py
@@ -8,7 +8,7 @@
 from io import BufferedIOBase
 
 from llama_index.core.async_utils import run_jobs
-from llama_index.core.bridge.pydantic import Field, validator
+from llama_index.core.bridge.pydantic import Field, field_validator
 from llama_index.core.constants import DEFAULT_BASE_URL
 from llama_index.core.readers.base import BasePydanticReader
 from llama_index.core.schema import Document
@@ -31,7 +31,11 @@
 class LlamaParse(BasePydanticReader):
     """A smart-parser for files."""
 
-    api_key: str = Field(default="", description="The API key for the LlamaParse API.")
+    api_key: str = Field(
+        default="",
+        description="The API key for the LlamaParse API.",
+        validate_default=True,
+    )
     base_url: str = Field(
         default=DEFAULT_BASE_URL,
         description="The base URL of the Llama Parsing API.",
@@ -134,7 +138,8 @@ class LlamaParse(BasePydanticReader):
         description="The model name for the vendor multimodal API.",
     )
 
-    @validator("api_key", pre=True, always=True)
+    @field_validator("api_key", mode="before", check_fields=True)
+    @classmethod
     def validate_api_key(cls, v: str) -> str:
         """Validate the API key."""
         if not v:
@@ -147,7 +152,8 @@ def validate_api_key(cls, v: str) -> str:
 
         return v
 
-    @validator("base_url", pre=True, always=True)
+    @field_validator("base_url", mode="before", check_fields=True)
+    @classmethod
     def validate_base_url(cls, v: str) -> str:
         """Validate the base URL."""
         url = os.getenv("LLAMA_CLOUD_BASE_URL", None)