40 lines
873 B
Python
40 lines
873 B
Python
from typing import List
|
|
|
|
from unstructured.documents.elements import Element
|
|
|
|
|
|
class MockUnstructuredClass(object):
|
|
@staticmethod
|
|
def partition_md() -> List[Element]:
|
|
element = Element(
|
|
category="title",
|
|
embeddings=[],
|
|
id="test",
|
|
metadata={},
|
|
text="test"
|
|
)
|
|
return [element]
|
|
|
|
@staticmethod
|
|
def partition_text() -> List[Element]:
|
|
element = Element(
|
|
category="title",
|
|
embeddings=[],
|
|
id="test",
|
|
metadata={},
|
|
text="test"
|
|
)
|
|
return [element]
|
|
|
|
|
|
@staticmethod
|
|
def chunk_by_title() -> List[Element]:
|
|
element = Element(
|
|
category="title",
|
|
embeddings=[],
|
|
id="test",
|
|
metadata={},
|
|
text="test"
|
|
)
|
|
return [element]
|