feat: add custom layer

petercat-ai · Apr 15, 2024 · fe59b8b · fe59b8b
1 parent 8d9c62e
commit fe59b8b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 11 deletions.
diff --git a/server/main.py b/server/main.py
@@ -15,10 +15,6 @@
 
 open_api_key = get_env_variable("OPENAI_API_KEY")
 is_dev = bool(get_env_variable("IS_DEV"))
-task_root = get_env_variable("LAMBDA_TASK_ROOT")
-
-if task_root:
-    os.environ["NLTK_DATA"] = os.path.join(task_root, "nltk_data")
 
 app = FastAPI( 
     title="Bo-meta Server",

diff --git a/server/routers/health_checker.py b/server/routers/health_checker.py
@@ -1,7 +1,4 @@
 from fastapi import APIRouter, Depends, HTTPException
-from uilts.env import get_env_variable
-
-task_root = get_env_variable("LAMBDA_TASK_ROOT")
 
 router = APIRouter(
     prefix="/api",
@@ -11,7 +8,4 @@
 
 @router.get("/health_checker")
 def health_checker():
-    return {
-        "Hello": "World",
-        "task_root": task_root,
-    }
+    return { "Hello": "World" }