tweak(platform): Disable docs endpoint when not local

autogpt_platform/backend/.env.example

@@ -98,3 +98,5 @@ ENABLE_CLOUD_LOGGING=false
 ENABLE_FILE_LOGGING=false
 # Use to manually set the log directory
 # LOG_DIR=./logs
+
+APP_ENV=local

autogpt_platform/backend/backend/server/rest_api.py

@@ -53,6 +53,7 @@ async def lifespan(self, _: FastAPI):
         await db.disconnect()
 
     def run_service(self):
+        docs_url = "/docs" if settings.config.app_env == "local" else None
         app = FastAPI(
             title="AutoGPT Agent Server",
             description=(
@@ -62,6 +63,7 @@ def run_service(self):
             summary="AutoGPT Agent Server",
             version="0.1",
             lifespan=self.lifespan,
+            docs_url=docs_url
         )
 
         if self._test_dependency_overrides:

autogpt_platform/backend/backend/server/ws_api.py

@@ -27,7 +27,9 @@ async def lifespan(app: FastAPI):
     await event_queue.close()
 
 
-app = FastAPI(lifespan=lifespan)
+docs_url = "/docs" if settings.config.app_env == "local" else None
+
+app = FastAPI(lifespan=lifespan, docs_url=docs_url)
 event_queue = AsyncRedisEventQueue()
 _connection_manager = None
 

autogpt_platform/backend/backend/util/settings.py

@@ -121,6 +121,11 @@ class Config(UpdateTrackingModel["Config"], BaseSettings):
         "This value is then used to generate redirect URLs for OAuth flows.",
     )
 
+    app_env: str = Field(
+        default="local",
+        description="The name of the app environment.",
+    )
+
     backend_cors_allow_origins: List[str] = Field(default_factory=list)
 
     @field_validator("backend_cors_allow_origins")

autogpt_platform/market/.env.example

@@ -7,4 +7,6 @@ SENTRY_DSN=https://[email protected].
 
 ENABLE_AUTH=true
 SUPABASE_JWT_SECRET=our-super-secret-jwt-token-with-at-least-32-characters-long
-BACKEND_CORS_ALLOW_ORIGINS="http://localhost:3000,http://127.0.0.1:3000"
+BACKEND_CORS_ALLOW_ORIGINS="http://localhost:3000,http://127.0.0.1:3000"
+
+APP_ENV=local

autogpt_platform/market/market/app.py

@@ -49,14 +49,15 @@ async def lifespan(app: fastapi.FastAPI):
     yield
     await db_client.disconnect()
 
-
+docs_url = "/docs" if os.environ.get("APP_ENV") == "local" else None
 app = fastapi.FastAPI(
     title="Marketplace API",
     description="AutoGPT Marketplace API is a service that allows users to share AI agents.",
     summary="Maketplace API",
     version="0.1",
     lifespan=lifespan,
     root_path="/api/v1/market",
+    docs_url=docs_url,
 )
 
 app.add_middleware(fastapi.middleware.gzip.GZipMiddleware, minimum_size=1000)