blob: 5c63ab824f9b768fdbf4be927bca6a03f1b6a6e9 (
about) (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
import uuid
from typing import Optional, Union
from pydantic import BaseModel
from r2r.base import AnalysisTypes, FilterCriteria
class R2RUpdatePromptRequest(BaseModel):
name: str
template: Optional[str] = None
input_types: Optional[dict[str, str]] = {}
class R2RIngestFilesRequest(BaseModel):
document_ids: Optional[list[uuid.UUID]] = None
metadatas: Optional[list[dict]] = None
versions: Optional[list[str]] = None
class R2RUpdateFilesRequest(BaseModel):
metadatas: Optional[list[dict]] = None
document_ids: Optional[list[uuid.UUID]] = None
class R2RSearchRequest(BaseModel):
query: str
vector_search_settings: Optional[dict] = None
kg_search_settings: Optional[dict] = None
class R2RRAGRequest(BaseModel):
query: str
vector_search_settings: Optional[dict] = None
kg_search_settings: Optional[dict] = None
rag_generation_config: Optional[dict] = None
class R2REvalRequest(BaseModel):
query: str
context: str
completion: str
class R2RDeleteRequest(BaseModel):
keys: list[str]
values: list[Union[bool, int, str]]
class R2RAnalyticsRequest(BaseModel):
filter_criteria: FilterCriteria
analysis_types: AnalysisTypes
class R2RUsersOverviewRequest(BaseModel):
user_ids: Optional[list[uuid.UUID]]
class R2RDocumentsOverviewRequest(BaseModel):
document_ids: Optional[list[uuid.UUID]]
user_ids: Optional[list[uuid.UUID]]
class R2RDocumentChunksRequest(BaseModel):
document_id: uuid.UUID
class R2RLogsRequest(BaseModel):
log_type_filter: Optional[str] = (None,)
max_runs_requested: int = 100
class R2RPrintRelationshipsRequest(BaseModel):
limit: int = 100
class R2RExtractionRequest(BaseModel):
entity_types: list[str]
relations: list[str]
|
