From 4eeb9449cb607c26688c6c2a63f5b53cb648ce01 Mon Sep 17 00:00:00 2001
From: gustavoschaedler <gustavopoa@gmail.com>
Date: Wed, 16 Aug 2023 20:20:18 +0100
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix(main.py):=20reformat=20impor?=
 =?UTF-8?q?t=20statements=20to=20improve=20readability=20and=20maintainabi?=
 =?UTF-8?q?lity=20=E2=9C=A8=20feat(main.py):=20add=20support=20for=20API?=
 =?UTF-8?q?=20key=20routes=20to=20enable=20API=20key=20management=20functi?=
 =?UTF-8?q?onality=20=F0=9F=93=9D=20docs(api=5Fkey.py):=20add=20API=20key?=
 =?UTF-8?q?=20routes=20for=20retrieving,=20creating,=20and=20deleting=20AP?=
 =?UTF-8?q?I=20keys?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/main.py            |  3 +-
 src/backend/langflow/routers/api_key.py | 49 +++++++++++++++++++++++++
 2 files changed, 51 insertions(+), 1 deletion(-)
 create mode 100644 src/backend/langflow/routers/api_key.py

diff --git a/src/backend/langflow/main.py b/src/backend/langflow/main.py
index b63caff24..7045ec99d 100644
--- a/src/backend/langflow/main.py
+++ b/src/backend/langflow/main.py
@@ -6,7 +6,7 @@ from fastapi.responses import FileResponse
 from fastapi.staticfiles import StaticFiles
 
 from langflow.api import router
-from langflow.routers import login, users, health
+from langflow.routers import api_key, login, users, health
 
 from langflow.interface.utils import setup_llm_caching
 from langflow.services.database.utils import initialize_database
@@ -32,6 +32,7 @@ def create_app():
     )
 
     app.include_router(login.router)
+    app.include_router(api_key.router)
     app.include_router(users.router)
     app.include_router(health.router)
 
diff --git a/src/backend/langflow/routers/api_key.py b/src/backend/langflow/routers/api_key.py
new file mode 100644
index 000000000..7cc712962
--- /dev/null
+++ b/src/backend/langflow/routers/api_key.py
@@ -0,0 +1,49 @@
+
+
+from fastapi import APIRouter
+
+
+
+router = APIRouter(tags=["APIKey"])
+
+
+@router.get("/api_key/{user_id}")
+def get_api_key(user_id: str):
+    return {
+        "total_count": 3,
+        "user_id": user_id,
+        "api_keys": [
+            {
+                "id": "4425707e-cce4-4d1b-a54e-bd2632064657",
+                "name": "my api_key name - 01",
+                "created_at": "2023-08-15T19:28:40.019613",
+                "last_used_at": "2023-08-16T18:38:20.875210",
+            },
+            {
+                "id": "6fb7282b-9f2e-4efe-9bda-0c3d8f899473",
+                "name": "my api_key name - 02",
+                "created_at": "2023-08-15T19:41:30.077942",
+                "last_used_at": "2023-08-15T19:45:32.067899",
+            },
+            {
+                "id": "c55f3b32-4920-42b6-a5cd-698b4251806e",
+                "name": "my api_key name - 03",
+                "created_at": "2023-08-15T20:29:40.577808",
+                "last_used_at": "2023-08-15T20:29:40.577816",
+            },
+        ],
+    }
+
+
+@router.post("/api_key/{user_id}")
+def create_api_key(user_id: str):
+    return {
+        "user_id": user_id,
+        "name": "my api-key 01",
+        "api_key": "lf-eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiI1YTBmODM1ZS0yMTQxLTQ2YWItYmQ4NS0yMWEzMjQ1MTE2ZDAiLCJleHAiOjE2OTIyMTUwMTN9.c_s0ZPRtjSI9yUrhi8ACIwyXf0feRLYfaeIZEbRVKQg",
+    }
+
+
+@router.delete("/api_key/{api_key_id}")
+def delete_api_key(api_key_id: str):
+    return {"detail": "API Key deleted"}