mirror of
				https://github.com/open-webui/openapi-servers
				synced 2025-06-26 18:17:04 +00:00 
			
		
		
		
	feat: get user info server
This commit is contained in:
		
							parent
							
								
									0d7cc22555
								
							
						
					
					
						commit
						81255d55eb
					
				
							
								
								
									
										34
									
								
								servers/get-user-info/.dockerignore
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								servers/get-user-info/.dockerignore
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,34 @@ | ||||
| # Include any files or directories that you don't want to be copied to your | ||||
| # container here (e.g., local build artifacts, temporary files, etc.). | ||||
| # | ||||
| # For more help, visit the .dockerignore file reference guide at | ||||
| # https://docs.docker.com/go/build-context-dockerignore/ | ||||
| 
 | ||||
| **/.DS_Store | ||||
| **/__pycache__ | ||||
| **/.venv | ||||
| **/.classpath | ||||
| **/.dockerignore | ||||
| **/.env | ||||
| **/.git | ||||
| **/.gitignore | ||||
| **/.project | ||||
| **/.settings | ||||
| **/.toolstarget | ||||
| **/.vs | ||||
| **/.vscode | ||||
| **/*.*proj.user | ||||
| **/*.dbmdl | ||||
| **/*.jfm | ||||
| **/bin | ||||
| **/charts | ||||
| **/docker-compose* | ||||
| **/compose.y*ml | ||||
| **/Dockerfile* | ||||
| **/node_modules | ||||
| **/npm-debug.log | ||||
| **/obj | ||||
| **/secrets.dev.yaml | ||||
| **/values.dev.yaml | ||||
| LICENSE | ||||
| README.md | ||||
							
								
								
									
										51
									
								
								servers/get-user-info/Dockerfile
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										51
									
								
								servers/get-user-info/Dockerfile
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,51 @@ | ||||
| # syntax=docker/dockerfile:1 | ||||
| 
 | ||||
| # Comments are provided throughout this file to help you get started. | ||||
| # If you need more help, visit the Dockerfile reference guide at | ||||
| # https://docs.docker.com/go/dockerfile-reference/ | ||||
| 
 | ||||
| # Want to help us make this template better? Share your feedback here: https://forms.gle/ybq9Krt8jtBL3iCk7 | ||||
| 
 | ||||
| ARG PYTHON_VERSION=3.10.12 | ||||
| FROM python:${PYTHON_VERSION}-slim as base | ||||
| 
 | ||||
| # Prevents Python from writing pyc files. | ||||
| ENV PYTHONDONTWRITEBYTECODE=1 | ||||
| 
 | ||||
| # Keeps Python from buffering stdout and stderr to avoid situations where | ||||
| # the application crashes without emitting any logs due to buffering. | ||||
| ENV PYTHONUNBUFFERED=1 | ||||
| 
 | ||||
| WORKDIR /app | ||||
| 
 | ||||
| # Create a non-privileged user that the app will run under. | ||||
| # See https://docs.docker.com/go/dockerfile-user-best-practices/ | ||||
| ARG UID=10001 | ||||
| RUN adduser \ | ||||
|     --disabled-password \ | ||||
|     --gecos "" \ | ||||
|     --home "/nonexistent" \ | ||||
|     --shell "/sbin/nologin" \ | ||||
|     --no-create-home \ | ||||
|     --uid "${UID}" \ | ||||
|     appuser | ||||
| 
 | ||||
| # Download dependencies as a separate step to take advantage of Docker's caching. | ||||
| # Leverage a cache mount to /root/.cache/pip to speed up subsequent builds. | ||||
| # Leverage a bind mount to requirements.txt to avoid having to copy them into | ||||
| # into this layer. | ||||
| RUN --mount=type=cache,target=/root/.cache/pip \ | ||||
|     --mount=type=bind,source=requirements.txt,target=requirements.txt \ | ||||
|     python -m pip install -r requirements.txt | ||||
| 
 | ||||
| # Switch to the non-privileged user to run the application. | ||||
| USER appuser | ||||
| 
 | ||||
| # Copy the source code into the container. | ||||
| COPY . . | ||||
| 
 | ||||
| # Expose the port that the application listens on. | ||||
| EXPOSE 8000 | ||||
| 
 | ||||
| # Run the application. | ||||
| CMD uvicorn 'main:app' --host=0.0.0.0 --port=8000 | ||||
							
								
								
									
										64
									
								
								servers/get-user-info/README.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										64
									
								
								servers/get-user-info/README.md
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,64 @@ | ||||
| # 🔐 User Info Proxy API | ||||
| 
 | ||||
| A lightweight FastAPI microservice that forwards an Authorization Bearer token to an internal authentication server and returns user details. | ||||
| 
 | ||||
| ## 🚀 Features | ||||
| 
 | ||||
| - 🔁 Forwards Bearer tokens to your internal auth endpoint | ||||
| - ☁️ Async HTTP support with httpx | ||||
| - 🔒 Built-in 401 & 502 error handling | ||||
| - 🌍 CORS-friendly by default | ||||
| 
 | ||||
| ## 📦 Endpoints | ||||
| 
 | ||||
| ### GET /get_user_info | ||||
| 
 | ||||
| Forward your existing Bearer token and get authenticated user details. | ||||
| 
 | ||||
| 📥 Headers: | ||||
| 
 | ||||
| Authorization: Bearer YOUR_TOKEN | ||||
| 
 | ||||
| 📤 Response: | ||||
| 
 | ||||
| { | ||||
|   "id": "user-id", | ||||
|   "email": "user@example.com", | ||||
|   "name": "Jane Doe", | ||||
|   ... | ||||
| } | ||||
| 
 | ||||
| ## ⚙️ Setup | ||||
| 
 | ||||
| 1. Set your auth backend base URL: | ||||
| 
 | ||||
| ``` | ||||
| export OPEN_WEBUI_BASE_URL=http://your-open-webui.com | ||||
| ``` | ||||
| 
 | ||||
| 2. Run the service: | ||||
| 
 | ||||
| ``` | ||||
| uvicorn main:app --host 0.0.0.0 --reload | ||||
| ``` | ||||
| 
 | ||||
| ## 🧩 Environment Variables | ||||
| 
 | ||||
| | Name                | Description                          | Default              | | ||||
| |---------------------|--------------------------------------|----------------------| | ||||
| | OPEN_WEBUI_BASE_URL | Base URL of the internal auth server | http://localhost:3000 | | ||||
| 
 | ||||
| ## 🍿 Example | ||||
| 
 | ||||
| curl -H "Authorization: Bearer <your_token>" http://localhost:8000/get_user_info | ||||
| 
 | ||||
| ## 🧪 Tech Stack | ||||
| 
 | ||||
| - Python 3.11+ | ||||
| - FastAPI ⚡ | ||||
| - httpx 🌐 | ||||
| - Uvicorn 🔥 | ||||
| 
 | ||||
| --- | ||||
| 
 | ||||
| Made with ❤️ by your backend team. | ||||
							
								
								
									
										7
									
								
								servers/get-user-info/compose.yaml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										7
									
								
								servers/get-user-info/compose.yaml
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,7 @@ | ||||
| services: | ||||
|   server: | ||||
|     build: | ||||
|       context: . | ||||
|     ports: | ||||
|       - 8000:8000 | ||||
| 
 | ||||
							
								
								
									
										63
									
								
								servers/get-user-info/main.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										63
									
								
								servers/get-user-info/main.py
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,63 @@ | ||||
| from fastapi import FastAPI, HTTPException, Request | ||||
| from fastapi.middleware.cors import CORSMiddleware | ||||
| import aiohttp | ||||
| import os | ||||
| 
 | ||||
| OPEN_WEBUI_BASE_URL = os.getenv("OPEN_WEBUI_BASE_URL", "http://localhost:8080") | ||||
| 
 | ||||
| app = FastAPI( | ||||
|     title="User Info Proxy API", | ||||
|     version="1.0.0", | ||||
|     description="Fetch user details from the internal authentication server.", | ||||
| ) | ||||
| 
 | ||||
| app.add_middleware( | ||||
|     CORSMiddleware, | ||||
|     allow_origins=["*"],  # You may restrict this to certain domains | ||||
|     allow_credentials=True, | ||||
|     allow_methods=["*"], | ||||
|     allow_headers=["*"], | ||||
| ) | ||||
| 
 | ||||
| 
 | ||||
| @app.get( | ||||
|     "/get_session_user_info", | ||||
|     summary="Forward auth token and retrieve session user details", | ||||
|     description="Get user info from internal auth service using Authorization Bearer token.", | ||||
| ) | ||||
| async def get_session_user_info(request: Request): | ||||
|     auth_header = request.headers.get("Authorization") | ||||
| 
 | ||||
|     print(f"Received Authorization header: {auth_header}") | ||||
| 
 | ||||
|     if not auth_header or not auth_header.startswith("Bearer "): | ||||
|         raise HTTPException( | ||||
|             status_code=401, detail="Missing or invalid Authorization header" | ||||
|         ) | ||||
| 
 | ||||
|     try: | ||||
|         async with aiohttp.ClientSession() as session: | ||||
|             async with session.get( | ||||
|                 f"{OPEN_WEBUI_BASE_URL}/api/v1/auths/", | ||||
|                 headers={"Authorization": auth_header}, | ||||
|                 timeout=aiohttp.ClientTimeout(total=10.0), | ||||
|             ) as resp: | ||||
| 
 | ||||
|                 if resp.status != 200: | ||||
|                     raise HTTPException( | ||||
|                         status_code=resp.status, detail="Failed to retrieve user info" | ||||
|                     ) | ||||
| 
 | ||||
|                 data = await resp.json() | ||||
| 
 | ||||
|                 return { | ||||
|                     "id": data.get("id"), | ||||
|                     "role": data.get("role"), | ||||
|                     "name": data.get("name"), | ||||
|                     "email": data.get("email"), | ||||
|                 } | ||||
| 
 | ||||
|     except aiohttp.ClientError as exc: | ||||
|         raise HTTPException( | ||||
|             status_code=502, detail=f"Error connecting to auth service: {exc}" | ||||
|         ) | ||||
							
								
								
									
										6
									
								
								servers/get-user-info/requirements.txt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										6
									
								
								servers/get-user-info/requirements.txt
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,6 @@ | ||||
| fastapi | ||||
| uvicorn[standard] | ||||
| pydantic | ||||
| python-multipart | ||||
| 
 | ||||
| aiohttp | ||||
		Loading…
	
		Reference in New Issue
	
	Block a user