Add SELDON_ENVIRONMENT to Index Name

SeldonIO · Aug 26, 2020 · 148db30 · 148db30
1 parent bc93191
commit 148db30
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/components/seldon-request-logger/app/log_helper.py b/components/seldon-request-logger/app/log_helper.py
@@ -40,9 +40,16 @@ def build_index_name(headers):
     index_name = os.getenv("INDEX_NAME")
     if index_name:
         return index_name
-
+
+    # Adding seldon_environment (dev/test/staging/prod) to index_name if defined as a environment variable
+    seldon_environment = os.getenv("SELDON_ENVIRONMENT")
+    if seldon_environment:
+       index_name = "inference-log-" + seldon_environment + "-" + serving_engine(headers)
+    else:
+       index_name = "inference-log-" + serving_engine(headers)
+
     # otherwise create an index per deployment
-    index_name = "inference-log-" + serving_engine(headers)
+    # index_name = "inference-log-" + serving_engine(headers)
     namespace = clean_header(NAMESPACE_HEADER_NAME, headers)
     if namespace is None:
         index_name = index_name + "-unknown-namespace"