26 lines
444 B
Python
26 lines
444 B
Python
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
|
||
|
|
class SourceOffset(BaseModel):
|
||
|
|
start: int
|
||
|
|
end: int
|
||
|
|
|
||
|
|
|
||
|
|
class TripleItem(BaseModel):
|
||
|
|
subject: str
|
||
|
|
predicate: str
|
||
|
|
object: str
|
||
|
|
source_snippet: str
|
||
|
|
source_offset: SourceOffset
|
||
|
|
|
||
|
|
|
||
|
|
class TextExtractRequest(BaseModel):
|
||
|
|
file_path: str
|
||
|
|
file_name: str
|
||
|
|
model: str | None = None
|
||
|
|
prompt_template: str | None = None
|
||
|
|
|
||
|
|
|
||
|
|
class TextExtractResponse(BaseModel):
|
||
|
|
items: list[TripleItem]
|