Add demo instance info to the server info api

Signed-off-by: Wei-Chun, Chang <wcchang@infuseai.io>
DataRecce · Apr 11, 2024 · 0c8e53f · 0c8e53f
1 parent 2791d62
commit 0c8e53f
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 9 deletions.
diff --git a/js/app/page.tsx b/js/app/page.tsx
@@ -163,12 +163,8 @@ function NavBar() {
           );
         })}
         <Spacer />
-        {!isDemoSite && (
-          <>
-            <StateImporter />
-            <StateExporter />
-          </>
-        )}
+        {!isDemoSite && <StateImporter />}
+        <StateExporter />
         <Box p="8px" mr="10px" color="gray.500">
           {version}
         </Box>

diff --git a/js/src/lib/api/info.ts b/js/src/lib/api/info.ts
@@ -120,6 +120,7 @@ export interface ServerInfoResult {
     base: LineageData;
     current: LineageData;
   };
+  demo: boolean;
 }
 
 export async function getServerInfo(): Promise<ServerInfoResult> {

diff --git a/js/src/lib/hooks/LineageGraphContext.tsx b/js/src/lib/hooks/LineageGraphContext.tsx
@@ -109,9 +109,7 @@ export function LineageGraphContextProvider({ children }: LineageGraphProps) {
 
   const errorMessage = error?.message;
   const lineage = data?.lineage;
-  const isDemoSite =
-    !!lineage?.current?.metadata.pr_url &&
-    lineage.current.metadata.pr_url.includes("cloud.datarecce.io");
+  const isDemoSite = data?.demo;
 
   return (
     <>

diff --git a/recce/server.py b/recce/server.py
@@ -1,6 +1,7 @@
 import asyncio
 import json
 import logging
+import os
 import uuid
 from contextlib import asynccontextmanager
 from dataclasses import dataclass
@@ -109,12 +110,14 @@ async def health_check(request: Request):
 @app.get("/api/info")
 async def get_info():
     dbt_context = default_dbt_context()
+    demo = os.environ.get('DEMO', False)
     try:
         return {
             'lineage': {
                 'base': dbt_context.get_lineage(base=True),
                 'current': dbt_context.get_lineage(base=False),
             },
+            'demo': bool(demo)
         }
     except Exception as e:
         raise HTTPException(status_code=400, detail=str(e))