Add a managed app deployment as an option (#268)

Co-authored-by: harjitdotsingh <singh.harjit@gmail.com>

Author: jgbradley1

.github/CODEOWNERS

@@ -0,0 +1,5 @@
+# These owners will be the default owners for everything in
+# the repo. Unless a later match takes precedence,
+# @graphrag-core-team and @societal-resilience-graphrag will be requested for
+# review when someone opens a pull request.
+*       @Azure-Samples/graphrag-core-team @Azure-Samples/societal-resilience-graphrag

.gitignore

@@ -1,3 +1,7 @@
+# ignore temporary artifact files that users may create
+infra/managed-app/artifacts/graphrag/
+openapi.json
+
 # ignore vscode config files
 .vscode/
 

CODEOWNERS

@@ -2,6 +2,7 @@
 # Licensed under the MIT License.
 
 import asyncio
+import os
 import re
 import traceback
 from math import ceil
@@ -26,12 +27,15 @@
     get_blob_container_client,
     get_cosmos_container_store_client,
     sanitize_name,
+    subscription_key_check,
 )
 
 data_route = APIRouter(
     prefix="/data",
     tags=["Data Management"],
 )
+if os.getenv("KUBERNETES_SERVICE_HOST"):
+    data_route.dependencies.append(Depends(subscription_key_check))
 
 
 @data_route.get(

backend/graphrag_app/api/data.py

@@ -1,6 +1,7 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
+import os
 import traceback
 
 from fastapi import (
@@ -14,13 +15,16 @@
 from graphrag_app.utils.azure_clients import AzureClientManager
 from graphrag_app.utils.common import (
     sanitize_name,
+    subscription_key_check,
     validate_index_file_exist,
 )
 
 graph_route = APIRouter(
     prefix="/graph",
     tags=["Graph Operations"],
 )
+if os.getenv("KUBERNETES_SERVICE_HOST"):
+    graph_route.dependencies.append(Depends(subscription_key_check))
 
 
 @graph_route.get(

backend/graphrag_app/api/graph.py

@@ -33,13 +33,16 @@
     delete_storage_container_if_exist,
     get_cosmos_container_store_client,
     sanitize_name,
+    subscription_key_check,
 )
 from graphrag_app.utils.pipeline import PipelineJob
 
 index_route = APIRouter(
     prefix="/index",
     tags=["Index Operations"],
 )
+if os.getenv("KUBERNETES_SERVICE_HOST"):
+    index_route.dependencies.append(Depends(subscription_key_check))
 
 
 @index_route.post(

backend/graphrag_app/api/index.py

@@ -1,6 +1,7 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
+import os
 import traceback
 from pathlib import Path
 
@@ -15,9 +16,11 @@
 
 from graphrag_app.logger.load_logger import load_pipeline_logger
 from graphrag_app.utils.azure_clients import AzureClientManager
-from graphrag_app.utils.common import sanitize_name
+from graphrag_app.utils.common import sanitize_name, subscription_key_check
 
 prompt_tuning_route = APIRouter(prefix="/index/config", tags=["Prompt Tuning"])
+if os.getenv("KUBERNETES_SERVICE_HOST"):
+    prompt_tuning_route.dependencies.append(Depends(subscription_key_check))
 
 
 @prompt_tuning_route.get(

backend/graphrag_app/api/prompt_tuning.py

@@ -1,12 +1,14 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
+import os
 import traceback
 from pathlib import Path
 
 import yaml
 from fastapi import (
     APIRouter,
+    Depends,
     HTTPException,
 )
 from graphrag.api.query import global_search, local_search
@@ -22,6 +24,7 @@
 from graphrag_app.utils.common import (
     get_df,
     sanitize_name,
+    subscription_key_check,
     validate_index_file_exist,
 )
 from graphrag_app.utils.pipeline import PipelineJob
@@ -30,6 +33,8 @@
     prefix="/query",
     tags=["Query Operations"],
 )
+if os.getenv("KUBERNETES_SERVICE_HOST"):
+    query_route.dependencies.append(Depends(subscription_key_check))
 
 
 @query_route.post(

backend/graphrag_app/api/query.py

@@ -10,6 +10,7 @@
 import yaml
 from fastapi import (
     APIRouter,
+    Depends,
     HTTPException,
 )
 from fastapi.responses import StreamingResponse
@@ -28,6 +29,7 @@
 from graphrag_app.utils.common import (
     get_df,
     sanitize_name,
+    subscription_key_check,
     validate_index_file_exist,
 )
 
@@ -37,6 +39,8 @@
     prefix="/query/streaming",
     tags=["Query Streaming Operations"],
 )
+if os.getenv("KUBERNETES_SERVICE_HOST"):
+    query_streaming_route.dependencies.append(Depends(subscription_key_check))
 
 
 @query_streaming_route.post(

backend/graphrag_app/api/query_streaming.py

@@ -1,7 +1,7 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
-
+import os
 import traceback
 
 import pandas as pd
@@ -18,6 +18,7 @@
 from graphrag_app.utils.common import (
     pandas_storage_options,
     sanitize_name,
+    subscription_key_check,
     validate_index_file_exist,
 )
 
@@ -26,6 +27,9 @@
     tags=["Sources"],
 )
 
+if os.getenv("KUBERNETES_SERVICE_HOST"):
+    source_route.dependencies.append(Depends(subscription_key_check))
+
 
 COMMUNITY_REPORT_TABLE = "output/create_final_community_reports.parquet"
 COVARIATES_TABLE = "output/create_final_covariates.parquet"