refactor(api_key): use api key on backend to set up user

Author: ArslanSaleem

backend/.env.example

@@ -5,3 +5,4 @@ USE_OPENAI_EMBEDDINGS=false # optional
 OPENAI_API_KEY=sk-xxxxxxxxxxxx # optional
 CHROMA_BATCH_SIZE=5 # optional
 MAX_FILE_SIZE=20971520 # optional
+PANDAETL_API_KEY=xxx-xxx-xxx-xxx # optional if you already have a PandaETL api key

backend/app/config.py

@@ -1,6 +1,7 @@
 import os
 from dotenv import load_dotenv
 from pydantic_settings import BaseSettings
+from typing import Optional
 
 # Load environment variables from .env file
 load_dotenv()
@@ -30,6 +31,9 @@ class Settings(BaseSettings):
     chat_extraction_doc_threshold: float = 0.5
     chat_extraction_max_docs: int = 50
 
+    # PandaETL api key
+    pandaetl_api_key: Optional[str] = None
+
     class Config:
         env_file = ".env"
 

backend/app/main.py

@@ -1,13 +1,14 @@
 from app import models
 from app.processing.process_queue import submit_process
-from app.repositories import process_repository, project_repository
+from app.repositories import process_repository, project_repository, user_repository
 from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 from .database import SessionLocal
 from fastapi.middleware.cors import CORSMiddleware
 from app.processing.file_preprocessing import process_file
 from .config import settings
 from .api import v1_router
+from app.schemas.user import APIKeyRequest
 
 # Initialize the FastAPI app
 app = FastAPI()
@@ -57,6 +58,26 @@ def startup_pending_processes():
         print(f"Error in startup_pending_processes: {e}")
 
 
+def setup_user():
+    try:
+        with SessionLocal() as db:
+
+            if settings.pandaetl_api_key:
+                user = user_repository.get_users(db, n=1)
+                api_key = user_repository.get_user_api_key(db)
+
+                if not user:
+                    user = user_repository.create_user(db, APIKeyRequest(email="[email protected]"))
+
+                if not api_key:
+                    user_repository.add_user_api_key(db, user.id, settings.pandaetl_api_key)
+
+                print("Successfully set up user from api key")
+
+    except Exception as e:
+        print(f"Error in setup user from api key: {e}")
+
+
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],  # Allow all origins (for development)
@@ -69,6 +90,6 @@ def startup_pending_processes():
 
 app.include_router(v1_router, prefix="/v1")
 
-
+setup_user()
 startup_pending_processes()
 startup_file_preprocessing()

frontend/.env.example

@@ -2,5 +2,4 @@ NEXT_PUBLIC_API_URL=http://localhost:3000/api/v1
 NEXT_PUBLIC_STORAGE_URL=http://localhost:3000/api/assets
 NEXT_PUBLIC_MIXPANEL_TOKEN=f2e8a71ab2bde33ebf346c5abf6ba9fa
 NEXT_PUBLIC_ROLLBAR_ACCESS_TOKEN=0df0bee895044430880278e2b2a5b2d2
-# NEXT_PUBLIC_API_KEY=xxx-xxx-xxx-xxx # Uncomment this if you have an API key
 # NEXT_PUBLIC_BACKEND_URL=http://backend:5328 # Uncomment this if you're working with a docker setup

frontend/src/middleware.ts

@@ -8,12 +8,9 @@ import { GetAPIKey } from "./services/user";
 export async function middleware(request: NextRequest) {
   let apiKey = null;
   try {
-    const envApiKey = process.env.NEXT_PUBLIC_API_KEY;
     const dockerBackendUrl = process.env.NEXT_PUBLIC_BACKEND_URL;
 
-    if (envApiKey) {
-      apiKey = { data: { api_key: envApiKey } };
-    } else if (dockerBackendUrl) {
+    if (dockerBackendUrl) {
       const response = await axios.get<{ data: APIKeyData }>(
         `${dockerBackendUrl}/v1/user/get-api-key`
       );