fix sdk call (#12)

Author: Coniferish

libs/unstructured/langchain_unstructured/document_loaders.py

@@ -11,8 +11,8 @@
 from langchain_core.document_loaders.base import BaseLoader
 from langchain_core.documents import Document
 from typing_extensions import TypeAlias
-from unstructured_client import UnstructuredClient  # type: ignore
-from unstructured_client.models import operations, shared  # type: ignore
+from unstructured_client import UnstructuredClient
+from unstructured_client.models import operations, shared
 
 Element: TypeAlias = Any
 
@@ -255,7 +255,7 @@ def _elements_via_api(self) -> list[dict[str, Any]]:
         """Retrieve a list of element dicts from the API using the SDK client."""
         client = self.client
         req = self._sdk_partition_request
-        response = client.general.partition(req)  # type: ignore
+        response = client.general.partition(request=req)
         if response.status_code == 200:
             return json.loads(response.raw_response.text)
         raise ValueError(

libs/unstructured/tests/integration_tests/test_document_loaders.py

@@ -7,10 +7,7 @@
 
 from langchain_unstructured import UnstructuredLoader
 
-EXAMPLE_DOCS_DIRECTORY = str(
-    Path(__file__).parent.parent.parent
-    / "example_docs/"
-)
+EXAMPLE_DOCS_DIRECTORY = str(Path(__file__).parent.parent.parent / "example_docs/")
 UNSTRUCTURED_API_KEY = os.getenv("UNSTRUCTURED_API_KEY")
 
 
@@ -205,7 +202,7 @@ def test_loader_partition_via_api_raises_TypeError_with_invalid_arg() -> None:
         mode="elements",
     )
 
-    with pytest.raises(TypeError, match="unexpected keyword argument 'mode'"):
+    with pytest.raises(TypeError):
         loader.load()
 
 

libs/unstructured/tests/unit_tests/test_document_loaders.py

@@ -4,7 +4,7 @@
 from unittest.mock import Mock, mock_open, patch
 
 import pytest
-from unstructured.documents.elements import Text  # type: ignore
+from unstructured.documents.elements import Text
 
 from langchain_unstructured.document_loaders import (
     UnstructuredLoader,