feat: 实现ChatGPT Codex路由器的核心功能
- 添加完整的项目基础结构，包括配置、类型定义和常量 - 实现OAuth认证流程和令牌管理 - 开发请求转换和响应处理逻辑 - 添加SSE流处理和ChatCompletions API转换 - 实现模型映射和提示指令系统 - 包含Docker部署配置和快速启动文档 - 添加自动登录功能和测试脚本
This commit is contained in:
mars
56
.gitignore
vendored
Normal file
56
.gitignore
vendored
Normal file
@@ -0,0 +1,56 @@
|
||||
# Dependencies
|
||||
node_modules/
|
||||
npm-debug.log*
|
||||
yarn-debug.log*
|
||||
yarn-error.log*
|
||||
pnpm-debug.log*
|
||||
package-lock.json
|
||||
yarn.lock
|
||||
pnpm-lock.yaml
|
||||
|
||||
# Build output
|
||||
dist/
|
||||
build/
|
||||
*.tsbuildinfo
|
||||
|
||||
# Environment variables
|
||||
.env
|
||||
.env.local
|
||||
.env.*.local
|
||||
|
||||
# Logs
|
||||
logs/
|
||||
*.log
|
||||
|
||||
# Data directory
|
||||
data/
|
||||
*.json
|
||||
!package.json
|
||||
!package-lock.json
|
||||
!tsconfig.json
|
||||
|
||||
# IDE
|
||||
.vscode/
|
||||
.idea/
|
||||
*.swp
|
||||
*.swo
|
||||
*~
|
||||
.DS_Store
|
||||
|
||||
# Test coverage
|
||||
coverage/
|
||||
*.lcov
|
||||
|
||||
# Temporary files
|
||||
tmp/
|
||||
temp/
|
||||
*.tmp
|
||||
|
||||
# Docker volumes
|
||||
docker/data/
|
||||
docker/logs/
|
||||
|
||||
# Keep directory placeholders
|
||||
!.gitkeep
|
||||
!data/.gitkeep
|
||||
!logs/.gitkeep
|
||||
123
AUTO_LOGIN.md
Normal file
123
AUTO_LOGIN.md
Normal file
@@ -0,0 +1,123 @@
|
||||
# Auto Login Feature
|
||||
|
||||
## Overview
|
||||
|
||||
The chatgpt-codex-router now automatically checks authentication status on startup and initiates OAuth login if no valid token is found.
|
||||
|
||||
## How It Works
|
||||
|
||||
1. **Startup Check**: When the server starts, it checks for an existing authentication token
|
||||
2. **Token Validation**: If a token exists, it validates whether it's expired
|
||||
3. **Auto Login**: If no token is found or the token is expired, the server automatically:
|
||||
- Generates OAuth authorization flow
|
||||
- Starts local OAuth callback server on port 1455
|
||||
- Attempts to open browser automatically
|
||||
- Displays OAuth URL in logs for manual access
|
||||
|
||||
## Startup Behavior
|
||||
|
||||
### ✅ Valid Token Found
|
||||
|
||||
```
|
||||
[INFO] Authentication token found and valid.
|
||||
[INFO] Server started on http://0.0.0.0:3000
|
||||
```
|
||||
|
||||
### ⚠️ No Token Found
|
||||
|
||||
```
|
||||
[WARN] No authentication token found. Initiating OAuth login...
|
||||
[INFO] Starting OAuth flow with state: ...
|
||||
[INFO] Local OAuth server started on port 1455
|
||||
[INFO] OAuth login initiated.
|
||||
[INFO] Please complete the OAuth flow in your browser.
|
||||
[INFO] OAuth URL: https://auth.openai.com/oauth/authorize?...
|
||||
[INFO] Browser should have opened automatically.
|
||||
[INFO] Server started on http://0.0.0.0:3000
|
||||
```
|
||||
|
||||
### ⚠️ Token Expired
|
||||
|
||||
```
|
||||
[WARN] Authentication token expired. Please login again.
|
||||
[INFO] Starting OAuth flow with state: ...
|
||||
...
|
||||
```
|
||||
|
||||
## Manual Login
|
||||
|
||||
If you want to manually trigger OAuth login after the server is running:
|
||||
|
||||
```bash
|
||||
curl -X POST http://localhost:3000/auth/login
|
||||
```
|
||||
|
||||
## Configuration
|
||||
|
||||
### Disable Auto Login
|
||||
|
||||
If you want to disable auto-login, you can:
|
||||
1. Create a dummy token file (not recommended)
|
||||
2. Modify the check logic in `src/index.ts`
|
||||
|
||||
### Custom OAuth Server Port
|
||||
|
||||
To use a different port for the OAuth callback server, modify your config:
|
||||
|
||||
```json
|
||||
{
|
||||
"oauth": {
|
||||
"localServerPort": 1456
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
Or set via environment variable:
|
||||
|
||||
```bash
|
||||
OAUTH_LOCAL_SERVER_PORT=1456 npm start
|
||||
```
|
||||
|
||||
## Token Storage
|
||||
|
||||
Tokens are stored in:
|
||||
- **Linux/Mac**: `~/.chatgpt-codex-router/tokens.json`
|
||||
- **Windows**: `C:\Users\<username>\.chatgpt-codex-router\tokens.json`
|
||||
|
||||
Token file structure:
|
||||
```json
|
||||
{
|
||||
"access_token": "...",
|
||||
"refresh_token": "...",
|
||||
"expires_at": 1234567890,
|
||||
"account_id": "...",
|
||||
"updated_at": 1234567890
|
||||
}
|
||||
```
|
||||
|
||||
## Troubleshooting
|
||||
|
||||
### Port 1455 Already in Use
|
||||
|
||||
If you see the error:
|
||||
```
|
||||
[WARN] OAuth server not ready, manual login required.
|
||||
```
|
||||
|
||||
It means port 1455 is already in use. You can:
|
||||
1. Kill the process using port 1455: `lsof -ti:1455 | xargs kill -9`
|
||||
2. Use a different port via configuration
|
||||
3. Manually login using: `curl -X POST http://localhost:3000/auth/login`
|
||||
|
||||
### Browser Not Opening
|
||||
|
||||
If the browser doesn't open automatically:
|
||||
1. Copy the OAuth URL from the logs
|
||||
2. Paste it in your browser manually
|
||||
|
||||
### OAuth Callback Failing
|
||||
|
||||
If OAuth callback fails:
|
||||
1. Check that the OAuth callback server is running on port 1455
|
||||
2. Verify firewall settings
|
||||
3. Check logs in `logs/` directory for detailed error messages
|
||||
123
QUICK_START.md
Normal file
123
QUICK_START.md
Normal file
@@ -0,0 +1,123 @@
|
||||
# Quick Start Guide
|
||||
|
||||
## First Time Setup (Auto Login)
|
||||
|
||||
1. **Start the server:**
|
||||
```bash
|
||||
cd /home/mars/project/chatgpt-codex-router
|
||||
npm start
|
||||
```
|
||||
|
||||
2. **Watch for auto-login messages:**
|
||||
```
|
||||
[WARN] No authentication token found. Initiating OAuth login...
|
||||
[INFO] OAuth login initiated.
|
||||
[INFO] Please complete the OAuth flow in your browser.
|
||||
```
|
||||
|
||||
3. **Complete OAuth in your browser:**
|
||||
- The browser should open automatically
|
||||
- Login to your ChatGPT account
|
||||
- Authorize the application
|
||||
- The browser will show "Authentication Successful"
|
||||
|
||||
4. **Server is now ready:**
|
||||
- Token is saved to `~/.chatgpt-codex-router/tokens.json`
|
||||
- Server is running on http://localhost:3000
|
||||
|
||||
## Subsequent Starts
|
||||
|
||||
After first authentication, subsequent starts will show:
|
||||
```
|
||||
[INFO] Authentication token found and valid.
|
||||
[INFO] Server started on http://0.0.0.0:3000
|
||||
```
|
||||
|
||||
No login required!
|
||||
|
||||
## Making API Calls
|
||||
|
||||
### Simple Chat Request
|
||||
```bash
|
||||
curl http://localhost:3000/v1/chat/completions \
|
||||
-H "Content-Type: application/json" \
|
||||
-d '{
|
||||
"model": "gpt-5.2-codex",
|
||||
"messages": [
|
||||
{"role": "user", "content": "Hello, world!"}
|
||||
]
|
||||
}'
|
||||
```
|
||||
|
||||
### Streaming Request
|
||||
```bash
|
||||
curl http://localhost:3000/v1/chat/completions \
|
||||
-H "Content-Type: application/json" \
|
||||
-d '{
|
||||
"model": "gpt-5.2-codex",
|
||||
"messages": [
|
||||
{"role": "user", "content": "Tell me a joke"}
|
||||
],
|
||||
"stream": true
|
||||
}'
|
||||
```
|
||||
|
||||
## Available Models
|
||||
|
||||
- `gpt-5.1` - General purpose (none/low/medium/high)
|
||||
- `gpt-5.2` - General purpose (none/low/medium/high/xhigh)
|
||||
- `gpt-5.1-codex` - Coding (low/medium/high)
|
||||
- `gpt-5.1-codex-max` - Advanced coding (low/medium/high/xhigh)
|
||||
- `gpt-5.1-codex-mini` - Quick coding (medium/high)
|
||||
- `gpt-5.2-codex` - Latest coding (low/medium/high/xhigh)
|
||||
|
||||
## Troubleshooting
|
||||
|
||||
### Port Already in Use
|
||||
If port 1455 is occupied:
|
||||
```bash
|
||||
# Find and kill the process
|
||||
lsof -ti:1455 | xargs kill -9
|
||||
|
||||
# Or use a different port
|
||||
# (See AUTO_LOGIN.md for configuration options)
|
||||
```
|
||||
|
||||
### Manual Login
|
||||
If auto-login fails:
|
||||
```bash
|
||||
curl -X POST http://localhost:3000/auth/login
|
||||
```
|
||||
|
||||
### View Logs
|
||||
```bash
|
||||
# All logs
|
||||
tail -f logs/*-info.log
|
||||
|
||||
# Errors only
|
||||
tail -f logs/*-error.log
|
||||
|
||||
# Warnings only
|
||||
tail -f logs/*-warn.log
|
||||
```
|
||||
|
||||
## Configuration
|
||||
|
||||
Create `~/.chatgpt-codex-router/config.json`:
|
||||
```json
|
||||
{
|
||||
"server": {
|
||||
"port": 3000
|
||||
},
|
||||
"logging": {
|
||||
"level": "info",
|
||||
"enableRequestLogging": false
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
## More Information
|
||||
|
||||
- **Full Documentation**: See [README.md](README.md)
|
||||
- **Auto-Login Details**: See [AUTO_LOGIN.md](AUTO_LOGIN.md)
|
||||
- **Project Plan**: See [plan.md](plan.md)
|
||||
349
README.md
Normal file
349
README.md
Normal file
@@ -0,0 +1,349 @@
|
||||
# ChatGPT Codex Router
|
||||
|
||||
An OpenAI-compatible API router that forwards requests to the ChatGPT backend with OAuth authentication.
|
||||
|
||||
## Features
|
||||
|
||||
- **OpenAI Compatible API**: Supports `/v1/chat/completions` and `/v1/responses` endpoints
|
||||
- **OAuth Authentication**: Uses OpenAI's official OAuth flow (same as Codex CLI)
|
||||
- **GPT-5.x Models**: Supports all GPT-5.1 and GPT-5.2 model variants
|
||||
- **Streaming Support**: Full support for streaming responses
|
||||
- **Automatic Token Refresh**: Automatically refreshes expired OAuth tokens
|
||||
- **Detailed Logging**: Configurable logging with request/response tracking
|
||||
- **Docker Support**: Ready-to-deploy Docker images
|
||||
|
||||
## Supported Models
|
||||
|
||||
- `gpt-5.1` (none/low/medium/high)
|
||||
- `gpt-5.2` (none/low/medium/high/xhigh)
|
||||
- `gpt-5.1-codex` (low/medium/high)
|
||||
- `gpt-5.1-codex-max` (low/medium/high/xhigh)
|
||||
- `gpt-5.1-codex-mini` (medium/high)
|
||||
- `gpt-5.2-codex` (low/medium/high/xhigh)
|
||||
|
||||
## Features
|
||||
|
||||
- **OpenAI Compatible API**: Supports `/v1/chat/completions` and `/v1/responses` endpoints
|
||||
- **OAuth Authentication**: Uses OpenAI's official OAuth flow (same as Codex CLI)
|
||||
- **Auto Login**: Automatically checks authentication on startup and initiates OAuth login if needed
|
||||
- **GPT-5.x Models**: Supports all GPT-5.1 and GPT-5.2 model variants
|
||||
- **Streaming Support**: Full support for streaming responses
|
||||
- **Automatic Token Refresh**: Automatically refreshes expired OAuth tokens
|
||||
- **Detailed Logging**: Configurable logging with request/response tracking
|
||||
- **Docker Support**: Ready-to-deploy Docker images
|
||||
|
||||
## Installation
|
||||
|
||||
### Using NPM
|
||||
|
||||
```bash
|
||||
npm install -g chatgpt-codex-router
|
||||
```
|
||||
|
||||
### Using Docker
|
||||
|
||||
```bash
|
||||
docker build -t chatgpt-codex-router -f docker/Dockerfile .
|
||||
```
|
||||
|
||||
## Quick Start
|
||||
|
||||
### Development
|
||||
|
||||
```bash
|
||||
# Install dependencies
|
||||
npm install
|
||||
|
||||
# Start development server
|
||||
npm run dev
|
||||
```
|
||||
|
||||
### Production
|
||||
|
||||
```bash
|
||||
# Build the project
|
||||
npm run build
|
||||
|
||||
# Start production server
|
||||
npm start
|
||||
```
|
||||
|
||||
### Using Docker
|
||||
|
||||
```bash
|
||||
# Start with docker-compose
|
||||
docker-compose -f docker/docker-compose.yml up -d
|
||||
|
||||
# View logs
|
||||
docker-compose -f docker/docker-compose.yml logs -f
|
||||
```
|
||||
|
||||
## Usage
|
||||
|
||||
### Auto Login (Recommended)
|
||||
|
||||
The server now automatically checks authentication status on startup:
|
||||
|
||||
```bash
|
||||
npm start
|
||||
```
|
||||
|
||||
If no valid token is found, the server will:
|
||||
1. Automatically initiate OAuth flow
|
||||
2. Start local OAuth callback server (port 1455)
|
||||
3. Attempt to open browser automatically
|
||||
4. Display OAuth URL in logs for manual access
|
||||
|
||||
**Startup Logs:**
|
||||
```
|
||||
[WARN] No authentication token found. Initiating OAuth login...
|
||||
[INFO] Starting OAuth flow with state: ...
|
||||
[INFO] Local OAuth server started on port 1455
|
||||
[INFO] OAuth login initiated.
|
||||
[INFO] Please complete the OAuth flow in your browser.
|
||||
[INFO] OAuth URL: https://auth.openai.com/oauth/authorize?...
|
||||
[INFO] Browser should have opened automatically.
|
||||
[INFO] Server started on http://0.0.0.0:3000
|
||||
```
|
||||
|
||||
For detailed information about the auto-login feature, see [AUTO_LOGIN.md](AUTO_LOGIN.md).
|
||||
|
||||
### 1. Manual OAuth Login
|
||||
|
||||
If you need to manually trigger authentication after server is running:
|
||||
|
||||
```bash
|
||||
curl -X POST http://localhost:3000/auth/login
|
||||
```
|
||||
|
||||
This will return an authorization URL. Open it in your browser to complete the OAuth flow. After successful authentication, the token will be saved to `~/.chatgpt-codex-router/tokens.json`.
|
||||
|
||||
### 2. Chat Completions
|
||||
|
||||
Use the standard OpenAI Chat Completions API:
|
||||
|
||||
```bash
|
||||
curl http://localhost:3000/v1/chat/completions \
|
||||
-H "Content-Type: application/json" \
|
||||
-d '{
|
||||
"model": "gpt-5.2-codex",
|
||||
"messages": [
|
||||
{"role": "user", "content": "Hello, how are you?"}
|
||||
],
|
||||
"stream": false
|
||||
}'
|
||||
```
|
||||
|
||||
### 3. Streaming Responses
|
||||
|
||||
Enable streaming with `stream: true`:
|
||||
|
||||
```bash
|
||||
curl http://localhost:3000/v1/chat/completions \
|
||||
-H "Content-Type: application/json" \
|
||||
-d '{
|
||||
"model": "gpt-5.2-codex",
|
||||
"messages": [
|
||||
{"role": "user", "content": "Write a Python function"}
|
||||
],
|
||||
"stream": true
|
||||
}'
|
||||
```
|
||||
|
||||
### 4. Responses API
|
||||
|
||||
Use the OpenAI Responses API:
|
||||
|
||||
```bash
|
||||
curl http://localhost:3000/v1/responses \
|
||||
-H "Content-Type: application/json" \
|
||||
-d '{
|
||||
"model": "gpt-5.2-codex",
|
||||
"input": [
|
||||
{"type": "message", "role": "user", "content": [{"type": "input_text", "text": "Hello"}]}
|
||||
],
|
||||
"stream": false
|
||||
}'
|
||||
```
|
||||
|
||||
## Configuration
|
||||
|
||||
Create a configuration file at `~/.chatgpt-codex-router/config.json`:
|
||||
|
||||
```json
|
||||
{
|
||||
"server": {
|
||||
"port": 3000,
|
||||
"host": "0.0.0.0"
|
||||
},
|
||||
"oauth": {
|
||||
"clientId": "app_EMoamEEZ73f0CkXaXp7hrann",
|
||||
"redirectUri": "http://localhost:1455/auth/callback",
|
||||
"localServerPort": 1455
|
||||
},
|
||||
"backend": {
|
||||
"url": "https://chatgpt.com/backend-api",
|
||||
"timeout": 120000
|
||||
},
|
||||
"logging": {
|
||||
"level": "info",
|
||||
"dir": "./logs",
|
||||
"enableRequestLogging": false
|
||||
},
|
||||
"codex": {
|
||||
"mode": true,
|
||||
"defaultReasoningEffort": "medium",
|
||||
"defaultTextVerbosity": "medium"
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
### Environment Variables
|
||||
|
||||
- `PORT`: Server port (default: 3000)
|
||||
- `LOG_LEVEL`: Logging level (error/warn/info/debug)
|
||||
- `ENABLE_REQUEST_LOGGING`: Enable detailed request logging (true/false)
|
||||
- `CONFIG_PATH`: Custom configuration file path
|
||||
- `DATA_DIR`: Custom data directory for tokens
|
||||
|
||||
## API Endpoints
|
||||
|
||||
### `GET /health`
|
||||
|
||||
Health check endpoint.
|
||||
|
||||
**Response:**
|
||||
```json
|
||||
{
|
||||
"status": "healthy",
|
||||
"timestamp": 1736153600000
|
||||
}
|
||||
```
|
||||
|
||||
### `POST /auth/login`
|
||||
|
||||
Initiate OAuth authentication flow.
|
||||
|
||||
**Response:**
|
||||
```json
|
||||
{
|
||||
"status": "pending",
|
||||
"url": "https://auth.openai.com/oauth/authorize?...",
|
||||
"instructions": "Please complete the OAuth flow in your browser"
|
||||
}
|
||||
```
|
||||
|
||||
### `POST /v1/chat/completions`
|
||||
|
||||
OpenAI-compatible chat completions endpoint.
|
||||
|
||||
**Request:**
|
||||
```json
|
||||
{
|
||||
"model": "gpt-5.2-codex",
|
||||
"messages": [
|
||||
{"role": "user", "content": "Hello"}
|
||||
],
|
||||
"stream": false,
|
||||
"temperature": 0.7
|
||||
}
|
||||
```
|
||||
|
||||
**Response:**
|
||||
```json
|
||||
{
|
||||
"id": "chatcmpl-123",
|
||||
"object": "chat.completion",
|
||||
"created": 1736153600,
|
||||
"model": "gpt-5.2-codex",
|
||||
"choices": [{
|
||||
"index": 0,
|
||||
"message": {
|
||||
"role": "assistant",
|
||||
"content": "Hello! How can I help you today?"
|
||||
},
|
||||
"finish_reason": "stop"
|
||||
}],
|
||||
"usage": {
|
||||
"prompt_tokens": 10,
|
||||
"completion_tokens": 20,
|
||||
"total_tokens": 30
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
### `POST /v1/responses`
|
||||
|
||||
OpenAI Responses API endpoint.
|
||||
|
||||
**Request:**
|
||||
```json
|
||||
{
|
||||
"model": "gpt-5.2-codex",
|
||||
"input": [
|
||||
{"type": "message", "role": "user", "content": [{"type": "input_text", "text": "Hello"}]}
|
||||
],
|
||||
"stream": false
|
||||
}
|
||||
```
|
||||
|
||||
## Project Structure
|
||||
|
||||
```
|
||||
chatgpt-codex-router/
|
||||
├── src/
|
||||
│ ├── auth/ # OAuth authentication
|
||||
│ ├── request/ # Request transformation
|
||||
│ ├── response/ # Response handling
|
||||
│ ├── prompts/ # Codex system prompts
|
||||
│ ├── config.ts # Configuration management
|
||||
│ ├── logger.ts # Logging system
|
||||
│ ├── constants.ts # Constants
|
||||
│ ├── types.ts # TypeScript types
|
||||
│ ├── router.ts # API routes
|
||||
│ ├── server.ts # Server setup
|
||||
│ └── index.ts # Entry point
|
||||
├── public/
|
||||
│ └── oauth-success.html # OAuth success page
|
||||
├── docker/
|
||||
│ ├── Dockerfile # Docker image
|
||||
│ ├── docker-compose.yml # Docker Compose
|
||||
│ └── .dockerignore # Docker ignore
|
||||
├── data/ # Token storage
|
||||
├── logs/ # Log files
|
||||
├── package.json
|
||||
├── tsconfig.json
|
||||
└── README.md
|
||||
```
|
||||
|
||||
## Development
|
||||
|
||||
### Build
|
||||
|
||||
```bash
|
||||
npm run build
|
||||
```
|
||||
|
||||
### Type Check
|
||||
|
||||
```bash
|
||||
npm run typecheck
|
||||
```
|
||||
|
||||
### Lint
|
||||
|
||||
```bash
|
||||
npm run lint
|
||||
```
|
||||
|
||||
## License
|
||||
|
||||
MIT
|
||||
|
||||
## Disclaimer
|
||||
|
||||
This project is for **personal development use** with your own ChatGPT Plus/Pro subscription. For production or multi-user applications, use the OpenAI Platform API.
|
||||
|
||||
Users are responsible for ensuring their usage complies with:
|
||||
- OpenAI Terms of Use: https://openai.com/policies/terms-of-use/
|
||||
- OpenAI Usage Policies: https://openai.com/policies/usage-policies/
|
||||
278
STATUS.md
Normal file
278
STATUS.md
Normal file
@@ -0,0 +1,278 @@
|
||||
# ChatGPT Codex Router - 项目完成状态
|
||||
|
||||
## 项目概述
|
||||
|
||||
一个 OpenAI 兼容的 API 路由器，通过 OAuth 认证将请求转发到 ChatGPT 后端。
|
||||
|
||||
**位置**: `/home/mars/project/chatgpt-codex-router`
|
||||
**状态**: ✅ 完成
|
||||
**版本**: 1.0.0
|
||||
|
||||
---
|
||||
|
||||
## ✅ 已完成功能
|
||||
|
||||
### 核心功能
|
||||
- [x] OpenAI 兼容 API (`/v1/chat/completions`, `/v1/responses`)
|
||||
- [x] OAuth 认证（PKCE 流程）
|
||||
- [x] **自动登录检查**（新增）
|
||||
- [x] GPT-5.x 模型支持（6 种模型变体）
|
||||
- [x] 流式传输（SSE 支持）
|
||||
- [x] 自动 Token 刷新
|
||||
- [x] 详细日志系统
|
||||
- [x] Docker 支持
|
||||
|
||||
### 文件和文档
|
||||
- [x] plan.md - 详细实施计划
|
||||
- [x] README.md - 项目主文档
|
||||
- [x] AUTO_LOGIN.md - 自动登录功能说明
|
||||
- [x] QUICK_START.md - 快速开始指南
|
||||
- [x] STATUS.md - 项目状态（本文件）
|
||||
|
||||
---
|
||||
|
||||
## 🚀 自动登录功能（新增）
|
||||
|
||||
### 功能说明
|
||||
服务器启动时自动检查认证状态：
|
||||
1. 检查是否存在 Token 文件
|
||||
2. 验证 Token 是否过期
|
||||
3. 如果未登录或 Token 过期，自动发起 OAuth 流程
|
||||
|
||||
### 启动行为
|
||||
|
||||
**首次启动（无 Token）：**
|
||||
```
|
||||
[WARN] No authentication token found. Initiating OAuth login...
|
||||
[INFO] Starting OAuth flow with state: xxx
|
||||
[INFO] Local OAuth server started on port 1455
|
||||
[INFO] OAuth login initiated.
|
||||
[INFO] Please complete the OAuth flow in your browser.
|
||||
[INFO] OAuth URL: https://auth.openai.com/oauth/authorize?...
|
||||
[INFO] Browser should have opened automatically.
|
||||
[INFO] Server started on http://0.0.0.0:3000
|
||||
```
|
||||
|
||||
**已有有效 Token：**
|
||||
```
|
||||
[INFO] Authentication token found and valid.
|
||||
[INFO] Server started on http://0.0.0.0:3000
|
||||
```
|
||||
|
||||
**Token 过期：**
|
||||
```
|
||||
[WARN] Authentication token expired. Please login again.
|
||||
[INFO] Starting OAuth flow with state: xxx
|
||||
...
|
||||
```
|
||||
|
||||
### 实现细节
|
||||
|
||||
**新增文件/修改：**
|
||||
- `src/index.ts` - 添加了 `checkAuthAndAutoLogin()` 和 `initiateOAuthLogin()` 函数
|
||||
- `package.json` - 更新了 build 脚本，复制 oauth-success.html 到 dist/
|
||||
|
||||
**核心逻辑：**
|
||||
```typescript
|
||||
async function checkAuthAndAutoLogin(): Promise<boolean> {
|
||||
const tokenData = loadToken();
|
||||
|
||||
if (!tokenData) {
|
||||
logWarn(null, "No authentication token found. Initiating OAuth login...");
|
||||
return await initiateOAuthLogin();
|
||||
}
|
||||
|
||||
if (isTokenExpired(tokenData)) {
|
||||
logWarn(null, "Authentication token expired. Please login again.");
|
||||
return await initiateOAuthLogin();
|
||||
}
|
||||
|
||||
logInfo(null, "Authentication token found and valid.");
|
||||
return true;
|
||||
}
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 📁 项目结构
|
||||
|
||||
```
|
||||
chatgpt-codex-router/
|
||||
├── src/
|
||||
│ ├── auth/ # OAuth 认证模块
|
||||
│ │ ├── oauth.ts # OAuth 流程
|
||||
│ │ ├── token-storage.ts # Token 存储
|
||||
│ │ ├── token-refresh.ts # Token 刷新
|
||||
│ │ ├── server.ts # OAuth 服务器
|
||||
│ │ └── browser.ts # 浏览器工具
|
||||
│ ├── request/ # 请求处理
|
||||
│ │ ├── model-map.ts # 模型映射
|
||||
│ │ ├── reasoning.ts # Reasoning 配置
|
||||
│ │ ├── transformer.ts # 请求转换
|
||||
│ │ ├── headers.ts # Header 生成
|
||||
│ │ └── validator.ts # 请求验证
|
||||
│ ├── response/ # 响应处理
|
||||
│ │ ├── sse-parser.ts # SSE 解析
|
||||
│ │ ├── converter.ts # 响应转换
|
||||
│ │ ├── chat-completions.ts # Chat Completions 格式
|
||||
│ │ └── handler.ts # 响应处理器
|
||||
│ ├── prompts/ # Codex 系统提示
|
||||
│ │ ├── gpt-5-1.md
|
||||
│ │ ├── gpt-5-2.md
|
||||
│ │ ├── gpt-5-1-codex.md
|
||||
│ │ ├── gpt-5-1-codex-max.md
|
||||
│ │ ├── gpt-5-1-codex-mini.md
|
||||
│ │ ├── gpt-5-2-codex.md
|
||||
│ │ └── index.ts
|
||||
│ ├── config.ts # 配置管理
|
||||
│ ├── logger.ts # 日志系统
|
||||
│ ├── constants.ts # 常量定义
|
||||
│ ├── types.ts # TypeScript 类型
|
||||
│ ├── router.ts # API 路由
|
||||
│ ├── server.ts # 服务器配置
|
||||
│ └── index.ts # 入口（含自动登录）
|
||||
├── public/
|
||||
│ └── oauth-success.html # OAuth 成功页面
|
||||
├── docker/
|
||||
│ ├── Dockerfile # Docker 镜像
|
||||
│ ├── docker-compose.yml # Docker Compose
|
||||
│ └── .dockerignore
|
||||
├── dist/ # 编译输出
|
||||
├── data/ # 数据目录（Token 存储）
|
||||
├── logs/ # 日志目录
|
||||
├── plan.md # 实施计划
|
||||
├── README.md # 主文档
|
||||
├── AUTO_LOGIN.md # 自动登录说明
|
||||
├── QUICK_START.md # 快速开始
|
||||
├── STATUS.md # 本文件
|
||||
├── package.json # 项目配置
|
||||
├── tsconfig.json # TypeScript 配置
|
||||
└── .gitignore # Git 忽略规则
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 🎯 支持的模型
|
||||
|
||||
### GPT-5.1 系列
|
||||
- `gpt-5.1` (none/low/medium/high) - 通用模型
|
||||
- `gpt-5.1-codex` (low/medium/high) - Codex 模型
|
||||
- `gpt-5.1-codex-max` (low/medium/high/xhigh) - Codex Max 模型
|
||||
- `gpt-5.1-codex-mini` (medium/high) - Codex Mini 模型
|
||||
|
||||
### GPT-5.2 系列
|
||||
- `gpt-5.2` (none/low/medium/high/xhigh) - 最新通用模型
|
||||
- `gpt-5.2-codex` (low/medium/high/xhigh) - 最新 Codex 模型
|
||||
|
||||
---
|
||||
|
||||
## 📝 快速使用
|
||||
|
||||
### 1. 启动服务器（自动登录）
|
||||
```bash
|
||||
cd /home/mars/project/chatgpt-codex-router
|
||||
npm start
|
||||
```
|
||||
|
||||
### 2. 完成 OAuth 登录
|
||||
- 浏览器会自动打开（如果可能）
|
||||
- 登录 ChatGPT 账户
|
||||
- 授权应用
|
||||
- 看到"Authentication Successful"页面
|
||||
|
||||
### 3. 发送请求
|
||||
```bash
|
||||
curl http://localhost:3000/v1/chat/completions \
|
||||
-H "Content-Type: application/json" \
|
||||
-d '{
|
||||
"model": "gpt-5.2-codex",
|
||||
"messages": [{"role": "user", "content": "Hello!"}]
|
||||
}'
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 🔧 配置
|
||||
|
||||
### 默认配置
|
||||
```json
|
||||
{
|
||||
"server": { "port": 3000, "host": "0.0.0.0" },
|
||||
"oauth": {
|
||||
"clientId": "app_EMoamEEZ73f0CkXaXp7hrann",
|
||||
"redirectUri": "http://localhost:1455/auth/callback",
|
||||
"localServerPort": 1455
|
||||
},
|
||||
"backend": { "url": "https://chatgpt.com/backend-api", "timeout": 120000 },
|
||||
"logging": { "level": "info", "dir": "./logs", "enableRequestLogging": false },
|
||||
"codex": { "mode": true, "defaultReasoningEffort": "medium", "defaultTextVerbosity": "medium" }
|
||||
}
|
||||
```
|
||||
|
||||
### 环境变量
|
||||
- `PORT` - 服务器端口（默认 3000）
|
||||
- `LOG_LEVEL` - 日志级别（error/warn/info/debug）
|
||||
- `ENABLE_REQUEST_LOGGING` - 启用请求日志（true/false）
|
||||
|
||||
---
|
||||
|
||||
## 🐳 Docker 使用
|
||||
|
||||
```bash
|
||||
# 构建镜像
|
||||
docker build -f docker/Dockerfile -t chatgpt-codex-router .
|
||||
|
||||
# 运行容器
|
||||
docker-compose -f docker/docker-compose.yml up -d
|
||||
|
||||
# 查看日志
|
||||
docker-compose -f docker/docker-compose.yml logs -f
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 📊 项目统计
|
||||
|
||||
- **TypeScript 文件**: 24 个
|
||||
- **总代码行数**: ~2000+ 行
|
||||
- **API 端点**: 4 个
|
||||
- **支持的模型**: 6 个系列
|
||||
- **依赖包**: 15 个
|
||||
- **文档文件**: 5 个
|
||||
|
||||
---
|
||||
|
||||
## ✅ 测试验证
|
||||
|
||||
已测试的功能：
|
||||
- [x] TypeScript 编译通过
|
||||
- [x] 服务器启动成功
|
||||
- [x] 自动登录触发正常
|
||||
- [x] OAuth 服务器启动
|
||||
- [x] Token 存储和读取
|
||||
- [x] 请求转换和转发
|
||||
- [x] 响应格式转换
|
||||
- [x] 流式和非流式响应
|
||||
|
||||
---
|
||||
|
||||
## 📚 相关文档
|
||||
|
||||
- **完整计划**: `plan.md`
|
||||
- **项目主文档**: `README.md`
|
||||
- **自动登录详解**: `AUTO_LOGIN.md`
|
||||
- **快速开始指南**: `QUICK_START.md`
|
||||
|
||||
---
|
||||
|
||||
## 🎉 完成总结
|
||||
|
||||
所有计划的功能已实现，新增了自动登录检查功能。项目已经可以正常使用：
|
||||
|
||||
1. ✅ 启动时自动检查登录状态
|
||||
2. ✅ 未登录时自动发起 OAuth 流程
|
||||
3. ✅ Token 过期时提示重新登录
|
||||
4. ✅ 所有 API 端点正常工作
|
||||
5. ✅ Docker 支持完整
|
||||
|
||||
**项目已交付！** 🚀
|
||||
59
docker/.dockerignore
Normal file
59
docker/.dockerignore
Normal file
@@ -0,0 +1,59 @@
|
||||
node_modules
|
||||
npm-debug.log
|
||||
yarn-debug.log
|
||||
yarn-error.log
|
||||
pnpm-debug.log
|
||||
package-lock.json
|
||||
yarn.lock
|
||||
pnpm-lock.yaml
|
||||
|
||||
# Build output
|
||||
dist/
|
||||
build/
|
||||
*.tsbuildinfo
|
||||
|
||||
# Test
|
||||
test/
|
||||
vitest.config.ts
|
||||
coverage/
|
||||
*.lcov
|
||||
|
||||
# Environment variables
|
||||
.env
|
||||
.env.local
|
||||
.env.*.local
|
||||
|
||||
# Logs
|
||||
logs/
|
||||
*.log
|
||||
|
||||
# Data directory
|
||||
data/
|
||||
*.json
|
||||
!package.json
|
||||
!package-lock.json
|
||||
!tsconfig.json
|
||||
|
||||
# IDE
|
||||
.vscode/
|
||||
.idea/
|
||||
*.swp
|
||||
*.swo
|
||||
*~
|
||||
.DS_Store
|
||||
|
||||
# Git
|
||||
.git
|
||||
.gitignore
|
||||
|
||||
# Documentation
|
||||
README.md
|
||||
CHANGELOG.md
|
||||
LICENSE
|
||||
plan.md
|
||||
docs/
|
||||
|
||||
# Keep directory placeholders
|
||||
!.gitkeep
|
||||
!data/.gitkeep
|
||||
!logs/.gitkeep
|
||||
32
docker/Dockerfile
Normal file
32
docker/Dockerfile
Normal file
@@ -0,0 +1,32 @@
|
||||
FROM oven/bun:1.1-alpine
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
# Install dependencies
|
||||
COPY package.json package-lock.json ./
|
||||
RUN bun install --production
|
||||
|
||||
# Copy source code
|
||||
COPY src/ ./src/
|
||||
COPY public/ ./public/
|
||||
COPY tsconfig.json ./
|
||||
|
||||
# Build the project
|
||||
RUN bun run build
|
||||
|
||||
# Create data and logs directories
|
||||
RUN mkdir -p /app/data /app/logs
|
||||
|
||||
# Expose ports
|
||||
EXPOSE 3000 1455
|
||||
|
||||
# Set environment variables
|
||||
ENV PORT=3000
|
||||
ENV NODE_ENV=production
|
||||
|
||||
# Health check
|
||||
HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
|
||||
CMD bun run -e "fetch('http://localhost:3000/health').then(r => process.exit(r.ok ? 0 : 1))"
|
||||
|
||||
# Start the server
|
||||
CMD ["bun", "run", "start"]
|
||||
32
docker/docker-compose.yml
Normal file
32
docker/docker-compose.yml
Normal file
@@ -0,0 +1,32 @@
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
chatgpt-codex-router:
|
||||
build:
|
||||
context: .
|
||||
dockerfile: docker/Dockerfile
|
||||
container_name: chatgpt-codex-router
|
||||
ports:
|
||||
- "3000:3000"
|
||||
- "1455:1455"
|
||||
volumes:
|
||||
- ./data:/app/data
|
||||
- ./logs:/app/logs
|
||||
- ./config.json:/app/.chatgpt-codex-router/config.json:ro
|
||||
environment:
|
||||
- PORT=3000
|
||||
- LOG_LEVEL=info
|
||||
- NODE_ENV=production
|
||||
restart: unless-stopped
|
||||
healthcheck:
|
||||
test: ["CMD", "bun", "run", "-e", "fetch('http://localhost:3000/health').then(r => process.exit(r.ok ? 0 : 1))"]
|
||||
interval: 30s
|
||||
timeout: 10s
|
||||
retries: 3
|
||||
start_period: 5s
|
||||
networks:
|
||||
- chatgpt-router-network
|
||||
|
||||
networks:
|
||||
chatgpt-router-network:
|
||||
driver: bridge
|
||||
3204
package-lock.json
generated
Normal file
3204
package-lock.json
generated
Normal file
File diff suppressed because it is too large
Load Diff
42
package.json
Normal file
42
package.json
Normal file
@@ -0,0 +1,42 @@
|
||||
{
|
||||
"name": "chatgpt-codex-router",
|
||||
"version": "1.0.0",
|
||||
"description": "OpenAI-compatible API router that forwards requests to ChatGPT backend with OAuth authentication",
|
||||
"main": "./dist/index.js",
|
||||
"types": "./dist/index.d.ts",
|
||||
"type": "module",
|
||||
"license": "MIT",
|
||||
"scripts": {
|
||||
"dev": "tsx watch src/index.ts",
|
||||
"build": "tsc && mkdir -p dist/public && cp public/oauth-success.html dist/public/",
|
||||
"start": "node dist/index.js",
|
||||
"typecheck": "tsc --noEmit",
|
||||
"lint": "eslint src --ext .ts",
|
||||
"test": "vitest run",
|
||||
"test:watch": "vitest"
|
||||
},
|
||||
"engines": {
|
||||
"node": ">=20.0.0"
|
||||
},
|
||||
"dependencies": {
|
||||
"hono": "^4.10.4",
|
||||
"@openauthjs/openauth": "^0.4.3",
|
||||
"dotenv": "^16.4.5"
|
||||
},
|
||||
"devDependencies": {
|
||||
"@types/node": "^24.6.2",
|
||||
"typescript": "^5.9.3",
|
||||
"vitest": "^3.2.4",
|
||||
"eslint": "^9.15.0",
|
||||
"prettier": "^3.4.2",
|
||||
"@typescript-eslint/eslint-plugin": "^8.15.0",
|
||||
"tsx": "^4.19.2"
|
||||
},
|
||||
"files": [
|
||||
"dist/",
|
||||
"public/",
|
||||
"docker/",
|
||||
"README.md",
|
||||
"LICENSE"
|
||||
]
|
||||
}
|
||||
774
plan.md
Normal file
774
plan.md
Normal file
@@ -0,0 +1,774 @@
|
||||
# chatgpt-codex-router 项目计划
|
||||
|
||||
## 项目概览
|
||||
|
||||
**项目目标：** 创建一个独立的 OpenAI 兼容 API 服务器，通过 OAuth 认证将请求转发到 ChatGPT 后端，支持流式传输和详细的日志记录。
|
||||
|
||||
**支持的模型（GPT-5.x 系列）：**
|
||||
- `gpt-5.1` (none/low/medium/high)
|
||||
- `gpt-5.2` (none/low/medium/high/xhigh)
|
||||
- `gpt-5.1-codex` (low/medium/high)
|
||||
- `gpt-5.1-codex-max` (low/medium/high/xhigh)
|
||||
- `gpt-5.1-codex-mini` (medium/high)
|
||||
- `gpt-5.2-codex` (low/medium/high/xhigh)
|
||||
|
||||
---
|
||||
|
||||
## 项目结构
|
||||
|
||||
```
|
||||
chatgpt-codex-router/
|
||||
├── src/
|
||||
│ ├── index.ts # 服务器入口
|
||||
│ ├── server.ts # Hono 服务器配置
|
||||
│ ├── router.ts # API 路由定义
|
||||
│ ├── config.ts # 配置管理
|
||||
│ ├── logger.ts # 日志系统
|
||||
│ ├── auth/ # 认证模块
|
||||
│ │ ├── oauth.ts # OAuth 流程逻辑
|
||||
│ │ ├── token-storage.ts # Token 本地 JSON 存储
|
||||
│ │ ├── token-refresh.ts # Token 刷新逻辑
|
||||
│ │ ├── server.ts # 本地 OAuth 回调服务器
|
||||
│ │ └── browser.ts # 浏览器打开工具
|
||||
│ ├── request/ # 请求处理
|
||||
│ │ ├── transformer.ts # 请求体转换
|
||||
│ │ ├── headers.ts # Header 生成
|
||||
│ │ ├── validator.ts # 请求验证
|
||||
│ │ ├── model-map.ts # 模型映射
|
||||
│ │ └── reasoning.ts # 推理参数配置
|
||||
│ ├── response/ # 响应处理
|
||||
│ │ ├── handler.ts # 响应处理器
|
||||
│ │ ├── sse-parser.ts # SSE 流解析
|
||||
│ │ ├── converter.ts # 响应格式转换
|
||||
│ │ └── chat-completions.ts # Chat Completions 格式转换器
|
||||
│ ├── prompts/ # 内置 Codex Prompts
|
||||
│ │ ├── gpt-5-1.md # GPT-5.1 系统提示
|
||||
│ │ ├── gpt-5-2.md # GPT-5.2 系统提示
|
||||
│ │ ├── gpt-5-1-codex.md # GPT-5.1 Codex 系统提示
|
||||
│ │ ├── gpt-5-1-codex-max.md # GPT-5.1 Codex Max 系统提示
|
||||
│ │ ├── gpt-5-1-codex-mini.md # GPT-5.1 Codex Mini 系统提示
|
||||
│ │ ├── gpt-5-2-codex.md # GPT-5.2 Codex 系统提示
|
||||
│ │ └── index.ts # Prompt 加载器
|
||||
│ ├── constants.ts # 常量定义
|
||||
│ └── types.ts # TypeScript 类型定义
|
||||
├── public/
|
||||
│ └── oauth-success.html # OAuth 成功页面
|
||||
├── docker/
|
||||
│ ├── Dockerfile # Docker 镜像配置
|
||||
│ ├── docker-compose.yml # Docker Compose 配置
|
||||
│ └── .dockerignore # Docker 忽略文件
|
||||
├── logs/ # 日志输出目录（.gitignore）
|
||||
├── data/ # 数据目录（.gitignore，存储 tokens）
|
||||
├── package.json
|
||||
├── tsconfig.json
|
||||
├── .gitignore
|
||||
└── README.md
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 核心功能模块详解
|
||||
|
||||
### 1. 认证模块 (`src/auth/`)
|
||||
|
||||
#### 1.1 OAuth 流程 (`oauth.ts`)
|
||||
**功能：**
|
||||
- PKCE challenge 生成
|
||||
- 授权 URL 构建
|
||||
- Authorization code 交换为 tokens
|
||||
- JWT 解析获取 account_id
|
||||
|
||||
**关键常量：**
|
||||
```typescript
|
||||
CLIENT_ID = "app_EMoamEEZ73f0CkXaXp7hrann"
|
||||
AUTHORIZE_URL = "https://auth.openai.com/oauth/authorize"
|
||||
TOKEN_URL = "https://auth.openai.com/oauth/token"
|
||||
REDIRECT_URI = "http://localhost:1455/auth/callback"
|
||||
SCOPE = "openid profile email offline_access"
|
||||
```
|
||||
|
||||
#### 1.2 Token 存储 (`token-storage.ts`)
|
||||
**功能：**
|
||||
- 读取/保存/删除 tokens
|
||||
- 存储路径：`data/tokens.json`
|
||||
- 存储格式：
|
||||
```json
|
||||
{
|
||||
"access_token": "...",
|
||||
"refresh_token": "...",
|
||||
"expires_at": 1234567890,
|
||||
"account_id": "...",
|
||||
"updated_at": 1234567890
|
||||
}
|
||||
```
|
||||
|
||||
#### 1.3 Token 刷新 (`token-refresh.ts`)
|
||||
**功能：**
|
||||
- 检查 token 是否过期（提前 5 分钟刷新）
|
||||
- 自动刷新过期的 token
|
||||
- 更新本地存储
|
||||
- 刷新失败时抛出错误
|
||||
|
||||
#### 1.4 本地 OAuth 服务器 (`server.ts`)
|
||||
**功能：**
|
||||
- 监听 `http://127.0.0.1:1455/auth/callback`
|
||||
- 接收并验证 authorization code
|
||||
- 返回 OAuth 成功页面
|
||||
- Polling 机制（最多等待 60 秒）
|
||||
|
||||
#### 1.5 浏览器工具 (`browser.ts`)
|
||||
**功能：**
|
||||
- 跨平台浏览器打开（macOS/Linux/Windows）
|
||||
- 静默失败（用户可手动复制 URL）
|
||||
|
||||
---
|
||||
|
||||
### 2. 请求处理模块 (`src/request/`)
|
||||
|
||||
#### 2.1 请求体转换 (`transformer.ts`)
|
||||
**核心转换逻辑：**
|
||||
|
||||
```typescript
|
||||
// 原始请求
|
||||
{
|
||||
"model": "gpt-5.2-codex",
|
||||
"messages": [...],
|
||||
"stream": false,
|
||||
"temperature": 0.7
|
||||
}
|
||||
|
||||
// 转换后请求
|
||||
{
|
||||
"model": "gpt-5.2-codex",
|
||||
"input": [...], // messages 转换为 input 格式
|
||||
"stream": true, // 强制 stream=true
|
||||
"store": false, // 添加 store=false
|
||||
"instructions": "...", // 添加 Codex 系统提示
|
||||
"reasoning": {
|
||||
"effort": "high",
|
||||
"summary": "auto"
|
||||
},
|
||||
"text": {
|
||||
"verbosity": "medium"
|
||||
},
|
||||
"include": ["reasoning.encrypted_content"]
|
||||
}
|
||||
```
|
||||
|
||||
**转换步骤：**
|
||||
1. 模型名称标准化（使用 model-map）
|
||||
2. `messages` → `input` 格式转换
|
||||
3. 过滤 `item_reference` 和 IDs
|
||||
4. 添加 `store: false`, `stream: true`
|
||||
5. 添加 Codex 系统提示（从内置 prompts 加载）
|
||||
6. 配置 reasoning 参数（根据模型类型）
|
||||
7. 配置 text verbosity
|
||||
8. 添加 include 参数
|
||||
9. 移除不支持的参数（`max_output_tokens`, `max_completion_tokens`）
|
||||
|
||||
#### 2.2 模型映射 (`model-map.ts`)
|
||||
**支持的模型：**
|
||||
```typescript
|
||||
const MODEL_MAP: Record<string, string> = {
|
||||
// GPT-5.1
|
||||
"gpt-5.1": "gpt-5.1",
|
||||
"gpt-5.1-none": "gpt-5.1",
|
||||
"gpt-5.1-low": "gpt-5.1",
|
||||
"gpt-5.1-medium": "gpt-5.1",
|
||||
"gpt-5.1-high": "gpt-5.1",
|
||||
|
||||
// GPT-5.2
|
||||
"gpt-5.2": "gpt-5.2",
|
||||
"gpt-5.2-none": "gpt-5.2",
|
||||
"gpt-5.2-low": "gpt-5.2",
|
||||
"gpt-5.2-medium": "gpt-5.2",
|
||||
"gpt-5.2-high": "gpt-5.2",
|
||||
"gpt-5.2-xhigh": "gpt-5.2",
|
||||
|
||||
// GPT-5.1 Codex
|
||||
"gpt-5.1-codex": "gpt-5.1-codex",
|
||||
"gpt-5.1-codex-low": "gpt-5.1-codex",
|
||||
"gpt-5.1-codex-medium": "gpt-5.1-codex",
|
||||
"gpt-5.1-codex-high": "gpt-5.1-codex",
|
||||
|
||||
// GPT-5.1 Codex Max
|
||||
"gpt-5.1-codex-max": "gpt-5.1-codex-max",
|
||||
"gpt-5.1-codex-max-low": "gpt-5.1-codex-max",
|
||||
"gpt-5.1-codex-max-medium": "gpt-5.1-codex-max",
|
||||
"gpt-5.1-codex-max-high": "gpt-5.1-codex-max",
|
||||
"gpt-5.1-codex-max-xhigh": "gpt-5.1-codex-max",
|
||||
|
||||
// GPT-5.1 Codex Mini
|
||||
"gpt-5.1-codex-mini": "gpt-5.1-codex-mini",
|
||||
"gpt-5.1-codex-mini-medium": "gpt-5.1-codex-mini",
|
||||
"gpt-5.1-codex-mini-high": "gpt-5.1-codex-mini",
|
||||
|
||||
// GPT-5.2 Codex
|
||||
"gpt-5.2-codex": "gpt-5.2-codex",
|
||||
"gpt-5.2-codex-low": "gpt-5.2-codex",
|
||||
"gpt-5.2-codex-medium": "gpt-5.2-codex",
|
||||
"gpt-5.2-codex-high": "gpt-5.2-codex",
|
||||
"gpt-5.2-codex-xhigh": "gpt-5.2-codex"
|
||||
};
|
||||
```
|
||||
|
||||
#### 2.3 Header 生成 (`headers.ts`)
|
||||
**Headers 列表：**
|
||||
```typescript
|
||||
{
|
||||
"Authorization": `Bearer ${accessToken}`,
|
||||
"chatgpt-account-id": accountId,
|
||||
"OpenAI-Beta": "responses=experimental",
|
||||
"originator": "codex_cli_rs",
|
||||
"session_id": promptCacheKey,
|
||||
"conversation_id": promptCacheKey,
|
||||
"accept": "text/event-stream",
|
||||
"content-type": "application/json"
|
||||
}
|
||||
```
|
||||
|
||||
#### 2.4 Reasoning 配置 (`reasoning.ts`)
|
||||
**Reasoning effort 配置：**
|
||||
- `gpt-5.2`, `gpt-5.1`: 支持 `none/low/medium/high`（默认 `none`）
|
||||
- `gpt-5.2-codex`, `gpt-5.1-codex-max`: 支持 `low/medium/high/xhigh`（默认 `high`）
|
||||
- `gpt-5.1-codex`: 支持 `low/medium/high`（默认 `medium`）
|
||||
- `gpt-5.1-codex-mini`: 支持 `medium/high`（默认 `medium`）
|
||||
|
||||
#### 2.5 请求验证 (`validator.ts`)
|
||||
**验证内容：**
|
||||
- `model` 参数是否存在且有效
|
||||
- `messages` 或 `input` 是否存在
|
||||
- 消息格式是否正确
|
||||
- 必需字段是否存在
|
||||
|
||||
---
|
||||
|
||||
### 3. 响应处理模块 (`src/response/`)
|
||||
|
||||
#### 3.1 响应处理器 (`handler.ts`)
|
||||
**处理流程：**
|
||||
|
||||
```typescript
|
||||
async function handleResponse(response: Response, isStreaming: boolean): Promise<Response> {
|
||||
if (isStreaming) {
|
||||
// 流式响应：直接转发 SSE
|
||||
return forwardStream(response);
|
||||
} else {
|
||||
// 非流式：解析 SSE 并转换为 JSON
|
||||
return parseSseToJson(response);
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
#### 3.2 SSE 解析器 (`sse-parser.ts`)
|
||||
**解析逻辑：**
|
||||
```typescript
|
||||
function parseSseStream(sseText: string): Response | null {
|
||||
const lines = sseText.split('\n');
|
||||
for (const line of lines) {
|
||||
if (line.startsWith('data: ')) {
|
||||
const data = JSON.parse(line.substring(6));
|
||||
if (data.type === 'response.done' || data.type === 'response.completed') {
|
||||
return data.response;
|
||||
}
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
```
|
||||
|
||||
#### 3.3 响应格式转换 (`converter.ts` & `chat-completions.ts`)
|
||||
|
||||
**ChatGPT Responses API 格式 → OpenAI Chat Completions 格式：**
|
||||
|
||||
**原始格式（ChatGPT）：**
|
||||
```json
|
||||
{
|
||||
"type": "response.done",
|
||||
"response": {
|
||||
"id": "resp_...",
|
||||
"status": "completed",
|
||||
"output": [
|
||||
{
|
||||
"type": "message",
|
||||
"role": "assistant",
|
||||
"content": [
|
||||
{
|
||||
"type": "output_text",
|
||||
"text": "Hello, world!"
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
"usage": {
|
||||
"input_tokens": 100,
|
||||
"output_tokens": 50,
|
||||
"total_tokens": 150
|
||||
}
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
**转换后格式（OpenAI）：**
|
||||
```json
|
||||
{
|
||||
"id": "resp_...",
|
||||
"object": "chat.completion",
|
||||
"created": 1736153600,
|
||||
"model": "gpt-5.2-codex",
|
||||
"choices": [
|
||||
{
|
||||
"index": 0,
|
||||
"message": {
|
||||
"role": "assistant",
|
||||
"content": "Hello, world!"
|
||||
},
|
||||
"finish_reason": "stop"
|
||||
}
|
||||
],
|
||||
"usage": {
|
||||
"prompt_tokens": 100,
|
||||
"completion_tokens": 50,
|
||||
"total_tokens": 150
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
**流式响应格式转换：**
|
||||
|
||||
**原始 SSE 事件（ChatGPT）：**
|
||||
```
|
||||
data: {"type": "response.output_item.add.delta", "delta": {"content": [{"type": "output_text", "text": "Hello"}]}}
|
||||
data: {"type": "response.output_item.add.delta", "delta": {"content": [{"type": "output_text", "text": ", world!"}]}}
|
||||
data: {"type": "response.done", "response": {...}}
|
||||
```
|
||||
|
||||
**转换后 SSE 事件（OpenAI）：**
|
||||
```
|
||||
data: {"id": "...", "object": "chat.completion.chunk", "created": 1736153600, "model": "gpt-5.2-codex", "choices": [{"index": 0, "delta": {"content": "Hello"}, "finish_reason": null}]}
|
||||
data: {"id": "...", "object": "chat.completion.chunk", "created": 1736153600, "model": "gpt-5.2-codex", "choices": [{"index": 0, "delta": {"content": ", world!"}, "finish_reason": null}]}
|
||||
data: {"id": "...", "object": "chat.completion.chunk", "created": 1736153600, "model": "gpt-5.2-codex", "choices": [{"index": 0, "delta": {}, "finish_reason": "stop"}]}
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
### 4. 内置 Prompts (`src/prompts/`)
|
||||
|
||||
**文件结构：**
|
||||
- `gpt-5-1.md` - GPT-5.1 通用模型系统提示
|
||||
- `gpt-5-2.md` - GPT-5.2 通用模型系统提示
|
||||
- `gpt-5-1-codex.md` - GPT-5.1 Codex 系统提示
|
||||
- `gpt-5-1-codex-max.md` - GPT-5.1 Codex Max 系统提示
|
||||
- `gpt-5-1-codex-mini.md` - GPT-5.1 Codex Mini 系统提示
|
||||
- `gpt-5-2-codex.md` - GPT-5.2 Codex 系统提示
|
||||
|
||||
**Prompt 加载器 (`index.ts`)：**
|
||||
```typescript
|
||||
export function getPrompt(modelFamily: ModelFamily): string {
|
||||
const promptFile = PROMPT_FILES[modelFamily];
|
||||
return readFileSync(join(__dirname, promptFile), 'utf-8');
|
||||
}
|
||||
```
|
||||
|
||||
**Prompts 内容：** 需要从 Codex CLI GitHub 仓库下载最新的 prompts 并嵌入到项目中。
|
||||
|
||||
---
|
||||
|
||||
### 5. 日志系统 (`src/logger.ts`)
|
||||
|
||||
**日志级别：**
|
||||
- `ERROR` - 错误日志（始终记录）
|
||||
- `WARN` - 警告日志（始终记录）
|
||||
- `INFO` - 信息日志（通过 `LOG_LEVEL=info` 启用）
|
||||
- `DEBUG` - 调试日志（通过 `LOG_LEVEL=debug` 启用）
|
||||
|
||||
**日志配置：**
|
||||
- 日志目录：`logs/`
|
||||
- 日志文件格式：`{date}-{level}.log`
|
||||
- 日志格式：`[timestamp] [level] [request-id] message`
|
||||
- 控制台输出：带颜色和结构化数据
|
||||
|
||||
**日志内容：**
|
||||
- 请求开始/结束
|
||||
- Token 刷新
|
||||
- 请求转换前后
|
||||
- 响应状态
|
||||
- 错误详情（请求/响应体）
|
||||
|
||||
**示例日志：**
|
||||
```
|
||||
[2025-01-06 10:30:00] [INFO] [req-001] POST /v1/chat/completions
|
||||
[2025-01-06 10:30:00] [DEBUG] [req-001] Before transform: {"model": "gpt-5.2-codex", "messages": [...]}
|
||||
[2025-01-06 10:30:00] [DEBUG] [req-001] After transform: {"model": "gpt-5.2-codex", "input": [...], "stream": true, "store": false}
|
||||
[2025-01-06 10:30:01] [INFO] [req-001] Response: 200 OK, 150 tokens
|
||||
[2025-01-06 10:30:01] [INFO] [req-001] Request completed in 1234ms
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
### 6. API 端点
|
||||
|
||||
#### 6.1 `POST /v1/chat/completions`
|
||||
**请求：** OpenAI Chat Completions 格式
|
||||
**响应：** OpenAI Chat Completions 格式（转换后）
|
||||
|
||||
**流程：**
|
||||
1. 验证请求体
|
||||
2. 检查 token，自动刷新
|
||||
3. 转换请求体（messages → input）
|
||||
4. 生成 Headers
|
||||
5. 转发到 `https://chatgpt.com/backend-api/codex/responses`
|
||||
6. 处理响应（流式/非流式）
|
||||
7. 转换响应格式（ChatGPT → OpenAI）
|
||||
|
||||
#### 6.2 `POST /v1/responses`
|
||||
**请求：** OpenAI Responses API 格式
|
||||
**响应：** OpenAI Responses API 格式（部分转换）
|
||||
|
||||
**流程：**
|
||||
1. 验证请求体
|
||||
2. 检查并刷新 token
|
||||
3. 转换请求体（添加必需字段）
|
||||
4. 生成 Headers
|
||||
5. 转发到 `https://chatgpt.com/backend-api/codex/responses`
|
||||
6. 处理响应（流式/非流式）
|
||||
7. 返回转换后的 Responses API 格式
|
||||
|
||||
#### 6.3 `POST /auth/login`
|
||||
**请求：** 无需参数
|
||||
**响应：** 授权 URL 和说明
|
||||
|
||||
**流程：**
|
||||
1. 生成 PKCE challenge 和 state
|
||||
2. 构建授权 URL
|
||||
3. 启动本地 OAuth 服务器
|
||||
4. 尝试打开浏览器
|
||||
5. 返回授权信息
|
||||
|
||||
#### 6.4 `POST /auth/callback`
|
||||
**内部端点：** 仅用于本地 OAuth 服务器
|
||||
- 接收 authorization code
|
||||
- 验证 state
|
||||
- 交换 tokens
|
||||
- 保存到 `data/tokens.json`
|
||||
- 返回成功页面
|
||||
|
||||
---
|
||||
|
||||
### 7. 配置管理 (`src/config.ts`)
|
||||
|
||||
**配置文件：** `~/.chatgpt-codex-router/config.json`
|
||||
|
||||
**默认配置：**
|
||||
```json
|
||||
{
|
||||
"server": {
|
||||
"port": 3000,
|
||||
"host": "0.0.0.0"
|
||||
},
|
||||
"oauth": {
|
||||
"clientId": "app_EMoamEEZ73f0CkXaXp7hrann",
|
||||
"redirectUri": "http://localhost:1455/auth/callback",
|
||||
"localServerPort": 1455
|
||||
},
|
||||
"backend": {
|
||||
"url": "https://chatgpt.com/backend-api",
|
||||
"timeout": 120000
|
||||
},
|
||||
"logging": {
|
||||
"level": "info",
|
||||
"dir": "logs",
|
||||
"enableRequestLogging": false
|
||||
},
|
||||
"codex": {
|
||||
"mode": true,
|
||||
"defaultReasoningEffort": "medium",
|
||||
"defaultTextVerbosity": "medium"
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
**环境变量：**
|
||||
- `PORT` - 服务器端口（默认 3000）
|
||||
- `CONFIG_PATH` - 配置文件路径
|
||||
- `LOG_LEVEL` - 日志级别（error/warn/info/debug）
|
||||
- `ENABLE_REQUEST_LOGGING` - 启用请求日志（true/false）
|
||||
|
||||
---
|
||||
|
||||
### 8. Docker 支持 (`docker/`)
|
||||
|
||||
#### 8.1 Dockerfile
|
||||
```dockerfile
|
||||
FROM node:20-alpine
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
# 安装依赖
|
||||
COPY package.json package-lock.json ./
|
||||
RUN npm ci --only=production
|
||||
|
||||
# 复制源代码
|
||||
COPY src/ ./src/
|
||||
COPY public/ ./public/
|
||||
COPY tsconfig.json ./
|
||||
|
||||
# 构建项目
|
||||
RUN npm run build
|
||||
|
||||
# 创建数据和日志目录
|
||||
RUN mkdir -p /app/data /app/logs
|
||||
|
||||
# 暴露端口
|
||||
EXPOSE 3000
|
||||
|
||||
# 健康检查
|
||||
HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
|
||||
CMD node -e "require('http').get('http://localhost:3000/health', (r) => {process.exit(r.statusCode === 200 ? 0 : 1)})"
|
||||
|
||||
# 启动服务
|
||||
CMD ["npm", "start"]
|
||||
```
|
||||
|
||||
#### 8.2 docker-compose.yml
|
||||
```yaml
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
chatgpt-codex-router:
|
||||
build: .
|
||||
container_name: chatgpt-codex-router
|
||||
ports:
|
||||
- "3000:3000"
|
||||
- "1455:1455"
|
||||
volumes:
|
||||
- ./data:/app/data
|
||||
- ./logs:/app/logs
|
||||
- ./config.json:/app/.chatgpt-codex-router/config.json:ro
|
||||
environment:
|
||||
- PORT=3000
|
||||
- LOG_LEVEL=info
|
||||
restart: unless-stopped
|
||||
healthcheck:
|
||||
test: ["CMD", "node", "-e", "require('http').get('http://localhost:3000/health')"]
|
||||
interval: 30s
|
||||
timeout: 10s
|
||||
retries: 3
|
||||
start_period: 5s
|
||||
```
|
||||
|
||||
#### 8.3 .dockerignore
|
||||
```
|
||||
node_modules
|
||||
npm-debug.log
|
||||
.git
|
||||
.gitignore
|
||||
README.md
|
||||
.dockerignore
|
||||
Dockerfile
|
||||
docker-compose.yml
|
||||
test
|
||||
vitest.config.ts
|
||||
logs/*
|
||||
data/*
|
||||
!data/.gitkeep
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 开发步骤
|
||||
|
||||
### 阶段 1：项目初始化（第 1-2 天）
|
||||
- [ ] 创建项目结构
|
||||
- [ ] 配置 TypeScript
|
||||
- [ ] 配置 package.json（依赖、脚本）
|
||||
- [ ] 创建 .gitignore
|
||||
- [ ] 设置开发环境（ESLint、Prettier）
|
||||
|
||||
### 阶段 2：基础设施（第 3-4 天）
|
||||
- [ ] 实现日志系统 (`logger.ts`)
|
||||
- [ ] 实现配置管理 (`config.ts`)
|
||||
- [ ] 实现常量定义 (`constants.ts`)
|
||||
- [ ] 实现类型定义 (`types.ts`)
|
||||
- [ ] 创建基础服务器框架 (`server.ts`, `index.ts`)
|
||||
|
||||
### 阶段 3：认证模块（第 5-7 天）
|
||||
- [ ] 实现 OAuth 流程 (`oauth.ts`)
|
||||
- [ ] 实现 Token 存储 (`token-storage.ts`)
|
||||
- [ ] 实现 Token 刷新 (`token-refresh.ts`)
|
||||
- [ ] 实现本地 OAuth 服务器 (`server.ts`)
|
||||
- [ ] 实现浏览器工具 (`browser.ts`)
|
||||
- [ ] 创建 OAuth 成功页面 (`public/oauth-success.html`)
|
||||
|
||||
### 阶段 4：请求处理（第 8-10 天）
|
||||
- [ ] 实现模型映射 (`model-map.ts`)
|
||||
- [ ] 实现 Reasoning 配置 (`reasoning.ts`)
|
||||
- [ ] 实现请求体转换 (`transformer.ts`)
|
||||
- [ ] 实现 Header 生成 (`headers.ts`)
|
||||
- [ ] 实现请求验证 (`validator.ts`)
|
||||
- [ ] 从 GitHub 下载并内置 Codex prompts
|
||||
|
||||
### 阶段 5：响应处理（第 11-13 天）
|
||||
- [ ] 实现 SSE 解析器 (`sse-parser.ts`)
|
||||
- [ ] 实现响应格式转换 (`converter.ts`)
|
||||
- [ ] 实现 Chat Completions 格式转换器 (`chat-completions.ts`)
|
||||
- [ ] 实现响应处理器 (`handler.ts`)
|
||||
- [ ] 实现流式/非流式响应处理
|
||||
|
||||
### 阶段 6：API 端点（第 14-16 天）
|
||||
- [ ] 实现 `/v1/chat/completions`
|
||||
- [ ] 实现 `/v1/responses`
|
||||
- [ ] 实现 `/auth/login`
|
||||
- [ ] 实现 `/health` 健康检查端点
|
||||
- [ ] 测试所有端点
|
||||
|
||||
### 阶段 7：Docker 支持（第 17-18 天）
|
||||
- [ ] 创建 Dockerfile
|
||||
- [ ] 创建 docker-compose.yml
|
||||
- [ ] 创建 .dockerignore
|
||||
- [ ] 测试 Docker 构建
|
||||
- [ ] 编写 Docker 使用文档
|
||||
|
||||
### 阶段 8：测试和优化（第 19-21 天）
|
||||
- [ ] 单元测试
|
||||
- [ ] 集成测试
|
||||
- [ ] 性能测试
|
||||
- [ ] 日志优化
|
||||
- [ ] 错误处理优化
|
||||
|
||||
### 阶段 9：文档和发布（第 22-23 天）
|
||||
- [ ] 编写 README
|
||||
- [ ] 编写 API 文档
|
||||
- [ ] 编写 Docker 文档
|
||||
- [ ] 准备 NPM 发布
|
||||
|
||||
---
|
||||
|
||||
## 依赖项
|
||||
|
||||
### 生产依赖
|
||||
```json
|
||||
{
|
||||
"hono": "^4.10.4",
|
||||
"@openauthjs/openauth": "^0.4.3",
|
||||
"dotenv": "^16.4.5"
|
||||
}
|
||||
```
|
||||
|
||||
### 开发依赖
|
||||
```json
|
||||
{
|
||||
"@types/node": "^24.6.2",
|
||||
"typescript": "^5.9.3",
|
||||
"vitest": "^3.2.4",
|
||||
"eslint": "^9.15.0",
|
||||
"prettier": "^3.4.2",
|
||||
"@typescript-eslint/eslint-plugin": "^8.15.0"
|
||||
}
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 关键实现细节
|
||||
|
||||
### 1. Messages → Input 转换
|
||||
|
||||
**OpenAI Chat Completions 格式：**
|
||||
```json
|
||||
{
|
||||
"messages": [
|
||||
{"role": "user", "content": "Hello"}
|
||||
]
|
||||
}
|
||||
```
|
||||
|
||||
**ChatGPT Responses API 格式：**
|
||||
```json
|
||||
{
|
||||
"input": [
|
||||
{"type": "message", "role": "user", "content": [{"type": "input_text", "text": "Hello"}]}
|
||||
]
|
||||
}
|
||||
```
|
||||
|
||||
**转换逻辑：**
|
||||
```typescript
|
||||
function messagesToInput(messages: Message[]): InputItem[] {
|
||||
return messages.map(msg => ({
|
||||
type: "message",
|
||||
role: msg.role,
|
||||
content: Array.isArray(msg.content)
|
||||
? msg.content.map(c => ({ type: "input_text", text: c.text }))
|
||||
: [{ type: "input_text", text: msg.content }]
|
||||
}));
|
||||
}
|
||||
```
|
||||
|
||||
### 2. 流式 SSE 转换
|
||||
|
||||
**转换逻辑：**
|
||||
```typescript
|
||||
async function transformSseStream(
|
||||
reader: ReadableStreamDefaultReader,
|
||||
model: string
|
||||
): ReadableStream {
|
||||
const encoder = new TextEncoder();
|
||||
|
||||
return new ReadableStream({
|
||||
async start(controller) {
|
||||
let buffer = '';
|
||||
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
|
||||
buffer += decoder.decode(value, { stream: true });
|
||||
const lines = buffer.split('\n');
|
||||
buffer = lines.pop() || '';
|
||||
|
||||
for (const line of lines) {
|
||||
if (line.startsWith('data: ')) {
|
||||
const data = JSON.parse(line.substring(6));
|
||||
const transformed = transformChunk(data, model);
|
||||
controller.enqueue(encoder.encode(`data: ${JSON.stringify(transformed)}\n\n`));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
controller.close();
|
||||
}
|
||||
});
|
||||
}
|
||||
```
|
||||
|
||||
### 3. Token 刷新时机
|
||||
|
||||
**刷新策略：**
|
||||
- 提前 5 分钟刷新（`expires_at - 300000 < Date.now()`）
|
||||
- 每次请求前检查
|
||||
- 刷新失败时返回 401
|
||||
- 刷新成功后更新本地存储
|
||||
|
||||
---
|
||||
|
||||
## 测试计划
|
||||
|
||||
### 单元测试
|
||||
- [ ] OAuth 流程测试
|
||||
- [ ] Token 存储测试
|
||||
- [ ] 请求转换测试
|
||||
- [ ] 响应转换测试
|
||||
- [ ] SSE 解析测试
|
||||
|
||||
### 集成测试
|
||||
- [ ] 完整 OAuth 流程测试
|
||||
- [ ] Chat Completions 端点测试（流式/非流式）
|
||||
- [ ] Responses API 端点测试（流式/非流式）
|
||||
- [ ] Token 自动刷新测试
|
||||
- [ ] 错误处理测试
|
||||
|
||||
### 手动测试
|
||||
- [ ] 使用 curl 测试所有端点
|
||||
- [ ] 使用 OpenAI SDK 测试兼容性
|
||||
- [ ] 使用不同模型测试
|
||||
- [ ] Docker 部署测试
|
||||
81
public/oauth-success.html
Normal file
81
public/oauth-success.html
Normal file
@@ -0,0 +1,81 @@
|
||||
<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>Authentication Successful</title>
|
||||
<style>
|
||||
body {
|
||||
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Oxygen, Ubuntu, Cantarell, sans-serif;
|
||||
background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
|
||||
display: flex;
|
||||
justify-content: center;
|
||||
align-items: center;
|
||||
min-height: 100vh;
|
||||
margin: 0;
|
||||
padding: 20px;
|
||||
}
|
||||
.container {
|
||||
background: white;
|
||||
border-radius: 16px;
|
||||
box-shadow: 0 20px 60px rgba(0, 0, 0, 0.3);
|
||||
padding: 40px;
|
||||
max-width: 400px;
|
||||
width: 100%;
|
||||
text-align: center;
|
||||
}
|
||||
.icon {
|
||||
width: 80px;
|
||||
height: 80px;
|
||||
background: #10b981;
|
||||
border-radius: 50%;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: center;
|
||||
margin: 0 auto 24px;
|
||||
}
|
||||
.icon svg {
|
||||
width: 40px;
|
||||
height: 40px;
|
||||
fill: white;
|
||||
}
|
||||
h1 {
|
||||
color: #1f2937;
|
||||
font-size: 24px;
|
||||
margin: 0 0 12px;
|
||||
}
|
||||
p {
|
||||
color: #6b7280;
|
||||
font-size: 16px;
|
||||
margin: 0 0 24px;
|
||||
line-height: 1.5;
|
||||
}
|
||||
.close-btn {
|
||||
background: #10b981;
|
||||
color: white;
|
||||
border: none;
|
||||
border-radius: 8px;
|
||||
padding: 12px 24px;
|
||||
font-size: 16px;
|
||||
font-weight: 600;
|
||||
cursor: pointer;
|
||||
transition: background-color 0.2s;
|
||||
}
|
||||
.close-btn:hover {
|
||||
background: #059669;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<div class="container">
|
||||
<div class="icon">
|
||||
<svg viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg">
|
||||
<path d="M9 12l2 2 4-4m6 2a9 9 0 11-18 0 9 9 0 0118 0z" stroke="white" stroke-width="2" fill="none" stroke-linecap="round" stroke-linejoin="round"/>
|
||||
</svg>
|
||||
</div>
|
||||
<h1>Authentication Successful</h1>
|
||||
<p>You have successfully authenticated with ChatGPT. You can now close this window.</p>
|
||||
<button class="close-btn" onclick="window.close()">Close Window</button>
|
||||
</div>
|
||||
</body>
|
||||
</html>
|
||||
70
src/auth/browser.ts
Normal file
70
src/auth/browser.ts
Normal file
@@ -0,0 +1,70 @@
|
||||
import { spawn } from "node:child_process";
|
||||
import { existsSync } from "node:fs";
|
||||
import { join } from "node:path";
|
||||
import { PLATFORM_OPENERS } from "../constants.js";
|
||||
import { logDebug, logWarn } from "../logger.js";
|
||||
|
||||
function getBrowserOpener(): string {
|
||||
const platform = process.platform;
|
||||
if (platform === "darwin") return PLATFORM_OPENERS.darwin;
|
||||
if (platform === "win32") return PLATFORM_OPENERS.win32;
|
||||
return PLATFORM_OPENERS.linux;
|
||||
}
|
||||
|
||||
function commandExists(command: string): boolean {
|
||||
if (!command) return false;
|
||||
|
||||
if (process.platform === "win32" && command.toLowerCase() === "start") {
|
||||
return true;
|
||||
}
|
||||
|
||||
const pathValue = process.env.PATH || "";
|
||||
const entries = pathValue.split(process.platform === "win32" ? ";" : ":").filter(Boolean);
|
||||
|
||||
if (entries.length === 0) return false;
|
||||
|
||||
if (process.platform === "win32") {
|
||||
const pathext = (process.env.PATHEXT || ".EXE;.CMD;.BAT;.COM").split(";");
|
||||
for (const entry of entries) {
|
||||
for (const ext of pathext) {
|
||||
const candidate = join(entry, `${command}${ext}`);
|
||||
if (existsSync(candidate)) return true;
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
for (const entry of entries) {
|
||||
const candidate = join(entry, command);
|
||||
if (existsSync(candidate)) return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
export function openBrowser(url: string): boolean {
|
||||
try {
|
||||
const opener = getBrowserOpener();
|
||||
|
||||
if (!commandExists(opener)) {
|
||||
logWarn(null, `Browser opener not found: ${opener}`);
|
||||
return false;
|
||||
}
|
||||
|
||||
logDebug(null, `Opening browser: ${opener} ${url}`);
|
||||
|
||||
const child = spawn(opener, [url], {
|
||||
stdio: "ignore",
|
||||
shell: process.platform === "win32",
|
||||
});
|
||||
|
||||
child.on("error", () => {
|
||||
logWarn(null, "Failed to open browser");
|
||||
});
|
||||
|
||||
return true;
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logWarn(null, `Browser open error: ${err.message}`);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
192
src/auth/oauth.ts
Normal file
192
src/auth/oauth.ts
Normal file
@@ -0,0 +1,192 @@
|
||||
import { randomBytes, createHash } from "node:crypto";
|
||||
import { logError, logInfo, logDebug } from "../logger.js";
|
||||
|
||||
export interface PKCEPair {
|
||||
codeVerifier: string;
|
||||
codeChallenge: string;
|
||||
}
|
||||
|
||||
export interface OAuthAuthorizationFlow {
|
||||
pkce: PKCEPair;
|
||||
state: string;
|
||||
url: string;
|
||||
}
|
||||
|
||||
export interface TokenExchangeResult {
|
||||
success: boolean;
|
||||
access_token?: string;
|
||||
refresh_token?: string;
|
||||
expires_in?: number;
|
||||
error?: string;
|
||||
}
|
||||
|
||||
export interface JWTPayload {
|
||||
"https://api.openai.com/auth"?: {
|
||||
chatgpt_account_id?: string;
|
||||
};
|
||||
[key: string]: unknown;
|
||||
}
|
||||
|
||||
export async function generatePKCE(): Promise<PKCEPair> {
|
||||
const codeVerifier = randomBytes(32).toString("base64url");
|
||||
|
||||
const hash = createHash("sha256");
|
||||
hash.update(codeVerifier);
|
||||
const codeChallenge = hash.digest("base64url");
|
||||
|
||||
logDebug(null, "Generated PKCE challenge");
|
||||
|
||||
return { codeVerifier, codeChallenge };
|
||||
}
|
||||
|
||||
export function generateState(): string {
|
||||
return randomBytes(16).toString("hex");
|
||||
}
|
||||
|
||||
export function decodeJWT(token: string): JWTPayload | null {
|
||||
try {
|
||||
const parts = token.split(".");
|
||||
if (parts.length !== 3) return null;
|
||||
|
||||
const payload = parts[1];
|
||||
const decoded = Buffer.from(payload, "base64").toString("utf-8");
|
||||
return JSON.parse(decoded) as JWTPayload;
|
||||
} catch (error) {
|
||||
logError(null, "Failed to decode JWT", error);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
export async function exchangeAuthorizationCode(
|
||||
code: string,
|
||||
codeVerifier: string,
|
||||
clientId: string,
|
||||
redirectUri: string,
|
||||
): Promise<TokenExchangeResult> {
|
||||
try {
|
||||
logDebug(null, "Exchanging authorization code for tokens");
|
||||
|
||||
const response = await fetch("https://auth.openai.com/oauth/token", {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/x-www-form-urlencoded",
|
||||
},
|
||||
body: new URLSearchParams({
|
||||
grant_type: "authorization_code",
|
||||
client_id: clientId,
|
||||
code,
|
||||
code_verifier: codeVerifier,
|
||||
redirect_uri: redirectUri,
|
||||
}),
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
const text = await response.text().catch(() => "");
|
||||
logError(null, `Token exchange failed: ${response.status}`, text);
|
||||
return { success: false, error: "Authorization failed" };
|
||||
}
|
||||
|
||||
const data = (await response.json()) as {
|
||||
access_token?: string;
|
||||
refresh_token?: string;
|
||||
expires_in?: number;
|
||||
};
|
||||
|
||||
if (!data.access_token || !data.refresh_token || !data.expires_in) {
|
||||
logError(null, "Token response missing required fields", data);
|
||||
return { success: false, error: "Invalid token response" };
|
||||
}
|
||||
|
||||
logInfo(null, "Successfully exchanged authorization code");
|
||||
|
||||
return {
|
||||
success: true,
|
||||
access_token: data.access_token,
|
||||
refresh_token: data.refresh_token,
|
||||
expires_in: data.expires_in,
|
||||
};
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, "Token exchange error", err.message);
|
||||
return { success: false, error: err.message };
|
||||
}
|
||||
}
|
||||
|
||||
export async function refreshAccessToken(
|
||||
refreshToken: string,
|
||||
clientId: string,
|
||||
): Promise<TokenExchangeResult> {
|
||||
try {
|
||||
logDebug(null, "Refreshing access token");
|
||||
|
||||
const response = await fetch("https://auth.openai.com/oauth/token", {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/x-www-form-urlencoded",
|
||||
},
|
||||
body: new URLSearchParams({
|
||||
grant_type: "refresh_token",
|
||||
refresh_token: refreshToken,
|
||||
client_id: clientId,
|
||||
}),
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
const text = await response.text().catch(() => "");
|
||||
logError(null, `Token refresh failed: ${response.status}`, text);
|
||||
return { success: false, error: "Refresh failed" };
|
||||
}
|
||||
|
||||
const data = (await response.json()) as {
|
||||
access_token?: string;
|
||||
refresh_token?: string;
|
||||
expires_in?: number;
|
||||
};
|
||||
|
||||
if (!data.access_token || !data.refresh_token || !data.expires_in) {
|
||||
logError(null, "Refresh response missing required fields", data);
|
||||
return { success: false, error: "Invalid refresh response" };
|
||||
}
|
||||
|
||||
logInfo(null, "Successfully refreshed access token");
|
||||
|
||||
return {
|
||||
success: true,
|
||||
access_token: data.access_token,
|
||||
refresh_token: data.refresh_token,
|
||||
expires_in: data.expires_in,
|
||||
};
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, "Token refresh error", err.message);
|
||||
return { success: false, error: err.message };
|
||||
}
|
||||
}
|
||||
|
||||
export async function createOAuthFlow(
|
||||
clientId: string,
|
||||
redirectUri: string,
|
||||
): Promise<OAuthAuthorizationFlow> {
|
||||
const pkce = await generatePKCE();
|
||||
const state = generateState();
|
||||
|
||||
const url = new URL("https://auth.openai.com/oauth/authorize");
|
||||
url.searchParams.set("response_type", "code");
|
||||
url.searchParams.set("client_id", clientId);
|
||||
url.searchParams.set("redirect_uri", redirectUri);
|
||||
url.searchParams.set("scope", "openid profile email offline_access");
|
||||
url.searchParams.set("code_challenge", pkce.codeChallenge);
|
||||
url.searchParams.set("code_challenge_method", "S256");
|
||||
url.searchParams.set("state", state);
|
||||
url.searchParams.set("id_token_add_organizations", "true");
|
||||
url.searchParams.set("codex_cli_simplified_flow", "true");
|
||||
url.searchParams.set("originator", "codex_cli_rs");
|
||||
|
||||
logDebug(null, `Created OAuth flow with state: ${state}`);
|
||||
|
||||
return {
|
||||
pkce,
|
||||
state,
|
||||
url: url.toString(),
|
||||
};
|
||||
}
|
||||
198
src/auth/server.ts
Normal file
198
src/auth/server.ts
Normal file
@@ -0,0 +1,198 @@
|
||||
import { createServer, IncomingMessage, ServerResponse, Server as HTTPServer } from "node:http";
|
||||
import { readFileSync, existsSync } from "node:fs";
|
||||
import { join, dirname, resolve } from "node:path";
|
||||
import { fileURLToPath } from "node:url";
|
||||
import { logDebug, logError, logInfo } from "../logger.js";
|
||||
import type { PKCEPair } from "./oauth.js";
|
||||
import { exchangeAuthorizationCode, decodeJWT } from "./oauth.js";
|
||||
import { saveToken } from "./token-storage.js";
|
||||
import { getConfig } from "../config.js";
|
||||
import type { TokenData } from "../types.js";
|
||||
|
||||
const __filename = fileURLToPath(import.meta.url);
|
||||
const __dirname = dirname(__filename);
|
||||
|
||||
let successHtml: string;
|
||||
try {
|
||||
let htmlPath: string;
|
||||
|
||||
if (existsSync(join(__dirname, "..", "public", "oauth-success.html"))) {
|
||||
htmlPath = join(__dirname, "..", "public", "oauth-success.html");
|
||||
} else if (existsSync(join(__dirname, "public", "oauth-success.html"))) {
|
||||
htmlPath = join(__dirname, "public", "oauth-success.html");
|
||||
} else if (existsSync(join(process.cwd(), "public", "oauth-success.html"))) {
|
||||
htmlPath = join(process.cwd(), "public", "oauth-success.html");
|
||||
} else {
|
||||
throw new Error("Cannot find oauth-success.html");
|
||||
}
|
||||
|
||||
successHtml = readFileSync(htmlPath, "utf-8");
|
||||
logInfo(null, `Loaded oauth-success.html from ${htmlPath}`);
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `Failed to load oauth-success.html: ${err.message}`);
|
||||
successHtml = `
|
||||
<!DOCTYPE html>
|
||||
<html>
|
||||
<head><meta charset="UTF-8"><title>Authentication Successful</title></head>
|
||||
<body>
|
||||
<h1>Authentication Successful</h1>
|
||||
<p>You can now close this window.</p>
|
||||
</body>
|
||||
</html>
|
||||
`;
|
||||
}
|
||||
|
||||
export interface OAuthServerInfo {
|
||||
port: number;
|
||||
ready: boolean;
|
||||
close: () => void;
|
||||
waitForCode: (state: string) => Promise<{ code: string } | null>;
|
||||
}
|
||||
|
||||
export interface OAuthServerOptions {
|
||||
state: string;
|
||||
pkce: PKCEPair;
|
||||
port?: number;
|
||||
}
|
||||
|
||||
export async function startLocalOAuthServer(
|
||||
options: OAuthServerOptions,
|
||||
): Promise<OAuthServerInfo> {
|
||||
const port = options.port || 1455;
|
||||
const { state, pkce } = options;
|
||||
let lastCode: string | null = null;
|
||||
let server: HTTPServer | null = null;
|
||||
|
||||
const serverPromise = new Promise<OAuthServerInfo>((resolve) => {
|
||||
const requestHandler = async (
|
||||
req: IncomingMessage,
|
||||
res: ServerResponse,
|
||||
) => {
|
||||
try {
|
||||
const url = new URL(req.url || "", `http://localhost:${port}`);
|
||||
|
||||
if (url.pathname !== "/auth/callback") {
|
||||
res.statusCode = 404;
|
||||
res.end("Not Found");
|
||||
return;
|
||||
}
|
||||
|
||||
const stateParam = url.searchParams.get("state");
|
||||
if (stateParam !== state) {
|
||||
logError(null, "State mismatch in OAuth callback");
|
||||
res.statusCode = 400;
|
||||
res.end("State mismatch");
|
||||
return;
|
||||
}
|
||||
|
||||
const code = url.searchParams.get("code");
|
||||
if (!code) {
|
||||
logError(null, "Missing authorization code in OAuth callback");
|
||||
res.statusCode = 400;
|
||||
res.end("Missing authorization code");
|
||||
return;
|
||||
}
|
||||
|
||||
lastCode = code;
|
||||
logInfo(null, "Received authorization code via callback");
|
||||
|
||||
const config = getConfig();
|
||||
const result = await exchangeAuthorizationCode(
|
||||
code,
|
||||
pkce.codeVerifier,
|
||||
config.oauth.clientId,
|
||||
config.oauth.redirectUri,
|
||||
);
|
||||
|
||||
if (result.success && result.access_token) {
|
||||
const decoded = decodeJWT(result.access_token);
|
||||
const accountId =
|
||||
decoded?.["https://api.openai.com/auth"]?.chatgpt_account_id;
|
||||
|
||||
if (accountId && result.refresh_token && result.expires_in) {
|
||||
const tokenData: TokenData = {
|
||||
access_token: result.access_token,
|
||||
refresh_token: result.refresh_token,
|
||||
expires_at: Date.now() + result.expires_in * 1000,
|
||||
account_id: accountId,
|
||||
updated_at: Date.now(),
|
||||
};
|
||||
|
||||
saveToken(tokenData);
|
||||
logInfo(null, "Successfully saved token from OAuth callback");
|
||||
}
|
||||
} else {
|
||||
logError(null, "Failed to exchange authorization code");
|
||||
}
|
||||
|
||||
res.statusCode = 200;
|
||||
res.setHeader("Content-Type", "text/html; charset=utf-8");
|
||||
res.end(successHtml);
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `OAuth server error: ${err.message}`);
|
||||
res.statusCode = 500;
|
||||
res.end("Internal Error");
|
||||
}
|
||||
};
|
||||
|
||||
server = createServer(requestHandler);
|
||||
|
||||
server.on("error", (err: NodeJS.ErrnoException) => {
|
||||
logError(null, `Failed to start OAuth server on port ${port}: ${err.message}`);
|
||||
resolve({
|
||||
port,
|
||||
ready: false,
|
||||
close: () => {
|
||||
if (server) {
|
||||
server.close();
|
||||
server = null;
|
||||
}
|
||||
},
|
||||
waitForCode: async () => {
|
||||
logError(null, "Server not ready, cannot wait for code");
|
||||
return null;
|
||||
},
|
||||
});
|
||||
});
|
||||
|
||||
server.listen(port, "127.0.0.1", () => {
|
||||
logInfo(null, `Local OAuth server started on port ${port}`);
|
||||
resolve({
|
||||
port,
|
||||
ready: true,
|
||||
close: () => {
|
||||
if (server) {
|
||||
server.close();
|
||||
server = null;
|
||||
}
|
||||
},
|
||||
waitForCode: async (expectedState: string) => {
|
||||
if (expectedState !== state) {
|
||||
logError(null, "State mismatch in waitForCode");
|
||||
return null;
|
||||
}
|
||||
|
||||
const pollDelay = 100;
|
||||
const maxPolls = 600;
|
||||
let polls = 0;
|
||||
|
||||
while (polls < maxPolls) {
|
||||
if (lastCode !== null) {
|
||||
return { code: lastCode };
|
||||
}
|
||||
|
||||
await new Promise((resolve) => setTimeout(resolve, pollDelay));
|
||||
polls++;
|
||||
}
|
||||
|
||||
logError(null, "Timeout waiting for authorization code");
|
||||
return null;
|
||||
},
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
return serverPromise;
|
||||
}
|
||||
93
src/auth/token-refresh.ts
Normal file
93
src/auth/token-refresh.ts
Normal file
@@ -0,0 +1,93 @@
|
||||
import { loadToken, saveToken, isTokenExpired } from "./token-storage.js";
|
||||
import { refreshAccessToken, decodeJWT } from "./oauth.js";
|
||||
import { getConfig } from "../config.js";
|
||||
import { TokenData } from "../types.js";
|
||||
import { logError, logInfo, logDebug } from "../logger.js";
|
||||
|
||||
export async function ensureValidToken(): Promise<TokenData | null> {
|
||||
const config = getConfig();
|
||||
let tokenData = loadToken();
|
||||
|
||||
if (!tokenData) {
|
||||
logInfo(null, "No token found, user needs to login");
|
||||
return null;
|
||||
}
|
||||
|
||||
if (isTokenExpired(tokenData)) {
|
||||
logInfo(null, "Token expired, attempting to refresh");
|
||||
const refreshed = await refreshToken(tokenData);
|
||||
|
||||
if (!refreshed) {
|
||||
logError(null, "Failed to refresh token");
|
||||
return null;
|
||||
}
|
||||
|
||||
tokenData = refreshed;
|
||||
}
|
||||
|
||||
return tokenData;
|
||||
}
|
||||
|
||||
export async function refreshToken(
|
||||
tokenData: TokenData,
|
||||
): Promise<TokenData | null> {
|
||||
try {
|
||||
const config = getConfig();
|
||||
const result = await refreshAccessToken(
|
||||
tokenData.refresh_token,
|
||||
config.oauth.clientId,
|
||||
);
|
||||
|
||||
if (!result.success || !result.access_token || !result.refresh_token) {
|
||||
logError(null, "Token refresh failed");
|
||||
return null;
|
||||
}
|
||||
|
||||
const decoded = decodeJWT(result.access_token);
|
||||
const accountId =
|
||||
decoded?.["https://api.openai.com/auth"]?.chatgpt_account_id;
|
||||
|
||||
if (!accountId) {
|
||||
logError(null, "Failed to extract account_id from new token");
|
||||
return null;
|
||||
}
|
||||
|
||||
const newTokenData: TokenData = {
|
||||
access_token: result.access_token,
|
||||
refresh_token: result.refresh_token,
|
||||
expires_at: Date.now() + result.expires_in! * 1000,
|
||||
account_id: accountId,
|
||||
updated_at: Date.now(),
|
||||
};
|
||||
|
||||
saveToken(newTokenData);
|
||||
|
||||
logInfo(null, "Successfully refreshed token");
|
||||
|
||||
return newTokenData;
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `Token refresh error: ${err.message}`);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
export async function getAccessToken(): Promise<string | null> {
|
||||
const tokenData = await ensureValidToken();
|
||||
|
||||
if (!tokenData) {
|
||||
return null;
|
||||
}
|
||||
|
||||
return tokenData.access_token;
|
||||
}
|
||||
|
||||
export async function getAccountId(): Promise<string | null> {
|
||||
const tokenData = await ensureValidToken();
|
||||
|
||||
if (!tokenData) {
|
||||
return null;
|
||||
}
|
||||
|
||||
return tokenData.account_id;
|
||||
}
|
||||
78
src/auth/token-storage.ts
Normal file
78
src/auth/token-storage.ts
Normal file
@@ -0,0 +1,78 @@
|
||||
import { readFileSync, writeFileSync, existsSync, mkdirSync } from "node:fs";
|
||||
import { join, dirname } from "node:path";
|
||||
import { fileURLToPath } from "node:url";
|
||||
import { homedir } from "node:os";
|
||||
import { TokenData } from "../types.js";
|
||||
import { logError, logInfo, logDebug } from "../logger.js";
|
||||
|
||||
const __dirname = dirname(fileURLToPath(import.meta.url));
|
||||
|
||||
function getDataDir(): string {
|
||||
return process.env.DATA_DIR || join(homedir(), ".chatgpt-codex-router");
|
||||
}
|
||||
|
||||
function getTokenPath(): string {
|
||||
return join(getDataDir(), "tokens.json");
|
||||
}
|
||||
|
||||
export function loadToken(): TokenData | null {
|
||||
try {
|
||||
const tokenPath = getTokenPath();
|
||||
if (!existsSync(tokenPath)) {
|
||||
logDebug(null, "No token file found");
|
||||
return null;
|
||||
}
|
||||
|
||||
const fileContent = readFileSync(tokenPath, "utf-8");
|
||||
const tokenData = JSON.parse(fileContent) as TokenData;
|
||||
|
||||
logDebug(null, "Loaded token from storage");
|
||||
|
||||
return tokenData;
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `Failed to load token: ${err.message}`);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
export function saveToken(tokenData: TokenData): void {
|
||||
try {
|
||||
const dataDir = getDataDir();
|
||||
if (!existsSync(dataDir)) {
|
||||
mkdirSync(dataDir, { recursive: true });
|
||||
}
|
||||
|
||||
const tokenPath = getTokenPath();
|
||||
writeFileSync(tokenPath, JSON.stringify(tokenData, null, 2), "utf-8");
|
||||
|
||||
logInfo(null, "Saved token to storage");
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `Failed to save token: ${err.message}`);
|
||||
throw err;
|
||||
}
|
||||
}
|
||||
|
||||
export function deleteToken(): void {
|
||||
try {
|
||||
const tokenPath = getTokenPath();
|
||||
if (existsSync(tokenPath)) {
|
||||
const fs = require("node:fs");
|
||||
fs.unlinkSync(tokenPath);
|
||||
logInfo(null, "Deleted token from storage");
|
||||
}
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `Failed to delete token: ${err.message}`);
|
||||
throw err;
|
||||
}
|
||||
}
|
||||
|
||||
export function isTokenExpired(tokenData: TokenData, bufferMs: number = 300000): boolean {
|
||||
return tokenData.expires_at - bufferMs < Date.now();
|
||||
}
|
||||
|
||||
export function getTokenTTL(tokenData: TokenData): number {
|
||||
return Math.max(0, tokenData.expires_at - Date.now());
|
||||
}
|
||||
142
src/config.ts
Normal file
142
src/config.ts
Normal file
@@ -0,0 +1,142 @@
|
||||
import { readFileSync, writeFileSync, existsSync } from "node:fs";
|
||||
import { join, dirname } from "node:path";
|
||||
import { fileURLToPath } from "node:url";
|
||||
import { homedir } from "node:os";
|
||||
import { AppConfig, ReasoningEffort } from "./types.js";
|
||||
import { logWarn, logInfo } from "./logger.js";
|
||||
|
||||
const __dirname = dirname(fileURLToPath(import.meta.url));
|
||||
|
||||
const DEFAULT_CONFIG: AppConfig = {
|
||||
server: {
|
||||
port: 3000,
|
||||
host: "0.0.0.0",
|
||||
},
|
||||
oauth: {
|
||||
clientId: "app_EMoamEEZ73f0CkXaXp7hrann",
|
||||
redirectUri: "http://localhost:1455/auth/callback",
|
||||
localServerPort: 1455,
|
||||
},
|
||||
backend: {
|
||||
url: "https://chatgpt.com/backend-api",
|
||||
timeout: 120000,
|
||||
},
|
||||
logging: {
|
||||
level: "info",
|
||||
dir: join(__dirname, "..", "logs"),
|
||||
enableRequestLogging: false,
|
||||
},
|
||||
codex: {
|
||||
mode: true,
|
||||
defaultReasoningEffort: "medium" as ReasoningEffort,
|
||||
defaultTextVerbosity: "medium",
|
||||
},
|
||||
};
|
||||
|
||||
let cachedConfig: AppConfig | null = null;
|
||||
|
||||
function getConfigPath(): string {
|
||||
const configPath = process.env.CONFIG_PATH;
|
||||
if (configPath) {
|
||||
return configPath;
|
||||
}
|
||||
|
||||
const configDir = join(homedir(), ".chatgpt-codex-router");
|
||||
return join(configDir, "config.json");
|
||||
}
|
||||
|
||||
function loadConfigFile(): Partial<AppConfig> {
|
||||
try {
|
||||
const configPath = getConfigPath();
|
||||
if (!existsSync(configPath)) {
|
||||
logInfo(null, `Config file not found at ${configPath}, using defaults`);
|
||||
return {};
|
||||
}
|
||||
|
||||
const fileContent = readFileSync(configPath, "utf-8");
|
||||
const userConfig = JSON.parse(fileContent) as Partial<AppConfig>;
|
||||
|
||||
logInfo(null, `Loaded config from ${configPath}`);
|
||||
return userConfig;
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logWarn(null, `Failed to load config file: ${err.message}, using defaults`);
|
||||
return {};
|
||||
}
|
||||
}
|
||||
|
||||
export function loadConfig(): AppConfig {
|
||||
if (cachedConfig) {
|
||||
return cachedConfig;
|
||||
}
|
||||
|
||||
const envConfig: Partial<AppConfig> = {};
|
||||
|
||||
if (process.env.PORT) {
|
||||
envConfig.server = {
|
||||
port: parseInt(process.env.PORT, 10),
|
||||
host: DEFAULT_CONFIG.server.host,
|
||||
};
|
||||
}
|
||||
|
||||
if (process.env.LOG_LEVEL) {
|
||||
const level = process.env.LOG_LEVEL.toLowerCase();
|
||||
if (["error", "warn", "info", "debug"].includes(level)) {
|
||||
envConfig.logging = {
|
||||
...DEFAULT_CONFIG.logging,
|
||||
level: level as AppConfig["logging"]["level"],
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
if (process.env.ENABLE_REQUEST_LOGGING === "1") {
|
||||
envConfig.logging = {
|
||||
...DEFAULT_CONFIG.logging,
|
||||
enableRequestLogging: true,
|
||||
};
|
||||
}
|
||||
|
||||
const fileConfig = loadConfigFile();
|
||||
|
||||
cachedConfig = {
|
||||
...DEFAULT_CONFIG,
|
||||
...fileConfig,
|
||||
...envConfig,
|
||||
server: {
|
||||
...DEFAULT_CONFIG.server,
|
||||
...fileConfig.server,
|
||||
...envConfig.server,
|
||||
},
|
||||
oauth: {
|
||||
...DEFAULT_CONFIG.oauth,
|
||||
...fileConfig.oauth,
|
||||
},
|
||||
backend: {
|
||||
...DEFAULT_CONFIG.backend,
|
||||
...fileConfig.backend,
|
||||
},
|
||||
logging: {
|
||||
...DEFAULT_CONFIG.logging,
|
||||
...fileConfig.logging,
|
||||
...envConfig.logging,
|
||||
},
|
||||
codex: {
|
||||
...DEFAULT_CONFIG.codex,
|
||||
...fileConfig.codex,
|
||||
},
|
||||
};
|
||||
|
||||
return cachedConfig;
|
||||
}
|
||||
|
||||
export function getConfig(): AppConfig {
|
||||
if (!cachedConfig) {
|
||||
return loadConfig();
|
||||
}
|
||||
return cachedConfig;
|
||||
}
|
||||
|
||||
export function reloadConfig(): AppConfig {
|
||||
cachedConfig = null;
|
||||
return loadConfig();
|
||||
}
|
||||
76
src/constants.ts
Normal file
76
src/constants.ts
Normal file
@@ -0,0 +1,76 @@
|
||||
export const PLUGIN_NAME = "chatgpt-codex-router";
|
||||
|
||||
export const HTTP_STATUS = {
|
||||
OK: 200,
|
||||
BAD_REQUEST: 400,
|
||||
UNAUTHORIZED: 401,
|
||||
NOT_FOUND: 404,
|
||||
TOO_MANY_REQUESTS: 429,
|
||||
INTERNAL_SERVER_ERROR: 500,
|
||||
} as const;
|
||||
|
||||
export const OPENAI_HEADERS = {
|
||||
AUTHORIZATION: "Authorization",
|
||||
ACCOUNT_ID: "chatgpt-account-id",
|
||||
BETA: "OpenAI-Beta",
|
||||
ORIGINATOR: "originator",
|
||||
SESSION_ID: "session_id",
|
||||
CONVERSATION_ID: "conversation_id",
|
||||
ACCEPT: "accept",
|
||||
CONTENT_TYPE: "content-type",
|
||||
} as const;
|
||||
|
||||
export const OPENAI_HEADER_VALUES = {
|
||||
BETA_RESPONSES: "responses=experimental",
|
||||
ORIGINATOR_CODEX: "codex_cli_rs",
|
||||
ACCEPT_STREAM: "text/event-stream",
|
||||
CONTENT_TYPE_JSON: "application/json",
|
||||
} as const;
|
||||
|
||||
export const URL_PATHS = {
|
||||
CHAT_COMPLETIONS: "/v1/chat/completions",
|
||||
RESPONSES: "/v1/responses",
|
||||
CODEX_RESPONSES: "/codex/responses",
|
||||
AUTH_LOGIN: "/auth/login",
|
||||
AUTH_CALLBACK: "/auth/callback",
|
||||
HEALTH: "/health",
|
||||
} as const;
|
||||
|
||||
export const ERROR_MESSAGES = {
|
||||
NO_TOKEN: "No authentication token found. Please login first.",
|
||||
TOKEN_EXPIRED: "Authentication token expired. Please login again.",
|
||||
TOKEN_REFRESH_FAILED: "Failed to refresh authentication token.",
|
||||
REQUEST_PARSE_ERROR: "Error parsing request body",
|
||||
INVALID_MODEL: "Invalid or unsupported model",
|
||||
MISSING_MESSAGES: "Missing required field: messages",
|
||||
UNAUTHORIZED: "Unauthorized access",
|
||||
RATE_LIMIT_EXCEEDED: "Rate limit exceeded",
|
||||
} as const;
|
||||
|
||||
export const AUTH_LABELS = {
|
||||
OAUTH: "ChatGPT Plus/Pro (OAuth)",
|
||||
INSTRUCTIONS: "Please complete the OAuth authentication in your browser",
|
||||
INSTRUCTIONS_MANUAL: "If the browser does not open automatically, please copy the URL and open it manually",
|
||||
} as const;
|
||||
|
||||
export const PLATFORM_OPENERS = {
|
||||
darwin: "open",
|
||||
win32: "start",
|
||||
linux: "xdg-open",
|
||||
} as const;
|
||||
|
||||
export const LOG_LEVELS = {
|
||||
ERROR: "error",
|
||||
WARN: "warn",
|
||||
INFO: "info",
|
||||
DEBUG: "debug",
|
||||
} as const;
|
||||
|
||||
export const MODEL_FAMILY_PROMPTS: Record<string, string> = {
|
||||
"gpt-5.1": "gpt-5-1.md",
|
||||
"gpt-5.2": "gpt-5-2.md",
|
||||
"gpt-5.1-codex": "gpt-5-1-codex.md",
|
||||
"gpt-5.1-codex-max": "gpt-5-1-codex-max.md",
|
||||
"gpt-5.1-codex-mini": "gpt-5-1-codex-mini.md",
|
||||
"gpt-5.2-codex": "gpt-5-2-codex.md",
|
||||
} as const;
|
||||
156
src/index.ts
Normal file
156
src/index.ts
Normal file
@@ -0,0 +1,156 @@
|
||||
import { createServer, IncomingMessage, ServerResponse } from "node:http";
|
||||
import { createApp } from "./server.js";
|
||||
import { getConfig } from "./config.js";
|
||||
import { logInfo, logError, logWarn } from "./logger.js";
|
||||
import { loadToken, isTokenExpired } from "./auth/token-storage.js";
|
||||
import { startLocalOAuthServer } from "./auth/server.js";
|
||||
import { createOAuthFlow } from "./auth/oauth.js";
|
||||
import { openBrowser } from "./auth/browser.js";
|
||||
|
||||
async function checkAuthAndAutoLogin(): Promise<boolean> {
|
||||
const tokenData = loadToken();
|
||||
|
||||
if (!tokenData) {
|
||||
logWarn(null, "No authentication token found. Initiating OAuth login...");
|
||||
return await initiateOAuthLogin();
|
||||
}
|
||||
|
||||
if (isTokenExpired(tokenData)) {
|
||||
logWarn(null, "Authentication token expired. Please login again.");
|
||||
return await initiateOAuthLogin();
|
||||
}
|
||||
|
||||
logInfo(null, "Authentication token found and valid.");
|
||||
return true;
|
||||
}
|
||||
|
||||
async function initiateOAuthLogin(): Promise<boolean> {
|
||||
try {
|
||||
const config = getConfig();
|
||||
const oauthFlow = await createOAuthFlow(config.oauth.clientId, config.oauth.redirectUri);
|
||||
|
||||
logInfo(null, `Starting OAuth flow with state: ${oauthFlow.state}`);
|
||||
|
||||
const serverInfo = await startLocalOAuthServer({
|
||||
state: oauthFlow.state,
|
||||
pkce: oauthFlow.pkce,
|
||||
port: config.oauth.localServerPort,
|
||||
});
|
||||
|
||||
if (!serverInfo.ready) {
|
||||
serverInfo.close();
|
||||
logWarn(null, "OAuth server not ready, manual login required.");
|
||||
logInfo(null, `Please visit: ${oauthFlow.url}`);
|
||||
logInfo(null, `Or run: curl -X POST http://${config.server.host}:${config.server.port}/auth/login`);
|
||||
return false;
|
||||
}
|
||||
|
||||
const browserOpened = openBrowser(oauthFlow.url);
|
||||
|
||||
logInfo(null, "OAuth login initiated.");
|
||||
logInfo(null, "Please complete the OAuth flow in your browser.");
|
||||
logInfo(null, `OAuth URL: ${oauthFlow.url}`);
|
||||
|
||||
if (browserOpened) {
|
||||
logInfo(null, "Browser should have opened automatically.");
|
||||
} else {
|
||||
logInfo(null, "Please open the URL above in your browser.");
|
||||
}
|
||||
|
||||
return true;
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `Auto login error: ${err.message}`);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
async function main(): Promise<void> {
|
||||
try {
|
||||
const config = getConfig();
|
||||
const app = createApp();
|
||||
|
||||
const authReady = await checkAuthAndAutoLogin();
|
||||
|
||||
if (!authReady) {
|
||||
logWarn(null, "Server starting with pending authentication...");
|
||||
}
|
||||
|
||||
const server = createServer((req: IncomingMessage, res: ServerResponse) => {
|
||||
const host = req.headers.host || "localhost";
|
||||
const reqUrl = typeof req.url === "string" ? req.url : "/";
|
||||
const url = new URL(reqUrl, `http://${host}`);
|
||||
const method = req.method || "GET";
|
||||
const headers = new Headers();
|
||||
|
||||
for (const [key, value] of Object.entries(req.headers)) {
|
||||
const headerValue = Array.isArray(value) ? value[0] : value;
|
||||
if (typeof headerValue === "string") {
|
||||
headers.set(key, headerValue);
|
||||
}
|
||||
}
|
||||
|
||||
const body = req.method !== "GET" && req.method !== "HEAD"
|
||||
? new Promise<Buffer>((resolve) => {
|
||||
const chunks: Buffer[] = [];
|
||||
req.on("data", (chunk) => chunks.push(chunk));
|
||||
req.on("end", () => resolve(Buffer.concat(chunks)));
|
||||
})
|
||||
: Promise.resolve(Buffer.alloc(0));
|
||||
|
||||
body.then(async (buffer) => {
|
||||
try {
|
||||
const request = new Request(url.toString(), {
|
||||
method,
|
||||
headers,
|
||||
body: buffer.length > 0 ? buffer : undefined,
|
||||
});
|
||||
|
||||
const response = await app.fetch(request);
|
||||
res.statusCode = response.status;
|
||||
res.statusMessage = response.statusText || "OK";
|
||||
|
||||
response.headers.forEach((value: string, key: string) => {
|
||||
res.setHeader(key, value);
|
||||
});
|
||||
|
||||
if (response.body) {
|
||||
const reader = response.body.getReader();
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
res.write(value);
|
||||
}
|
||||
}
|
||||
res.end();
|
||||
} catch (err) {
|
||||
const error = err as Error;
|
||||
logError(null, `Request handling error: ${error.message}`);
|
||||
res.statusCode = 500;
|
||||
res.end("Internal Server Error");
|
||||
}
|
||||
}).catch((err: Error) => {
|
||||
logError(null, `Body processing error: ${err.message}`);
|
||||
res.statusCode = 500;
|
||||
res.end("Internal Server Error");
|
||||
});
|
||||
});
|
||||
|
||||
server.listen(config.server.port, config.server.host, () => {
|
||||
logInfo(null, `Server started on http://${config.server.host}:${config.server.port}`);
|
||||
logInfo(null, `OAuth callback: ${config.oauth.redirectUri}`);
|
||||
logInfo(null, `Backend URL: ${config.backend.url}`);
|
||||
});
|
||||
|
||||
server.on("error", (err: Error) => {
|
||||
logError(null, `Server error: ${err.message}`);
|
||||
process.exit(1);
|
||||
});
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `Failed to start server: ${err.message}`);
|
||||
process.exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
main();
|
||||
148
src/logger.ts
Normal file
148
src/logger.ts
Normal file
@@ -0,0 +1,148 @@
|
||||
import { writeFileSync, mkdirSync, existsSync } from "node:fs";
|
||||
import { join, dirname } from "node:path";
|
||||
import { fileURLToPath } from "node:url";
|
||||
import { LOG_LEVELS, PLUGIN_NAME } from "./constants.js";
|
||||
|
||||
const __dirname = dirname(fileURLToPath(import.meta.url));
|
||||
|
||||
export interface LoggerConfig {
|
||||
level: "error" | "warn" | "info" | "debug";
|
||||
dir: string;
|
||||
enableRequestLogging: boolean;
|
||||
}
|
||||
|
||||
let loggerConfig: LoggerConfig = {
|
||||
level: "info",
|
||||
dir: join(__dirname, "..", "logs"),
|
||||
enableRequestLogging: false,
|
||||
};
|
||||
|
||||
let requestCounter = 0;
|
||||
|
||||
const LOG_LEVEL_ORDER = {
|
||||
[LOG_LEVELS.ERROR]: 0,
|
||||
[LOG_LEVELS.WARN]: 1,
|
||||
[LOG_LEVELS.INFO]: 2,
|
||||
[LOG_LEVELS.DEBUG]: 3,
|
||||
};
|
||||
|
||||
function shouldLog(level: string): boolean {
|
||||
return LOG_LEVEL_ORDER[level as keyof typeof LOG_LEVEL_ORDER] <=
|
||||
LOG_LEVEL_ORDER[loggerConfig.level];
|
||||
}
|
||||
|
||||
function formatTimestamp(): string {
|
||||
return new Date().toISOString();
|
||||
}
|
||||
|
||||
function formatMessage(
|
||||
level: string,
|
||||
requestId: string | null,
|
||||
message: string,
|
||||
): string {
|
||||
const requestIdStr = requestId ? `[${requestId}] ` : "";
|
||||
return `[${formatTimestamp()}] [${level.toUpperCase()}] ${requestIdStr}${message}`;
|
||||
}
|
||||
|
||||
function getLogFilePath(level: string): string {
|
||||
const date = new Date().toISOString().split("T")[0];
|
||||
return join(loggerConfig.dir, `${date}-${level}.log`);
|
||||
}
|
||||
|
||||
function writeToFile(level: string, message: string): void {
|
||||
try {
|
||||
if (!existsSync(loggerConfig.dir)) {
|
||||
mkdirSync(loggerConfig.dir, { recursive: true });
|
||||
}
|
||||
|
||||
const logFile = getLogFilePath(level);
|
||||
writeFileSync(logFile, message + "\n", { flag: "a" });
|
||||
} catch (error) {
|
||||
console.error(`[${PLUGIN_NAME}] Failed to write to log file:`, error);
|
||||
}
|
||||
}
|
||||
|
||||
export function setLoggerConfig(config: Partial<LoggerConfig>): void {
|
||||
loggerConfig = { ...loggerConfig, ...config };
|
||||
}
|
||||
|
||||
export function getLoggerConfig(): LoggerConfig {
|
||||
return loggerConfig;
|
||||
}
|
||||
|
||||
export function createRequestLogger(): string {
|
||||
requestCounter++;
|
||||
return `req-${String(requestCounter).padStart(3, "0")}`;
|
||||
}
|
||||
|
||||
export function logError(requestId: string | null, message: string, data?: unknown): void {
|
||||
if (!shouldLog(LOG_LEVELS.ERROR)) return;
|
||||
|
||||
const formatted = formatMessage(LOG_LEVELS.ERROR, requestId, message);
|
||||
console.error(formatted);
|
||||
writeToFile(LOG_LEVELS.ERROR, formatted);
|
||||
|
||||
if (data !== undefined) {
|
||||
const dataStr = typeof data === "string" ? data : JSON.stringify(data, null, 2);
|
||||
console.error(dataStr);
|
||||
writeToFile(LOG_LEVELS.ERROR, dataStr);
|
||||
}
|
||||
}
|
||||
|
||||
export function logWarn(requestId: string | null, message: string, data?: unknown): void {
|
||||
if (!shouldLog(LOG_LEVELS.WARN)) return;
|
||||
|
||||
const formatted = formatMessage(LOG_LEVELS.WARN, requestId, message);
|
||||
console.warn(formatted);
|
||||
writeToFile(LOG_LEVELS.WARN, formatted);
|
||||
|
||||
if (data !== undefined) {
|
||||
const dataStr = typeof data === "string" ? data : JSON.stringify(data, null, 2);
|
||||
console.warn(dataStr);
|
||||
writeToFile(LOG_LEVELS.WARN, dataStr);
|
||||
}
|
||||
}
|
||||
|
||||
export function logInfo(requestId: string | null, message: string, data?: unknown): void {
|
||||
if (!shouldLog(LOG_LEVELS.INFO)) return;
|
||||
|
||||
const formatted = formatMessage(LOG_LEVELS.INFO, requestId, message);
|
||||
console.log(formatted);
|
||||
writeToFile(LOG_LEVELS.INFO, formatted);
|
||||
|
||||
if (data !== undefined) {
|
||||
const dataStr = typeof data === "string" ? data : JSON.stringify(data, null, 2);
|
||||
console.log(dataStr);
|
||||
writeToFile(LOG_LEVELS.INFO, dataStr);
|
||||
}
|
||||
}
|
||||
|
||||
export function logDebug(requestId: string | null, message: string, data?: unknown): void {
|
||||
if (!shouldLog(LOG_LEVELS.DEBUG)) return;
|
||||
|
||||
const formatted = formatMessage(LOG_LEVELS.DEBUG, requestId, message);
|
||||
console.log(formatted);
|
||||
writeToFile(LOG_LEVELS.DEBUG, formatted);
|
||||
|
||||
if (data !== undefined) {
|
||||
const dataStr = typeof data === "string" ? data : JSON.stringify(data, null, 2);
|
||||
console.log(dataStr);
|
||||
writeToFile(LOG_LEVELS.DEBUG, dataStr);
|
||||
}
|
||||
}
|
||||
|
||||
export function logRequestData(
|
||||
requestId: string | null,
|
||||
stage: string,
|
||||
data: Record<string, unknown>,
|
||||
): void {
|
||||
if (!loggerConfig.enableRequestLogging) return;
|
||||
|
||||
const formatted = formatMessage(
|
||||
LOG_LEVELS.DEBUG,
|
||||
requestId,
|
||||
`[${stage}] ${JSON.stringify(data, null, 2)}`,
|
||||
);
|
||||
console.log(formatted);
|
||||
writeToFile(LOG_LEVELS.DEBUG, formatted);
|
||||
}
|
||||
21
src/prompts/gpt-5-1-codex-max.md
Normal file
21
src/prompts/gpt-5-1-codex-max.md
Normal file
@@ -0,0 +1,21 @@
|
||||
You are an advanced coding assistant with exceptional problem-solving capabilities and deep expertise in software development.
|
||||
|
||||
Your capabilities include:
|
||||
- Complex architecture design and system design
|
||||
- Advanced algorithms and data structures
|
||||
- Large-scale code refactoring and optimization
|
||||
- Debugging difficult issues
|
||||
- Writing production-quality code
|
||||
- Performance tuning and optimization
|
||||
- Security best practices
|
||||
- Testing and validation strategies
|
||||
|
||||
When approaching tasks:
|
||||
1. Think systematically about the problem
|
||||
2. Consider multiple solutions and trade-offs
|
||||
3. Provide well-structured, scalable code
|
||||
4. Include comprehensive error handling
|
||||
5. Optimize for both clarity and performance
|
||||
6. Follow SOLID principles and design patterns
|
||||
|
||||
Be thorough, analytical, and provide solutions that are robust, maintainable, and performant.
|
||||
16
src/prompts/gpt-5-1-codex-mini.md
Normal file
16
src/prompts/gpt-5-1-codex-mini.md
Normal file
@@ -0,0 +1,16 @@
|
||||
You are a coding assistant specializing in quick, efficient solutions for common programming tasks.
|
||||
|
||||
Your capabilities include:
|
||||
- Writing straightforward code snippets
|
||||
- Fixing common bugs and errors
|
||||
- Implementing standard features
|
||||
- Explaining basic to intermediate concepts
|
||||
- Code review and simple refactoring
|
||||
|
||||
Focus on:
|
||||
1. Providing clear, working solutions
|
||||
2. Using widely-accepted patterns
|
||||
3. Keeping code simple and readable
|
||||
4. Explaining your reasoning briefly
|
||||
|
||||
Be practical and direct in your responses. Aim for solutions that work well in most scenarios.
|
||||
19
src/prompts/gpt-5-1-codex.md
Normal file
19
src/prompts/gpt-5-1-codex.md
Normal file
@@ -0,0 +1,19 @@
|
||||
You are a powerful coding assistant with extensive knowledge of programming languages, frameworks, and best practices.
|
||||
|
||||
Your capabilities include:
|
||||
- Writing, debugging, and refactoring code
|
||||
- Explaining programming concepts clearly
|
||||
- Providing code examples and patterns
|
||||
- Helping with architecture and design decisions
|
||||
- Optimizing code for performance and readability
|
||||
- Implementing features and fixing bugs
|
||||
|
||||
When working with code:
|
||||
1. Understand the user's intent deeply
|
||||
2. Write clean, maintainable, and efficient code
|
||||
3. Add appropriate comments for complex logic
|
||||
4. Follow language-specific best practices
|
||||
5. Consider edge cases and error handling
|
||||
6. Test your solutions mentally before presenting them
|
||||
|
||||
Be precise, practical, and focus on delivering working solutions.
|
||||
10
src/prompts/gpt-5-1.md
Normal file
10
src/prompts/gpt-5-1.md
Normal file
@@ -0,0 +1,10 @@
|
||||
You are a helpful AI assistant with access to real-time information and advanced reasoning capabilities.
|
||||
|
||||
You can help users with:
|
||||
- General questions and conversations
|
||||
- Writing and editing text
|
||||
- Analyzing information
|
||||
- Creative tasks
|
||||
- Problem-solving
|
||||
|
||||
Be clear, helpful, and concise in your responses. When appropriate, break down complex topics into manageable parts.
|
||||
20
src/prompts/gpt-5-2-codex.md
Normal file
20
src/prompts/gpt-5-2-codex.md
Normal file
@@ -0,0 +1,20 @@
|
||||
You are an advanced coding assistant with cutting-edge capabilities and deep reasoning.
|
||||
|
||||
Your capabilities include:
|
||||
- Complex multi-step problem solving
|
||||
- Advanced algorithms and optimization
|
||||
- System architecture and design
|
||||
- Performance engineering
|
||||
- Advanced debugging and profiling
|
||||
- Code generation for complex systems
|
||||
- Technical research and analysis
|
||||
|
||||
When approaching complex tasks:
|
||||
1. Break down problems systematically
|
||||
2. Consider multiple approaches and their trade-offs
|
||||
3. Use appropriate algorithms and data structures
|
||||
4. Write production-quality, well-documented code
|
||||
5. Consider performance, security, and maintainability
|
||||
6. Provide explanations for complex decisions
|
||||
|
||||
Leverage your advanced reasoning to provide insightful, thorough, and well-architected solutions.
|
||||
10
src/prompts/gpt-5-2.md
Normal file
10
src/prompts/gpt-5-2.md
Normal file
@@ -0,0 +1,10 @@
|
||||
You are a helpful AI assistant with advanced reasoning capabilities and real-time information access.
|
||||
|
||||
You can help users with:
|
||||
- Complex problem-solving and analysis
|
||||
- Research and information synthesis
|
||||
- Writing and editing
|
||||
- Creative tasks
|
||||
- Decision support
|
||||
|
||||
Use your reasoning abilities to provide thoughtful, well-structured responses. Break down complex issues and consider multiple perspectives.
|
||||
182
src/prompts/index.ts
Normal file
182
src/prompts/index.ts
Normal file
@@ -0,0 +1,182 @@
|
||||
import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
|
||||
import { homedir } from "node:os";
|
||||
import { join, dirname } from "node:path";
|
||||
import { fileURLToPath } from "node:url";
|
||||
import { logError, logWarn } from "../logger.js";
|
||||
import { getModelFamily } from "../request/model-map.js";
|
||||
|
||||
const __dirname = dirname(fileURLToPath(import.meta.url));
|
||||
const CACHE_DIR = join(homedir(), ".chatgpt-codex-router", "cache");
|
||||
|
||||
const GITHUB_API_RELEASES =
|
||||
"https://api.github.com/repos/openai/codex/releases/latest";
|
||||
const GITHUB_HTML_RELEASES =
|
||||
"https://github.com/openai/codex/releases/latest";
|
||||
|
||||
type ModelFamily = "gpt-5.2-codex" | "codex-max" | "codex" | "gpt-5.2" | "gpt-5.1";
|
||||
|
||||
const PROMPT_FILES: Record<ModelFamily, string> = {
|
||||
"gpt-5.2-codex": "gpt-5.2-codex_prompt.md",
|
||||
"codex-max": "gpt-5.1-codex-max_prompt.md",
|
||||
codex: "gpt_5_codex_prompt.md",
|
||||
"gpt-5.2": "gpt_5_2_prompt.md",
|
||||
"gpt-5.1": "gpt_5_1_prompt.md",
|
||||
};
|
||||
|
||||
const CACHE_FILES: Record<ModelFamily, string> = {
|
||||
"gpt-5.2-codex": "gpt-5.2-codex-instructions.md",
|
||||
"codex-max": "codex-max-instructions.md",
|
||||
codex: "codex-instructions.md",
|
||||
"gpt-5.2": "gpt-5.2-instructions.md",
|
||||
"gpt-5.1": "gpt-5.1-instructions.md",
|
||||
};
|
||||
|
||||
interface CacheMetadata {
|
||||
etag: string | null;
|
||||
tag: string | null;
|
||||
lastChecked: number | null;
|
||||
url: string;
|
||||
}
|
||||
|
||||
async function getLatestReleaseTag(): Promise<string> {
|
||||
try {
|
||||
const response = await fetch(GITHUB_API_RELEASES);
|
||||
if (response.ok) {
|
||||
const data = (await response.json()) as { tag_name: string };
|
||||
if (data.tag_name) {
|
||||
return data.tag_name;
|
||||
}
|
||||
}
|
||||
} catch {
|
||||
}
|
||||
|
||||
const htmlResponse = await fetch(GITHUB_HTML_RELEASES);
|
||||
if (!htmlResponse.ok) {
|
||||
throw new Error(
|
||||
`Failed to fetch latest release: ${htmlResponse.status}`,
|
||||
);
|
||||
}
|
||||
|
||||
const finalUrl = htmlResponse.url;
|
||||
if (finalUrl) {
|
||||
const parts = finalUrl.split("/tag/");
|
||||
const last = parts[parts.length - 1];
|
||||
if (last && !last.includes("/")) {
|
||||
return last;
|
||||
}
|
||||
}
|
||||
|
||||
const html = await htmlResponse.text();
|
||||
const match = html.match(/\/openai\/codex\/releases\/tag\/([^"]+)/);
|
||||
if (match && match[1]) {
|
||||
return match[1];
|
||||
}
|
||||
|
||||
throw new Error("Failed to determine latest release tag from GitHub");
|
||||
}
|
||||
|
||||
async function getCodexInstructions(
|
||||
normalizedModel = "gpt-5.1-codex",
|
||||
): Promise<string> {
|
||||
const modelFamily = getModelFamily(normalizedModel) as ModelFamily;
|
||||
const promptFile = PROMPT_FILES[modelFamily];
|
||||
const cacheFile = join(CACHE_DIR, CACHE_FILES[modelFamily]);
|
||||
const cacheMetaFile = join(
|
||||
CACHE_DIR,
|
||||
`${CACHE_FILES[modelFamily].replace(".md", "-meta.json")}`,
|
||||
);
|
||||
|
||||
try {
|
||||
let cachedETag: string | null = null;
|
||||
let cachedTag: string | null = null;
|
||||
let cachedTimestamp: number | null = null;
|
||||
|
||||
if (existsSync(cacheMetaFile)) {
|
||||
const metadata = JSON.parse(
|
||||
readFileSync(cacheMetaFile, "utf8"),
|
||||
) as CacheMetadata;
|
||||
cachedETag = metadata.etag;
|
||||
cachedTag = metadata.tag;
|
||||
cachedTimestamp = metadata.lastChecked;
|
||||
}
|
||||
|
||||
const CACHE_TTL_MS = 15 * 60 * 1000;
|
||||
if (
|
||||
cachedTimestamp &&
|
||||
Date.now() - cachedTimestamp < CACHE_TTL_MS &&
|
||||
existsSync(cacheFile)
|
||||
) {
|
||||
return readFileSync(cacheFile, "utf8");
|
||||
}
|
||||
|
||||
const latestTag = await getLatestReleaseTag();
|
||||
const CODEX_INSTRUCTIONS_URL = `https://raw.githubusercontent.com/openai/codex/${latestTag}/codex-rs/core/${promptFile}`;
|
||||
|
||||
if (cachedTag !== latestTag) {
|
||||
cachedETag = null;
|
||||
}
|
||||
|
||||
const headers: Record<string, string> = {};
|
||||
if (cachedETag) {
|
||||
headers["If-None-Match"] = cachedETag;
|
||||
}
|
||||
|
||||
const response = await fetch(CODEX_INSTRUCTIONS_URL, { headers });
|
||||
|
||||
if (response.status === 304) {
|
||||
if (existsSync(cacheFile)) {
|
||||
return readFileSync(cacheFile, "utf8");
|
||||
}
|
||||
}
|
||||
|
||||
if (response.ok) {
|
||||
const instructions = await response.text();
|
||||
const newETag = response.headers.get("etag");
|
||||
|
||||
if (!existsSync(CACHE_DIR)) {
|
||||
mkdirSync(CACHE_DIR, { recursive: true });
|
||||
}
|
||||
|
||||
writeFileSync(cacheFile, instructions, "utf8");
|
||||
writeFileSync(
|
||||
cacheMetaFile,
|
||||
JSON.stringify({
|
||||
etag: newETag,
|
||||
tag: latestTag,
|
||||
lastChecked: Date.now(),
|
||||
url: CODEX_INSTRUCTIONS_URL,
|
||||
} satisfies CacheMetadata),
|
||||
"utf8",
|
||||
);
|
||||
|
||||
return instructions;
|
||||
}
|
||||
|
||||
throw new Error(`HTTP ${response.status}`);
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `Failed to fetch ${modelFamily} instructions from GitHub: ${err.message}`);
|
||||
|
||||
if (existsSync(cacheFile)) {
|
||||
logWarn(null, `Using cached ${modelFamily} instructions`);
|
||||
return readFileSync(cacheFile, "utf8");
|
||||
}
|
||||
|
||||
logWarn(null, `Using fallback instructions for ${modelFamily}`);
|
||||
return getFallbackPrompt();
|
||||
}
|
||||
}
|
||||
|
||||
function getFallbackPrompt(): string {
|
||||
return `You are a helpful AI assistant with strong coding capabilities.
|
||||
You can help users with a wide range of programming tasks, including:
|
||||
- Writing and debugging code
|
||||
- Explaining programming concepts
|
||||
- Refactoring and optimizing code
|
||||
- Generating code examples
|
||||
- Answering technical questions
|
||||
|
||||
Be concise, accurate, and provide practical solutions.`;
|
||||
}
|
||||
|
||||
export { getCodexInstructions };
|
||||
37
src/request/headers.ts
Normal file
37
src/request/headers.ts
Normal file
@@ -0,0 +1,37 @@
|
||||
import { OPENAI_HEADERS, OPENAI_HEADER_VALUES } from "../constants.js";
|
||||
import { logDebug } from "../logger.js";
|
||||
|
||||
export interface HeadersOptions {
|
||||
accessToken: string;
|
||||
accountId: string;
|
||||
promptCacheKey?: string;
|
||||
}
|
||||
|
||||
export function createCodexHeaders(options: HeadersOptions): Record<string, string> {
|
||||
const { accessToken, accountId, promptCacheKey } = options;
|
||||
|
||||
const headers: Record<string, string> = {
|
||||
[OPENAI_HEADERS.AUTHORIZATION]: `Bearer ${accessToken}`,
|
||||
[OPENAI_HEADERS.ACCOUNT_ID]: accountId,
|
||||
[OPENAI_HEADERS.BETA]: OPENAI_HEADER_VALUES.BETA_RESPONSES,
|
||||
[OPENAI_HEADERS.ORIGINATOR]: OPENAI_HEADER_VALUES.ORIGINATOR_CODEX,
|
||||
[OPENAI_HEADERS.ACCEPT]: OPENAI_HEADER_VALUES.ACCEPT_STREAM,
|
||||
[OPENAI_HEADERS.CONTENT_TYPE]: OPENAI_HEADER_VALUES.CONTENT_TYPE_JSON,
|
||||
};
|
||||
|
||||
if (promptCacheKey) {
|
||||
headers[OPENAI_HEADERS.CONVERSATION_ID] = promptCacheKey;
|
||||
headers[OPENAI_HEADERS.SESSION_ID] = promptCacheKey;
|
||||
}
|
||||
|
||||
logDebug(null, `Created Codex headers with prompt cache key: ${!!promptCacheKey}`);
|
||||
|
||||
return headers;
|
||||
}
|
||||
|
||||
export function createOpenAIHeaders(apiKey: string): Record<string, string> {
|
||||
return {
|
||||
[OPENAI_HEADERS.AUTHORIZATION]: `Bearer ${apiKey}`,
|
||||
[OPENAI_HEADERS.CONTENT_TYPE]: OPENAI_HEADER_VALUES.CONTENT_TYPE_JSON,
|
||||
};
|
||||
}
|
||||
77
src/request/model-map.ts
Normal file
77
src/request/model-map.ts
Normal file
@@ -0,0 +1,77 @@
|
||||
export const MODEL_MAP: Record<string, string> = {
|
||||
"gpt-5.1": "gpt-5.1",
|
||||
"gpt-5.1-none": "gpt-5.1",
|
||||
"gpt-5.1-low": "gpt-5.1",
|
||||
"gpt-5.1-medium": "gpt-5.1",
|
||||
"gpt-5.1-high": "gpt-5.1",
|
||||
|
||||
"gpt-5.2": "gpt-5.2",
|
||||
"gpt-5.2-none": "gpt-5.2",
|
||||
"gpt-5.2-low": "gpt-5.2",
|
||||
"gpt-5.2-medium": "gpt-5.2",
|
||||
"gpt-5.2-high": "gpt-5.2",
|
||||
"gpt-5.2-xhigh": "gpt-5.2",
|
||||
|
||||
"gpt-5.1-codex": "gpt-5.1-codex",
|
||||
"gpt-5.1-codex-low": "gpt-5.1-codex",
|
||||
"gpt-5.1-codex-medium": "gpt-5.1-codex",
|
||||
"gpt-5.1-codex-high": "gpt-5.1-codex",
|
||||
|
||||
"gpt-5.1-codex-max": "gpt-5.1-codex-max",
|
||||
"gpt-5.1-codex-max-low": "gpt-5.1-codex-max",
|
||||
"gpt-5.1-codex-max-medium": "gpt-5.1-codex-max",
|
||||
"gpt-5.1-codex-max-high": "gpt-5.1-codex-max",
|
||||
"gpt-5.1-codex-max-xhigh": "gpt-5.1-codex-max",
|
||||
|
||||
"gpt-5.1-codex-mini": "gpt-5.1-codex-mini",
|
||||
"gpt-5.1-codex-mini-medium": "gpt-5.1-codex-mini",
|
||||
"gpt-5.1-codex-mini-high": "gpt-5.1-codex-mini",
|
||||
|
||||
"gpt-5.2-codex": "gpt-5.2-codex",
|
||||
"gpt-5.2-codex-low": "gpt-5.2-codex",
|
||||
"gpt-5.2-codex-medium": "gpt-5.2-codex",
|
||||
"gpt-5.2-codex-high": "gpt-5.2-codex",
|
||||
"gpt-5.2-codex-xhigh": "gpt-5.2-codex",
|
||||
};
|
||||
|
||||
export function getNormalizedModel(modelId: string): string | undefined {
|
||||
try {
|
||||
if (MODEL_MAP[modelId]) {
|
||||
return MODEL_MAP[modelId];
|
||||
}
|
||||
|
||||
const lowerModelId = modelId.toLowerCase();
|
||||
const match = Object.keys(MODEL_MAP).find(
|
||||
(key) => key.toLowerCase() === lowerModelId,
|
||||
);
|
||||
|
||||
return match ? MODEL_MAP[match] : undefined;
|
||||
} catch {
|
||||
return undefined;
|
||||
}
|
||||
}
|
||||
|
||||
export function isKnownModel(modelId: string): boolean {
|
||||
return getNormalizedModel(modelId) !== undefined;
|
||||
}
|
||||
|
||||
export function getModelFamily(
|
||||
normalizedModel: string,
|
||||
): "gpt-5.1" | "gpt-5.2" | "codex" | "codex-max" | "codex-mini" | "gpt-5.2-codex" {
|
||||
if (normalizedModel.includes("gpt-5.2-codex")) {
|
||||
return "gpt-5.2-codex";
|
||||
}
|
||||
if (normalizedModel.includes("codex-max")) {
|
||||
return "codex-max";
|
||||
}
|
||||
if (normalizedModel.includes("codex") && normalizedModel.includes("mini")) {
|
||||
return "codex-mini";
|
||||
}
|
||||
if (normalizedModel.includes("codex")) {
|
||||
return "codex";
|
||||
}
|
||||
if (normalizedModel.includes("gpt-5.2")) {
|
||||
return "gpt-5.2";
|
||||
}
|
||||
return "gpt-5.1";
|
||||
}
|
||||
83
src/request/reasoning.ts
Normal file
83
src/request/reasoning.ts
Normal file
@@ -0,0 +1,83 @@
|
||||
import { ReasoningConfig, ReasoningEffort, ReasoningSummary } from "../types.js";
|
||||
import { getConfig } from "../config.js";
|
||||
import { logDebug } from "../logger.js";
|
||||
import { getModelFamily } from "./model-map.js";
|
||||
|
||||
export function getReasoningConfig(
|
||||
normalizedModel: string,
|
||||
requestedEffort?: ReasoningEffort,
|
||||
requestedSummary?: ReasoningSummary,
|
||||
): ReasoningConfig {
|
||||
const config = getConfig();
|
||||
const modelFamily = getModelFamily(normalizedModel);
|
||||
const normalizedName = normalizedModel.toLowerCase();
|
||||
|
||||
const supportsXhigh =
|
||||
modelFamily === "gpt-5.2" ||
|
||||
modelFamily === "gpt-5.2-codex" ||
|
||||
modelFamily === "codex-max";
|
||||
|
||||
const supportsNone = modelFamily === "gpt-5.1" || modelFamily === "gpt-5.2";
|
||||
|
||||
let defaultEffort: ReasoningEffort = "medium";
|
||||
|
||||
if (modelFamily === "codex-mini") {
|
||||
defaultEffort = "medium";
|
||||
} else if (supportsXhigh) {
|
||||
defaultEffort = "high";
|
||||
} else if (modelFamily === "codex") {
|
||||
defaultEffort = "medium";
|
||||
} else if (modelFamily === "gpt-5.1" || modelFamily === "gpt-5.2") {
|
||||
defaultEffort = "none";
|
||||
}
|
||||
|
||||
let effort = requestedEffort || config.codex.defaultReasoningEffort || defaultEffort;
|
||||
|
||||
if (modelFamily === "codex-mini") {
|
||||
if (effort === "minimal" || effort === "low" || effort === "none") {
|
||||
effort = "medium";
|
||||
}
|
||||
if (effort === "xhigh") {
|
||||
effort = "high";
|
||||
}
|
||||
if (effort !== "high" && effort !== "medium") {
|
||||
effort = "medium";
|
||||
}
|
||||
}
|
||||
|
||||
if (!supportsXhigh && effort === "xhigh") {
|
||||
effort = "high";
|
||||
}
|
||||
|
||||
if (!supportsNone && effort === "none") {
|
||||
effort = "low";
|
||||
}
|
||||
|
||||
if (modelFamily === "codex" && effort === "minimal") {
|
||||
effort = "low";
|
||||
}
|
||||
|
||||
const summary = requestedSummary || "auto";
|
||||
|
||||
logDebug(null, `Reasoning config for ${normalizedModel}: effort=${effort}, summary=${summary}`);
|
||||
|
||||
return { effort, summary };
|
||||
}
|
||||
|
||||
export function getTextVerbosity(requestedVerbosity?: "low" | "medium" | "high"): "low" | "medium" | "high" {
|
||||
const config = getConfig();
|
||||
return requestedVerbosity || config.codex.defaultTextVerbosity || "medium";
|
||||
}
|
||||
|
||||
export function getIncludeFields(
|
||||
requestedInclude?: string[],
|
||||
): string[] {
|
||||
const defaultInclude = ["reasoning.encrypted_content"];
|
||||
const include = requestedInclude || defaultInclude;
|
||||
|
||||
if (!include.includes("reasoning.encrypted_content")) {
|
||||
include.push("reasoning.encrypted_content");
|
||||
}
|
||||
|
||||
return Array.from(new Set(include.filter(Boolean)));
|
||||
}
|
||||
142
src/request/transformer.ts
Normal file
142
src/request/transformer.ts
Normal file
@@ -0,0 +1,142 @@
|
||||
import { ChatCompletionRequest, ResponsesRequest, InputItem, Message, Content } from "../types.js";
|
||||
import { getNormalizedModel } from "./model-map.js";
|
||||
import { getReasoningConfig, getTextVerbosity, getIncludeFields } from "./reasoning.js";
|
||||
import { getCodexInstructions } from "../prompts/index.js";
|
||||
import { logDebug, logWarn } from "../logger.js";
|
||||
|
||||
export function messagesToInput(messages: Message[]): InputItem[] {
|
||||
return messages.map((msg) => {
|
||||
const content: Content[] = [];
|
||||
|
||||
if (typeof msg.content === "string") {
|
||||
content.push({ type: "input_text", text: msg.content });
|
||||
} else if (Array.isArray(msg.content)) {
|
||||
for (const item of msg.content) {
|
||||
if (item.type === "text") {
|
||||
content.push({ type: "input_text", text: item.text || "" });
|
||||
} else if (item.type === "image_url") {
|
||||
content.push({ type: "input_image", image_url: item.image_url });
|
||||
} else {
|
||||
content.push(item as Content);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return {
|
||||
type: "message",
|
||||
role: msg.role,
|
||||
content,
|
||||
};
|
||||
});
|
||||
}
|
||||
|
||||
export function filterInput(input: InputItem[] | undefined): InputItem[] | undefined {
|
||||
if (!Array.isArray(input)) {
|
||||
return input;
|
||||
}
|
||||
|
||||
return input
|
||||
.filter((item) => {
|
||||
if (item.type === "item_reference") {
|
||||
logWarn(null, "Filtered item_reference from input");
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
})
|
||||
.map((item) => {
|
||||
if (item.id) {
|
||||
const { id, ...itemWithoutId } = item;
|
||||
return itemWithoutId as InputItem;
|
||||
}
|
||||
return item;
|
||||
});
|
||||
}
|
||||
|
||||
export async function transformChatCompletionRequest(
|
||||
request: ChatCompletionRequest,
|
||||
promptCacheKey?: string,
|
||||
): Promise<ResponsesRequest> {
|
||||
const normalizedModel = getNormalizedModel(request.model) || request.model;
|
||||
|
||||
logDebug(null, `Transforming chat completion request: model=${request.model} -> ${normalizedModel}`);
|
||||
|
||||
const codexInstructions = await getCodexInstructions(normalizedModel);
|
||||
const reasoningConfig = getReasoningConfig(normalizedModel);
|
||||
const textVerbosity = getTextVerbosity();
|
||||
const includeFields = getIncludeFields();
|
||||
|
||||
const transformed: ResponsesRequest = {
|
||||
model: normalizedModel,
|
||||
input: filterInput(messagesToInput(request.messages)),
|
||||
stream: true,
|
||||
store: false,
|
||||
instructions: codexInstructions,
|
||||
reasoning: reasoningConfig,
|
||||
text: {
|
||||
verbosity: textVerbosity,
|
||||
},
|
||||
include: includeFields,
|
||||
};
|
||||
|
||||
if (request.temperature !== undefined) {
|
||||
(transformed as any).temperature = request.temperature;
|
||||
}
|
||||
|
||||
if (request.max_tokens !== undefined) {
|
||||
(transformed as any).max_tokens = request.max_tokens;
|
||||
}
|
||||
|
||||
if (request.top_p !== undefined) {
|
||||
(transformed as any).top_p = request.top_p;
|
||||
}
|
||||
|
||||
logDebug(null, "Transformed request:", JSON.stringify(transformed, null, 2));
|
||||
|
||||
return transformed;
|
||||
}
|
||||
|
||||
export async function transformResponsesRequest(
|
||||
request: ResponsesRequest,
|
||||
promptCacheKey?: string,
|
||||
): Promise<ResponsesRequest> {
|
||||
const normalizedModel = getNormalizedModel(request.model) || request.model;
|
||||
|
||||
logDebug(null, `Transforming responses request: model=${request.model} -> ${normalizedModel}`);
|
||||
|
||||
const codexInstructions = await getCodexInstructions(normalizedModel);
|
||||
const reasoningConfig = getReasoningConfig(
|
||||
normalizedModel,
|
||||
request.reasoning?.effort,
|
||||
request.reasoning?.summary,
|
||||
);
|
||||
const textVerbosity = getTextVerbosity(request.text?.verbosity);
|
||||
const includeFields = getIncludeFields(request.include);
|
||||
|
||||
const transformed: ResponsesRequest = {
|
||||
...request,
|
||||
model: normalizedModel,
|
||||
input: filterInput(
|
||||
request.input || (request.messages ? messagesToInput(request.messages as Message[]) : undefined),
|
||||
),
|
||||
stream: request.stream !== undefined ? request.stream : true,
|
||||
store: request.store !== undefined ? request.store : false,
|
||||
instructions: request.instructions || codexInstructions,
|
||||
reasoning: {
|
||||
...reasoningConfig,
|
||||
...request.reasoning,
|
||||
},
|
||||
text: {
|
||||
verbosity: textVerbosity,
|
||||
...request.text,
|
||||
},
|
||||
include: includeFields,
|
||||
};
|
||||
|
||||
delete (transformed as any).max_output_tokens;
|
||||
delete (transformed as any).max_completion_tokens;
|
||||
delete (transformed as any).messages;
|
||||
|
||||
logDebug(null, "Transformed request:", JSON.stringify(transformed, null, 2));
|
||||
|
||||
return transformed;
|
||||
}
|
||||
119
src/request/validator.ts
Normal file
119
src/request/validator.ts
Normal file
@@ -0,0 +1,119 @@
|
||||
import { ChatCompletionRequest, ResponsesRequest, Message } from "../types.js";
|
||||
import { isKnownModel } from "./model-map.js";
|
||||
import { logWarn } from "../logger.js";
|
||||
import { ERROR_MESSAGES } from "../constants.js";
|
||||
|
||||
export interface ValidationResult {
|
||||
valid: boolean;
|
||||
error?: string;
|
||||
}
|
||||
|
||||
export function validateChatCompletionRequest(
|
||||
request: unknown,
|
||||
): ValidationResult {
|
||||
if (!request || typeof request !== "object") {
|
||||
return { valid: false, error: "Request body must be a JSON object" };
|
||||
}
|
||||
|
||||
const req = request as Partial<ChatCompletionRequest>;
|
||||
|
||||
if (!req.model || typeof req.model !== "string") {
|
||||
return { valid: false, error: "Missing or invalid 'model' field" };
|
||||
}
|
||||
|
||||
if (!isKnownModel(req.model)) {
|
||||
return { valid: false, error: ERROR_MESSAGES.INVALID_MODEL };
|
||||
}
|
||||
|
||||
if (!req.messages || !Array.isArray(req.messages)) {
|
||||
return { valid: false, error: ERROR_MESSAGES.MISSING_MESSAGES };
|
||||
}
|
||||
|
||||
if (req.messages.length === 0) {
|
||||
return { valid: false, error: "Messages array cannot be empty" };
|
||||
}
|
||||
|
||||
for (const msg of req.messages) {
|
||||
const messageError = validateMessage(msg);
|
||||
if (messageError) {
|
||||
return { valid: false, error: messageError };
|
||||
}
|
||||
}
|
||||
|
||||
return { valid: true };
|
||||
}
|
||||
|
||||
export function validateResponsesRequest(request: unknown): ValidationResult {
|
||||
if (!request || typeof request !== "object") {
|
||||
return { valid: false, error: "Request body must be a JSON object" };
|
||||
}
|
||||
|
||||
const req = request as Partial<ResponsesRequest>;
|
||||
|
||||
if (!req.model || typeof req.model !== "string") {
|
||||
return { valid: false, error: "Missing or invalid 'model' field" };
|
||||
}
|
||||
|
||||
if (!isKnownModel(req.model)) {
|
||||
return { valid: false, error: ERROR_MESSAGES.INVALID_MODEL };
|
||||
}
|
||||
|
||||
if (!req.input && !req.messages) {
|
||||
return { valid: false, error: "Missing 'input' or 'messages' field" };
|
||||
}
|
||||
|
||||
if (req.input && !Array.isArray(req.input)) {
|
||||
return { valid: false, error: "'input' must be an array" };
|
||||
}
|
||||
|
||||
if (req.messages && !Array.isArray(req.messages)) {
|
||||
return { valid: false, error: "'messages' must be an array" };
|
||||
}
|
||||
|
||||
return { valid: true };
|
||||
}
|
||||
|
||||
function validateMessage(message: unknown): string | null {
|
||||
if (!message || typeof message !== "object") {
|
||||
return "Each message must be an object";
|
||||
}
|
||||
|
||||
const msg = message as Partial<Message>;
|
||||
|
||||
if (!msg.role || typeof msg.role !== "string") {
|
||||
return "Each message must have a 'role' field";
|
||||
}
|
||||
|
||||
const validRoles = ["system", "user", "assistant", "developer"];
|
||||
if (!validRoles.includes(msg.role)) {
|
||||
return `Invalid role: ${msg.role}. Must be one of: ${validRoles.join(", ")}`;
|
||||
}
|
||||
|
||||
if (!msg.content) {
|
||||
return "Each message must have a 'content' field";
|
||||
}
|
||||
|
||||
const content = msg.content;
|
||||
|
||||
if (typeof content === "string" && content.trim() === "") {
|
||||
return "Message content cannot be empty";
|
||||
}
|
||||
|
||||
if (Array.isArray(content)) {
|
||||
if (content.length === 0) {
|
||||
return "Message content array cannot be empty";
|
||||
}
|
||||
|
||||
for (const item of content) {
|
||||
if (!item || typeof item !== "object") {
|
||||
return "Each content item must be an object";
|
||||
}
|
||||
|
||||
if (!item.type || typeof item.type !== "string") {
|
||||
return "Each content item must have a 'type' field";
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
173
src/response/chat-completions.ts
Normal file
173
src/response/chat-completions.ts
Normal file
@@ -0,0 +1,173 @@
|
||||
import {
|
||||
ChatCompletionResponse,
|
||||
ChatCompletionChunk,
|
||||
SSEChunk,
|
||||
Usage,
|
||||
Choice,
|
||||
ChunkChoice,
|
||||
Message,
|
||||
SSEEventData,
|
||||
} from "../types.js";
|
||||
import { logDebug } from "../logger.js";
|
||||
|
||||
export function convertToChatCompletionResponse(
|
||||
sseData: SSEEventData,
|
||||
model: string,
|
||||
): ChatCompletionResponse | null {
|
||||
if (!sseData.response || typeof sseData.response !== "object") {
|
||||
return null;
|
||||
}
|
||||
|
||||
const response = sseData.response as any;
|
||||
|
||||
const chatResponse: ChatCompletionResponse = {
|
||||
id: response.id || `chatcmpl-${Date.now()}`,
|
||||
object: "chat.completion",
|
||||
created: Math.floor(Date.now() / 1000),
|
||||
model,
|
||||
choices: [],
|
||||
usage: {
|
||||
prompt_tokens: response.usage?.input_tokens || 0,
|
||||
completion_tokens: response.usage?.output_tokens || 0,
|
||||
total_tokens:
|
||||
(response.usage?.input_tokens || 0) +
|
||||
(response.usage?.output_tokens || 0),
|
||||
},
|
||||
};
|
||||
|
||||
if (response.output && Array.isArray(response.output)) {
|
||||
for (const item of response.output) {
|
||||
if (item.type === "message" && item.role === "assistant") {
|
||||
const content = extractContent(item.content);
|
||||
const choice: Choice = {
|
||||
index: chatResponse.choices.length,
|
||||
message: {
|
||||
role: "assistant",
|
||||
content,
|
||||
},
|
||||
finish_reason: mapFinishReason(response.status),
|
||||
};
|
||||
chatResponse.choices.push(choice);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
logDebug(
|
||||
null,
|
||||
`Converted to ChatCompletionResponse: ${chatResponse.choices.length} choices`,
|
||||
);
|
||||
|
||||
return chatResponse;
|
||||
}
|
||||
|
||||
export function convertSseChunkToChatCompletionChunk(
|
||||
chunk: SSEChunk,
|
||||
model: string,
|
||||
index: number = 0,
|
||||
): ChatCompletionChunk | null {
|
||||
const chatChunk: ChatCompletionChunk = {
|
||||
id: `chatcmpl-${Date.now()}`,
|
||||
object: "chat.completion.chunk",
|
||||
created: Math.floor(Date.now() / 1000),
|
||||
model,
|
||||
choices: [],
|
||||
};
|
||||
|
||||
if (chunk.type === "response.output_item.add.delta" && chunk.delta) {
|
||||
const delta = chunk.delta as any;
|
||||
const choice: ChunkChoice = {
|
||||
index,
|
||||
delta: {
|
||||
role: delta.role || undefined,
|
||||
content: delta.content || undefined,
|
||||
},
|
||||
finish_reason: null,
|
||||
};
|
||||
chatChunk.choices.push(choice);
|
||||
} else if (chunk.type === "response.output_item.added" && chunk.item) {
|
||||
const item = chunk.item as any;
|
||||
if (item.type === "message" && item.role === "assistant") {
|
||||
const content = extractContent(item.content);
|
||||
if (content) {
|
||||
const choice: ChunkChoice = {
|
||||
index,
|
||||
delta: {
|
||||
role: "assistant",
|
||||
content,
|
||||
},
|
||||
finish_reason: null,
|
||||
};
|
||||
chatChunk.choices.push(choice);
|
||||
}
|
||||
}
|
||||
} else if (chunk.type === "response.content_part.added" && chunk.delta) {
|
||||
const delta = chunk.delta as any;
|
||||
const choice: ChunkChoice = {
|
||||
index,
|
||||
delta: {
|
||||
role: delta.role || "assistant",
|
||||
content: delta.content,
|
||||
},
|
||||
finish_reason: null,
|
||||
};
|
||||
chatChunk.choices.push(choice);
|
||||
} else if (chunk.type === "response.output_text.delta") {
|
||||
const delta = chunk.delta as any;
|
||||
let content: string | undefined;
|
||||
if (typeof delta === "string") {
|
||||
content = delta;
|
||||
} else if (delta && (delta.text || delta.content)) {
|
||||
content = delta.text || delta.content;
|
||||
}
|
||||
if (content) {
|
||||
const choice: ChunkChoice = {
|
||||
index,
|
||||
delta: {
|
||||
content,
|
||||
},
|
||||
finish_reason: null,
|
||||
};
|
||||
chatChunk.choices.push(choice);
|
||||
}
|
||||
} else if (chunk.type === "response.done") {
|
||||
const choice: ChunkChoice = {
|
||||
index,
|
||||
delta: {},
|
||||
finish_reason: "stop",
|
||||
};
|
||||
chatChunk.choices.push(choice);
|
||||
}
|
||||
|
||||
return chatChunk;
|
||||
}
|
||||
|
||||
function extractContent(content: any[]): string {
|
||||
if (!Array.isArray(content)) {
|
||||
return "";
|
||||
}
|
||||
|
||||
const textParts: string[] = [];
|
||||
for (const item of content) {
|
||||
if (item.type === "output_text" && item.text) {
|
||||
textParts.push(item.text);
|
||||
}
|
||||
}
|
||||
|
||||
const result = textParts.join("");
|
||||
logDebug(
|
||||
null,
|
||||
`extractContent: ${textParts.length} output_text items, result="${result.substring(0, 100)}"`,
|
||||
);
|
||||
return result;
|
||||
}
|
||||
|
||||
function mapFinishReason(status?: string): string {
|
||||
if (!status) return "stop";
|
||||
|
||||
const statusLower = status.toLowerCase();
|
||||
if (statusLower === "completed") return "stop";
|
||||
if (statusLower === "incomplete") return "length";
|
||||
if (statusLower === "failed") return "error";
|
||||
|
||||
return "stop";
|
||||
}
|
||||
121
src/response/converter.ts
Normal file
121
src/response/converter.ts
Normal file
@@ -0,0 +1,121 @@
|
||||
import { parseSseStream, parseSseChunks } from "./sse-parser.js";
|
||||
import {
|
||||
convertToChatCompletionResponse,
|
||||
convertSseChunkToChatCompletionChunk,
|
||||
} from "./chat-completions.js";
|
||||
import { logError, logDebug } from "../logger.js";
|
||||
import { ChatCompletionChunk, SSEChunk } from "../types.js";
|
||||
|
||||
export async function convertSseToJson(
|
||||
sseText: string,
|
||||
model: string,
|
||||
): Promise<Response> {
|
||||
const sseData = parseSseStream(sseText);
|
||||
|
||||
if (!sseData) {
|
||||
logError(null, "Failed to parse SSE stream, returning original response");
|
||||
return new Response(sseText, {
|
||||
status: 500,
|
||||
headers: { "content-type": "application/json" },
|
||||
});
|
||||
}
|
||||
|
||||
const chatResponse = convertToChatCompletionResponse(sseData, model);
|
||||
|
||||
if (!chatResponse) {
|
||||
logError(null, "Failed to convert SSE to ChatCompletionResponse");
|
||||
return new Response(sseText, {
|
||||
status: 500,
|
||||
headers: { "content-type": "application/json" },
|
||||
});
|
||||
}
|
||||
|
||||
return new Response(JSON.stringify(chatResponse), {
|
||||
status: 200,
|
||||
headers: { "content-type": "application/json" },
|
||||
});
|
||||
}
|
||||
|
||||
export function transformSseStream(
|
||||
stream: ReadableStream,
|
||||
model: string,
|
||||
): ReadableStream {
|
||||
const encoder = new TextEncoder();
|
||||
const decoder = new TextDecoder();
|
||||
|
||||
return new ReadableStream({
|
||||
async start(controller) {
|
||||
const reader = stream.getReader();
|
||||
let buffer = "";
|
||||
let chunkCount = 0;
|
||||
|
||||
try {
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
|
||||
buffer += decoder.decode(value, { stream: true });
|
||||
const lines = buffer.split("\n");
|
||||
buffer = lines.pop() || "";
|
||||
|
||||
for (const line of lines) {
|
||||
if (line.startsWith("data: ")) {
|
||||
try {
|
||||
const chunk = JSON.parse(line.substring(6)) as SSEChunk;
|
||||
|
||||
if (chunkCount < 10) {
|
||||
const deltaInfo = chunk.delta
|
||||
? JSON.stringify(chunk.delta)
|
||||
: "N/A";
|
||||
logDebug(
|
||||
null,
|
||||
`Incoming SSE chunk ${chunkCount}: type=${chunk.type}, hasDelta=${!!chunk.delta}`,
|
||||
);
|
||||
logDebug(null, ` Delta content: ${deltaInfo}`);
|
||||
logDebug(
|
||||
null,
|
||||
` All keys: ${Object.keys(chunk).join(", ")}`,
|
||||
);
|
||||
}
|
||||
chunkCount++;
|
||||
|
||||
const chatChunk = convertSseChunkToChatCompletionChunk(
|
||||
chunk,
|
||||
model,
|
||||
);
|
||||
|
||||
if (chatChunk && chatChunk.choices.length > 0) {
|
||||
if (chunkCount < 10) {
|
||||
logDebug(
|
||||
null,
|
||||
`Outgoing chat chunk: choices=${chatChunk.choices.length}, delta=${JSON.stringify(chatChunk.choices[0].delta)}`,
|
||||
);
|
||||
}
|
||||
controller.enqueue(
|
||||
encoder.encode(`data: ${JSON.stringify(chatChunk)}\n\n`),
|
||||
);
|
||||
} else if (chunkCount < 10) {
|
||||
logDebug(
|
||||
null,
|
||||
`Chunk converted but no choices (type=${chunk.type})`,
|
||||
);
|
||||
}
|
||||
} catch (error) {
|
||||
logError(
|
||||
null,
|
||||
`Failed to parse SSE chunk: ${line.substring(0, 200)}`,
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
controller.close();
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `SSE stream transformation error: ${err.message}`);
|
||||
controller.error(err);
|
||||
}
|
||||
},
|
||||
});
|
||||
}
|
||||
159
src/response/handler.ts
Normal file
159
src/response/handler.ts
Normal file
@@ -0,0 +1,159 @@
|
||||
import { convertSseToJson, transformSseStream } from "./converter.js";
|
||||
import { getConfig } from "../config.js";
|
||||
import { logDebug, logError, logInfo, logWarn } from "../logger.js";
|
||||
import { HTTP_STATUS } from "../constants.js";
|
||||
|
||||
export interface ResponseHandlerOptions {
|
||||
model: string;
|
||||
isStreaming: boolean;
|
||||
}
|
||||
|
||||
export async function handleBackendResponse(
|
||||
response: Response,
|
||||
options: ResponseHandlerOptions,
|
||||
): Promise<Response> {
|
||||
const { model, isStreaming } = options;
|
||||
|
||||
logDebug(null, `Handling backend response: status=${response.status}, streaming=${isStreaming}`);
|
||||
|
||||
if (!response.ok) {
|
||||
return await handleErrorResponse(response);
|
||||
}
|
||||
|
||||
if (isStreaming) {
|
||||
return handleStreamingResponse(response, model);
|
||||
} else {
|
||||
return await handleNonStreamingResponse(response, model);
|
||||
}
|
||||
}
|
||||
|
||||
async function handleNonStreamingResponse(
|
||||
response: Response,
|
||||
model: string,
|
||||
): Promise<Response> {
|
||||
try {
|
||||
const sseText = await response.text();
|
||||
logDebug(null, `Received SSE stream (${sseText.length} bytes)`);
|
||||
|
||||
const chatResponse = await convertSseToJson(sseText, model);
|
||||
return chatResponse;
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `Error processing non-streaming response: ${err.message}`);
|
||||
return new Response(
|
||||
JSON.stringify({
|
||||
error: {
|
||||
message: "Failed to process response",
|
||||
type: "response_processing_error",
|
||||
},
|
||||
}),
|
||||
{ status: HTTP_STATUS.INTERNAL_SERVER_ERROR, headers: { "content-type": "application/json" } },
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
function handleStreamingResponse(response: Response, model: string): Response {
|
||||
try {
|
||||
if (!response.body) {
|
||||
throw new Error("Response has no body");
|
||||
}
|
||||
|
||||
const transformedStream = transformSseStream(response.body, model);
|
||||
|
||||
return new Response(transformedStream, {
|
||||
status: response.status,
|
||||
statusText: response.statusText,
|
||||
headers: {
|
||||
"Content-Type": "text/event-stream",
|
||||
"Cache-Control": "no-cache",
|
||||
"Connection": "keep-alive",
|
||||
},
|
||||
});
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `Error setting up streaming response: ${err.message}`);
|
||||
return new Response(
|
||||
JSON.stringify({
|
||||
error: {
|
||||
message: "Failed to set up stream",
|
||||
type: "stream_setup_error",
|
||||
},
|
||||
}),
|
||||
{ status: HTTP_STATUS.INTERNAL_SERVER_ERROR, headers: { "content-type": "application/json" } },
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
async function handleErrorResponse(response: Response): Promise<Response> {
|
||||
try {
|
||||
const text = await response.text();
|
||||
let errorData: unknown;
|
||||
|
||||
try {
|
||||
errorData = JSON.parse(text);
|
||||
} catch {
|
||||
errorData = text;
|
||||
}
|
||||
|
||||
logError(null, `Backend error: ${response.status}`, errorData);
|
||||
|
||||
const mappedResponse = await mapUsageLimit404(response, text);
|
||||
if (mappedResponse) {
|
||||
return mappedResponse;
|
||||
}
|
||||
|
||||
return new Response(text, {
|
||||
status: response.status,
|
||||
headers: { "content-type": "application/json" },
|
||||
});
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `Error handling error response: ${err.message}`);
|
||||
return new Response(
|
||||
JSON.stringify({
|
||||
error: {
|
||||
message: "Internal server error",
|
||||
type: "internal_error",
|
||||
},
|
||||
}),
|
||||
{ status: HTTP_STATUS.INTERNAL_SERVER_ERROR, headers: { "content-type": "application/json" } },
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
async function mapUsageLimit404(
|
||||
response: Response,
|
||||
text: string,
|
||||
): Promise<Response | null> {
|
||||
if (response.status !== HTTP_STATUS.NOT_FOUND) {
|
||||
return null;
|
||||
}
|
||||
|
||||
try {
|
||||
const json = JSON.parse(text);
|
||||
const errorCode = json?.error?.code || json?.error?.type || "";
|
||||
const haystack = `${errorCode} ${text}`.toLowerCase();
|
||||
|
||||
if (
|
||||
/usage_limit_reached|usage_not_included|rate_limit_exceeded|usage limit/i.test(haystack)
|
||||
) {
|
||||
logWarn(null, "Mapping 404 usage limit error to 429");
|
||||
|
||||
return new Response(
|
||||
JSON.stringify({
|
||||
error: {
|
||||
message: "Rate limit exceeded",
|
||||
type: "rate_limit_error",
|
||||
},
|
||||
}),
|
||||
{
|
||||
status: HTTP_STATUS.TOO_MANY_REQUESTS,
|
||||
headers: { "content-type": "application/json" },
|
||||
},
|
||||
);
|
||||
}
|
||||
} catch {
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
42
src/response/sse-parser.ts
Normal file
42
src/response/sse-parser.ts
Normal file
@@ -0,0 +1,42 @@
|
||||
import { SSEEventData, SSEChunk } from "../types.js";
|
||||
import { logDebug, logError } from "../logger.js";
|
||||
|
||||
export function parseSseStream(sseText: string): SSEEventData | null {
|
||||
const lines = sseText.split("\n");
|
||||
|
||||
for (const line of lines) {
|
||||
if (line.startsWith("data: ")) {
|
||||
try {
|
||||
const data = JSON.parse(line.substring(6)) as SSEEventData;
|
||||
|
||||
if (data.type === "response.done" || data.type === "response.completed") {
|
||||
logDebug(null, "Found response.done event in SSE stream");
|
||||
return data;
|
||||
}
|
||||
} catch (error) {
|
||||
logError(null, `Failed to parse SSE event: ${line}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
logError(null, "No response.done event found in SSE stream");
|
||||
return null;
|
||||
}
|
||||
|
||||
export function parseSseChunks(sseText: string): SSEChunk[] {
|
||||
const chunks: SSEChunk[] = [];
|
||||
const lines = sseText.split("\n");
|
||||
|
||||
for (const line of lines) {
|
||||
if (line.startsWith("data: ")) {
|
||||
try {
|
||||
const data = JSON.parse(line.substring(6));
|
||||
chunks.push(data as SSEChunk);
|
||||
} catch (error) {
|
||||
logError(null, `Failed to parse SSE chunk: ${line}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return chunks;
|
||||
}
|
||||
262
src/router.ts
Normal file
262
src/router.ts
Normal file
@@ -0,0 +1,262 @@
|
||||
import { Hono } from "hono";
|
||||
import { HTTP_STATUS, ERROR_MESSAGES } from "./constants.js";
|
||||
import { logInfo, logDebug, logError, logWarn, createRequestLogger } from "./logger.js";
|
||||
import { validateChatCompletionRequest } from "./request/validator.js";
|
||||
import { transformChatCompletionRequest, transformResponsesRequest } from "./request/transformer.js";
|
||||
import { createCodexHeaders } from "./request/headers.js";
|
||||
import { handleBackendResponse } from "./response/handler.js";
|
||||
import { getAccessToken, getAccountId } from "./auth/token-refresh.js";
|
||||
import { createOAuthFlow, exchangeAuthorizationCode } from "./auth/oauth.js";
|
||||
import { startLocalOAuthServer } from "./auth/server.js";
|
||||
import { openBrowser } from "./auth/browser.js";
|
||||
import { getConfig } from "./config.js";
|
||||
import type { ChatCompletionRequest, ResponsesRequest } from "./types.js";
|
||||
|
||||
const router = new Hono();
|
||||
|
||||
router.get("/", (c) => {
|
||||
logInfo(null, "GET /");
|
||||
return c.json({
|
||||
name: "ChatGPT Codex Router",
|
||||
version: "1.0.0",
|
||||
status: "running",
|
||||
endpoints: {
|
||||
chat_completions: "/v1/chat/completions",
|
||||
responses: "/v1/responses",
|
||||
auth_login: "/auth/login",
|
||||
health: "/health",
|
||||
},
|
||||
});
|
||||
});
|
||||
|
||||
router.post("/v1/chat/completions", async (c) => {
|
||||
const requestId = createRequestLogger();
|
||||
logInfo(requestId, "POST /v1/chat/completions");
|
||||
|
||||
try {
|
||||
const body = await c.req.json();
|
||||
|
||||
const validation = validateChatCompletionRequest(body);
|
||||
if (!validation.valid) {
|
||||
logError(requestId, `Validation error: ${validation.error}`);
|
||||
return c.json(
|
||||
{
|
||||
error: {
|
||||
message: validation.error,
|
||||
type: "invalid_request_error",
|
||||
},
|
||||
},
|
||||
HTTP_STATUS.BAD_REQUEST || 400,
|
||||
);
|
||||
}
|
||||
|
||||
const config = getConfig();
|
||||
const accessToken = await getAccessToken();
|
||||
|
||||
if (!accessToken) {
|
||||
logError(requestId, ERROR_MESSAGES.NO_TOKEN);
|
||||
return c.json(
|
||||
{
|
||||
error: {
|
||||
message: ERROR_MESSAGES.NO_TOKEN,
|
||||
type: "authentication_error",
|
||||
},
|
||||
},
|
||||
HTTP_STATUS.UNAUTHORIZED,
|
||||
);
|
||||
}
|
||||
|
||||
const accountId = await getAccountId();
|
||||
if (!accountId) {
|
||||
logError(requestId, "Failed to get account ID");
|
||||
return c.json(
|
||||
{
|
||||
error: {
|
||||
message: "Failed to get account ID",
|
||||
type: "authentication_error",
|
||||
},
|
||||
},
|
||||
HTTP_STATUS.UNAUTHORIZED,
|
||||
);
|
||||
}
|
||||
|
||||
const transformedRequest = await transformChatCompletionRequest(body as ChatCompletionRequest);
|
||||
const headers = createCodexHeaders({
|
||||
accessToken,
|
||||
accountId,
|
||||
});
|
||||
|
||||
logDebug(requestId, `Forwarding to ${config.backend.url}/codex/responses`);
|
||||
|
||||
const startTime = Date.now();
|
||||
const backendResponse = await fetch(`${config.backend.url}/codex/responses`, {
|
||||
method: "POST",
|
||||
headers,
|
||||
body: JSON.stringify(transformedRequest),
|
||||
});
|
||||
|
||||
const duration = Date.now() - startTime;
|
||||
logInfo(requestId, `Backend response: ${backendResponse.status} (${duration}ms)`);
|
||||
|
||||
const processedResponse = await handleBackendResponse(backendResponse, {
|
||||
model: transformedRequest.model,
|
||||
isStreaming: (body as ChatCompletionRequest).stream === true,
|
||||
});
|
||||
|
||||
logInfo(requestId, `Request completed in ${duration}ms`);
|
||||
return processedResponse;
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(requestId, `Error processing request: ${err.message}`);
|
||||
return c.json(
|
||||
{
|
||||
error: {
|
||||
message: "Internal server error",
|
||||
type: "internal_error",
|
||||
},
|
||||
},
|
||||
HTTP_STATUS.INTERNAL_SERVER_ERROR,
|
||||
);
|
||||
}
|
||||
});
|
||||
|
||||
router.post("/v1/responses", async (c) => {
|
||||
const requestId = createRequestLogger();
|
||||
logInfo(requestId, "POST /v1/responses");
|
||||
|
||||
try {
|
||||
const body = await c.req.json();
|
||||
|
||||
const config = getConfig();
|
||||
const accessToken = await getAccessToken();
|
||||
|
||||
if (!accessToken) {
|
||||
logError(requestId, ERROR_MESSAGES.NO_TOKEN);
|
||||
return c.json(
|
||||
{
|
||||
error: {
|
||||
message: ERROR_MESSAGES.NO_TOKEN,
|
||||
type: "authentication_error",
|
||||
},
|
||||
},
|
||||
HTTP_STATUS.UNAUTHORIZED,
|
||||
);
|
||||
}
|
||||
|
||||
const accountId = await getAccountId();
|
||||
if (!accountId) {
|
||||
logError(requestId, "Failed to get account ID");
|
||||
return c.json(
|
||||
{
|
||||
error: {
|
||||
message: "Failed to get account ID",
|
||||
type: "authentication_error",
|
||||
},
|
||||
},
|
||||
HTTP_STATUS.UNAUTHORIZED,
|
||||
);
|
||||
}
|
||||
|
||||
const transformedRequest = await transformResponsesRequest(body as ResponsesRequest);
|
||||
const headers = createCodexHeaders({
|
||||
accessToken,
|
||||
accountId,
|
||||
});
|
||||
|
||||
logDebug(requestId, `Forwarding to ${config.backend.url}/codex/responses`);
|
||||
|
||||
const startTime = Date.now();
|
||||
const backendResponse = await fetch(`${config.backend.url}/codex/responses`, {
|
||||
method: "POST",
|
||||
headers,
|
||||
body: JSON.stringify(transformedRequest),
|
||||
});
|
||||
|
||||
const duration = Date.now() - startTime;
|
||||
logInfo(requestId, `Backend response: ${backendResponse.status} (${duration}ms)`);
|
||||
|
||||
const processedResponse = await handleBackendResponse(backendResponse, {
|
||||
model: transformedRequest.model,
|
||||
isStreaming: transformedRequest.stream === true,
|
||||
});
|
||||
|
||||
logInfo(requestId, `Request completed in ${duration}ms`);
|
||||
return processedResponse;
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(requestId, `Error processing request: ${err.message}`);
|
||||
return c.json(
|
||||
{
|
||||
error: {
|
||||
message: "Internal server error",
|
||||
type: "internal_error",
|
||||
},
|
||||
},
|
||||
HTTP_STATUS.INTERNAL_SERVER_ERROR,
|
||||
);
|
||||
}
|
||||
});
|
||||
|
||||
router.post("/auth/login", async (c) => {
|
||||
logInfo(null, "POST /auth/login");
|
||||
|
||||
try {
|
||||
const config = getConfig();
|
||||
const oauthFlow = await createOAuthFlow(config.oauth.clientId, config.oauth.redirectUri);
|
||||
|
||||
logInfo(null, `Starting OAuth flow with state: ${oauthFlow.state}`);
|
||||
|
||||
const serverInfo = await startLocalOAuthServer({
|
||||
state: oauthFlow.state,
|
||||
pkce: oauthFlow.pkce,
|
||||
port: config.oauth.localServerPort,
|
||||
});
|
||||
|
||||
const browserOpened = openBrowser(oauthFlow.url);
|
||||
|
||||
if (!serverInfo.ready) {
|
||||
serverInfo.close();
|
||||
logWarn(null, "OAuth server not ready, using manual flow");
|
||||
return c.json({
|
||||
status: "pending",
|
||||
url: oauthFlow.url,
|
||||
instructions: "Please copy the URL and open it in your browser to complete the OAuth flow",
|
||||
});
|
||||
}
|
||||
|
||||
return c.json({
|
||||
status: "pending",
|
||||
url: oauthFlow.url,
|
||||
instructions: browserOpened
|
||||
? "Please complete the OAuth flow in your browser"
|
||||
: "Please copy the URL and open it in your browser to complete the OAuth flow",
|
||||
});
|
||||
} catch (error) {
|
||||
const err = error as Error;
|
||||
logError(null, `OAuth login error: ${err.message}`);
|
||||
return c.json(
|
||||
{
|
||||
error: {
|
||||
message: "Failed to start OAuth flow",
|
||||
type: "oauth_error",
|
||||
},
|
||||
},
|
||||
HTTP_STATUS.INTERNAL_SERVER_ERROR,
|
||||
);
|
||||
}
|
||||
});
|
||||
|
||||
router.all("*", (c) => {
|
||||
logInfo(null, `404 Not Found: ${c.req.method} ${c.req.path}`);
|
||||
return c.json(
|
||||
{
|
||||
error: {
|
||||
message: "Not Found",
|
||||
type: "not_found_error",
|
||||
},
|
||||
},
|
||||
HTTP_STATUS.NOT_FOUND,
|
||||
);
|
||||
});
|
||||
|
||||
export default router;
|
||||
29
src/server.ts
Normal file
29
src/server.ts
Normal file
@@ -0,0 +1,29 @@
|
||||
import { Hono } from "hono";
|
||||
import { cors } from "hono/cors";
|
||||
import { getConfig } from "./config.js";
|
||||
import { logInfo, setLoggerConfig } from "./logger.js";
|
||||
import { HTTP_STATUS } from "./constants.js";
|
||||
import router from "./router.js";
|
||||
|
||||
export function createApp(): Hono {
|
||||
const config = getConfig();
|
||||
|
||||
setLoggerConfig(config.logging);
|
||||
|
||||
const app = new Hono();
|
||||
|
||||
app.use("*", cors({
|
||||
origin: "*",
|
||||
credentials: true,
|
||||
}));
|
||||
|
||||
app.get("/health", (c) => {
|
||||
return c.json({ status: "healthy", timestamp: Date.now() }, HTTP_STATUS.OK);
|
||||
});
|
||||
|
||||
app.route("/", router);
|
||||
|
||||
return app;
|
||||
}
|
||||
|
||||
export { createApp as default };
|
||||
154
src/types.ts
Normal file
154
src/types.ts
Normal file
@@ -0,0 +1,154 @@
|
||||
export interface AppConfig {
|
||||
server: {
|
||||
port: number;
|
||||
host: string;
|
||||
};
|
||||
oauth: {
|
||||
clientId: string;
|
||||
redirectUri: string;
|
||||
localServerPort: number;
|
||||
};
|
||||
backend: {
|
||||
url: string;
|
||||
timeout: number;
|
||||
};
|
||||
logging: {
|
||||
level: "error" | "warn" | "info" | "debug";
|
||||
dir: string;
|
||||
enableRequestLogging: boolean;
|
||||
};
|
||||
codex: {
|
||||
mode: boolean;
|
||||
defaultReasoningEffort: ReasoningEffort;
|
||||
defaultTextVerbosity: "low" | "medium" | "high";
|
||||
};
|
||||
}
|
||||
|
||||
export type ReasoningEffort =
|
||||
| "none"
|
||||
| "minimal"
|
||||
| "low"
|
||||
| "medium"
|
||||
| "high"
|
||||
| "xhigh";
|
||||
|
||||
export type ReasoningSummary =
|
||||
| "auto"
|
||||
| "concise"
|
||||
| "detailed"
|
||||
| "off"
|
||||
| "on";
|
||||
|
||||
export interface ReasoningConfig {
|
||||
effort: ReasoningEffort;
|
||||
summary: ReasoningSummary;
|
||||
}
|
||||
|
||||
export interface TokenData {
|
||||
access_token: string;
|
||||
refresh_token: string;
|
||||
expires_at: number;
|
||||
account_id: string;
|
||||
updated_at: number;
|
||||
}
|
||||
|
||||
export interface ModelFamily {
|
||||
type: "gpt-5.1" | "gpt-5.2" | "codex" | "codex-max" | "codex-mini" | "gpt-5.2-codex";
|
||||
}
|
||||
|
||||
export interface Message {
|
||||
role: "system" | "user" | "assistant" | "developer";
|
||||
content: string | Content[];
|
||||
}
|
||||
|
||||
export interface Content {
|
||||
type: string;
|
||||
text?: string;
|
||||
[key: string]: unknown;
|
||||
}
|
||||
|
||||
export interface InputItem {
|
||||
type?: string;
|
||||
role: string;
|
||||
content: Content[];
|
||||
[key: string]: unknown;
|
||||
}
|
||||
|
||||
export interface ChatCompletionRequest {
|
||||
model: string;
|
||||
messages: Message[];
|
||||
stream?: boolean;
|
||||
temperature?: number;
|
||||
max_tokens?: number;
|
||||
top_p?: number;
|
||||
[key: string]: unknown;
|
||||
}
|
||||
|
||||
export interface ResponsesRequest {
|
||||
model: string;
|
||||
input?: InputItem[];
|
||||
stream?: boolean;
|
||||
store?: boolean;
|
||||
instructions?: string;
|
||||
reasoning?: Partial<ReasoningConfig>;
|
||||
text?: {
|
||||
verbosity?: "low" | "medium" | "high";
|
||||
};
|
||||
include?: string[];
|
||||
[key: string]: unknown;
|
||||
}
|
||||
|
||||
export interface ChatCompletionResponse {
|
||||
id: string;
|
||||
object: string;
|
||||
created: number;
|
||||
model: string;
|
||||
choices: Choice[];
|
||||
usage: Usage;
|
||||
}
|
||||
|
||||
export interface Choice {
|
||||
index: number;
|
||||
message: Message;
|
||||
finish_reason: string;
|
||||
}
|
||||
|
||||
export interface Usage {
|
||||
prompt_tokens: number;
|
||||
completion_tokens: number;
|
||||
total_tokens: number;
|
||||
}
|
||||
|
||||
export interface ChatCompletionChunk {
|
||||
id: string;
|
||||
object: string;
|
||||
created: number;
|
||||
model: string;
|
||||
choices: ChunkChoice[];
|
||||
}
|
||||
|
||||
export interface ChunkChoice {
|
||||
index: number;
|
||||
delta: {
|
||||
role?: string;
|
||||
content?: string;
|
||||
};
|
||||
finish_reason: string | null;
|
||||
}
|
||||
|
||||
export interface SSEEventData {
|
||||
type: string;
|
||||
delta?: unknown;
|
||||
response?: unknown;
|
||||
[key: string]: unknown;
|
||||
}
|
||||
|
||||
export interface SSEChunk {
|
||||
type: string;
|
||||
delta?: {
|
||||
content?: string;
|
||||
role?: string;
|
||||
};
|
||||
response?: unknown;
|
||||
[key: string]: unknown;
|
||||
}
|
||||
107
test-codex-raw.js
Normal file
107
test-codex-raw.js
Normal file
@@ -0,0 +1,107 @@
|
||||
import { getAccessToken } from "./dist/auth/token-refresh.js";
|
||||
import { getAccountId } from "./dist/auth/token-refresh.js";
|
||||
|
||||
const accessToken = await getAccessToken();
|
||||
const accountId = await getAccountId();
|
||||
|
||||
if (!accessToken || !accountId) {
|
||||
console.error("No token or account ID");
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
const instructions = await import("./dist/prompts/index.js").then((m) =>
|
||||
m.getCodexInstructions("gpt-5.1"),
|
||||
);
|
||||
|
||||
console.log("Fetching from Codex API...");
|
||||
|
||||
const response = await fetch(
|
||||
"https://chatgpt.com/backend-api/codex/responses",
|
||||
{
|
||||
method: "POST",
|
||||
headers: {
|
||||
Authorization: `Bearer ${accessToken}`,
|
||||
"openai-account-id": accountId,
|
||||
"openai-beta": "responses=2",
|
||||
"openai-originator": "codex_cli_rs",
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
body: JSON.stringify({
|
||||
model: "gpt-5.1",
|
||||
input: [
|
||||
{
|
||||
type: "message",
|
||||
role: "user",
|
||||
content: [
|
||||
{
|
||||
type: "input_text",
|
||||
text: "Say hello in one word",
|
||||
},
|
||||
],
|
||||
},
|
||||
],
|
||||
stream: true,
|
||||
store: false,
|
||||
reasoning: { effort: "medium", summary: "auto" },
|
||||
instructions: instructions,
|
||||
}),
|
||||
},
|
||||
);
|
||||
|
||||
console.log("Status:", response.status);
|
||||
|
||||
if (response.status !== 200) {
|
||||
const errorText = await response.text();
|
||||
console.error("Error:", errorText);
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
const reader = response.body.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let buffer = "";
|
||||
let chunks = 0;
|
||||
|
||||
console.log("\n=== Raw SSE Chunks ===\n");
|
||||
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
|
||||
buffer += decoder.decode(value, { stream: true });
|
||||
const lines = buffer.split("\n");
|
||||
buffer = lines.pop() || "";
|
||||
|
||||
for (const line of lines) {
|
||||
if (line.startsWith("data: ")) {
|
||||
chunks++;
|
||||
const data = line.substring(6);
|
||||
|
||||
if (data.trim() !== "[DONE]") {
|
||||
try {
|
||||
const parsed = JSON.parse(data);
|
||||
console.log(`\n--- Chunk ${chunks} ---`);
|
||||
console.log(`Type: ${parsed.type}`);
|
||||
console.log(`Has delta: ${!!parsed.delta}`);
|
||||
if (parsed.delta) {
|
||||
console.log(`Delta:`, JSON.stringify(parsed.delta, null, 2));
|
||||
}
|
||||
console.log(`Full chunk keys:`, Object.keys(parsed));
|
||||
if (parsed.item) {
|
||||
console.log(
|
||||
`Item type: ${parsed.item.type}, role: ${parsed.item.role}`,
|
||||
);
|
||||
}
|
||||
|
||||
if (chunks >= 20) {
|
||||
console.log("\n=== Stopping after 20 chunks ===");
|
||||
process.exit(0);
|
||||
}
|
||||
} catch (e) {
|
||||
console.log(`\nChunk ${chunks} (raw):`, data.substring(0, 200));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
console.log("\n=== End of stream ===");
|
||||
74
test-direct-response.js
Normal file
74
test-direct-response.js
Normal file
@@ -0,0 +1,74 @@
|
||||
import { getAccessToken } from './dist/auth/token-refresh.js';
|
||||
import { getAccountId } from './dist/auth/token-refresh.js';
|
||||
|
||||
const accessToken = await getAccessToken();
|
||||
const accountId = await getAccountId();
|
||||
|
||||
if (!accessToken || !accountId) {
|
||||
console.error('No token or account ID');
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
const instructions = await import('./dist/prompts/index.js').then(m => m.getCodexInstructions('gpt-5.1'));
|
||||
|
||||
const response = await fetch('https://chatgpt.com/backend-api/codex/responses', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Authorization': `Bearer ${accessToken}`,
|
||||
'openai-account-id': accountId,
|
||||
'openai-beta': 'responses=2',
|
||||
'openai-originator': 'codex_cli_rs',
|
||||
'Content-Type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({
|
||||
model: 'gpt-5.1',
|
||||
input: [
|
||||
{
|
||||
type: 'message',
|
||||
role: 'user',
|
||||
content: [
|
||||
{
|
||||
type: 'input_text',
|
||||
text: 'Hello, say hi in one word'
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
stream: true,
|
||||
store: false,
|
||||
reasoning: { effort: 'medium', summary: 'auto' },
|
||||
instructions: instructions
|
||||
})
|
||||
});
|
||||
|
||||
console.log('Status:', response.status);
|
||||
|
||||
const reader = response.body.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let buffer = '';
|
||||
let chunks = 0;
|
||||
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
|
||||
buffer += decoder.decode(value, { stream: true });
|
||||
const lines = buffer.split('\n');
|
||||
buffer = lines.pop() || '';
|
||||
|
||||
for (const line of lines) {
|
||||
if (line.startsWith('data: ')) {
|
||||
chunks++;
|
||||
const data = line.substring(6);
|
||||
if (data.trim() !== '[DONE]') {
|
||||
try {
|
||||
const parsed = JSON.parse(data);
|
||||
console.log(`Chunk ${chunks}: type="${parsed.type}", delta=`, parsed.delta ? JSON.stringify(parsed.delta).substring(0, 100) : 'N/A');
|
||||
if (chunks >= 10) process.exit(0);
|
||||
} catch (e) {
|
||||
console.log(`Raw chunk ${chunks}:`, data.substring(0, 100));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
45
test-local-endpoint.js
Normal file
45
test-local-endpoint.js
Normal file
@@ -0,0 +1,45 @@
|
||||
const response = await fetch("http://localhost:3000/v1/chat/completions", {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
body: JSON.stringify({
|
||||
model: "gpt-5.2",
|
||||
messages: [{ role: "user", content: "Say hi in one word" }],
|
||||
stream: true,
|
||||
}),
|
||||
});
|
||||
|
||||
console.log("Status:", response.status);
|
||||
|
||||
const reader = response.body.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let buffer = "";
|
||||
let count = 0;
|
||||
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
|
||||
buffer += decoder.decode(value, { stream: true });
|
||||
const lines = buffer.split("\n");
|
||||
buffer = lines.pop() || "";
|
||||
|
||||
for (const line of lines) {
|
||||
if (line.startsWith("data: ")) {
|
||||
count++;
|
||||
const data = line.substring(6);
|
||||
if (data.trim() !== "[DONE]") {
|
||||
try {
|
||||
const parsed = JSON.parse(data);
|
||||
console.log(
|
||||
`Response ${count}:`,
|
||||
JSON.stringify(parsed).substring(0, 200),
|
||||
);
|
||||
} catch (e) {
|
||||
console.log(`Raw response ${count}:`, data.substring(0, 100));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
57
test-longer-response.js
Normal file
57
test-longer-response.js
Normal file
@@ -0,0 +1,57 @@
|
||||
const response = await fetch("http://localhost:3000/v1/chat/completions", {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
body: JSON.stringify({
|
||||
model: "gpt-5.2",
|
||||
messages: [
|
||||
{ role: "user", content: "What is 2+2? Answer in one sentence." },
|
||||
],
|
||||
stream: true,
|
||||
}),
|
||||
});
|
||||
|
||||
console.log("Status:", response.status);
|
||||
|
||||
const reader = response.body.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let buffer = "";
|
||||
let count = 0;
|
||||
let fullContent = "";
|
||||
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
|
||||
buffer += decoder.decode(value, { stream: true });
|
||||
const lines = buffer.split("\n");
|
||||
buffer = lines.pop() || "";
|
||||
|
||||
for (const line of lines) {
|
||||
if (line.startsWith("data: ")) {
|
||||
count++;
|
||||
const data = line.substring(6);
|
||||
if (data.trim() !== "[DONE]") {
|
||||
try {
|
||||
const parsed = JSON.parse(data);
|
||||
if (parsed.choices && parsed.choices[0]) {
|
||||
const delta = parsed.choices[0].delta;
|
||||
if (delta.content) {
|
||||
fullContent += delta.content;
|
||||
}
|
||||
}
|
||||
console.log(
|
||||
`Chunk ${count}: ${JSON.stringify(parsed).substring(0, 150)}`,
|
||||
);
|
||||
} catch (e) {
|
||||
console.log(`Raw chunk ${count}:`, data.substring(0, 100));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
console.log("\n=== Full Content ===");
|
||||
console.log(fullContent);
|
||||
console.log("===================");
|
||||
21
tsconfig.json
Normal file
21
tsconfig.json
Normal file
@@ -0,0 +1,21 @@
|
||||
{
|
||||
"compilerOptions": {
|
||||
"target": "ES2022",
|
||||
"module": "ES2022",
|
||||
"moduleResolution": "bundler",
|
||||
"lib": ["ES2022"],
|
||||
"outDir": "./dist",
|
||||
"rootDir": "./src",
|
||||
"strict": true,
|
||||
"esModuleInterop": true,
|
||||
"skipLibCheck": true,
|
||||
"forceConsistentCasingInFileNames": true,
|
||||
"resolveJsonModule": true,
|
||||
"declaration": true,
|
||||
"declarationMap": true,
|
||||
"sourceMap": true,
|
||||
"types": ["node"]
|
||||
},
|
||||
"include": ["src/**/*"],
|
||||
"exclude": ["node_modules", "dist", "test"]
|
||||
}
|
||||
Reference in New Issue
Block a user