blob: 3622b22d5542d4a1216c599e64222109985f89ac (
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
|
from typing import Optional
from pydantic import BaseModel
from r2r.base import (
AsyncPipe,
EmbeddingProvider,
EvalProvider,
KGProvider,
LLMProvider,
PromptProvider,
VectorDBProvider,
)
from r2r.pipelines import (
EvalPipeline,
IngestionPipeline,
RAGPipeline,
SearchPipeline,
)
class R2RProviders(BaseModel):
vector_db: Optional[VectorDBProvider]
embedding: Optional[EmbeddingProvider]
llm: Optional[LLMProvider]
prompt: Optional[PromptProvider]
eval: Optional[EvalProvider]
kg: Optional[KGProvider]
class Config:
arbitrary_types_allowed = True
class R2RPipes(BaseModel):
parsing_pipe: Optional[AsyncPipe]
embedding_pipe: Optional[AsyncPipe]
vector_storage_pipe: Optional[AsyncPipe]
vector_search_pipe: Optional[AsyncPipe]
rag_pipe: Optional[AsyncPipe]
streaming_rag_pipe: Optional[AsyncPipe]
eval_pipe: Optional[AsyncPipe]
kg_pipe: Optional[AsyncPipe]
kg_storage_pipe: Optional[AsyncPipe]
kg_agent_search_pipe: Optional[AsyncPipe]
class Config:
arbitrary_types_allowed = True
class R2RPipelines(BaseModel):
eval_pipeline: EvalPipeline
ingestion_pipeline: IngestionPipeline
search_pipeline: SearchPipeline
rag_pipeline: RAGPipeline
streaming_rag_pipeline: RAGPipeline
class Config:
arbitrary_types_allowed = True
|
