fix: added default values to skill Dockerfile

* USE_KG_DATA and USE_BOT_KG_DATA
deeppavlov · Dec 8, 2023 · 42fd228 · 42fd228
1 parent ef11d37
commit 42fd228
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/skills/dff_knowledge_prompted_skill/Dockerfile b/skills/dff_knowledge_prompted_skill/Dockerfile
@@ -34,10 +34,10 @@ ENV N_UTTERANCES_CONTEXT ${N_UTTERANCES_CONTEXT}
 ARG ENVVARS_TO_SEND
 ENV ENVVARS_TO_SEND ${ENVVARS_TO_SEND}
 
-ARG USE_KG_DATA
+ARG USE_KG_DATA=0
 ENV USE_KG_DATA=$USE_KG_DATA
 
-ARG USE_BOT_KG_DATA
+ARG USE_BOT_KG_DATA=0
 ENV USE_BOT_KG_DATA=$USE_BOT_KG_DATA
 
 CMD gunicorn --workers=1 server:app -b 0.0.0.0:${SERVICE_PORT} --reload --timeout ${GENERATIVE_TIMEOUT:-30}
diff --git a/skills/dff_knowledge_prompted_skill/scenario/response.py b/skills/dff_knowledge_prompted_skill/scenario/response.py
@@ -18,9 +18,9 @@
 GENERATIVE_TIMEOUT = float(getenv("GENERATIVE_TIMEOUT", 5))
 GENERATIVE_SERVICE_URL = getenv("GENERATIVE_SERVICE_URL")
 GENERATIVE_SERVICE_CONFIG = getenv("GENERATIVE_SERVICE_CONFIG")
-USE_KG_DATA = bool(getenv("USE_KG_DATA", 0))
+USE_KG_DATA = int(getenv("USE_KG_DATA", 0))
 USER_KG_SERVICE_URL = getenv("USER_KG_SERVICE_URL")
-USE_BOT_KG_DATA = bool(getenv("USE_BOT_KG_DATA", 0))
+USE_BOT_KG_DATA = int(getenv("USE_BOT_KG_DATA", 0))
 if GENERATIVE_SERVICE_CONFIG:
     with open(f"common/generative_configs/{GENERATIVE_SERVICE_CONFIG}", "r") as f:
         GENERATIVE_SERVICE_CONFIG = json.load(f)
@@ -94,7 +94,7 @@ def gathering_responses(reply, confidence, human_attr, bot_attr, attr):
     user_kg = int_ctx.get_last_human_utterance(ctx, actor).get("annotations", {}).get("user_knowledge_memorizer")
     logger.info(f"custom_el: {custom_el}")
     logger.info(f"user_kg: {user_kg}")
-    # logger.info(f"USE_KG_DATA: {USE_KG_DATA}")
+    logger.info(f"USE_KG_DATA: {USE_KG_DATA}")
 
     if USE_KG_DATA and user_kg and (kg_prompt := user_kg["kg_prompt"]):
         kg_prompt = re.sub(r"[-\n]", "", kg_prompt[0].lower()).split(".")
@@ -107,7 +107,7 @@ def gathering_responses(reply, confidence, human_attr, bot_attr, attr):
 
     logger.info(f"bot_custom_el: {bot_custom_el}")
     logger.info(f"bot_kg: {bot_kg}")
-    # logger.info(f"USE_BOT_KG_DATA: {USE_BOT_KG_DATA}")
+    logger.info(f"USE_BOT_KG_DATA: {USE_BOT_KG_DATA}")
 
     if USE_BOT_KG_DATA and bot_kg and (kg_prompt := bot_kg["kg_prompt"]):
         logger.info(f"kg_prompt - {kg_prompt}")