-
Notifications
You must be signed in to change notification settings - Fork 112
/
configurable_pipeline.py
65 lines (51 loc) · 1.53 KB
/
configurable_pipeline.py
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
import argparse
import os
from r2r import (
R2RApp,
R2RConfig,
R2RPipeFactory,
R2RPipelineFactory,
R2RProviderFactory,
)
current_file_path = os.path.dirname(__file__)
configs_path = os.path.join(current_file_path, "..", "configs")
def default_app(): # config_name: str = "default", pipe_name: str = "qna"):
# config_name = os.getenv("CONFIG_OPTION") or config_name
# pipe_name = os.getenv("PIPELINE_OPTION") or pipe_name
config = R2RConfig.from_json()
providers = R2RProviderFactory(config).create_providers()
pipes = R2RPipeFactory(config, providers).create_pipes()
pipelines = R2RPipelineFactory(config, pipes).create_pipelines()
r2r = R2RApp(
config=config,
providers=providers,
pipelines=pipelines,
)
return r2r
r2r = default_app() # args.config)
app = r2r.app
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="R2R Pipe")
parser.add_argument(
"--host",
type=str,
default="0.0.0.0",
help="Port to serve deployed pipe on.",
)
parser.add_argument(
"--port",
type=str,
default="8000",
help="Port to serve deployed pipe on.",
)
# parser.add_argument(
# "--config",
# type=str,
# default="default",
# choices=CONFIG_OPTIONS.keys(),
# help="Configuration option for the pipe",
# )
args, _ = parser.parse_known_args()
host = os.getenv("HOST") or args.host
port = os.getenv("PORT") or args.port
r2r.serve()