From c1f6eb2ef353c81cc2d84ac409df79882de68290 Mon Sep 17 00:00:00 2001
From: Hemang <hemang@invariantlabs.ai>
Date: Tue, 4 Feb 2025 16:14:57 +0100
Subject: [PATCH] Add route for auto-generated docs.

---
 proxy/serve.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/proxy/serve.py b/proxy/serve.py
index b665c55..a0216c3 100644
--- a/proxy/serve.py
+++ b/proxy/serve.py
@@ -6,7 +6,11 @@ from routes.anthropic import proxy as anthropic_proxy
 from routes.open_ai import proxy as open_ai_proxy
 from starlette_compress import CompressMiddleware
 
-app = fastapi.FastAPI()
+app = fastapi.app = fastapi.FastAPI(
+    docs_url="/api/v1/proxy/docs",
+    redoc_url="/api/v1/proxy/redoc",
+    openapi_url="/api/v1/proxy/openapi.json",
+)
 app.add_middleware(CompressMiddleware)
 
 router = fastapi.APIRouter(prefix="/api/v1")