Spaces:
Runtime error
Runtime error
Commit
Β·
289ada1
1
Parent(s):
d397017
Update config.py
Browse files
config.py
CHANGED
|
@@ -13,6 +13,9 @@ class DemoConfig:
|
|
| 13 |
|
| 14 |
summary_model_endpoint: str
|
| 15 |
|
|
|
|
|
|
|
|
|
|
| 16 |
@classmethod
|
| 17 |
def load_from_env(cls) -> "DemoConfig":
|
| 18 |
return DemoConfig(
|
|
@@ -21,4 +24,6 @@ class DemoConfig:
|
|
| 21 |
raven_endpoint=getenv("RAVEN_ENDPOINT"),
|
| 22 |
hf_token=getenv("HF_TOKEN"),
|
| 23 |
summary_model_endpoint=getenv("SUMMARY_MODEL_ENDPOINT"),
|
|
|
|
|
|
|
| 24 |
)
|
|
|
|
| 13 |
|
| 14 |
summary_model_endpoint: str
|
| 15 |
|
| 16 |
+
mongo_endpoint: str
|
| 17 |
+
mongo_collection: str
|
| 18 |
+
|
| 19 |
@classmethod
|
| 20 |
def load_from_env(cls) -> "DemoConfig":
|
| 21 |
return DemoConfig(
|
|
|
|
| 24 |
raven_endpoint=getenv("RAVEN_ENDPOINT"),
|
| 25 |
hf_token=getenv("HF_TOKEN"),
|
| 26 |
summary_model_endpoint=getenv("SUMMARY_MODEL_ENDPOINT"),
|
| 27 |
+
mongo_endpoint=getenv("MONGO_ENDPOINT"),
|
| 28 |
+
mongo_collection=getenv("MONGO_COLLECTION"),
|
| 29 |
)
|