feat: Make OpenAPI spec usable by custom GPTs (#462)

run-llama · Dec 11, 2024 · 25667d4 · 25667d4
1 parent d31910a
commit 25667d4
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/.changeset/famous-snails-destroy.md b/.changeset/famous-snails-destroy.md
@@ -0,0 +1,5 @@
+---
+"create-llama": patch
+---
+
+Make OpenAPI spec usable by custom GPTs
diff --git a/templates/types/streaming/fastapi/app/api/routers/query.py b/templates/types/streaming/fastapi/app/api/routers/query.py
@@ -16,7 +16,11 @@ def get_query_engine() -> BaseQueryEngine:
     return index.as_query_engine()
 
 
-@r.get("/")
+@r.get(
+    "/",
+    summary="Get information from the knowledge base",
+    description="Retrieves relevant information from the knowledge base based on the provided search query. Returns a text response containing the matched information.",
+)
 async def query_request(
     query: str,
 ) -> str:

diff --git a/templates/types/streaming/fastapi/main.py b/templates/types/streaming/fastapi/main.py
@@ -16,7 +16,11 @@
 from fastapi.responses import RedirectResponse
 from fastapi.staticfiles import StaticFiles
 
-app = FastAPI()
+servers = []
+app_name = os.getenv("FLY_APP_NAME")
+if app_name:
+    servers = [{"url": f"https://{app_name}.fly.dev"}]
+app = FastAPI(servers=servers)
 
 init_settings()
 init_observability()