Skip to content

Commit a3e2dda

Browse files
committed
modify import of FileData type to come from new types submodule
1 parent edddcf1 commit a3e2dda

File tree

7 files changed

+12
-7
lines changed

7 files changed

+12
-7
lines changed

test/api/test_api.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,12 @@
44
import pytest
55
from fastapi.testclient import TestClient
66
from pydantic import BaseModel
7-
from unstructured_ingest.v2.interfaces import BatchFileData, BatchItem, FileData, SourceIdentifiers
7+
from unstructured_ingest.v2.types.file_data import (
8+
BatchFileData,
9+
BatchItem,
10+
FileData,
11+
SourceIdentifiers,
12+
)
813

914
from unstructured_platform_plugins.etl_uvicorn.api_generator import (
1015
EtlApiException,

test/assets/dataclass_response.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from pathlib import Path
33
from typing import Any, Optional, TypedDict, Union
44

5-
from unstructured_ingest.v2.interfaces import BatchFileData, FileData
5+
from unstructured_ingest.v2.types.file_data import BatchFileData, FileData
66

77

88
class SampleFunctionResponse(TypedDict):

test/assets/filedata_meta.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from typing import Optional, Union
33

44
from pydantic import BaseModel
5-
from unstructured_ingest.v2.interfaces import BatchFileData, FileData, SourceIdentifiers
5+
from unstructured_ingest.v2.types.file_data import BatchFileData, FileData, SourceIdentifiers
66

77
from unstructured_platform_plugins.schema import FileDataMeta, NewRecord
88

test/test_schema.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import pytest
77
from pydantic import BaseModel
88
from typing_extensions import TypedDict
9-
from unstructured_ingest.v2.interfaces import FileData
9+
from unstructured_ingest.v2.types.file_data import FileData
1010

1111
import unstructured_platform_plugins.schema.json_schema as js
1212
from unstructured_platform_plugins.schema.model import is_valid_input_dict, is_valid_response_dict

test/test_utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
import pytest
66
from pydantic import BaseModel
7-
from unstructured_ingest.v2.interfaces import FileData, SourceIdentifiers
7+
from unstructured_ingest.v2.types.file_data import FileData, SourceIdentifiers
88
from uvicorn.importer import import_from_string
99

1010
from unstructured_platform_plugins.etl_uvicorn import utils

unstructured_platform_plugins/schema/filedata_meta.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
from typing import Any
22

33
from pydantic import BaseModel, Field
4-
from unstructured_ingest.v2.interfaces import FileData
4+
from unstructured_ingest.v2.types.file_data import FileData
55

66

77
class NewRecord(BaseModel):

unstructured_platform_plugins/schema/json_schema.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
from pydantic import BaseModel, create_model
1010
from pydantic.fields import FieldInfo, PydanticUndefined
11-
from unstructured_ingest.v2.interfaces import BatchFileData, FileData
11+
from unstructured_ingest.v2.types.file_data import BatchFileData, FileData
1212

1313
from unstructured_platform_plugins.schema.utils import TypedParameter
1414
from unstructured_platform_plugins.type_hints import get_type_hints

0 commit comments

Comments
 (0)