feat: Add SERVICE_NAME to metadata to track in LiteralAI

navapbc · Jun 9, 2024 · 565a507 · 565a507
1 parent 2468f25
commit 565a507
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/.github/workflows/push-image.yml b/.github/workflows/push-image.yml
@@ -152,6 +152,7 @@ jobs:
                     "command": [],
                     "environment": {
                         "ENV": "PROD",
+                        "SERVICE_NAME": "$SERVICE_NAME",
                         "DEPLOYMENT_DATE": "$DEPLOYMENT_DATE",
                         "BUILD_DATE": "$BUILD_DATE",
                         "BUILD_FOLDER": "${{ inputs.dockerfile_folder }}",

diff --git a/05-assistive-chatbot/chatbot-chainlit.py b/05-assistive-chatbot/chatbot-chainlit.py
@@ -34,14 +34,20 @@ async def init_chat():
     git_sha = os.environ.get("GIT_SHA", "")
     build_date = os.environ.get("BUILD_DATE", "unknown")
     initial_settings = chatbot.create_init_settings()
+    service_name = os.environ.get("SERVICE_NAME", "n/a")
     metadata = {
         **initial_settings,
         "build_date": build_date,
         "git_sha": git_sha,
+        "service_name": service_name,
         "hostname": socket.gethostname(),
     }
 
-    await cl.Message(metadata=metadata, content=f"Welcome to the Assistive Chat prototype (built {build_date})").send()
+    await cl.Message(
+        metadata=metadata,
+        disable_feedback=True,
+        content=f"Welcome to the Assistive Chat prototype (built {build_date})",
+    ).send()
 
     available_llms = llms.available_llms()
     # https://docs.chainlit.io/api-reference/chat-settings

diff --git a/05-assistive-chatbot/chatbot_api.py b/05-assistive-chatbot/chatbot_api.py
@@ -60,10 +60,11 @@ async def healthcheck(request: Request):
 
     git_sha = os.environ.get("GIT_SHA", "")
     build_date = os.environ.get("BUILD_DATE", "")
+    service_name = os.environ.get("SERVICE_NAME", "")
     hostname = f"{platform.node()} {socket.gethostname()}"
 
     logger.info("Returning: Healthy %s %s", build_date, git_sha)
-    return HTMLResponse(f"Healthy {git_sha} built at {build_date}<br/>{hostname}")
+    return HTMLResponse(f"Healthy {git_sha} built at {build_date}<br/>{service_name} {hostname}")
 
 
 ALLOWED_ENV_VARS = [