brian-yu-nexusflow commited on
Commit
289ada1
β€’
1 Parent(s): d397017

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +5 -0
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
  )