[ { "path": ".dockerignore", "content": "**/node_modules\n**/dist" }, { "path": ".github/workflows/docker-image.yml", "content": "name: e2e test\n\non:\n push:\n branches: \"**\"\n pull_request:\n branches: \"**\"\nenv:\n PG_DB: postgres\n PG_USER: postgres\n PG_HOST: localhost\n PG_PASS: thisisapassword\n\njobs:\n build_chat:\n runs-on: ubuntu-latest\n steps:\n - uses: actions/checkout@v3\n\n - name: check locale\n run: python scripts/locale_missing_key.py web/src/locales --base zh-CN\n\n - name: Use Node.js\n uses: actions/setup-node@v3\n with:\n node-version: \"18.x\"\n\n - name: build web\n run: |\n npm install\n npm run test\n npm run build\n working-directory: web\n - name: copy to api/static\n run: |\n cp -R web/dist/* api/static/\n - name: Set up Go\n uses: actions/setup-go@v3\n with:\n go-version: 1.19\n - name: Build API\n run: go build -v ./...\n working-directory: api\n - name: Test API\n run: go test -v ./...\n working-directory: api\n - name: Build Chat image\n run: |\n docker build . --file Dockerfile -t ghcr.io/swuecho/chat:${GITHUB_SHA}\n docker tag ghcr.io/swuecho/chat:${GITHUB_SHA} ghcr.io/swuecho/chat:latest\n\n - name: docker compose\n run: docker compose up -d\n\n - name: show docker ps\n run: docker compose ps\n\n - name: show docker logs\n run: docker compose logs\n\n # Setup cache for node_modules\n - name: Cache node modules\n uses: actions/cache@v3\n with:\n path: e2e/node_modules\n key: ${{ runner.os }}-node-${{ hashFiles('e2e/package-lock.json') }}\n restore-keys: |\n ${{ runner.os }}-node-\n\n - name: Install dependencies\n run: npm ci\n working-directory: e2e\n\n - name: Install playwright browsers\n run: npx playwright install --with-deps\n working-directory: e2e\n\n - run: npx playwright test\n working-directory: e2e\n" }, { "path": ".github/workflows/fly.yml", "content": "name: Fly Deploy\non:\n push:\n branches:\n - master\njobs:\n deploy:\n name: Deploy app\n runs-on: ubuntu-latest\n steps:\n - uses: actions/checkout@v3\n - uses: superfly/flyctl-actions/setup-flyctl@master\n - run: flyctl deploy --remote-only\n env:\n FLY_API_TOKEN: ${{ secrets.FLY_API_TOKEN }}\n" }, { "path": ".github/workflows/mobile-build.yml", "content": "name: Build Mobile Package\n\non:\n workflow_dispatch:\n push:\n paths:\n - \"mobile/**\"\n - \".github/workflows/mobile-build.yml\"\n pull_request:\n paths:\n - \"mobile/**\"\n - \".github/workflows/mobile-build.yml\"\n\njobs:\n build-android:\n runs-on: ubuntu-latest\n defaults:\n run:\n working-directory: mobile\n steps:\n - uses: actions/checkout@v3\n - name: Set up Flutter\n uses: subosito/flutter-action@v2\n with:\n channel: \"stable\"\n flutter-version: \"3.35.4\"\n - name: Install dependencies\n run: flutter pub get\n - name: Build Android APK\n run: flutter build apk --release\n - name: Build Android App Bundle\n run: flutter build appbundle --release\n - name: Upload Android artifacts\n uses: actions/upload-artifact@v4\n with:\n name: chat-mobile-android\n path: |\n mobile/build/app/outputs/flutter-apk/app-release.apk\n mobile/build/app/outputs/bundle/release/app-release.aab\n build-ios:\n runs-on: macos-latest\n defaults:\n run:\n working-directory: mobile\n steps:\n - uses: actions/checkout@v3\n - name: Set up Flutter\n uses: subosito/flutter-action@v2\n with:\n channel: \"stable\"\n flutter-version: \"3.35.4\"\n - name: Install dependencies\n run: flutter pub get\n - name: Build iOS app (no codesign)\n run: flutter build ios --simulator\n - name: Upload iOS artifact\n uses: actions/upload-artifact@v4\n with:\n name: chat-mobile-ios\n path: mobile/build/ios/iphonesimulator/Runner.app\n" }, { "path": ".github/workflows/publish.yml", "content": "name: Publish\n\non:\n push:\n tags:\n - \"v*\"\n\njobs:\n build_api:\n runs-on: ubuntu-latest\n steps:\n - uses: actions/checkout@v3\n - name: Use Node.js\n uses: actions/setup-node@v3\n with:\n node-version: \"18.x\"\n - name: build web\n run: |\n npm install\n npm run test \n npm run build\n working-directory: web\n - name: copy to api/static\n run: |\n cp -R web/dist/* api/static/\n - name: Set up Go\n uses: actions/setup-go@v3\n with:\n go-version: 1.24\n - name: Build Chat Binary\n run: go build -v ./...\n working-directory: api\n - name: Test Chat\n run: go test -v ./...\n working-directory: api\n # use root folder docker\n - name: Build Chat image\n run: |\n docker build . --file Dockerfile -t ghcr.io/swuecho/chat:${GITHUB_REF#refs/tags/}\n - name: Login to GitHub Container Registry\n run: echo \"${{ secrets.GHCR_TOKEN }}\" | docker login ghcr.io -u ${{ github.actor }} --password-stdin\n - name: Push API image to GitHub Container Registry\n run: |\n docker push ghcr.io/swuecho/chat:${GITHUB_REF#refs/tags/}\n docker tag ghcr.io/swuecho/chat:${GITHUB_REF#refs/tags/} ghcr.io/swuecho/chat:latest\n docker push ghcr.io/swuecho/chat:latest\n - name: Login to Docker Hub\n uses: docker/login-action@v2\n with:\n username: ${{ secrets.DOCKERHUB_USERNAME }}\n password: ${{ secrets.DOCKERHUB_TOKEN }}\n - name: push to docker\n run: |\n docker tag ghcr.io/swuecho/chat:${GITHUB_REF#refs/tags/} echowuhao/chat:${GITHUB_REF#refs/tags/}\n docker tag ghcr.io/swuecho/chat:${GITHUB_REF#refs/tags/} echowuhao/chat:latest\n docker push echowuhao/chat:${GITHUB_REF#refs/tags/}\n docker push echowuhao/chat:latest\n" }, { "path": ".gitignore", "content": ".DS_Store\n.env*\nenv.sh\nenv.ps\ndata\n.python-version\n.aider*\n" }, { "path": "AGENTS.md", "content": "# Chat - Multi-LLM Chat Interface\n\nA full-stack chat application that provides a unified interface for interacting with multiple large language models (LLMs) including OpenAI, Claude, Gemini, and Ollama.\n\n## Project Overview\n\nThis project is a ChatGPT wrapper that extends beyond OpenAI to support multiple LLM providers. It features a Vue.js frontend with a Go backend, using PostgreSQL for data persistence.\n\n### Key Features\n\n- **Multi-LLM Support**: OpenAI, Claude, Gemini, and Ollama models\n- **Chat Sessions**: Persistent conversation history and context management\n- **Workspaces**: Organize chat sessions into customizable workspaces with colors and icons\n- **User Management**: Authentication, rate limiting, and admin controls\n- **File Uploads**: Support for text and multimedia files (model-dependent)\n- **Snapshots**: Shareable conversation snapshots with full-text search\n- **Prompt Management**: Built-in prompt templates with '/' shortcut\n- **Internationalization**: Support for multiple languages\n\n## Architecture\n\n### Frontend (Vue.js)\n- **Framework**: Vue 3 with Composition API\n- **State Management**: Pinia\n- **UI Library**: Naive UI\n- **Routing**: Vue Router\n- **Build Tool**: Rsbuild\n- **Styling**: Tailwind CSS + Less\n\n### Backend (Go)\n- **Framework**: Standard Go HTTP with Gorilla Mux\n- **Database**: PostgreSQL with SQLC for type-safe queries\n- **Authentication**: JWT tokens\n- **Rate Limiting**: Built-in rate limiting (100 calls/10min default)\n- **File Upload**: Support for various file types\n\n### Database (PostgreSQL)\n- **Schema**: Located in `api/sqlc/schema.sql`\n- **Queries**: Type-safe queries generated by SQLC\n- **Tables**: Users, sessions, messages, models, prompts, snapshots, workspaces, etc.\n\n## Project Structure\n\n```\nchat/\n├── api/ # Go backend\n│ ├── main.go # Application entry point\n│ ├── llm/ # LLM provider integrations\n│ │ ├── openai/ # OpenAI integration\n│ │ ├── claude/ # Claude integration\n│ │ └── gemini/ # Gemini integration\n│ ├── sqlc/ # Database schema and queries\n│ ├── sqlc_queries/ # Generated type-safe queries\n│ └── static/ # Static assets\n├── web/ # Vue.js frontend\n│ ├── src/\n│ │ ├── components/ # Reusable components\n│ │ ├── views/ # Page components\n│ │ ├── store/ # Pinia stores\n│ │ ├── api/ # API client\n│ │ └── utils/ # Utility functions\n│ └── dist/ # Built frontend assets\n├── docs/ # Documentation\n├── e2e/ # End-to-end tests\n└── data/ # Database dumps\n```\n\n## Development Setup\n\n### Prerequisites\n- Go 1.19+\n- Node.js 18+\n- PostgreSQL\n- (Optional) Docker\n\n### Backend Setup\n```bash\ncd api\ngo mod tidy\ngo run main.go\n```\n\n### Frontend Setup\n```bash\ncd web\nnpm install\nnpm run dev\n```\n\n### server reload\n\nboth frontend and backend will be auto-reload whe code change.\n\n### Database Setup\n\n1. Create a PostgreSQL database\n2. Run the schema from `api/sqlc/schema.sql`\n3. Configure database connection in environment variables\n\n## Configuration\n\n### Environment Variables\n- `OPENAI_API_KEY`: OpenAI API key\n- `CLAUDE_API_KEY`: Claude API key\n- `GEMINI_API_KEY`: Gemini API key\n- `DATABASE_URL`: PostgreSQL connection string\n- `JWT_SECRET`: JWT signing secret\n- `OPENAI_RATELIMIT`: Rate limit (default: 100)\n\n## API Endpoints\n\n### Authentication\n- `POST /api/auth/login` - User login\n- `POST /api/auth/register` - User registration\n- `POST /api/auth/refresh` - Token refresh\n\n### Chat\n\n- `GET /api/chat/sessions` - Get user sessions\n- `POST /api/chat/sessions` - Create new session\n- `GET /api/chat/messages` - Get session messages\n- `POST /api/chat/messages` - Send message\n- `DELETE /api/chat/sessions/:id` - Delete session\n\n### Workspaces\n- `GET /api/workspaces` - Get user workspaces\n- `POST /api/workspaces` - Create new workspace\n- `GET /api/workspaces/{uuid}` - Get workspace by UUID\n- `PUT /api/workspaces/{uuid}` - Update workspace\n- `DELETE /api/workspaces/{uuid}` - Delete workspace\n- `PUT /api/workspaces/{uuid}/reorder` - Update workspace order\n- `PUT /api/workspaces/{uuid}/set-default` - Set default workspace\n- `POST /api/workspaces/{uuid}/sessions` - Create session in workspace\n- `GET /api/workspaces/{uuid}/sessions` - Get sessions in workspace\n- `POST /api/workspaces/default` - Ensure default workspace exists\n\n### Models\n- `GET /api/models` - List available models\n- `POST /api/models` - Add new model (admin)\n- `PUT /api/models/:id` - Update model (admin)\n\n### File Upload\n- `POST /api/files` - Upload file\n- `GET /api/files/:id` - Get file\n\n## Key Components\n\n### Frontend Components\n- `chat/index.vue`: Main chat interface\n- `chat/components/Message/`: Message rendering\n- `chat/components/Session/`: Session management\n- `admin/index.vue`: Admin dashboard\n- `prompt/creator.vue`: Prompt management\n\n### Backend Handlers\n- `chat_main_handler.go`: Core chat functionality\n- `chat_session_handler.go`: Session management\n- `chat_message_handler.go`: Message handling\n- `chat_workspace_handler.go`: Workspace management\n- `auth_handler.go`: Authentication\n- `admin_handler.go`: Admin operations\n\n## Testing\n\n### Backend Tests\n```bash\ncd api\ngo test ./...\n```\n\n### Frontend Tests\n```bash\ncd e2e\nnpm test\n```\n\n### E2E Tests\n```bash\ncd e2e\nnpm test\n```\n\n\n## Workspaces\n\nThe application supports organizing chat sessions into workspaces for better organization and context management.\n\n### Workspace Features\n\n- **Custom Organization**: Create themed workspaces for different projects or topics\n- **Visual Customization**: Set custom colors and icons for easy identification\n- **Session Management**: Sessions are automatically associated with workspaces\n- **Default Workspace**: Each user has a default \"General\" workspace that's created automatically\n- **Ordering**: Workspaces can be reordered for personal preference\n- **Permission Control**: Users can only access their own workspaces\n\n### Workspace Properties\n\n- **Name**: Human-readable workspace name\n- **Description**: Optional description for workspace purpose\n- **Color**: Hex color code for visual theming (default: #6366f1)\n- **Icon**: Icon identifier for visual representation (default: folder)\n- **Default**: Boolean flag indicating if this is the user's default workspace\n- **Order Position**: Integer for custom workspace ordering\n\n### Database Schema\n\nThe `chat_workspace` table includes:\n- `id`: Primary key\n- `uuid`: Unique identifier for API operations\n- `user_id`: Foreign key to owner user\n- `name`: Workspace name (required)\n- `description`: Optional description (default: empty string)\n- `color`: Hex color code (default: #6366f1)\n- `icon`: Icon identifier (default: folder)\n- `is_default`: Boolean default flag\n- `order_position`: Integer for ordering\n- `created_at`/`updated_at`: Timestamps\n\nSessions are linked to workspaces via the `workspace_id` foreign key in the `chat_session` table.\n\n## Model Support\n\n- **OpenAI**: GPT-3.5, GPT-4 models\n- **Claude**: Claude 3 Opus, Sonnet, Haiku, Claude 4\n- **Gemini**: Gemini Pro\n- **Ollama**: Local model hosting support" }, { "path": "CLAUDE.md", "content": "# CLAUDE.md\n\nThis file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.\n\n## Project Overview\n\nMulti-LLM chat interface with Vue.js frontend, Go backend, and PostgreSQL database. Supports OpenAI, Claude, Gemini, and Ollama models with features like workspaces, snapshots, and file uploads.\n\n## Development Commands\n\n### Backend (Go)\n```bash\ncd api\n\n# Install dependencies and hot-reload tool\nmake install\n\n# Run server with hot reload (uses Air)\nmake serve\n\n# Build\nmake build\n\n# Format code\nmake fmt\n\n# Run tests\ngo test ./...\n\n# Regenerate SQLC code (after modifying queries or schema)\nsqlc generate\n```\n\n**Important**: The backend uses Air for hot-reloading during development. Configuration is in `api/.air.toml`.\n\n### Frontend (Vue.js)\n```bash\ncd web\n\n# Install dependencies\nnpm install\n\n# Development server with hot reload\nnpm run dev\n\n# Build for production\nnpm run build\n\n# Run linter\nnpm run lint\n\n# Fix linting issues\nnpm run lint:fix\n\n# Run tests\nnpm test\n```\n\n### E2E Tests (Playwright)\n```bash\ncd e2e\n\n# Run all tests\nnpx playwright test\n\n# Run with UI\nnpx playwright test --ui\n```\n\n## Architecture\n\n### Request Flow\n```\nHTTP Request → Mux Router → Handler → Service → SQLC Queries → PostgreSQL\n ↓\n LLM Provider (OpenAI/Claude/Gemini/Ollama)\n```\n\n### Backend Architecture (Go)\n\n**Key Pattern**: The backend follows a handler → service → repository (SQLC) pattern:\n\n1. **Handlers** (`*_handler.go`): HTTP request/response handling\n - `chat_main_handler.go`: Core chat functionality\n - `chat_session_handler.go`: Session CRUD operations\n - `chat_message_handler.go`: Message operations\n - `chat_workspace_handler.go`: Workspace management\n - `chat_auth_user_handler.go`: Authentication\n - `admin_handler.go`: Admin operations\n\n2. **Services** (`*_service.go`): Business logic layer\n - `chat_main_service.go`: Chat orchestration and LLM routing\n\n3. **SQLC Generated Code** (`sqlc_queries/`): Type-safe database queries\n - Schema: `api/sqlc/schema.sql`\n - Queries: `api/sqlc/queries/*.sql`\n - Generated Go: `api/sqlc_queries/*.go`\n - Config: `api/sqlc.yaml`\n\n4. **LLM Integrations** (`llm/`):\n - `llm/openai/`: OpenAI API client\n - `llm/claude/`: Claude API client\n - `llm/gemini/`: Gemini API client\n - Each provider has its own request/response formatting\n\n**Router**: Uses Gorilla Mux for routing (configured in `main.go`)\n\n### Frontend Architecture (Vue.js)\n\n**Stack**: Vue 3 (Composition API) + Pinia + Naive UI + Rsbuild + Tailwind CSS\n\n**Key Directories**:\n- `web/src/views/`: Page components\n- `web/src/components/`: Reusable components\n- `web/src/store/modules/`: Pinia stores for state management\n- `web/src/api/`: API client functions\n- `web/src/views/chat/composables/`: Chat feature composables (refactored from monolithic component)\n\n**Chat Composables Pattern**: The main chat interface uses a composable-based architecture for better separation of concerns:\n- `useStreamHandling.ts`: Handles LLM streaming responses\n- `useConversationFlow.ts`: Manages conversation lifecycle\n- `useRegenerate.ts`: Message regeneration\n- `useSearchAndPrompts.ts`: Search and prompt templates\n- `useChatActions.ts`: Snapshot, bot creation, file uploads\n- `useErrorHandling.ts`: Centralized error management\n- `useValidation.ts`: Input validation rules\n- `usePerformanceOptimizations.ts`: Debouncing, memoization\n\nThis pattern reduced the main component from 738 to 293 lines while adding better error handling and type safety.\n\n### Database (PostgreSQL + SQLC)\n\n**SQLC Workflow**:\n1. Define schema in `api/sqlc/schema.sql`\n2. Write SQL queries in `api/sqlc/queries/*.sql`\n3. Run `sqlc generate` to create type-safe Go code\n4. Use generated code in services\n\n**Key Tables**:\n- `auth_user`: User accounts (first registered user becomes admin)\n- `chat_session`: Chat sessions\n- `chat_message`: Messages within sessions\n- `chat_workspace`: Workspace organization\n- `chat_model`: Available LLM models\n- `chat_prompt`: Prompt templates\n- `chat_snapshot`: Shareable conversation snapshots\n- `chat_file`: File uploads\n\n**Default Context**: Latest 4 messages are included in context by default.\n\n## Environment Variables\n\nRequired variables (set in shell or `.env`):\n```bash\n# Database (required)\nDATABASE_URL=postgres://user:pass@host:port/dbname?sslmode=disable\n\n# LLM API Keys (at least one required)\nOPENAI_API_KEY=sk-...\nCLAUDE_API_KEY=...\nGEMINI_API_KEY=...\nDEEPSEEK_API_KEY=...\n\n# Optional\nOPENAI_RATELIMIT=100 # Calls per 10 minutes (default: 100)\nJWT_SECRET=... # For JWT token signing\n```\n\n**Note**: The \"debug\" model doesn't require API keys for testing.\n\n## Key Features & Patterns\n\n### Authentication & Authorization\n- JWT-based authentication\n- First registered user becomes administrator (`is_superuser=true`)\n- Rate limiting per user (default: 100 calls/10min, configurable via `OPENAI_RATELIMIT`)\n- Per-model rate limiting available for specific models (GPT-4, etc.)\n\n### Workspaces\n- Sessions are organized into workspaces\n- Each user has a default \"General\" workspace\n- Custom colors and icons for visual organization\n- Workspace-specific session isolation\n\n### Chat Flow\n1. First message in a session is the system message (prompt)\n2. User sends message → Handler validates → Service routes to appropriate LLM provider\n3. LLM streams response → Server-Sent Events (SSE) → Frontend renders incrementally\n4. Messages stored in PostgreSQL with full history\n\n### File Uploads\n- Text files supported for all models\n- Multimedia files require model support (GPT-4 Vision, Claude 3+, Gemini)\n- Files associated with messages via `chat_file` table\n\n### Snapshots\n- Create shareable static pages from conversations (like ShareGPT)\n- Full-text search support (English) for organizing conversation history\n- Can continue conversations from snapshots\n\n### Prompt Management\n- Built-in prompt templates stored in `chat_prompt` table\n- Quick access via '/' shortcut in chat interface\n\n## Testing\n\n### Running Backend Tests\n```bash\ncd api\ngo test ./...\n```\n\n### Running Frontend Tests\n```bash\ncd web\nnpm test\n```\n\n### Running E2E Tests\n```bash\ncd e2e\nexport DATABASE_URL=postgres://...\nnpx playwright test\n```\n\n## Adding a New LLM Model\n\nSee documentation: `docs/add_model_en.md` and `docs/add_model_zh.md`\n\n**Summary**:\n1. Add model configuration to `chat_model` table (via admin UI or SQL)\n2. Implement provider in `api/llm//` if new provider type\n3. Update routing logic in `chat_main_service.go` if needed\n4. Set appropriate `api_type` field: `openai`, `claude`, `gemini`, `ollama`, or `custom`\n\n## Common Gotchas\n\n1. **SQLC Code Generation**: After modifying `schema.sql` or query files, always run `sqlc generate` from the `api/` directory\n2. **Hot Reload**: Both frontend (Rsbuild) and backend (Air) auto-reload on code changes\n3. **Database Migrations**: Schema changes are handled via `ALTER TABLE IF NOT EXISTS` statements in `schema.sql`\n4. **Rate Limiting**: Applies globally (100/10min) unless per-model rate limiting is enabled\n5. **Model API Types**: The `api_type` column determines which LLM provider client is used\n6. **Session Context**: By default, only the latest 4 messages are sent to the LLM (plus system prompt)\n7. **Title Generation**: Conversation titles are optionally generated by `gemini-2.0-flash`; if not configured, uses first 100 chars of prompt\n\n## Documentation\n\n- Local development: `docs/dev_locally_en.md`, `docs/dev_locally_zh.md`\n- Deployment: `docs/deployment_en.md`, `docs/deployment_zh.md`\n- Ollama integration: `docs/ollama_en.md`, `docs/ollama_zh.md`\n- Snapshots vs ChatBots: `docs/snapshots_vs_chatbots_en.md`\n- Adding models: `docs/add_model_en.md`\n- Dev documentation: `docs/dev/` (VFS, error handling, integration guides)\n\n## Technology Stack Summary\n\n**Frontend**: Vue 3, Pinia, Naive UI, Rsbuild, Tailwind CSS, TypeScript\n**Backend**: Go, Gorilla Mux, SQLC, PostgreSQL, Air (hot reload)\n**Testing**: Playwright (E2E), Vitest (frontend unit tests)\n**LLM SDKs**: Custom HTTP clients for each provider\n" }, { "path": "Dockerfile", "content": "FROM node:16 as frontend_builder\n\n# Set the working directory to /app\nWORKDIR /app\n\n# Copy the package.json and package-lock.json files to the container\nCOPY web/package*.json ./\n\n# Install dependencies\nRUN npm install\n\n# Copy the remaining application files to the container\nCOPY web/ .\n# Build the application\nRUN npm run build\n\nFROM golang:1.24-alpine3.20 AS builder\n\nWORKDIR /app\n\nCOPY api/go.mod api/go.sum ./\nRUN go mod download\n\nCOPY api/ .\n# cp -rf /app/dist/* /app/static/\nCOPY --from=frontend_builder /app/dist/ ./static/\n\nRUN CGO_ENABLED=0 GOARCH=amd64 GOOS=linux go build -a -installsuffix cgo -o /app/app\n\nFROM alpine:3.20\n\nWORKDIR /app\n\nCOPY --from=builder /app/app /app\n# for go timezone work\nCOPY --from=builder /usr/local/go/lib/time/zoneinfo.zip /app/zoneinfo.zip\nENV ZONEINFO=/app/zoneinfo.zip \n\nEXPOSE 8080\n\nENTRYPOINT [\"/app/app\"]\n" }, { "path": "README.md", "content": "## Demo\n\n\n\"image\"\n\n\n\"image\"\n\n\"thinking\"\n\n\n![image](https://github.com/user-attachments/assets/ad38194e-dd13-4eb0-b946-81c29a37955d)\n\n\n\"image\"\n\n\"Screenshot\n\n\n![image](https://github.com/user-attachments/assets/5b3751e4-eaa1-4a79-b47a-9b073c63eb04)\n\n\"image\"\n\n\"chat\n\n\"chat\n\n\n\n\n\n## 规则\n\n- 第一个消息是系统消息（prompt）\n- 上下文默认附带最新创建的4条消息\n- 第一个注册的用户是管理员\n- 默认限流 100 chatGPT call /10分钟 (OPENAI_RATELIMIT=100)\n- 根据对话生成可以分享的静态页面(like ShareGPT), 也可以继续会话. \n- 对话快照目录(对话集), 支持全文查找(English), 方便整理, 搜索会话记录.\n- 支持OPEN AI, Claude 模型 \n- 支持Ollama host模型, 配置参考: https://github.com/swuecho/chat/discussions/396\n- 支持上传文本文件\n- 支持多媒体文件, 需要模型支持\n- 提示词管理, 提示词快捷键 '/'\n\n> （可选）对话标题用 `gemini-2.0-flash` 生成， 所以需要配置该模型， 不配置默认用提示词前100个字符\n\n## 文档\n\n- [添加新模型指南](https://github.com/swuecho/chat/blob/master/docs/add_model_zh.md)\n- [快照 vs 聊天机器人](https://github.com/swuecho/chat/blob/master/docs/snapshots_vs_chatbots_zh.md)\n- [使用本地Ollama](https://github.com/swuecho/chat/blob/master/docs/ollama_zh.md)\n- [论坛](https://github.com/swuecho/chat/discussions)\n\n## 开发指南\n\n- [本地开发指南](https://github.com/swuecho/chat/blob/master/docs/dev_locally_zh.md)\n\n## 部署指南\n\n- [部署指南](https://github.com/swuecho/chat/blob/master/docs/deployment_zh.md)\n\n## 致谢\n\n- web: [ChatGPT-Web](https://github.com/Chanzhaoyu/chatgpt-web) 复制过来的 。\n- api : 参考 [Kerwin1202](https://github.com/Kerwin1202)'s [Chanzhaoyu/chatgpt-web#589](https://github.com/Chanzhaoyu/chatgpt-web/pull/589) 的node版本在chatgpt帮助下写的\n\n## LICENCE: MIT\n\n## Rules\n\n- The first message is a system message (prompt)\n- By default, the latest 4 messages are included in context\n- The first registered user becomes administrator\n- Default rate limit: 100 ChatGPT calls per 10 minutes (OPENAI_RATELIMIT=100)\n- Generate shareable static pages from conversations (like ShareGPT), or continue conversations\n- Conversation snapshots directory supports full-text search (English), making it easy to organize and search conversation history\n- Supports OpenAI and Claude models\n- Supports Ollama host models, configuration reference: https://github.com/swuecho/chat/discussions/396\n- Supports text file uploads\n- Supports multimedia files (requires model support)\n- Prompt management with '/' shortcut\n\n> (Optional) Conversation titles are generated by `gemini-2.0-flash`, so this model needs to be configured. If not configured, the first 100 characters of the prompt will be used as the title.\n\n## Documentation\n\n- [Adding New Models Guide](https://github.com/swuecho/chat/blob/master/docs/add_model_en.md)\n- [Snapshots vs ChatBots](https://github.com/swuecho/chat/blob/master/docs/snapshots_vs_chatbots_en.md)\n- [Using Local Ollama](https://github.com/swuecho/chat/blob/master/docs/ollama_en.md)\n- [Community Discussions](https://github.com/swuecho/chat/discussions)\n\n## Development Guide\n\n- [Local Development Guide](https://github.com/swuecho/chat/blob/master/docs/dev_locally_en.md)\n\n## Deployment Guide\n\n- [Deployment Guide](https://github.com/swuecho/chat/blob/master/docs/deployment_en.md)\n\n## Acknowledgments\n\n- web: copied from chatgpt-web \n- api: based on the node version of [Kerwin1202](https://github.com/Kerwin1202)'s [Chanzhaoyu/chatgpt-web#589](https://github.com/Chanzhaoyu/chatgpt-web/pull/589)\nand written with the help of chatgpt.\n" }, { "path": "api/.air.toml", "content": "root = \".\"\ntestdata_dir = \"testdata\"\ntmp_dir = \"tmp\"\n\n[build]\n args_bin = []\n #bin = \"./tmp/main\"\n #cmd = \"go build -o ./tmp/main .\"\n delay = 0\n exclude_dir = [\"assets\", \"tmp\", \"vendor\", \"testdata\"]\n exclude_file = []\n exclude_regex = [\"_test.go\"]\n exclude_unchanged = false\n follow_symlink = false\n full_bin = \"\"\n include_dir = []\n include_ext = [\"go\", \"tpl\", \"tmpl\", \"html\"]\n include_file = []\n kill_delay = \"0s\"\n log = \"build-errors.log\"\n rerun = false\n rerun_delay = 500\n send_interrupt = false\n stop_on_error = false\n\n[color]\n app = \"\"\n build = \"yellow\"\n main = \"magenta\"\n runner = \"green\"\n watcher = \"cyan\"\n\n[log]\n main_only = false\n time = false\n\n[misc]\n clean_on_exit = false\n\n[screen]\n clear_on_rebuild = false\n keep_scroll = true\n" }, { "path": "api/.github/workflows/go.yml", "content": "# This workflow will build a golang project\n# For more information see: https://docs.github.com/en/actions/automating-builds-and-tests/building-and-testing-go\n\nname: Go\n\non:\n push:\n branches: [ \"master\" ]\n pull_request:\n branches: [ \"master\" ]\n\njobs:\n\n build:\n runs-on: ubuntu-latest\n steps:\n - uses: actions/checkout@v3\n\n - name: Set up Go\n uses: actions/setup-go@v3\n with:\n go-version: 1.24\n\n - name: Build\n run: go build -v ./...\n\n - name: Test\n run: go test -v ./...\n\n # build docker image base on Dockerfile\n \n\n" }, { "path": "api/.gitignore", "content": "tmp/\nchat_backend\nenv.sh\nstatic/static" }, { "path": "api/.vscode/settings.json", "content": "{\n \"editor.fontFamily\": \"Go Mono\",\n \"go.useLanguageServer\": true,\n \"files.watcherExclude\": {\n \"**/target\": true\n }\n}" }, { "path": "api/LICENSE", "content": "MIT License\n\nCopyright (c) 2023-2024 Hao Wu\n\nPermission is hereby granted, free of charge, to any person obtaining a copy\nof this software and associated documentation files (the \"Software\"), to deal\nin the Software without restriction, including without limitation the rights\nto use, copy, modify, merge, publish, distribute, sublicense, and/or sell\ncopies of the Software, and to permit persons to whom the Software is\nfurnished to do so, subject to the following conditions:\n\nThe above copyright notice and this permission notice shall be included in all\ncopies or substantial portions of the Software.\n\nTHE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\nIMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\nFITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\nAUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\nLIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\nOUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\nSOFTWARE.\n" }, { "path": "api/Makefile", "content": ".DEFAULT_GOAL:=build\n\nfmt:\n\tgo fmt ./...\n\n.PHONY: fmt\n\nlint: fmt\n\tgolint ./...\n\n.PHONY: lint\n\nvet: fmt\n\tgo vet ./...\n\n.PHONY: vet\n\nbuild: vet\n\tgo build\n.PHONY: build\n\ninstall:\n\tgo install github.com/air-verse/air@latest\n\tgo mod tidy\n \nserve:\n\t@echo \"Starting server...\"\n\techo 'sudo lsof -i -P -n | grep 8080'\n\techo $(OPENAI_API_KEY)\n\techo $(PG_HOST)\n\tair\n \n\n\n" }, { "path": "api/README.md", "content": "# architecture\n\nrequest -> mux(router) -> sql generated code -> database -> sql\n\n## library used\n\n1. sqlc to connect go code to sql (sql is mostly generated by chatgpt)\n2. mux as router\n\n## ChatGPT version\n\nWhen it comes to building web applications with Go, there are several choices for libraries and frameworks. One of the most popular options for creating a web server is mux, which provides a flexible and powerful routing system.\n\nIn addition to handling incoming requests and routing them to the appropriate handler, a web application will typically need to interact with a database. This is where sqlc comes in, a Go library that helps connect your Go code to your database using SQL.\n\nUsing these two tools together, you can quickly create a powerful and efficient web application that uses SQL for data storage and retrieval.\n\nHere's how it all fits together. When a request comes into your application, it's handled by mux. Mux inspects the incoming request and routes it to the appropriate function or handler in your Go code.\n\nYour Go code, in turn, uses sqlc generated go code (based on SQL) that interacts with your database. This generated go code is used to fetch data, store new data, and update existing data. The SQL code is compiled and executed by your database, and the result is returned to your Go code.\n\nOverall, this architecture provides a clean and modular approach to building web applications with Go. By leveraging powerful libraries like mux and sqlc, you can focus on writing application logic rather than worrying about the low-level details of routing and database access.\n\nIn summary, if you're building a web application with Go, you should definitely consider using mux as your router and sqlc to connect your Go code to your database. The combination of these two libraries makes it easy to build scalable and reliable web applications that are both easy to maintain and performant." }, { "path": "api/admin_handler.go", "content": "package main\n\nimport (\n\t\"encoding/json\"\n\t\"net/http\"\n\t\"strconv\"\n\n\t\"github.com/gorilla/mux\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype AdminHandler struct {\n\tservice *AuthUserService\n}\n\nfunc NewAdminHandler(service *AuthUserService) *AdminHandler {\n\treturn &AdminHandler{\n\t\tservice: service,\n\t}\n}\n\nfunc (h *AdminHandler) RegisterRoutes(router *mux.Router) {\n\t// admin routes (without /admin prefix since router already handles it)\n\trouter.HandleFunc(\"/users\", h.CreateUser).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/users\", h.UpdateUser).Methods(http.MethodPut)\n\trouter.HandleFunc(\"/rate_limit\", h.UpdateRateLimit).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/user_stats\", h.UserStatHandler).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/user_analysis/{email}\", h.UserAnalysisHandler).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/user_session_history/{email}\", h.UserSessionHistoryHandler).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/session_messages/{sessionUuid}\", h.SessionMessagesHandler).Methods(http.MethodGet)\n}\n\nfunc (h *AdminHandler) CreateUser(w http.ResponseWriter, r *http.Request) {\n\tvar userParams sqlc_queries.CreateAuthUserParams\n\terr := json.NewDecoder(r.Body).Decode(&userParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tuser, err := h.service.CreateAuthUser(r.Context(), userParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to create user\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(user)\n}\n\nfunc (h *AdminHandler) UpdateUser(w http.ResponseWriter, r *http.Request) {\n\tvar userParams sqlc_queries.UpdateAuthUserByEmailParams\n\terr := json.NewDecoder(r.Body).Decode(&userParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tuser, err := h.service.q.UpdateAuthUserByEmail(r.Context(), userParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to update user\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(user)\n}\n\nfunc (h *AdminHandler) UserStatHandler(w http.ResponseWriter, r *http.Request) {\n\tvar pagination Pagination\n\terr := json.NewDecoder(r.Body).Decode(&pagination)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tuserStatsRows, total, err := h.service.GetUserStats(r.Context(), pagination, int32(appConfig.OPENAI.RATELIMIT))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get user stats\"))\n\t\treturn\n\t}\n\n\t// Create a new []interface{} slice with same length as userStatsRows\n\tdata := make([]interface{}, len(userStatsRows))\n\n\t// Copy the contents of userStatsRows into data\n\tfor i, v := range userStatsRows {\n\t\tdivider := v.TotalChatMessages3Days\n\t\tvar avg int64\n\t\tif divider > 0 {\n\t\t\tavg = v.TotalTokenCount3Days / v.TotalChatMessages3Days\n\t\t} else {\n\t\t\tavg = 0\n\t\t}\n\t\tdata[i] = UserStat{\n\t\t\tEmail: v.UserEmail,\n\t\t\tFirstName: v.FirstName,\n\t\t\tLastName: v.LastName,\n\t\t\tTotalChatMessages: v.TotalChatMessages,\n\t\t\tTotalChatMessages3Days: v.TotalChatMessages3Days,\n\t\t\tRateLimit: v.RateLimit,\n\t\t\tTotalChatMessagesTokenCount: v.TotalTokenCount,\n\t\t\tTotalChatMessages3DaysTokenCount: v.TotalTokenCount3Days,\n\t\t\tAvgChatMessages3DaysTokenCount: avg,\n\t\t}\n\t}\n\n\tjson.NewEncoder(w).Encode(Pagination{\n\t\tPage: pagination.Page,\n\t\tSize: pagination.Size,\n\t\tTotal: total,\n\t\tData: data,\n\t})\n}\n\nfunc (h *AdminHandler) UpdateRateLimit(w http.ResponseWriter, r *http.Request) {\n\tvar rateLimitRequest RateLimitRequest\n\terr := json.NewDecoder(r.Body).Decode(&rateLimitRequest)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\trate, err := h.service.q.UpdateAuthUserRateLimitByEmail(r.Context(),\n\t\tsqlc_queries.UpdateAuthUserRateLimitByEmailParams{\n\t\t\tEmail: rateLimitRequest.Email,\n\t\t\tRateLimit: rateLimitRequest.RateLimit,\n\t\t})\n\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to update rate limit\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(\n\t\tmap[string]int32{\n\t\t\t\"rate\": rate,\n\t\t})\n}\n\nfunc (h *AdminHandler) UserAnalysisHandler(w http.ResponseWriter, r *http.Request) {\n\tvars := mux.Vars(r)\n\temail := vars[\"email\"]\n\n\tif email == \"\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Email parameter is required\"))\n\t\treturn\n\t}\n\n\tanalysisData, err := h.service.GetUserAnalysis(r.Context(), email, int32(appConfig.OPENAI.RATELIMIT))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to get user analysis\"))\n\t\treturn\n\t}\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tjson.NewEncoder(w).Encode(analysisData)\n}\n\ntype SessionHistoryResponse struct {\n\tData []SessionHistoryInfo `json:\"data\"`\n\tTotal int64 `json:\"total\"`\n\tPage int32 `json:\"page\"`\n\tSize int32 `json:\"size\"`\n}\n\nfunc (h *AdminHandler) UserSessionHistoryHandler(w http.ResponseWriter, r *http.Request) {\n\tvars := mux.Vars(r)\n\temail := vars[\"email\"]\n\n\tif email == \"\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Email parameter is required\"))\n\t\treturn\n\t}\n\n\t// Parse pagination parameters\n\tpageStr := r.URL.Query().Get(\"page\")\n\tsizeStr := r.URL.Query().Get(\"size\")\n\n\tpage := int32(1)\n\tsize := int32(10)\n\n\tif pageStr != \"\" {\n\t\tif p, err := strconv.Atoi(pageStr); err == nil && p > 0 {\n\t\t\tpage = int32(p)\n\t\t}\n\t}\n\n\tif sizeStr != \"\" {\n\t\tif s, err := strconv.Atoi(sizeStr); err == nil && s > 0 && s <= 100 {\n\t\t\tsize = int32(s)\n\t\t}\n\t}\n\n\tsessionHistory, total, err := h.service.GetUserSessionHistory(r.Context(), email, page, size)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to get user session history\"))\n\t\treturn\n\t}\n\n\tresponse := SessionHistoryResponse{\n\t\tData: sessionHistory,\n\t\tTotal: total,\n\t\tPage: page,\n\t\tSize: size,\n\t}\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tjson.NewEncoder(w).Encode(response)\n}\n\nfunc (h *AdminHandler) SessionMessagesHandler(w http.ResponseWriter, r *http.Request) {\n\tvars := mux.Vars(r)\n\tsessionUuid := vars[\"sessionUuid\"]\n\n\tif sessionUuid == \"\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Session UUID parameter is required\"))\n\t\treturn\n\t}\n\n\tmessages, err := h.service.q.GetChatMessagesBySessionUUIDForAdmin(r.Context(), sessionUuid)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to get session messages\"))\n\t\treturn\n\t}\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tjson.NewEncoder(w).Encode(messages)\n}\n" }, { "path": "api/ai/model.go", "content": "package ai\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n)\n\ntype Role int\n\nconst (\n\tSystem Role = iota\n\tUser\n\tAssistant\n)\n\nfunc (r Role) String() string {\n\tswitch r {\n\tcase System:\n\t\treturn \"system\"\n\tcase User:\n\t\treturn \"user\"\n\tcase Assistant:\n\t\treturn \"assistant\"\n\tdefault:\n\t\treturn \"\"\n\t}\n}\n\nfunc StringToRole(s string) (Role, error) {\n\tswitch s {\n\tcase \"system\":\n\t\treturn System, nil\n\tcase \"user\":\n\t\treturn User, nil\n\tcase \"assistant\":\n\t\treturn Assistant, nil\n\tdefault:\n\t\treturn 0, fmt.Errorf(\"invalid role string: %s\", s)\n\t}\n}\n\nfunc (r *Role) UnmarshalJSON(data []byte) error {\n\tvar roleStr string\n\terr := json.Unmarshal(data, &roleStr)\n\tif err != nil {\n\t\treturn err\n\t}\n\tswitch roleStr {\n\tcase \"system\":\n\t\t*r = System\n\tcase \"user\":\n\t\t*r = User\n\tcase \"assistant\":\n\t\t*r = Assistant\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid role string: %s\", roleStr)\n\t}\n\treturn nil\n}\n\nfunc (r Role) MarshalJSON() ([]byte, error) {\n\tswitch r {\n\tcase System:\n\t\treturn json.Marshal(\"system\")\n\tcase User:\n\t\treturn json.Marshal(\"user\")\n\tcase Assistant:\n\t\treturn json.Marshal(\"assistant\")\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"invalid role value: %d\", r)\n\t}\n}\n" }, { "path": "api/artifact_instruction.txt", "content": "ARTIFACT CREATION GUIDELINES - MANDATORY COMPLIANCE REQUIRED:\n\n⚠️ CRITICAL: These formatting rules are REQUIRED for artifact rendering. Deviation will cause display failures.\n\n## MANDATORY ARTIFACT FORMATS (EXACT SYNTAX REQUIRED):\n\n### HTML Applications:\n```html \n[Complete HTML content with inline CSS and JavaScript(Preact)]\n```\n\n### SVG Graphics:\n```svg \n[Complete SVG markup]\n```\n\n### Mermaid Diagrams:\n```mermaid \n[Mermaid diagram syntax]\n```\n\n### JSON Data:\n```json \n[Valid JSON data]\n```\n\n### Executable Code:\n```javascript \n[JavaScript/TypeScript code]\n```\n\n```python \n[Python code]\n```\n\n## FORMATTING COMPLIANCE CHECKLIST:\n\n✅ Comment MUST be on the SAME LINE as opening ```\n✅ Use EXACT format: `` or ``\n✅ Include descriptive, specific title explaining functionality\n✅ No extra spaces or characters in comment syntax\n✅ Complete, self-contained code within blocks\n\n❌ COMMON ERRORS TO AVOID:\n- Comment on separate line from ```\n- Missing or incorrect comment format\n- Generic titles like \"Code\" or \"Example\"\n- Incomplete or broken code\n- External dependencies in HTML artifacts\n\n## WHEN TO CREATE ARTIFACTS:\n\n### ALWAYS create artifacts for:\n- Interactive web applications, forms, games, tools\n- Data visualizations, charts, graphs, dashboards\n- Diagrams, flowcharts, visual representations\n- Working code examples demonstrating functionality\n- Calculators, converters, utility applications\n- Rich data displays or formatted outputs\n- Any content meant to be rendered/executed\n\n### NEVER create artifacts for:\n- Simple text responses\n- Code snippets for reference only\n- Incomplete or pseudo-code\n- Content requiring external files\n\n## HTML ARTIFACT STANDARDS:\n\n### REQUIRED STRUCTURE:\n```html \n\n\n\n \n \n [App Title]\n \n\n\n \n \n\n\n```\n\n### TECHNICAL REQUIREMENTS:\n- Use Preact with HTM: `import { html, render } from 'https://unpkg.com/htm/preact/standalone.module.js'`\n- Modern ES6+ syntax only\n- Responsive design with proper viewport meta\n- Semantic HTML5 elements\n- Accessible UI with proper ARIA labels\n- Error handling for user interactions\n\n## EXECUTABLE CODE STANDARDS:\n\n### JavaScript/TypeScript FEATURES:\n- Output: console.log(), console.error(), console.warn()\n- Graphics: createCanvas(width, height) for visualizations\n- Libraries: `// @import libraryName` (lodash, d3, chart.js, moment, axios, rxjs, p5, three, fabric)\n- Return values automatically displayed\n- Built-in timeout and resource monitoring\n\n### Python FEATURES:\n- Output: print() for all results (auto-captured)\n- Plotting: matplotlib plots auto-displayed as PNG\n- Libraries: numpy, pandas, matplotlib, scipy, scikit-learn, requests, seaborn, plotly\n- Memory and execution monitoring included\n- No file/network access (sandboxed)\n\n## QUALITY ASSURANCE:\n\n### PRE-SUBMISSION CHECKLIST:\n1. ✅ Verify exact comment syntax on same line as ```\n2. ✅ Test all interactive functionality\n3. ✅ Ensure complete self-contained code\n4. ✅ Validate responsive design (HTML)\n5. ✅ Confirm proper error handling\n6. ✅ Check accessibility features\n7. ✅ Verify cross-browser compatibility\n\n### ARTIFACT TITLE GUIDELINES:\n- Be specific and descriptive\n- Include primary function/purpose\n- Avoid generic terms\n- Examples:\n - ✅ \"Interactive BMI Calculator with Health Recommendations\"\n - ✅ \"Real-time Stock Price Chart with Technical Indicators\"\n - ❌ \"Calculator\"\n - ❌ \"Chart\"\n\n## RENDERER BEHAVIOR:\n\nThe artifact viewer uses specialized renderers:\n- **HTML**: Full browser environment with Preact support\n- **SVG**: Native SVG rendering with interactive capabilities\n- **Mermaid**: Diagram engine with theme support\n- **JSON**: Formatted tree view with syntax highlighting\n- **JavaScript**: Node.js-like environment with canvas support\n- **Python**: Scientific computing sandbox with plot display\n\n⚠️ FINAL REMINDER: Artifacts that don't follow these exact formatting rules will fail to render. Always double-check syntax before submission." }, { "path": "api/auth/auth.go", "content": "package auth\n\nimport (\n\t\"crypto/rand\"\n\t\"crypto/sha256\"\n\t\"crypto/subtle\"\n\t\"encoding/base64\"\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/rotisserie/eris\"\n\t\"golang.org/x/crypto/pbkdf2\"\n)\n\nconst (\n\titerations = 260000\n\tsaltSize = 16\n\tkeySize = 32\n)\n\nfunc generateSalt() ([]byte, error) {\n\tsalt := make([]byte, saltSize)\n\t_, err := rand.Read(salt)\n\treturn salt, err\n}\n\nfunc GeneratePasswordHash(password string) (string, error) {\n\tsalt, err := generateSalt()\n\tif err != nil {\n\t\treturn \"\", eris.Wrap(err, \"error generating salt: \")\n\t}\n\n\thash := pbkdf2.Key([]byte(password), salt, iterations, keySize, sha256.New)\n\tencodedHash := base64.StdEncoding.EncodeToString(hash)\n\tencodedSalt := base64.StdEncoding.EncodeToString(salt)\n\n\tpasswordHash := fmt.Sprintf(\"pbkdf2_sha256$%d$%s$%s\", iterations, encodedSalt, encodedHash)\n\n\treturn passwordHash, nil\n}\n\nfunc ValidatePassword(password, hash string) bool {\n\tfields := strings.Split(hash, \"$\")\n\tif len(fields) != 4 || fields[0] != \"pbkdf2_sha256\" || fields[1] != fmt.Sprintf(\"%d\", iterations) {\n\t\treturn false\n\t}\n\tencodedSalt := fields[2]\n\tdecodedSalt, err := base64.StdEncoding.DecodeString(encodedSalt)\n\tif err != nil {\n\t\treturn false\n\t}\n\n\tencodedHash := fields[3]\n\tdecodedHash, err := base64.StdEncoding.DecodeString(encodedHash)\n\tif err != nil {\n\t\treturn false\n\t}\n\tcomputedHash := pbkdf2.Key([]byte(password), decodedSalt, iterations, keySize, sha256.New)\n\treturn subtle.ConstantTimeCompare(decodedHash, computedHash) == 1\n}\n\nfunc GenerateRandomPassword() (string, error) {\n\tconst letters = \"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890\"\n\tpassword := make([]byte, 12)\n\t_, err := rand.Read(password)\n\tif err != nil {\n\t\treturn \"\", eris.Wrap(err, \"failed to generate random password\")\n\t}\n\tfor i := 0; i < len(password); i++ {\n\t\tpassword[i] = letters[int(password[i])%len(letters)]\n\t}\n\treturn string(password), nil\n}\n" }, { "path": "api/auth/auth_test.go", "content": "package auth\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\t\"testing\"\n)\n\nfunc TestGeneratePasswordHash(t *testing.T) {\n\tpassword := \"mypassword\"\n\n\thash, err := GeneratePasswordHash(password)\n\tif err != nil {\n\t\tt.Fatalf(\"error generating password hash: %v\", err)\n\t}\n\tfmt.Println(hash)\n\t// Check that the hash has the correct format\n\tfields := strings.Split(hash, \"$\")\n\tif len(fields) != 4 || fields[0] != \"pbkdf2_sha256\" || fields[1] != \"260000\" {\n\t\tt.Errorf(\"unexpected hash format: %s\", hash)\n\t}\n\n\t// Check that we can successfully validate the password using the hash\n\tvalid := ValidatePassword(password, hash)\n\tif !valid {\n\t\tt.Error(\"generated hash does not validate password\")\n\t}\n}\n\nfunc TestGeneratePasswordHash2(t *testing.T) {\n\tpassword := \"@WuHao5\"\n\n\thash, err := GeneratePasswordHash(password)\n\tif err != nil {\n\t\tt.Fatalf(\"error generating password hash: %v\", err)\n\t}\n\tfmt.Println(hash)\n\t// Check that the hash has the correct format\n\tfields := strings.Split(hash, \"$\")\n\tif len(fields) != 4 || fields[0] != \"pbkdf2_sha256\" || fields[1] != \"260000\" {\n\t\tt.Errorf(\"unexpected hash format: %s\", hash)\n\t}\n\n\t// Check that we can successfully validate the password using the hash\n\tvalid := ValidatePassword(password, hash)\n\tif !valid {\n\t\tt.Error(\"generated hash does not validate password\")\n\t}\n}\n\nfunc TestPass(t *testing.T) {\n\thash := \"pbkdf2_sha256$260000$TSefBGfPi5fY+4whotY5sQ==$/1CeWE2PG6aYdW2DSxYyVol+HEZBmAfDj7zMgEMlxgg=\"\n\tpassword := \"using555\"\n\t// Check that we can successfully validate the password using the hash\n\tvalid := ValidatePassword(password, hash)\n\tif !valid {\n\t\tt.Error(\"generated hash does not validate password\")\n\t}\n\n}\n" }, { "path": "api/auth/token.go", "content": "package auth\n\nimport (\n\t\"encoding/base64\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"net/http\"\n\t\"strconv\"\n\t\"time\"\n\n\tjwt \"github.com/golang-jwt/jwt/v5\"\n\t\"github.com/google/uuid\"\n)\n\nfunc NewUUID() string {\n\tuuidv7, err := uuid.NewV7()\n\tif err != nil {\n\t\treturn uuid.NewString()\n\t}\n\treturn uuidv7.String()\n}\n\nvar ErrInvalidToken = errors.New(\"invalid token\")\n\nconst (\n\tTokenTypeAccess = \"access\"\n\tTokenTypeRefresh = \"refresh\"\n)\n\nfunc GenJwtSecretAndAudience() (string, string) {\n\t// Generate a random byte string to use as the secret\n\tsecretBytes := make([]byte, 32)\n\trand.Read(secretBytes)\n\n\t// Convert the byte string to a base64 encoded string\n\tsecret := base64.StdEncoding.EncodeToString(secretBytes)\n\n\t// Generate a random string to use as the audience\n\tconst letters = \"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789\"\n\taudienceBytes := make([]byte, 32)\n\tfor i := range audienceBytes {\n\t\taudienceBytes[i] = letters[rand.Intn(len(letters))]\n\t}\n\taudience := string(audienceBytes)\n\treturn secret, audience\n}\n\nfunc GenerateToken(userID int32, role string, secret, jwt_audience string, lifetime time.Duration, tokenType string) (string, error) {\n\tif tokenType == \"\" {\n\t\ttokenType = TokenTypeAccess\n\t}\n\n\texpires := time.Now().Add(lifetime).Unix()\n\tnotBefore := time.Now().Unix()\n\tissuer := \"https://www.bestqa.net\"\n\n\tclaims := jwt.MapClaims{\n\t\t\"user_id\": strconv.FormatInt(int64(userID), 10),\n\t\t\"exp\": expires,\n\t\t\"role\": role,\n\t\t\"jti\": NewUUID(),\n\t\t\"iss\": issuer,\n\t\t\"nbf\": notBefore,\n\t\t\"aud\": jwt_audience,\n\t\t\"token_type\": tokenType,\n\t}\n\n\ttoken := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)\n\ttoken.Header[\"kid\"] = \"dfsafdsafdsafadsfdasdfs\"\n\tsignedToken, err := token.SignedString([]byte(secret))\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\treturn signedToken, nil\n}\n\nfunc ValidateToken(tokenString string, secret string, expectedTokenType string) (int32, error) {\n\ttoken, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) {\n\t\t// Verify the signing algorithm and secret key used to sign the token\n\t\tif _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {\n\t\t\treturn nil, fmt.Errorf(\"unexpected signing method: %v\", token)\n\t\t}\n\t\treturn []byte(secret), nil\n\t})\n\tif err != nil {\n\t\treturn 0, ErrInvalidToken\n\t}\n\n\tclaims, ok := token.Claims.(jwt.MapClaims)\n\n\tif !ok || !token.Valid {\n\t\treturn 0, ErrInvalidToken\n\t}\n\n\tuserIDStr, ok := claims[\"user_id\"].(string)\n\tif !ok {\n\t\treturn 0, ErrInvalidToken\n\t}\n\n\ttokenType, ok := claims[\"token_type\"].(string)\n\tif !ok {\n\t\t// Support legacy tokens that were minted without token_type; treat them as access tokens\n\t\t// so existing forever tokens continue to work.\n\t\tif expectedTokenType == \"\" || expectedTokenType == TokenTypeAccess {\n\t\t\ttokenType = TokenTypeAccess\n\t\t} else {\n\t\t\treturn 0, ErrInvalidToken\n\t\t}\n\t}\n\n\tif expectedTokenType != \"\" && tokenType != expectedTokenType {\n\t\treturn 0, ErrInvalidToken\n\t}\n\n\ti, err := strconv.Atoi(userIDStr)\n\tif err != nil {\n\t\treturn -1, err\n\t}\n\n\treturn int32(i), nil\n}\n\nfunc GetExpireSecureCookie(value string, isHttps bool) *http.Cookie {\n\tutcOffset := time.Now().UTC().Add(-24 * time.Hour)\n\toptions := &http.Cookie{\n\t\tName: \"jwt\",\n\t\tValue: value,\n\t\tPath: \"/\",\n\t\tHttpOnly: true,\n\t\tSecure: isHttps,\n\t\tSameSite: http.SameSiteStrictMode,\n\t\tExpires: utcOffset,\n\t}\n\treturn options\n}\n" }, { "path": "api/auth/token_test.go", "content": "package auth\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n)\n\nfunc TestGenerateToken(t *testing.T) {\n\tuser_id := int32(0)\n\tsecret := \"abedefg\"\n\tlifetime := 8 * time.Hour\n\ttoken, err := GenerateToken(user_id, \"user\", secret, \"aud\", lifetime, TokenTypeAccess)\n\tif err != nil {\n\t\tt.Fatalf(\"error generating password hash: %v\", err)\n\t}\n\t// Check that the hash has the correct format\n\t// Check that we can successfully validate the password using the hash\n\tfmt.Println(token)\n\tuser_id_after_valid, err := ValidateToken(token, secret, TokenTypeAccess)\n\tif err != nil {\n\t\tt.Error(\"generated token does not validate \")\n\t}\n\tif user_id != user_id_after_valid {\n\t\tt.Error(\"generated token does not validate \")\n\t}\n}\n" }, { "path": "api/bot_answer_history_handler.go", "content": "package main\n\nimport (\n\t\"encoding/json\"\n\t\"net/http\"\n\t\"strconv\"\n\n\t\"github.com/gorilla/mux\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype BotAnswerHistoryHandler struct {\n\tservice *BotAnswerHistoryService\n}\n\nfunc NewBotAnswerHistoryHandler(q *sqlc_queries.Queries) *BotAnswerHistoryHandler {\n\tservice := NewBotAnswerHistoryService(q)\n\treturn &BotAnswerHistoryHandler{service: service}\n}\n\nfunc (h *BotAnswerHistoryHandler) Register(router *mux.Router) {\n\trouter.HandleFunc(\"/bot_answer_history\", h.CreateBotAnswerHistory).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/bot_answer_history/{id}\", h.GetBotAnswerHistoryByID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/bot_answer_history/bot/{bot_uuid}\", h.GetBotAnswerHistoryByBotUUID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/bot_answer_history/user/{user_id}\", h.GetBotAnswerHistoryByUserID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/bot_answer_history/{id}\", h.UpdateBotAnswerHistory).Methods(http.MethodPut)\n\trouter.HandleFunc(\"/bot_answer_history/{id}\", h.DeleteBotAnswerHistory).Methods(http.MethodDelete)\n\trouter.HandleFunc(\"/bot_answer_history/bot/{bot_uuid}/count\", h.GetBotAnswerHistoryCountByBotUUID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/bot_answer_history/user/{user_id}/count\", h.GetBotAnswerHistoryCountByUserID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/bot_answer_history/bot/{bot_uuid}/latest\", h.GetLatestBotAnswerHistoryByBotUUID).Methods(http.MethodGet)\n}\n\nfunc (h *BotAnswerHistoryHandler) CreateBotAnswerHistory(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tvar params sqlc_queries.CreateBotAnswerHistoryParams\n\tif err := json.NewDecoder(r.Body).Decode(¶ms); err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Invalid request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Set the user ID from context\n\tparams.UserID = userID\n\n\thistory, err := h.service.CreateBotAnswerHistory(ctx, params)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to create bot answer history\"))\n\t\treturn\n\t}\n\n\tRespondWithJSON(w, http.StatusCreated, history)\n}\n\nfunc (h *BotAnswerHistoryHandler) GetBotAnswerHistoryByID(w http.ResponseWriter, r *http.Request) {\n\tid := mux.Vars(r)[\"id\"]\n\tif id == \"\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"ID is required\"))\n\t\treturn\n\t}\n\n\tidInt, err := strconv.ParseInt(id, 10, 32)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Invalid ID format\"))\n\t\treturn\n\t}\n\n\thistory, err := h.service.GetBotAnswerHistoryByID(r.Context(), int32(idInt))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to get bot answer history\"))\n\t\treturn\n\t}\n\n\tRespondWithJSON(w, http.StatusOK, history)\n}\n\nfunc (h *BotAnswerHistoryHandler) GetBotAnswerHistoryByBotUUID(w http.ResponseWriter, r *http.Request) {\n\tbotUUID := mux.Vars(r)[\"bot_uuid\"]\n\tif botUUID == \"\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Bot UUID is required\"))\n\t\treturn\n\t}\n\n\tlimit, offset := getPaginationParams(r)\n\thistory, err := h.service.GetBotAnswerHistoryByBotUUID(r.Context(), botUUID, limit, offset)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to get bot answer history\"))\n\t\treturn\n\t}\n\n\t// Get total count for pagination\n\ttotalCount, err := h.service.GetBotAnswerHistoryCountByBotUUID(r.Context(), botUUID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to get bot answer history count\"))\n\t\treturn\n\t}\n\n\t// Calculate total pages\n\ttotalPages := totalCount / int64(limit)\n\tif totalCount%int64(limit) > 0 {\n\t\ttotalPages++\n\t}\n\n\t// Return paginated response\n\tRespondWithJSON(w, http.StatusOK, map[string]interface{}{\n\t\t\"items\": history,\n\t\t\"totalPages\": totalPages,\n\t\t\"totalCount\": totalCount,\n\t})\n}\n\nfunc (h *BotAnswerHistoryHandler) GetBotAnswerHistoryByUserID(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tlimit, offset := getPaginationParams(r)\n\thistory, err := h.service.GetBotAnswerHistoryByUserID(ctx, userID, limit, offset)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to get bot answer history\"))\n\t\treturn\n\t}\n\n\tRespondWithJSON(w, http.StatusOK, history)\n}\n\nfunc (h *BotAnswerHistoryHandler) UpdateBotAnswerHistory(w http.ResponseWriter, r *http.Request) {\n\tid := mux.Vars(r)[\"id\"]\n\tif id == \"\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"ID is required\"))\n\t\treturn\n\t}\n\n\tvar params sqlc_queries.UpdateBotAnswerHistoryParams\n\tif err := json.NewDecoder(r.Body).Decode(¶ms); err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Invalid request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\thistory, err := h.service.UpdateBotAnswerHistory(r.Context(), params.ID, params.Answer, params.TokensUsed)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to update bot answer history\"))\n\t\treturn\n\t}\n\n\tRespondWithJSON(w, http.StatusOK, history)\n}\n\nfunc (h *BotAnswerHistoryHandler) DeleteBotAnswerHistory(w http.ResponseWriter, r *http.Request) {\n\tid := mux.Vars(r)[\"id\"]\n\tif id == \"\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"ID is required\"))\n\t\treturn\n\t}\n\n\tidInt, err := strconv.ParseInt(id, 10, 32)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Invalid ID format\"))\n\t\treturn\n\t}\n\n\tif err := h.service.DeleteBotAnswerHistory(r.Context(), int32(idInt)); err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to delete bot answer history\"))\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusNoContent)\n}\n\nfunc (h *BotAnswerHistoryHandler) GetBotAnswerHistoryCountByBotUUID(w http.ResponseWriter, r *http.Request) {\n\tbotUUID := mux.Vars(r)[\"bot_uuid\"]\n\tif botUUID == \"\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Bot UUID is required\"))\n\t\treturn\n\t}\n\n\tcount, err := h.service.GetBotAnswerHistoryCountByBotUUID(r.Context(), botUUID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to get bot answer history count\"))\n\t\treturn\n\t}\n\n\tRespondWithJSON(w, http.StatusOK, map[string]int64{\"count\": count})\n}\n\nfunc (h *BotAnswerHistoryHandler) GetBotAnswerHistoryCountByUserID(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tcount, err := h.service.GetBotAnswerHistoryCountByUserID(ctx, userID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to get bot answer history count\"))\n\t\treturn\n\t}\n\n\tRespondWithJSON(w, http.StatusOK, map[string]int64{\"count\": count})\n}\n\nfunc (h *BotAnswerHistoryHandler) GetLatestBotAnswerHistoryByBotUUID(w http.ResponseWriter, r *http.Request) {\n\tbotUUID := mux.Vars(r)[\"bot_uuid\"]\n\tif botUUID == \"\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Bot UUID is required\"))\n\t\treturn\n\t}\n\n\tlimit := getLimitParam(r, 1)\n\thistory, err := h.service.GetLatestBotAnswerHistoryByBotUUID(r.Context(), botUUID, limit)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to get latest bot answer history\"))\n\t\treturn\n\t}\n\n\tRespondWithJSON(w, http.StatusOK, history)\n}\n" }, { "path": "api/bot_answer_history_service.go", "content": "package main\n\nimport (\n\t\"context\"\n\n\t\"github.com/rotisserie/eris\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype BotAnswerHistoryService struct {\n\tq *sqlc_queries.Queries\n}\n\n// NewBotAnswerHistoryService creates a new BotAnswerHistoryService\nfunc NewBotAnswerHistoryService(q *sqlc_queries.Queries) *BotAnswerHistoryService {\n\treturn &BotAnswerHistoryService{q: q}\n}\n\n// CreateBotAnswerHistory creates a new bot answer history entry\nfunc (s *BotAnswerHistoryService) CreateBotAnswerHistory(ctx context.Context, params sqlc_queries.CreateBotAnswerHistoryParams) (sqlc_queries.BotAnswerHistory, error) {\n\thistory, err := s.q.CreateBotAnswerHistory(ctx, params)\n\tif err != nil {\n\t\treturn sqlc_queries.BotAnswerHistory{}, eris.Wrap(err, \"failed to create bot answer history\")\n\t}\n\treturn history, nil\n}\n\n// GetBotAnswerHistoryByID gets a bot answer history entry by ID\nfunc (s *BotAnswerHistoryService) GetBotAnswerHistoryByID(ctx context.Context, id int32) (sqlc_queries.GetBotAnswerHistoryByIDRow, error) {\n\thistory, err := s.q.GetBotAnswerHistoryByID(ctx, id)\n\tif err != nil {\n\t\treturn sqlc_queries.GetBotAnswerHistoryByIDRow{}, eris.Wrap(err, \"failed to get bot answer history by ID\")\n\t}\n\treturn history, nil\n}\n\n// GetBotAnswerHistoryByBotUUID gets paginated bot answer history for a specific bot\nfunc (s *BotAnswerHistoryService) GetBotAnswerHistoryByBotUUID(ctx context.Context, botUUID string, limit, offset int32) ([]sqlc_queries.GetBotAnswerHistoryByBotUUIDRow, error) {\n\tparams := sqlc_queries.GetBotAnswerHistoryByBotUUIDParams{\n\t\tBotUuid: botUUID,\n\t\tLimit: limit,\n\t\tOffset: offset,\n\t}\n\thistory, err := s.q.GetBotAnswerHistoryByBotUUID(ctx, params)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to get bot answer history by bot UUID\")\n\t}\n\treturn history, nil\n}\n\n// GetBotAnswerHistoryByUserID gets paginated bot answer history for a specific user\nfunc (s *BotAnswerHistoryService) GetBotAnswerHistoryByUserID(ctx context.Context, userID, limit, offset int32) ([]sqlc_queries.GetBotAnswerHistoryByUserIDRow, error) {\n\tparams := sqlc_queries.GetBotAnswerHistoryByUserIDParams{\n\t\tUserID: userID,\n\t\tLimit: limit,\n\t\tOffset: offset,\n\t}\n\thistory, err := s.q.GetBotAnswerHistoryByUserID(ctx, params)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to get bot answer history by user ID\")\n\t}\n\treturn history, nil\n}\n\n// UpdateBotAnswerHistory updates an existing bot answer history entry\nfunc (s *BotAnswerHistoryService) UpdateBotAnswerHistory(ctx context.Context, id int32, answer string, tokensUsed int32) (sqlc_queries.BotAnswerHistory, error) {\n\tparams := sqlc_queries.UpdateBotAnswerHistoryParams{\n\t\tID: id,\n\t\tAnswer: answer,\n\t\tTokensUsed: tokensUsed,\n\t}\n\thistory, err := s.q.UpdateBotAnswerHistory(ctx, params)\n\tif err != nil {\n\t\treturn sqlc_queries.BotAnswerHistory{}, eris.Wrap(err, \"failed to update bot answer history\")\n\t}\n\treturn history, nil\n}\n\n// DeleteBotAnswerHistory deletes a bot answer history entry by ID\nfunc (s *BotAnswerHistoryService) DeleteBotAnswerHistory(ctx context.Context, id int32) error {\n\terr := s.q.DeleteBotAnswerHistory(ctx, id)\n\tif err != nil {\n\t\treturn eris.Wrap(err, \"failed to delete bot answer history\")\n\t}\n\treturn nil\n}\n\n// GetBotAnswerHistoryCountByBotUUID gets the count of history entries for a bot\nfunc (s *BotAnswerHistoryService) GetBotAnswerHistoryCountByBotUUID(ctx context.Context, botUUID string) (int64, error) {\n\tcount, err := s.q.GetBotAnswerHistoryCountByBotUUID(ctx, botUUID)\n\tif err != nil {\n\t\treturn 0, eris.Wrap(err, \"failed to get bot answer history count by bot UUID\")\n\t}\n\treturn count, nil\n}\n\n// GetBotAnswerHistoryCountByUserID gets the count of history entries for a user\nfunc (s *BotAnswerHistoryService) GetBotAnswerHistoryCountByUserID(ctx context.Context, userID int32) (int64, error) {\n\tcount, err := s.q.GetBotAnswerHistoryCountByUserID(ctx, userID)\n\tif err != nil {\n\t\treturn 0, eris.Wrap(err, \"failed to get bot answer history count by user ID\")\n\t}\n\treturn count, nil\n}\n\n// GetLatestBotAnswerHistoryByBotUUID gets the latest history entries for a bot\nfunc (s *BotAnswerHistoryService) GetLatestBotAnswerHistoryByBotUUID(ctx context.Context, botUUID string, limit int32) ([]sqlc_queries.GetLatestBotAnswerHistoryByBotUUIDRow, error) {\n\tparams := sqlc_queries.GetLatestBotAnswerHistoryByBotUUIDParams{\n\t\tBotUuid: botUUID,\n\t\tLimit: limit,\n\t}\n\thistory, err := s.q.GetLatestBotAnswerHistoryByBotUUID(ctx, params)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to get latest bot answer history by bot UUID\")\n\t}\n\treturn history, nil\n}\n" }, { "path": "api/chat_artifact.go", "content": "package main\n\nimport (\n\t\"regexp\"\n\t\"strings\"\n)\n\n// extractArtifacts detects and extracts artifacts from message content\nfunc extractArtifacts(content string) []Artifact {\n\tvar artifacts []Artifact\n\n\t// Pattern for HTML artifacts (check specific types first)\n\t// Example: ```html \n\thtmlArtifactRegex := regexp.MustCompile(`(?s)` + \"```\" + `html\\s*\\s*\\n(.*?)\\n` + \"```\")\n\thtmlMatches := htmlArtifactRegex.FindAllStringSubmatch(content, -1)\n\n\tfor _, match := range htmlMatches {\n\t\ttitle := strings.TrimSpace(match[1])\n\t\tartifactContent := strings.TrimSpace(match[2])\n\t\tartifact := Artifact{\n\t\t\tUUID: NewUUID(),\n\t\t\tType: \"html\",\n\t\t\tTitle: title,\n\t\t\tContent: artifactContent,\n\t\t\tLanguage: \"html\",\n\t\t}\n\t\tartifacts = append(artifacts, artifact)\n\t}\n\n\t// Pattern for SVG artifacts\n\t// Example: ```svg \n\tsvgArtifactRegex := regexp.MustCompile(`(?s)` + \"```\" + `svg\\s*\\s*\\n(.*?)\\n` + \"```\")\n\tsvgMatches := svgArtifactRegex.FindAllStringSubmatch(content, -1)\n\n\tfor _, match := range svgMatches {\n\t\ttitle := strings.TrimSpace(match[1])\n\t\tartifactContent := strings.TrimSpace(match[2])\n\n\t\tartifact := Artifact{\n\t\t\tUUID: NewUUID(),\n\t\t\tType: \"svg\",\n\t\t\tTitle: title,\n\t\t\tContent: artifactContent,\n\t\t\tLanguage: \"svg\",\n\t\t}\n\t\tartifacts = append(artifacts, artifact)\n\t}\n\n\t// Pattern for Mermaid diagrams\n\t// Example: ```mermaid \n\tmermaidArtifactRegex := regexp.MustCompile(`(?s)` + \"```\" + `mermaid\\s*\\s*\\n(.*?)\\n` + \"```\")\n\tmermaidMatches := mermaidArtifactRegex.FindAllStringSubmatch(content, -1)\n\n\tfor _, match := range mermaidMatches {\n\t\ttitle := strings.TrimSpace(match[1])\n\t\tartifactContent := strings.TrimSpace(match[2])\n\n\t\tartifact := Artifact{\n\t\t\tUUID: NewUUID(),\n\t\t\tType: \"mermaid\",\n\t\t\tTitle: title,\n\t\t\tContent: artifactContent,\n\t\t\tLanguage: \"mermaid\",\n\t\t}\n\t\tartifacts = append(artifacts, artifact)\n\t}\n\n\t// Pattern for JSON artifacts\n\t// Example: ```json \n\tjsonArtifactRegex := regexp.MustCompile(`(?s)` + \"```\" + `json\\s*\\s*\\n(.*?)\\n` + \"```\")\n\tjsonMatches := jsonArtifactRegex.FindAllStringSubmatch(content, -1)\n\n\tfor _, match := range jsonMatches {\n\t\ttitle := strings.TrimSpace(match[1])\n\t\tartifactContent := strings.TrimSpace(match[2])\n\n\t\tartifact := Artifact{\n\t\t\tUUID: NewUUID(),\n\t\t\tType: \"json\",\n\t\t\tTitle: title,\n\t\t\tContent: artifactContent,\n\t\t\tLanguage: \"json\",\n\t\t}\n\t\tartifacts = append(artifacts, artifact)\n\t}\n\n\t// Pattern for executable code artifacts\n\t// Example: ```javascript \n\texecutableArtifactRegex := regexp.MustCompile(`(?s)` + \"```\" + `(\\w+)?\\s*\\s*\\n(.*?)\\n` + \"```\")\n\texecutableMatches := executableArtifactRegex.FindAllStringSubmatch(content, -1)\n\n\tfor _, match := range executableMatches {\n\t\tlanguage := match[1]\n\t\ttitle := strings.TrimSpace(match[2])\n\t\tartifactContent := strings.TrimSpace(match[3])\n\n\t\t// Skip if already processed as HTML, SVG, Mermaid, or JSON\n\t\tif language == \"html\" || language == \"svg\" || language == \"mermaid\" || language == \"json\" {\n\t\t\tcontinue\n\t\t}\n\n\t\tif language == \"\" {\n\t\t\tlanguage = \"javascript\" // Default to JavaScript for executable code\n\t\t}\n\n\t\t// Only create executable artifacts for supported languages\n\t\tif isExecutableLanguage(language) {\n\t\t\tartifact := Artifact{\n\t\t\t\tUUID: NewUUID(),\n\t\t\t\tType: \"executable-code\",\n\t\t\t\tTitle: title,\n\t\t\t\tContent: artifactContent,\n\t\t\t\tLanguage: language,\n\t\t\t}\n\t\t\tartifacts = append(artifacts, artifact)\n\t\t}\n\t}\n\n\t// Pattern for general code artifacts (exclude html and svg which are handled above)\n\t// Example: ```javascript \n\tcodeArtifactRegex := regexp.MustCompile(`(?s)` + \"```\" + `(\\w+)?\\s*\\s*\\n(.*?)\\n` + \"```\")\n\tmatches := codeArtifactRegex.FindAllStringSubmatch(content, -1)\n\n\tfor _, match := range matches {\n\t\tlanguage := match[1]\n\t\ttitle := strings.TrimSpace(match[2])\n\t\tartifactContent := strings.TrimSpace(match[3])\n\n\t\t// Skip if already processed as HTML, SVG, Mermaid, JSON, or executable\n\t\tif language == \"html\" || language == \"svg\" || language == \"mermaid\" || language == \"json\" {\n\t\t\tcontinue\n\t\t}\n\n\t\tif language == \"\" {\n\t\t\tlanguage = \"text\"\n\t\t}\n\n\t\t// Check if this should be an executable artifact for supported languages\n\t\tartifactType := \"code\"\n\t\tif isExecutableLanguage(language) {\n\t\t\t// For supported languages, make them executable by default if they contain certain patterns\n\t\t\tif containsExecutablePatterns(artifactContent) {\n\t\t\t\tartifactType = \"executable-code\"\n\t\t\t}\n\t\t}\n\n\t\tartifact := Artifact{\n\t\t\tUUID: NewUUID(),\n\t\t\tType: artifactType,\n\t\t\tTitle: title,\n\t\t\tContent: artifactContent,\n\t\t\tLanguage: language,\n\t\t}\n\t\tartifacts = append(artifacts, artifact)\n\t}\n\n\treturn artifacts\n}\n\n// isExecutableLanguage checks if a language is supported for code execution\nfunc isExecutableLanguage(language string) bool {\n\texecutableLanguages := []string{\n\t\t\"javascript\", \"js\", \"typescript\", \"ts\",\n\t\t\"python\", \"py\",\n\t}\n\n\tlanguage = strings.ToLower(strings.TrimSpace(language))\n\tfor _, execLang := range executableLanguages {\n\t\tif language == execLang {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// containsExecutablePatterns checks if code contains patterns that suggest it should be executable\nfunc containsExecutablePatterns(content string) bool {\n\t// Patterns that suggest the code is meant to be executed\n\texecutablePatterns := []string{\n\n\t\t// JavaScript patterns\n\t\t\"console.log\",\n\t\t\"console.error\",\n\t\t\"console.warn\",\n\t\t\"function\",\n\t\t\"const \",\n\t\t\"let \",\n\t\t\"var \",\n\t\t\"=>\",\n\t\t\"if (\",\n\t\t\"for (\",\n\t\t\"while (\",\n\t\t\"return \",\n\n\t\t// Python patterns\n\t\t\"print(\",\n\t\t\"import \",\n\t\t\"from \",\n\t\t\"def \",\n\t\t\"if __name__\",\n\t\t\"class \",\n\t\t\"for \",\n\t\t\"while \",\n\t}\n\n\tcontentLower := strings.ToLower(content)\n\tfor _, pattern := range executablePatterns {\n\t\tif strings.Contains(contentLower, pattern) {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n" }, { "path": "api/chat_auth_user_handler.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"net/http\"\n\t\"os\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/gorilla/mux\"\n\tlog \"github.com/sirupsen/logrus\"\n\t\"github.com/swuecho/chat_backend/auth\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// Constants for token management\nconst (\n\tAccessTokenLifetime = 30 * time.Minute\n\tRefreshTokenLifetime = 7 * 24 * time.Hour // 7 days\n\tRefreshTokenName = \"refresh_token\"\n)\n\ntype AuthUserHandler struct {\n\tservice *AuthUserService\n}\n\n// isHTTPS checks if the request is using HTTPS or if we're in production\nfunc isHTTPS(r *http.Request) bool {\n\t// Check if request is HTTPS\n\tif r.TLS != nil {\n\t\treturn true\n\t}\n\n\t// Check common proxy headers\n\tif r.Header.Get(\"X-Forwarded-Proto\") == \"https\" {\n\t\treturn true\n\t}\n\n\tif r.Header.Get(\"X-Forwarded-Ssl\") == \"on\" {\n\t\treturn true\n\t}\n\n\t// Check if environment indicates production\n\tenv := os.Getenv(\"ENV\")\n\tif env == \"\" {\n\t\tenv = os.Getenv(\"ENVIRONMENT\")\n\t}\n\tif env == \"\" {\n\t\tenv = os.Getenv(\"NODE_ENV\")\n\t}\n\n\treturn env == \"production\" || env == \"prod\"\n}\n\n// createSecureRefreshCookie creates a secure httpOnly cookie for refresh tokens\nfunc createSecureRefreshCookie(name, value string, maxAge int, r *http.Request) *http.Cookie {\n\t// Determine the appropriate SameSite setting based on environment\n\tsameSite := http.SameSiteLaxMode // More permissive for development\n\tif isHTTPS(r) {\n\t\tsameSite = http.SameSiteStrictMode // Strict for HTTPS\n\t}\n\n\t// Determine domain based on environment\n\tvar domain string\n\thost := r.Host\n\tif host != \"\" && !strings.HasPrefix(host, \"localhost\") && !strings.HasPrefix(host, \"127.0.0.1\") {\n\t\t// For production, set domain without port\n\t\tif strings.Contains(host, \":\") {\n\t\t\tdomain = strings.Split(host, \":\")[0]\n\t\t} else {\n\t\t\tdomain = host\n\t\t}\n\t}\n\n\tcookie := &http.Cookie{\n\t\tName: name,\n\t\tValue: value,\n\t\tHttpOnly: true,\n\t\tSecure: isHTTPS(r),\n\t\tSameSite: sameSite,\n\t\tPath: \"/\",\n\t\tMaxAge: maxAge,\n\t}\n\n\t// Only set domain if it's not localhost\n\tif domain != \"\" && domain != \"localhost\" && domain != \"127.0.0.1\" {\n\t\tcookie.Domain = domain\n\t}\n\n\treturn cookie\n}\n\nfunc NewAuthUserHandler(sqlc_q *sqlc_queries.Queries) *AuthUserHandler {\n\tuserService := NewAuthUserService(sqlc_q)\n\treturn &AuthUserHandler{service: userService}\n}\n\nfunc (h *AuthUserHandler) Register(router *mux.Router) {\n\t// Authenticated user routes\n\trouter.HandleFunc(\"/users\", h.GetUserByID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/users/{id}\", h.UpdateSelf).Methods(http.MethodPut)\n\trouter.HandleFunc(\"/token_10years\", h.ForeverToken).Methods(http.MethodGet)\n}\n\nfunc (h *AuthUserHandler) RegisterPublicRoutes(router *mux.Router) {\n\t// Public routes (no authentication required)\n\trouter.HandleFunc(\"/signup\", h.SignUp).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/login\", h.Login).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/auth/refresh\", h.RefreshToken).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/logout\", h.Logout).Methods(http.MethodPost)\n}\n\nfunc (h *AuthUserHandler) CreateUser(w http.ResponseWriter, r *http.Request) {\n\tvar userParams sqlc_queries.CreateAuthUserParams\n\terr := json.NewDecoder(r.Body).Decode(&userParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tuser, err := h.service.CreateAuthUser(r.Context(), userParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to create user\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(user)\n}\n\nfunc (h *AuthUserHandler) GetUserByID(w http.ResponseWriter, r *http.Request) {\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tuser, err := h.service.GetAuthUserByID(r.Context(), userID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrResourceNotFound(\"user\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(user)\n}\n\nfunc (h *AuthUserHandler) UpdateSelf(w http.ResponseWriter, r *http.Request) {\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tvar userParams sqlc_queries.UpdateAuthUserParams\n\terr = json.NewDecoder(r.Body).Decode(&userParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tuserParams.ID = userID\n\tuser, err := h.service.q.UpdateAuthUser(r.Context(), userParams)\n\tif err != nil {\n\t\tlog.WithError(err).Error(\"Failed to update user\")\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to update user\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(user)\n}\n\nfunc (h *AuthUserHandler) UpdateUser(w http.ResponseWriter, r *http.Request) {\n\t// get user id from var\n\t// to int32\n\tvar userParams sqlc_queries.UpdateAuthUserByEmailParams\n\terr := json.NewDecoder(r.Body).Decode(&userParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tuser, err := h.service.q.UpdateAuthUserByEmail(r.Context(), userParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to update user\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(user)\n}\n\ntype LoginParams struct {\n\tEmail string `json:\"email\"`\n\tPassword string `json:\"password\"`\n}\n\nfunc (h *AuthUserHandler) SignUp(w http.ResponseWriter, r *http.Request) {\n\tvar params LoginParams\n\tif err := json.NewDecoder(r.Body).Decode(¶ms); err != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"error\": err.Error(),\n\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\"action\": \"signup_decode_error\",\n\t\t}).Warn(\"Failed to decode signup request\")\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Invalid request: unable to decode JSON body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tlog.WithFields(log.Fields{\n\t\t\"email\": params.Email,\n\t\t\"ip\": r.RemoteAddr,\n\t\t\"action\": \"signup_attempt\",\n\t}).Info(\"User signup attempt\")\n\n\thash, err := auth.GeneratePasswordHash(params.Password)\n\tif err != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"email\": params.Email,\n\t\t\t\"error\": err.Error(),\n\t\t}).Error(\"Failed to generate password hash\")\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to generate password hash\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tuserParams := sqlc_queries.CreateAuthUserParams{\n\t\tPassword: hash,\n\t\tEmail: params.Email,\n\t\tUsername: params.Email,\n\t}\n\n\tuser, err := h.service.CreateAuthUser(r.Context(), userParams)\n\tif err != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"email\": params.Email,\n\t\t\t\"error\": err.Error(),\n\t\t}).Error(\"Failed to create user\")\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to create user\"))\n\t\treturn\n\t}\n\n\t// Generate access token using constant\n\ttokenString, err := auth.GenerateToken(user.ID, user.Role(), jwtSecretAndAud.Secret, jwtSecretAndAud.Audience, AccessTokenLifetime, auth.TokenTypeAccess)\n\tif err != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"user_id\": user.ID,\n\t\t\t\"error\": err.Error(),\n\t\t}).Error(\"Failed to generate access token\")\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to generate token\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Generate refresh token using constant\n\trefreshToken, err := auth.GenerateToken(user.ID, user.Role(), jwtSecretAndAud.Secret, jwtSecretAndAud.Audience, RefreshTokenLifetime, auth.TokenTypeRefresh)\n\tif err != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"user_id\": user.ID,\n\t\t\t\"error\": err.Error(),\n\t\t}).Error(\"Failed to generate refresh token\")\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to generate refresh token\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Use helper function to create refresh token cookie\n\trefreshCookie := createSecureRefreshCookie(RefreshTokenName, refreshToken, int(RefreshTokenLifetime.Seconds()), r)\n\thttp.SetCookie(w, refreshCookie)\n\n\tlog.WithFields(log.Fields{\n\t\t\"user_id\": user.ID,\n\t\t\"email\": user.Email,\n\t\t\"action\": \"signup_success\",\n\t}).Info(\"User signup successful\")\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\texpiresIn := time.Now().Add(AccessTokenLifetime).Unix()\n\tjson.NewEncoder(w).Encode(TokenResult{AccessToken: tokenString, ExpiresIn: int(expiresIn)})\n\tw.WriteHeader(http.StatusOK)\n}\n\nfunc (h *AuthUserHandler) Login(w http.ResponseWriter, r *http.Request) {\n\tvar loginParams LoginParams\n\terr := json.NewDecoder(r.Body).Decode(&loginParams)\n\tif err != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"error\": err.Error(),\n\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\"action\": \"login_decode_error\",\n\t\t}).Warn(\"Failed to decode login request\")\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tlog.WithFields(log.Fields{\n\t\t\"email\": loginParams.Email,\n\t\t\"ip\": r.RemoteAddr,\n\t\t\"action\": \"login_attempt\",\n\t}).Info(\"User login attempt\")\n\n\tuser, err := h.service.Authenticate(r.Context(), loginParams.Email, loginParams.Password)\n\tif err != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"email\": loginParams.Email,\n\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\"error\": err.Error(),\n\t\t\t\"action\": \"login_failed\",\n\t\t}).Warn(\"User login failed\")\n\t\tRespondWithAPIError(w, ErrAuthInvalidEmailOrPassword.WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Generate access token using constant\n\taccessToken, err := auth.GenerateToken(user.ID, user.Role(), jwtSecretAndAud.Secret, jwtSecretAndAud.Audience, AccessTokenLifetime, auth.TokenTypeAccess)\n\tif err != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"user_id\": user.ID,\n\t\t\t\"error\": err.Error(),\n\t\t}).Error(\"Failed to generate access token\")\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to generate access token\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Generate refresh token using constant\n\trefreshToken, err := auth.GenerateToken(user.ID, user.Role(), jwtSecretAndAud.Secret, jwtSecretAndAud.Audience, RefreshTokenLifetime, auth.TokenTypeRefresh)\n\tif err != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"user_id\": user.ID,\n\t\t\t\"error\": err.Error(),\n\t\t}).Error(\"Failed to generate refresh token\")\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to generate refresh token\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Use helper function to create refresh token cookie\n\trefreshCookie := createSecureRefreshCookie(RefreshTokenName, refreshToken, int(RefreshTokenLifetime.Seconds()), r)\n\thttp.SetCookie(w, refreshCookie)\n\n\t// Debug: Log cookie details\n\tlog.WithFields(log.Fields{\n\t\t\"user_id\": user.ID,\n\t\t\"name\": refreshCookie.Name,\n\t\t\"domain\": refreshCookie.Domain,\n\t\t\"path\": refreshCookie.Path,\n\t\t\"secure\": refreshCookie.Secure,\n\t\t\"sameSite\": refreshCookie.SameSite,\n\t\t\"action\": \"login_cookie_set\",\n\t}).Info(\"Refresh token cookie set\")\n\n\tlog.WithFields(log.Fields{\n\t\t\"user_id\": user.ID,\n\t\t\"email\": user.Email,\n\t\t\"action\": \"login_success\",\n\t}).Info(\"User login successful\")\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\texpiresIn := time.Now().Add(AccessTokenLifetime).Unix()\n\tjson.NewEncoder(w).Encode(TokenResult{AccessToken: accessToken, ExpiresIn: int(expiresIn)})\n\tw.WriteHeader(http.StatusOK)\n}\n\nfunc (h *AuthUserHandler) ForeverToken(w http.ResponseWriter, r *http.Request) {\n\n\tlifetime := time.Duration(10*365*24) * time.Hour\n\tuserId, _ := getUserID(r.Context())\n\tuserRole := r.Context().Value(userContextKey).(string)\n\ttoken, err := auth.GenerateToken(userId, userRole, jwtSecretAndAud.Secret, jwtSecretAndAud.Audience, lifetime, auth.TokenTypeAccess)\n\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to generate token\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\texpiresIn := time.Now().Add(lifetime).Unix()\n\tjson.NewEncoder(w).Encode(TokenResult{AccessToken: token, ExpiresIn: int(expiresIn)})\n\tw.WriteHeader(http.StatusOK)\n\n}\n\nfunc (h *AuthUserHandler) RefreshToken(w http.ResponseWriter, r *http.Request) {\n\tlog.WithFields(log.Fields{\n\t\t\"ip\": r.RemoteAddr,\n\t\t\"action\": \"refresh_attempt\",\n\t}).Info(\"Token refresh attempt\")\n\n\t// Debug: Log all cookies to help diagnose the issue\n\tallCookies := r.Cookies()\n\tlog.WithFields(log.Fields{\n\t\t\"ip\": r.RemoteAddr,\n\t\t\"cookies\": len(allCookies),\n\t\t\"action\": \"refresh_debug_cookies\",\n\t}).Info(\"All cookies received\")\n\n\tfor _, cookie := range allCookies {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\"name\": cookie.Name,\n\t\t\t\"domain\": cookie.Domain,\n\t\t\t\"path\": cookie.Path,\n\t\t\t\"action\": \"refresh_debug_cookie\",\n\t\t}).Info(\"Cookie details\")\n\t}\n\n\t// Get refresh token from httpOnly cookie\n\trefreshCookie, err := r.Cookie(RefreshTokenName)\n\tif err != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\"error\": err.Error(),\n\t\t\t\"action\": \"refresh_missing_cookie\",\n\t\t}).Warn(\"Missing refresh token cookie\")\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"Missing refresh token\"))\n\t\treturn\n\t}\n\n\t// Validate refresh token\n\tresult := parseAndValidateJWT(refreshCookie.Value, auth.TokenTypeRefresh)\n\tif result.Error != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\"error\": result.Error.Detail,\n\t\t\t\"action\": \"refresh_invalid_token\",\n\t\t}).Warn(\"Invalid refresh token\")\n\t\tRespondWithAPIError(w, *result.Error)\n\t\treturn\n\t}\n\n\t// Convert UserID string back to int32\n\tuserIDInt, err := strconv.ParseInt(result.UserID, 10, 32)\n\tif err != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"user_id\": result.UserID,\n\t\t\t\"error\": err.Error(),\n\t\t\t\"action\": \"refresh_invalid_user_id\",\n\t\t}).Error(\"Invalid user ID in refresh token\")\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"Invalid user ID in token\"))\n\t\treturn\n\t}\n\n\t// Generate new access token using constant\n\taccessToken, err := auth.GenerateToken(int32(userIDInt), result.Role, jwtSecretAndAud.Secret, jwtSecretAndAud.Audience, AccessTokenLifetime, auth.TokenTypeAccess)\n\tif err != nil {\n\t\tlog.WithFields(log.Fields{\n\t\t\t\"user_id\": userIDInt,\n\t\t\t\"error\": err.Error(),\n\t\t}).Error(\"Failed to generate access token during refresh\")\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to generate access token\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tlog.WithFields(log.Fields{\n\t\t\"user_id\": userIDInt,\n\t\t\"action\": \"refresh_success\",\n\t}).Info(\"Token refresh successful\")\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\texpiresIn := time.Now().Add(AccessTokenLifetime).Unix()\n\tjson.NewEncoder(w).Encode(TokenResult{AccessToken: accessToken, ExpiresIn: int(expiresIn)})\n}\n\nfunc (h *AuthUserHandler) Logout(w http.ResponseWriter, r *http.Request) {\n\tlog.WithFields(log.Fields{\n\t\t\"ip\": r.RemoteAddr,\n\t\t\"action\": \"logout_attempt\",\n\t}).Info(\"User logout attempt\")\n\n\t// Clear refresh token cookie using the same domain logic as creation\n\trefreshCookie := createSecureRefreshCookie(RefreshTokenName, \"\", -1, r)\n\thttp.SetCookie(w, refreshCookie)\n\n\tlog.WithFields(log.Fields{\n\t\t\"ip\": r.RemoteAddr,\n\t\t\"action\": \"logout_success\",\n\t}).Info(\"User logout successful\")\n\n\tw.WriteHeader(http.StatusOK)\n}\n\ntype TokenRequest struct {\n\tToken string `json:\"token\"`\n}\n\ntype ResetPasswordRequest struct {\n\tEmail string `json:\"email\"`\n}\n\nfunc (h *AuthUserHandler) ResetPasswordHandler(w http.ResponseWriter, r *http.Request) {\n\tif r.Method != http.MethodPost {\n\t\tw.WriteHeader(http.StatusMethodNotAllowed)\n\t\treturn\n\t}\n\n\tvar req ResetPasswordRequest\n\terr := json.NewDecoder(r.Body).Decode(&req)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Retrieve user account from the database by email address\n\tuser, err := h.service.q.GetUserByEmail(context.Background(), req.Email)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrResourceNotFound(\"user\"))\n\t\treturn\n\t}\n\n\t// Generate temporary password\n\ttempPassword, err := auth.GenerateRandomPassword()\n\tif err != nil {\n\t\tlog.WithError(err).Error(\"Failed to generate temporary password\")\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to generate temporary password\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Hash temporary password\n\thashedPassword, err := auth.GeneratePasswordHash(tempPassword)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to hash password\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Update user account with new hashed password\n\terr = h.service.q.UpdateUserPassword(\n\t\tcontext.Background(),\n\t\tsqlc_queries.UpdateUserPasswordParams{\n\t\t\tEmail: req.Email,\n\t\t\tPassword: hashedPassword,\n\t\t})\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to update password\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Send email to the user with temporary password and instructions\n\terr = SendPasswordResetEmail(user.Email, tempPassword)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to send password reset email\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusOK)\n}\n\nfunc SendPasswordResetEmail(email, tempPassword string) error {\n\treturn nil\n}\n\ntype ChangePasswordRequest struct {\n\tEmail string `json:\"email\"`\n\tNewPassword string `json:\"new_password\"`\n}\n\nfunc (h *AuthUserHandler) ChangePasswordHandler(w http.ResponseWriter, r *http.Request) {\n\tif r.Method != http.MethodPost {\n\t\tw.WriteHeader(http.StatusMethodNotAllowed)\n\t\treturn\n\t}\n\n\tvar req ChangePasswordRequest\n\terr := json.NewDecoder(r.Body).Decode(&req)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Hash new password\n\thashedPassword, err := auth.GeneratePasswordHash(req.NewPassword)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to hash password\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Update password in the database\n\terr = h.service.q.UpdateUserPassword(context.Background(), sqlc_queries.UpdateUserPasswordParams{\n\t\tEmail: req.Email,\n\t\tPassword: string(hashedPassword),\n\t})\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to update password\"))\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusOK)\n}\n\ntype UserStat struct {\n\tEmail string `json:\"email\"`\n\tFirstName string `json:\"firstName\"`\n\tLastName string `json:\"lastName\"`\n\tTotalChatMessages int64 `json:\"totalChatMessages\"`\n\tTotalChatMessagesTokenCount int64 `json:\"totalChatMessagesTokenCount\"`\n\tTotalChatMessages3Days int64 `json:\"totalChatMessages3Days\"`\n\tTotalChatMessages3DaysTokenCount int64 `json:\"totalChatMessages3DaysTokenCount\"`\n\tAvgChatMessages3DaysTokenCount int64 `json:\"avgChatMessages3DaysTokenCount\"`\n\tRateLimit int32 `json:\"rateLimit\"`\n}\n\nfunc (h *AuthUserHandler) UserStatHandler(w http.ResponseWriter, r *http.Request) {\n\tvar pagination Pagination\n\terr := json.NewDecoder(r.Body).Decode(&pagination)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tuserStatsRows, total, err := h.service.GetUserStats(r.Context(), pagination, int32(appConfig.OPENAI.RATELIMIT))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get user stats\"))\n\t\treturn\n\t}\n\n\t// Create a new []interface{} slice with same length as userStatsRows\n\tdata := make([]interface{}, len(userStatsRows))\n\n\t// Copy the contents of userStatsRows into data\n\tfor i, v := range userStatsRows {\n\t\tdivider := v.TotalChatMessages3Days\n\t\tvar avg int64\n\t\tif divider > 0 {\n\t\t\tavg = v.TotalTokenCount3Days / v.TotalChatMessages3Days\n\t\t} else {\n\t\t\tavg = 0\n\t\t}\n\t\tdata[i] = UserStat{\n\t\t\tEmail: v.UserEmail,\n\t\t\tFirstName: v.FirstName,\n\t\t\tLastName: v.LastName,\n\t\t\tTotalChatMessages: v.TotalChatMessages,\n\t\t\tTotalChatMessages3Days: v.TotalChatMessages3Days,\n\t\t\tRateLimit: v.RateLimit,\n\t\t\tTotalChatMessagesTokenCount: v.TotalTokenCount,\n\t\t\tTotalChatMessages3DaysTokenCount: v.TotalTokenCount3Days,\n\t\t\tAvgChatMessages3DaysTokenCount: avg,\n\t\t}\n\t}\n\n\tjson.NewEncoder(w).Encode(Pagination{\n\t\tPage: pagination.Page,\n\t\tSize: pagination.Size,\n\t\tTotal: total,\n\t\tData: data,\n\t})\n}\n\ntype RateLimitRequest struct {\n\tEmail string `json:\"email\"`\n\tRateLimit int32 `json:\"rateLimit\"`\n}\n\nfunc (h *AuthUserHandler) UpdateRateLimit(w http.ResponseWriter, r *http.Request) {\n\tvar rateLimitRequest RateLimitRequest\n\terr := json.NewDecoder(r.Body).Decode(&rateLimitRequest)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\trate, err := h.service.q.UpdateAuthUserRateLimitByEmail(r.Context(),\n\t\tsqlc_queries.UpdateAuthUserRateLimitByEmailParams{\n\t\t\tEmail: rateLimitRequest.Email,\n\t\t\tRateLimit: rateLimitRequest.RateLimit,\n\t\t})\n\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to update rate limit\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(\n\t\tmap[string]int32{\n\t\t\t\"rate\": rate,\n\t\t})\n}\nfunc (h *AuthUserHandler) GetRateLimit(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\trate, err := h.service.q.GetRateLimit(ctx, userID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get rate limit\"))\n\t\treturn\n\t}\n\n\tjson.NewEncoder(w).Encode(map[string]int32{\n\t\t\"rate\": rate,\n\t})\n}\n" }, { "path": "api/chat_auth_user_service.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/rotisserie/eris\"\n\t\"github.com/swuecho/chat_backend/auth\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype AuthUserService struct {\n\tq *sqlc_queries.Queries\n}\n\n// NewAuthUserService creates a new AuthUserService.\nfunc NewAuthUserService(q *sqlc_queries.Queries) *AuthUserService {\n\treturn &AuthUserService{q: q}\n}\n\n// CreateAuthUser creates a new authentication user record.\nfunc (s *AuthUserService) CreateAuthUser(ctx context.Context, auth_user_params sqlc_queries.CreateAuthUserParams) (sqlc_queries.AuthUser, error) {\n\ttotalUserCount, err := s.q.GetTotalActiveUserCount(ctx)\n\tif err != nil {\n\t\treturn sqlc_queries.AuthUser{}, errors.New(\"failed to retrieve total user count\")\n\t}\n\tif totalUserCount == 0 {\n\t\tauth_user_params.IsSuperuser = true\n\t\tfmt.Println(\"First user is superuser.\")\n\t}\n\tauth_user, err := s.q.CreateAuthUser(ctx, auth_user_params)\n\tif err != nil {\n\t\treturn sqlc_queries.AuthUser{}, err\n\t}\n\treturn auth_user, nil\n}\n\n// GetAuthUserByID returns an authentication user record by ID.\nfunc (s *AuthUserService) GetAuthUserByID(ctx context.Context, id int32) (sqlc_queries.AuthUser, error) {\n\tauth_user, err := s.q.GetAuthUserByID(ctx, id)\n\tif err != nil {\n\t\treturn sqlc_queries.AuthUser{}, errors.New(\"failed to retrieve authentication user\")\n\t}\n\treturn auth_user, nil\n}\n\n// GetAllAuthUsers returns all authentication user records.\nfunc (s *AuthUserService) GetAllAuthUsers(ctx context.Context) ([]sqlc_queries.AuthUser, error) {\n\tauth_users, err := s.q.GetAllAuthUsers(ctx)\n\tif err != nil {\n\t\treturn nil, errors.New(\"failed to retrieve authentication users\")\n\t}\n\treturn auth_users, nil\n}\n\nfunc (s *AuthUserService) Authenticate(ctx context.Context, email, password string) (sqlc_queries.AuthUser, error) {\n\tuser, err := s.q.GetUserByEmail(ctx, email)\n\tif err != nil {\n\t\treturn sqlc_queries.AuthUser{}, err\n\t}\n\tif !auth.ValidatePassword(password, user.Password) {\n\t\treturn sqlc_queries.AuthUser{}, ErrAuthInvalidCredentials\n\t}\n\treturn user, nil\n}\n\nfunc (s *AuthUserService) Logout(tokenString string) (*http.Cookie, error) {\n\tuserID, err := auth.ValidateToken(tokenString, jwtSecretAndAud.Secret, auth.TokenTypeAccess)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\t// Implement a mechanism to track invalidated tokens for the given user ID\n\t// auth.AddInvalidToken(userID, \"insert-invalidated-token-here\")\n\tcookie := auth.GetExpireSecureCookie(strconv.Itoa(int(userID)), false)\n\treturn cookie, nil\n}\n\n// backend api\n// GetUserStat(page, page_size) -> {data: [{user_email, total_sessions, total_messages, total_sessions_3_days, total_messages_3_days, rate_limit}], total: 100}\n// GetTotalUserCount\n// GetUserStat(page, page_size) ->[{user_email, total_sessions, total_messages, total_sessions_3_days, total_messages_3_days, rate_limit}]\nfunc (s *AuthUserService) GetUserStats(ctx context.Context, p Pagination, defaultRateLimit int32) ([]sqlc_queries.GetUserStatsRow, int64, error) {\n\tauth_users_stat, err := s.q.GetUserStats(ctx,\n\t\tsqlc_queries.GetUserStatsParams{\n\t\t\tOffset: p.Offset(),\n\t\t\tLimit: p.Size,\n\t\t\tDefaultRateLimit: defaultRateLimit,\n\t\t})\n\tif err != nil {\n\t\treturn nil, 0, eris.Wrap(err, \"failed to retrieve user stats \")\n\t}\n\ttotal, err := s.q.GetTotalActiveUserCount(ctx)\n\tif err != nil {\n\t\treturn nil, 0, errors.New(\"failed to retrieve total active user count\")\n\t}\n\treturn auth_users_stat, total, nil\n}\n\n// UpdateRateLimit(user_email, rate_limit) -> { rate_limit: 100 }\nfunc (s *AuthUserService) UpdateRateLimit(ctx context.Context, user_email string, rate_limit int32) (int32, error) {\n\tauth_user_params := sqlc_queries.UpdateAuthUserRateLimitByEmailParams{\n\t\tEmail: user_email,\n\t\tRateLimit: rate_limit,\n\t}\n\trate, err := s.q.UpdateAuthUserRateLimitByEmail(ctx, auth_user_params)\n\tif err != nil {\n\t\treturn -1, errors.New(\"failed to update authentication user\")\n\t}\n\treturn rate, nil\n}\n\n// get ratelimit for user_id\nfunc (s *AuthUserService) GetRateLimit(ctx context.Context, user_id int32) (int32, error) {\n\trate, err := s.q.GetRateLimit(ctx, user_id)\n\tif err != nil {\n\t\treturn -1, errors.New(\"failed to get rate limit\")\n\n\t}\n\treturn rate, nil\n}\n\n// UserAnalysisData represents the complete user analysis response\ntype UserAnalysisData struct {\n\tUserInfo UserAnalysisInfo `json:\"userInfo\"`\n\tModelUsage []ModelUsageInfo `json:\"modelUsage\"`\n\tRecentActivity []ActivityInfo `json:\"recentActivity\"`\n}\n\ntype UserAnalysisInfo struct {\n\tEmail string `json:\"email\"`\n\tTotalMessages int64 `json:\"totalMessages\"`\n\tTotalTokens int64 `json:\"totalTokens\"`\n\tTotalSessions int64 `json:\"totalSessions\"`\n\tMessages3Days int64 `json:\"messages3Days\"`\n\tTokens3Days int64 `json:\"tokens3Days\"`\n\tRateLimit int32 `json:\"rateLimit\"`\n}\n\ntype ModelUsageInfo struct {\n\tModel string `json:\"model\"`\n\tMessageCount int64 `json:\"messageCount\"`\n\tTokenCount int64 `json:\"tokenCount\"`\n\tPercentage float64 `json:\"percentage\"`\n\tLastUsed time.Time `json:\"lastUsed\"`\n}\n\ntype ActivityInfo struct {\n\tDate time.Time `json:\"date\"`\n\tMessages int64 `json:\"messages\"`\n\tTokens int64 `json:\"tokens\"`\n\tSessions int64 `json:\"sessions\"`\n}\n\ntype SessionHistoryInfo struct {\n\tSessionID string `json:\"sessionId\"`\n\tModel string `json:\"model\"`\n\tMessageCount int64 `json:\"messageCount\"`\n\tTokenCount int64 `json:\"tokenCount\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n}\n\n// GetUserAnalysis retrieves comprehensive user analysis data\nfunc (s *AuthUserService) GetUserAnalysis(ctx context.Context, email string, defaultRateLimit int32) (*UserAnalysisData, error) {\n\t// Get basic user info\n\tuserInfo, err := s.q.GetUserAnalysisByEmail(ctx, sqlc_queries.GetUserAnalysisByEmailParams{\n\t\tEmail: email,\n\t\tDefaultRateLimit: defaultRateLimit,\n\t})\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to get user analysis\")\n\t}\n\n\t// Get model usage\n\tmodelUsageRows, err := s.q.GetUserModelUsageByEmail(ctx, email)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to get user model usage\")\n\t}\n\n\t// Calculate total tokens for percentage calculation\n\tvar totalTokens int64\n\tfor _, row := range modelUsageRows {\n\t\tif row.TokenCount != nil {\n\t\t\tif tc, ok := row.TokenCount.(int64); ok {\n\t\t\t\ttotalTokens += tc\n\t\t\t}\n\t\t}\n\t}\n\n\tmodelUsage := make([]ModelUsageInfo, len(modelUsageRows))\n\tfor i, row := range modelUsageRows {\n\t\t// Convert interface{} to int64 safely\n\t\ttokenCount := int64(0)\n\t\tif row.TokenCount != nil {\n\t\t\tif tc, ok := row.TokenCount.(int64); ok {\n\t\t\t\ttokenCount = tc\n\t\t\t}\n\t\t}\n\n\t\tpercentage := float64(0)\n\t\tif totalTokens > 0 {\n\t\t\tpercentage = float64(tokenCount) / float64(totalTokens) * 100\n\t\t}\n\t\tmodelUsage[i] = ModelUsageInfo{\n\t\t\tModel: row.Model,\n\t\t\tMessageCount: row.MessageCount,\n\t\t\tTokenCount: tokenCount,\n\t\t\tPercentage: percentage,\n\t\t\tLastUsed: row.LastUsed,\n\t\t}\n\t}\n\n\t// Get recent activity\n\tactivityRows, err := s.q.GetUserRecentActivityByEmail(ctx, email)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to get user recent activity\")\n\t}\n\n\trecentActivity := make([]ActivityInfo, len(activityRows))\n\tfor i, row := range activityRows {\n\t\t// Convert interface{} to int64 safely\n\t\ttokens := int64(0)\n\t\tif row.Tokens != nil {\n\t\t\tif t, ok := row.Tokens.(int64); ok {\n\t\t\t\ttokens = t\n\t\t\t}\n\t\t}\n\n\t\trecentActivity[i] = ActivityInfo{\n\t\t\tDate: row.ActivityDate,\n\t\t\tMessages: row.Messages,\n\t\t\tTokens: tokens,\n\t\t\tSessions: row.Sessions,\n\t\t}\n\t}\n\n\tanalysisData := &UserAnalysisData{\n\t\tUserInfo: UserAnalysisInfo{\n\t\t\tEmail: userInfo.UserEmail,\n\t\t\tTotalMessages: userInfo.TotalMessages,\n\t\t\tTotalTokens: userInfo.TotalTokens,\n\t\t\tTotalSessions: userInfo.TotalSessions,\n\t\t\tMessages3Days: userInfo.Messages3Days,\n\t\t\tTokens3Days: userInfo.Tokens3Days,\n\t\t\tRateLimit: userInfo.RateLimit,\n\t\t},\n\t\tModelUsage: modelUsage,\n\t\tRecentActivity: recentActivity,\n\t}\n\n\treturn analysisData, nil\n}\n\n// GetUserSessionHistory retrieves paginated session history for a user\nfunc (s *AuthUserService) GetUserSessionHistory(ctx context.Context, email string, page, pageSize int32) ([]SessionHistoryInfo, int64, error) {\n\toffset := (page - 1) * pageSize\n\n\t// Get session history with pagination\n\tsessionRows, err := s.q.GetUserSessionHistoryByEmail(ctx, sqlc_queries.GetUserSessionHistoryByEmailParams{\n\t\tEmail: email,\n\t\tLimit: pageSize,\n\t\tOffset: offset,\n\t})\n\tif err != nil {\n\t\treturn nil, 0, eris.Wrap(err, \"failed to get user session history\")\n\t}\n\n\t// Get total count\n\ttotalCount, err := s.q.GetUserSessionHistoryCountByEmail(ctx, email)\n\tif err != nil {\n\t\treturn nil, 0, eris.Wrap(err, \"failed to get user session history count\")\n\t}\n\n\tsessionHistory := make([]SessionHistoryInfo, len(sessionRows))\n\tfor i, row := range sessionRows {\n\t\t// Convert interface{} to int64 safely\n\t\tmessageCount := int64(0)\n\t\tif row.MessageCount != nil {\n\t\t\tif mc, ok := row.MessageCount.(int64); ok {\n\t\t\t\tmessageCount = mc\n\t\t\t}\n\t\t}\n\n\t\ttokenCount := int64(0)\n\t\tif row.TokenCount != nil {\n\t\t\tif tc, ok := row.TokenCount.(int64); ok {\n\t\t\t\ttokenCount = tc\n\t\t\t}\n\t\t}\n\n\t\tsessionHistory[i] = SessionHistoryInfo{\n\t\t\tSessionID: row.SessionID,\n\t\t\tModel: row.Model,\n\t\t\tMessageCount: messageCount,\n\t\t\tTokenCount: tokenCount,\n\t\t\tCreatedAt: row.CreatedAt,\n\t\t\tUpdatedAt: row.UpdatedAt,\n\t\t}\n\t}\n\n\treturn sessionHistory, totalCount, nil\n}\n" }, { "path": "api/chat_comment_handler.go", "content": "package main\n\nimport (\n\t\"encoding/json\"\n\t\"net/http\"\n\n\t\"github.com/google/uuid\"\n\t\"github.com/gorilla/mux\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype ChatCommentHandler struct {\n\tservice *ChatCommentService\n}\n\nfunc NewChatCommentHandler(sqlc_q *sqlc_queries.Queries) *ChatCommentHandler {\n\tchatCommentService := NewChatCommentService(sqlc_q)\n\treturn &ChatCommentHandler{\n\t\tservice: chatCommentService,\n\t}\n}\n\nfunc (h *ChatCommentHandler) Register(router *mux.Router) {\n\trouter.HandleFunc(\"/uuid/chat_sessions/{sessionUUID}/chat_messages/{messageUUID}/comments\", h.CreateChatComment).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/uuid/chat_sessions/{sessionUUID}/comments\", h.GetCommentsBySessionUUID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/uuid/chat_messages/{messageUUID}/comments\", h.GetCommentsByMessageUUID).Methods(http.MethodGet)\n}\n\nfunc (h *ChatCommentHandler) CreateChatComment(w http.ResponseWriter, r *http.Request) {\n\tvars := mux.Vars(r)\n\tsessionUUID := vars[\"sessionUUID\"]\n\tmessageUUID := vars[\"messageUUID\"]\n\n\tvar req struct {\n\t\tContent string `json:\"content\"`\n\t}\n\tif err := json.NewDecoder(r.Body).Decode(&req); err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"unauthorized\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tcomment, err := h.service.CreateChatComment(r.Context(), sqlc_queries.CreateChatCommentParams{\n\t\tUuid: uuid.New().String(),\n\t\tChatSessionUuid: sessionUUID,\n\t\tChatMessageUuid: messageUUID,\n\t\tContent: req.Content,\n\t\tCreatedBy: userID,\n\t})\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to create chat comment\"))\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusCreated)\n\tjson.NewEncoder(w).Encode(comment)\n}\n\nfunc (h *ChatCommentHandler) GetCommentsBySessionUUID(w http.ResponseWriter, r *http.Request) {\n\tsessionUUID := mux.Vars(r)[\"sessionUUID\"]\n\n\tcomments, err := h.service.GetCommentsBySessionUUID(r.Context(), sessionUUID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get comments by session\"))\n\t\treturn\n\t}\n\n\tjson.NewEncoder(w).Encode(comments)\n}\n\nfunc (h *ChatCommentHandler) GetCommentsByMessageUUID(w http.ResponseWriter, r *http.Request) {\n\tmessageUUID := mux.Vars(r)[\"messageUUID\"]\n\n\tcomments, err := h.service.GetCommentsByMessageUUID(r.Context(), messageUUID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get comments by message\"))\n\t\treturn\n\t}\n\n\tjson.NewEncoder(w).Encode(comments)\n}\n" }, { "path": "api/chat_comment_service.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/rotisserie/eris\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype ChatCommentService struct {\n\tq *sqlc_queries.Queries\n}\n\nfunc NewChatCommentService(q *sqlc_queries.Queries) *ChatCommentService {\n\treturn &ChatCommentService{q: q}\n}\n\n// CreateChatComment creates a new chat comment\nfunc (s *ChatCommentService) CreateChatComment(ctx context.Context, params sqlc_queries.CreateChatCommentParams) (sqlc_queries.ChatComment, error) {\n\tcomment, err := s.q.CreateChatComment(ctx, params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatComment{}, eris.Wrap(err, \"failed to create comment\")\n\t}\n\treturn comment, nil\n}\n\n// GetCommentsBySessionUUID returns comments for a session with author info\nfunc (s *ChatCommentService) GetCommentsBySessionUUID(ctx context.Context, sessionUUID string) ([]sqlc_queries.GetCommentsBySessionUUIDRow, error) {\n\tcomments, err := s.q.GetCommentsBySessionUUID(ctx, sessionUUID)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to get comments by session UUID\")\n\t}\n\treturn comments, nil\n}\n\n// GetCommentsByMessageUUID returns comments for a message with author info\nfunc (s *ChatCommentService) GetCommentsByMessageUUID(ctx context.Context, messageUUID string) ([]sqlc_queries.GetCommentsByMessageUUIDRow, error) {\n\tcomments, err := s.q.GetCommentsByMessageUUID(ctx, messageUUID)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to get comments by message UUID\")\n\t}\n\treturn comments, nil\n}\n\n// CommentWithAuthor represents a comment with author information\ntype CommentWithAuthor struct {\n\tUUID string `json:\"uuid\"`\n\tContent string `json:\"content\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tAuthorUsername string `json:\"authorUsername\"`\n\tAuthorEmail string `json:\"authorEmail\"`\n}\n\n// GetCommentsBySession returns comments for a session with author info\nfunc (s *ChatCommentService) GetCommentsBySession(ctx context.Context, sessionUUID string) ([]CommentWithAuthor, error) {\n\tcomments, err := s.q.GetCommentsBySessionUUID(ctx, sessionUUID)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to get comments by session\")\n\t}\n\n\tresult := make([]CommentWithAuthor, len(comments))\n\tfor i, c := range comments {\n\t\tresult[i] = CommentWithAuthor{\n\t\t\tUUID: c.Uuid,\n\t\t\tContent: c.Content,\n\t\t\tCreatedAt: c.CreatedAt,\n\t\t\tAuthorUsername: c.AuthorUsername,\n\t\t\tAuthorEmail: c.AuthorEmail,\n\t\t}\n\t}\n\treturn result, nil\n}\n\n// GetCommentsByMessage returns comments for a message with author info\nfunc (s *ChatCommentService) GetCommentsByMessage(ctx context.Context, messageUUID string) ([]CommentWithAuthor, error) {\n\tcomments, err := s.q.GetCommentsByMessageUUID(ctx, messageUUID)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to get comments by message\")\n\t}\n\n\tresult := make([]CommentWithAuthor, len(comments))\n\tfor i, c := range comments {\n\t\tresult[i] = CommentWithAuthor{\n\t\t\tUUID: c.Uuid,\n\t\t\tContent: c.Content,\n\t\t\tCreatedAt: c.CreatedAt,\n\t\t\tAuthorUsername: c.AuthorUsername,\n\t\t\tAuthorEmail: c.AuthorEmail,\n\t\t}\n\t}\n\treturn result, nil\n}\n" }, { "path": "api/chat_main_handler.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\tlog \"github.com/sirupsen/logrus\"\n\t\"net/http\"\n\t\"strings\"\n\t\"time\"\n\n\tmapset \"github.com/deckarep/golang-set/v2\"\n\topenai \"github.com/sashabaranov/go-openai\"\n\n\t\"github.com/gorilla/mux\"\n\n\t\"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype ChatHandler struct {\n\tservice *ChatService\n\tchatfileService *ChatFileService\n\trequestCtx context.Context // Store the request context for streaming\n}\n\nconst sessionTitleGenerationTimeout = 30 * time.Second\n\nfunc NewChatHandler(sqlc_q *sqlc_queries.Queries) *ChatHandler {\n\t// create a new ChatService instance\n\tchatService := NewChatService(sqlc_q)\n\tChatFileService := NewChatFileService(sqlc_q)\n\treturn &ChatHandler{\n\t\tservice: chatService,\n\t\tchatfileService: ChatFileService,\n\t\trequestCtx: context.Background(),\n\t}\n}\n\nfunc (h *ChatHandler) Register(router *mux.Router) {\n\trouter.HandleFunc(\"/chat_stream\", h.ChatCompletionHandler).Methods(http.MethodPost)\n\t// for bot\n\t// given a chat_uuid, a user message, return the answer\n\t//\n\trouter.HandleFunc(\"/chatbot\", h.ChatBotCompletionHandler).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/chat_instructions\", h.GetChatInstructions).Methods(http.MethodGet)\n}\n\ntype ChatRequest struct {\n\tPrompt string\n\tSessionUuid string\n\tChatUuid string\n\tRegenerate bool\n\tStream bool `json:\"stream,omitempty\"`\n}\n\ntype ChatCompletionResponse struct {\n\tID string `json:\"id\"`\n\tObject string `json:\"object\"`\n\tCreated int `json:\"created\"`\n\tModel string `json:\"model\"`\n\tUsage struct {\n\t\tPromptTokens int `json:\"prompt_tokens\"`\n\t\tCompletionTokens int `json:\"completion_tokens\"`\n\t\tTotalTokens int `json:\"total_tokens\"`\n\t} `json:\"usage\"`\n\tChoices []Choice `json:\"choices\"`\n}\n\ntype Choice struct {\n\tMessage openai.ChatCompletionMessage `json:\"message\"`\n\tFinishReason any `json:\"finish_reason\"`\n\tIndex int `json:\"index\"`\n}\n\ntype OpenaiChatRequest struct {\n\tModel string `json:\"model\"`\n\tMessages []openai.ChatCompletionMessage `json:\"messages\"`\n}\n\ntype BotRequest struct {\n\tMessage string `json:\"message\"`\n\tSnapshotUuid string `json:\"snapshot_uuid\"`\n\tStream bool `json:\"stream\"`\n}\n\ntype ChatInstructionResponse struct {\n\tArtifactInstruction string `json:\"artifactInstruction\"`\n}\n\nfunc (h *ChatHandler) GetChatInstructions(w http.ResponseWriter, r *http.Request) {\n\tartifactInstruction, err := loadArtifactInstruction()\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to load artifact instruction: %v\", err)\n\t\tartifactInstruction = \"\"\n\t}\n\n\tjson.NewEncoder(w).Encode(ChatInstructionResponse{\n\t\tArtifactInstruction: artifactInstruction,\n\t})\n}\n\n// ChatCompletionHandler is an HTTP handler that sends the stream to the client as Server-Sent Events (SSE)\nfunc (h *ChatHandler) ChatBotCompletionHandler(w http.ResponseWriter, r *http.Request) {\n\tvar req BotRequest\n\tif err := json.NewDecoder(r.Body).Decode(&req); err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tsnapshotUuid := req.SnapshotUuid\n\tnewQuestion := req.Message\n\n\tlog.Printf(\"snapshotUuid: %s\", snapshotUuid)\n\tlog.Printf(\"newQuestion: %s\", newQuestion)\n\n\tctx := r.Context()\n\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tlog.Printf(\"Error getting user ID: %v\", err)\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tfmt.Printf(\"userID: %d\", userID)\n\n\tchatSnapshot, err := h.service.q.ChatSnapshotByUserIdAndUuid(ctx, sqlc_queries.ChatSnapshotByUserIdAndUuidParams{\n\t\tUserID: userID,\n\t\tUuid: snapshotUuid,\n\t})\n\tif err != nil {\n\t\tapiErr := ErrResourceNotFound(\"Chat snapshot\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tfmt.Printf(\"chatSnapshot: %+v\", chatSnapshot)\n\n\tvar session sqlc_queries.ChatSession\n\terr = json.Unmarshal(chatSnapshot.Session, &session)\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to deserialize chat session\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tvar simpleChatMessages []SimpleChatMessage\n\terr = json.Unmarshal(chatSnapshot.Conversation, &simpleChatMessages)\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to deserialize conversation\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tgenBotAnswer(h, w, session, simpleChatMessages, snapshotUuid, newQuestion, userID, req.Stream)\n\n}\n\n// ChatCompletionHandler is an HTTP handler that sends the stream to the client as Server-Sent Events (SSE)\nfunc (h *ChatHandler) ChatCompletionHandler(w http.ResponseWriter, r *http.Request) {\n\tvar req ChatRequest\n\tif err := json.NewDecoder(r.Body).Decode(&req); err != nil {\n\t\tlog.Printf(\"Error decoding request: %v\", err)\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid request format\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tchatSessionUuid := req.SessionUuid\n\tchatUuid := req.ChatUuid\n\tnewQuestion := req.Prompt\n\n\tlog.Printf(\"chatSessionUuid: %s\", chatSessionUuid)\n\tlog.Printf(\"chatUuid: %s\", chatUuid)\n\tlog.Printf(\"newQuestion: %s\", newQuestion)\n\n\tctx := r.Context()\n\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tlog.Printf(\"Error getting user ID: %v\", err)\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tif req.Regenerate {\n\t\tregenerateAnswer(h, w, ctx, chatSessionUuid, chatUuid, req.Stream)\n\t} else {\n\t\tgenAnswer(h, w, ctx, chatSessionUuid, chatUuid, newQuestion, userID, req.Stream)\n\t}\n\n}\n\n// validateChatSession validates the chat session and returns the session and model info.\n// It performs comprehensive validation including:\n// - Session existence check\n// - Model availability verification\n// - Base URL extraction\n// - UUID validation\n// Returns: session, model, baseURL, success\nfunc (h *ChatHandler) validateChatSession(ctx context.Context, w http.ResponseWriter, chatSessionUuid string) (*sqlc_queries.ChatSession, *sqlc_queries.ChatModel, string, bool) {\n\tchatSession, err := h.service.q.GetChatSessionByUUID(ctx, chatSessionUuid)\n\tif err != nil {\n\t\tlog.Printf(\"Invalid session UUID: %s, error: %v\", chatSessionUuid, err)\n\t\tRespondWithAPIError(w, ErrResourceNotFound(\"chat session\").WithMessage(chatSessionUuid))\n\t\treturn nil, nil, \"\", false\n\t}\n\n\tchatModel, err := h.service.q.ChatModelByName(ctx, chatSession.Model)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrResourceNotFound(\"chat model: \"+chatSession.Model))\n\t\treturn nil, nil, \"\", false\n\t}\n\n\tbaseURL, _ := getModelBaseUrl(chatModel.Url)\n\n\tif chatSession.Uuid == \"\" {\n\t\tlog.Printf(\"Empty session UUID for chat: %s\", chatSessionUuid)\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Invalid session UUID\"))\n\t\treturn nil, nil, \"\", false\n\t}\n\n\treturn &chatSession, &chatModel, baseURL, true\n}\n\n// handlePromptCreation handles creating new prompt or adding user message to existing conversation.\n// This function manages the logic for:\n// - Detecting existing prompts in the session\n// - Creating new prompts for fresh conversations\n// - Adding user messages to ongoing conversations\n// - Handling empty questions for regeneration scenarios\nfunc (h *ChatHandler) handlePromptCreation(ctx context.Context, w http.ResponseWriter, chatSession *sqlc_queries.ChatSession, chatUuid, newQuestion string, userID int32, baseURL string) bool {\n\texistingPrompt := true\n\tprompt, err := h.service.q.GetOneChatPromptBySessionUUID(ctx, chatSession.Uuid)\n\tif err != nil {\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\tlog.Printf(\"No existing prompt found for session: %s\", chatSession.Uuid)\n\t\t\texistingPrompt = false\n\t\t} else {\n\t\t\tlog.Printf(\"Error checking prompt for session %s: %v\", chatSession.Uuid, err)\n\t\t\tRespondWithAPIError(w, createAPIError(ErrInternalUnexpected, \"Failed to get prompt\", err.Error()))\n\t\t\treturn false\n\t\t}\n\t} else {\n\t\tlog.Printf(\"Found existing prompt ID %d for session %s\", prompt.ID, chatSession.Uuid)\n\t}\n\n\tif existingPrompt {\n\t\tif newQuestion != \"\" {\n\t\t\t_, err := h.service.CreateChatMessageSimple(ctx, chatSession.Uuid, chatUuid, \"user\", newQuestion, \"\", chatSession.Model, userID, baseURL, chatSession.SummarizeMode)\n\t\t\tif err != nil {\n\t\t\t\tRespondWithAPIError(w, createAPIError(ErrInternalUnexpected, \"Failed to create message\", err.Error()))\n\t\t\t\treturn false\n\t\t\t}\n\t\t} else {\n\t\t\tlog.Println(\"no new question, regenerate answer\")\n\t\t}\n\t} else {\n\t\tchatPrompt, err := h.service.CreateChatPromptSimple(ctx, chatSession.Uuid, DefaultSystemPromptText, userID)\n\t\tif err != nil {\n\t\t\tRespondWithAPIError(w, createAPIError(ErrInternalUnexpected, \"Failed to create prompt\", err.Error()))\n\t\t\treturn false\n\t\t}\n\t\tlog.Printf(\"%+v\\n\", chatPrompt)\n\n\t\tif newQuestion != \"\" {\n\t\t\t_, err := h.service.CreateChatMessageSimple(ctx, chatSession.Uuid, chatUuid, \"user\", newQuestion, \"\", chatSession.Model, userID, baseURL, chatSession.SummarizeMode)\n\t\t\tif err != nil {\n\t\t\t\tRespondWithAPIError(w, createAPIError(ErrInternalUnexpected, \"Failed to create message\", err.Error()))\n\t\t\t\treturn false\n\t\t\t}\n\t\t}\n\n\t\t// Update session title with first 10 words of the first user message.\n\t\tif newQuestion != \"\" {\n\t\t\tsessionTitle := firstNWords(newQuestion, 10)\n\t\t\tif sessionTitle != \"\" {\n\t\t\t\tupdateParams := sqlc_queries.UpdateChatSessionTopicByUUIDParams{\n\t\t\t\t\tUuid: chatSession.Uuid,\n\t\t\t\t\tUserID: userID,\n\t\t\t\t\tTopic: sessionTitle,\n\t\t\t\t}\n\t\t\t\t_, err := h.service.q.UpdateChatSessionTopicByUUID(ctx, updateParams)\n\t\t\t\tif err != nil {\n\t\t\t\t\tlog.Printf(\"Warning: Failed to update session title for session %s: %v\", chatSession.Uuid, err)\n\t\t\t\t} else {\n\t\t\t\t\tlog.Printf(\"Updated session %s title to: %s\", chatSession.Uuid, sessionTitle)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\treturn true\n}\n\n// generateAndSaveAnswer generates the LLM response and saves it to the database\nfunc (h *ChatHandler) generateAndSaveAnswer(ctx context.Context, w http.ResponseWriter, chatSession *sqlc_queries.ChatSession, chatUuid string, userID int32, baseURL string, streamOutput bool) bool {\n\tmsgs, err := h.service.getAskMessages(*chatSession, chatUuid, false)\n\tif err != nil {\n\t\tlog.Printf(\"Error collecting messages for session %s: %v\", chatSession.Uuid, err)\n\t\tRespondWithAPIError(w, createAPIError(ErrInternalUnexpected, \"Failed to collect messages\", err.Error()))\n\t\treturn false\n\t}\n\tlog.Printf(\"Collected messages for processing - SessionUUID: %s, MessageCount: %d, Model: %s\", chatSession.Uuid, len(msgs), chatSession.Model)\n\n\t// Store the request context so models can access it\n\th.requestCtx = ctx\n\tmodel := h.chooseChatModel(*chatSession, msgs)\n\tLLMAnswer, err := model.Stream(w, *chatSession, msgs, chatUuid, false, streamOutput)\n\tif err != nil {\n\t\tlog.Printf(\"Error generating answer: %v\", err)\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to generate answer\"))\n\t\treturn false\n\t}\n\tif LLMAnswer == nil {\n\t\tlog.Printf(\"Error generating answer: LLMAnswer is nil\")\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"LLMAnswer is nil\"))\n\t\treturn false\n\t}\n\n\tif !isTest(msgs) {\n\t\tlog.Printf(\"LLMAnswer: %+v\", LLMAnswer)\n\t\th.service.logChat(*chatSession, msgs, LLMAnswer.ReasoningContent+LLMAnswer.Answer)\n\t}\n\n\tchatMessage, err := h.service.CreateChatMessageWithSuggestedQuestions(ctx, chatSession.Uuid, LLMAnswer.AnswerId, \"assistant\", LLMAnswer.Answer, LLMAnswer.ReasoningContent, chatSession.Model, userID, baseURL, chatSession.SummarizeMode, chatSession.ExploreMode, msgs)\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(ErrInternalUnexpected, \"Failed to create message\", err.Error()))\n\t\treturn false\n\t}\n\n\t// Send suggested questions as a separate streaming event if streaming is enabled and exploreMode is on\n\tif streamOutput && chatSession.ExploreMode && chatMessage.SuggestedQuestions != nil {\n\t\th.sendSuggestedQuestionsStream(w, LLMAnswer.AnswerId, chatMessage.SuggestedQuestions)\n\t}\n\n\t// Generate a better title using LLM for the first exchange (async, non-blocking).\n\t// Detach from the request context so the follow-up DB/model call can finish\n\t// after the streaming response closes.\n\tgo h.generateSessionTitle(chatSession, userID)\n\n\treturn true\n}\n\n// generateSessionTitle regenerates the session title from the latest conversation state.\nfunc (h *ChatHandler) generateSessionTitle(chatSession *sqlc_queries.ChatSession, userID int32) {\n\tctx, cancel := context.WithTimeout(context.Background(), sessionTitleGenerationTimeout)\n\tdefer cancel()\n\n\t// Regenerate from the latest conversation after each assistant response.\n\tmessages, err := h.service.q.GetChatMessagesBySessionUUID(ctx, sqlc_queries.GetChatMessagesBySessionUUIDParams{\n\t\tUuid: chatSession.Uuid,\n\t\tOffset: 0,\n\t\tLimit: 100,\n\t})\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to get messages for title generation: %v\", err)\n\t\treturn\n\t}\n\n\tvar chatText string\n\tfor _, msg := range messages {\n\t\tif msg.Role == \"user\" {\n\t\t\tchatText += \"user: \" + msg.Content + \"\\n\"\n\t\t} else if msg.Role == \"assistant\" {\n\t\t\tchatText += \"assistant: \" + msg.Content + \"\\n\"\n\t\t}\n\t}\n\n\tif strings.TrimSpace(chatText) == \"\" {\n\t\treturn\n\t}\n\n\t// Use the same approach as chat_snapshot_service.go - check if gemini-2.0-flash is available\n\tmodel := \"gemini-2.0-flash\"\n\t_, err = h.service.q.ChatModelByName(ctx, model)\n\tif err != nil {\n\t\t// Model not available, skip title generation\n\t\treturn\n\t}\n\n\t// Generate title using Gemini\n\tgenTitle, err := GenerateChatTitle(ctx, model, chatText)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to generate session title: %v\", err)\n\t\treturn\n\t}\n\n\tif genTitle == \"\" {\n\t\treturn\n\t}\n\n\t// Update the session title\n\tupdateParams := sqlc_queries.UpdateChatSessionTopicByUUIDParams{\n\t\tUuid: chatSession.Uuid,\n\t\tUserID: userID,\n\t\tTopic: genTitle,\n\t}\n\t_, err = h.service.q.UpdateChatSessionTopicByUUID(ctx, updateParams)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to update session title: %v\", err)\n\t\treturn\n\t}\n\n\tlog.Printf(\"Generated LLM title for session %s: %s\", chatSession.Uuid, genTitle)\n}\n\n// sendSuggestedQuestionsStream sends suggested questions as a separate streaming event\nfunc (h *ChatHandler) sendSuggestedQuestionsStream(w http.ResponseWriter, answerID string, suggestedQuestionsJSON json.RawMessage) {\n\t// Parse the suggested questions JSON\n\tvar suggestedQuestions []string\n\tif err := json.Unmarshal(suggestedQuestionsJSON, &suggestedQuestions); err != nil {\n\t\tlog.Printf(\"Warning: Failed to parse suggested questions for streaming: %v\", err)\n\t\treturn\n\t}\n\n\t// Only send if we have questions\n\tif len(suggestedQuestions) == 0 {\n\t\treturn\n\t}\n\n\t// Get the flusher for streaming\n\tflusher, ok := w.(http.Flusher)\n\tif !ok {\n\t\tlog.Printf(\"Warning: Response writer does not support flushing, cannot send suggested questions stream\")\n\t\treturn\n\t}\n\n\t// Create a special response with suggested questions\n\tsuggestedQuestionsResponse := map[string]interface{}{\n\t\t\"id\": answerID,\n\t\t\"object\": \"chat.completion.chunk\",\n\t\t\"choices\": []map[string]interface{}{\n\t\t\t{\n\t\t\t\t\"index\": 0,\n\t\t\t\t\"delta\": map[string]interface{}{\n\t\t\t\t\t\"content\": \"\", // Empty content\n\t\t\t\t\t\"suggestedQuestions\": suggestedQuestions,\n\t\t\t\t},\n\t\t\t\t\"finish_reason\": nil,\n\t\t\t},\n\t\t},\n\t}\n\n\tdata, err := json.Marshal(suggestedQuestionsResponse)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to marshal suggested questions response: %v\", err)\n\t\treturn\n\t}\n\n\t// Send the streaming event\n\tfmt.Fprintf(w, \"data: %v\\n\\n\", string(data))\n\tflusher.Flush()\n\n\tlog.Printf(\"Sent suggested questions stream for answer ID: %s, questions: %v\", answerID, suggestedQuestions)\n}\n\n// genAnswer is an HTTP handler that sends the stream to the client as Server-Sent Events (SSE)\n// if there is no prompt yet, it will create a new prompt and use it as request\n// otherwise, it will create a message, use prompt + get latest N message + newQuestion as request\nfunc genAnswer(h *ChatHandler, w http.ResponseWriter, ctx context.Context, chatSessionUuid string, chatUuid string, newQuestion string, userID int32, streamOutput bool) {\n\n\t// Validate chat session and get model info\n\tchatSession, _, baseURL, ok := h.validateChatSession(ctx, w, chatSessionUuid)\n\tif !ok {\n\t\treturn\n\t}\n\tlog.Printf(\"Processing chat session - SessionUUID: %s, UserID: %d, Model: %s\", chatSession.Uuid, userID, chatSession.Model)\n\n\t// Handle prompt creation or user message addition\n\tif !h.handlePromptCreation(ctx, w, chatSession, chatUuid, newQuestion, userID, baseURL) {\n\t\treturn\n\t}\n\n\t// Generate and save the answer\n\th.generateAndSaveAnswer(ctx, w, chatSession, chatUuid, userID, baseURL, streamOutput)\n}\n\nfunc genBotAnswer(h *ChatHandler, w http.ResponseWriter, session sqlc_queries.ChatSession, simpleChatMessages []SimpleChatMessage, snapshotUuid, newQuestion string, userID int32, streamOutput bool) {\n\t_, err := h.service.q.ChatModelByName(context.Background(), session.Model)\n\tif err != nil {\n\t\tapiErr := ErrResourceNotFound(\"Chat model: \" + session.Model)\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tmessages := simpleChatMessagesToMessages(simpleChatMessages)\n\tmessages = append(messages, models.Message{\n\t\tRole: \"user\",\n\t\tContent: newQuestion,\n\t})\n\tmodel := h.chooseChatModel(session, messages)\n\n\tLLMAnswer, err := model.Stream(w, session, messages, \"\", false, streamOutput)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"Failed to generate answer\"))\n\t\treturn\n\t}\n\n\tctx := context.Background()\n\n\t// Save to bot answer history\n\thistoryParams := sqlc_queries.CreateBotAnswerHistoryParams{\n\t\tBotUuid: snapshotUuid,\n\t\tUserID: userID,\n\t\tPrompt: newQuestion,\n\t\tAnswer: LLMAnswer.Answer,\n\t\tModel: session.Model,\n\t\tTokensUsed: int32(len(LLMAnswer.Answer)) / 4, // Approximate token count\n\t}\n\tif _, err := h.service.q.CreateBotAnswerHistory(ctx, historyParams); err != nil {\n\t\tlog.Printf(\"Failed to save bot answer history: %v\", err)\n\t\t// Don't fail the request, just log the error\n\t}\n\n\tif !isTest(messages) {\n\t\th.service.logChat(session, messages, LLMAnswer.Answer)\n\t}\n}\n\n// Helper function to convert SimpleChatMessage to Message\nfunc simpleChatMessagesToMessages(simpleChatMessages []SimpleChatMessage) []models.Message {\n\tmessages := make([]models.Message, len(simpleChatMessages))\n\tfor i, scm := range simpleChatMessages {\n\t\trole := \"user\"\n\t\tif scm.Inversion {\n\t\t\trole = \"assistant\"\n\t\t}\n\t\tif i == 0 {\n\t\t\trole = \"system\"\n\t\t}\n\t\tmessages[i] = models.Message{\n\t\t\tRole: role,\n\t\t\tContent: scm.Text,\n\t\t}\n\t}\n\treturn messages\n}\n\nfunc regenerateAnswer(h *ChatHandler, w http.ResponseWriter, ctx context.Context, chatSessionUuid string, chatUuid string, stream bool) {\n\n\t// Validate chat session\n\tchatSession, _, _, ok := h.validateChatSession(ctx, w, chatSessionUuid)\n\tif !ok {\n\t\treturn\n\t}\n\n\tmsgs, err := h.service.getAskMessages(*chatSession, chatUuid, true)\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to get chat messages\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Store the request context so models can access it\n\th.requestCtx = ctx\n\tmodel := h.chooseChatModel(*chatSession, msgs)\n\tLLMAnswer, err := model.Stream(w, *chatSession, msgs, chatUuid, true, stream)\n\tif err != nil {\n\t\tlog.Printf(\"Error regenerating answer: %v\", err)\n\t\treturn\n\t}\n\n\th.service.logChat(*chatSession, msgs, LLMAnswer.Answer)\n\n\tif err := h.service.UpdateChatMessageContent(ctx, chatUuid, LLMAnswer.Answer); err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to update message\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Generate suggested questions if explore mode is enabled\n\tif chatSession.ExploreMode {\n\t\tsuggestedQuestions := h.service.generateSuggestedQuestions(LLMAnswer.Answer, msgs)\n\t\tif len(suggestedQuestions) > 0 {\n\t\t\t// Update the message with suggested questions in database\n\t\t\tquestionsJSON, err := json.Marshal(suggestedQuestions)\n\t\t\tif err == nil {\n\t\t\t\th.service.UpdateChatMessageSuggestions(ctx, chatUuid, questionsJSON)\n\n\t\t\t\t// Stream suggested questions to frontend\n\t\t\t\tif stream {\n\t\t\t\t\th.sendSuggestedQuestionsStream(w, LLMAnswer.AnswerId, questionsJSON)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n}\n\n// GetRequestContext returns the current request context for streaming operations\nfunc (h *ChatHandler) GetRequestContext() context.Context {\n\treturn h.requestCtx\n}\n\nfunc (h *ChatHandler) chooseChatModel(chat_session sqlc_queries.ChatSession, msgs []models.Message) ChatModel {\n\tmodel := chat_session.Model\n\tisTestChat := isTest(msgs)\n\n\t// If this is a test chat, return the test model immediately\n\tif isTestChat {\n\t\treturn &TestChatModel{h: h}\n\t}\n\n\t// Get the chat model from database to access api_type field\n\tchatModel, err := GetChatModel(h.service.q, model)\n\tif err != nil {\n\t\t// Fallback to OpenAI if model not found in database\n\t\treturn &OpenAIChatModel{h: h}\n\t}\n\n\t// Use api_type field from database instead of string prefix matching\n\tapiType := chatModel.ApiType\n\n\tcompletionModel := mapset.NewSet[string]()\n\t// completionModel.Add(openai.GPT3TextDavinci002)\n\tisCompletion := completionModel.Contains(model)\n\n\tvar chatModelImpl ChatModel\n\tswitch apiType {\n\tcase \"claude\":\n\t\tchatModelImpl = &Claude3ChatModel{h: h}\n\tcase \"ollama\":\n\t\tchatModelImpl = &OllamaChatModel{h: h}\n\tcase \"gemini\":\n\t\tchatModelImpl = NewGeminiChatModel(h)\n\tcase \"custom\":\n\t\tchatModelImpl = &CustomChatModel{h: h}\n\tcase \"openai\":\n\t\tif isCompletion {\n\t\t\tchatModelImpl = &CompletionChatModel{h: h}\n\t\t} else {\n\t\t\tchatModelImpl = &OpenAIChatModel{h: h}\n\t\t}\n\tdefault:\n\t\t// Default to OpenAI for unknown api types\n\t\tchatModelImpl = &OpenAIChatModel{h: h}\n\t}\n\treturn chatModelImpl\n}\n\n// isTest determines if the chat messages indicate this is a test scenario\nfunc isTest(msgs []models.Message) bool {\n\tif len(msgs) == 0 {\n\t\treturn false\n\t}\n\n\tlastMsgs := msgs[len(msgs)-1]\n\tpromptMsg := msgs[0]\n\n\t// Check if either first or last message contains test demo marker\n\treturn (len(promptMsg.Content) >= TestPrefixLength && promptMsg.Content[:TestPrefixLength] == TestDemoPrefix) ||\n\t\t(len(lastMsgs.Content) >= TestPrefixLength && lastMsgs.Content[:TestPrefixLength] == TestDemoPrefix)\n}\n\nfunc (h *ChatHandler) CheckModelAccess(w http.ResponseWriter, chatSessionUuid string, model string, userID int32) bool {\n\tchatModel, err := h.service.q.ChatModelByName(context.Background(), model)\n\tif err != nil {\n\t\tlog.WithError(err).WithField(\"model\", model).Error(\"Chat model not found\")\n\t\tRespondWithAPIError(w, ErrResourceNotFound(\"chat model: \"+model))\n\t\treturn true\n\t}\n\tlog.Printf(\"%+v\", chatModel)\n\tif !chatModel.EnablePerModeRatelimit {\n\t\treturn false\n\t}\n\tctx := context.Background()\n\trate, err := h.service.q.RateLimiteByUserAndSessionUUID(ctx,\n\t\tsqlc_queries.RateLimiteByUserAndSessionUUIDParams{\n\t\t\tUuid: chatSessionUuid,\n\t\t\tUserID: userID,\n\t\t})\n\tlog.Printf(\"%+v\", rate)\n\tif err != nil {\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\t// If no rate limit is found, use a default value instead of returning an error\n\t\t\tlog.Printf(\"No rate limit found for user %d and session %s, using default\", userID, chatSessionUuid)\n\t\t\treturn false\n\t\t}\n\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to get rate limit\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn true\n\t}\n\n\t// get last model usage in 10min\n\tusage10Min, err := h.service.q.GetChatMessagesCountByUserAndModel(ctx,\n\t\tsqlc_queries.GetChatMessagesCountByUserAndModelParams{\n\t\t\tUserID: userID,\n\t\t\tModel: rate.ChatModelName,\n\t\t})\n\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to get usage data\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn true\n\t}\n\n\tlog.Printf(\"%+v\", usage10Min)\n\n\tif int32(usage10Min) > rate.RateLimit {\n\t\tapiErr := ErrTooManyRequests\n\t\tapiErr.Message = fmt.Sprintf(\"Rate limit exceeded for %s\", rate.ChatModelName)\n\t\tapiErr.Detail = fmt.Sprintf(\"Usage: %d, Limit: %d\", usage10Min, rate.RateLimit)\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn true\n\t}\n\treturn false\n}\n" }, { "path": "api/chat_main_service.go", "content": "package main\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"log\"\n\t\"net/http\"\n\t\"os\"\n\t\"strings\"\n\t\"time\"\n\n\t_ \"embed\"\n\t\"github.com/rotisserie/eris\"\n\t\"github.com/samber/lo\"\n\topenai \"github.com/sashabaranov/go-openai\"\n\t\"github.com/swuecho/chat_backend/llm/gemini\"\n\tmodels \"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype ChatService struct {\n\tq *sqlc_queries.Queries\n}\n\n//go:embed artifact_instruction.txt\nvar artifactInstructionText string\n\n// NewChatService creates a new ChatService with database queries.\nfunc NewChatService(q *sqlc_queries.Queries) *ChatService {\n\treturn &ChatService{q: q}\n}\n\n// loadArtifactInstruction loads the artifact instruction from file.\n// Returns the instruction content or an error if the file cannot be read.\nfunc loadArtifactInstruction() (string, error) {\n\tif artifactInstructionText == \"\" {\n\t\treturn \"\", eris.New(\"artifact instruction text is empty\")\n\t}\n\treturn artifactInstructionText, nil\n}\n\nfunc appendInstructionToSystemMessage(msgs []models.Message, instruction string) {\n\tif instruction == \"\" || len(msgs) == 0 {\n\t\treturn\n\t}\n\n\tsystemMsgFound := false\n\tfor i, msg := range msgs {\n\t\tif msg.Role == \"system\" {\n\t\t\tmsgs[i].Content = msg.Content + \"\\n\" + instruction\n\t\t\tmsgs[i].SetTokenCount(int32(len(msgs[i].Content) / TokenEstimateRatio))\n\t\t\tsystemMsgFound = true\n\t\t\tbreak\n\t\t}\n\t}\n\n\tif !systemMsgFound {\n\t\tmsgs[0].Content = msgs[0].Content + \"\\n\" + instruction\n\t\tmsgs[0].SetTokenCount(int32(len(msgs[0].Content) / TokenEstimateRatio))\n\t}\n}\n\n// getAskMessages retrieves and processes chat messages for LLM requests.\n// It combines prompts and messages, applies length limits, and adds artifact instructions (unless explore mode is enabled).\n// Parameters:\n// - chatSession: The chat session containing configuration\n// - chatUuid: UUID for message identification (used in regenerate mode)\n// - regenerate: If true, excludes the target message from history\n//\n// Returns combined message array or error.\nfunc (s *ChatService) getAskMessages(chatSession sqlc_queries.ChatSession, chatUuid string, regenerate bool) ([]models.Message, error) {\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second*RequestTimeoutSeconds)\n\tdefer cancel()\n\n\tchatSessionUuid := chatSession.Uuid\n\n\tlastN := chatSession.MaxLength\n\tif chatSession.MaxLength == 0 {\n\t\tlastN = DefaultMaxLength\n\t}\n\n\tchat_prompts, err := s.q.GetChatPromptsBySessionUUID(ctx, chatSessionUuid)\n\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"fail to get prompt: \")\n\t}\n\n\tvar chatMessages []sqlc_queries.ChatMessage\n\tif regenerate {\n\t\tchatMessages, err = s.q.GetLastNChatMessages(ctx,\n\t\t\tsqlc_queries.GetLastNChatMessagesParams{\n\t\t\t\tChatSessionUuid: chatSessionUuid,\n\t\t\t\tUuid: chatUuid,\n\t\t\t\tLimit: lastN,\n\t\t\t})\n\n\t} else {\n\t\tchatMessages, err = s.q.GetLatestMessagesBySessionUUID(ctx,\n\t\t\tsqlc_queries.GetLatestMessagesBySessionUUIDParams{ChatSessionUuid: chatSession.Uuid, Limit: lastN})\n\t}\n\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"fail to get messages: \")\n\t}\n\tchatPromptMsgs := lo.Map(chat_prompts, func(m sqlc_queries.ChatPrompt, _ int) models.Message {\n\t\tmsg := models.Message{Role: m.Role, Content: m.Content}\n\t\tmsg.SetTokenCount(int32(m.TokenCount))\n\t\treturn msg\n\t})\n\tchatMessageMsgs := lo.Map(chatMessages, func(m sqlc_queries.ChatMessage, _ int) models.Message {\n\t\tmsg := models.Message{Role: m.Role, Content: m.Content}\n\t\tmsg.SetTokenCount(int32(m.TokenCount))\n\t\treturn msg\n\t})\n\tmsgs := append(chatPromptMsgs, chatMessageMsgs...)\n\n\t// Add artifact instruction to system messages only if artifact mode is enabled\n\tif chatSession.ArtifactEnabled {\n\t\tartifactInstruction, err := loadArtifactInstruction()\n\t\tif err != nil {\n\t\t\tlog.Printf(\"Warning: Failed to load artifact instruction: %v\", err)\n\t\t\tartifactInstruction = \"\" // Use empty string if file can't be loaded\n\t\t}\n\n\t\tappendInstructionToSystemMessage(msgs, artifactInstruction)\n\t}\n\n\treturn msgs, nil\n}\n\n// CreateChatPromptSimple creates a new chat prompt for a session.\n// This is typically used to start a new conversation with a system message.\nfunc (s *ChatService) CreateChatPromptSimple(ctx context.Context, chatSessionUuid string, newQuestion string, userID int32) (sqlc_queries.ChatPrompt, error) {\n\ttokenCount, err := getTokenCount(newQuestion)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to get token count for prompt: %v\", err)\n\t\ttokenCount = len(newQuestion) / TokenEstimateRatio // Fallback estimate\n\t}\n\tchatPrompt, err := s.q.CreateChatPrompt(ctx,\n\t\tsqlc_queries.CreateChatPromptParams{\n\t\t\tUuid: NewUUID(),\n\t\t\tChatSessionUuid: chatSessionUuid,\n\t\t\tRole: \"system\",\n\t\t\tContent: newQuestion,\n\t\t\tUserID: userID,\n\t\t\tCreatedBy: userID,\n\t\t\tUpdatedBy: userID,\n\t\t\tTokenCount: int32(tokenCount),\n\t\t})\n\treturn chatPrompt, err\n}\n\n// CreateChatMessageSimple creates a new chat message with optional summarization and artifact extraction.\n// Handles token counting, content summarization for long messages, and artifact parsing.\n// Parameters:\n// - ctx: Request context for cancellation\n// - sessionUuid, uuid: Message and session identifiers\n// - role: Message role (user/assistant/system)\n// - content, reasoningContent: Message content and reasoning (if any)\n// - model: LLM model name\n// - userId: User ID for ownership\n// - baseURL: API base URL for summarization\n// - is_summarize_mode: Whether to enable automatic summarization\n//\n// Returns created message or error.\nfunc (s *ChatService) CreateChatMessageSimple(ctx context.Context, sessionUuid, uuid, role, content, reasoningContent, model string, userId int32, baseURL string, is_summarize_mode bool) (sqlc_queries.ChatMessage, error) {\n\tnumTokens, err := getTokenCount(content)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to get token count: %v\", err)\n\t\tnumTokens = len(content) / TokenEstimateRatio // Fallback estimate\n\t}\n\n\tsummary := \"\"\n\n\tif is_summarize_mode && numTokens > SummarizeThreshold {\n\t\tlog.Println(\"summarizing\")\n\t\tsummary = llm_summarize_with_timeout(baseURL, content)\n\t\tlog.Println(\"summarizing: \" + summary)\n\t}\n\n\t// Extract artifacts from content\n\tartifacts := extractArtifacts(content)\n\tartifactsJSON, err := json.Marshal(artifacts)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to marshal artifacts: %v\", err)\n\t\tartifactsJSON = json.RawMessage([]byte(\"[]\"))\n\t}\n\n\tchatMessage := sqlc_queries.CreateChatMessageParams{\n\t\tChatSessionUuid: sessionUuid,\n\t\tUuid: uuid,\n\t\tRole: role,\n\t\tContent: content,\n\t\tReasoningContent: reasoningContent,\n\t\tModel: model,\n\t\tUserID: userId,\n\t\tCreatedBy: userId,\n\t\tUpdatedBy: userId,\n\t\tLlmSummary: summary,\n\t\tTokenCount: int32(numTokens),\n\t\tRaw: json.RawMessage([]byte(\"{}\")),\n\t\tArtifacts: artifactsJSON,\n\t\tSuggestedQuestions: json.RawMessage([]byte(\"[]\")),\n\t}\n\tmessage, err := s.q.CreateChatMessage(ctx, chatMessage)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatMessage{}, eris.Wrap(err, \"failed to create message \")\n\t}\n\treturn message, nil\n}\n\n// CreateChatMessageWithSuggestedQuestions creates a chat message with optional suggested questions for explore mode\nfunc (s *ChatService) CreateChatMessageWithSuggestedQuestions(ctx context.Context, sessionUuid, uuid, role, content, reasoningContent, model string, userId int32, baseURL string, is_summarize_mode, exploreMode bool, messages []models.Message) (sqlc_queries.ChatMessage, error) {\n\tnumTokens, err := getTokenCount(content)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to get token count: %v\", err)\n\t\tnumTokens = len(content) / TokenEstimateRatio // Fallback estimate\n\t}\n\n\tsummary := \"\"\n\tif is_summarize_mode && numTokens > SummarizeThreshold {\n\t\tlog.Println(\"summarizing\")\n\t\tsummary = llm_summarize_with_timeout(baseURL, content)\n\t\tlog.Println(\"summarizing: \" + summary)\n\t}\n\n\t// Extract artifacts from content\n\tartifacts := extractArtifacts(content)\n\tartifactsJSON, err := json.Marshal(artifacts)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to marshal artifacts: %v\", err)\n\t\tartifactsJSON = json.RawMessage([]byte(\"[]\"))\n\t}\n\n\t// Generate suggested questions if explore mode is enabled and role is assistant\n\tsuggestedQuestions := json.RawMessage([]byte(\"[]\"))\n\tif exploreMode && role == \"assistant\" && messages != nil {\n\t\tquestions := s.generateSuggestedQuestions(content, messages)\n\t\tif questionsJSON, err := json.Marshal(questions); err == nil {\n\t\t\tsuggestedQuestions = questionsJSON\n\t\t} else {\n\t\t\tlog.Printf(\"Warning: Failed to marshal suggested questions: %v\", err)\n\t\t}\n\t}\n\n\tchatMessage := sqlc_queries.CreateChatMessageParams{\n\t\tChatSessionUuid: sessionUuid,\n\t\tUuid: uuid,\n\t\tRole: role,\n\t\tContent: content,\n\t\tReasoningContent: reasoningContent,\n\t\tModel: model,\n\t\tUserID: userId,\n\t\tCreatedBy: userId,\n\t\tUpdatedBy: userId,\n\t\tLlmSummary: summary,\n\t\tTokenCount: int32(numTokens),\n\t\tRaw: json.RawMessage([]byte(\"{}\")),\n\t\tArtifacts: artifactsJSON,\n\t\tSuggestedQuestions: suggestedQuestions,\n\t}\n\tmessage, err := s.q.CreateChatMessage(ctx, chatMessage)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatMessage{}, eris.Wrap(err, \"failed to create message \")\n\t}\n\treturn message, nil\n}\n\n// generateSuggestedQuestions generates follow-up questions based on the conversation context\nfunc (s *ChatService) generateSuggestedQuestions(content string, messages []models.Message) []string {\n\t// Create a simplified prompt to generate follow-up questions\n\tprompt := `Based on the following conversation, generate 3 thoughtful follow-up questions that would help explore the topic further. Return only the questions, one per line, without numbering or bullet points.\n\nConversation context:\n`\n\n\t// Add the last few messages for context (limit to avoid token overflow)\n\tcontextMessages := messages\n\tif len(messages) > 6 {\n\t\tcontextMessages = messages[len(messages)-6:]\n\t}\n\n\tfor _, msg := range contextMessages {\n\t\tprompt += fmt.Sprintf(\"%s: %s\\n\", msg.Role, msg.Content)\n\t}\n\n\tprompt += fmt.Sprintf(\"assistant: %s\\n\\nGenerate 3 follow-up questions:\", content)\n\n\t// Use the preferred models (deepseek-chat or gemini-2.0-flash) to generate suggestions\n\tquestions := s.callLLMForSuggestions(prompt)\n\n\t// Parse the response into individual questions\n\tlines := strings.Split(strings.TrimSpace(questions), \"\\n\")\n\tvar result []string\n\tfor _, line := range lines {\n\t\tline = strings.TrimSpace(line)\n\t\tif line != \"\" && len(result) < 3 {\n\t\t\t// Clean up any numbering or bullet points that might remain\n\t\t\tline = strings.TrimPrefix(line, \"1. \")\n\t\t\tline = strings.TrimPrefix(line, \"2. \")\n\t\t\tline = strings.TrimPrefix(line, \"3. \")\n\t\t\tline = strings.TrimPrefix(line, \"- \")\n\t\t\tline = strings.TrimPrefix(line, \"• \")\n\t\t\tresult = append(result, line)\n\t\t}\n\t}\n\n\treturn result\n}\n\n// callLLMForSuggestions makes a simple API call to generate suggested questions\nfunc (s *ChatService) callLLMForSuggestions(prompt string) string {\n\tctx := context.Background()\n\n\t// Get all models and find preferred models for suggestions\n\tallModels, err := s.q.ListChatModels(ctx)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to list models for suggestions: %v\", err)\n\t\treturn \"\"\n\t}\n\n\t// Filter for enabled models and prioritize deepseek-chat or gemini-2.0-flash\n\tvar selectedModel sqlc_queries.ChatModel\n\tvar foundPreferred bool\n\n\t// First pass: look for preferred models\n\tfor _, model := range allModels {\n\t\tif !model.IsEnable {\n\t\t\tcontinue\n\t\t}\n\t\tmodelNameLower := strings.ToLower(model.Name)\n\t\tif strings.Contains(modelNameLower, \"deepseek-chat\") || strings.Contains(modelNameLower, \"gemini-2.0-flash\") {\n\t\t\tselectedModel = model\n\t\t\tfoundPreferred = true\n\t\t\tbreak\n\t\t}\n\t}\n\n\t// Second pass: fallback to any gemini or openai model if preferred not found\n\tif !foundPreferred {\n\t\tfor _, model := range allModels {\n\t\t\tif !model.IsEnable {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tapiType := strings.ToLower(model.ApiType)\n\t\t\tmodelName := strings.ToLower(model.Name)\n\n\t\t\t// Prefer gemini models, then openai\n\t\t\tif apiType == \"gemini\" || (apiType == \"openai\" && strings.Contains(modelName, \"gpt\")) {\n\t\t\t\tselectedModel = model\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t}\n\n\tif selectedModel.ID == 0 {\n\t\tlog.Printf(\"Warning: No suitable models available for suggestions\")\n\t\treturn \"\"\n\t}\n\n\t// Use different API calls based on model type\n\tapiType := strings.ToLower(selectedModel.ApiType)\n\tmodelName := strings.ToLower(selectedModel.Name)\n\n\tif apiType == \"gemini\" || strings.Contains(modelName, \"gemini\") {\n\t\treturn s.callGeminiForSuggestions(ctx, selectedModel, prompt)\n\t} else if strings.Contains(modelName, \"deepseek\") || apiType == \"openai\" {\n\t\treturn s.callOpenAICompatibleForSuggestions(ctx, selectedModel, prompt)\n\t}\n\n\tlog.Printf(\"Warning: Unsupported model type for suggestions: %s\", selectedModel.ApiType)\n\treturn \"\"\n}\n\n// callGeminiForSuggestions makes a Gemini API call for suggestions\nfunc (s *ChatService) callGeminiForSuggestions(ctx context.Context, model sqlc_queries.ChatModel, prompt string) string {\n\t// Validate API key\n\tapiKey := os.Getenv(\"GEMINI_API_KEY\")\n\tif apiKey == \"\" {\n\t\tlog.Printf(\"Warning: GEMINI_API_KEY environment variable not set\")\n\t\treturn \"\"\n\t}\n\n\t// Create messages for Gemini\n\tmessages := []models.Message{\n\t\t{\n\t\t\tRole: \"user\",\n\t\t\tContent: prompt,\n\t\t},\n\t}\n\n\t// Generate Gemini payload\n\tpayloadBytes, err := gemini.GenGemminPayload(messages, nil)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to generate Gemini payload for suggestions: %v\", err)\n\t\treturn \"\"\n\t}\n\n\t// Build URL\n\turl := gemini.BuildAPIURL(model.Name, false)\n\treq, err := http.NewRequestWithContext(ctx, \"POST\", url, bytes.NewBuffer(payloadBytes))\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to create Gemini request for suggestions: %v\", err)\n\t\treturn \"\"\n\t}\n\treq.Header.Set(\"Content-Type\", \"application/json\")\n\n\t// Make the API call with timeout\n\tctx, cancel := context.WithTimeout(ctx, 30*time.Second)\n\tdefer cancel()\n\n\tanswer, err := gemini.HandleRegularResponse(http.Client{Timeout: 30 * time.Second}, req)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to get Gemini response for suggestions: %v\", err)\n\t\treturn \"\"\n\t}\n\n\tif answer == nil || answer.Answer == \"\" {\n\t\tlog.Printf(\"Warning: Empty response from Gemini for suggestions\")\n\t\treturn \"\"\n\t}\n\n\treturn answer.Answer\n}\n\n// callOpenAICompatibleForSuggestions makes an OpenAI-compatible API call for suggestions (including deepseek)\nfunc (s *ChatService) callOpenAICompatibleForSuggestions(ctx context.Context, model sqlc_queries.ChatModel, prompt string) string {\n\t// Generate OpenAI client configuration\n\tconfig, err := genOpenAIConfig(model)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to generate OpenAI configuration for suggestions: %v\", err)\n\t\treturn \"\"\n\t}\n\n\tclient := openai.NewClientWithConfig(config)\n\n\t// Create a simple chat completion request for generating suggestions\n\treq := openai.ChatCompletionRequest{\n\t\tModel: model.Name,\n\t\tTemperature: DefaultTemperature,\n\t\tMessages: []openai.ChatCompletionMessage{\n\t\t\t{\n\t\t\t\tRole: \"user\",\n\t\t\t\tContent: prompt,\n\t\t\t},\n\t\t},\n\t\tMaxTokens: 200, // Keep suggestions concise\n\t}\n\n\t// Make the API call with timeout\n\tctx, cancel := context.WithTimeout(ctx, 30*time.Second)\n\tdefer cancel()\n\n\tresp, err := client.CreateChatCompletion(ctx, req)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to generate suggested questions with %s: %v\", model.Name, err)\n\t\treturn \"\"\n\t}\n\n\tif len(resp.Choices) == 0 {\n\t\tlog.Printf(\"Warning: No response choices returned for suggested questions from %s\", model.Name)\n\t\treturn \"\"\n\t}\n\n\treturn resp.Choices[0].Message.Content\n}\n\n// UpdateChatMessageContent updates the content of an existing chat message.\n// Recalculates token count for the updated content.\nfunc (s *ChatService) UpdateChatMessageContent(ctx context.Context, uuid, content string) error {\n\t// encode\n\t// num_tokens\n\tnum_tokens, err := getTokenCount(content)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to get token count for update: %v\", err)\n\t\tnum_tokens = len(content) / TokenEstimateRatio // Fallback estimate\n\t}\n\n\terr = s.q.UpdateChatMessageContent(ctx, sqlc_queries.UpdateChatMessageContentParams{\n\t\tUuid: uuid,\n\t\tContent: content,\n\t\tTokenCount: int32(num_tokens),\n\t})\n\treturn err\n}\n\n// UpdateChatMessageSuggestions updates the suggested questions for a chat message\nfunc (s *ChatService) UpdateChatMessageSuggestions(ctx context.Context, uuid string, suggestedQuestions json.RawMessage) error {\n\t_, err := s.q.UpdateChatMessageSuggestions(ctx, sqlc_queries.UpdateChatMessageSuggestionsParams{\n\t\tUuid: uuid,\n\t\tSuggestedQuestions: suggestedQuestions,\n\t})\n\treturn err\n}\n\n// logChat creates a chat log entry for analytics and debugging.\n// Logs the session, messages, and LLM response for audit purposes.\nfunc (s *ChatService) logChat(chatSession sqlc_queries.ChatSession, msgs []models.Message, answerText string) {\n\t// log chat\n\tsessionRaw := chatSession.ToRawMessage()\n\tif sessionRaw == nil {\n\t\tlog.Println(\"failed to marshal chat session\")\n\t\treturn\n\t}\n\tquestion, err := json.Marshal(msgs)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to marshal chat messages: %v\", err)\n\t\treturn // Skip logging if marshalling fails\n\t}\n\tanswerRaw, err := json.Marshal(answerText)\n\tif err != nil {\n\t\tlog.Printf(\"Warning: Failed to marshal answer: %v\", err)\n\t\treturn // Skip logging if marshalling fails\n\t}\n\n\ts.q.CreateChatLog(context.Background(), sqlc_queries.CreateChatLogParams{\n\t\tSession: *sessionRaw,\n\t\tQuestion: question,\n\t\tAnswer: answerRaw,\n\t})\n}\n" }, { "path": "api/chat_message_handler.go", "content": "package main\n\nimport (\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"net/http\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/gorilla/mux\"\n\t\"github.com/samber/lo\"\n\t\"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype ChatMessageHandler struct {\n\tservice *ChatMessageService\n}\n\nfunc NewChatMessageHandler(sqlc_q *sqlc_queries.Queries) *ChatMessageHandler {\n\tchatMessageService := NewChatMessageService(sqlc_q)\n\treturn &ChatMessageHandler{\n\t\tservice: chatMessageService,\n\t}\n}\n\nfunc (h *ChatMessageHandler) Register(router *mux.Router) {\n\trouter.HandleFunc(\"/chat_messages\", h.CreateChatMessage).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/chat_messages/{id}\", h.GetChatMessageByID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/chat_messages/{id}\", h.UpdateChatMessage).Methods(http.MethodPut)\n\trouter.HandleFunc(\"/chat_messages/{id}\", h.DeleteChatMessage).Methods(http.MethodDelete)\n\trouter.HandleFunc(\"/chat_messages\", h.GetAllChatMessages).Methods(http.MethodGet)\n\n\trouter.HandleFunc(\"/uuid/chat_messages/{uuid}\", h.GetChatMessageByUUID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/uuid/chat_messages/{uuid}\", h.UpdateChatMessageByUUID).Methods(http.MethodPut)\n\trouter.HandleFunc(\"/uuid/chat_messages/{uuid}\", h.DeleteChatMessageByUUID).Methods(http.MethodDelete)\n\trouter.HandleFunc(\"/uuid/chat_messages/{uuid}/generate-suggestions\", h.GenerateMoreSuggestions).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/uuid/chat_messages/chat_sessions/{uuid}\", h.GetChatHistoryBySessionUUID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/uuid/chat_messages/chat_sessions/{uuid}\", h.DeleteChatMessagesBySesionUUID).Methods(http.MethodDelete)\n}\n\n//type userIdContextKey string\n\n//const userIDKey = userIdContextKey(\"userID\")\n\nfunc (h *ChatMessageHandler) CreateChatMessage(w http.ResponseWriter, r *http.Request) {\n\tvar messageParams sqlc_queries.CreateChatMessageParams\n\terr := json.NewDecoder(r.Body).Decode(&messageParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tmessage, err := h.service.CreateChatMessage(r.Context(), messageParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to create chat message\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(message)\n}\n\nfunc (h *ChatMessageHandler) GetChatMessageByID(w http.ResponseWriter, r *http.Request) {\n\tidStr := mux.Vars(r)[\"id\"]\n\tid, err := strconv.Atoi(idStr)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"invalid chat message ID\"))\n\t\treturn\n\t}\n\tmessage, err := h.service.GetChatMessageByID(r.Context(), int32(id))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get chat message\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(message)\n}\n\nfunc (h *ChatMessageHandler) UpdateChatMessage(w http.ResponseWriter, r *http.Request) {\n\tidStr := mux.Vars(r)[\"id\"]\n\tid, err := strconv.Atoi(idStr)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"invalid chat message ID\"))\n\t\treturn\n\t}\n\tvar messageParams sqlc_queries.UpdateChatMessageParams\n\terr = json.NewDecoder(r.Body).Decode(&messageParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tmessageParams.ID = int32(id)\n\tmessage, err := h.service.UpdateChatMessage(r.Context(), messageParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to update chat message\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(message)\n}\n\nfunc (h *ChatMessageHandler) DeleteChatMessage(w http.ResponseWriter, r *http.Request) {\n\tidStr := mux.Vars(r)[\"id\"]\n\tid, err := strconv.Atoi(idStr)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"invalid chat message ID\"))\n\t\treturn\n\t}\n\terr = h.service.DeleteChatMessage(r.Context(), int32(id))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to delete chat message\"))\n\t\treturn\n\t}\n\tw.WriteHeader(http.StatusOK)\n}\n\nfunc (h *ChatMessageHandler) GetAllChatMessages(w http.ResponseWriter, r *http.Request) {\n\tmessages, err := h.service.GetAllChatMessages(r.Context())\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get chat messages\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(messages)\n}\n\n// GetChatMessageByUUID get chat message by uuid\nfunc (h *ChatMessageHandler) GetChatMessageByUUID(w http.ResponseWriter, r *http.Request) {\n\tuuidStr := mux.Vars(r)[\"uuid\"]\n\tmessage, err := h.service.GetChatMessageByUUID(r.Context(), uuidStr)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get chat message\"))\n\t\treturn\n\t}\n\n\tjson.NewEncoder(w).Encode(message)\n}\n\n// UpdateChatMessageByUUID update chat message by uuid\nfunc (h *ChatMessageHandler) UpdateChatMessageByUUID(w http.ResponseWriter, r *http.Request) {\n\tvar simple_msg SimpleChatMessage\n\terr := json.NewDecoder(r.Body).Decode(&simple_msg)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tvar messageParams sqlc_queries.UpdateChatMessageByUUIDParams\n\tmessageParams.Uuid = simple_msg.Uuid\n\tmessageParams.Content = simple_msg.Text\n\ttokenCount, _ := getTokenCount(simple_msg.Text)\n\tmessageParams.TokenCount = int32(tokenCount)\n\tmessageParams.IsPin = simple_msg.IsPin\n\tmessage, err := h.service.UpdateChatMessageByUUID(r.Context(), messageParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to update chat message\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(message)\n}\n\n// DeleteChatMessageByUUID delete chat message by uuid\nfunc (h *ChatMessageHandler) DeleteChatMessageByUUID(w http.ResponseWriter, r *http.Request) {\n\tuuidStr := mux.Vars(r)[\"uuid\"]\n\terr := h.service.DeleteChatMessageByUUID(r.Context(), uuidStr)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to delete chat message\"))\n\t\treturn\n\t}\n\tw.WriteHeader(http.StatusOK)\n}\n\n// GetChatMessagesBySessionUUID get chat messages by session uuid\nfunc (h *ChatMessageHandler) GetChatMessagesBySessionUUID(w http.ResponseWriter, r *http.Request) {\n\tuuidStr := mux.Vars(r)[\"uuid\"]\n\tpageNum, err := strconv.Atoi(r.URL.Query().Get(\"page\"))\n\tif err != nil {\n\t\tpageNum = 1\n\t}\n\tpageSize, err := strconv.Atoi(r.URL.Query().Get(\"page_size\"))\n\tif err != nil {\n\t\tpageSize = 200\n\t}\n\n\tmessages, err := h.service.GetChatMessagesBySessionUUID(r.Context(), uuidStr, int32(pageNum), int32(pageSize))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get chat messages\"))\n\t\treturn\n\t}\n\n\tsimple_msgs := lo.Map(messages, func(message sqlc_queries.ChatMessage, _ int) SimpleChatMessage {\n\t\t// Extract artifacts from database\n\t\tvar artifacts []Artifact\n\t\tif message.Artifacts != nil {\n\t\t\terr := json.Unmarshal(message.Artifacts, &artifacts)\n\t\t\tif err != nil {\n\t\t\t\t// Log error but don't fail the request\n\t\t\t\tartifacts = []Artifact{}\n\t\t\t}\n\t\t}\n\n\t\treturn SimpleChatMessage{\n\t\t\tDateTime: message.UpdatedAt.Format(time.RFC3339),\n\t\t\tText: message.Content,\n\t\t\tInversion: message.Role != \"user\",\n\t\t\tError: false,\n\t\t\tLoading: false,\n\t\t\tArtifacts: artifacts,\n\t\t}\n\t})\n\tjson.NewEncoder(w).Encode(simple_msgs)\n}\n\n// GetChatMessagesBySessionUUID get chat messages by session uuid\nfunc (h *ChatMessageHandler) GetChatHistoryBySessionUUID(w http.ResponseWriter, r *http.Request) {\n\tuuidStr := mux.Vars(r)[\"uuid\"]\n\tpageNum, err := strconv.Atoi(r.URL.Query().Get(\"page\"))\n\tif err != nil {\n\t\tpageNum = 1\n\t}\n\tpageSize, err := strconv.Atoi(r.URL.Query().Get(\"page_size\"))\n\tif err != nil {\n\t\tpageSize = 200\n\t}\n\tsimple_msgs, err := h.service.q.GetChatHistoryBySessionUUID(r.Context(), uuidStr, int32(pageNum), int32(pageSize))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get chat history\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(simple_msgs)\n}\n\n// DeleteChatMessagesBySesionUUID delete chat messages by session uuid\nfunc (h *ChatMessageHandler) DeleteChatMessagesBySesionUUID(w http.ResponseWriter, r *http.Request) {\n\tuuidStr := mux.Vars(r)[\"uuid\"]\n\terr := h.service.DeleteChatMessagesBySesionUUID(r.Context(), uuidStr)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to delete chat messages\"))\n\t\treturn\n\t}\n\tw.WriteHeader(http.StatusOK)\n}\n\n// GenerateMoreSuggestions generates additional suggested questions for a message\nfunc (h *ChatMessageHandler) GenerateMoreSuggestions(w http.ResponseWriter, r *http.Request) {\n\tmessageUUID := mux.Vars(r)[\"uuid\"]\n\n\t// Get the existing message\n\tmessage, err := h.service.q.GetChatMessageByUUID(r.Context(), messageUUID)\n\tif err != nil {\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\tRespondWithAPIError(w, ErrChatMessageNotFound.WithMessage(\"Message not found\").WithDebugInfo(err.Error()))\n\t\t} else {\n\t\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get message\"))\n\t\t}\n\t\treturn\n\t}\n\n\t// Only allow suggestions for assistant messages\n\tif message.Role != \"assistant\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Suggestions can only be generated for assistant messages\"))\n\t\treturn\n\t}\n\n\t// Get the session to check if explore mode is enabled\n\tsession, err := h.service.q.GetChatSessionByUUID(r.Context(), message.ChatSessionUuid)\n\tif err != nil {\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\tRespondWithAPIError(w, ErrChatSessionNotFound.WithMessage(\"Session not found\").WithDebugInfo(err.Error()))\n\t\t} else {\n\t\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get session\"))\n\t\t}\n\t\treturn\n\t}\n\n\t// Check if explore mode is enabled\n\tif !session.ExploreMode {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Suggestions are only available in explore mode\"))\n\t\treturn\n\t}\n\n\t// Get conversation context - last 6 messages\n\tcontextMessages, err := h.service.q.GetLatestMessagesBySessionUUID(r.Context(),\n\t\tsqlc_queries.GetLatestMessagesBySessionUUIDParams{\n\t\t\tChatSessionUuid: session.Uuid,\n\t\t\tLimit: 6,\n\t\t})\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get conversation context\"))\n\t\treturn\n\t}\n\n\t// Convert to models.Message format for suggestion generation\n\tvar msgs []models.Message\n\tfor _, msg := range contextMessages {\n\t\tmsgs = append(msgs, models.Message{\n\t\t\tRole: msg.Role,\n\t\t\tContent: msg.Content,\n\t\t})\n\t}\n\n\t// Create a new ChatService to access suggestion generation methods\n\tchatService := NewChatService(h.service.q)\n\n\t// Generate new suggested questions\n\tnewSuggestions := chatService.generateSuggestedQuestions(message.Content, msgs)\n\tif len(newSuggestions) == 0 {\n\t\tRespondWithAPIError(w, createAPIError(ErrInternalUnexpected, \"Failed to generate suggestions\", \"no suggestions returned\"))\n\t\treturn\n\t}\n\n\t// Parse existing suggestions\n\tvar existingSuggestions []string\n\tif len(message.SuggestedQuestions) > 0 {\n\t\tif err := json.Unmarshal(message.SuggestedQuestions, &existingSuggestions); err != nil {\n\t\t\t// If unmarshal fails, treat as empty array\n\t\t\texistingSuggestions = []string{}\n\t\t}\n\t}\n\n\t// Combine existing and new suggestions (avoiding duplicates)\n\tallSuggestions := append(existingSuggestions, newSuggestions...)\n\n\t// Remove duplicates\n\tseenSuggestions := make(map[string]bool)\n\tvar uniqueSuggestions []string\n\tfor _, suggestion := range allSuggestions {\n\t\tif !seenSuggestions[suggestion] {\n\t\t\tseenSuggestions[suggestion] = true\n\t\t\tuniqueSuggestions = append(uniqueSuggestions, suggestion)\n\t\t}\n\t}\n\n\t// Update the message with new suggestions\n\tsuggestionsJSON, err := json.Marshal(uniqueSuggestions)\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(ErrInternalUnexpected, \"Failed to serialize suggestions\", err.Error()))\n\t\treturn\n\t}\n\n\t_, err = h.service.q.UpdateChatMessageSuggestions(r.Context(),\n\t\tsqlc_queries.UpdateChatMessageSuggestionsParams{\n\t\t\tUuid: messageUUID,\n\t\t\tSuggestedQuestions: suggestionsJSON,\n\t\t})\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to update message with suggestions\"))\n\t\treturn\n\t}\n\n\t// Return the new suggestions to the client\n\tresponse := map[string]interface{}{\n\t\t\"newSuggestions\": newSuggestions,\n\t\t\"allSuggestions\": uniqueSuggestions,\n\t}\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tjson.NewEncoder(w).Encode(response)\n}\n" }, { "path": "api/chat_message_service.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\n\t\"github.com/rotisserie/eris\"\n\t\"github.com/swuecho/chat_backend/ai\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype ChatMessageService struct {\n\tq *sqlc_queries.Queries\n}\n\n// NewChatMessageService creates a new ChatMessageService.\nfunc NewChatMessageService(q *sqlc_queries.Queries) *ChatMessageService {\n\treturn &ChatMessageService{q: q}\n}\n\n// CreateChatMessage creates a new chat message.\nfunc (s *ChatMessageService) CreateChatMessage(ctx context.Context, message_params sqlc_queries.CreateChatMessageParams) (sqlc_queries.ChatMessage, error) {\n\tmessage, err := s.q.CreateChatMessage(ctx, message_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatMessage{}, eris.Wrap(err, \"failed to create message \")\n\t}\n\treturn message, nil\n}\n\n// GetChatMessageByID returns a chat message by ID.\nfunc (s *ChatMessageService) GetChatMessageByID(ctx context.Context, id int32) (sqlc_queries.ChatMessage, error) {\n\tmessage, err := s.q.GetChatMessageByID(ctx, id)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatMessage{}, eris.Wrap(err, \"failed to create message \")\n\t}\n\treturn message, nil\n}\n\n// UpdateChatMessage updates an existing chat message.\nfunc (s *ChatMessageService) UpdateChatMessage(ctx context.Context, message_params sqlc_queries.UpdateChatMessageParams) (sqlc_queries.ChatMessage, error) {\n\tmessage_u, err := s.q.UpdateChatMessage(ctx, message_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatMessage{}, eris.Wrap(err, \"failed to update message \")\n\t}\n\treturn message_u, nil\n}\n\n// DeleteChatMessage deletes a chat message by ID.\nfunc (s *ChatMessageService) DeleteChatMessage(ctx context.Context, id int32) error {\n\terr := s.q.DeleteChatMessage(ctx, id)\n\tif err != nil {\n\t\treturn eris.Wrap(err, \"failed to delete message \")\n\t}\n\treturn nil\n}\n\n// DeleteChatMessageByUUID deletes a chat message by uuid\nfunc (s *ChatMessageService) DeleteChatMessageByUUID(ctx context.Context, uuid string) error {\n\terr := s.q.DeleteChatMessageByUUID(ctx, uuid)\n\tif err != nil {\n\t\treturn eris.Wrap(err, \"failed to delete message \")\n\t}\n\treturn nil\n}\n\n// GetAllChatMessages returns all chat messages.\nfunc (s *ChatMessageService) GetAllChatMessages(ctx context.Context) ([]sqlc_queries.ChatMessage, error) {\n\tmessages, err := s.q.GetAllChatMessages(ctx)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to retrieve messages \")\n\t}\n\treturn messages, nil\n}\n\nfunc (s *ChatMessageService) GetLatestMessagesBySessionID(ctx context.Context, chatSessionUuid string, limit int32) ([]sqlc_queries.ChatMessage, error) {\n\tparams := sqlc_queries.GetLatestMessagesBySessionUUIDParams{ChatSessionUuid: chatSessionUuid, Limit: limit}\n\tmsgs, err := s.q.GetLatestMessagesBySessionUUID(ctx, params)\n\tif err != nil {\n\t\treturn []sqlc_queries.ChatMessage{}, err\n\t}\n\treturn msgs, nil\n}\n\nfunc (s *ChatMessageService) GetFirstMessageBySessionUUID(ctx context.Context, chatSessionUuid string) (sqlc_queries.ChatMessage, error) {\n\tmsg, err := s.q.GetFirstMessageBySessionUUID(ctx, chatSessionUuid)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatMessage{}, err\n\t}\n\treturn msg, nil\n}\n\nfunc (s *ChatMessageService) AddMessage(ctx context.Context, chatSessionUuid string, uuid string, role ai.Role, content string, raw []byte) (sqlc_queries.ChatMessage, error) {\n\tparams := sqlc_queries.CreateChatMessageParams{\n\t\tChatSessionUuid: chatSessionUuid,\n\t\tUuid: uuid,\n\t\tRole: role.String(),\n\t\tContent: content,\n\t\tRaw: json.RawMessage(raw),\n\t}\n\tmsg, err := s.q.CreateChatMessage(ctx, params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatMessage{}, err\n\t}\n\treturn msg, nil\n}\n\n// GetChatMessageByUUID returns a chat message by ID.\nfunc (s *ChatMessageService) GetChatMessageByUUID(ctx context.Context, uuid string) (sqlc_queries.ChatMessage, error) {\n\tmessage, err := s.q.GetChatMessageByUUID(ctx, uuid)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatMessage{}, errors.New(\"failed to retrieve message\")\n\t}\n\treturn message, nil\n}\n\n// UpdateChatMessageByUUID updates an existing chat message.\nfunc (s *ChatMessageService) UpdateChatMessageByUUID(ctx context.Context, message_params sqlc_queries.UpdateChatMessageByUUIDParams) (sqlc_queries.ChatMessage, error) {\n\tmessage_u, err := s.q.UpdateChatMessageByUUID(ctx, message_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatMessage{}, eris.Wrap(err, \"failed to update message \")\n\t}\n\treturn message_u, nil\n}\n\n// GetChatMessagesBySessionUUID returns a chat message by session uuid.\nfunc (s *ChatMessageService) GetChatMessagesBySessionUUID(ctx context.Context, uuid string, pageNum, pageSize int32) ([]sqlc_queries.ChatMessage, error) {\n\tparam := sqlc_queries.GetChatMessagesBySessionUUIDParams{\n\t\tUuid: uuid,\n\t\tOffset: pageNum - 1,\n\t\tLimit: pageSize,\n\t}\n\tmessage, err := s.q.GetChatMessagesBySessionUUID(ctx, param)\n\tif err != nil {\n\t\treturn []sqlc_queries.ChatMessage{}, eris.Wrap(err, \"failed to retrieve message \")\n\t}\n\treturn message, nil\n}\n\n// DeleteChatMessagesBySesionUUID deletes chat messages by session uuid.\nfunc (s *ChatMessageService) DeleteChatMessagesBySesionUUID(ctx context.Context, uuid string) error {\n\terr := s.q.DeleteChatMessagesBySesionUUID(ctx, uuid)\n\treturn err\n}\n\nfunc (s *ChatMessageService) GetChatMessagesCount(ctx context.Context, userID int32) (int32, error) {\n\tcount, err := s.q.GetChatMessagesCount(ctx, userID)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn int32(count), nil\n}\n" }, { "path": "api/chat_message_service_test.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\nfunc TestChatMessageService(t *testing.T) {\n\t// Create a new ChatMessageService with the test database connection\n\tq := sqlc_queries.New(db)\n\tservice := NewChatMessageService(q)\n\n\t// Insert a new chat message into the database\n\tmsg_params := sqlc_queries.CreateChatMessageParams{\n\t\tChatSessionUuid: \"1\",\n\t\tUuid: \"test-uuid-1\",\n\t\tRole: \"Test Role\",\n\t\tContent: \"Test Message\",\n\t\tReasoningContent: \"\",\n\t\tModel: \"test-model\",\n\t\tTokenCount: 100,\n\t\tScore: 0.5,\n\t\tUserID: 1,\n\t\tCreatedBy: 1,\n\t\tUpdatedBy: 1,\n\t\tLlmSummary: \"\",\n\t\tRaw: json.RawMessage([]byte(\"{}\")),\n\t\tArtifacts: json.RawMessage([]byte(\"[]\")),\n\t\tSuggestedQuestions: json.RawMessage([]byte(\"[]\")),\n\t}\n\tmsg, err := service.CreateChatMessage(context.Background(), msg_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat message: %v\", err)\n\t}\n\n\t// Retrieve the inserted chat message from the database and check that it matches the expected values\n\tretrieved_msg, err := service.GetChatMessageByID(context.Background(), msg.ID)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to retrieve chat message: %v\", err)\n\t}\n\tif retrieved_msg.ID != msg.ID || retrieved_msg.ChatSessionUuid != msg.ChatSessionUuid ||\n\t\tretrieved_msg.Role != msg.Role || retrieved_msg.Content != msg.Content || retrieved_msg.Score != msg.Score ||\n\t\tretrieved_msg.UserID != msg.UserID || !retrieved_msg.CreatedAt.Equal(msg.CreatedAt) || !retrieved_msg.UpdatedAt.Equal(msg.UpdatedAt) ||\n\t\tretrieved_msg.CreatedBy != msg.CreatedBy || retrieved_msg.UpdatedBy != msg.UpdatedBy {\n\t\tt.Error(\"retrieved chat message does not match expected values\")\n\t}\n\n\t// Delete the chat prompt and check that it was deleted from the database\n\tif err := service.DeleteChatMessage(context.Background(), msg.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat prompt: %v\", err)\n\t}\n\t_, err = service.GetChatMessageByID(context.Background(), msg.ID)\n\tif err == nil || !errors.Is(err, sql.ErrNoRows) {\n\t\tt.Error(\"expected error due to missing chat prompt, but got no error or different error\")\n\t}\n}\n\nfunc TestGetChatMessagesBySessionID(t *testing.T) {\n\n\t// Create a new ChatMessageService with the test database connection\n\tq := sqlc_queries.New(db)\n\tservice := NewChatMessageService(q)\n\n\t// Insert two chat messages into the database with different chat session IDs\n\tmsg1_params := sqlc_queries.CreateChatMessageParams{\n\t\tChatSessionUuid: \"1\",\n\t\tUuid: \"test-uuid-1\",\n\t\tRole: \"Test Role 1\",\n\t\tContent: \"Test Message 1\",\n\t\tReasoningContent: \"\",\n\t\tModel: \"test-model\",\n\t\tTokenCount: 100,\n\t\tScore: 0.5,\n\t\tUserID: 1,\n\t\tCreatedBy: 1,\n\t\tUpdatedBy: 1,\n\t\tLlmSummary: \"\",\n\t\tRaw: json.RawMessage([]byte(\"{}\")),\n\t\tArtifacts: json.RawMessage([]byte(\"[]\")),\n\t\tSuggestedQuestions: json.RawMessage([]byte(\"[]\")),\n\t}\n\tmsg1, err := service.CreateChatMessage(context.Background(), msg1_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat message: %v\", err)\n\t}\n\tmsg2_params := sqlc_queries.CreateChatMessageParams{\n\t\tChatSessionUuid: \"2\",\n\t\tUuid: \"test-uuid-2\",\n\t\tRole: \"Test Role 2\",\n\t\tContent: \"Test Message 2\",\n\t\tReasoningContent: \"\",\n\t\tModel: \"test-model\",\n\t\tTokenCount: 100,\n\t\tScore: 0.75,\n\t\tUserID: 2,\n\t\tCreatedBy: 2,\n\t\tUpdatedBy: 2,\n\t\tLlmSummary: \"\",\n\t\tRaw: json.RawMessage([]byte(\"{}\")),\n\t\tArtifacts: json.RawMessage([]byte(\"[]\")),\n\t\tSuggestedQuestions: json.RawMessage([]byte(\"[]\")),\n\t}\n\tmsg2, err := service.CreateChatMessage(context.Background(), msg2_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat message: %v\", err)\n\t}\n\n\t// Retrieve chat messages by chat session ID and check that they match the expected values\n\t// skip because of there is no chatSession with uuid \"1\" avaialble\n\t// chatSessionID := \"1\"\n\t// msgs, err := service.GetChatMessagesBySessionUUID(context.Background(), chatSessionID, 1, 10)\n\t// if err != nil {\n\t// \tt.Fatalf(\"failed to retrieve chat messages: %v\", err)\n\t// }\n\t// if len(msgs) != 1 {\n\t// \tt.Errorf(\"expected 1 chat message, but got %d\", len(msgs))\n\t// }\n\t// if msgs[0].ChatSessionUuid != msg1.ChatSessionUuid || msgs[0].Role != msg1.Role || msgs[0].Content != msg1.Content ||\n\t// \tmsgs[0].Score != msg1.Score || msgs[0].UserID != msg1.UserID {\n\t// \tt.Error(\"retrieved chat messages do not match expected values\")\n\t// }\n\t// Delete the chat prompt and check that it was deleted from the database\n\tif err := service.DeleteChatMessage(context.Background(), msg1.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat prompt: %v\", err)\n\t}\n\t// Delete the chat prompt and check that it was deleted from the database\n\tif err := service.DeleteChatMessage(context.Background(), msg2.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat prompt: %v\", err)\n\t}\n\n\t_, err = service.GetChatMessageByID(context.Background(), msg1.ID)\n\tif err == nil || !errors.Is(err, sql.ErrNoRows) {\n\t\tt.Error(\"expected error due to missing chat prompt, but got no error or different error\")\n\t}\n\t_, err = service.GetChatMessageByID(context.Background(), msg2.ID)\n\tif err == nil || !errors.Is(err, sql.ErrNoRows) {\n\t\tt.Error(\"expected error due to missing chat prompt, but got no error or different error\")\n\t}\n}\n" }, { "path": "api/chat_model_handler.go", "content": "package main\n\nimport (\n\t\"encoding/json\"\n\t\"net/http\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/gorilla/mux\"\n\t\"github.com/samber/lo\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype ChatModelHandler struct {\n\tdb *sqlc_queries.Queries\n}\n\nfunc NewChatModelHandler(db *sqlc_queries.Queries) *ChatModelHandler {\n\treturn &ChatModelHandler{\n\t\tdb: db,\n\t}\n}\n\nfunc (h *ChatModelHandler) Register(r *mux.Router) {\n\n\t// Assuming db is an instance of the SQLC generated DB struct\n\t//handler := NewChatModelHandler(db)\n\t// r := mux.NewRouter()\n\n\t// TODO: user can read, remove user_id field from the response\n\tr.HandleFunc(\"/chat_model\", h.ListSystemChatModels).Methods(\"GET\")\n\tr.HandleFunc(\"/chat_model/default\", h.GetDefaultChatModel).Methods(\"GET\")\n\tr.HandleFunc(\"/chat_model/{id}\", h.ChatModelByID).Methods(\"GET\")\n\t// create delete update self's chat model\n\tr.HandleFunc(\"/chat_model\", h.CreateChatModel).Methods(\"POST\")\n\tr.HandleFunc(\"/chat_model/{id}\", h.UpdateChatModel).Methods(\"PUT\")\n\tr.HandleFunc(\"/chat_model/{id}\", h.DeleteChatModel).Methods(\"DELETE\")\n}\n\nfunc (h *ChatModelHandler) ListSystemChatModels(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\tChatModels, err := h.db.ListSystemChatModels(ctx)\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to list chat models\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tlatestUsageTimeOfModels, err := h.db.GetLatestUsageTimeOfModel(ctx, \"30 days\")\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to get model usage data\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\t// create a map of model id to usage time\n\tusageTimeMap := make(map[string]sqlc_queries.GetLatestUsageTimeOfModelRow)\n\tfor _, usageTime := range latestUsageTimeOfModels {\n\t\tusageTimeMap[usageTime.Model] = usageTime\n\t}\n\n\t// create a ChatModelWithUsage struct\n\ttype ChatModelWithUsage struct {\n\t\tsqlc_queries.ChatModel\n\t\tLastUsageTime time.Time `json:\"lastUsageTime,omitempty\"`\n\t\tMessageCount int64 `json:\"messageCount\"`\n\t}\n\n\t// merge ChatModels and usageTimeMap with pre-allocated slice\n\tchatModelsWithUsage := lo.Map(ChatModels, func(model sqlc_queries.ChatModel, _ int) ChatModelWithUsage {\n\t\tusage := usageTimeMap[model.Name]\n\t\treturn ChatModelWithUsage{\n\t\t\tChatModel: model,\n\t\t\tLastUsageTime: usage.LatestMessageTime,\n\t\t\tMessageCount: usage.MessageCount,\n\t\t}\n\t})\n\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(chatModelsWithUsage)\n}\n\nfunc (h *ChatModelHandler) ChatModelByID(w http.ResponseWriter, r *http.Request) {\n\tvars := mux.Vars(r)\n\tctx := r.Context()\n\tid, err := strconv.Atoi(vars[\"id\"])\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid chat model ID\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tChatModel, err := h.db.ChatModelByID(ctx, int32(id))\n\tif err != nil {\n\t\tapiErr := ErrResourceNotFound(\"Chat model\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(ChatModel)\n}\n\nfunc (h *ChatModelHandler) CreateChatModel(w http.ResponseWriter, r *http.Request) {\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tvar input struct {\n\t\tName string `json:\"name\"`\n\t\tLabel string `json:\"label\"`\n\t\tIsDefault bool `json:\"isDefault\"`\n\t\tURL string `json:\"url\"`\n\t\tApiAuthHeader string `json:\"apiAuthHeader\"`\n\t\tApiAuthKey string `json:\"apiAuthKey\"`\n\t\tEnablePerModeRatelimit bool `json:\"enablePerModeRatelimit\"`\n\t\tApiType string `json:\"apiType\"`\n\t}\n\n\terr = json.NewDecoder(r.Body).Decode(&input)\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Failed to parse request body\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Set default api_type if not provided\n\tapiType := input.ApiType\n\tif apiType == \"\" {\n\t\tapiType = \"openai\" // default api type\n\t}\n\n\t// Validate api_type\n\tvalidApiTypes := map[string]bool{\n\t\t\"openai\": true,\n\t\t\"claude\": true,\n\t\t\"gemini\": true,\n\t\t\"ollama\": true,\n\t\t\"custom\": true,\n\t}\n\n\tif !validApiTypes[apiType] {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid API type. Valid types are: openai, claude, gemini, ollama, custom\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tChatModel, err := h.db.CreateChatModel(r.Context(), sqlc_queries.CreateChatModelParams{\n\t\tName: input.Name,\n\t\tLabel: input.Label,\n\t\tIsDefault: input.IsDefault,\n\t\tUrl: input.URL,\n\t\tApiAuthHeader: input.ApiAuthHeader,\n\t\tApiAuthKey: input.ApiAuthKey,\n\t\tUserID: userID,\n\t\tEnablePerModeRatelimit: input.EnablePerModeRatelimit,\n\t\tMaxToken: 4096, // default max token\n\t\tDefaultToken: 2048, // default token\n\t\tOrderNumber: 0, // default order\n\t\tHttpTimeOut: 120, // default timeout\n\t\tApiType: apiType,\n\t})\n\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to create chat model\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusCreated)\n\tjson.NewEncoder(w).Encode(ChatModel)\n}\n\nfunc (h *ChatModelHandler) UpdateChatModel(w http.ResponseWriter, r *http.Request) {\n\tvars := mux.Vars(r)\n\tid, err := strconv.Atoi(vars[\"id\"])\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid chat model ID\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tvar input struct {\n\t\tName string `json:\"name\"`\n\t\tLabel string `json:\"label\"`\n\t\tIsDefault bool `json:\"isDefault\"`\n\t\tURL string `json:\"url\"`\n\t\tApiAuthHeader string `json:\"apiAuthHeader\"`\n\t\tApiAuthKey string `json:\"apiAuthKey\"`\n\t\tEnablePerModeRatelimit bool `json:\"enablePerModeRatelimit\"`\n\t\tOrderNumber int32 `json:\"orderNumber\"`\n\t\tDefaultToken int32 `json:\"defaultToken\"`\n\t\tMaxToken int32 `json:\"maxToken\"`\n\t\tHttpTimeOut int32 `json:\"httpTimeOut\"`\n\t\tIsEnable bool `json:\"isEnable\"`\n\t\tApiType string `json:\"apiType\"`\n\t}\n\terr = json.NewDecoder(r.Body).Decode(&input)\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Failed to parse request body\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Set default api_type if not provided\n\tapiType := input.ApiType\n\tif apiType == \"\" {\n\t\tapiType = \"openai\" // default api type\n\t}\n\n\t// Validate api_type\n\tvalidApiTypes := map[string]bool{\n\t\t\"openai\": true,\n\t\t\"claude\": true,\n\t\t\"gemini\": true,\n\t\t\"ollama\": true,\n\t\t\"custom\": true,\n\t}\n\n\tif !validApiTypes[apiType] {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid API type. Valid types are: openai, claude, gemini, ollama, custom\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tChatModel, err := h.db.UpdateChatModel(r.Context(), sqlc_queries.UpdateChatModelParams{\n\t\tID: int32(id),\n\t\tName: input.Name,\n\t\tLabel: input.Label,\n\t\tIsDefault: input.IsDefault,\n\t\tUrl: input.URL,\n\t\tApiAuthHeader: input.ApiAuthHeader,\n\t\tApiAuthKey: input.ApiAuthKey,\n\t\tUserID: userID,\n\t\tEnablePerModeRatelimit: input.EnablePerModeRatelimit,\n\t\tOrderNumber: input.OrderNumber,\n\t\tDefaultToken: input.DefaultToken,\n\t\tMaxToken: input.MaxToken,\n\t\tHttpTimeOut: input.HttpTimeOut,\n\t\tIsEnable: input.IsEnable,\n\t\tApiType: apiType,\n\t})\n\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to update chat model\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(ChatModel)\n}\n\nfunc (h *ChatModelHandler) DeleteChatModel(w http.ResponseWriter, r *http.Request) {\n\tvars := mux.Vars(r)\n\tid, err := strconv.Atoi(vars[\"id\"])\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid chat model ID\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\terr = h.db.DeleteChatModel(r.Context(),\n\t\tsqlc_queries.DeleteChatModelParams{\n\t\t\tID: int32(id),\n\t\t\tUserID: userID,\n\t\t})\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to delete chat model\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusOK)\n}\n\nfunc (h *ChatModelHandler) GetDefaultChatModel(w http.ResponseWriter, r *http.Request) {\n\tChatModel, err := h.db.GetDefaultChatModel(r.Context())\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to retrieve default chat model\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(ChatModel)\n}\n" }, { "path": "api/chat_model_handler_test.go", "content": "package main\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"net/http/httptest\"\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp/cmpopts\"\n\t\"github.com/gorilla/mux\"\n\t\"github.com/samber/lo\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n\t\"gotest.tools/v3/assert\"\n)\n\nfunc createTwoChatModel(q *sqlc_queries.Queries) (sqlc_queries.AuthUser, []sqlc_queries.ChatModel) {\n\t// add a system user\n\tadmin, err := q.CreateAuthUser(context.Background(), sqlc_queries.CreateAuthUserParams{\n\t\tEmail: \"admin@a.com\",\n\t\tUsername: \"test\",\n\t\tPassword: \"test\",\n\t\tIsSuperuser: true,\n\t})\n\n\tif err != nil {\n\t\tfmt.Printf(\"Error creating test data: %s\", err.Error())\n\t}\n\texpectedResults := []sqlc_queries.ChatModel{\n\t\t{\n\t\t\tName: \"Test API 1\",\n\t\t\tLabel: \"Test Label 1\",\n\t\t\tIsDefault: false,\n\t\t\tUrl: \"http://test.url.com\",\n\t\t\tApiAuthHeader: \"Authorization\",\n\t\t\tApiAuthKey: \"TestKey1\",\n\t\t\tUserID: admin.ID,\n\t\t},\n\t\t{\n\t\t\tName: \"Test API 2\",\n\t\t\tLabel: \"Test Label 2\",\n\t\t\tIsDefault: false,\n\t\t\tUrl: \"http://test.url2.com\",\n\t\t\tApiAuthHeader: \"Authorization\",\n\t\t\tApiAuthKey: \"TestKey2\",\n\t\t\tUserID: admin.ID,\n\t\t},\n\t}\n\n\tfor _, api := range expectedResults {\n\t\t_, err := q.CreateChatModel(context.Background(), sqlc_queries.CreateChatModelParams{\n\t\t\tName: api.Name,\n\t\t\tLabel: api.Label,\n\t\t\tIsDefault: api.IsDefault,\n\t\t\tUrl: api.Url,\n\t\t\tApiAuthHeader: api.ApiAuthHeader,\n\t\t\tApiAuthKey: api.ApiAuthKey,\n\t\t\tUserID: api.UserID,\n\t\t})\n\t\tif err != nil {\n\t\t\tfmt.Printf(\"Error creating test data: %s\", err.Error())\n\t\t}\n\t}\n\treturn admin, expectedResults\n}\nfunc clearChatModelsIfExists(q *sqlc_queries.Queries) {\n\tdefaultApis, _ := q.ListChatModels(context.Background())\n\n\tfor _, api := range defaultApis {\n\t\tq.DeleteChatModel(context.Background(),\n\t\t\tsqlc_queries.DeleteChatModelParams{\n\t\t\t\tID: api.ID,\n\t\t\t\tUserID: api.UserID,\n\t\t\t})\n\t}\n}\n\nfunc unmarshalResponseToChatModel(t *testing.T, rr *httptest.ResponseRecorder) []sqlc_queries.ChatModel {\n\t// read the response body\n\t// unmarshal the response body into a list of ChatModel\n\tvar results []sqlc_queries.ChatModel\n\terr := json.NewDecoder(rr.Body).Decode(&results)\n\tassert.NilError(t, err)\n\n\treturn results\n}\n\n// the code below do db update directly in instead of using handler, please change to use handler\nfunc TestChatModelTest(t *testing.T) {\n\tq := sqlc_queries.New(db)\n\th := NewChatModelHandler(q) // create a new ChatModelHandler instance for testing\n\trouter := mux.NewRouter()\n\th.Register(router)\n\t// delete all existing chat APIs\n\tclearChatModelsIfExists(q)\n\n\t// Now let's create our expected results. Create two results and insert them into the database using the queries.\n\tadmin, expectedResults := createTwoChatModel(q)\n\n\t// ensure that we get an array of two chat APIs in the response body\n\t// ensure the returned values are what we expect them to be\n\tresults := checkGetModels(t, router, expectedResults)\n\n\t// Now lets update the the first element of our expected results array and call PUT on the endpoint\n\n\t// Create an HTTP request so we can simulate a PUT with the payload\n\t// ensure the new values are returned and were also updated in the database\n\tfirstRecordID := results[0].ID\n\tupdateFirstRecord(t, router, firstRecordID, admin, expectedResults[0])\n\n\t// delete first model\n\tdeleteReq, _ := http.NewRequest(\"DELETE\", fmt.Sprintf(\"/chat_model/%d\", firstRecordID), nil)\n\tdeleteReq = deleteReq.WithContext(getContextWithUser(int(admin.ID)))\n\tdeleteRR := httptest.NewRecorder()\n\trouter.ServeHTTP(deleteRR, deleteReq)\n\tassert.Equal(t, deleteRR.Code, http.StatusOK)\n\n\t// check only one model left\n\treq, _ := http.NewRequest(\"GET\", \"/chat_model\", nil)\n\trr := httptest.NewRecorder()\n\trouter.ServeHTTP(rr, req)\n\t// ensure that we get an array of one chat API in the response body\n\tresults = unmarshalResponseToChatModel(t, rr)\n\tassert.Equal(t, len(results), 1)\n\tassert.Equal(t, results[0].Name, \"Test API 1\")\n\n\t// delete the last model\n\tdeleteRequest, _ := http.NewRequest(\"DELETE\", fmt.Sprintf(\"/chat_model/%d\", results[0].ID), nil)\n\tcontextWithUser := getContextWithUser(int(admin.ID))\n\tdeleteRequest = deleteRequest.WithContext(contextWithUser)\n\tdeleteResponseRecorder := httptest.NewRecorder()\n\trouter.ServeHTTP(deleteResponseRecorder, deleteRequest)\n\tassert.Equal(t, deleteResponseRecorder.Code, http.StatusOK)\n\n\t// check no models left\n\tgetRequest, _ := http.NewRequest(\"GET\", \"/chat_model\", nil)\n\t// Create a ResponseRecorder to record the response\n\tgetResponseRecorder := httptest.NewRecorder()\n\trouter.ServeHTTP(getResponseRecorder, getRequest)\n\tresults = unmarshalResponseToChatModel(t, getResponseRecorder)\n\tassert.Equal(t, len(results), 0)\n}\n\nfunc checkGetModels(t *testing.T, router *mux.Router, expectedResults []sqlc_queries.ChatModel) []sqlc_queries.ChatModel {\n\treq, _ := http.NewRequest(\"GET\", \"/chat_model\", nil)\n\trr := httptest.NewRecorder()\n\trouter.ServeHTTP(rr, req)\n\n\tassert.Equal(t, rr.Code, http.StatusOK)\n\tvar results []sqlc_queries.ChatModel\n\terr := json.NewDecoder(rr.Body).Decode(&results)\n\tif err != nil {\n\t\tt.Errorf(\"error parsing response body: %s\", err.Error())\n\t}\n\tassert.Equal(t, len(results), 2)\n\tassert.DeepEqual(t, lo.Reverse(expectedResults), results, cmpopts.IgnoreFields(sqlc_queries.ChatModel{}, \"ID\", \"IsEnable\"))\n\treturn results\n}\n\nfunc updateFirstRecord(t *testing.T, router *mux.Router, chatModelID int32, admin sqlc_queries.AuthUser, rec sqlc_queries.ChatModel) {\n\trec.Name = \"Test API 1 Updated\"\n\trec.Label = \"Test Label 1 Updated\"\n\n\tupdateBytes, err := json.Marshal(rec)\n\tif err != nil {\n\t\tt.Errorf(\"Error marshaling update payload: %s\", err.Error())\n\t}\n\n\tupdateReq, _ := http.NewRequest(\"PUT\", fmt.Sprintf(\"/chat_model/%d\", chatModelID), bytes.NewBuffer(updateBytes))\n\tupdateReq = updateReq.WithContext(getContextWithUser(int(admin.ID)))\n\n\tupdateRR := httptest.NewRecorder()\n\n\trouter.ServeHTTP(updateRR, updateReq)\n\n\tassert.Equal(t, updateRR.Code, http.StatusOK)\n\n\tvar updatedResult sqlc_queries.ChatModel\n\terr = json.Unmarshal(updateRR.Body.Bytes(), &updatedResult)\n\tif err != nil {\n\t\tt.Errorf(\"Error parsing response body: %s\", err.Error())\n\t}\n\n\tassert.Equal(t, rec.Name, updatedResult.Name)\n\tassert.Equal(t, rec.Label, updatedResult.Label)\n}\n" }, { "path": "api/chat_model_privilege_handler.go", "content": "package main\n\nimport (\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"log\"\n\t\"net/http\"\n\t\"strconv\"\n\n\t\"github.com/gorilla/mux\"\n\t\"github.com/samber/lo\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// UserChatModelPrivilegeHandler handles requests related to user chat model privileges\ntype UserChatModelPrivilegeHandler struct {\n\tdb *sqlc_queries.Queries\n}\n\n// NewUserChatModelPrivilegeHandler creates a new handler instance\nfunc NewUserChatModelPrivilegeHandler(db *sqlc_queries.Queries) *UserChatModelPrivilegeHandler {\n\treturn &UserChatModelPrivilegeHandler{\n\t\tdb: db,\n\t}\n}\n\n// Register sets up the handler routes\nfunc (h *UserChatModelPrivilegeHandler) Register(r *mux.Router) {\n\tr.HandleFunc(\"/admin/user_chat_model_privilege\", h.ListUserChatModelPrivileges).Methods(http.MethodGet)\n\tr.HandleFunc(\"/admin/user_chat_model_privilege\", h.CreateUserChatModelPrivilege).Methods(http.MethodPost)\n\tr.HandleFunc(\"/admin/user_chat_model_privilege/{id}\", h.DeleteUserChatModelPrivilege).Methods(http.MethodDelete)\n\tr.HandleFunc(\"/admin/user_chat_model_privilege/{id}\", h.UpdateUserChatModelPrivilege).Methods(http.MethodPut)\n}\n\ntype ChatModelPrivilege struct {\n\tID int32 `json:\"id\"`\n\tFullName string `json:\"fullName\"`\n\tUserEmail string `json:\"userEmail\"`\n\tChatModelName string `json:\"chatModelName\"`\n\tRateLimit int32 `json:\"rateLimit\"`\n}\n\n// ListUserChatModelPrivileges handles GET requests to list all user chat model privileges\nfunc (h *UserChatModelPrivilegeHandler) ListUserChatModelPrivileges(w http.ResponseWriter, r *http.Request) {\n\t// TODO: check user is super_user\n\tuserChatModelRows, err := h.db.ListUserChatModelPrivilegesRateLimit(r.Context())\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to list user chat model privileges\"))\n\t\treturn\n\t}\n\n\tlog.Printf(\"Listing user chat model privileges\")\n\toutput := lo.Map(userChatModelRows, func(r sqlc_queries.ListUserChatModelPrivilegesRateLimitRow, idx int) ChatModelPrivilege {\n\t\treturn ChatModelPrivilege{\n\t\t\tID: r.ID,\n\t\t\tFullName: r.FullName,\n\t\t\tUserEmail: r.UserEmail,\n\t\t\tChatModelName: r.ChatModelName,\n\t\t\tRateLimit: r.RateLimit,\n\t\t}\n\t})\n\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(output)\n}\n\nfunc (h *UserChatModelPrivilegeHandler) UserChatModelPrivilegeByID(w http.ResponseWriter, r *http.Request) {\n\tvars := mux.Vars(r)\n\tid, err := strconv.Atoi(vars[\"id\"])\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"invalid user chat model privilege ID\"))\n\t\treturn\n\t}\n\n\tuserChatModelPrivilege, err := h.db.UserChatModelPrivilegeByID(r.Context(), int32(id))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to get user chat model privilege\"))\n\t\treturn\n\t}\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(userChatModelPrivilege)\n}\n\n// CreateUserChatModelPrivilege handles POST requests to create a new user chat model privilege\nfunc (h *UserChatModelPrivilegeHandler) CreateUserChatModelPrivilege(w http.ResponseWriter, r *http.Request) {\n\tvar input ChatModelPrivilege\n\terr := json.NewDecoder(r.Body).Decode(&input)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"failed to parse request body\"))\n\t\treturn\n\t}\n\n\t// Validate input\n\tif input.UserEmail == \"\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"user email is required\"))\n\t\treturn\n\t}\n\tif input.ChatModelName == \"\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"chat model name is required\"))\n\t\treturn\n\t}\n\tif input.RateLimit <= 0 {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"rate limit must be positive\").WithMessage(\n\t\t\tfmt.Sprintf(\"invalid rate limit: %d\", input.RateLimit)))\n\t\treturn\n\t}\n\n\tlog.Printf(\"Creating chat model privilege for user %s with model %s\",\n\t\tinput.UserEmail, input.ChatModelName)\n\n\tuser, err := h.db.GetAuthUserByEmail(r.Context(), input.UserEmail)\n\tif err != nil {\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\tRespondWithAPIError(w, ErrResourceNotFound(\"user\").WithMessage(\n\t\t\t\tfmt.Sprintf(\"user with email %s not found\", input.UserEmail)))\n\t\t} else {\n\t\t\tRespondWithAPIError(w, WrapError(err, \"failed to get user by email\"))\n\t\t}\n\t\treturn\n\t}\n\n\tchatModel, err := h.db.ChatModelByName(r.Context(), input.ChatModelName)\n\tif err != nil {\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\tRespondWithAPIError(w, ErrChatModelNotFound.WithMessage(fmt.Sprintf(\"chat model %s not found\", input.ChatModelName)))\n\t\t} else {\n\t\t\tRespondWithAPIError(w, WrapError(err, \"failed to get chat model\"))\n\t\t}\n\t\treturn\n\t}\n\n\tuserChatModelPrivilege, err := h.db.CreateUserChatModelPrivilege(r.Context(), sqlc_queries.CreateUserChatModelPrivilegeParams{\n\t\tUserID: user.ID,\n\t\tChatModelID: chatModel.ID,\n\t\tRateLimit: input.RateLimit,\n\t\tCreatedBy: user.ID,\n\t\tUpdatedBy: user.ID,\n\t})\n\n\tif err != nil {\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\tRespondWithAPIError(w, ErrResourceNotFound(\"chat model privilege\"))\n\t\t} else {\n\t\t\tRespondWithAPIError(w, WrapError(err, \"failed to create user chat model privilege\"))\n\t\t}\n\t\treturn\n\t}\n\n\toutput := ChatModelPrivilege{\n\t\tID: userChatModelPrivilege.ID,\n\t\tUserEmail: user.Email,\n\t\tChatModelName: chatModel.Name,\n\t\tRateLimit: userChatModelPrivilege.RateLimit,\n\t}\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(output)\n}\n\n// UpdateUserChatModelPrivilege handles PUT requests to update a user chat model privilege\nfunc (h *UserChatModelPrivilegeHandler) UpdateUserChatModelPrivilege(w http.ResponseWriter, r *http.Request) {\n\tvars := mux.Vars(r)\n\tid, err := strconv.Atoi(vars[\"id\"])\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"invalid user chat model privilege ID\"))\n\t\treturn\n\t}\n\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"missing or invalid user ID\"))\n\t\treturn\n\t}\n\n\tvar input ChatModelPrivilege\n\terr = json.NewDecoder(r.Body).Decode(&input)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"failed to parse request body\"))\n\t\treturn\n\t}\n\n\t// Validate input\n\tif input.RateLimit <= 0 {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"rate limit must be positive\"))\n\t\treturn\n\t}\n\n\tlog.Printf(\"Updating chat model privilege %d for user %d\", id, userID)\n\n\tuserChatModelPrivilege, err := h.db.UpdateUserChatModelPrivilege(r.Context(), sqlc_queries.UpdateUserChatModelPrivilegeParams{\n\t\tID: int32(id),\n\t\tRateLimit: input.RateLimit,\n\t\tUpdatedBy: userID,\n\t})\n\n\tif err != nil {\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\tRespondWithAPIError(w, ErrResourceNotFound(\"chat model privilege\"))\n\t\t} else {\n\t\t\tRespondWithAPIError(w, WrapError(err, \"failed to update user chat model privilege\"))\n\t\t}\n\t\treturn\n\t}\n\toutput := ChatModelPrivilege{\n\t\tID: userChatModelPrivilege.ID,\n\t\tUserEmail: input.UserEmail,\n\t\tChatModelName: input.ChatModelName,\n\t\tRateLimit: userChatModelPrivilege.RateLimit,\n\t}\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(output)\n}\n\nfunc (h *UserChatModelPrivilegeHandler) DeleteUserChatModelPrivilege(w http.ResponseWriter, r *http.Request) {\n\tvars := mux.Vars(r)\n\tid, err := strconv.Atoi(vars[\"id\"])\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"invalid user chat model privilege ID\"))\n\t\treturn\n\t}\n\n\terr = h.db.DeleteUserChatModelPrivilege(r.Context(), int32(id))\n\tif err != nil {\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\tRespondWithAPIError(w, ErrResourceNotFound(\"chat model privilege\"))\n\t\t} else {\n\t\t\tRespondWithAPIError(w, WrapError(err, \"failed to delete user chat model privilege\"))\n\t\t}\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusNoContent)\n}\n\nfunc (h *UserChatModelPrivilegeHandler) UserChatModelPrivilegeByUserAndModelID(w http.ResponseWriter, r *http.Request) {\n\t_, err := getUserID(r.Context())\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"missing or invalid user ID\"))\n\t\treturn\n\t}\n\n\tvar input struct {\n\t\tUserID int32\n\t\tChatModelID int32\n\t}\n\terr = json.NewDecoder(r.Body).Decode(&input)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"failed to parse request body\"))\n\t\treturn\n\t}\n\n\tuserChatModelPrivilege, err := h.db.UserChatModelPrivilegeByUserAndModelID(r.Context(),\n\t\tsqlc_queries.UserChatModelPrivilegeByUserAndModelIDParams{\n\t\t\tUserID: input.UserID,\n\t\t\tChatModelID: input.ChatModelID,\n\t\t})\n\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to get user chat model privilege\"))\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(userChatModelPrivilege)\n}\n\nfunc (h *UserChatModelPrivilegeHandler) ListUserChatModelPrivilegesByUserID(w http.ResponseWriter, r *http.Request) {\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"missing or invalid user ID\"))\n\t\treturn\n\t}\n\n\tprivileges, err := h.db.ListUserChatModelPrivilegesByUserID(r.Context(), int32(userID))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to list privileges for user\"))\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(privileges)\n}\n" }, { "path": "api/chat_prompt_hander.go", "content": "package main\n\nimport (\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"net/http\"\n\t\"strconv\"\n\n\t\"github.com/gorilla/mux\"\n\t\"github.com/jackc/pgconn\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype ChatPromptHandler struct {\n\tservice *ChatPromptService\n}\n\nfunc NewChatPromptHandler(sqlc_q *sqlc_queries.Queries) *ChatPromptHandler {\n\tpromptService := NewChatPromptService(sqlc_q)\n\treturn &ChatPromptHandler{\n\t\tservice: promptService,\n\t}\n}\n\nfunc (h *ChatPromptHandler) Register(router *mux.Router) {\n\trouter.HandleFunc(\"/chat_prompts\", h.CreateChatPrompt).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/chat_prompts/users\", h.GetChatPromptsByUserID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/chat_prompts/{id}\", h.GetChatPromptByID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/chat_prompts/{id}\", h.UpdateChatPrompt).Methods(http.MethodPut)\n\trouter.HandleFunc(\"/chat_prompts/{id}\", h.DeleteChatPrompt).Methods(http.MethodDelete)\n\trouter.HandleFunc(\"/chat_prompts\", h.GetAllChatPrompts).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/uuid/chat_prompts/{uuid}\", h.DeleteChatPromptByUUID).Methods(http.MethodDelete)\n\trouter.HandleFunc(\"/uuid/chat_prompts/{uuid}\", h.UpdateChatPromptByUUID).Methods(http.MethodPut)\n}\n\nfunc (h *ChatPromptHandler) CreateChatPrompt(w http.ResponseWriter, r *http.Request) {\n\tvar promptParams sqlc_queries.CreateChatPromptParams\n\terr := json.NewDecoder(r.Body).Decode(&promptParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Always trust authenticated user identity over client-provided values.\n\tpromptParams.UserID = userID\n\tpromptParams.CreatedBy = userID\n\tpromptParams.UpdatedBy = userID\n\n\t// Idempotent creation for session system prompt:\n\t// return existing prompt instead of inserting duplicates when concurrent\n\t// frontend/backend requests race on a fresh session.\n\tif promptParams.ChatSessionUuid != \"\" && promptParams.Role == \"system\" {\n\t\texistingPrompt, getErr := h.service.q.GetOneChatPromptBySessionUUID(r.Context(), promptParams.ChatSessionUuid)\n\t\tif getErr == nil {\n\t\t\tjson.NewEncoder(w).Encode(existingPrompt)\n\t\t\treturn\n\t\t}\n\t\tif !errors.Is(getErr, sql.ErrNoRows) {\n\t\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(getErr), \"Failed to check existing chat prompt\"))\n\t\t\treturn\n\t\t}\n\t}\n\n\tprompt, err := h.service.CreateChatPrompt(r.Context(), promptParams)\n\tif err != nil {\n\t\t// Handle race: another request inserted the same session system prompt\n\t\t// between our read check and insert attempt.\n\t\tvar pgErr *pgconn.PgError\n\t\tif promptParams.ChatSessionUuid != \"\" && promptParams.Role == \"system\" &&\n\t\t\terrors.As(err, &pgErr) && pgErr.Code == \"23505\" {\n\t\t\texistingPrompt, getErr := h.service.q.GetOneChatPromptBySessionUUID(r.Context(), promptParams.ChatSessionUuid)\n\t\t\tif getErr == nil {\n\t\t\t\tjson.NewEncoder(w).Encode(existingPrompt)\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to create chat prompt\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(prompt)\n}\n\nfunc (h *ChatPromptHandler) GetChatPromptByID(w http.ResponseWriter, r *http.Request) {\n\tidStr := mux.Vars(r)[\"id\"]\n\tid, err := strconv.Atoi(idStr)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"invalid chat prompt ID\"))\n\t\treturn\n\t}\n\tprompt, err := h.service.GetChatPromptByID(r.Context(), int32(id))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get chat prompt\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(prompt)\n}\n\nfunc (h *ChatPromptHandler) UpdateChatPrompt(w http.ResponseWriter, r *http.Request) {\n\tidStr := mux.Vars(r)[\"id\"]\n\tid, err := strconv.Atoi(idStr)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"invalid chat prompt ID\"))\n\t\treturn\n\t}\n\tvar promptParams sqlc_queries.UpdateChatPromptParams\n\terr = json.NewDecoder(r.Body).Decode(&promptParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tpromptParams.ID = int32(id)\n\tprompt, err := h.service.UpdateChatPrompt(r.Context(), promptParams)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to update chat prompt\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(prompt)\n}\n\nfunc (h *ChatPromptHandler) DeleteChatPrompt(w http.ResponseWriter, r *http.Request) {\n\tidStr := mux.Vars(r)[\"id\"]\n\tid, err := strconv.Atoi(idStr)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"invalid chat prompt ID\"))\n\t\treturn\n\t}\n\terr = h.service.DeleteChatPrompt(r.Context(), int32(id))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to delete chat prompt\"))\n\t\treturn\n\t}\n\tw.WriteHeader(http.StatusOK)\n}\n\nfunc (h *ChatPromptHandler) GetAllChatPrompts(w http.ResponseWriter, r *http.Request) {\n\tprompts, err := h.service.GetAllChatPrompts(r.Context())\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get chat prompts\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(prompts)\n}\n\nfunc (h *ChatPromptHandler) GetChatPromptsByUserID(w http.ResponseWriter, r *http.Request) {\n\tidStr := mux.Vars(r)[\"id\"]\n\tid, err := strconv.Atoi(idStr)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"invalid user ID\"))\n\t\treturn\n\t}\n\tprompts, err := h.service.GetChatPromptsByUserID(r.Context(), int32(id))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to get chat prompts by user\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(prompts)\n}\n\nfunc (h *ChatPromptHandler) DeleteChatPromptByUUID(w http.ResponseWriter, r *http.Request) {\n\tidStr := mux.Vars(r)[\"uuid\"]\n\terr := h.service.DeleteChatPromptByUUID(r.Context(), idStr)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to delete chat prompt\"))\n\t\treturn\n\t}\n\tw.WriteHeader(http.StatusOK)\n}\n\nfunc (h *ChatPromptHandler) UpdateChatPromptByUUID(w http.ResponseWriter, r *http.Request) {\n\tvar simple_msg SimpleChatMessage\n\terr := json.NewDecoder(r.Body).Decode(&simple_msg)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tprompt, err := h.service.UpdateChatPromptByUUID(r.Context(), simple_msg.Uuid, simple_msg.Text)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to update chat prompt\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(prompt)\n}\n" }, { "path": "api/chat_prompt_service.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"errors\"\n\n\t\"github.com/rotisserie/eris\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype ChatPromptService struct {\n\tq *sqlc_queries.Queries\n}\n\n// NewChatPromptService creates a new ChatPromptService.\nfunc NewChatPromptService(q *sqlc_queries.Queries) *ChatPromptService {\n\treturn &ChatPromptService{q: q}\n}\n\n// CreateChatPrompt creates a new chat prompt.\nfunc (s *ChatPromptService) CreateChatPrompt(ctx context.Context, prompt_params sqlc_queries.CreateChatPromptParams) (sqlc_queries.ChatPrompt, error) {\n\tprompt, err := s.q.CreateChatPrompt(ctx, prompt_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatPrompt{}, eris.Wrap(err, \"failed to create prompt: \")\n\t}\n\treturn prompt, nil\n}\n\nfunc (s *ChatPromptService) CreateChatPromptWithUUID(ctx context.Context, uuid string, role, content string) (sqlc_queries.ChatPrompt, error) {\n\tparams := sqlc_queries.CreateChatPromptParams{\n\t\tChatSessionUuid: uuid,\n\t\tRole: role,\n\t\tContent: content,\n\t}\n\tprompt, err := s.q.CreateChatPrompt(ctx, params)\n\treturn prompt, err\n}\n\n// GetChatPromptByID returns a chat prompt by ID.\nfunc (s *ChatPromptService) GetChatPromptByID(ctx context.Context, id int32) (sqlc_queries.ChatPrompt, error) {\n\tprompt, err := s.q.GetChatPromptByID(ctx, id)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatPrompt{}, eris.Wrap(err, \"failed to create prompt: \")\n\t}\n\treturn prompt, nil\n}\n\n// UpdateChatPrompt updates an existing chat prompt.\nfunc (s *ChatPromptService) UpdateChatPrompt(ctx context.Context, prompt_params sqlc_queries.UpdateChatPromptParams) (sqlc_queries.ChatPrompt, error) {\n\tprompt_u, err := s.q.UpdateChatPrompt(ctx, prompt_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatPrompt{}, errors.New(\"failed to update prompt\")\n\t}\n\treturn prompt_u, nil\n}\n\n// DeleteChatPrompt deletes a chat prompt by ID.\nfunc (s *ChatPromptService) DeleteChatPrompt(ctx context.Context, id int32) error {\n\terr := s.q.DeleteChatPrompt(ctx, id)\n\tif err != nil {\n\t\treturn errors.New(\"failed to delete prompt\")\n\t}\n\treturn nil\n}\n\n// GetAllChatPrompts returns all chat prompts.\nfunc (s *ChatPromptService) GetAllChatPrompts(ctx context.Context) ([]sqlc_queries.ChatPrompt, error) {\n\tprompts, err := s.q.GetAllChatPrompts(ctx)\n\tif err != nil {\n\t\treturn nil, errors.New(\"failed to retrieve prompts\")\n\t}\n\treturn prompts, nil\n}\n\nfunc (s *ChatPromptService) GetChatPromptsByUserID(ctx context.Context, userID int32) ([]sqlc_queries.ChatPrompt, error) {\n\tprompts, err := s.q.GetChatPromptsByUserID(ctx, userID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn prompts, nil\n}\n\nfunc (s *ChatPromptService) GetChatPromptsBySessionUUID(ctx context.Context, session_uuid string) ([]sqlc_queries.ChatPrompt, error) {\n\tprompts, err := s.q.GetChatPromptsBySessionUUID(ctx, session_uuid)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn prompts, nil\n}\n\n// DeleteChatPromptByUUID\nfunc (s *ChatPromptService) DeleteChatPromptByUUID(ctx context.Context, uuid string) error {\n\terr := s.q.DeleteChatPromptByUUID(ctx, uuid)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\n// UpdateChatPromptByUUID\nfunc (s *ChatPromptService) UpdateChatPromptByUUID(ctx context.Context, uuid string, content string) (sqlc_queries.ChatPrompt, error) {\n\ttokenCount, _ := getTokenCount(content)\n\tparams := sqlc_queries.UpdateChatPromptByUUIDParams{\n\t\tUuid: uuid,\n\t\tContent: content,\n\t\tTokenCount: int32(tokenCount),\n\t}\n\tprompt, err := s.q.UpdateChatPromptByUUID(ctx, params)\n\treturn prompt, err\n}\n" }, { "path": "api/chat_prompt_service_test.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t_ \"github.com/lib/pq\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\nfunc TestChatPromptService(t *testing.T) {\n\t// Create a new ChatPromptService with the test database connection\n\tq := sqlc_queries.New(db)\n\tservice := NewChatPromptService(q)\n\n\t// Insert a new chat prompt into the database\n\tprompt_params := sqlc_queries.CreateChatPromptParams{ChatSessionUuid: \"Test Topic\", Role: \"Test Role\", Content: \"Test Content\", UserID: 1}\n\tprompt, err := service.CreateChatPrompt(context.Background(), prompt_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat prompt: %v\", err)\n\t}\n\n\t// Retrieve the chat prompt by ID and check that it matches the expected values\n\tretrievedPrompt, err := service.GetChatPromptByID(context.Background(), prompt.ID)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to get chat prompt: %v\", err)\n\t}\n\tif retrievedPrompt.ChatSessionUuid != prompt.ChatSessionUuid || retrievedPrompt.Role != prompt.Role || retrievedPrompt.Content != prompt.Content || retrievedPrompt.Score != prompt.Score || retrievedPrompt.UserID != prompt.UserID {\n\t\tt.Error(\"retrieved chat prompt does not match expected values\")\n\t}\n\n\t// Update the chat prompt and check that it was updated in the database\n\tupdated_params := sqlc_queries.UpdateChatPromptParams{ID: prompt.ID, ChatSessionUuid: \"Updated Test Topic\", Role: \"Updated Test Role\", Content: \"Updated Test Content\", Score: 0.75}\n\tif _, err := service.UpdateChatPrompt(context.Background(), updated_params); err != nil {\n\t\tt.Fatalf(\"failed to update chat prompt: %v\", err)\n\t}\n\tretrievedPrompt, err = service.GetChatPromptByID(context.Background(), prompt.ID)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to get chat prompt: %v\", err)\n\t}\n\tif retrievedPrompt.ChatSessionUuid != updated_params.ChatSessionUuid || retrievedPrompt.Role != updated_params.Role || retrievedPrompt.Content != updated_params.Content || retrievedPrompt.Score != updated_params.Score {\n\t\tt.Error(\"retrieved chat prompt does not match expected values\")\n\t}\n\n\t// Delete the chat prompt and check that it was deleted from the database\n\tif err := service.DeleteChatPrompt(context.Background(), prompt.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat prompt: %v\", err)\n\t}\n\t_, err = service.GetChatPromptByID(context.Background(), prompt.ID)\n\tif err != nil && errors.Is(err, sql.ErrNoRows) {\n\t\tprint(\"Chat prompt deleted successfully\")\n\t}\n\tif err == nil || !errors.Is(err, sql.ErrNoRows) {\n\t\tt.Error(\"expected error due to missing chat prompt, but got no error or different error\")\n\t}\n\n\t_, err = service.q.GetChatPromptsBySessionUUID(context.Background(), \"12324\")\n\n\tif err != nil {\n\t\tt.Error(\"expected error due to missing chat prompt, but got no error or different error\")\n\t}\n}\n\nfunc TestGetAllChatPrompts(t *testing.T) {\n\tq := sqlc_queries.New(db)\n\tservice := NewChatPromptService(q)\n\n\t// Insert two chat prompts into the database\n\tprompt1_params := sqlc_queries.CreateChatPromptParams{ChatSessionUuid: \"Test Topic 1\", Role: \"Test Role 1\", Content: \"Test Content 1\", UserID: 1}\n\tprompt1, err := service.CreateChatPrompt(context.Background(), prompt1_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat prompt: %v\", err)\n\t}\n\tprompt2_params := sqlc_queries.CreateChatPromptParams{ChatSessionUuid: \"Test Topic 2\", Role: \"Test Role 2\", Content: \"Test Content 2\", UserID: 2}\n\tprompt2, err := service.CreateChatPrompt(context.Background(), prompt2_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat prompt: %v\", err)\n\t}\n\n\t// Retrieve all chat prompts and check that they match the expected values\n\tprompts, err := service.GetAllChatPrompts(context.Background())\n\tif err != nil {\n\t\tt.Fatalf(\"failed to retrieve chat prompts: %v\", err)\n\t}\n\tif len(prompts) != 2 {\n\t\tt.Errorf(\"expected 2 chat prompts, but got %d\", len(prompts))\n\t}\n\tif prompts[0].ChatSessionUuid != prompt1.ChatSessionUuid || prompts[0].Role != prompt1.Role || prompts[0].Content != prompt1.Content || prompts[0].Score != prompt1.Score || prompts[0].UserID != prompt1.UserID ||\n\t\tprompts[1].ChatSessionUuid != prompt2.ChatSessionUuid || prompts[1].Role != prompt2.Role || prompts[1].Content != prompt2.Content || prompts[1].Score != prompt2.Score || prompts[1].UserID != prompt2.UserID {\n\t\tt.Error(\"retrieved chat prompts do not match expected values\")\n\t}\n\n\t// Delete the chat prompt and check that it was deleted from the database\n\tif err := service.DeleteChatPrompt(context.Background(), prompt1.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat prompt: %v\", err)\n\t}\n\n\tif err := service.DeleteChatPrompt(context.Background(), prompt2.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat prompt: %v\", err)\n\t}\n\n\tpromptsAfterDelete, _ := service.GetAllChatPrompts(context.Background())\n\tif len(promptsAfterDelete) != 0 {\n\t\tt.Error(\"retrieved chat prompts\")\n\t}\n\tfmt.Printf(\"%+v\", promptsAfterDelete)\n\n}\n\nfunc TestGetChatPromptsByTopic(t *testing.T) {\n\n\t// Create a new ChatPromptService with the test database connection\n\tq := sqlc_queries.New(db)\n\tservice := NewChatPromptService(q)\n\n\t// Insert two chat prompts into the database with different topics\n\tprompt1_params := sqlc_queries.CreateChatPromptParams{ChatSessionUuid: \"Test Topic 1\", Role: \"Test Role 1\", Content: \"Test Content 1\", UserID: 1}\n\tprompt1, err := service.CreateChatPrompt(context.Background(), prompt1_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat prompt: %v\", err)\n\t}\n\tprompt2_params := sqlc_queries.CreateChatPromptParams{ChatSessionUuid: \"Test Topic 2\", Role: \"Test Role 2\", Content: \"Test Content 2\", UserID: 2}\n\tprompt2, err := service.CreateChatPrompt(context.Background(), prompt2_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat prompt: %v\", err)\n\t}\n\n\t// Retrieve chat prompts by topic and check that they match the expected values\n\ttopic := \"Test Topic 1\"\n\tprompts, err := service.GetChatPromptsBySessionUUID(context.Background(), topic)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to retrieve chat prompts: %v\", err)\n\t}\n\tif len(prompts) != 1 {\n\t\tt.Errorf(\"expected 1 chat prompt, but got %d\", len(prompts))\n\t}\n\tif prompts[0].ChatSessionUuid != prompt1.ChatSessionUuid || prompts[0].Role != prompt1.Role || prompts[0].Content != prompt1.Content || prompts[0].Score != prompt1.Score || prompts[0].UserID != prompt1.UserID {\n\t\tt.Error(\"retrieved chat prompts do not match expected values\")\n\t}\n\n\t// Delete the chat prompt and check that it was deleted from the database\n\tif err := service.DeleteChatPrompt(context.Background(), prompt1.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat prompt: %v\", err)\n\t}\n\n\tif err := service.DeleteChatPrompt(context.Background(), prompt2.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat prompt: %v\", err)\n\t}\n\n\tpromptsAfterDelete, _ := service.GetAllChatPrompts(context.Background())\n\tif len(promptsAfterDelete) != 0 {\n\t\tt.Error(\"retrieved chat prompts\")\n\t}\n\tfmt.Printf(\"%+v\", promptsAfterDelete)\n\n}\n\nfunc TestGetChatPromptsByUserID(t *testing.T) {\n\t// Create a new ChatPromptService with the test database connection\n\tq := sqlc_queries.New(db)\n\tservice := NewChatPromptService(q)\n\n\t// Insert two chat prompts into the database with different user IDs\n\tprompt1_params := sqlc_queries.CreateChatPromptParams{ChatSessionUuid: \"Test Topic 1\", Role: \"Test Role 1\", Content: \"Test Content 1\", UserID: 1}\n\tprompt1, err := service.CreateChatPrompt(context.Background(), prompt1_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat prompt: %v\", err)\n\t}\n\tprompt2_params := sqlc_queries.CreateChatPromptParams{ChatSessionUuid: \"Test Topic 2\", Role: \"Test Role 2\", Content: \"Test Content 2\", UserID: 2}\n\tprompt2, err := service.CreateChatPrompt(context.Background(), prompt2_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat prompt: %v\", err)\n\t}\n\n\t// Retrieve chat prompts by user ID and check that they match the expected values\n\tuserID := int32(1)\n\tprompts, err := service.GetChatPromptsByUserID(context.Background(), userID)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to retrieve chat prompts: %v\", err)\n\t}\n\tif len(prompts) != 1 {\n\t\tt.Errorf(\"expected 1 chat prompt, but got %d\", len(prompts))\n\t}\n\tif prompts[0].ChatSessionUuid != prompt1.ChatSessionUuid || prompts[0].Role != prompt1.Role || prompts[0].Content != prompt1.Content || prompts[0].Score != prompt1.Score || prompts[0].UserID != prompt1.UserID {\n\t\tt.Error(\"retrieved chat prompts do not match expected values\")\n\t}\n\n\t// Delete the chat prompt and check that it was deleted from the database\n\tif err := service.DeleteChatPrompt(context.Background(), prompt1.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat prompt: %v\", err)\n\t}\n\n\tif err := service.DeleteChatPrompt(context.Background(), prompt2.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat prompt: %v\", err)\n\t}\n\n\tpromptsAfterDelete, _ := service.GetAllChatPrompts(context.Background())\n\tif len(promptsAfterDelete) != 0 {\n\t\tt.Error(\"retrieved chat prompts\")\n\t}\n\tfmt.Printf(\"%+v\", promptsAfterDelete)\n}\n" }, { "path": "api/chat_session_handler.go", "content": "package main\n\nimport (\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"net/http\"\n\t\"strconv\"\n\n\t\"github.com/google/uuid\"\n\n\t\"github.com/gorilla/mux\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype ChatSessionHandler struct {\n\tservice *ChatSessionService\n}\n\nfunc NewChatSessionHandler(sqlc_q *sqlc_queries.Queries) *ChatSessionHandler {\n\t// create a new ChatSessionService instance\n\tchatSessionService := NewChatSessionService(sqlc_q)\n\treturn &ChatSessionHandler{\n\t\tservice: chatSessionService,\n\t}\n}\n\nfunc (h *ChatSessionHandler) Register(router *mux.Router) {\n\trouter.HandleFunc(\"/chat_sessions/user\", h.getSimpleChatSessionsByUserID).Methods(http.MethodGet)\n\n\trouter.HandleFunc(\"/uuid/chat_sessions/max_length/{uuid}\", h.updateSessionMaxLength).Methods(\"PUT\")\n\trouter.HandleFunc(\"/uuid/chat_sessions/topic/{uuid}\", h.updateChatSessionTopicByUUID).Methods(\"PUT\")\n\trouter.HandleFunc(\"/uuid/chat_sessions/{uuid}\", h.getChatSessionByUUID).Methods(\"GET\")\n\trouter.HandleFunc(\"/uuid/chat_sessions/{uuid}\", h.createOrUpdateChatSessionByUUID).Methods(\"PUT\")\n\trouter.HandleFunc(\"/uuid/chat_sessions/{uuid}\", h.deleteChatSessionByUUID).Methods(\"DELETE\")\n\trouter.HandleFunc(\"/uuid/chat_sessions\", h.createChatSessionByUUID).Methods(\"POST\")\n\trouter.HandleFunc(\"/uuid/chat_session_from_snapshot/{uuid}\", h.createChatSessionFromSnapshot).Methods(http.MethodPost)\n}\n\n// getChatSessionByUUID returns a chat session by its UUID\nfunc (h *ChatSessionHandler) getChatSessionByUUID(w http.ResponseWriter, r *http.Request) {\n\tuuid := mux.Vars(r)[\"uuid\"]\n\tsession, err := h.service.GetChatSessionByUUID(r.Context(), uuid)\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\tapiErr := ErrResourceNotFound(\"Chat session\")\n\t\t\tapiErr.Message = \"Session not found with UUID: \" + uuid\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t} else {\n\t\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to get chat session\")\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\t}\n\n\tsession_resp := &ChatSessionResponse{\n\t\tUuid: session.Uuid,\n\t\tTopic: session.Topic,\n\t\tMaxLength: session.MaxLength,\n\t\tCreatedAt: session.CreatedAt,\n\t\tUpdatedAt: session.UpdatedAt,\n\t\tArtifactEnabled: session.ArtifactEnabled,\n\t}\n\tjson.NewEncoder(w).Encode(session_resp)\n}\n\n// createChatSessionByUUID creates a chat session by its UUID (idempotent)\nfunc (h *ChatSessionHandler) createChatSessionByUUID(w http.ResponseWriter, r *http.Request) {\n\tvar req struct {\n\t\tUuid string `json:\"uuid\"`\n\t\tTopic string `json:\"topic\"`\n\t\tModel string `json:\"model\"`\n\t\tDefaultSystemPrompt string `json:\"defaultSystemPrompt\"`\n\t}\n\terr := json.NewDecoder(r.Body).Decode(&req)\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid request format\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tctx := r.Context()\n\tuserIDInt, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Get or create default workspace for the user\n\tworkspaceService := NewChatWorkspaceService(h.service.q)\n\tdefaultWorkspace, err := workspaceService.EnsureDefaultWorkspaceExists(ctx, userIDInt)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to ensure default workspace exists\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Use CreateOrUpdateChatSessionByUUID for idempotent session creation\n\tcreateOrUpdateParams := sqlc_queries.CreateOrUpdateChatSessionByUUIDParams{\n\t\tUuid: req.Uuid,\n\t\tUserID: userIDInt,\n\t\tTopic: req.Topic,\n\t\tMaxLength: DefaultMaxLength,\n\t\tTemperature: DefaultTemperature,\n\t\tModel: req.Model,\n\t\tMaxTokens: DefaultMaxTokens,\n\t\tTopP: DefaultTopP,\n\t\tN: DefaultN,\n\t\tDebug: false,\n\t\tSummarizeMode: false,\n\t\tExploreMode: false,\n\t\tArtifactEnabled: false,\n\t\tWorkspaceID: sql.NullInt32{Int32: defaultWorkspace.ID, Valid: true},\n\t}\n\n\tsession, err := h.service.CreateOrUpdateChatSessionByUUID(r.Context(), createOrUpdateParams)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to create or update chat session\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t_, err = h.service.EnsureDefaultSystemPrompt(ctx, session.Uuid, userIDInt, req.DefaultSystemPrompt)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to create default system prompt\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// set active chat session when creating a new chat session (use unified approach)\n\t_, err = h.service.q.UpsertUserActiveSession(r.Context(),\n\t\tsqlc_queries.UpsertUserActiveSessionParams{\n\t\t\tUserID: session.UserID,\n\t\t\tWorkspaceID: sql.NullInt32{Valid: false},\n\t\t\tChatSessionUuid: session.Uuid,\n\t\t})\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to update or create active user session record\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(session)\n}\n\ntype UpdateChatSessionRequest struct {\n\tUuid string `json:\"uuid\"`\n\tTopic string `json:\"topic\"`\n\tMaxLength int32 `json:\"maxLength\"`\n\tTemperature float64 `json:\"temperature\"`\n\tModel string `json:\"model\"`\n\tTopP float64 `json:\"topP\"`\n\tN int32 `json:\"n\"`\n\tMaxTokens int32 `json:\"maxTokens\"`\n\tDebug bool `json:\"debug\"`\n\tSummarizeMode bool `json:\"summarizeMode\"`\n\tArtifactEnabled bool `json:\"artifactEnabled\"`\n\tExploreMode bool `json:\"exploreMode\"`\n\tWorkspaceUUID string `json:\"workspaceUuid,omitempty\"`\n}\n\n// UpdateChatSessionByUUID updates a chat session by its UUID\nfunc (h *ChatSessionHandler) createOrUpdateChatSessionByUUID(w http.ResponseWriter, r *http.Request) {\n\tvar sessionReq UpdateChatSessionRequest\n\terr := json.NewDecoder(r.Body).Decode(&sessionReq)\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid request format\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tif sessionReq.MaxLength == 0 {\n\t\tsessionReq.MaxLength = DefaultMaxLength\n\t}\n\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tvar sessionParams sqlc_queries.CreateOrUpdateChatSessionByUUIDParams\n\n\tsessionParams.MaxLength = sessionReq.MaxLength\n\tsessionParams.Topic = sessionReq.Topic\n\tsessionParams.Uuid = sessionReq.Uuid\n\tsessionParams.UserID = userID\n\tsessionParams.Temperature = sessionReq.Temperature\n\tsessionParams.Model = sessionReq.Model\n\tsessionParams.TopP = sessionReq.TopP\n\tsessionParams.N = sessionReq.N\n\tsessionParams.MaxTokens = sessionReq.MaxTokens\n\tsessionParams.Debug = sessionReq.Debug\n\tsessionParams.SummarizeMode = sessionReq.SummarizeMode\n\tsessionParams.ArtifactEnabled = sessionReq.ArtifactEnabled\n\tsessionParams.ExploreMode = sessionReq.ExploreMode\n\n\t// Handle workspace\n\tif sessionReq.WorkspaceUUID != \"\" {\n\t\tworkspaceService := NewChatWorkspaceService(h.service.q)\n\t\tworkspace, err := workspaceService.GetWorkspaceByUUID(ctx, sessionReq.WorkspaceUUID)\n\t\tif err != nil {\n\t\t\tapiErr := WrapError(MapDatabaseError(err), \"Invalid workspace UUID\")\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\t\tsessionParams.WorkspaceID = sql.NullInt32{Int32: workspace.ID, Valid: true}\n\t} else {\n\t\t// Ensure default workspace exists\n\t\tworkspaceService := NewChatWorkspaceService(h.service.q)\n\t\tdefaultWorkspace, err := workspaceService.EnsureDefaultWorkspaceExists(ctx, userID)\n\t\tif err != nil {\n\t\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to ensure default workspace exists\")\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\t\tsessionParams.WorkspaceID = sql.NullInt32{Int32: defaultWorkspace.ID, Valid: true}\n\t}\n\tsession, err := h.service.CreateOrUpdateChatSessionByUUID(r.Context(), sessionParams)\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to create or update chat session\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(session)\n}\n\n// deleteChatSessionByUUID deletes a chat session by its UUID\nfunc (h *ChatSessionHandler) deleteChatSessionByUUID(w http.ResponseWriter, r *http.Request) {\n\tuuid := mux.Vars(r)[\"uuid\"]\n\terr := h.service.DeleteChatSessionByUUID(r.Context(), uuid)\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to delete chat session\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tw.WriteHeader(http.StatusOK)\n}\n\n// getSimpleChatSessionsByUserID returns a list of simple chat sessions by user ID\nfunc (h *ChatSessionHandler) getSimpleChatSessionsByUserID(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\tidStr := ctx.Value(userContextKey).(string)\n\tid, err := strconv.Atoi(idStr)\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid user ID\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tsessions, err := h.service.GetSimpleChatSessionsByUserID(ctx, int32(id))\n\tif err != nil {\n\t\tapiErr := ErrResourceNotFound(\"Chat sessions\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(sessions)\n}\n\n// updateChatSessionTopicByUUID updates a chat session topic by its UUID\nfunc (h *ChatSessionHandler) updateChatSessionTopicByUUID(w http.ResponseWriter, r *http.Request) {\n\tuuid := mux.Vars(r)[\"uuid\"]\n\tvar sessionParams sqlc_queries.UpdateChatSessionTopicByUUIDParams\n\terr := json.NewDecoder(r.Body).Decode(&sessionParams)\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid request format\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tsessionParams.Uuid = uuid\n\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tsessionParams.UserID = userID\n\n\tsession, err := h.service.UpdateChatSessionTopicByUUID(r.Context(), sessionParams)\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to update chat session topic\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(session)\n}\n\n// updateSessionMaxLength\nfunc (h *ChatSessionHandler) updateSessionMaxLength(w http.ResponseWriter, r *http.Request) {\n\tuuid := mux.Vars(r)[\"uuid\"]\n\tvar sessionParams sqlc_queries.UpdateSessionMaxLengthParams\n\terr := json.NewDecoder(r.Body).Decode(&sessionParams)\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid request format\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tsessionParams.Uuid = uuid\n\n\tsession, err := h.service.UpdateSessionMaxLength(r.Context(), sessionParams)\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to update session max length\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(session)\n}\n\n// CreateChatSessionFromSnapshot ($uuid)\n// create a new session with title of snapshot,\n// create a prompt with the first message of snapshot\n// create messages based on the rest of messages.\n// return the new session uuid\n\nfunc (h *ChatSessionHandler) createChatSessionFromSnapshot(w http.ResponseWriter, r *http.Request) {\n\tvars := mux.Vars(r)\n\tsnapshot_uuid := vars[\"uuid\"]\n\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tsnapshot, err := h.service.q.ChatSnapshotByUUID(r.Context(), snapshot_uuid)\n\tif err != nil {\n\t\tapiErr := ErrResourceNotFound(\"Chat snapshot\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tsessionTitle := snapshot.Title\n\tconversions := snapshot.Conversation\n\tvar conversionsSimpleMessages []SimpleChatMessage\n\tjson.Unmarshal(conversions, &conversionsSimpleMessages)\n\tpromptMsg := conversionsSimpleMessages[0]\n\tchatPrompt, err := h.service.q.GetChatPromptByUUID(r.Context(), promptMsg.Uuid)\n\tif err != nil {\n\t\tapiErr := ErrResourceNotFound(\"Chat prompt\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\toriginSession, err := h.service.q.GetChatSessionByUUIDWithInActive(r.Context(), chatPrompt.ChatSessionUuid)\n\tif err != nil {\n\t\tapiErr := ErrResourceNotFound(\"Original chat session\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tsessionUUID := uuid.New().String()\n\n\tsession, err := h.service.q.CreateOrUpdateChatSessionByUUID(r.Context(), sqlc_queries.CreateOrUpdateChatSessionByUUIDParams{\n\t\tUuid: sessionUUID,\n\t\tUserID: userID,\n\t\tTopic: sessionTitle,\n\t\tMaxLength: originSession.MaxLength,\n\t\tTemperature: originSession.Temperature,\n\t\tModel: originSession.Model,\n\t\tMaxTokens: originSession.MaxTokens,\n\t\tTopP: originSession.TopP,\n\t\tDebug: originSession.Debug,\n\t\tSummarizeMode: originSession.SummarizeMode,\n\t\tExploreMode: originSession.ExploreMode,\n\t\tWorkspaceID: originSession.WorkspaceID,\n\t\tN: 1,\n\t})\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to create chat session from snapshot\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t_, err = h.service.q.CreateChatPrompt(r.Context(), sqlc_queries.CreateChatPromptParams{\n\t\tUuid: NewUUID(),\n\t\tChatSessionUuid: sessionUUID,\n\t\tRole: \"system\",\n\t\tContent: promptMsg.Text,\n\t\tUserID: userID,\n\t\tCreatedBy: userID,\n\t\tUpdatedBy: userID,\n\t})\n\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to create prompt for chat session from snapshot\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tfor _, message := range conversionsSimpleMessages[1:] {\n\t\t// if inversion is true, the role is user, otherwise assistant\n\t\t// Determine the role based on the inversion flag\n\n\t\tmessageParam := sqlc_queries.CreateChatMessageParams{\n\t\t\tChatSessionUuid: sessionUUID,\n\t\t\tUuid: NewUUID(),\n\t\t\tRole: message.GetRole(),\n\t\t\tContent: message.Text,\n\t\t\tUserID: userID,\n\t\t\tRaw: json.RawMessage([]byte(\"{}\")),\n\t\t}\n\t\t_, err = h.service.q.CreateChatMessage(r.Context(), messageParam)\n\t\tif err != nil {\n\t\t\tapiErr := ErrInternalUnexpected\n\t\t\tapiErr.Detail = \"Failed to create messages for chat session from snapshot\"\n\t\t\tapiErr.DebugInfo = err.Error()\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\n\t}\n\n\t// set active session using simplified service\n\tactiveSessionService := NewUserActiveChatSessionService(h.service.q)\n\t_, err = activeSessionService.UpsertActiveSession(r.Context(), userID, nil, session.Uuid)\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to update active session\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(map[string]string{\"SessionUuid\": session.Uuid})\n}\n" }, { "path": "api/chat_session_service.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"strings\"\n\n\t\"github.com/rotisserie/eris\"\n\t\"github.com/samber/lo\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// ChatSessionService provides methods for interacting with chat sessions.\ntype ChatSessionService struct {\n\tq *sqlc_queries.Queries\n}\n\n// NewChatSessionService creates a new ChatSessionService.\nfunc NewChatSessionService(q *sqlc_queries.Queries) *ChatSessionService {\n\treturn &ChatSessionService{q: q}\n}\n\n// CreateChatSession creates a new chat session.\nfunc (s *ChatSessionService) CreateChatSession(ctx context.Context, session_params sqlc_queries.CreateChatSessionParams) (sqlc_queries.ChatSession, error) {\n\tsession, err := s.q.CreateChatSession(ctx, session_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatSession{}, err\n\t}\n\treturn session, nil\n}\n\n// GetChatSessionByID returns a chat session by ID.\nfunc (s *ChatSessionService) GetChatSessionByID(ctx context.Context, id int32) (sqlc_queries.ChatSession, error) {\n\tsession, err := s.q.GetChatSessionByID(ctx, id)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatSession{}, eris.Wrap(err, \"failed to retrieve session: \")\n\t}\n\treturn session, nil\n}\n\n// UpdateChatSession updates an existing chat session.\nfunc (s *ChatSessionService) UpdateChatSession(ctx context.Context, session_params sqlc_queries.UpdateChatSessionParams) (sqlc_queries.ChatSession, error) {\n\tsession_u, err := s.q.UpdateChatSession(ctx, session_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatSession{}, eris.Wrap(err, \"failed to update session\")\n\t}\n\treturn session_u, nil\n}\n\n// DeleteChatSession deletes a chat session by ID.\nfunc (s *ChatSessionService) DeleteChatSession(ctx context.Context, id int32) error {\n\terr := s.q.DeleteChatSession(ctx, id)\n\tif err != nil {\n\t\treturn eris.Wrap(err, \"failed to delete session by id\")\n\t}\n\treturn nil\n}\n\n// GetAllChatSessions returns all chat sessions.\nfunc (s *ChatSessionService) GetAllChatSessions(ctx context.Context) ([]sqlc_queries.ChatSession, error) {\n\tsessions, err := s.q.GetAllChatSessions(ctx)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to retrieve sessions\")\n\t}\n\treturn sessions, nil\n}\n\nfunc (s *ChatSessionService) GetChatSessionsByUserID(ctx context.Context, userID int32) ([]sqlc_queries.ChatSession, error) {\n\tsessions, err := s.q.GetChatSessionsByUserID(ctx, userID)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to retrieve sessions\")\n\t}\n\treturn sessions, nil\n}\n\nfunc (s *ChatSessionService) GetSimpleChatSessionsByUserID(ctx context.Context, userID int32) ([]SimpleChatSession, error) {\n\tsessions, err := s.q.GetSessionsGroupedByWorkspace(ctx, userID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tsimple_sessions := lo.Map(sessions, func(session sqlc_queries.GetSessionsGroupedByWorkspaceRow, _idx int) SimpleChatSession {\n\t\tworkspaceUuid := \"\"\n\t\tif session.WorkspaceUuid.Valid {\n\t\t\tworkspaceUuid = session.WorkspaceUuid.String\n\t\t}\n\n\t\treturn SimpleChatSession{\n\t\t\tUuid: session.Uuid,\n\t\t\tIsEdit: false,\n\t\t\tTitle: session.Topic,\n\t\t\tMaxLength: int(session.MaxLength),\n\t\t\tTemperature: float64(session.Temperature),\n\t\t\tTopP: float64(session.TopP),\n\t\t\tN: session.N,\n\t\t\tMaxTokens: session.MaxTokens,\n\t\t\tDebug: session.Debug,\n\t\t\tModel: session.Model,\n\t\t\tSummarizeMode: session.SummarizeMode,\n\t\t\tArtifactEnabled: session.ArtifactEnabled,\n\t\t\tWorkspaceUuid: workspaceUuid,\n\t\t}\n\t})\n\treturn simple_sessions, nil\n}\n\n// GetChatSessionByUUID returns an authentication user record by ID.\nfunc (s *ChatSessionService) GetChatSessionByUUID(ctx context.Context, uuid string) (sqlc_queries.ChatSession, error) {\n\tchatSession, err := s.q.GetChatSessionByUUID(ctx, uuid)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatSession{}, eris.Wrap(err, \"failed to retrieve session by uuid, \")\n\t}\n\treturn chatSession, nil\n}\n\n// UpdateChatSessionByUUID updates an existing chat session.\nfunc (s *ChatSessionService) UpdateChatSessionByUUID(ctx context.Context, session_params sqlc_queries.UpdateChatSessionByUUIDParams) (sqlc_queries.ChatSession, error) {\n\tsession_u, err := s.q.UpdateChatSessionByUUID(ctx, session_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatSession{}, eris.Wrap(err, \"failed to update session, \")\n\t}\n\treturn session_u, nil\n}\n\n// UpdateChatSessionTopicByUUID updates an existing chat session topic.\nfunc (s *ChatSessionService) UpdateChatSessionTopicByUUID(ctx context.Context, session_params sqlc_queries.UpdateChatSessionTopicByUUIDParams) (sqlc_queries.ChatSession, error) {\n\tsession_u, err := s.q.UpdateChatSessionTopicByUUID(ctx, session_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatSession{}, eris.Wrap(err, \"failed to update session, \")\n\t}\n\treturn session_u, nil\n}\n\n// CreateOrUpdateChatSessionByUUID updates an existing chat session.\nfunc (s *ChatSessionService) CreateOrUpdateChatSessionByUUID(ctx context.Context, session_params sqlc_queries.CreateOrUpdateChatSessionByUUIDParams) (sqlc_queries.ChatSession, error) {\n\tsession_u, err := s.q.CreateOrUpdateChatSessionByUUID(ctx, session_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatSession{}, eris.Wrap(err, \"failed to update session, \")\n\t}\n\treturn session_u, nil\n}\n\n// DeleteChatSessionByUUID deletes a chat session by UUID.\nfunc (s *ChatSessionService) DeleteChatSessionByUUID(ctx context.Context, uuid string) error {\n\terr := s.q.DeleteChatSessionByUUID(ctx, uuid)\n\tif err != nil {\n\t\treturn eris.Wrap(err, \"failed to delete session by uuid, \")\n\n\t}\n\treturn nil\n}\n\n// UpdateSessionMaxLength\nfunc (s *ChatSessionService) UpdateSessionMaxLength(ctx context.Context, session_params sqlc_queries.UpdateSessionMaxLengthParams) (sqlc_queries.ChatSession, error) {\n\tsession_u, err := s.q.UpdateSessionMaxLength(ctx, session_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatSession{}, eris.Wrap(err, \"failed to update session, \")\n\t}\n\treturn session_u, nil\n}\n\n// EnsureDefaultSystemPrompt ensures a session has exactly one active system prompt.\n// It is safe to call repeatedly and tolerates concurrent callers.\nfunc (s *ChatSessionService) EnsureDefaultSystemPrompt(ctx context.Context, chatSessionUUID string, userID int32, systemPrompt string) (sqlc_queries.ChatPrompt, error) {\n\texistingPrompt, err := s.q.GetOneChatPromptBySessionUUID(ctx, chatSessionUUID)\n\tif err == nil {\n\t\treturn existingPrompt, nil\n\t}\n\tif !errors.Is(err, sql.ErrNoRows) {\n\t\treturn sqlc_queries.ChatPrompt{}, eris.Wrap(err, \"failed to check existing session prompt\")\n\t}\n\n\tpromptText := strings.TrimSpace(systemPrompt)\n\tif promptText == \"\" {\n\t\tpromptText = DefaultSystemPromptText\n\t}\n\n\ttokenCount, tokenErr := getTokenCount(promptText)\n\tif tokenErr != nil {\n\t\ttokenCount = len(promptText) / TokenEstimateRatio\n\t}\n\tif tokenCount <= 0 {\n\t\ttokenCount = 1\n\t}\n\n\tprompt, createErr := s.q.CreateChatPrompt(ctx, sqlc_queries.CreateChatPromptParams{\n\t\tUuid: NewUUID(),\n\t\tChatSessionUuid: chatSessionUUID,\n\t\tRole: \"system\",\n\t\tContent: promptText,\n\t\tTokenCount: int32(tokenCount),\n\t\tUserID: userID,\n\t\tCreatedBy: userID,\n\t\tUpdatedBy: userID,\n\t})\n\tif createErr == nil {\n\t\treturn prompt, nil\n\t}\n\n\t// Handle concurrent creation race by returning the now-existing prompt.\n\texistingPrompt, err = s.q.GetOneChatPromptBySessionUUID(ctx, chatSessionUUID)\n\tif err == nil {\n\t\treturn existingPrompt, nil\n\t}\n\n\treturn sqlc_queries.ChatPrompt{}, eris.Wrap(createErr, \"failed to create default system prompt\")\n}\n" }, { "path": "api/chat_session_service_test.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t_ \"github.com/lib/pq\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\nfunc TestChatSessionService(t *testing.T) {\n\tsqlc_q := sqlc_queries.New(db)\n\tservice := NewChatSessionService(sqlc_q)\n\t// Create a new database connection\n\n\t// Insert a new chat session into the database\n\tsession_params := sqlc_queries.CreateChatSessionParams{UserID: 1, Topic: \"Test Session\", MaxLength: 100}\n\tsession, err := service.CreateChatSession(context.Background(), session_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat session: %v\", err)\n\t}\n\n\t// Retrieve the chat session by ID and check that it matches the expected values\n\tretrievedSession, err := service.GetChatSessionByID(context.Background(), session.ID)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to get chat session: %v\", err)\n\t}\n\tif retrievedSession.UserID != session.UserID || retrievedSession.Topic != session.Topic || retrievedSession.MaxLength != session.MaxLength {\n\t\tt.Error(\"retrieved chat session does not match expected values\")\n\t}\n\n\t// Update the chat session and check that it was updated in the database\n\tupdated_params := sqlc_queries.UpdateChatSessionParams{ID: session.ID,\n\t\tUserID: session.UserID,\n\t\tTopic: \"Updated Test Session\",\n\t}\n\tif _, err := service.UpdateChatSession(context.Background(), updated_params); err != nil {\n\t\tt.Fatalf(\"failed to update chat session: %v\", err)\n\t}\n\tretrievedSession, err = service.GetChatSessionByID(context.Background(), session.ID)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to get chat session: %v\", err)\n\t}\n\n\t// Check that updated chat session matches expected values\n\t// UpdatedAt is generated by the database; verify updated fields instead of strict timestamp equality.\n\tif retrievedSession.Topic != updated_params.Topic {\n\t\tt.Errorf(\"chat session mismatch: expected Topic=%s, got Topic=%s \",\n\t\t\tupdated_params.Topic, retrievedSession.Topic)\n\t}\n\n\t// Delete the chat session and check that it was deleted from the database\n\n\tif err := service.DeleteChatSession(context.Background(), session.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat session: %v\", err)\n\t}\n\tdeletedSession, err := service.GetChatSessionByID(context.Background(), session.ID)\n\tif err == nil || !errors.Is(err, sql.ErrNoRows) {\n\t\tfmt.Printf(\"%+v\", deletedSession)\n\t\tt.Error(\"expected error due to missing chat session, but got no error or different error\")\n\t}\n}\n\nfunc TestGetChatSessionsByUserID(t *testing.T) {\n\tsqlc_q := sqlc_queries.New(db)\n\tservice := NewChatSessionService(sqlc_q)\n\n\t// Insert two chat sessions into the database with different user IDs\n\tsession1_params := sqlc_queries.CreateChatSessionParams{UserID: 1, Topic: \"Test Session 1\", MaxLength: 100, Uuid: \"uuid1\"}\n\tsession1, err := service.CreateChatSession(context.Background(), session1_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat session: %v\", err)\n\t}\n\tsession2_params := sqlc_queries.CreateChatSessionParams{UserID: 2, Topic: \"Test Session 2\", MaxLength: 150, Uuid: \"uuid2\"}\n\tsession2, err := service.CreateChatSession(context.Background(), session2_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat session: %v\", err)\n\t}\n\n\t// Retrieve chat sessions by user ID and check that they match the expected values\n\tuserID := int32(1)\n\tsessions, err := service.GetChatSessionsByUserID(context.Background(), userID)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to retrieve chat sessions: %v\", err)\n\t}\n\tif len(sessions) != 1 {\n\t\tt.Errorf(\"expected 1 chat session, but got %d\", len(sessions))\n\t}\n\tif sessions[0].UserID != session1.UserID || sessions[0].Topic != session1.Topic || sessions[0].MaxLength != session1.MaxLength {\n\t\tt.Error(\"retrieved chat sessions do not match expected values\")\n\t}\n\tif err := service.DeleteChatSession(context.Background(), session1.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat session: %v\", err)\n\t}\n\tif err := service.DeleteChatSession(context.Background(), session2.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat session: %v\", err)\n\t}\n}\n\nfunc TestGetAllChatSessions(t *testing.T) {\n\n\t// Create a new ChatSessionService with the test database connection\n\tq := sqlc_queries.New(db)\n\tservice := NewChatSessionService(q)\n\n\tsession1_params := sqlc_queries.CreateChatSessionParams{UserID: 1, Topic: \"Test Session 1\", MaxLength: 100, Uuid: \"uuid1\"}\n\tsession1, err := service.CreateChatSession(context.Background(), session1_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat session: %v\", err)\n\t}\n\tsession2_params := sqlc_queries.CreateChatSessionParams{UserID: 2, Topic: \"Test Session 2\", MaxLength: 150, Uuid: \"uuid2\"}\n\tsession2, err := service.CreateChatSession(context.Background(), session2_params)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create chat session: %v\", err)\n\t}\n\n\t// Retrieve all chat sessions and check that they match the expected values\n\tsessions, err := service.GetAllChatSessions(context.Background())\n\tif err != nil {\n\t\tt.Fatalf(\"failed to retrieve chat sessions: %v\", err)\n\t}\n\tif len(sessions) != 2 {\n\t\tt.Errorf(\"expected 2 chat sessions, but got %d\", len(sessions))\n\t}\n\tif sessions[0].Topic != session1.Topic || sessions[1].Topic != session2.Topic {\n\t\tt.Error(\"retrieved chat sessions do not match expected values\")\n\t}\n\n\tif err := service.DeleteChatSession(context.Background(), session1.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat session: %v\", err)\n\t}\n\tif err := service.DeleteChatSession(context.Background(), session2.ID); err != nil {\n\t\tt.Fatalf(\"failed to delete chat session: %v\", err)\n\t}\n}\n" }, { "path": "api/chat_snapshot_handler.go", "content": "package main\n\nimport (\n\t\"encoding/json\"\n\t\"net/http\"\n\t\"strconv\"\n\n\t\"github.com/gorilla/mux\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// ChatSnapshotHandler handles requests related to chat snapshots\ntype ChatSnapshotHandler struct {\n\tservice *ChatSnapshotService\n}\n\n// NewChatSnapshotHandler creates a new handler instance\nfunc NewChatSnapshotHandler(sqlc_q *sqlc_queries.Queries) *ChatSnapshotHandler {\n\treturn &ChatSnapshotHandler{\n\t\tservice: NewChatSnapshotService(sqlc_q),\n\t}\n}\n\nfunc (h *ChatSnapshotHandler) Register(router *mux.Router) {\n\trouter.HandleFunc(\"/uuid/chat_snapshot/all\", h.ChatSnapshotMetaByUserID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/uuid/chat_snapshot/{uuid}\", h.GetChatSnapshot).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/uuid/chat_snapshot/{uuid}\", h.CreateChatSnapshot).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/uuid/chat_snapshot/{uuid}\", h.UpdateChatSnapshotMetaByUUID).Methods(http.MethodPut)\n\trouter.HandleFunc(\"/uuid/chat_snapshot/{uuid}\", h.DeleteChatSnapshot).Methods(http.MethodDelete)\n\trouter.HandleFunc(\"/uuid/chat_snapshot_search\", h.ChatSnapshotSearch).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/uuid/chat_bot/{uuid}\", h.CreateChatBot).Methods(http.MethodPost)\n}\n\nfunc (h *ChatSnapshotHandler) CreateChatSnapshot(w http.ResponseWriter, r *http.Request) {\n\tchatSessionUuid := mux.Vars(r)[\"uuid\"]\n\tuser_id, err := getUserID(r.Context())\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tuuid, err := h.service.CreateChatSnapshot(r.Context(), chatSessionUuid, user_id)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to create chat snapshot\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(\n\t\tmap[string]interface{}{\n\t\t\t\"uuid\": uuid,\n\t\t})\n\n}\n\nfunc (h *ChatSnapshotHandler) CreateChatBot(w http.ResponseWriter, r *http.Request) {\n\tchatSessionUuid := mux.Vars(r)[\"uuid\"]\n\tuser_id, err := getUserID(r.Context())\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tuuid, err := h.service.CreateChatBot(r.Context(), chatSessionUuid, user_id)\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to create chat bot\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(\n\t\tmap[string]interface{}{\n\t\t\t\"uuid\": uuid,\n\t\t})\n\n}\n\nfunc (h *ChatSnapshotHandler) GetChatSnapshot(w http.ResponseWriter, r *http.Request) {\n\tuuidStr := mux.Vars(r)[\"uuid\"]\n\tsnapshot, err := h.service.q.ChatSnapshotByUUID(r.Context(), uuidStr)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to get chat snapshot\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(snapshot)\n\n}\n\nfunc (h *ChatSnapshotHandler) ChatSnapshotMetaByUserID(w http.ResponseWriter, r *http.Request) {\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\t// get type from query\n\ttyp := r.URL.Query().Get(\"type\")\n\n\t// Parse pagination parameters\n\tpageStr := r.URL.Query().Get(\"page\")\n\tpageSizeStr := r.URL.Query().Get(\"page_size\")\n\n\tpage := int32(1) // Default to page 1\n\tpageSize := int32(20) // Default to 20 items per page\n\n\tif pageStr != \"\" {\n\t\tparsedPage, err := strconv.Atoi(pageStr)\n\t\tif err == nil && parsedPage > 0 {\n\t\t\tpage = int32(parsedPage)\n\t\t}\n\t}\n\n\tif pageSizeStr != \"\" {\n\t\tparsedPageSize, err := strconv.Atoi(pageSizeStr)\n\t\tif err == nil && parsedPageSize > 0 && parsedPageSize <= 100 {\n\t\t\tpageSize = int32(parsedPageSize)\n\t\t}\n\t}\n\n\toffset := (page - 1) * pageSize\n\n\tchatSnapshots, err := h.service.q.ChatSnapshotMetaByUserID(r.Context(), sqlc_queries.ChatSnapshotMetaByUserIDParams{\n\t\tUserID: userID,\n\t\tTyp: typ,\n\t\tLimit: pageSize,\n\t\tOffset: offset,\n\t})\n\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to retrieve chat snapshots\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Get total count for pagination\n\ttotalCount, err := h.service.q.ChatSnapshotCountByUserIDAndType(r.Context(), sqlc_queries.ChatSnapshotCountByUserIDAndTypeParams{\n\t\tUserID: userID,\n\t\tColumn2: typ,\n\t})\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to retrieve snapshot count\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(map[string]interface{}{\n\t\t\"data\": chatSnapshots,\n\t\t\"page\": page,\n\t\t\"page_size\": pageSize,\n\t\t\"total\": totalCount,\n\t})\n}\nfunc (h *ChatSnapshotHandler) UpdateChatSnapshotMetaByUUID(w http.ResponseWriter, r *http.Request) {\n\tuuid := mux.Vars(r)[\"uuid\"]\n\tvar input struct {\n\t\tTitle string `json:\"title\"`\n\t\tSummary string `json:\"summary\"`\n\t}\n\terr := json.NewDecoder(r.Body).Decode(&input)\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Failed to parse request body\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\terr = h.service.q.UpdateChatSnapshotMetaByUUID(r.Context(), sqlc_queries.UpdateChatSnapshotMetaByUUIDParams{\n\t\tUuid: uuid,\n\t\tTitle: input.Title,\n\t\tSummary: input.Summary,\n\t\tUserID: userID,\n\t})\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to update chat snapshot metadata\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tsnapshot, err := h.service.q.ChatSnapshotByUUID(r.Context(), uuid)\n\tif err != nil {\n\t\tapiErr := ErrResourceNotFound(\"Chat snapshot\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(snapshot)\n\n}\n\nfunc (h *ChatSnapshotHandler) DeleteChatSnapshot(w http.ResponseWriter, r *http.Request) {\n\tvars := mux.Vars(r)\n\tuuid := vars[\"uuid\"]\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t_, err = h.service.q.DeleteChatSnapshot(r.Context(), sqlc_queries.DeleteChatSnapshotParams{\n\t\tUuid: uuid,\n\t\tUserID: userID,\n\t})\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to delete chat snapshot\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n}\n\nfunc (h *ChatSnapshotHandler) ChatSnapshotSearch(w http.ResponseWriter, r *http.Request) {\n\tsearch := r.URL.Query().Get(\"search\")\n\tif search == \"\" {\n\t\tw.WriteHeader(http.StatusOK)\n\t\tvar emptySlice []any // create an empty slice of integers\n\t\tjson.NewEncoder(w).Encode(emptySlice)\n\t\treturn\n\t}\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tchatSnapshots, err := h.service.q.ChatSnapshotSearch(r.Context(), sqlc_queries.ChatSnapshotSearchParams{\n\t\tUserID: userID,\n\t\tSearch: search,\n\t})\n\tif err != nil {\n\t\tapiErr := ErrInternalUnexpected\n\t\tapiErr.Detail = \"Failed to search chat snapshots\"\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(chatSnapshots)\n}\n" }, { "path": "api/chat_snapshot_handler_test.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"net/http/httptest\"\n\t\"testing\"\n\n\t\"github.com/gorilla/mux\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n\t\"gotest.tools/v3/assert\"\n)\n\n// the code below do db update directly in instead of using handler, please change to use handler\n\n// TestChatSnapshot tests the ChatSnapshotHandler\nfunc TestChatSnapshot(t *testing.T) {\n\tconst snapshotPath = \"/uuid/chat_snapshot/%s\" // API path for snapshots\n\n\t// Create a chat service for testing\n\tq := sqlc_queries.New(db)\n\th := NewChatSnapshotHandler(q) // Create a ChatSnapshotHandler\n\n\t// Register snapshot API routes\n\trouter := mux.NewRouter()\n\th.Register(router)\n\n\t// Add a test user\n\tuserID := 1\n\n\t// Generate a random UUID for the snapshot\n\tsnapshotUUID := NewUUID()\n\n\t// Create a test snapshot\n\tsnapshot, err := h.service.q.CreateChatSnapshot(context.Background(), sqlc_queries.CreateChatSnapshotParams{\n\t\tUuid: snapshotUUID, // Use the generated UUID\n\t\tModel: \"gpt3\",\n\t\tTitle: \"test chat snapshot\",\n\t\tUserID: int32(userID),\n\t\tSession: json.RawMessage([]byte(\"{}\")),\n\t\tTags: json.RawMessage([]byte(\"{}\")),\n\t\tText: \"test chat snapshot text\",\n\t\tConversation: json.RawMessage([]byte(\"{}\")),\n\t})\n\tif err != nil {\n\t\treturn\n\t}\n\tassert.Equal(t, snapshot.Uuid, snapshotUUID)\n\n\t// Test GET snapshot - should succeed\n\treq, _ := http.NewRequest(\"GET\", fmt.Sprintf(snapshotPath, snapshot.Uuid), nil)\n\trr := httptest.NewRecorder()\n\trouter.ServeHTTP(rr, req)\n\tassert.Equal(t, http.StatusOK, rr.Code)\n\n\t// Test DELETE snapshot without auth - should fail\n\treqDelete, _ := http.NewRequest(\"DELETE\", fmt.Sprintf(snapshotPath, snapshot.Uuid), nil)\n\trr = httptest.NewRecorder()\n\trouter.ServeHTTP(rr, reqDelete)\n\tassert.Equal(t, http.StatusUnauthorized, rr.Code)\n\n\t// Test DELETE snapshot with auth - should succeed\n\treqDeleteWithAuth, _ := http.NewRequest(\"DELETE\", fmt.Sprintf(snapshotPath, snapshot.Uuid), nil)\n\tctx := getContextWithUser(userID) // Get auth context\n\treqDeleteWithAuth = reqDeleteWithAuth.WithContext(ctx)\n\trr = httptest.NewRecorder()\n\trouter.ServeHTTP(rr, reqDeleteWithAuth)\n\tassert.Equal(t, http.StatusOK, rr.Code)\n\n}\n" }, { "path": "api/chat_snapshot_service.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"log\"\n\n\t\"github.com/google/uuid\"\n\t\"github.com/samber/lo\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// ChatSnapshotService provides methods for interacting with chat sessions.\ntype ChatSnapshotService struct {\n\tq *sqlc_queries.Queries\n}\n\n// NewChatSnapshotService creates a new ChatSnapshotService.\nfunc NewChatSnapshotService(q *sqlc_queries.Queries) *ChatSnapshotService {\n\treturn &ChatSnapshotService{q: q}\n}\n\nfunc (s *ChatSnapshotService) CreateChatSnapshot(ctx context.Context, chatSessionUuid string, userId int32) (string, error) {\n\tchatSession, err := s.q.GetChatSessionByUUID(ctx, chatSessionUuid)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\t// TODO: fix hardcode\n\t// Get chat history\n\tsimple_msgs, err := s.q.GetChatHistoryBySessionUUID(ctx, chatSessionUuid, int32(1), int32(10000))\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\ttext := lo.Reduce(simple_msgs, func(acc string, curr sqlc_queries.SimpleChatMessage, _ int) string {\n\t\treturn acc + curr.Text\n\t}, \"\")\n\ttitle := GenTitle(s.q, ctx, chatSession, text)\n\t// simple_msgs to RawMessage\n\tsimple_msgs_raw, err := json.Marshal(simple_msgs)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\tsnapshot_uuid := uuid.New().String()\n\tchatSessionMessage, err := json.Marshal(chatSession)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\tone, err := s.q.CreateChatSnapshot(ctx, sqlc_queries.CreateChatSnapshotParams{\n\t\tUuid: snapshot_uuid,\n\t\tModel: chatSession.Model,\n\t\tTitle: title,\n\t\tUserID: userId,\n\t\tSession: chatSessionMessage,\n\t\tTags: json.RawMessage([]byte(\"{}\")),\n\t\tText: text,\n\t\tConversation: simple_msgs_raw,\n\t})\n\tif err != nil {\n\t\tlog.Println(err)\n\t\treturn \"\", err\n\t}\n\treturn one.Uuid, nil\n\n}\n\nfunc GenTitle(q *sqlc_queries.Queries, ctx context.Context, chatSession sqlc_queries.ChatSession, text string) string {\n\ttitle := firstN(chatSession.Topic, 100)\n\t// generate title using\n\tmodel := \"gemini-2.0-flash\"\n\t_, err := q.ChatModelByName(ctx, model)\n\tif err == nil {\n\t\tgenTitle, err := GenerateChatTitle(ctx, model, text)\n\t\tif err != nil {\n\t\t\tlog.Println(err)\n\t\t}\n\t\tif genTitle != \"\" {\n\t\t\ttitle = genTitle\n\t\t}\n\t}\n\treturn title\n}\n\nfunc (s *ChatSnapshotService) CreateChatBot(ctx context.Context, chatSessionUuid string, userId int32) (string, error) {\n\tchatSession, err := s.q.GetChatSessionByUUID(ctx, chatSessionUuid)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\t// TODO: fix hardcode\n\tsimple_msgs, err := s.q.GetChatHistoryBySessionUUID(ctx, chatSessionUuid, int32(1), int32(10000))\n\ttext := lo.Reduce(simple_msgs, func(acc string, curr sqlc_queries.SimpleChatMessage, _ int) string {\n\t\treturn acc + curr.Text\n\t}, \"\")\n\t// save all simple_msgs to a jsonb field in chat_snapshot\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\t// simple_msgs to RawMessage\n\tsimple_msgs_raw, err := json.Marshal(simple_msgs)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\tsnapshot_uuid := uuid.New().String()\n\tchatSessionMessage, err := json.Marshal(chatSession)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\ttitle := GenTitle(s.q, ctx, chatSession, text)\n\tone, err := s.q.CreateChatBot(ctx, sqlc_queries.CreateChatBotParams{\n\t\tUuid: snapshot_uuid,\n\t\tModel: chatSession.Model,\n\t\tTyp: \"chatbot\",\n\t\tTitle: title,\n\t\tUserID: userId,\n\t\tSession: chatSessionMessage,\n\t\tTags: json.RawMessage([]byte(\"{}\")),\n\t\tText: text,\n\t\tConversation: simple_msgs_raw,\n\t})\n\tif err != nil {\n\t\tlog.Println(err)\n\t\treturn \"\", err\n\t}\n\treturn one.Uuid, nil\n\n}\n" }, { "path": "api/chat_user_active_chat_session_handler.go", "content": "package main\n\nimport (\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"log\"\n\t\"net/http\"\n\t\"regexp\"\n\n\t\"github.com/gorilla/mux\"\n\tsqlc \"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// UserActiveChatSessionHandler handles requests related to active chat sessions\ntype UserActiveChatSessionHandler struct {\n\tservice *UserActiveChatSessionService\n}\n\n// NewUserActiveChatSessionHandler creates a new handler instance\nfunc NewUserActiveChatSessionHandler(sqlc_q *sqlc.Queries) *UserActiveChatSessionHandler {\n\tactiveSessionService := NewUserActiveChatSessionService(sqlc_q)\n\treturn &UserActiveChatSessionHandler{\n\t\tservice: activeSessionService,\n\t}\n}\n\n// Register sets up the handler routes\nfunc (h *UserActiveChatSessionHandler) Register(router *mux.Router) {\n\trouter.HandleFunc(\"/uuid/user_active_chat_session\", h.GetUserActiveChatSessionHandler).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/uuid/user_active_chat_session\", h.CreateOrUpdateUserActiveChatSessionHandler).Methods(http.MethodPut)\n\n\t// Per-workspace active session endpoints\n\t// Note: More specific routes must come before parameterized routes to avoid shadowing\n\trouter.HandleFunc(\"/workspaces/active-sessions\", h.GetAllWorkspaceActiveSessionsHandler).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/workspaces/{workspaceUuid}/active-session\", h.GetWorkspaceActiveSessionHandler).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/workspaces/{workspaceUuid}/active-session\", h.SetWorkspaceActiveSessionHandler).Methods(http.MethodPut)\n}\n\n// GetUserActiveChatSessionHandler handles GET requests to get a session by user_id\nfunc (h *UserActiveChatSessionHandler) GetUserActiveChatSessionHandler(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\n\t// Get and validate user ID\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"missing or invalid user ID\"))\n\t\treturn\n\t}\n\n\tlog.Printf(\"Getting active chat session for user %d\", userID)\n\n\t// Get session from service (use unified approach for global session)\n\tsession, err := h.service.GetActiveSession(r.Context(), userID, nil)\n\tif err != nil {\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\tRespondWithAPIError(w, ErrChatSessionNotFound.WithMessage(fmt.Sprintf(\"no active session for user %d\", userID)))\n\t\t} else {\n\t\t\tRespondWithAPIError(w, WrapError(err, \"failed to get active chat session\"))\n\t\t}\n\t\treturn\n\t}\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tif err := json.NewEncoder(w).Encode(session); err != nil {\n\t\tlog.Printf(\"Failed to encode response: %v\", err)\n\t}\n}\n\n// UUID validation regex\nvar uuidRegex = regexp.MustCompile(`^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$`)\n\n// CreateOrUpdateUserActiveChatSessionHandler handles PUT requests to create/update a session\nfunc (h *UserActiveChatSessionHandler) CreateOrUpdateUserActiveChatSessionHandler(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\n\t// Get and validate user ID\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"missing or invalid user ID\"))\n\t\treturn\n\t}\n\n\t// Parse request body\n\tvar reqBody struct {\n\t\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\t}\n\tif err := json.NewDecoder(r.Body).Decode(&reqBody); err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"failed to parse request body\"))\n\t\treturn\n\t}\n\n\t// Validate session UUID format\n\tif !uuidRegex.MatchString(reqBody.ChatSessionUuid) {\n\t\tRespondWithAPIError(w, ErrChatSessionInvalid.WithMessage(\n\t\t\tfmt.Sprintf(\"invalid session UUID format: %s\", reqBody.ChatSessionUuid)))\n\t\treturn\n\t}\n\n\tlog.Printf(\"Creating/updating active chat session for user %d\", userID)\n\n\t// Create/update session (use unified approach for global session)\n\tsession, err := h.service.UpsertActiveSession(r.Context(), userID, nil, reqBody.ChatSessionUuid)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to create or update active chat session\"))\n\t\treturn\n\t}\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tif err := json.NewEncoder(w).Encode(session); err != nil {\n\t\tlog.Printf(\"Failed to encode response: %v\", err)\n\t}\n}\n\n// Per-workspace active session handlers\n\n// GetWorkspaceActiveSessionHandler gets the active session for a specific workspace\nfunc (h *UserActiveChatSessionHandler) GetWorkspaceActiveSessionHandler(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\tworkspaceUuid := mux.Vars(r)[\"workspaceUuid\"]\n\n\t// Get and validate user ID\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"missing or invalid user ID\"))\n\t\treturn\n\t}\n\n\t// Check workspace permission\n\tworkspaceService := NewChatWorkspaceService(h.service.q)\n\thasPermission, err := workspaceService.HasWorkspacePermission(ctx, workspaceUuid, userID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to check workspace permission\"))\n\t\treturn\n\t}\n\tif !hasPermission {\n\t\tRespondWithAPIError(w, ErrAuthAccessDenied.WithMessage(\"access denied to workspace\"))\n\t\treturn\n\t}\n\n\t// Get workspace to get its ID\n\tworkspace, err := workspaceService.GetWorkspaceByUUID(ctx, workspaceUuid)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrResourceNotFound(\"Workspace\").WithMessage(\"workspace not found\"))\n\t\treturn\n\t}\n\n\t// Get workspace active session\n\tsession, err := h.service.GetActiveSession(ctx, userID, &workspace.ID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrResourceNotFound(\"Active Session\").WithMessage(\"no active session for workspace\"))\n\t\treturn\n\t}\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tjson.NewEncoder(w).Encode(map[string]interface{}{\n\t\t\"chatSessionUuid\": session.ChatSessionUuid,\n\t\t\"workspaceUuid\": workspaceUuid,\n\t\t\"updatedAt\": session.UpdatedAt,\n\t})\n}\n\n// SetWorkspaceActiveSessionHandler sets the active session for a specific workspace\nfunc (h *UserActiveChatSessionHandler) SetWorkspaceActiveSessionHandler(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\tworkspaceUuid := mux.Vars(r)[\"workspaceUuid\"]\n\n\t// Get and validate user ID\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"missing or invalid user ID\"))\n\t\treturn\n\t}\n\n\t// Check workspace permission\n\tworkspaceService := NewChatWorkspaceService(h.service.q)\n\thasPermission, err := workspaceService.HasWorkspacePermission(ctx, workspaceUuid, userID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to check workspace permission\"))\n\t\treturn\n\t}\n\tif !hasPermission {\n\t\tRespondWithAPIError(w, ErrAuthAccessDenied.WithMessage(\"access denied to workspace\"))\n\t\treturn\n\t}\n\n\t// Parse request body\n\tvar requestBody struct {\n\t\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\t}\n\tif err := json.NewDecoder(r.Body).Decode(&requestBody); err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"failed to parse request body\"))\n\t\treturn\n\t}\n\n\t// Validate session UUID format\n\tif !uuidRegex.MatchString(requestBody.ChatSessionUuid) {\n\t\tRespondWithAPIError(w, ErrChatSessionInvalid.WithMessage(\"invalid session UUID format\"))\n\t\treturn\n\t}\n\n\t// Get workspace to get its ID\n\tworkspace, err := workspaceService.GetWorkspaceByUUID(ctx, workspaceUuid)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrResourceNotFound(\"Workspace\").WithMessage(\"workspace not found\"))\n\t\treturn\n\t}\n\n\tsessionService := NewChatSessionService(h.service.q)\n\tsession, err := sessionService.GetChatSessionByUUID(ctx, requestBody.ChatSessionUuid)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrResourceNotFound(\"Chat Session\").WithMessage(\"chat session not found\"))\n\t\treturn\n\t}\n\tif !session.WorkspaceID.Valid || session.WorkspaceID.Int32 != workspace.ID {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"session does not belong to workspace\"))\n\t\treturn\n\t}\n\n\t// Set workspace active session\n\tactiveSession, err := h.service.UpsertActiveSession(ctx, userID, &workspace.ID, requestBody.ChatSessionUuid)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to set workspace active session\"))\n\t\treturn\n\t}\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tjson.NewEncoder(w).Encode(map[string]interface{}{\n\t\t\"chatSessionUuid\": activeSession.ChatSessionUuid,\n\t\t\"workspaceUuid\": workspaceUuid,\n\t\t\"updatedAt\": activeSession.UpdatedAt,\n\t})\n}\n\n// GetAllWorkspaceActiveSessionsHandler gets all workspace active sessions for a user\nfunc (h *UserActiveChatSessionHandler) GetAllWorkspaceActiveSessionsHandler(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\n\t// Get and validate user ID\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"missing or invalid user ID\"))\n\t\treturn\n\t}\n\n\t// Get all workspace active sessions\n\tsessions, err := h.service.GetAllActiveSessions(ctx, userID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to get workspace active sessions\"))\n\t\treturn\n\t}\n\n\t// Convert to response format with workspace UUIDs\n\tworkspaceService := NewChatWorkspaceService(h.service.q)\n\tworkspaces, err := workspaceService.GetWorkspacesByUserID(ctx, userID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to get workspaces\"))\n\t\treturn\n\t}\n\n\t// Create a map for workspace ID to UUID lookup\n\tworkspaceMap := make(map[int32]string)\n\tfor _, workspace := range workspaces {\n\t\tworkspaceMap[workspace.ID] = workspace.Uuid\n\t}\n\n\t// Build response\n\tvar response []map[string]interface{}\n\tfor _, session := range sessions {\n\t\tif session.WorkspaceID.Valid {\n\t\t\tif workspaceUuid, exists := workspaceMap[session.WorkspaceID.Int32]; exists {\n\t\t\t\tresponse = append(response, map[string]interface{}{\n\t\t\t\t\t\"workspaceUuid\": workspaceUuid,\n\t\t\t\t\t\"chatSessionUuid\": session.ChatSessionUuid,\n\t\t\t\t\t\"updatedAt\": session.UpdatedAt,\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\t}\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tjson.NewEncoder(w).Encode(response)\n}\n" }, { "path": "api/chat_user_active_chat_session_sevice.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"github.com/rotisserie/eris\"\n\tsqlc \"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype UserActiveChatSessionService struct {\n\tq *sqlc.Queries\n}\n\nfunc NewUserActiveChatSessionService(q *sqlc.Queries) *UserActiveChatSessionService {\n\treturn &UserActiveChatSessionService{q: q}\n}\n\n// Simplified unified methods\n\n// UpsertActiveSession creates or updates an active session for a user in a specific workspace (or global if workspaceID is nil)\nfunc (s *UserActiveChatSessionService) UpsertActiveSession(ctx context.Context, userID int32, workspaceID *int32, sessionUUID string) (sqlc.UserActiveChatSession, error) {\n\tvar nullWorkspaceID sql.NullInt32\n\tif workspaceID != nil {\n\t\tnullWorkspaceID = sql.NullInt32{Int32: *workspaceID, Valid: true}\n\t}\n\n\tsession, err := s.q.UpsertUserActiveSession(ctx, sqlc.UpsertUserActiveSessionParams{\n\t\tUserID: userID,\n\t\tWorkspaceID: nullWorkspaceID,\n\t\tChatSessionUuid: sessionUUID,\n\t})\n\tif err != nil {\n\t\treturn sqlc.UserActiveChatSession{}, eris.Wrap(err, \"failed to upsert active session\")\n\t}\n\treturn session, nil\n}\n\n// GetActiveSession retrieves the active session for a user in a specific workspace (or global if workspaceID is nil)\nfunc (s *UserActiveChatSessionService) GetActiveSession(ctx context.Context, userID int32, workspaceID *int32) (sqlc.UserActiveChatSession, error) {\n\tvar workspaceParam int32\n\tif workspaceID != nil {\n\t\tworkspaceParam = *workspaceID\n\t}\n\n\tsession, err := s.q.GetUserActiveSession(ctx, sqlc.GetUserActiveSessionParams{\n\t\tUserID: userID,\n\t\tColumn2: workspaceParam, // SQLC generated this awkward name due to the complex WHERE clause\n\t})\n\tif err != nil {\n\t\treturn sqlc.UserActiveChatSession{}, eris.Wrap(err, \"failed to get active session\")\n\t}\n\treturn session, nil\n}\n\n// GetAllActiveSessions retrieves all active sessions for a user (both global and workspace-specific)\nfunc (s *UserActiveChatSessionService) GetAllActiveSessions(ctx context.Context, userID int32) ([]sqlc.UserActiveChatSession, error) {\n\tsessions, err := s.q.GetAllUserActiveSessions(ctx, userID)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to get all active sessions\")\n\t}\n\treturn sessions, nil\n}\n\n// DeleteActiveSession deletes the active session for a user in a specific workspace (or global if workspaceID is nil)\nfunc (s *UserActiveChatSessionService) DeleteActiveSession(ctx context.Context, userID int32, workspaceID *int32) error {\n\tvar workspaceParam int32\n\tif workspaceID != nil {\n\t\tworkspaceParam = *workspaceID\n\t}\n\n\terr := s.q.DeleteUserActiveSession(ctx, sqlc.DeleteUserActiveSessionParams{\n\t\tUserID: userID,\n\t\tColumn2: workspaceParam, // SQLC generated this awkward name\n\t})\n\tif err != nil {\n\t\treturn eris.Wrap(err, \"failed to delete active session\")\n\t}\n\treturn nil\n}\n" }, { "path": "api/chat_workspace_handler.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"log\"\n\t\"net/http\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\t\"github.com/gorilla/mux\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype ChatWorkspaceHandler struct {\n\tservice *ChatWorkspaceService\n}\n\nfunc NewChatWorkspaceHandler(sqlc_q *sqlc_queries.Queries) *ChatWorkspaceHandler {\n\tworkspaceService := NewChatWorkspaceService(sqlc_q)\n\treturn &ChatWorkspaceHandler{\n\t\tservice: workspaceService,\n\t}\n}\n\nfunc (h *ChatWorkspaceHandler) Register(router *mux.Router) {\n\trouter.HandleFunc(\"/workspaces\", h.getWorkspacesByUserID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/workspaces\", h.createWorkspace).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/workspaces/{uuid}\", h.getWorkspaceByUUID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/workspaces/{uuid}\", h.updateWorkspace).Methods(http.MethodPut)\n\trouter.HandleFunc(\"/workspaces/{uuid}\", h.deleteWorkspace).Methods(http.MethodDelete)\n\trouter.HandleFunc(\"/workspaces/{uuid}/reorder\", h.updateWorkspaceOrder).Methods(http.MethodPut)\n\trouter.HandleFunc(\"/workspaces/{uuid}/set-default\", h.setDefaultWorkspace).Methods(http.MethodPut)\n\trouter.HandleFunc(\"/workspaces/{uuid}/sessions\", h.createSessionInWorkspace).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/workspaces/{uuid}/sessions\", h.getSessionsByWorkspace).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/workspaces/default\", h.ensureDefaultWorkspace).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/workspaces/auto-migrate\", h.autoMigrateLegacySessions).Methods(http.MethodPost)\n}\n\ntype CreateWorkspaceRequest struct {\n\tName string `json:\"name\"`\n\tDescription string `json:\"description\"`\n\tColor string `json:\"color\"`\n\tIcon string `json:\"icon\"`\n\tIsDefault bool `json:\"isDefault\"`\n}\n\ntype UpdateWorkspaceRequest struct {\n\tName string `json:\"name\"`\n\tDescription string `json:\"description\"`\n\tColor string `json:\"color\"`\n\tIcon string `json:\"icon\"`\n}\n\ntype UpdateWorkspaceOrderRequest struct {\n\tOrderPosition int32 `json:\"orderPosition\"`\n}\n\ntype WorkspaceResponse struct {\n\tUuid string `json:\"uuid\"`\n\tName string `json:\"name\"`\n\tDescription string `json:\"description\"`\n\tColor string `json:\"color\"`\n\tIcon string `json:\"icon\"`\n\tIsDefault bool `json:\"isDefault\"`\n\tOrderPosition int32 `json:\"orderPosition\"`\n\tSessionCount int64 `json:\"sessionCount,omitempty\"`\n\tCreatedAt string `json:\"createdAt\"`\n\tUpdatedAt string `json:\"updatedAt\"`\n}\n\n// createWorkspace creates a new workspace\nfunc (h *ChatWorkspaceHandler) createWorkspace(w http.ResponseWriter, r *http.Request) {\n\tvar req CreateWorkspaceRequest\n\terr := json.NewDecoder(r.Body).Decode(&req)\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid request format\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Validate required fields\n\tif req.Name == \"\" {\n\t\tapiErr := ErrValidationInvalidInput(\"Workspace name is required\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Default values\n\tif req.Color == \"\" {\n\t\treq.Color = \"#6366f1\"\n\t}\n\tif req.Icon == \"\" {\n\t\treq.Icon = \"folder\"\n\t}\n\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tworkspaceUUID := uuid.New().String()\n\tmakeDefault := req.IsDefault\n\tparams := sqlc_queries.CreateWorkspaceParams{\n\t\tUuid: workspaceUUID,\n\t\tUserID: userID,\n\t\tName: req.Name,\n\t\tDescription: req.Description,\n\t\tColor: req.Color,\n\t\tIcon: req.Icon,\n\t\t// Set default in a dedicated step to keep exactly one default per user.\n\t\tIsDefault: false,\n\t\tOrderPosition: 0, // Will be updated if needed\n\t}\n\n\tworkspace, err := h.service.CreateWorkspace(ctx, params)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to create workspace\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tif makeDefault {\n\t\tworkspace, err = h.setWorkspaceAsDefaultForUser(ctx, userID, workspace.Uuid)\n\t\tif err != nil {\n\t\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to set default workspace\")\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\t}\n\n\tresponse := WorkspaceResponse{\n\t\tUuid: workspace.Uuid,\n\t\tName: workspace.Name,\n\t\tDescription: workspace.Description,\n\t\tColor: workspace.Color,\n\t\tIcon: workspace.Icon,\n\t\tIsDefault: workspace.IsDefault,\n\t\tOrderPosition: workspace.OrderPosition,\n\t\tCreatedAt: workspace.CreatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t\tUpdatedAt: workspace.UpdatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t}\n\n\tjson.NewEncoder(w).Encode(response)\n}\n\n// getWorkspaceByUUID returns a workspace by its UUID\nfunc (h *ChatWorkspaceHandler) getWorkspaceByUUID(w http.ResponseWriter, r *http.Request) {\n\tworkspaceUUID := mux.Vars(r)[\"uuid\"]\n\tlog.Printf(\"🔍 DEBUG: getWorkspaceByUUID called with UUID=%s\", workspaceUUID)\n\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tlog.Printf(\"🔍 DEBUG: getWorkspaceByUUID userID=%d\", userID)\n\n\t// Check permission\n\thasPermission, err := h.service.HasWorkspacePermission(ctx, workspaceUUID, userID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to check workspace permission\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tif !hasPermission {\n\t\tapiErr := ErrAuthAccessDenied\n\t\tapiErr.Message = \"Access denied to workspace\"\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tworkspace, err := h.service.GetWorkspaceByUUID(ctx, workspaceUUID)\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\tapiErr := ErrResourceNotFound(\"Workspace\")\n\t\t\tapiErr.Message = \"Workspace not found with UUID: \" + workspaceUUID\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to get workspace\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tresponse := WorkspaceResponse{\n\t\tUuid: workspace.Uuid,\n\t\tName: workspace.Name,\n\t\tDescription: workspace.Description,\n\t\tColor: workspace.Color,\n\t\tIcon: workspace.Icon,\n\t\tIsDefault: workspace.IsDefault,\n\t\tOrderPosition: workspace.OrderPosition,\n\t\tCreatedAt: workspace.CreatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t\tUpdatedAt: workspace.UpdatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t}\n\n\tjson.NewEncoder(w).Encode(response)\n}\n\n// getWorkspacesByUserID returns all workspaces for the authenticated user\nfunc (h *ChatWorkspaceHandler) getWorkspacesByUserID(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tworkspaces, err := h.service.GetWorkspaceWithSessionCount(ctx, userID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to get workspaces\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tresponses := make([]WorkspaceResponse, 0)\n\tfor _, workspace := range workspaces {\n\t\tresponse := WorkspaceResponse{\n\t\t\tUuid: workspace.Uuid,\n\t\t\tName: workspace.Name,\n\t\t\tDescription: workspace.Description,\n\t\t\tColor: workspace.Color,\n\t\t\tIcon: workspace.Icon,\n\t\t\tIsDefault: workspace.IsDefault,\n\t\t\tOrderPosition: workspace.OrderPosition,\n\t\t\tSessionCount: workspace.SessionCount,\n\t\t\tCreatedAt: workspace.CreatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t\t\tUpdatedAt: workspace.UpdatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t\t}\n\t\tresponses = append(responses, response)\n\t}\n\n\tjson.NewEncoder(w).Encode(responses)\n}\n\n// updateWorkspace updates an existing workspace\nfunc (h *ChatWorkspaceHandler) updateWorkspace(w http.ResponseWriter, r *http.Request) {\n\tworkspaceUUID := mux.Vars(r)[\"uuid\"]\n\n\tvar req UpdateWorkspaceRequest\n\terr := json.NewDecoder(r.Body).Decode(&req)\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid request format\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Check permission\n\thasPermission, err := h.service.HasWorkspacePermission(ctx, workspaceUUID, userID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to check workspace permission\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tif !hasPermission {\n\t\tapiErr := ErrAuthAccessDenied\n\t\tapiErr.Message = \"Access denied to workspace\"\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tparams := sqlc_queries.UpdateWorkspaceParams{\n\t\tUuid: workspaceUUID,\n\t\tName: req.Name,\n\t\tDescription: req.Description,\n\t\tColor: req.Color,\n\t\tIcon: req.Icon,\n\t}\n\n\tworkspace, err := h.service.UpdateWorkspace(ctx, params)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to update workspace\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tresponse := WorkspaceResponse{\n\t\tUuid: workspace.Uuid,\n\t\tName: workspace.Name,\n\t\tDescription: workspace.Description,\n\t\tColor: workspace.Color,\n\t\tIcon: workspace.Icon,\n\t\tIsDefault: workspace.IsDefault,\n\t\tOrderPosition: workspace.OrderPosition,\n\t\tCreatedAt: workspace.CreatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t\tUpdatedAt: workspace.UpdatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t}\n\n\tjson.NewEncoder(w).Encode(response)\n}\n\n// updateWorkspaceOrder updates the order position of a workspace\nfunc (h *ChatWorkspaceHandler) updateWorkspaceOrder(w http.ResponseWriter, r *http.Request) {\n\tworkspaceUUID := mux.Vars(r)[\"uuid\"]\n\n\tvar req UpdateWorkspaceOrderRequest\n\terr := json.NewDecoder(r.Body).Decode(&req)\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid request format\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Check permission\n\thasPermission, err := h.service.HasWorkspacePermission(ctx, workspaceUUID, userID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to check workspace permission\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tif !hasPermission {\n\t\tapiErr := ErrAuthAccessDenied\n\t\tapiErr.Message = \"Access denied to workspace\"\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tparams := sqlc_queries.UpdateWorkspaceOrderParams{\n\t\tUuid: workspaceUUID,\n\t\tOrderPosition: req.OrderPosition,\n\t}\n\n\tworkspace, err := h.service.UpdateWorkspaceOrder(ctx, params)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to update workspace order\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tresponse := WorkspaceResponse{\n\t\tUuid: workspace.Uuid,\n\t\tName: workspace.Name,\n\t\tDescription: workspace.Description,\n\t\tColor: workspace.Color,\n\t\tIcon: workspace.Icon,\n\t\tIsDefault: workspace.IsDefault,\n\t\tOrderPosition: workspace.OrderPosition,\n\t\tCreatedAt: workspace.CreatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t\tUpdatedAt: workspace.UpdatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t}\n\n\tjson.NewEncoder(w).Encode(response)\n}\n\n// deleteWorkspace deletes a workspace\nfunc (h *ChatWorkspaceHandler) deleteWorkspace(w http.ResponseWriter, r *http.Request) {\n\tworkspaceUUID := mux.Vars(r)[\"uuid\"]\n\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Check permission\n\thasPermission, err := h.service.HasWorkspacePermission(ctx, workspaceUUID, userID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to check workspace permission\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tif !hasPermission {\n\t\tapiErr := ErrAuthAccessDenied\n\t\tapiErr.Message = \"Access denied to workspace\"\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Get workspace to check if it's default\n\tworkspace, err := h.service.GetWorkspaceByUUID(ctx, workspaceUUID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to get workspace\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Prevent deletion of default workspace\n\tif workspace.IsDefault {\n\t\tapiErr := ErrValidationInvalidInput(\"Cannot delete default workspace\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\terr = h.service.DeleteWorkspace(ctx, workspaceUUID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to delete workspace\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(map[string]string{\"message\": \"Workspace deleted successfully\"})\n}\n\n// setDefaultWorkspace sets a workspace as the default\nfunc (h *ChatWorkspaceHandler) setDefaultWorkspace(w http.ResponseWriter, r *http.Request) {\n\tworkspaceUUID := mux.Vars(r)[\"uuid\"]\n\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Check permission\n\thasPermission, err := h.service.HasWorkspacePermission(ctx, workspaceUUID, userID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to check workspace permission\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tif !hasPermission {\n\t\tapiErr := ErrAuthAccessDenied\n\t\tapiErr.Message = \"Access denied to workspace\"\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tworkspace, err := h.setWorkspaceAsDefaultForUser(ctx, userID, workspaceUUID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to set default workspace\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tresponse := WorkspaceResponse{\n\t\tUuid: workspace.Uuid,\n\t\tName: workspace.Name,\n\t\tDescription: workspace.Description,\n\t\tColor: workspace.Color,\n\t\tIcon: workspace.Icon,\n\t\tIsDefault: workspace.IsDefault,\n\t\tOrderPosition: workspace.OrderPosition,\n\t\tCreatedAt: workspace.CreatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t\tUpdatedAt: workspace.UpdatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t}\n\n\tjson.NewEncoder(w).Encode(response)\n}\n\nfunc (h *ChatWorkspaceHandler) setWorkspaceAsDefaultForUser(ctx context.Context, userID int32, workspaceUUID string) (sqlc_queries.ChatWorkspace, error) {\n\tworkspaces, err := h.service.GetWorkspacesByUserID(ctx, userID)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatWorkspace{}, err\n\t}\n\n\tfor _, ws := range workspaces {\n\t\tif ws.IsDefault && ws.Uuid != workspaceUUID {\n\t\t\t_, err = h.service.SetDefaultWorkspace(ctx, sqlc_queries.SetDefaultWorkspaceParams{\n\t\t\t\tUuid: ws.Uuid,\n\t\t\t\tIsDefault: false,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\treturn sqlc_queries.ChatWorkspace{}, err\n\t\t\t}\n\t\t}\n\t}\n\n\treturn h.service.SetDefaultWorkspace(ctx, sqlc_queries.SetDefaultWorkspaceParams{\n\t\tUuid: workspaceUUID,\n\t\tIsDefault: true,\n\t})\n}\n\n// ensureDefaultWorkspace ensures the user has a default workspace\nfunc (h *ChatWorkspaceHandler) ensureDefaultWorkspace(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tworkspace, err := h.service.EnsureDefaultWorkspaceExists(ctx, userID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to ensure default workspace\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tresponse := WorkspaceResponse{\n\t\tUuid: workspace.Uuid,\n\t\tName: workspace.Name,\n\t\tDescription: workspace.Description,\n\t\tColor: workspace.Color,\n\t\tIcon: workspace.Icon,\n\t\tIsDefault: workspace.IsDefault,\n\t\tOrderPosition: workspace.OrderPosition,\n\t\tCreatedAt: workspace.CreatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t\tUpdatedAt: workspace.UpdatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t}\n\n\tjson.NewEncoder(w).Encode(response)\n}\n\ntype CreateSessionInWorkspaceRequest struct {\n\tTopic string `json:\"topic\"`\n\tModel string `json:\"model\"`\n\tDefaultSystemPrompt string `json:\"defaultSystemPrompt\"`\n}\n\n// createSessionInWorkspace creates a new session in a specific workspace\nfunc (h *ChatWorkspaceHandler) createSessionInWorkspace(w http.ResponseWriter, r *http.Request) {\n\tworkspaceUUID := mux.Vars(r)[\"uuid\"]\n\n\tvar req CreateSessionInWorkspaceRequest\n\terr := json.NewDecoder(r.Body).Decode(&req)\n\tif err != nil {\n\t\tapiErr := ErrValidationInvalidInput(\"Invalid request format\")\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Check workspace permission\n\thasPermission, err := h.service.HasWorkspacePermission(ctx, workspaceUUID, userID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to check workspace permission\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tif !hasPermission {\n\t\tapiErr := ErrAuthAccessDenied\n\t\tapiErr.Message = \"Access denied to workspace\"\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Get workspace\n\tworkspace, err := h.service.GetWorkspaceByUUID(ctx, workspaceUUID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to get workspace\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Create session\n\tsessionUUID := uuid.New().String()\n\tsessionService := NewChatSessionService(h.service.q)\n\tactiveSessionService := NewUserActiveChatSessionService(h.service.q)\n\n\tsessionParams := sqlc_queries.CreateChatSessionInWorkspaceParams{\n\t\tUserID: userID,\n\t\tUuid: sessionUUID,\n\t\tTopic: req.Topic,\n\t\tCreatedAt: time.Now(),\n\t\tActive: true,\n\t\tMaxLength: 10,\n\t\tModel: req.Model,\n\t\tWorkspaceID: sql.NullInt32{Int32: workspace.ID, Valid: true},\n\t}\n\n\tsession, err := sessionService.q.CreateChatSessionInWorkspace(ctx, sessionParams)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to create session in workspace\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t_, err = sessionService.EnsureDefaultSystemPrompt(ctx, session.Uuid, userID, req.DefaultSystemPrompt)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to create default system prompt\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Set as active session (use unified approach)\n\t_, err = activeSessionService.UpsertActiveSession(ctx, userID, &workspace.ID, sessionUUID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to set active session\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tjson.NewEncoder(w).Encode(map[string]interface{}{\n\t\t\"uuid\": session.Uuid,\n\t\t\"topic\": session.Topic,\n\t\t\"model\": session.Model,\n\t\t\"artifactEnabled\": session.ArtifactEnabled,\n\t\t\"workspaceUuid\": workspaceUUID,\n\t\t\"createdAt\": session.CreatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t})\n}\n\n// getSessionsByWorkspace returns all sessions in a specific workspace\nfunc (h *ChatWorkspaceHandler) getSessionsByWorkspace(w http.ResponseWriter, r *http.Request) {\n\tworkspaceUUID := mux.Vars(r)[\"uuid\"]\n\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.DebugInfo = err.Error()\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Check workspace permission\n\thasPermission, err := h.service.HasWorkspacePermission(ctx, workspaceUUID, userID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to check workspace permission\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\tif !hasPermission {\n\t\tapiErr := ErrAuthAccessDenied\n\t\tapiErr.Message = \"Access denied to workspace\"\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Get workspace\n\tworkspace, err := h.service.GetWorkspaceByUUID(ctx, workspaceUUID)\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to get workspace\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\t// Get sessions in workspace\n\tsessionService := NewChatSessionService(h.service.q)\n\tsessions, err := sessionService.q.GetSessionsByWorkspaceID(ctx, sql.NullInt32{Int32: workspace.ID, Valid: true})\n\tif err != nil {\n\t\tapiErr := WrapError(MapDatabaseError(err), \"Failed to get sessions\")\n\t\tRespondWithAPIError(w, apiErr)\n\t\treturn\n\t}\n\n\tsessionResponses := make([]map[string]interface{}, 0)\n\tfor _, session := range sessions {\n\t\tsessionResponse := map[string]interface{}{\n\t\t\t\"uuid\": session.Uuid,\n\t\t\t\"title\": session.Topic, // Use \"title\" to match the original API\n\t\t\t\"isEdit\": false,\n\t\t\t\"model\": session.Model,\n\t\t\t\"workspaceUuid\": workspaceUUID,\n\t\t\t\"maxLength\": session.MaxLength,\n\t\t\t\"temperature\": session.Temperature,\n\t\t\t\"maxTokens\": session.MaxTokens,\n\t\t\t\"topP\": session.TopP,\n\t\t\t\"n\": session.N,\n\t\t\t\"debug\": session.Debug,\n\t\t\t\"summarizeMode\": session.SummarizeMode,\n\t\t\t\"exploreMode\": session.ExploreMode,\n\t\t\t\"artifactEnabled\": session.ArtifactEnabled,\n\t\t\t\"createdAt\": session.CreatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t\t\t\"updatedAt\": session.UpdatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t\t}\n\t\tsessionResponses = append(sessionResponses, sessionResponse)\n\t}\n\n\tjson.NewEncoder(w).Encode(sessionResponses)\n}\n\n// autoMigrateLegacySessions automatically detects and migrates legacy sessions without workspace_id\nfunc (h *ChatWorkspaceHandler) autoMigrateLegacySessions(w http.ResponseWriter, r *http.Request) {\n\tctx := r.Context()\n\tuserID, err := getUserID(ctx)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Check if user has any legacy sessions (sessions without workspace_id)\n\tsessionService := NewChatSessionService(h.service.q)\n\tlegacySessions, err := sessionService.q.GetSessionsWithoutWorkspace(ctx, userID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to check for legacy sessions\"))\n\t\treturn\n\t}\n\n\tresponse := map[string]interface{}{\n\t\t\"hasLegacySessions\": len(legacySessions) > 0,\n\t\t\"migratedSessions\": 0,\n\t}\n\n\t// If no legacy sessions, return early\n\tif len(legacySessions) == 0 {\n\t\tw.Header().Set(\"Content-Type\", \"application/json\")\n\t\tjson.NewEncoder(w).Encode(response)\n\t\treturn\n\t}\n\n\t// Ensure default workspace exists\n\tdefaultWorkspace, err := h.service.EnsureDefaultWorkspaceExists(ctx, userID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to ensure default workspace\"))\n\t\treturn\n\t}\n\n\t// Migrate all legacy sessions to default workspace\n\terr = h.service.MigrateSessionsToDefaultWorkspace(ctx, userID, defaultWorkspace.ID)\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(MapDatabaseError(err), \"Failed to migrate legacy sessions\"))\n\t\treturn\n\t}\n\n\t// Also migrate any legacy active sessions\n\tactiveSessionService := NewUserActiveChatSessionService(h.service.q)\n\tlegacyActiveSessions, err := activeSessionService.q.GetAllUserActiveSessions(ctx, userID)\n\tif err == nil {\n\t\tfor _, activeSession := range legacyActiveSessions {\n\t\t\tif !activeSession.WorkspaceID.Valid {\n\t\t\t\t// This is a legacy global active session, migrate it to default workspace\n\t\t\t\t_, _ = activeSessionService.UpsertActiveSession(ctx, userID, &defaultWorkspace.ID, activeSession.ChatSessionUuid)\n\t\t\t\t// Delete the old global active session by setting workspace to NULL and deleting\n\t\t\t\t_ = activeSessionService.DeleteActiveSession(ctx, userID, nil)\n\t\t\t}\n\t\t}\n\t}\n\n\tresponse[\"migratedSessions\"] = len(legacySessions)\n\tresponse[\"defaultWorkspace\"] = WorkspaceResponse{\n\t\tUuid: defaultWorkspace.Uuid,\n\t\tName: defaultWorkspace.Name,\n\t\tDescription: defaultWorkspace.Description,\n\t\tColor: defaultWorkspace.Color,\n\t\tIcon: defaultWorkspace.Icon,\n\t\tIsDefault: defaultWorkspace.IsDefault,\n\t\tOrderPosition: defaultWorkspace.OrderPosition,\n\t\tCreatedAt: defaultWorkspace.CreatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t\tUpdatedAt: defaultWorkspace.UpdatedAt.Format(\"2006-01-02T15:04:05Z\"),\n\t}\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tjson.NewEncoder(w).Encode(response)\n}\n" }, { "path": "api/chat_workspace_service.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"log\"\n\n\t\"github.com/google/uuid\"\n\t\"github.com/rotisserie/eris\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// ChatWorkspaceService provides methods for interacting with chat workspaces.\ntype ChatWorkspaceService struct {\n\tq *sqlc_queries.Queries\n}\n\n// NewChatWorkspaceService creates a new ChatWorkspaceService.\nfunc NewChatWorkspaceService(q *sqlc_queries.Queries) *ChatWorkspaceService {\n\treturn &ChatWorkspaceService{q: q}\n}\n\n// CreateWorkspace creates a new workspace.\nfunc (s *ChatWorkspaceService) CreateWorkspace(ctx context.Context, params sqlc_queries.CreateWorkspaceParams) (sqlc_queries.ChatWorkspace, error) {\n\tworkspace, err := s.q.CreateWorkspace(ctx, params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatWorkspace{}, eris.Wrap(err, \"failed to create workspace\")\n\t}\n\treturn workspace, nil\n}\n\n// GetWorkspaceByUUID returns a workspace by UUID.\nfunc (s *ChatWorkspaceService) GetWorkspaceByUUID(ctx context.Context, workspaceUUID string) (sqlc_queries.ChatWorkspace, error) {\n\tworkspace, err := s.q.GetWorkspaceByUUID(ctx, workspaceUUID)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatWorkspace{}, eris.Wrap(err, \"failed to retrieve workspace\")\n\t}\n\treturn workspace, nil\n}\n\n// GetWorkspacesByUserID returns all workspaces for a user.\nfunc (s *ChatWorkspaceService) GetWorkspacesByUserID(ctx context.Context, userID int32) ([]sqlc_queries.ChatWorkspace, error) {\n\tworkspaces, err := s.q.GetWorkspacesByUserID(ctx, userID)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to retrieve workspaces\")\n\t}\n\treturn workspaces, nil\n}\n\n// GetWorkspaceWithSessionCount returns all workspaces with session counts for a user.\nfunc (s *ChatWorkspaceService) GetWorkspaceWithSessionCount(ctx context.Context, userID int32) ([]sqlc_queries.GetWorkspaceWithSessionCountRow, error) {\n\tworkspaces, err := s.q.GetWorkspaceWithSessionCount(ctx, userID)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"failed to retrieve workspaces with session count\")\n\t}\n\treturn workspaces, nil\n}\n\n// UpdateWorkspace updates an existing workspace.\nfunc (s *ChatWorkspaceService) UpdateWorkspace(ctx context.Context, params sqlc_queries.UpdateWorkspaceParams) (sqlc_queries.ChatWorkspace, error) {\n\tworkspace, err := s.q.UpdateWorkspace(ctx, params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatWorkspace{}, eris.Wrap(err, \"failed to update workspace\")\n\t}\n\treturn workspace, nil\n}\n\n// UpdateWorkspaceOrder updates the order position of a workspace.\nfunc (s *ChatWorkspaceService) UpdateWorkspaceOrder(ctx context.Context, params sqlc_queries.UpdateWorkspaceOrderParams) (sqlc_queries.ChatWorkspace, error) {\n\tworkspace, err := s.q.UpdateWorkspaceOrder(ctx, params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatWorkspace{}, eris.Wrap(err, \"failed to update workspace order\")\n\t}\n\treturn workspace, nil\n}\n\n// DeleteWorkspace deletes a workspace by UUID.\nfunc (s *ChatWorkspaceService) DeleteWorkspace(ctx context.Context, workspaceUUID string) error {\n\terr := s.q.DeleteWorkspace(ctx, workspaceUUID)\n\tif err != nil {\n\t\treturn eris.Wrap(err, \"failed to delete workspace\")\n\t}\n\treturn nil\n}\n\n// GetDefaultWorkspaceByUserID returns the default workspace for a user.\nfunc (s *ChatWorkspaceService) GetDefaultWorkspaceByUserID(ctx context.Context, userID int32) (sqlc_queries.ChatWorkspace, error) {\n\tworkspace, err := s.q.GetDefaultWorkspaceByUserID(ctx, userID)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatWorkspace{}, eris.Wrap(err, \"failed to retrieve default workspace\")\n\t}\n\treturn workspace, nil\n}\n\n// SetDefaultWorkspace sets a workspace as the default.\nfunc (s *ChatWorkspaceService) SetDefaultWorkspace(ctx context.Context, params sqlc_queries.SetDefaultWorkspaceParams) (sqlc_queries.ChatWorkspace, error) {\n\tworkspace, err := s.q.SetDefaultWorkspace(ctx, params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatWorkspace{}, eris.Wrap(err, \"failed to set default workspace\")\n\t}\n\treturn workspace, nil\n}\n\n// CreateDefaultWorkspace creates a default workspace for a user.\nfunc (s *ChatWorkspaceService) CreateDefaultWorkspace(ctx context.Context, userID int32) (sqlc_queries.ChatWorkspace, error) {\n\tworkspaceUUID := uuid.New().String()\n\tparams := sqlc_queries.CreateDefaultWorkspaceParams{\n\t\tUuid: workspaceUUID,\n\t\tUserID: userID,\n\t}\n\tworkspace, err := s.q.CreateDefaultWorkspace(ctx, params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatWorkspace{}, eris.Wrap(err, \"failed to create default workspace\")\n\t}\n\treturn workspace, nil\n}\n\n// EnsureDefaultWorkspaceExists ensures a user has a default workspace, creating one if needed.\nfunc (s *ChatWorkspaceService) EnsureDefaultWorkspaceExists(ctx context.Context, userID int32) (sqlc_queries.ChatWorkspace, error) {\n\t// Try to get existing default workspace\n\tworkspace, err := s.GetDefaultWorkspaceByUserID(ctx, userID)\n\tif err != nil {\n\t\t// If no default workspace exists, create one\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\treturn s.CreateDefaultWorkspace(ctx, userID)\n\t\t}\n\t\treturn sqlc_queries.ChatWorkspace{}, err\n\t}\n\treturn workspace, nil\n}\n\n// HasWorkspacePermission checks if a user has permission to access a workspace.\nfunc (s *ChatWorkspaceService) HasWorkspacePermission(ctx context.Context, workspaceUUID string, userID int32) (bool, error) {\n\tlog.Printf(\"🔍 DEBUG: Checking permission for workspace=%s, user=%d\", workspaceUUID, userID)\n\n\tresult, err := s.q.HasWorkspacePermission(ctx, sqlc_queries.HasWorkspacePermissionParams{\n\t\tUuid: workspaceUUID,\n\t\tUserID: userID,\n\t})\n\tif err != nil {\n\t\tlog.Printf(\"❌ DEBUG: Permission check failed: %v\", err)\n\t\treturn false, eris.Wrap(err, \"failed to check workspace permission\")\n\t}\n\n\tlog.Printf(\"✅ DEBUG: Permission result=%t for workspace=%s, user=%d\", result, workspaceUUID, userID)\n\treturn result, nil\n}\n\n// MigrateSessionsToDefaultWorkspace migrates all sessions without workspace to default workspace.\nfunc (s *ChatWorkspaceService) MigrateSessionsToDefaultWorkspace(ctx context.Context, userID int32, workspaceID int32) error {\n\terr := s.q.MigrateSessionsToDefaultWorkspace(ctx, sqlc_queries.MigrateSessionsToDefaultWorkspaceParams{\n\t\tUserID: userID,\n\t\tWorkspaceID: sql.NullInt32{Int32: workspaceID, Valid: true},\n\t})\n\tif err != nil {\n\t\treturn eris.Wrap(err, \"failed to migrate sessions to default workspace\")\n\t}\n\treturn nil\n}\n" }, { "path": "api/constants.go", "content": "// Package main provides constants used throughout the chat application.\n// This file contains all magic numbers, timeouts, and configuration values\n// to improve code maintainability and avoid scattered constants.\npackage main\n\nimport \"time\"\n\n// API and request constants\nconst (\n\t// Timeout settings\n\tDefaultRequestTimeout = 5 * time.Minute\n\n\t// Loop limits and safety guards\n\tMaxStreamingLoopIterations = 10000\n\n\t// Content buffering and flushing\n\tSmallAnswerThreshold = 200\n\tFlushCharacterThreshold = 500\n\tTestPrefixLength = 16\n\n\t// Pagination\n\tDefaultPageSize = 200\n\tMaxHistoryItems = 10000\n\n\t// Rate limiting\n\tDefaultPageLimit = 30\n\n\t// Test constants\n\tTestDemoPrefix = \"test_demo_bestqa\"\n\n\t// Service constants\n\tDefaultMaxLength = 10\n\tDefaultTemperature = 0.7\n\tDefaultMaxTokens = 4096\n\tDefaultTopP = 1.0\n\tDefaultN = 1\n\tRequestTimeoutSeconds = 10\n\tTokenEstimateRatio = 4\n\tSummarizeThreshold = 300\n\tDefaultSystemPromptText = \"You are a helpful, concise assistant. Ask clarifying questions when needed. Provide accurate answers with short reasoning and actionable steps. If unsure, say so and suggest how to verify.\"\n)\n\n// Error message constants\nconst (\n\tErrorStreamUnsupported = \"Streaming unsupported by client\"\n\tErrorNoContent = \"no content in answer\"\n\tErrorEndOfStream = \"End of stream reached\"\n\tErrorDoneBreak = \"DONE break\"\n)\n\n// HTTP constants\nconst (\n\tContentTypeJSON = \"application/json\"\n\tAcceptEventStream = \"text/event-stream\"\n\tCacheControlNoCache = \"no-cache\"\n\tConnectionKeepAlive = \"keep-alive\"\n)\n" }, { "path": "api/embed_debug_test.go", "content": "package main\n\nimport \"testing\"\n\nfunc TestEmbedInstructions(t *testing.T) {\n\tif artifactInstructionText == \"\" {\n\t\tt.Fatalf(\"artifactInstructionText is empty\")\n\t}\n}\n" }, { "path": "api/errors.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"log\"\n\t\"net/http\"\n\t\"sort\"\n\t\"strings\"\n\n\t\"github.com/jackc/pgconn\"\n)\n\n// APIError represents a standardized error response for the API\n// It includes both user-facing and internal debugging information\ntype APIError struct {\n\tHTTPCode int `json:\"-\"` // HTTP status code (not exposed in response)\n\tCode string `json:\"code\"` // Application-specific error code following format: DOMAIN_NNN\n\tMessage string `json:\"message\"` // Human-readable message for end users\n\tDetail string `json:\"detail,omitempty\"` // Optional error details for debugging\n\tDebugInfo string `json:\"-\"` // Internal debugging info (not exposed in responses)\n}\n\n// NewAPIError creates a new APIError with the given parameters\nfunc NewAPIError(httpCode int, code, message string) APIError {\n\treturn APIError{\n\t\tHTTPCode: httpCode,\n\t\tCode: code,\n\t\tMessage: message,\n\t}\n}\n\n// withMessage adds a message to an APIError\nfunc (e APIError) WithMessage(message string) APIError {\n\te.Message = message\n\treturn e\n}\n\n// WithMessage adds detail to an APIError\nfunc (e APIError) WithDetail(detail string) APIError {\n\te.Detail = detail\n\treturn e\n}\n\n// WithDebugInfo adds debug info to an APIError\nfunc (e APIError) WithDebugInfo(debugInfo string) APIError {\n\te.DebugInfo = debugInfo\n\treturn e\n}\n\nfunc (e APIError) Error() string {\n\treturn fmt.Sprintf(\"[%s] %s %s\", e.Code, e.Message, e.Detail)\n}\n\n// Error code prefixes by domain\nconst (\n\tErrAuth = \"AUTH\" // Authentication/Authorization errors (100-199)\n\tErrValidation = \"VALD\" // Validation errors (200-299)\n\tErrResource = \"RES\" // Resource-related errors (300-399)\n\tErrDatabase = \"DB\" // Database errors (400-499)\n\tErrExternal = \"EXT\" // External service errors (500-599)\n\tErrInternal = \"INTN\" // Internal application errors (600-699)\n\tErrModel = \"MODEL\" // Model related errors (700-799)\n)\n\n// Error code ranges:\n// - Each domain has 100 codes available (000-099)\n// - Codes should be sequential within each domain\n// - New errors should use the next available code in their domain\n\n// Define external service errors\nvar (\n\tErrExternalTimeout = APIError{\n\t\tHTTPCode: http.StatusGatewayTimeout,\n\t\tCode: ErrExternal + \"_001\",\n\t\tMessage: \"External service timed out\",\n\t}\n\n\tErrExternalUnavailable = APIError{\n\t\tHTTPCode: http.StatusServiceUnavailable,\n\t\tCode: ErrExternal + \"_002\",\n\t\tMessage: \"External service unavailable\",\n\t}\n)\n\n// Define all API errors\nvar (\n\t// Auth errors\n\tErrAuthInvalidCredentials = APIError{\n\t\tHTTPCode: http.StatusUnauthorized,\n\t\tCode: ErrAuth + \"_001\",\n\t\tMessage: \"Invalid credentials\",\n\t}\n\tErrAuthExpiredToken = APIError{\n\t\tHTTPCode: http.StatusUnauthorized,\n\t\tCode: ErrAuth + \"_002\",\n\t\tMessage: \"Token has expired\",\n\t}\n\tErrAuthAdminRequired = APIError{\n\t\tHTTPCode: http.StatusForbidden,\n\t\tCode: ErrAuth + \"_003\",\n\t\tMessage: \"Admin privileges required\",\n\t}\n\tErrAuthInvalidEmailOrPassword = APIError{\n\t\tHTTPCode: http.StatusForbidden,\n\t\tCode: ErrAuth + \"_004\",\n\t\tMessage: \"invalid email or password\",\n\t}\n\tErrAuthAccessDenied = APIError{\n\t\tHTTPCode: http.StatusForbidden,\n\t\tCode: ErrAuth + \"_005\",\n\t\tMessage: \"Access denied\",\n\t}\n\t// Resource errors\n\tErrResourceNotFoundGeneric = APIError{\n\t\tHTTPCode: http.StatusNotFound,\n\t\tCode: ErrResource + \"_001\",\n\t\tMessage: \"Resource not found\",\n\t}\n\tErrResourceAlreadyExistsGeneric = APIError{\n\t\tHTTPCode: http.StatusConflict,\n\t\tCode: ErrResource + \"_002\",\n\t\tMessage: \"Resource already exists\",\n\t}\n\tErrChatSessionNotFound = APIError{\n\t\tHTTPCode: http.StatusNotFound,\n\t\tCode: ErrResource + \"_004\",\n\t\tMessage: \"Chat session not found\",\n\t}\n\tErrChatMessageNotFound = APIError{\n\t\tHTTPCode: http.StatusNotFound,\n\t\tCode: ErrResource + \"_007\",\n\t\tMessage: \"Chat message not found\",\n\t}\n\tErrChatStreamFailed = APIError{\n\t\tHTTPCode: http.StatusInternalServerError,\n\t\tCode: ErrInternal + \"_004\",\n\t\tMessage: \"Failed to stream chat response\",\n\t}\n\tErrChatRequestFailed = APIError{\n\t\tHTTPCode: http.StatusInternalServerError,\n\t\tCode: ErrInternal + \"_005\",\n\t\tMessage: \"Failed to make chat request\",\n\t}\n\tErrChatFileNotFound = APIError{\n\t\tHTTPCode: http.StatusNotFound,\n\t\tCode: ErrResource + \"_005\",\n\t\tMessage: \"Chat file not found\",\n\t}\n\tErrChatModelNotFound = APIError{\n\t\tHTTPCode: http.StatusNotFound,\n\t\tCode: ErrResource + \"_006\",\n\t\tMessage: \"Chat model not found\",\n\t}\n\tErrChatFileTooLarge = APIError{\n\t\tHTTPCode: http.StatusBadRequest,\n\t\tCode: ErrValidation + \"_002\",\n\t\tMessage: \"File too large\",\n\t}\n\tErrChatFileInvalidType = APIError{\n\t\tHTTPCode: http.StatusBadRequest,\n\t\tCode: ErrValidation + \"_003\",\n\t\tMessage: \"Invalid file type\",\n\t}\n\tErrChatSessionInvalid = APIError{\n\t\tHTTPCode: http.StatusBadRequest,\n\t\tCode: ErrValidation + \"_004\",\n\t\tMessage: \"Invalid chat session\",\n\t}\n\tErrTooManyRequests = APIError{\n\t\tHTTPCode: http.StatusTooManyRequests,\n\t\tCode: ErrResource + \"_003\",\n\t\tMessage: \"Rate limit exceeded\",\n\t\tDetail: \"Too many requests in the given time period\",\n\t}\n\n\t// Validation errors\n\tErrValidationInvalidInputGeneric = APIError{\n\t\tHTTPCode: http.StatusBadRequest,\n\t\tCode: ErrValidation + \"_001\",\n\t\tMessage: \"Invalid input\",\n\t}\n\n\t// Database errors\n\tErrDatabaseQuery = APIError{\n\t\tHTTPCode: http.StatusInternalServerError,\n\t\tCode: ErrDatabase + \"_001\",\n\t\tMessage: \"Database query failed\",\n\t\tDebugInfo: \"Database operation failed - check logs for details\",\n\t}\n\tErrDatabaseConnection = APIError{\n\t\tHTTPCode: http.StatusServiceUnavailable,\n\t\tCode: ErrDatabase + \"_002\",\n\t\tMessage: \"Database connection failed\",\n\t\tDebugInfo: \"Could not connect to database - check connection settings\",\n\t}\n\tErrDatabaseForeignKey = APIError{\n\t\tHTTPCode: http.StatusBadRequest,\n\t\tCode: ErrDatabase + \"_003\",\n\t\tMessage: \"Referenced resource does not exist\",\n\t\tDebugInfo: \"Foreign key violation\",\n\t}\n\t// model related errors\n\tErrSystemMessageError = APIError{\n\t\tHTTPCode: http.StatusBadRequest,\n\t\tCode: ErrModel + \"_001\",\n\t\tMessage: \"Usage error, system message input, not user input\",\n\t}\n\tErrClaudeStreamFailed = APIError{\n\t\tHTTPCode: http.StatusInternalServerError,\n\t\tCode: ErrModel + \"_002\",\n\t\tMessage: \"Failed to stream Claude response\",\n\t}\n\tErrClaudeRequestFailed = APIError{\n\t\tHTTPCode: http.StatusInternalServerError,\n\t\tCode: ErrModel + \"_003\",\n\t\tMessage: \"Failed to make Claude request\",\n\t}\n\tErrClaudeInvalidResponse = APIError{\n\t\tHTTPCode: http.StatusInternalServerError,\n\t\tCode: ErrModel + \"_004\",\n\t\tMessage: \"Invalid response from Claude API\",\n\t}\n\tErrClaudeResponseFaild = APIError{\n\t\tHTTPCode: http.StatusInternalServerError,\n\t\tCode: ErrModel + \"_005\",\n\t\tMessage: \"Failed to stream Claude response\",\n\t}\n\n\t// OpenAI specific errors\n\tErrOpenAIStreamFailed = APIError{\n\t\tHTTPCode: http.StatusInternalServerError,\n\t\tCode: ErrModel + \"_006\",\n\t\tMessage: \"Failed to stream OpenAI response\",\n\t}\n\tErrOpenAIRequestFailed = APIError{\n\t\tHTTPCode: http.StatusInternalServerError,\n\t\tCode: ErrModel + \"_007\",\n\t\tMessage: \"Failed to make OpenAI request\",\n\t}\n\tErrOpenAIInvalidResponse = APIError{\n\t\tHTTPCode: http.StatusInternalServerError,\n\t\tCode: ErrModel + \"_008\",\n\t\tMessage: \"Invalid response from OpenAI API\",\n\t}\n\tErrOpenAIConfigFailed = APIError{\n\t\tHTTPCode: http.StatusInternalServerError,\n\t\tCode: ErrModel + \"_009\",\n\t\tMessage: \"Failed to configure OpenAI client\",\n\t}\n\n\t// Internal errors\n\tErrInternalUnexpected = APIError{\n\t\tHTTPCode: http.StatusInternalServerError,\n\t\tCode: ErrInternal + \"_001\",\n\t\tMessage: \"An unexpected error occurred\",\n\t\tDebugInfo: \"Unexpected internal error - check logs for stack trace\",\n\t}\n)\n\n// Helper functions to create specific errors with dynamic content\nfunc ErrResourceNotFound(resource string) APIError {\n\terr := ErrResourceNotFoundGeneric\n\terr.Message = resource + \" not found\"\n\treturn err\n}\n\nfunc ErrResourceAlreadyExists(resource string) APIError {\n\terr := ErrResourceAlreadyExistsGeneric\n\terr.Message = resource + \" already exists\"\n\treturn err\n}\n\nfunc ErrValidationInvalidInput(detail string) APIError {\n\terr := ErrValidationInvalidInputGeneric\n\terr.Detail = detail\n\treturn err\n}\n\n// RespondWithAPIError writes an APIError response to the client\n// It:\n// - Sets the appropriate HTTP status code\n// - Returns a JSON response with error details\n// - Logs the error with debug info\nfunc RespondWithAPIError(w http.ResponseWriter, err APIError) {\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tw.WriteHeader(err.HTTPCode)\n\n\t// Error response structure\n\tresponse := struct {\n\t\tCode string `json:\"code\"` // Application error code\n\t\tMessage string `json:\"message\"` // Human-readable error message\n\t\tDetail string `json:\"detail,omitempty\"` // Additional error details\n\t}{\n\t\tCode: err.Code,\n\t\tMessage: err.Message,\n\t\tDetail: err.Detail + \" \" + err.DebugInfo,\n\t}\n\n\t// Log error with debug info if available\n\tif err.DebugInfo != \"\" {\n\t\tlog.Printf(\"Error [%s]: %s - %s\", err.Code, err.Message, err.DebugInfo)\n\t}\n\n\t// Write JSON response\n\tif err := json.NewEncoder(w).Encode(response); err != nil {\n\t\tlog.Printf(\"Failed to write error response: %v\", err)\n\t}\n}\n\nfunc MapDatabaseError(err error) error {\n\t// Map common database errors to appropriate application errors\n\tif errors.Is(err, sql.ErrNoRows) {\n\t\treturn ErrResourceNotFound(\"Record\")\n\t}\n\n\t// Check for connection errors\n\tif strings.Contains(err.Error(), \"connection refused\") ||\n\t\tstrings.Contains(err.Error(), \"no such host\") ||\n\t\tstrings.Contains(err.Error(), \"connection reset by peer\") {\n\t\tdbErr := ErrDatabaseConnection\n\t\tdbErr.DebugInfo = err.Error()\n\t\treturn dbErr\n\t}\n\n\t// Check for other specific database errors\n\tvar pgErr *pgconn.PgError\n\tif errors.As(err, &pgErr) {\n\t\tswitch pgErr.Code {\n\t\tcase \"23505\": // Unique violation\n\t\t\treturn ErrResourceAlreadyExists(\"Record\")\n\t\tcase \"23503\": // Foreign key violation\n\t\t\tdbErr := ErrDatabaseForeignKey\n\t\t\tdbErr.DebugInfo = fmt.Sprintf(\"Foreign key violation: %s\", pgErr.Detail)\n\t\t\treturn dbErr\n\t\tcase \"42P01\": // Undefined table\n\t\t\tdbErr := ErrDatabaseQuery\n\t\t\tdbErr.Message = \"Database schema error\"\n\t\t\tdbErr.DebugInfo = fmt.Sprintf(\"Table does not exist: %s\", pgErr.Detail)\n\t\t\treturn dbErr\n\t\tcase \"42703\": // Undefined column\n\t\t\tdbErr := ErrDatabaseQuery\n\t\t\tdbErr.Message = \"Database schema error\"\n\t\t\tdbErr.DebugInfo = fmt.Sprintf(\"Column does not exist: %s\", pgErr.Detail)\n\t\t\treturn dbErr\n\t\tcase \"53300\": // Too many connections\n\t\t\tdbErr := ErrDatabaseConnection\n\t\t\tdbErr.Message = \"Database connection limit reached\"\n\t\t\tdbErr.DebugInfo = pgErr.Detail\n\t\t\treturn dbErr\n\t\t}\n\t}\n\n\t// Log the unhandled database error\n\tlog.Printf(\"Unhandled database error: %v\", err)\n\n\t// Return generic database error\n\tdbErr := ErrDatabaseQuery\n\tdbErr.DebugInfo = err.Error()\n\treturn dbErr\n}\n\n// ErrorCatalog holds all error codes for documentation purposes\nvar ErrorCatalog = map[string]APIError{\n\t// Auth errors\n\tErrAuthInvalidCredentials.Code: ErrAuthInvalidCredentials,\n\tErrAuthExpiredToken.Code: ErrAuthExpiredToken,\n\tErrAuthAdminRequired.Code: ErrAuthAdminRequired,\n\n\t// Resource errors\n\tErrResourceNotFoundGeneric.Code: ErrResourceNotFoundGeneric,\n\tErrResourceAlreadyExistsGeneric.Code: ErrResourceAlreadyExistsGeneric,\n\tErrTooManyRequests.Code: ErrTooManyRequests,\n\n\t// Validation errors\n\tErrValidationInvalidInputGeneric.Code: ErrValidationInvalidInputGeneric,\n\n\t// Database errors\n\tErrDatabaseQuery.Code: ErrDatabaseQuery,\n\tErrDatabaseConnection.Code: ErrDatabaseConnection,\n\tErrDatabaseForeignKey.Code: ErrDatabaseForeignKey,\n\n\t// External service errors\n\tErrExternalTimeout.Code: ErrExternalTimeout,\n\tErrExternalUnavailable.Code: ErrExternalUnavailable,\n\n\t// External service errors\n\tErrExternalTimeout.Code: ErrExternalTimeout,\n\tErrExternalUnavailable.Code: ErrExternalUnavailable,\n\n\t// Internal errors\n\tErrInternalUnexpected.Code: ErrInternalUnexpected,\n\tErrInternal + \"_002\": {HTTPCode: http.StatusGatewayTimeout, Code: ErrInternal + \"_002\", Message: \"Request timed out\"},\n\tErrInternal + \"_003\": {HTTPCode: http.StatusRequestTimeout, Code: ErrInternal + \"_003\", Message: \"Request was canceled\"},\n\tErrInternal + \"_004\": ErrChatStreamFailed,\n\tErrInternal + \"_005\": ErrChatRequestFailed,\n\tErrResource + \"_004\": ErrChatSessionNotFound,\n\tErrResource + \"_005\": ErrChatFileNotFound,\n\tErrResource + \"_006\": ErrChatModelNotFound,\n\tErrResource + \"_007\": ErrChatMessageNotFound,\n\n\t// model related errors\n\tErrModel + \"_001\": ErrSystemMessageError,\n\tErrModel + \"_002\": ErrClaudeStreamFailed,\n\tErrModel + \"_003\": ErrClaudeRequestFailed,\n\tErrModel + \"_004\": ErrClaudeInvalidResponse,\n\tErrModel + \"_005\": ErrClaudeResponseFaild,\n\tErrModel + \"_006\": ErrOpenAIStreamFailed,\n\tErrModel + \"_007\": ErrOpenAIRequestFailed,\n\tErrModel + \"_008\": ErrOpenAIInvalidResponse,\n\tErrModel + \"_009\": ErrOpenAIConfigFailed,\n}\n\n// WrapError converts a standard error into an APIError\n// It handles:\n// - Context cancellation/timeout errors\n// - Existing APIErrors (preserves original error details)\n// - Unknown errors (converts to internal server error)\n// Parameters:\n// - err: The original error to wrap\n// - detail: Additional context about where the error occurred\n//\n// Returns:\n// - APIError: A standardized error response\nfunc WrapError(err error, detail string) APIError {\n\tvar apiErr APIError\n\n\t// Handle context errors\n\tswitch {\n\tcase errors.Is(err, context.DeadlineExceeded):\n\t\tapiErr = APIError{\n\t\t\tHTTPCode: http.StatusGatewayTimeout,\n\t\t\tCode: ErrInternal + \"_002\",\n\t\t\tMessage: \"Request timed out\",\n\t\t\tDetail: detail,\n\t\t\tDebugInfo: \"Context deadline exceeded\",\n\t\t}\n\t\treturn apiErr\n\tcase errors.Is(err, context.Canceled):\n\t\tapiErr = APIError{\n\t\t\tHTTPCode: http.StatusRequestTimeout,\n\t\t\tCode: ErrInternal + \"_003\",\n\t\t\tMessage: \"Request was canceled\",\n\t\t\tDetail: detail,\n\t\t\tDebugInfo: \"Context was canceled\",\n\t\t}\n\t\treturn apiErr\n\t}\n\n\t// Handle APIError types\n\tswitch e := err.(type) {\n\tcase APIError:\n\t\tapiErr = e\n\t\tif detail != \"\" {\n\t\t\tif apiErr.Detail != \"\" {\n\t\t\t\tapiErr.Detail = fmt.Sprintf(\"%s: %s\", detail, apiErr.Detail)\n\t\t\t} else {\n\t\t\t\tapiErr.Detail = detail\n\t\t\t}\n\t\t}\n\tdefault:\n\t\t// Convert unknown errors to internal server error\n\t\tapiErr = ErrInternalUnexpected\n\t\tapiErr.Detail = detail\n\t\tapiErr.DebugInfo = err.Error()\n\t}\n\n\treturn apiErr\n}\n\n// IsErrorCode checks if an error is an APIError with the specified code\nfunc IsErrorCode(err error, code string) bool {\n\tif apiErr, ok := err.(APIError); ok {\n\t\treturn apiErr.Code == code\n\t}\n\treturn false\n}\n\n// Add a handler to serve the error catalog\nfunc ErrorCatalogHandler(w http.ResponseWriter, r *http.Request) {\n\ttype ErrorDoc struct {\n\t\tCode string `json:\"code\"`\n\t\tHTTPCode int `json:\"http_code\"`\n\t\tMessage string `json:\"message\"`\n\t}\n\n\tdocs := make([]ErrorDoc, 0, len(ErrorCatalog))\n\tfor code, info := range ErrorCatalog {\n\t\tdocs = append(docs, ErrorDoc{\n\t\t\tCode: code,\n\t\t\tHTTPCode: info.HTTPCode,\n\t\t\tMessage: info.Message,\n\t\t})\n\t}\n\n\t// Sort by error code\n\tsort.Slice(docs, func(i, j int) bool {\n\t\treturn docs[i].Code < docs[j].Code\n\t})\n\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tw.WriteHeader(http.StatusOK)\n\tjson.NewEncoder(w).Encode(docs)\n}\n\n// createAPIError creates a consistent API error with optional debug info\nfunc createAPIError(baseErr APIError, detail string, debugInfo string) APIError {\n\tapiErr := baseErr\n\tif detail != \"\" {\n\t\tapiErr.Detail = detail\n\t}\n\tif debugInfo != \"\" {\n\t\tapiErr.DebugInfo = debugInfo\n\t}\n\treturn apiErr\n}\n" }, { "path": "api/file_upload_handler.go", "content": "package main\n\nimport (\n\t\"bytes\"\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"log\"\n\t\"net/http\"\n\t\"strconv\"\n\t\"strings\"\n\n\t\"github.com/gorilla/mux\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype ChatFileHandler struct {\n\tservice *ChatFileService\n}\n\nfunc NewChatFileHandler(sqlc_q *sqlc_queries.Queries) *ChatFileHandler {\n\tChatFileService := NewChatFileService(sqlc_q)\n\treturn &ChatFileHandler{\n\t\tservice: ChatFileService,\n\t}\n}\n\nfunc (h *ChatFileHandler) Register(router *mux.Router) {\n\trouter.HandleFunc(\"/upload\", h.ReceiveFile).Methods(http.MethodPost)\n\trouter.HandleFunc(\"/chat_file/{uuid}/list\", h.ChatFilesBySessionUUID).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/download/{id}\", h.DownloadFile).Methods(http.MethodGet)\n\trouter.HandleFunc(\"/download/{id}\", h.DeleteFile).Methods(http.MethodDelete)\n}\n\nconst (\n\tmaxUploadSize = 32 << 20 // 32 MB\n)\n\nvar allowedTypes = map[string]string{\n\t\"image/jpeg\": \".jpg\",\n\t\"image/png\": \".png\",\n\t\"application/pdf\": \".pdf\",\n\t\"text/plain\": \".txt\",\n\t\"application/json\": \".json\",\n}\n\n// isValidFileType checks if the file type is allowed and matches the extension\nfunc isValidFileType(mimeType, fileName string) bool {\n\t// Get expected extension for mime type\n\texpectedExt, ok := allowedTypes[mimeType]\n\tif !ok {\n\t\treturn false\n\t}\n\n\t// Check if file has the expected extension\n\treturn strings.HasSuffix(strings.ToLower(fileName), expectedExt)\n}\n\nfunc (h *ChatFileHandler) ReceiveFile(w http.ResponseWriter, r *http.Request) {\n\t// Parse multipart form with size limit\n\tif err := r.ParseMultipartForm(maxUploadSize); err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(fmt.Sprintf(\"file too large, max size is %d bytes\", maxUploadSize)))\n\t\treturn\n\t}\n\n\t// Get session UUID\n\tsessionUUID := r.FormValue(\"session-uuid\")\n\tif sessionUUID == \"\" {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"missing session UUID\"))\n\t\treturn\n\t}\n\n\t// Get user ID\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"missing or invalid user ID\"))\n\t\treturn\n\t}\n\n\t// Get uploaded file\n\tfile, header, err := r.FormFile(\"file\")\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to get uploaded file\"))\n\t\treturn\n\t}\n\tdefer func() {\n\t\tif err := file.Close(); err != nil {\n\t\t\tlog.Printf(\"Error closing uploaded file: %v\", err)\n\t\t}\n\t}()\n\n\t// Validate file type and extension\n\tmimeType := header.Header.Get(\"Content-Type\")\n\tif !isValidFileType(mimeType, header.Filename) {\n\t\tRespondWithAPIError(w, ErrChatFileInvalidType.WithMessage(\n\t\t\tfmt.Sprintf(\"unsupported file type: %s or invalid extension for type\", mimeType)))\n\t\treturn\n\t}\n\n\tlog.Printf(\"Uploading file: %s (%s, %d bytes)\",\n\t\theader.Filename, mimeType, header.Size)\n\n\t// Validate file size\n\tif header.Size > maxUploadSize {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(fmt.Sprintf(\"file too large, max size is %d bytes\", maxUploadSize)))\n\t\treturn\n\t}\n\n\t// Read file into buffer with size limit\n\tvar buf bytes.Buffer\n\tlimitedReader := &io.LimitedReader{R: file, N: maxUploadSize}\n\tif _, err := io.Copy(&buf, limitedReader); err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to read uploaded file\"))\n\t\treturn\n\t}\n\n\t// Check if we hit the size limit\n\tif limitedReader.N <= 0 {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\n\t\t\tfmt.Sprintf(\"file exceeds maximum size of %d bytes\", maxUploadSize)))\n\t\treturn\n\t}\n\t// Create chat file record\n\tchatFile, err := h.service.q.CreateChatFile(r.Context(), sqlc_queries.CreateChatFileParams{\n\t\tChatSessionUuid: sessionUUID,\n\t\tUserID: userID,\n\t\tName: header.Filename,\n\t\tData: buf.Bytes(),\n\t\tMimeType: mimeType,\n\t})\n\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to create chat file record\"))\n\t\treturn\n\t}\n\n\t// Clean up buffer\n\tbuf.Reset()\n\n\t// Return success response\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tw.WriteHeader(http.StatusCreated)\n\tjson.NewEncoder(w).Encode(map[string]string{\n\t\t\"url\": fmt.Sprintf(\"/download/%d\", chatFile.ID),\n\t\t\"name\": header.Filename,\n\t\t\"type\": mimeType,\n\t\t\"size\": fmt.Sprintf(\"%d\", header.Size),\n\t})\n}\n\nfunc (h *ChatFileHandler) DownloadFile(w http.ResponseWriter, r *http.Request) {\n\tfileID := mux.Vars(r)[\"id\"]\n\tfileIdInt, err := strconv.ParseInt(fileID, 10, 32)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"invalid file ID\"))\n\t\treturn\n\t}\n\n\tfile, err := h.service.q.GetChatFileByID(r.Context(), int32(fileIdInt))\n\tif err != nil {\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\tRespondWithAPIError(w, ErrChatFileNotFound.WithMessage(fmt.Sprintf(\"file ID %d not found\", fileIdInt)))\n\t\t} else {\n\t\t\tRespondWithAPIError(w, WrapError(err, \"failed to get chat file\"))\n\t\t}\n\t\treturn\n\t}\n\n\t// Set proper content type from stored mime type\n\tw.Header().Set(\"Content-Disposition\", fmt.Sprintf(\"attachment; filename=%s\", file.Name))\n\tw.Header().Set(\"Content-Length\", strconv.Itoa(len(file.Data)))\n\n\tif _, err := w.Write(file.Data); err != nil {\n\t\tlog.Printf(\"Failed to write file data: %v\", err)\n\t}\n}\n\nfunc (h *ChatFileHandler) DeleteFile(w http.ResponseWriter, r *http.Request) {\n\tfileID := mux.Vars(r)[\"id\"]\n\tfileIdInt, _ := strconv.ParseInt(fileID, 10, 32)\n\t_, err := h.service.q.DeleteChatFile(r.Context(), int32(fileIdInt))\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to delete chat file\"))\n\t\treturn\n\t}\n\tw.WriteHeader(http.StatusOK)\n}\n\nfunc (h *ChatFileHandler) ChatFilesBySessionUUID(w http.ResponseWriter, r *http.Request) {\n\tsessionUUID := mux.Vars(r)[\"uuid\"]\n\tuserID, err := getUserID(r.Context())\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrAuthInvalidCredentials.WithMessage(\"missing or invalid user ID\"))\n\t\treturn\n\t}\n\tchatFiles, err := h.service.q.ListChatFilesBySessionUUID(r.Context(), sqlc_queries.ListChatFilesBySessionUUIDParams{\n\t\tChatSessionUuid: sessionUUID,\n\t\tUserID: userID,\n\t})\n\tif err != nil {\n\t\tRespondWithAPIError(w, WrapError(err, \"failed to list chat files for session\"))\n\t\treturn\n\t}\n\tw.WriteHeader(http.StatusOK)\n\n\tif len(chatFiles) == 0 {\n\t\tw.Write([]byte(\"[]\"))\n\t\treturn\n\t}\n\tjson.NewEncoder(w).Encode(chatFiles)\n}\n" }, { "path": "api/file_upload_service.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"log\"\n\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// ChatFileService handles operations related to chat file uploads\ntype ChatFileService struct {\n\tq *sqlc_queries.Queries\n}\n\n// NewChatFileService creates a new ChatFileService instance\nfunc NewChatFileService(q *sqlc_queries.Queries) *ChatFileService {\n\treturn &ChatFileService{q: q}\n}\n\n// CreateChatUpload handles creating a new chat file upload\nfunc (s *ChatFileService) CreateChatUpload(ctx context.Context, params sqlc_queries.CreateChatFileParams) (sqlc_queries.ChatFile, error) {\n\t// Validate input\n\tif params.ChatSessionUuid == \"\" {\n\t\treturn sqlc_queries.ChatFile{}, ErrValidationInvalidInput(\"missing session UUID\")\n\t}\n\tif params.UserID <= 0 {\n\t\treturn sqlc_queries.ChatFile{}, ErrValidationInvalidInput(\"invalid user ID\")\n\t}\n\tif params.Name == \"\" {\n\t\treturn sqlc_queries.ChatFile{}, ErrValidationInvalidInput(\"missing file name\")\n\t}\n\tif len(params.Data) == 0 {\n\t\treturn sqlc_queries.ChatFile{}, ErrValidationInvalidInput(\"empty file data\")\n\t}\n\n\tlog.Printf(\"Creating chat file upload for session %s, user %d\",\n\t\tparams.ChatSessionUuid, params.UserID)\n\n\tupload, err := s.q.CreateChatFile(ctx, params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatFile{}, WrapError(err, \"failed to create chat file\")\n\t}\n\n\tlog.Printf(\"Created chat file upload ID %d\", upload.ID)\n\treturn upload, nil\n}\n\n// GetChatFile retrieves a chat file by ID\nfunc (s *ChatFileService) GetChatFile(ctx context.Context, id int32) (sqlc_queries.GetChatFileByIDRow, error) {\n\tif id <= 0 {\n\t\treturn sqlc_queries.GetChatFileByIDRow{}, ErrValidationInvalidInput(\"invalid file ID\")\n\t}\n\n\tlog.Printf(\"Retrieving chat file ID %d\", id)\n\n\tfile, err := s.q.GetChatFileByID(ctx, id)\n\tif err != nil {\n\t\treturn sqlc_queries.GetChatFileByIDRow{}, WrapError(err, \"failed to get chat file\")\n\t}\n\n\treturn file, nil\n}\n\n// DeleteChatFile deletes a chat file by ID\nfunc (s *ChatFileService) DeleteChatFile(ctx context.Context, id int32) error {\n\tif id <= 0 {\n\t\treturn ErrValidationInvalidInput(\"invalid file ID\")\n\t}\n\n\tlog.Printf(\"Deleting chat file ID %d\", id)\n\n\t_, err := s.q.DeleteChatFile(ctx, id)\n\tif err != nil {\n\t\treturn WrapError(err, \"failed to delete chat file\")\n\t}\n\n\treturn nil\n}\n\n// ListChatFilesBySession retrieves chat files for a session\nfunc (s *ChatFileService) ListChatFilesBySession(ctx context.Context, sessionUUID string, userID int32) ([]sqlc_queries.ListChatFilesBySessionUUIDRow, error) {\n\tif sessionUUID == \"\" {\n\t\treturn nil, ErrValidationInvalidInput(\"missing session UUID\")\n\t}\n\tif userID <= 0 {\n\t\treturn nil, ErrValidationInvalidInput(\"invalid user ID\")\n\t}\n\n\tlog.Printf(\"Listing chat files for session %s, user %d\", sessionUUID, userID)\n\n\tfiles, err := s.q.ListChatFilesBySessionUUID(ctx, sqlc_queries.ListChatFilesBySessionUUIDParams{\n\t\tChatSessionUuid: sessionUUID,\n\t\tUserID: userID,\n\t})\n\tif err != nil {\n\t\treturn nil, WrapError(err, \"failed to list chat files\")\n\t}\n\n\treturn files, nil\n}\n" }, { "path": "api/go.mod", "content": "module github.com/swuecho/chat_backend\n\ngo 1.19\n\nrequire (\n\tgithub.com/deckarep/golang-set/v2 v2.6.0\n\tgithub.com/golang-jwt/jwt/v5 v5.2.1\n\tgithub.com/google/go-cmp v0.6.0\n\tgithub.com/google/uuid v1.6.0\n\tgithub.com/gorilla/mux v1.8.0\n\tgithub.com/jackc/pgconn v1.14.3\n\tgithub.com/lib/pq v1.10.9\n\tgithub.com/ory/dockertest/v3 v3.10.0\n\tgithub.com/pkoukk/tiktoken-go v0.1.6\n\tgithub.com/rotisserie/eris v0.5.4\n\tgithub.com/samber/lo v1.39.0\n\tgithub.com/sashabaranov/go-openai v1.36.1\n\tgithub.com/sirupsen/logrus v1.9.3\n\tgithub.com/spf13/viper v1.18.2\n\tgithub.com/tmc/langchaingo v0.0.0-20230610024316-06cb7b57ea80\n\tgolang.org/x/crypto v0.23.0\n\tgotest.tools/v3 v3.4.0\n)\n\nrequire (\n\tgithub.com/Masterminds/goutils v1.1.1 // indirect\n\tgithub.com/Masterminds/semver/v3 v3.2.0 // indirect\n\tgithub.com/Masterminds/sprig/v3 v3.2.3 // indirect\n\tgithub.com/dlclark/regexp2 v1.10.0 // indirect\n\tgithub.com/felixge/httpsnoop v1.0.3 // indirect\n\tgithub.com/fsnotify/fsnotify v1.7.0 // indirect\n\tgithub.com/hashicorp/hcl v1.0.0 // indirect\n\tgithub.com/huandu/xstrings v1.3.3 // indirect\n\tgithub.com/jackc/chunkreader/v2 v2.0.1 // indirect\n\tgithub.com/jackc/pgio v1.0.0 // indirect\n\tgithub.com/jackc/pgpassfile v1.0.0 // indirect\n\tgithub.com/jackc/pgproto3/v2 v2.3.3 // indirect\n\tgithub.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a // indirect\n\tgithub.com/magiconair/properties v1.8.7 // indirect\n\tgithub.com/mitchellh/copystructure v1.0.0 // indirect\n\tgithub.com/mitchellh/reflectwalk v1.0.0 // indirect\n\tgithub.com/pelletier/go-toml/v2 v2.1.0 // indirect\n\tgithub.com/sagikazarmark/locafero v0.4.0 // indirect\n\tgithub.com/sagikazarmark/slog-shim v0.1.0 // indirect\n\tgithub.com/shopspring/decimal v1.2.0 // indirect\n\tgithub.com/sourcegraph/conc v0.3.0 // indirect\n\tgithub.com/spf13/afero v1.11.0 // indirect\n\tgithub.com/spf13/cast v1.6.0 // indirect\n\tgithub.com/spf13/pflag v1.0.5 // indirect\n\tgithub.com/subosito/gotenv v1.6.0 // indirect\n\tgo.starlark.net v0.0.0-20230302034142-4b1e35fe2254 // indirect\n\tgo.uber.org/atomic v1.9.0 // indirect\n\tgo.uber.org/multierr v1.9.0 // indirect\n\tgolang.org/x/mod v0.12.0 // indirect\n\tgolang.org/x/text v0.15.0 // indirect\n\tgolang.org/x/tools v0.13.0 // indirect\n\tgopkg.in/ini.v1 v1.67.0 // indirect\n\tgopkg.in/yaml.v3 v3.0.1 // indirect\n)\n\nrequire (\n\tgithub.com/Azure/go-ansiterm v0.0.0-20230124172434-306776ec8161 // indirect\n\tgithub.com/Microsoft/go-winio v0.6.1 // indirect\n\tgithub.com/Nvveen/Gotty v0.0.0-20120604004816-cd527374f1e5 // indirect\n\tgithub.com/cenkalti/backoff/v4 v4.2.1 // indirect\n\tgithub.com/containerd/continuity v0.3.0 // indirect\n\tgithub.com/docker/cli v23.0.4+incompatible // indirect\n\tgithub.com/docker/docker v23.0.4+incompatible // indirect\n\tgithub.com/docker/go-connections v0.4.0 // indirect\n\tgithub.com/docker/go-units v0.5.0 // indirect\n\tgithub.com/gogo/protobuf v1.3.2 // indirect\n\tgithub.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510 // indirect\n\tgithub.com/gorilla/handlers v1.5.1\n\tgithub.com/imdario/mergo v0.3.15 // indirect\n\tgithub.com/mitchellh/mapstructure v1.5.0 // indirect\n\tgithub.com/moby/term v0.0.0-20221205130635-1aeaba878587 // indirect\n\tgithub.com/opencontainers/go-digest v1.0.0 // indirect\n\tgithub.com/opencontainers/image-spec v1.0.2 // indirect\n\tgithub.com/opencontainers/runc v1.1.6 // indirect\n\tgithub.com/pkg/errors v0.9.1 // indirect\n\tgithub.com/xeipuuv/gojsonpointer v0.0.0-20190905194746-02993c407bfb // indirect\n\tgithub.com/xeipuuv/gojsonreference v0.0.0-20180127040603-bd5ef7bd5415 // indirect\n\tgithub.com/xeipuuv/gojsonschema v1.2.0 // indirect\n\tgolang.org/x/exp v0.0.0-20230905200255-921286631fa9 // indirect\n\tgolang.org/x/sys v0.20.0 // indirect\n\tgolang.org/x/time v0.5.0\n\tgopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c // indirect\n\tgopkg.in/yaml.v2 v2.4.0 // indirect\n)\n" }, { "path": "api/go.sum", "content": "cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=\ngithub.com/Azure/go-ansiterm v0.0.0-20230124172434-306776ec8161 h1:L/gRVlceqvL25UVaW/CKtUDjefjrs0SPonmDGUVOYP0=\ngithub.com/Azure/go-ansiterm v0.0.0-20230124172434-306776ec8161/go.mod h1:xomTg63KZ2rFqZQzSB4Vz2SUXa1BpHTVz9L5PTmPC4E=\ngithub.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=\ngithub.com/Masterminds/goutils v1.1.1 h1:5nUrii3FMTL5diU80unEVvNevw1nH4+ZV4DSLVJLSYI=\ngithub.com/Masterminds/goutils v1.1.1/go.mod h1:8cTjp+g8YejhMuvIA5y2vz3BpJxksy863GQaJW2MFNU=\ngithub.com/Masterminds/semver/v3 v3.2.0 h1:3MEsd0SM6jqZojhjLWWeBY+Kcjy9i6MQAeY7YgDP83g=\ngithub.com/Masterminds/semver/v3 v3.2.0/go.mod h1:qvl/7zhW3nngYb5+80sSMF+FG2BjYrf8m9wsX0PNOMQ=\ngithub.com/Masterminds/sprig/v3 v3.2.3 h1:eL2fZNezLomi0uOLqjQoN6BfsDD+fyLtgbJMAj9n6YA=\ngithub.com/Masterminds/sprig/v3 v3.2.3/go.mod h1:rXcFaZ2zZbLRJv/xSysmlgIM1u11eBaRMhvYXJNkGuM=\ngithub.com/Microsoft/go-winio v0.6.1 h1:9/kr64B9VUZrLm5YYwbGtUJnMgqWVOdUAXu6Migciow=\ngithub.com/Microsoft/go-winio v0.6.1/go.mod h1:LRdKpFKfdobln8UmuiYcKPot9D2v6svN5+sAH+4kjUM=\ngithub.com/Nvveen/Gotty v0.0.0-20120604004816-cd527374f1e5 h1:TngWCqHvy9oXAN6lEVMRuU21PR1EtLVZJmdB18Gu3Rw=\ngithub.com/Nvveen/Gotty v0.0.0-20120604004816-cd527374f1e5/go.mod h1:lmUJ/7eu/Q8D7ML55dXQrVaamCz2vxCfdQBasLZfHKk=\ngithub.com/cenkalti/backoff/v4 v4.2.1 h1:y4OZtCnogmCPw98Zjyt5a6+QwPLGkiQsYW5oUqylYbM=\ngithub.com/cenkalti/backoff/v4 v4.2.1/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=\ngithub.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=\ngithub.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=\ngithub.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=\ngithub.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=\ngithub.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw=\ngithub.com/containerd/continuity v0.3.0 h1:nisirsYROK15TAMVukJOUyGJjz4BNQJBVsNvAXZJ/eg=\ngithub.com/containerd/continuity v0.3.0/go.mod h1:wJEAIwKOm/pBZuBd0JmeTvnLquTB1Ag8espWhkykbPM=\ngithub.com/creack/pty v1.1.18 h1:n56/Zwd5o6whRC5PMGretI4IdRLlmBXYNjScPaBgsbY=\ngithub.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=\ngithub.com/deckarep/golang-set/v2 v2.6.0 h1:XfcQbWM1LlMB8BsJ8N9vW5ehnnPVIw0je80NsVHagjM=\ngithub.com/deckarep/golang-set/v2 v2.6.0/go.mod h1:VAky9rY/yGXJOLEDv3OMci+7wtDpOF4IN+y82NBOac4=\ngithub.com/dlclark/regexp2 v1.10.0 h1:+/GIL799phkJqYW+3YbOd8LCcbHzT0Pbo8zl70MHsq0=\ngithub.com/dlclark/regexp2 v1.10.0/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=\ngithub.com/docker/cli v23.0.4+incompatible h1:xClB7PsiATttDHj8ce5qvJcikiApNy7teRR1XkoBZGs=\ngithub.com/docker/cli v23.0.4+incompatible/go.mod h1:JLrzqnKDaYBop7H2jaqPtU4hHvMKP+vjCwu2uszcLI8=\ngithub.com/docker/docker v23.0.4+incompatible h1:Kd3Bh9V/rO+XpTP/BLqM+gx8z7+Yb0AA2Ibj+nNo4ek=\ngithub.com/docker/docker v23.0.4+incompatible/go.mod h1:eEKB0N0r5NX/I1kEveEz05bcu8tLC/8azJZsviup8Sk=\ngithub.com/docker/go-connections v0.4.0 h1:El9xVISelRB7BuFusrZozjnkIM5YnzCViNKohAFqRJQ=\ngithub.com/docker/go-connections v0.4.0/go.mod h1:Gbd7IOopHjR8Iph03tsViu4nIes5XhDvyHbTtUxmeec=\ngithub.com/docker/go-units v0.5.0 h1:69rxXcBk27SvSaaxTtLh/8llcHD8vYHT7WSdRZ/jvr4=\ngithub.com/docker/go-units v0.5.0/go.mod h1:fgPhTUdO+D/Jk86RDLlptpiXQzgHJF7gydDDbaIK4Dk=\ngithub.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=\ngithub.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=\ngithub.com/felixge/httpsnoop v1.0.1/go.mod h1:m8KPJKqk1gH5J9DgRY2ASl2lWCfGKXixSwevea8zH2U=\ngithub.com/felixge/httpsnoop v1.0.3 h1:s/nj+GCswXYzN5v2DpNMuMQYe+0DDwt5WVCU6CWBdXk=\ngithub.com/felixge/httpsnoop v1.0.3/go.mod h1:m8KPJKqk1gH5J9DgRY2ASl2lWCfGKXixSwevea8zH2U=\ngithub.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=\ngithub.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nosvA=\ngithub.com/fsnotify/fsnotify v1.7.0/go.mod h1:40Bi/Hjc2AVfZrqy+aj+yEI+/bRxZnMJyTJwOpGvigM=\ngithub.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE=\ngithub.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q=\ngithub.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=\ngithub.com/golang-jwt/jwt/v5 v5.2.1 h1:OuVbFODueb089Lh128TAcimifWaLhJwVflnrgM17wHk=\ngithub.com/golang-jwt/jwt/v5 v5.2.1/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk=\ngithub.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=\ngithub.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=\ngithub.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8=\ngithub.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA=\ngithub.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs=\ngithub.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w=\ngithub.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0=\ngithub.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8=\ngithub.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=\ngithub.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=\ngithub.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=\ngithub.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.1/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=\ngithub.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=\ngithub.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510 h1:El6M4kTTCOh6aBiKaUGG7oYTSPP8MxqL4YI3kZKwcP4=\ngithub.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510/go.mod h1:pupxD2MaaD3pAXIBCelhxNneeOaAeabZDe5s4K6zSpQ=\ngithub.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=\ngithub.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/gorilla/handlers v1.5.1 h1:9lRY6j8DEeeBT10CvO9hGW0gmky0BprnvDI5vfhUHH4=\ngithub.com/gorilla/handlers v1.5.1/go.mod h1:t8XrUpc4KVXb7HGyJ4/cEnwQiaxrX/hz1Zv/4g96P1Q=\ngithub.com/gorilla/mux v1.8.0 h1:i40aqfkR1h2SlN9hojwV5ZA91wcXFOvkdNIeFDP5koI=\ngithub.com/gorilla/mux v1.8.0/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB71So=\ngithub.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4=\ngithub.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ=\ngithub.com/huandu/xstrings v1.3.3 h1:/Gcsuc1x8JVbJ9/rlye4xZnVAbEkGauT8lbebqcQws4=\ngithub.com/huandu/xstrings v1.3.3/go.mod h1:y5/lhBue+AyNmUVz9RLU9xbLR0o4KIIExikq4ovT0aE=\ngithub.com/imdario/mergo v0.3.11/go.mod h1:jmQim1M+e3UYxmgPu/WyfjB3N3VflVyUjjjwH0dnCYA=\ngithub.com/imdario/mergo v0.3.15 h1:M8XP7IuFNsqUx6VPK2P9OSmsYsI/YFaGil0uD21V3dM=\ngithub.com/imdario/mergo v0.3.15/go.mod h1:WBLT9ZmE3lPoWsEzCh9LPo3TiwVN+ZKEjmz+hD27ysY=\ngithub.com/jackc/chunkreader/v2 v2.0.0/go.mod h1:odVSm741yZoC3dpHEUXIqA9tQRhFrgOHwnPIn9lDKlk=\ngithub.com/jackc/chunkreader/v2 v2.0.1 h1:i+RDz65UE+mmpjTfyz0MoVTnzeYxroil2G82ki7MGG8=\ngithub.com/jackc/chunkreader/v2 v2.0.1/go.mod h1:odVSm741yZoC3dpHEUXIqA9tQRhFrgOHwnPIn9lDKlk=\ngithub.com/jackc/pgconn v1.14.3 h1:bVoTr12EGANZz66nZPkMInAV/KHD2TxH9npjXXgiB3w=\ngithub.com/jackc/pgconn v1.14.3/go.mod h1:RZbme4uasqzybK2RK5c65VsHxoyaml09lx3tXOcO/VM=\ngithub.com/jackc/pgio v1.0.0 h1:g12B9UwVnzGhueNavwioyEEpAmqMe1E/BN9ES+8ovkE=\ngithub.com/jackc/pgio v1.0.0/go.mod h1:oP+2QK2wFfUWgr+gxjoBH9KGBb31Eio69xUb0w5bYf8=\ngithub.com/jackc/pgmock v0.0.0-20210724152146-4ad1a8207f65 h1:DadwsjnMwFjfWc9y5Wi/+Zz7xoE5ALHsRQlOctkOiHc=\ngithub.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=\ngithub.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=\ngithub.com/jackc/pgproto3/v2 v2.3.3 h1:1HLSx5H+tXR9pW3in3zaztoEwQYRC9SQaYUHjTSUOag=\ngithub.com/jackc/pgproto3/v2 v2.3.3/go.mod h1:WfJCnwN3HIg9Ish/j3sgWXnAfK8A9Y0bwXYU5xKaEdA=\ngithub.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a h1:bbPeKD0xmW/Y25WS6cokEszi5g+S0QxI/d45PkRi7Nk=\ngithub.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a/go.mod h1:5TJZWKEWniPve33vlWYSoGYefn3gLQRzjfDlhSJ9ZKM=\ngithub.com/kisielk/errcheck v1.5.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8=\ngithub.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=\ngithub.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=\ngithub.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=\ngithub.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=\ngithub.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=\ngithub.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=\ngithub.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw=\ngithub.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=\ngithub.com/magiconair/properties v1.8.7 h1:IeQXZAiQcpL9mgcAe1Nu6cX9LLw6ExEHKjN0VQdvPDY=\ngithub.com/magiconair/properties v1.8.7/go.mod h1:Dhd985XPs7jluiymwWYZ0G4Z61jb3vdS329zhj2hYo0=\ngithub.com/mitchellh/copystructure v1.0.0 h1:Laisrj+bAB6b/yJwB5Bt3ITZhGJdqmxquMKeZ+mmkFQ=\ngithub.com/mitchellh/copystructure v1.0.0/go.mod h1:SNtv71yrdKgLRyLFxmLdkAbkKEFWgYaq1OVrnRcwhnw=\ngithub.com/mitchellh/mapstructure v1.5.0 h1:jeMsZIYE/09sWLaz43PL7Gy6RuMjD2eJVyuac5Z2hdY=\ngithub.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo=\ngithub.com/mitchellh/reflectwalk v1.0.0 h1:9D+8oIskB4VJBN5SFlmc27fSlIBZaov1Wpk/IfikLNY=\ngithub.com/mitchellh/reflectwalk v1.0.0/go.mod h1:mSTlrgnPZtwu0c4WaC2kGObEpuNDbx0jmZXqmk4esnw=\ngithub.com/moby/term v0.0.0-20221205130635-1aeaba878587 h1:HfkjXDfhgVaN5rmueG8cL8KKeFNecRCXFhaJ2qZ5SKA=\ngithub.com/moby/term v0.0.0-20221205130635-1aeaba878587/go.mod h1:8FzsFHVUBGZdbDsJw/ot+X+d5HLUbvklYLJ9uGfcI3Y=\ngithub.com/opencontainers/go-digest v1.0.0 h1:apOUWs51W5PlhuyGyz9FCeeBIOUDA/6nW8Oi/yOhh5U=\ngithub.com/opencontainers/go-digest v1.0.0/go.mod h1:0JzlMkj0TRzQZfJkVvzbP0HBR3IKzErnv2BNG4W4MAM=\ngithub.com/opencontainers/image-spec v1.0.2 h1:9yCKha/T5XdGtO0q9Q9a6T5NUCsTn/DrBg0D7ufOcFM=\ngithub.com/opencontainers/image-spec v1.0.2/go.mod h1:BtxoFyWECRxE4U/7sNtV5W15zMzWCbyJoFRP3s7yZA0=\ngithub.com/opencontainers/runc v1.1.6 h1:XbhB8IfG/EsnhNvZtNdLB0GBw92GYEFvKlhaJk9jUgA=\ngithub.com/opencontainers/runc v1.1.6/go.mod h1:CbUumNnWCuTGFukNXahoo/RFBZvDAgRh/smNYNOhA50=\ngithub.com/ory/dockertest/v3 v3.10.0 h1:4K3z2VMe8Woe++invjaTB7VRyQXQy5UY+loujO4aNE4=\ngithub.com/ory/dockertest/v3 v3.10.0/go.mod h1:nr57ZbRWMqfsdGdFNLHz5jjNdDb7VVFnzAeW1n5N1Lg=\ngithub.com/pelletier/go-toml/v2 v2.1.0 h1:FnwAJ4oYMvbT/34k9zzHuZNrhlz48GB3/s6at6/MHO4=\ngithub.com/pelletier/go-toml/v2 v2.1.0/go.mod h1:tJU2Z3ZkXwnxa4DPO899bsyIoywizdUvyaeZurnPPDc=\ngithub.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=\ngithub.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pkoukk/tiktoken-go v0.1.6 h1:JF0TlJzhTbrI30wCvFuiw6FzP2+/bR+FIxUdgEAcUsw=\ngithub.com/pkoukk/tiktoken-go v0.1.6/go.mod h1:9NiV+i9mJKGj1rYOT+njbv+ZwA/zJxYdewGl6qVatpg=\ngithub.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=\ngithub.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U=\ngithub.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=\ngithub.com/rotisserie/eris v0.5.4 h1:Il6IvLdAapsMhvuOahHWiBnl1G++Q0/L5UIkI5mARSk=\ngithub.com/rotisserie/eris v0.5.4/go.mod h1:Z/kgYTJiJtocxCbFfvRmO+QejApzG6zpyky9G1A4g9s=\ngithub.com/sagikazarmark/locafero v0.4.0 h1:HApY1R9zGo4DBgr7dqsTH/JJxLTTsOt7u6keLGt6kNQ=\ngithub.com/sagikazarmark/locafero v0.4.0/go.mod h1:Pe1W6UlPYUk/+wc/6KFhbORCfqzgYEpgQ3O5fPuL3H4=\ngithub.com/sagikazarmark/slog-shim v0.1.0 h1:diDBnUNK9N/354PgrxMywXnAwEr1QZcOr6gto+ugjYE=\ngithub.com/sagikazarmark/slog-shim v0.1.0/go.mod h1:SrcSrq8aKtyuqEI1uvTDTK1arOWRIczQRv+GVI1AkeQ=\ngithub.com/samber/lo v1.39.0 h1:4gTz1wUhNYLhFSKl6O+8peW0v2F4BCY034GRpU9WnuA=\ngithub.com/samber/lo v1.39.0/go.mod h1:+m/ZKRl6ClXCE2Lgf3MsQlWfh4bn1bz6CXEOxnEXnEA=\ngithub.com/sashabaranov/go-openai v1.36.1 h1:EVfRXwIlW2rUzpx6vR+aeIKCK/xylSrVYAx1TMTSX3g=\ngithub.com/sashabaranov/go-openai v1.36.1/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg=\ngithub.com/shopspring/decimal v1.2.0 h1:abSATXmQEYyShuxI4/vyW3tV1MrKAJzCZ/0zLUXYbsQ=\ngithub.com/shopspring/decimal v1.2.0/go.mod h1:DKyhrW/HYNuLGql+MJL6WCR6knT2jwCFRcu2hWCYk4o=\ngithub.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ=\ngithub.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=\ngithub.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=\ngithub.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=\ngithub.com/spf13/afero v1.11.0 h1:WJQKhtpdm3v2IzqG8VMqrr6Rf3UYpEF239Jy9wNepM8=\ngithub.com/spf13/afero v1.11.0/go.mod h1:GH9Y3pIexgf1MTIWtNGyogA5MwRIDXGUr+hbWNoBjkY=\ngithub.com/spf13/cast v1.3.1/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE=\ngithub.com/spf13/cast v1.6.0 h1:GEiTHELF+vaR5dhz3VqZfFSzZjYbgeKDpBxQVS4GYJ0=\ngithub.com/spf13/cast v1.6.0/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo=\ngithub.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=\ngithub.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=\ngithub.com/spf13/viper v1.18.2 h1:LUXCnvUvSM6FXAsj6nnfc8Q2tp1dIgUfY9Kc8GsSOiQ=\ngithub.com/spf13/viper v1.18.2/go.mod h1:EKmWIqdnk5lOcmR72yw6hS+8OPYcwD0jteitLMVB+yk=\ngithub.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=\ngithub.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=\ngithub.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=\ngithub.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=\ngithub.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=\ngithub.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=\ngithub.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=\ngithub.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=\ngithub.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=\ngithub.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=\ngithub.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8=\ngithub.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU=\ngithub.com/tmc/langchaingo v0.0.0-20230610024316-06cb7b57ea80 h1:Y+a76dNVbdWduw3gznOr2O2OSZkdwDRYPKTDpG/vM9I=\ngithub.com/tmc/langchaingo v0.0.0-20230610024316-06cb7b57ea80/go.mod h1:6l1WoyqVDwkv7cFlY3gfcTv8yVowVyuutKv8PGlQCWI=\ngithub.com/xeipuuv/gojsonpointer v0.0.0-20180127040702-4e3ac2762d5f/go.mod h1:N2zxlSyiKSe5eX1tZViRH5QA0qijqEDrYZiPEAiq3wU=\ngithub.com/xeipuuv/gojsonpointer v0.0.0-20190905194746-02993c407bfb h1:zGWFAtiMcyryUHoUjUJX0/lt1H2+i2Ka2n+D3DImSNo=\ngithub.com/xeipuuv/gojsonpointer v0.0.0-20190905194746-02993c407bfb/go.mod h1:N2zxlSyiKSe5eX1tZViRH5QA0qijqEDrYZiPEAiq3wU=\ngithub.com/xeipuuv/gojsonreference v0.0.0-20180127040603-bd5ef7bd5415 h1:EzJWgHovont7NscjpAxXsDA8S8BMYve8Y5+7cuRE7R0=\ngithub.com/xeipuuv/gojsonreference v0.0.0-20180127040603-bd5ef7bd5415/go.mod h1:GwrjFmJcFw6At/Gs6z4yjiIwzuJ1/+UwLxMQDVQXShQ=\ngithub.com/xeipuuv/gojsonschema v1.2.0 h1:LhYJRs+L4fBtjZUfuSZIKGeVu0QRy8e5Xi7D17UxZ74=\ngithub.com/xeipuuv/gojsonschema v1.2.0/go.mod h1:anYRn/JVcOK2ZgGU+IjEV4nwlhoK5sQluxsYJ78Id3Y=\ngithub.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=\ngithub.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=\ngithub.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=\ngo.starlark.net v0.0.0-20230302034142-4b1e35fe2254 h1:Ss6D3hLXTM0KobyBYEAygXzFfGcjnmfEJOBgSbemCtg=\ngo.starlark.net v0.0.0-20230302034142-4b1e35fe2254/go.mod h1:jxU+3+j+71eXOW14274+SmmuW82qJzl6iZSeqEtTGds=\ngo.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE=\ngo.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=\ngo.uber.org/multierr v1.9.0 h1:7fIwc/ZtS0q++VgcfqFDxSBZVv/Xo49/SYnDFupUwlI=\ngo.uber.org/multierr v1.9.0/go.mod h1:X2jQV1h+kxSjClGpnseKVIxpmcjrj7MNnI0bnlfKTVQ=\ngolang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=\ngolang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=\ngolang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=\ngolang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=\ngolang.org/x/crypto v0.3.0/go.mod h1:hebNnKkNXi2UzZN1eVRvBB7co0a+JxK6XbPiWVs/3J4=\ngolang.org/x/crypto v0.23.0 h1:dIJU/v2J8Mdglj/8rJ6UUOM3Zc9zLZxVZwwxMooUSAI=\ngolang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8=\ngolang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/exp v0.0.0-20230905200255-921286631fa9 h1:GoHiUyI/Tp2nVkLI2mCxVkOjsbSXD66ic0XW0js0R9g=\ngolang.org/x/exp v0.0.0-20230905200255-921286631fa9/go.mod h1:S2oDrQGGwySpoQPVqRShND87VCbxmc6bL1Yd2oYrm6k=\ngolang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=\ngolang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=\ngolang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=\ngolang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=\ngolang.org/x/mod v0.12.0 h1:rmsUpXtvNzj340zd98LZ4KntptpfRHwpFOHG188oHXc=\ngolang.org/x/mod v0.12.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=\ngolang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=\ngolang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=\ngolang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=\ngolang.org/x/net v0.2.0/go.mod h1:KqCZLdyyvdV855qA2rE3GC2aiw5xGR5TEjj8smXukLY=\ngolang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=\ngolang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.5.0 h1:60k92dhOjHxJkrqnwsfl8KuaHbn/5dl0lUPUklKo3qE=\ngolang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210119212857-b64e53b001e4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210616094352-59db8d763f22/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.2.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.20.0 h1:Od9JTbYCk261bKm4M/mw7AklTlFYIa0bIp9BgSm1S8Y=\ngolang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=\ngolang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=\ngolang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=\ngolang.org/x/term v0.0.0-20220526004731-065cf7ba2467/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=\ngolang.org/x/term v0.2.0/go.mod h1:TVmDHMZPmdnySmBfhjOoOdhjzdE1h4u1VwSiw2l1Nuc=\ngolang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=\ngolang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=\ngolang.org/x/text v0.4.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=\ngolang.org/x/text v0.15.0 h1:h1V/4gjBv8v9cjcR6+AR5+/cIYK5N/WAgiv4xlsEtAk=\ngolang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=\ngolang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=\ngolang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=\ngolang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=\ngolang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=\ngolang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.1.0/go.mod h1:xkSsbof2nBLbhDlRMhhhyNLN/zl3eTqcnHD5viDpcZ0=\ngolang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=\ngolang.org/x/tools v0.13.0 h1:Iey4qkscZuv0VvIt8E0neZjtPVQFSc870HQ448QgEmQ=\ngolang.org/x/tools v0.13.0/go.mod h1:HvlwmtVNQAhOuCjW7xxvovg8wbNq7LwfXh/k7wXUl58=\ngolang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngoogle.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=\ngoogle.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=\ngoogle.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=\ngoogle.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc=\ngoogle.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo=\ngoogle.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c=\ngoogle.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg=\ngoogle.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk=\ngoogle.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8=\ngoogle.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0=\ngoogle.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM=\ngoogle.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE=\ngoogle.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo=\ngoogle.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c=\ngoogle.golang.org/protobuf v1.31.0 h1:g0LDEJHgrBl9N9r17Ru3sqWhkIx2NB67okBHPwC7hs8=\ngopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=\ngopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=\ngopkg.in/ini.v1 v1.67.0 h1:Dgnx+6+nfE+IfzjUEISNeydPJh9AXNNsWbGP9KzCsOA=\ngopkg.in/ini.v1 v1.67.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=\ngopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=\ngopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=\ngopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\ngopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=\ngopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\ngotest.tools/v3 v3.4.0 h1:ZazjZUfuVeZGLAmlKKuyv3IKP5orXcwtOwDQH6YVr6o=\ngotest.tools/v3 v3.4.0/go.mod h1:CtbdzLSsqVhDgMtKsx03ird5YTGB3ar27v0u/yKBW5g=\nhonnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nhonnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\n" }, { "path": "api/handle_tts.go", "content": "package main\n\nimport (\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"os\"\n)\n\nfunc handleTTSRequest(w http.ResponseWriter, r *http.Request) {\n\t// Create a new HTTP request with the same method, URL, and body as the original request\n\ttargetURL := r.URL\n\thostEnvVarName := \"TTS_HOST\"\n\tportEnvVarName := \"TTS_PORT\"\n\trealHost := fmt.Sprintf(\"http://%s:%s/api\", os.Getenv(hostEnvVarName), os.Getenv(portEnvVarName))\n\tfullURL := realHost + targetURL.String()\n\tprint(fullURL)\n\tproxyReq, err := http.NewRequest(r.Method, fullURL, r.Body)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Error creating proxy request\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\n\t// Copy the headers from the original request to the proxy request\n\tfor name, values := range r.Header {\n\t\tfor _, value := range values {\n\t\t\tproxyReq.Header.Add(name, value)\n\t\t}\n\t}\n\tvar customTransport = http.DefaultTransport\n\n\t// Send the proxy request using the custom transport\n\tresp, err := customTransport.RoundTrip(proxyReq)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Error sending proxy request\").WithDebugInfo(err.Error()))\n\t\treturn\n\t}\n\tdefer resp.Body.Close()\n\n\t// Copy the headers from the proxy response to the original response\n\tfor name, values := range resp.Header {\n\t\tfor _, value := range values {\n\t\t\tw.Header().Add(name, value)\n\t\t}\n\t}\n\n\t// Set the status code of the original response to the status code of the proxy response\n\tw.WriteHeader(resp.StatusCode)\n\n\t// Copy the body of the proxy response to the original response\n\tio.Copy(w, resp.Body)\n}\n" }, { "path": "api/jwt_secret_service.go", "content": "package main\n\n// check if jwt_secret and jwt_aud available for 'chat' in database\n// if not, create them\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\n\t\"github.com/rotisserie/eris\"\n\t\"github.com/swuecho/chat_backend/auth\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype JWTSecretService struct {\n\tq *sqlc_queries.Queries\n}\n\n// NewJWTSecretService creates a new JWTSecretService.\nfunc NewJWTSecretService(q *sqlc_queries.Queries) *JWTSecretService {\n\treturn &JWTSecretService{q: q}\n}\n\n// GetJWTSecret returns a jwt_secret by name.\nfunc (s *JWTSecretService) GetJwtSecret(ctx context.Context, name string) (sqlc_queries.JwtSecret, error) {\n\tsecret, err := s.q.GetJwtSecret(ctx, name)\n\tif err != nil {\n\t\treturn sqlc_queries.JwtSecret{}, eris.Wrap(err, \"failed to get secret \")\n\t}\n\treturn secret, nil\n}\n\n// GetOrCreateJwtSecret returns a jwt_secret by name.\n// if jwt_secret does not exist, create it\nfunc (s *JWTSecretService) GetOrCreateJwtSecret(ctx context.Context, name string) (sqlc_queries.JwtSecret, error) {\n\tsecret, err := s.q.GetJwtSecret(ctx, name)\n\tif err != nil {\n\t\t// no row found, create it\n\t\tif errors.Is(err, sql.ErrNoRows) {\n\t\t\tsecret_str, aud_str := auth.GenJwtSecretAndAudience()\n\t\t\tsecret, err = s.q.CreateJwtSecret(ctx, sqlc_queries.CreateJwtSecretParams{\n\t\t\t\tName: name,\n\t\t\t\tSecret: secret_str,\n\t\t\t\tAudience: aud_str,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\treturn sqlc_queries.JwtSecret{}, eris.Wrap(err, \"failed to create secret \")\n\t\t\t}\n\t\t} else {\n\t\t\treturn sqlc_queries.JwtSecret{}, eris.Wrap(err, \"failed to create secret \")\n\t\t}\n\t}\n\treturn secret, nil\n}\n" }, { "path": "api/llm/claude/claude.go", "content": "package claude\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"strings\"\n\n\tmodels \"github.com/swuecho/chat_backend/models\"\n)\n\ntype Delta struct {\n\tType string `json:\"type\"`\n\tText string `json:\"text\"`\n}\n\ntype ContentBlockDelta struct {\n\tType string `json:\"type\"`\n\tIndex int `json:\"index\"`\n\tDelta Delta `json:\"delta\"`\n}\n\ntype ContentBlock struct {\n\tType string `json:\"type\"`\n\tText string `json:\"text\"`\n}\n\ntype StartBlock struct {\n\tType string `json:\"type\"`\n\tIndex int `json:\"index\"`\n\tContentBlock ContentBlock `json:\"content_block\"`\n}\n\nfunc AnswerFromBlockDelta(line []byte) string {\n\tvar response ContentBlockDelta\n\t_ = json.Unmarshal(line, &response)\n\treturn response.Delta.Text\n}\n\nfunc AnswerFromBlockStart(line []byte) string {\n\tvar response StartBlock\n\t_ = json.Unmarshal(line, &response)\n\treturn response.ContentBlock.Text\n}\n\nfunc FormatClaudePrompt(chat_compeletion_messages []models.Message) string {\n\tvar sb strings.Builder\n\n\tfor _, message := range chat_compeletion_messages {\n\n\t\tif message.Role != \"assistant\" {\n\t\t\tsb.WriteString(fmt.Sprintf(\"\\n\\nHuman: %s\\n\\nAssistant: \", message.Content))\n\t\t} else {\n\n\t\t\tsb.WriteString(fmt.Sprintf(\"%s\\n\", message.Content))\n\t\t}\n\t}\n\tprompt := sb.String()\n\treturn prompt\n}\n\n// response (not stream)\n\ntype Response struct {\n\tID string `json:\"id\"`\n\tType string `json:\"type\"`\n\tRole string `json:\"role\"`\n\tModel string `json:\"model\"`\n\tContent []Content `json:\"content\"`\n\tStopReason string `json:\"stop_reason\"`\n\tStopSequence interface{} `json:\"stop_sequence\"`\n\tUsage Usage `json:\"usage\"`\n}\n\ntype Content struct {\n\tType string `json:\"type\"`\n\tText string `json:\"text\"`\n}\n\ntype Usage struct {\n\tInputTokens int `json:\"input_tokens\"`\n\tOutputTokens int `json:\"output_tokens\"`\n}\n" }, { "path": "api/llm/gemini/gemini.go", "content": "package gemini\n\nimport (\n\tb64 \"encoding/base64\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"log\"\n\t\"net/http\"\n\t\"os\"\n\t\"strings\"\n\n\tmapset \"github.com/deckarep/golang-set/v2\"\n\t\"github.com/samber/lo\"\n\tmodels \"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype Part interface {\n\ttoPart() string\n}\n\ntype PartString struct {\n\tText string `json:\"text\"`\n}\n\nfunc TextData(text string) PartString {\n\treturn PartString{\n\t\tText: text,\n\t}\n}\n\nfunc (p *PartString) toPart() string {\n\treturn p.Text\n}\n\ntype PartBlob struct {\n\tBlob Blob `json:\"inlineData\"`\n}\n\nfunc (p PartBlob) toPart() string {\n\tb := p.Blob\n\treturn fmt.Sprintf(\"data:%s;base64,%s\", b.MIMEType, b.Data)\n}\n\n// from https://github.com/google/generative-ai-go/blob/main/genai/generativelanguagepb_veneer.gen.go#L56\n// Blob contains raw media bytes.\n//\n// Text should not be sent as raw bytes, use the 'text' field.\ntype Blob struct {\n\t// The IANA standard MIME type of the source data.\n\t// Examples:\n\t// - image/png\n\t// - image/jpeg\n\t//\n\t// If an unsupported MIME type is provided, an error will be returned. For a\n\t// complete list of supported types, see [Supported file\n\t// formats](https://ai.google.dev/gemini-api/docs/prompting_with_media#supported_file_formats).\n\tMIMEType string `json:\"mimeType\"`\n\t// Raw bytes for media formats.\n\tData string `json:\"data\"`\n}\n\nfunc ImageData(mimeType string, data []byte) Blob {\n\treturn Blob{\n\t\tMIMEType: mimeType,\n\t\tData: b64.StdEncoding.EncodeToString(data),\n\t}\n}\n\ntype GeminiMessage struct {\n\tRole string `json:\"role\"`\n\tParts []Part `json:\"parts\"`\n}\n\ntype GeminPayload struct {\n\tContents []GeminiMessage `json:\"contents\"`\n}\n\ntype Content struct {\n\tParts []struct {\n\t\tText string `json:\"text\"`\n\t\tThought bool `json:\"thought\"`\n\t} `json:\"parts\"`\n\tRole string `json:\"role\"`\n}\n\ntype SafetyRating struct {\n\tCategory string `json:\"category\"`\n\tProbability string `json:\"probability\"`\n}\n\ntype Candidate struct {\n\tContent Content `json:\"content\"`\n\tFinishReason string `json:\"finishReason\"`\n\tIndex int `json:\"index\"`\n\tSafetyRatings []SafetyRating `json:\"safetyRatings\"`\n}\n\ntype PromptFeedback struct {\n\tSafetyRatings []SafetyRating `json:\"safetyRatings\"`\n}\n\ntype ResponseBody struct {\n\tCandidates []Candidate `json:\"candidates\"`\n\tPromptFeedback PromptFeedback `json:\"promptFeedback\"`\n}\n\nfunc ParseRespLine(line []byte, answer string) string {\n\tvar resp ResponseBody\n\tif err := json.Unmarshal(line, &resp); err != nil {\n\t\tfmt.Println(\"Failed to parse request body:\", err)\n\t}\n\n\tfor _, candidate := range resp.Candidates {\n\t\tfor idx, part := range candidate.Content.Parts {\n\t\t\tif idx > 0 {\n\t\t\t\tanswer += \"\\n\\n\"\n\t\t\t}\n\t\t\tif part.Thought {\n\t\t\t\tanswer += (\"\" + part.Text + \"\")\n\t\t\t} else {\n\t\t\t\tanswer += part.Text\n\t\t\t}\n\t\t}\n\n\t}\n\treturn answer\n}\n\n// ParseRespLineDelta extracts only the delta content from a response line without accumulating\nfunc ParseRespLineDelta(line []byte) string {\n\tvar resp ResponseBody\n\tif err := json.Unmarshal(line, &resp); err != nil {\n\t\tfmt.Println(\"Failed to parse request body:\", err)\n\t\treturn \"\"\n\t}\n\n\tvar delta string\n\tfor _, candidate := range resp.Candidates {\n\t\tfor idx, part := range candidate.Content.Parts {\n\t\t\tif idx > 0 {\n\t\t\t\tdelta += \"\\n\\n\"\n\t\t\t}\n\t\t\tif part.Thought {\n\t\t\t\tdelta += (\"\" + part.Text + \"\")\n\t\t\t} else {\n\t\t\t\tdelta += part.Text\n\t\t\t}\n\t\t}\n\t}\n\treturn delta\n}\n\nfunc SupportedMimeTypes() mapset.Set[string] {\n\treturn mapset.NewSet(\n\t\t\"image/png\",\n\t\t\"image/jpeg\",\n\t\t\"image/webp\",\n\t\t\"image/heic\",\n\t\t\"image/heif\",\n\t\t\"audio/wav\",\n\t\t\"audio/mp3\",\n\t\t\"audio/aiff\",\n\t\t\"audio/aac\",\n\t\t\"audio/ogg\",\n\t\t\"audio/flac\",\n\t\t\"video/mp4\",\n\t\t\"video/mpeg\",\n\t\t\"video/mov\",\n\t\t\"video/avi\",\n\t\t\"video/x-flv\",\n\t\t\"video/mpg\",\n\t\t\"video/webm\",\n\t\t\"video/wmv\",\n\t\t\"video/3gpp\",\n\t)\n}\n\nfunc GenGemminPayload(chat_compeletion_messages []models.Message, chatFiles []sqlc_queries.ChatFile) ([]byte, error) {\n\tpayload := GeminPayload{\n\t\tContents: make([]GeminiMessage, len(chat_compeletion_messages)),\n\t}\n\tfor i, message := range chat_compeletion_messages {\n\t\tgeminiMessage := GeminiMessage{\n\t\t\tRole: message.Role,\n\t\t\tParts: []Part{\n\t\t\t\t&PartString{Text: message.Content},\n\t\t\t},\n\t\t}\n\t\tif message.Role == \"assistant\" {\n\t\t\tgeminiMessage.Role = \"model\"\n\t\t} else if message.Role == \"system\" {\n\t\t\tgeminiMessage.Role = \"user\"\n\t\t}\n\t\tpayload.Contents[i] = geminiMessage\n\t}\n\n\tif len(chatFiles) > 0 {\n\t\tpartsFromFiles := lo.Map(chatFiles, func(chatFile sqlc_queries.ChatFile, _ int) Part {\n\t\t\timageExt := SupportedMimeTypes()\n\t\t\tif imageExt.Contains(chatFile.MimeType) {\n\t\t\t\treturn &PartBlob{Blob: ImageData(chatFile.MimeType, chatFile.Data)}\n\t\t\t} else {\n\t\t\t\treturn &PartString{Text: \"file: \" + chatFile.Name + \"\\n<<<\" + string(chatFile.Data) + \">>>\\n\"}\n\t\t\t}\n\t\t})\n\t\tfmt.Printf(\"partsFromFiles: %+v\\n\", partsFromFiles)\n\t\tpayload.Contents[0].Parts = append(payload.Contents[0].Parts, partsFromFiles...)\n\t}\n\n\tpayloadBytes, err := json.Marshal(payload)\n\tlog.Printf(\"\\n%s\\n\", string(payloadBytes))\n\tif err != nil {\n\t\tfmt.Println(\"Error marshalling payload:\", err)\n\t\t// handle err\n\t\treturn nil, err\n\t}\n\treturn payloadBytes, nil\n}\n\ntype ErrorResponse struct {\n\tError struct {\n\t\tCode int `json:\"code\"`\n\t\tMessage string `json:\"message\"`\n\t\tStatus string `json:\"status\"`\n\t} `json:\"error\"`\n}\n\nfunc HandleRegularResponse(client http.Client, req *http.Request) (*models.LLMAnswer, error) {\n\t// Make the request\n\tresp, err := client.Do(req)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to send Gemini API request: %w\", err)\n\t}\n\tdefer resp.Body.Close()\n\n\t// Read response body\n\tbody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to read Gemini response body: %w\", err)\n\t}\n\n\t// Handle non-200 status codes\n\tif resp.StatusCode != http.StatusOK {\n\t\tvar errResp ErrorResponse\n\t\tif jsonErr := json.Unmarshal(body, &errResp); jsonErr == nil && errResp.Error.Message != \"\" {\n\t\t\treturn nil, fmt.Errorf(\"gemini API error: %s (status: %s, code: %d)\",\n\t\t\t\terrResp.Error.Message, errResp.Error.Status, errResp.Error.Code)\n\t\t}\n\t\treturn nil, fmt.Errorf(\"gemini API returned status %d: %s\", resp.StatusCode, string(body))\n\t}\n\n\t// Parse successful response\n\tvar geminiResp ResponseBody\n\tif err := json.Unmarshal(body, &geminiResp); err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to parse Gemini response: %w\", err)\n\t}\n\n\t// Validate response structure\n\tif len(geminiResp.Candidates) == 0 {\n\t\treturn nil, fmt.Errorf(\"no candidates in Gemini response\")\n\t}\n\n\t// Extract answer text\n\tvar answer strings.Builder\n\tfor _, candidate := range geminiResp.Candidates {\n\t\tfor _, part := range candidate.Content.Parts {\n\t\t\tif part.Text != \"\" {\n\t\t\t\tif answer.Len() > 0 {\n\t\t\t\t\tanswer.WriteString(\"\\n\\n\")\n\t\t\t\t}\n\t\t\t\tanswer.WriteString(part.Text)\n\t\t\t}\n\t\t}\n\t}\n\n\tif answer.Len() == 0 {\n\t\treturn nil, fmt.Errorf(\"empty response from Gemini\")\n\t}\n\n\treturn &models.LLMAnswer{\n\t\tAnswer: answer.String(),\n\t\tAnswerId: \"\", // Gemini doesn't provide an ID\n\t}, nil\n}\n\nfunc BuildAPIURL(model string, stream bool) string {\n\tendpoint := \"generateContent\"\n\turl := fmt.Sprintf(\"https://generativelanguage.googleapis.com/v1beta/models/%s:%s?key=$GEMINI_API_KEY\", model, endpoint)\n\tif stream {\n\t\tendpoint = \"streamGenerateContent?alt=sse\"\n\t\turl = fmt.Sprintf(\"https://generativelanguage.googleapis.com/v1beta/models/%s:%s&key=$GEMINI_API_KEY\", model, endpoint)\n\t}\n\treturn os.ExpandEnv(url)\n}\n\ntype GoogleApiError struct {\n\tError struct {\n\t\tCode int `json:\"code\"`\n\t\tMessage string `json:\"message\"`\n\t\tStatus string `json:\"status\"`\n\t\tDetails string `json:\"details,omitempty\"`\n\t} `json:\"error\"`\n}\n\nfunc (gae *GoogleApiError) String() string {\n\tif gae.Error.Message == \"\" {\n\t\treturn \"Unknown Google API Error\"\n\t}\n\treturn fmt.Sprintf(\"Google API Error: Code=%d, Status=%s, Message=%s, Details=[%s]\",\n\t\tgae.Error.Code, gae.Error.Status, gae.Error.Message, gae.Error.Details)\n}\n" }, { "path": "api/llm/gemini/gemini_test.go", "content": "package gemini\n\nimport (\n\t\"os\"\n\t\"testing\"\n)\n\nfunc TestBuildAPIURL(t *testing.T) {\n\t// Set test environment variable\n\tos.Setenv(\"GEMINI_API_KEY\", \"test-key\")\n\tdefer os.Unsetenv(\"GEMINI_API_KEY\")\n\n\ttests := []struct {\n\t\tname string\n\t\tmodel string\n\t\tstream bool\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tname: \"non-streaming request\",\n\t\t\tmodel: \"gemini-pro\",\n\t\t\tstream: false,\n\t\t\texpected: \"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:\" +\n\t\t\t\t\"generateContent?key=test-key\",\n\t\t},\n\t\t{\n\t\t\tname: \"streaming request\",\n\t\t\tmodel: \"gemini-pro\",\n\t\t\tstream: true,\n\t\t\texpected: \"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:\" +\n\t\t\t\t\"streamGenerateContent?alt=sse&key=test-key\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := BuildAPIURL(tt.model, tt.stream)\n\t\t\tif got != tt.expected {\n\t\t\t\tt.Errorf(\"buildAPIURL() = %v, want %v\", got, tt.expected)\n\t\t\t}\n\t\t})\n\t}\n}\n" }, { "path": "api/llm/openai/chat.go", "content": "package openai\n\nimport (\n\t\"encoding/json\"\n\t\"errors\"\n)\n\n// Chat message role defined by the OpenAI API.\nconst (\n\tChatMessageRoleSystem = \"system\"\n\tChatMessageRoleUser = \"user\"\n\tChatMessageRoleAssistant = \"assistant\"\n\tChatMessageRoleFunction = \"function\"\n\tChatMessageRoleTool = \"tool\"\n)\n\nconst chatCompletionsSuffix = \"/chat/completions\"\n\nvar (\n\tErrChatCompletionInvalidModel = errors.New(\"this model is not supported with this method, please use CreateCompletion client method instead\") //nolint:lll\n\tErrChatCompletionStreamNotSupported = errors.New(\"streaming is not supported with this method, please use CreateChatCompletionStream\") //nolint:lll\n\tErrContentFieldsMisused = errors.New(\"can't use both Content and MultiContent properties simultaneously\")\n)\n\ntype Hate struct {\n\tFiltered bool `json:\"filtered\"`\n\tSeverity string `json:\"severity,omitempty\"`\n}\ntype SelfHarm struct {\n\tFiltered bool `json:\"filtered\"`\n\tSeverity string `json:\"severity,omitempty\"`\n}\ntype Sexual struct {\n\tFiltered bool `json:\"filtered\"`\n\tSeverity string `json:\"severity,omitempty\"`\n}\ntype Violence struct {\n\tFiltered bool `json:\"filtered\"`\n\tSeverity string `json:\"severity,omitempty\"`\n}\n\ntype JailBreak struct {\n\tFiltered bool `json:\"filtered\"`\n\tDetected bool `json:\"detected\"`\n}\n\ntype Profanity struct {\n\tFiltered bool `json:\"filtered\"`\n\tDetected bool `json:\"detected\"`\n}\n\ntype ContentFilterResults struct {\n\tHate Hate `json:\"hate,omitempty\"`\n\tSelfHarm SelfHarm `json:\"self_harm,omitempty\"`\n\tSexual Sexual `json:\"sexual,omitempty\"`\n\tViolence Violence `json:\"violence,omitempty\"`\n\tJailBreak JailBreak `json:\"jailbreak,omitempty\"`\n\tProfanity Profanity `json:\"profanity,omitempty\"`\n}\n\ntype PromptAnnotation struct {\n\tPromptIndex int `json:\"prompt_index,omitempty\"`\n\tContentFilterResults ContentFilterResults `json:\"content_filter_results,omitempty\"`\n}\n\ntype ImageURLDetail string\n\nconst (\n\tImageURLDetailHigh ImageURLDetail = \"high\"\n\tImageURLDetailLow ImageURLDetail = \"low\"\n\tImageURLDetailAuto ImageURLDetail = \"auto\"\n)\n\ntype ChatMessageImageURL struct {\n\tURL string `json:\"url,omitempty\"`\n\tDetail ImageURLDetail `json:\"detail,omitempty\"`\n}\n\ntype ChatMessagePartType string\n\nconst (\n\tChatMessagePartTypeText ChatMessagePartType = \"text\"\n\tChatMessagePartTypeImageURL ChatMessagePartType = \"image_url\"\n)\n\ntype ChatMessagePart struct {\n\tType ChatMessagePartType `json:\"type,omitempty\"`\n\tText string `json:\"text,omitempty\"`\n\tImageURL *ChatMessageImageURL `json:\"image_url,omitempty\"`\n}\n\ntype ChatCompletionMessage struct {\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n\tRefusal string `json:\"refusal,omitempty\"`\n\tMultiContent []ChatMessagePart\n\n\t// This property isn't in the official documentation, but it's in\n\t// the documentation for the official library for python:\n\t// - https://github.com/openai/openai-python/blob/main/chatml.md\n\t// - https://github.com/openai/openai-cookbook/blob/main/examples/How_to_count_tokens_with_tiktoken.ipynb\n\tName string `json:\"name,omitempty\"`\n\n\tFunctionCall *FunctionCall `json:\"function_call,omitempty\"`\n\n\t// For Role=assistant prompts this may be set to the tool calls generated by the model, such as function calls.\n\tToolCalls []ToolCall `json:\"tool_calls,omitempty\"`\n\n\t// For Role=tool prompts this should be set to the ID given in the assistant's prior request to call a tool.\n\tToolCallID string `json:\"tool_call_id,omitempty\"`\n}\n\nfunc (m ChatCompletionMessage) MarshalJSON() ([]byte, error) {\n\tif m.Content != \"\" && m.MultiContent != nil {\n\t\treturn nil, ErrContentFieldsMisused\n\t}\n\tif len(m.MultiContent) > 0 {\n\t\tmsg := struct {\n\t\t\tRole string `json:\"role\"`\n\t\t\tContent string `json:\"-\"`\n\t\t\tRefusal string `json:\"refusal,omitempty\"`\n\t\t\tMultiContent []ChatMessagePart `json:\"content,omitempty\"`\n\t\t\tName string `json:\"name,omitempty\"`\n\t\t\tFunctionCall *FunctionCall `json:\"function_call,omitempty\"`\n\t\t\tToolCalls []ToolCall `json:\"tool_calls,omitempty\"`\n\t\t\tToolCallID string `json:\"tool_call_id,omitempty\"`\n\t\t}(m)\n\t\treturn json.Marshal(msg)\n\t}\n\n\tmsg := struct {\n\t\tRole string `json:\"role\"`\n\t\tContent string `json:\"content\"`\n\t\tRefusal string `json:\"refusal,omitempty\"`\n\t\tMultiContent []ChatMessagePart `json:\"-\"`\n\t\tName string `json:\"name,omitempty\"`\n\t\tFunctionCall *FunctionCall `json:\"function_call,omitempty\"`\n\t\tToolCalls []ToolCall `json:\"tool_calls,omitempty\"`\n\t\tToolCallID string `json:\"tool_call_id,omitempty\"`\n\t}(m)\n\treturn json.Marshal(msg)\n}\n\nfunc (m *ChatCompletionMessage) UnmarshalJSON(bs []byte) error {\n\tmsg := struct {\n\t\tRole string `json:\"role\"`\n\t\tContent string `json:\"content\"`\n\t\tRefusal string `json:\"refusal,omitempty\"`\n\t\tMultiContent []ChatMessagePart\n\t\tName string `json:\"name,omitempty\"`\n\t\tFunctionCall *FunctionCall `json:\"function_call,omitempty\"`\n\t\tToolCalls []ToolCall `json:\"tool_calls,omitempty\"`\n\t\tToolCallID string `json:\"tool_call_id,omitempty\"`\n\t}{}\n\n\tif err := json.Unmarshal(bs, &msg); err == nil {\n\t\t*m = ChatCompletionMessage(msg)\n\t\treturn nil\n\t}\n\tmultiMsg := struct {\n\t\tRole string `json:\"role\"`\n\t\tContent string\n\t\tRefusal string `json:\"refusal,omitempty\"`\n\t\tMultiContent []ChatMessagePart `json:\"content\"`\n\t\tName string `json:\"name,omitempty\"`\n\t\tFunctionCall *FunctionCall `json:\"function_call,omitempty\"`\n\t\tToolCalls []ToolCall `json:\"tool_calls,omitempty\"`\n\t\tToolCallID string `json:\"tool_call_id,omitempty\"`\n\t}{}\n\tif err := json.Unmarshal(bs, &multiMsg); err != nil {\n\t\treturn err\n\t}\n\t*m = ChatCompletionMessage(multiMsg)\n\treturn nil\n}\n\ntype ToolCall struct {\n\t// Index is not nil only in chat completion chunk object\n\tIndex *int `json:\"index,omitempty\"`\n\tID string `json:\"id,omitempty\"`\n\tType ToolType `json:\"type\"`\n\tFunction FunctionCall `json:\"function\"`\n}\n\ntype FunctionCall struct {\n\tName string `json:\"name,omitempty\"`\n\t// call function with arguments in JSON format\n\tArguments string `json:\"arguments,omitempty\"`\n}\n\ntype ChatCompletionResponseFormatType string\n\nconst (\n\tChatCompletionResponseFormatTypeJSONObject ChatCompletionResponseFormatType = \"json_object\"\n\tChatCompletionResponseFormatTypeJSONSchema ChatCompletionResponseFormatType = \"json_schema\"\n\tChatCompletionResponseFormatTypeText ChatCompletionResponseFormatType = \"text\"\n)\n\ntype ChatCompletionResponseFormat struct {\n\tType ChatCompletionResponseFormatType `json:\"type,omitempty\"`\n\tJSONSchema *ChatCompletionResponseFormatJSONSchema `json:\"json_schema,omitempty\"`\n}\n\ntype ChatCompletionResponseFormatJSONSchema struct {\n\tName string `json:\"name\"`\n\tDescription string `json:\"description,omitempty\"`\n\tSchema json.Marshaler `json:\"schema\"`\n\tStrict bool `json:\"strict\"`\n}\n\n// ChatCompletionRequest represents a request structure for chat completion API.\ntype ChatCompletionRequest struct {\n\tModel string `json:\"model\"`\n\tMessages []ChatCompletionMessage `json:\"messages\"`\n\t// MaxTokens The maximum number of tokens that can be generated in the chat completion.\n\t// This value can be used to control costs for text generated via API.\n\t// This value is now deprecated in favor of max_completion_tokens, and is not compatible with o1 series models.\n\t// refs: https://platform.openai.com/docs/api-reference/chat/create#chat-create-max_tokens\n\tMaxTokens int `json:\"max_tokens,omitempty\"`\n\t// MaxCompletionTokens An upper bound for the number of tokens that can be generated for a completion,\n\t// including visible output tokens and reasoning tokens https://platform.openai.com/docs/guides/reasoning\n\tMaxCompletionTokens int `json:\"max_completion_tokens,omitempty\"`\n\tTemperature float32 `json:\"temperature,omitempty\"`\n\tTopP float32 `json:\"top_p,omitempty\"`\n\tN int `json:\"n,omitempty\"`\n\tStream bool `json:\"stream,omitempty\"`\n\tStop []string `json:\"stop,omitempty\"`\n\tPresencePenalty float32 `json:\"presence_penalty,omitempty\"`\n\tResponseFormat *ChatCompletionResponseFormat `json:\"response_format,omitempty\"`\n\tSeed *int `json:\"seed,omitempty\"`\n\tFrequencyPenalty float32 `json:\"frequency_penalty,omitempty\"`\n\t// LogitBias is must be a token id string (specified by their token ID in the tokenizer), not a word string.\n\t// incorrect: `\"logit_bias\":{\"You\": 6}`, correct: `\"logit_bias\":{\"1639\": 6}`\n\t// refs: https://platform.openai.com/docs/api-reference/chat/create#chat/create-logit_bias\n\tLogitBias map[string]int `json:\"logit_bias,omitempty\"`\n\t// LogProbs indicates whether to return log probabilities of the output tokens or not.\n\t// If true, returns the log probabilities of each output token returned in the content of message.\n\t// This option is currently not available on the gpt-4-vision-preview model.\n\tLogProbs bool `json:\"logprobs,omitempty\"`\n\t// TopLogProbs is an integer between 0 and 5 specifying the number of most likely tokens to return at each\n\t// token position, each with an associated log probability.\n\t// logprobs must be set to true if this parameter is used.\n\tTopLogProbs int `json:\"top_logprobs,omitempty\"`\n\tUser string `json:\"user,omitempty\"`\n\t// Deprecated: use Tools instead.\n\tFunctions []FunctionDefinition `json:\"functions,omitempty\"`\n\t// Deprecated: use ToolChoice instead.\n\tFunctionCall any `json:\"function_call,omitempty\"`\n\tTools []Tool `json:\"tools,omitempty\"`\n\t// This can be either a string or an ToolChoice object.\n\tToolChoice any `json:\"tool_choice,omitempty\"`\n\t// Options for streaming response. Only set this when you set stream: true.\n\tStreamOptions *StreamOptions `json:\"stream_options,omitempty\"`\n\t// Disable the default behavior of parallel tool calls by setting it: false.\n\tParallelToolCalls any `json:\"parallel_tool_calls,omitempty\"`\n\t// Store can be set to true to store the output of this completion request for use in distillations and evals.\n\t// https://platform.openai.com/docs/api-reference/chat/create#chat-create-store\n\tStore bool `json:\"store,omitempty\"`\n\t// Metadata to store with the completion.\n\tMetadata map[string]string `json:\"metadata,omitempty\"`\n}\n\ntype StreamOptions struct {\n\t// If set, an additional chunk will be streamed before the data: [DONE] message.\n\t// The usage field on this chunk shows the token usage statistics for the entire request,\n\t// and the choices field will always be an empty array.\n\t// All other chunks will also include a usage field, but with a null value.\n\tIncludeUsage bool `json:\"include_usage,omitempty\"`\n}\n\ntype ToolType string\n\nconst (\n\tToolTypeFunction ToolType = \"function\"\n)\n\ntype Tool struct {\n\tType ToolType `json:\"type\"`\n\tFunction *FunctionDefinition `json:\"function,omitempty\"`\n}\n\ntype ToolChoice struct {\n\tType ToolType `json:\"type\"`\n\tFunction ToolFunction `json:\"function,omitempty\"`\n}\n\ntype ToolFunction struct {\n\tName string `json:\"name\"`\n}\n\ntype FunctionDefinition struct {\n\tName string `json:\"name\"`\n\tDescription string `json:\"description,omitempty\"`\n\tStrict bool `json:\"strict,omitempty\"`\n\t// Parameters is an object describing the function.\n\t// You can pass json.RawMessage to describe the schema,\n\t// or you can pass in a struct which serializes to the proper JSON schema.\n\t// The jsonschema package is provided for convenience, but you should\n\t// consider another specialized library if you require more complex schemas.\n\tParameters any `json:\"parameters\"`\n}\n\n// Deprecated: use FunctionDefinition instead.\ntype FunctionDefine = FunctionDefinition\n\ntype TopLogProbs struct {\n\tToken string `json:\"token\"`\n\tLogProb float64 `json:\"logprob\"`\n\tBytes []byte `json:\"bytes,omitempty\"`\n}\n\n// LogProb represents the probability information for a token.\ntype LogProb struct {\n\tToken string `json:\"token\"`\n\tLogProb float64 `json:\"logprob\"`\n\tBytes []byte `json:\"bytes,omitempty\"` // Omitting the field if it is null\n\t// TopLogProbs is a list of the most likely tokens and their log probability, at this token position.\n\t// In rare cases, there may be fewer than the number of requested top_logprobs returned.\n\tTopLogProbs []TopLogProbs `json:\"top_logprobs\"`\n}\n\n// LogProbs is the top-level structure containing the log probability information.\ntype LogProbs struct {\n\t// Content is a list of message content tokens with log probability information.\n\tContent []LogProb `json:\"content\"`\n}\n\ntype FinishReason string\n\nconst (\n\tFinishReasonStop FinishReason = \"stop\"\n\tFinishReasonLength FinishReason = \"length\"\n\tFinishReasonFunctionCall FinishReason = \"function_call\"\n\tFinishReasonToolCalls FinishReason = \"tool_calls\"\n\tFinishReasonContentFilter FinishReason = \"content_filter\"\n\tFinishReasonNull FinishReason = \"null\"\n)\n\nfunc (r FinishReason) MarshalJSON() ([]byte, error) {\n\tif r == FinishReasonNull || r == \"\" {\n\t\treturn []byte(\"null\"), nil\n\t}\n\treturn []byte(`\"` + string(r) + `\"`), nil // best effort to not break future API changes\n}\n\ntype ChatCompletionChoice struct {\n\tIndex int `json:\"index\"`\n\tMessage ChatCompletionMessage `json:\"message\"`\n\t// FinishReason\n\t// stop: API returned complete message,\n\t// or a message terminated by one of the stop sequences provided via the stop parameter\n\t// length: Incomplete model output due to max_tokens parameter or token limit\n\t// function_call: The model decided to call a function\n\t// content_filter: Omitted content due to a flag from our content filters\n\t// null: API response still in progress or incomplete\n\tFinishReason FinishReason `json:\"finish_reason\"`\n\tLogProbs *LogProbs `json:\"logprobs,omitempty\"`\n\tContentFilterResults ContentFilterResults `json:\"content_filter_results,omitempty\"`\n}\n\n// ChatCompletionResponse represents a response structure for chat completion API.\ntype ChatCompletionResponse struct {\n\tID string `json:\"id\"`\n\tObject string `json:\"object\"`\n\tCreated int64 `json:\"created\"`\n\tModel string `json:\"model\"`\n\tChoices []ChatCompletionChoice `json:\"choices\"`\n\tUsage Usage `json:\"usage\"`\n\tSystemFingerprint string `json:\"system_fingerprint\"`\n\tPromptFilterResults []PromptFilterResult `json:\"prompt_filter_results,omitempty\"`\n\n\thttpHeader\n}\n" }, { "path": "api/llm/openai/client.go", "content": "package openai\n\nimport \"net/http\"\n\ntype httpHeader http.Header\n\nfunc (h *httpHeader) SetHeader(header http.Header) {\n\t*h = httpHeader(header)\n}\n\nfunc (h *httpHeader) Header() http.Header {\n\treturn http.Header(*h)\n}\n" }, { "path": "api/llm/openai/common.go", "content": "package openai\n\n// common.go defines common types used throughout the OpenAI API.\n\n// Usage Represents the total token usage per request to OpenAI.\ntype Usage struct {\n\tPromptTokens int `json:\"prompt_tokens\"`\n\tCompletionTokens int `json:\"completion_tokens\"`\n\tTotalTokens int `json:\"total_tokens\"`\n\tPromptTokensDetails *PromptTokensDetails `json:\"prompt_tokens_details\"`\n\tCompletionTokensDetails *CompletionTokensDetails `json:\"completion_tokens_details\"`\n}\n\n// CompletionTokensDetails Breakdown of tokens used in a completion.\ntype CompletionTokensDetails struct {\n\tAudioTokens int `json:\"audio_tokens\"`\n\tReasoningTokens int `json:\"reasoning_tokens\"`\n}\n\n// PromptTokensDetails Breakdown of tokens used in the prompt.\ntype PromptTokensDetails struct {\n\tAudioTokens int `json:\"audio_tokens\"`\n\tCachedTokens int `json:\"cached_tokens\"`\n}\n" }, { "path": "api/llm/openai/openai.go", "content": "package openai\n\n// code is copied from openai go to add reasoningContent field\ntype ChatCompletionStreamChoiceDelta struct {\n\tContent string `json:\"content,omitempty\"`\n\tReasoningContent string `json:\"reasoning_content,omitempty\"`\n\tRole string `json:\"role,omitempty\"`\n\tFunctionCall *FunctionCall `json:\"function_call,omitempty\"`\n\tToolCalls []ToolCall `json:\"tool_calls,omitempty\"`\n\tRefusal string `json:\"refusal,omitempty\"`\n}\n\ntype ChatCompletionStreamChoiceLogprobs struct {\n\tContent []ChatCompletionTokenLogprob `json:\"content,omitempty\"`\n\tRefusal []ChatCompletionTokenLogprob `json:\"refusal,omitempty\"`\n}\n\ntype ChatCompletionTokenLogprob struct {\n\tToken string `json:\"token\"`\n\tBytes []int64 `json:\"bytes,omitempty\"`\n\tLogprob float64 `json:\"logprob,omitempty\"`\n\tTopLogprobs []ChatCompletionTokenLogprobTopLogprob `json:\"top_logprobs\"`\n}\n\ntype ChatCompletionTokenLogprobTopLogprob struct {\n\tToken string `json:\"token\"`\n\tBytes []int64 `json:\"bytes\"`\n\tLogprob float64 `json:\"logprob\"`\n}\n\ntype ChatCompletionStreamChoice struct {\n\tIndex int `json:\"index\"`\n\tDelta ChatCompletionStreamChoiceDelta `json:\"delta\"`\n\tLogprobs *ChatCompletionStreamChoiceLogprobs `json:\"logprobs,omitempty\"`\n\tFinishReason FinishReason `json:\"finish_reason\"`\n\tContentFilterResults ContentFilterResults `json:\"content_filter_results,omitempty\"`\n}\n\ntype PromptFilterResult struct {\n\tIndex int `json:\"index\"`\n\tContentFilterResults ContentFilterResults `json:\"content_filter_results,omitempty\"`\n}\n\ntype ChatCompletionStreamResponse struct {\n\tID string `json:\"id\"`\n\tObject string `json:\"object\"`\n\tCreated int64 `json:\"created\"`\n\tModel string `json:\"model\"`\n\tChoices []ChatCompletionStreamChoice `json:\"choices\"`\n\tSystemFingerprint string `json:\"system_fingerprint\"`\n\tPromptAnnotations []PromptAnnotation `json:\"prompt_annotations,omitempty\"`\n\tPromptFilterResults []PromptFilterResult `json:\"prompt_filter_results,omitempty\"`\n\t// An optional field that will only be present when you set stream_options: {\"include_usage\": true} in your request.\n\t// When present, it contains a null value except for the last chunk which contains the token usage statistics\n\t// for the entire request.\n\tUsage *Usage `json:\"usage,omitempty\"`\n}\n" }, { "path": "api/llm_openai.go", "content": "package main\n\nimport (\n\t\"encoding/base64\"\n\t\"fmt\"\n\t\"log\"\n\t\"net/http\"\n\t\"net/url\"\n\t\"os\"\n\t\"strings\"\n\t\"time\"\n\n\tmapset \"github.com/deckarep/golang-set/v2\"\n\t\"github.com/samber/lo\"\n\topenai \"github.com/sashabaranov/go-openai\"\n\tmodels \"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\nfunc SupportedMimeTypes() mapset.Set[string] {\n\treturn mapset.NewSet(\n\t\t\"image/png\",\n\t\t\"image/jpeg\",\n\t\t\"image/webp\",\n\t\t\"image/heic\",\n\t\t\"image/heif\",\n\t\t\"audio/wav\",\n\t\t\"audio/mp3\",\n\t\t\"audio/aiff\",\n\t\t\"audio/aac\",\n\t\t\"audio/ogg\",\n\t\t\"audio/flac\",\n\t\t\"video/mp4\",\n\t\t\"video/mpeg\",\n\t\t\"video/mov\",\n\t\t\"video/avi\",\n\t\t\"video/x-flv\",\n\t\t\"video/mpg\",\n\t\t\"video/webm\",\n\t\t\"video/wmv\",\n\t\t\"video/3gpp\",\n\t)\n}\n\nfunc messagesToOpenAIMesages(messages []models.Message, chatFiles []sqlc_queries.ChatFile) []openai.ChatCompletionMessage {\n\topen_ai_msgs := lo.Map(messages, func(m models.Message, _ int) openai.ChatCompletionMessage {\n\t\treturn openai.ChatCompletionMessage{Role: m.Role, Content: m.Content}\n\t})\n\tif len(chatFiles) == 0 {\n\t\treturn open_ai_msgs\n\t}\n\tparts := lo.Map(chatFiles, func(m sqlc_queries.ChatFile, _ int) openai.ChatMessagePart {\n\t\tif SupportedMimeTypes().Contains(m.MimeType) {\n\t\t\treturn openai.ChatMessagePart{\n\t\t\t\tType: openai.ChatMessagePartTypeImageURL,\n\t\t\t\tImageURL: &openai.ChatMessageImageURL{\n\t\t\t\t\tURL: byteToImageURL(m.MimeType, m.Data),\n\t\t\t\t\tDetail: openai.ImageURLDetailAuto,\n\t\t\t\t},\n\t\t\t}\n\t\t} else {\n\t\t\treturn openai.ChatMessagePart{\n\t\t\t\tType: openai.ChatMessagePartTypeText,\n\t\t\t\tText: \"file: \" + m.Name + \"\\n<<<\" + string(m.Data) + \">>>\\n\",\n\t\t\t}\n\t\t}\n\t})\n\t// first user message\n\tfirstUserMessage, idx, found := lo.FindIndexOf(open_ai_msgs, func(msg openai.ChatCompletionMessage) bool { return msg.Role == \"user\" })\n\n\tif found {\n\t\tlog.Printf(\"firstUserMessage: %+v\\n\", firstUserMessage)\n\t\topen_ai_msgs[idx].MultiContent = append(\n\t\t\t[]openai.ChatMessagePart{\n\t\t\t\t{Type: openai.ChatMessagePartTypeText, Text: firstUserMessage.Content},\n\t\t\t}, parts...)\n\t\topen_ai_msgs[idx].Content = \"\"\n\t\tlog.Printf(\"firstUserMessage: %+v\\n\", firstUserMessage)\n\t}\n\n\treturn open_ai_msgs\n}\n\nfunc byteToImageURL(mimeType string, data []byte) string {\n\tb64 := fmt.Sprintf(\"data:%s;base64,%s\", mimeType,\n\t\tbase64.StdEncoding.EncodeToString(data))\n\treturn b64\n}\n\nfunc getModelBaseUrl(apiUrl string) (string, error) {\n\tif apiUrl == \"https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions\" {\n\t\treturn \"https://dashscope.aliyuncs.com/compatible-mode/v1\", nil\n\t}\n\t// open router\n\t// https://openrouter.ai/api/v1\n\tif strings.Contains(apiUrl, \"openrouter\") {\n\t\t// keep the url until /v1\n\t\tslashIndex := strings.Index(apiUrl, \"/v1\")\n\t\tif slashIndex > 0 {\n\t\t\treturn apiUrl[:slashIndex] + \"/v1\", nil\n\t\t}\n\t\treturn apiUrl, nil\n\t}\n\tparsedUrl, err := url.Parse(apiUrl)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\tslashIndex := strings.Index(parsedUrl.Path[1:], \"/\")\n\tversion := \"\"\n\t// https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions\n\tif slashIndex > 0 {\n\t\tversion = parsedUrl.Path[1 : slashIndex+1]\n\t}\n\treturn fmt.Sprintf(\"%s://%s/%s\", parsedUrl.Scheme, parsedUrl.Host, version), nil\n}\n\nfunc configOpenAIProxy(config *openai.ClientConfig) {\n\tproxyUrlStr := appConfig.OPENAI.PROXY_URL\n\tif proxyUrlStr != \"\" {\n\t\tproxyUrl, err := url.Parse(proxyUrlStr)\n\t\tif err != nil {\n\t\t\tlog.Printf(\"Error parsing proxy URL: %v\", err)\n\t\t}\n\t\ttransport := &http.Transport{\n\t\t\tProxy: http.ProxyURL(proxyUrl),\n\t\t}\n\t\tconfig.HTTPClient = &http.Client{\n\t\t\tTransport: transport,\n\t\t\tTimeout: 120 * time.Second,\n\t\t}\n\t}\n}\n\nfunc genOpenAIConfig(chatModel sqlc_queries.ChatModel) (openai.ClientConfig, error) {\n\ttoken := os.Getenv(chatModel.ApiAuthKey)\n\tbaseUrl, err := getModelBaseUrl(chatModel.Url)\n\tlog.Printf(\"baseUrl: %s\\n\", baseUrl)\n\tif err != nil {\n\t\treturn openai.ClientConfig{}, err\n\t}\n\n\tvar config openai.ClientConfig\n\tif os.Getenv(\"AZURE_RESOURCE_NAME\") != \"\" {\n\t\tconfig = openai.DefaultAzureConfig(token, chatModel.Url)\n\t\tconfig.AzureModelMapperFunc = func(model string) string {\n\t\t\tazureModelMapping := map[string]string{\n\t\t\t\t\"gpt-3.5-turbo\": os.Getenv(\"AZURE_RESOURCE_NAME\"),\n\t\t\t}\n\t\t\treturn azureModelMapping[model]\n\t\t}\n\t} else {\n\t\tconfig = openai.DefaultConfig(token)\n\t\tconfig.BaseURL = baseUrl\n\t\t// two minutes timeout\n\t\t// config.HTTPClient.Timeout = 120 * time.Second\n\t\tconfigOpenAIProxy(&config)\n\t}\n\treturn config, err\n}\n" }, { "path": "api/llm_summary.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"log\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/tmc/langchaingo/chains\"\n\t\"github.com/tmc/langchaingo/documentloaders\"\n\t\"github.com/tmc/langchaingo/llms/openai\"\n\t\"github.com/tmc/langchaingo/textsplitter\"\n)\n\nfunc llm_summarize_with_timeout(baseURL, content string) string {\n\t// Create a context with a 20 second timeout\n\tctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)\n\tdefer cancel()\n\n\t// Call the summarize function with the context\n\tsummary := llm_summarize(ctx, baseURL, content)\n\n\treturn summary\n}\n\nfunc llm_summarize(ctx context.Context, baseURL string, doc string) string {\n\tbaseURL = strings.TrimSuffix(baseURL, \"/v1\")\n\tllm, err := openai.New(\n\t\topenai.WithToken(appConfig.OPENAI.API_KEY),\n\t\topenai.WithBaseURL(baseURL),\n\t)\n\tif err != nil {\n\t\tlog.Printf(\"failed to create openai client %s: %v\", baseURL, err)\n\t\treturn \"\"\n\t}\n\n\tllmSummarizationChain := chains.LoadRefineSummarization(llm)\n\tdocs, _ := documentloaders.NewText(strings.NewReader(doc)).LoadAndSplit(ctx,\n\t\ttextsplitter.NewRecursiveCharacter(),\n\t)\n\toutputValues, err := chains.Call(ctx, llmSummarizationChain, map[string]any{\"input_documents\": docs})\n\tif err != nil {\n\t\tlog.Printf(\"failed to call chain: %s, %v\", baseURL, err)\n\t\treturn \"\"\n\t}\n\tout := outputValues[\"text\"].(string)\n\treturn out\n}\n" }, { "path": "api/main.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t_ \"embed\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"os\"\n\t\"path/filepath\"\n\t\"reflect\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/gorilla/handlers\"\n\t\"github.com/gorilla/mux\"\n\t_ \"github.com/lib/pq\"\n\tlog \"github.com/sirupsen/logrus\"\n\t\"github.com/spf13/viper\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n\t\"github.com/swuecho/chat_backend/static\"\n\t\"golang.org/x/time/rate\"\n)\n\nvar logger *log.Logger\n\ntype AppConfig struct {\n\tOPENAI struct {\n\t\tAPI_KEY string\n\t\tRATELIMIT int\n\t\tPROXY_URL string\n\t}\n\tCLAUDE struct {\n\t\tAPI_KEY string\n\t}\n\tPG struct {\n\t\tHOST string\n\t\tPORT int\n\t\tUSER string\n\t\tPASS string\n\t\tDB string\n\t}\n}\n\nvar appConfig AppConfig\nvar jwtSecretAndAud sqlc_queries.JwtSecret\n\nfunc getFlattenKeys(prefix string, v reflect.Value) (keys []string) {\n\tswitch v.Kind() {\n\tcase reflect.Struct:\n\t\tfor i := 0; i < v.NumField(); i++ {\n\t\t\tfield := v.Field(i)\n\t\t\tname := v.Type().Field(i).Name\n\t\t\tkeys = append(keys, getFlattenKeys(prefix+name+\".\", field)...)\n\t\t}\n\tdefault:\n\t\tkeys = append(keys, prefix[:len(prefix)-1])\n\t}\n\treturn keys\n}\n\nfunc bindEnvironmentVariables() {\n\tappConfig = AppConfig{}\n\tfor _, key := range getFlattenKeys(\"\", reflect.ValueOf(appConfig)) {\n\t\tenvKey := strings.ToUpper(strings.ReplaceAll(key, \".\", \"_\"))\n\t\terr := viper.BindEnv(key, envKey)\n\t\tif err != nil {\n\t\t\tlogger.Fatal(\"config: unable to bind env: \" + err.Error())\n\t\t}\n\t}\n}\n\n//go:embed sqlc/schema.sql\nvar schemaBytes []byte\n\n// lastRequest tracks the last time a request was received\nvar lastRequest time.Time\nvar openAIRateLimiter *rate.Limiter\n\nfunc main() {\n\n\t// Allow only 3000 requests per minute, with burst 500\n\topenAIRateLimiter = rate.NewLimiter(rate.Every(time.Minute/3000), 500)\n\n\t// A buffered channel with capacity 1\n\t// This ensures only one API call can proceed at a time\n\n\tlastRequest = time.Now()\n\t// Configure viper to read environment variables\n\tbindEnvironmentVariables()\n\tviper.AutomaticEnv()\n\n\tif err := viper.Unmarshal(&appConfig); err != nil {\n\t\tlogger.Fatal(\"config: unable to decode into struct: \" + err.Error())\n\t}\n\n\tlog.Printf(\"%+v\", appConfig)\n\tlogger = log.New()\n\tlogger.Formatter = &log.JSONFormatter{}\n\n\t// Establish a database connection\n\tdbURL := os.Getenv(\"DATABASE_URL\")\n\tvar connStr string\n\tif dbURL == \"\" {\n\t\tpg := appConfig.PG\n\t\tconnStr = fmt.Sprintf(\"host=%s port=%d user=%s password=%s dbname=%s sslmode=disable\",\n\t\t\tpg.HOST, pg.PORT, pg.USER, pg.PASS, pg.DB)\n\t\tprint(connStr)\n\t} else {\n\t\tconnStr = dbURL\n\t}\n\tpgdb, err := sql.Open(\"postgres\", connStr)\n\tif err != nil {\n\t\tlog.Fatal(err)\n\t}\n\tdefer pgdb.Close()\n\n\t// Get current executable file path\n\tex, err := os.Executable()\n\tif err != nil {\n\t\tlog.WithError(err).Fatal(\"Failed to get executable path\")\n\t}\n\n\t// Get current project directory\n\tprojectDir := filepath.Dir(ex)\n\n\t// Print project directory\n\tfmt.Println(projectDir)\n\n\tsqlStatements := string(schemaBytes)\n\n\t// Execute SQL statements\n\t_, err = pgdb.Exec(sqlStatements)\n\tif err != nil {\n\t\tlog.WithError(err).Fatal(\"Failed to execute SQL schema statements\")\n\t}\n\tfmt.Println(\"SQL statements executed successfully\")\n\n\t// create a new Gorilla Mux router instance\n\t// Create a new router\n\trouter := mux.NewRouter()\n\n\tapiRouter := router.PathPrefix(\"/api\").Subrouter()\n\n\tsqlc_q := sqlc_queries.New(pgdb)\n\tsecretService := NewJWTSecretService(sqlc_q)\n\tjwtSecretAndAud, err = secretService.GetOrCreateJwtSecret(context.Background(), \"chat\")\n\tif err != nil {\n\t\tlog.Fatal(err)\n\t}\n\n\t// Create separate subrouters for admin and user routes\n\tadminRouter := apiRouter.PathPrefix(\"/admin\").Subrouter()\n\tuserRouter := apiRouter.NewRoute().Subrouter()\n\n\t// Apply different middleware to admin and user routes\n\tadminRouter.Use(AdminAuthMiddleware)\n\tuserRouter.Use(UserAuthMiddleware)\n\n\tChatModelHandler := NewChatModelHandler(sqlc_q)\n\tChatModelHandler.Register(userRouter) // Chat models for regular users\n\n\t// create a new AuthUserHandler instance for user routes\n\tuserHandler := NewAuthUserHandler(sqlc_q)\n\t// register authenticated routes with the user router\n\tuserHandler.Register(userRouter)\n\t// register public routes (login/signup) with the api router (no auth required)\n\tuserHandler.RegisterPublicRoutes(apiRouter)\n\n\t// create a new AdminHandler instance for admin-only routes\n\tauthUserService := NewAuthUserService(sqlc_q)\n\tadminHandler := NewAdminHandler(authUserService)\n\t// register the AdminHandler with the admin router (will remove /admin prefix automatically)\n\tadminHandler.RegisterRoutes(adminRouter)\n\n\tpromptHandler := NewChatPromptHandler(sqlc_q)\n\tpromptHandler.Register(userRouter)\n\n\tchatSessionHandler := NewChatSessionHandler(sqlc_q)\n\tchatSessionHandler.Register(userRouter)\n\n\t// Register active session handler before workspace handler to avoid route shadowing\n\tactiveSessionHandler := NewUserActiveChatSessionHandler(sqlc_q)\n\tactiveSessionHandler.Register(userRouter)\n\n\tchatWorkspaceHandler := NewChatWorkspaceHandler(sqlc_q)\n\tchatWorkspaceHandler.Register(userRouter)\n\n\tchatMessageHandler := NewChatMessageHandler(sqlc_q)\n\tchatMessageHandler.Register(userRouter)\n\n\tchatSnapshotHandler := NewChatSnapshotHandler(sqlc_q)\n\tchatSnapshotHandler.Register(userRouter)\n\n\t// create a new ChatHandler instance\n\tchatHandler := NewChatHandler(sqlc_q)\n\tchatHandler.Register(userRouter)\n\n\tuser_model_privilege_handler := NewUserChatModelPrivilegeHandler(sqlc_q)\n\tuser_model_privilege_handler.Register(userRouter)\n\n\tchatFileHandler := NewChatFileHandler(sqlc_q)\n\tchatFileHandler.Register(userRouter)\n\n\tchatCommentHandler := NewChatCommentHandler(sqlc_q)\n\tchatCommentHandler.Register(userRouter)\n\n\tbotAnswerHistoryHandler := NewBotAnswerHistoryHandler(sqlc_q)\n\tbotAnswerHistoryHandler.Register(userRouter)\n\n\tapiRouter.HandleFunc(\"/tts\", handleTTSRequest)\n\tapiRouter.HandleFunc(\"/errors\", ErrorCatalogHandler)\n\n\t// Embed static/* directory\n\tfs := http.FileServer(http.FS(static.StaticFiles))\n\n\t// Set cache headers for static/assets files\n\tcacheHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tif strings.HasPrefix(r.URL.Path, \"/static/\") {\n\t\t\tw.Header().Set(\"Cache-Control\", \"max-age=31536000\") // 1 year\n\t\t} else if r.URL.Path == \"\" {\n\t\t\tw.Header().Set(\"Cache-Control\", \"no-cache, no-store, must-revalidate\")\n\t\t\tw.Header().Set(\"Pragma\", \"no-cache\")\n\t\t\tw.Header().Set(\"Expires\", \"0\")\n\t\t}\n\t\tfs.ServeHTTP(w, r)\n\t})\n\n\trouter.PathPrefix(\"/\").Handler(makeGzipHandler(cacheHandler))\n\n\t// fly.io\n\tif os.Getenv(\"FLY_APP_NAME\") != \"\" {\n\t\trouter.Use(UpdateLastRequestTime)\n\t}\n\n\t// Apply rate limiting to authenticated routes only\n\tlimitedRouter := RateLimitByUserID(sqlc_q)\n\tadminRouter.Use(limitedRouter)\n\tuserRouter.Use(limitedRouter)\n\n\t// Public routes (apiRouter) don't need authentication or rate limiting\n\t// TTS and errors endpoints are public\n\t// Add CORS middleware to handle cross-origin requests\n\tdefaultOrigins := []string{\"http://localhost:9002\", \"http://localhost:3000\"}\n\tallowedOrigins := append([]string{}, defaultOrigins...)\n\trestrictToConfigured := false\n\tif corsOrigins := os.Getenv(\"CORS_ALLOWED_ORIGINS\"); corsOrigins != \"\" {\n\t\tparts := strings.Split(corsOrigins, \",\")\n\t\tallowedOrigins = allowedOrigins[:0]\n\t\tfor _, origin := range parts {\n\t\t\ttrimmed := strings.TrimSpace(origin)\n\t\t\tif trimmed == \"\" {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tallowedOrigins = append(allowedOrigins, trimmed)\n\t\t}\n\t\trestrictToConfigured = true\n\t}\n\n\toriginValidator := func(origin string) bool {\n\t\tif len(allowedOrigins) == 0 {\n\t\t\treturn true\n\t\t}\n\t\tfor _, allowed := range allowedOrigins {\n\t\t\tif allowed == \"*\" {\n\t\t\t\treturn true\n\t\t\t}\n\t\t\tif strings.EqualFold(origin, allowed) {\n\t\t\t\treturn true\n\t\t\t}\n\t\t}\n\t\tif !restrictToConfigured {\n\t\t\tif strings.HasPrefix(origin, \"http://localhost:\") || strings.HasPrefix(origin, \"http://127.0.0.1:\") {\n\t\t\t\treturn true\n\t\t\t}\n\t\t\tif strings.HasPrefix(origin, \"https://localhost:\") || strings.HasPrefix(origin, \"https://127.0.0.1:\") {\n\t\t\t\treturn true\n\t\t\t}\n\t\t}\n\t\treturn false\n\t}\n\n\tcorsOptions := handlers.CORS(\n\t\thandlers.AllowedOriginValidator(originValidator),\n\t\thandlers.AllowedMethods([]string{\"GET\", \"POST\", \"PUT\", \"DELETE\", \"OPTIONS\"}),\n\t\thandlers.AllowedHeaders([]string{\"Content-Type\", \"Authorization\", \"Cache-Control\", \"Connection\", \"Pragma\", \"Accept\", \"Accept-Language\", \"Origin\", \"Referer\"}),\n\t\thandlers.AllowCredentials(),\n\t)\n\n\t// Wrap the router with CORS and logging middleware\n\tcorsRouter := corsOptions(router)\n\tloggedRouter := handlers.LoggingHandler(logger.Out, corsRouter)\n\n\trouter.Walk(func(route *mux.Route, router *mux.Router, ancestors []*mux.Route) error {\n\t\ttpl, err1 := route.GetPathTemplate()\n\t\tmet, err2 := route.GetMethods()\n\t\tfmt.Println(tpl, err1, met, err2)\n\t\treturn nil\n\t})\n\t// fly.io\n\n\tif os.Getenv(\"FLY_APP_NAME\") != \"\" {\n\t\t// read env var FLY_RESTART_INTERVAL_IF_IDLE if not set, set to 30 minutes\n\t\trestartInterval := os.Getenv(\"FLY_RESTART_INTERVAL_IF_IDLE\")\n\n\t\t// If not set, default to 30 minutes\n\t\tif restartInterval == \"\" {\n\t\t\trestartInterval = \"30m\"\n\t\t}\n\n\t\tduration, err := time.ParseDuration(restartInterval)\n\t\tif err != nil {\n\t\t\tlog.Println(\"Invalid FLY_RESTART_INTERVAL_IF_IDLE value. Exiting.\")\n\t\t}\n\t\t// Use a goroutine to check for inactivity and exit\n\t\tgo func() {\n\t\t\tfor {\n\t\t\t\ttime.Sleep(1 * time.Minute) // Check every minute\n\t\t\t\tif time.Since(lastRequest) > duration {\n\t\t\t\t\tfmt.Printf(\"No activity for %s. Exiting.\", restartInterval)\n\t\t\t\t\tos.Exit(0)\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t}\n\t\t}()\n\n\t}\n\n\terr = http.ListenAndServe(\":8080\", loggedRouter)\n\tif err != nil {\n\t\tlog.Fatal(err)\n\t}\n}\n" }, { "path": "api/main_test.go", "content": "package main\n\nimport (\n\t\"database/sql\"\n\t\"fmt\"\n\t\"io\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t_ \"github.com/lib/pq\"\n\t\"github.com/ory/dockertest/v3\"\n\t\"github.com/ory/dockertest/v3/docker\"\n\tlog \"github.com/sirupsen/logrus\"\n)\n\nvar db *sql.DB\n\nfunc TestMain(m *testing.M) {\n\t// uses a sensible default on windows (tcp/http) and linux/osx (socket)\n\tpool, err := dockertest.NewPool(\"\")\n\tif err != nil {\n\t\tlog.Fatalf(\"Could not construct pool: %s\", err)\n\t}\n\n\terr = pool.Client.Ping()\n\tif err != nil {\n\t\tlog.Fatalf(\"Could not connect to Docker: %s\", err)\n\t}\n\n\t// pulls an image, creates a container based on it and runs it\n\tresource, err := pool.RunWithOptions(&dockertest.RunOptions{\n\t\tRepository: \"postgres\",\n\t\tTag: \"12\",\n\t\tEnv: []string{\n\t\t\t\"POSTGRES_PASSWORD=secret\",\n\t\t\t\"POSTGRES_USER=user_name\",\n\t\t\t\"POSTGRES_DB=dbname\",\n\t\t\t\"listen_addresses = '*'\",\n\t\t},\n\t}, func(config *docker.HostConfig) {\n\t\t// set AutoRemove to true so that stopped container goes away by itself\n\t\tconfig.AutoRemove = true\n\t\tconfig.RestartPolicy = docker.RestartPolicy{Name: \"no\"}\n\t})\n\tif err != nil {\n\t\tlog.Fatalf(\"Could not start resource: %s\", err)\n\t}\n\n\thostAndPort := resource.GetHostPort(\"5432/tcp\")\n\tdatabaseUrl := fmt.Sprintf(\"postgres://user_name:secret@%s/dbname?sslmode=disable\", hostAndPort)\n\n\tlog.Println(\"Connecting to database on url: \", databaseUrl)\n\n\tresource.Expire(120) // Tell docker to hard kill the container in 120 seconds\n\n\t// exponential backoff-retry, because the application in the container might not be ready to accept connections yet\n\tpool.MaxWait = 120 * time.Second\n\tif err = pool.Retry(func() error {\n\t\tdb, err = sql.Open(\"postgres\", databaseUrl)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn db.Ping()\n\t}); err != nil {\n\t\tlog.Fatalf(\"Could not connect to docker: %s\", err)\n\t}\n\t//////// init schema\n\tfile, err := os.Open(\"./sqlc/schema.sql\")\n\n\tif err != nil {\n\n\t\tlog.Fatalf(\"Could not open file: %s\", err)\n\t}\n\tdefer file.Close()\n\n\tbytes, err := io.ReadAll(file)\n\tif err != nil {\n\t\tlog.Fatalf(\"Could not read file: %s\", err)\n\t}\n\n\t_, err = db.Exec(string(bytes))\n\tif err != nil {\n\t\tlog.Fatalf(\"Could not execute SQL: %s\", err)\n\t}\n\n\t//Run tests\n\tcode := m.Run()\n\n\t// You can't defer this because os.Exit doesn't care for defer\n\tif err := pool.Purge(resource); err != nil {\n\t\tlog.Fatalf(\"Could not purge resource: %s\", err)\n\t}\n\n\tos.Exit(code)\n}\n" }, { "path": "api/middleware_authenticate.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"strings\"\n\n\tjwt \"github.com/golang-jwt/jwt/v5\"\n\t\"github.com/swuecho/chat_backend/auth\"\n)\n\nfunc CheckPermission(userID int, ctx context.Context) bool {\n\tcontextUserID, ok := ctx.Value(\"user_id\").(int)\n\tif !ok {\n\t\treturn false\n\t}\n\trole, ok := ctx.Value(\"role\").(string)\n\tif !ok {\n\t\treturn false\n\t}\n\n\tswitch role {\n\tcase \"admin\":\n\t\treturn true\n\tcase \"member\":\n\t\treturn userID == contextUserID\n\tdefault:\n\t\treturn false\n\t}\n}\n\ntype AuthTokenResult struct {\n\tToken *jwt.Token\n\tClaims jwt.MapClaims\n\tUserID string\n\tRole string\n\tTokenType string\n\tValid bool\n\tError *APIError\n}\n\nfunc extractBearerToken(r *http.Request) string {\n\t// Extract from Authorization header for access tokens\n\tbearerToken := r.Header.Get(\"Authorization\")\n\ttokenParts := strings.Split(bearerToken, \" \")\n\tif len(tokenParts) == 2 {\n\t\treturn tokenParts[1]\n\t}\n\treturn \"\"\n}\n\nfunc createUserContext(r *http.Request, userID, role string) *http.Request {\n\tctx := context.WithValue(r.Context(), userContextKey, userID)\n\tctx = context.WithValue(ctx, roleContextKey, role)\n\treturn r.WithContext(ctx)\n}\n\nfunc parseAndValidateJWT(bearerToken string, expectedTokenType string) *AuthTokenResult {\n\tresult := &AuthTokenResult{}\n\n\tif bearerToken == \"\" {\n\t\terr := ErrAuthInvalidCredentials\n\t\terr.Detail = \"Authorization token required\"\n\t\tresult.Error = &err\n\t\treturn result\n\t}\n\n\tjwtSigningKey := []byte(jwtSecretAndAud.Secret)\n\ttoken, err := jwt.Parse(bearerToken, func(token *jwt.Token) (any, error) {\n\t\tif _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {\n\t\t\treturn nil, fmt.Errorf(\"invalid JWT signing method\")\n\t\t}\n\t\treturn jwtSigningKey, nil\n\t})\n\n\tif err != nil {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.Detail = \"Invalid authorization token\"\n\t\tresult.Error = &apiErr\n\t\treturn result\n\t}\n\n\tif !token.Valid {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.Detail = \"Token is not valid\"\n\t\tresult.Error = &apiErr\n\t\treturn result\n\t}\n\n\tclaims, ok := token.Claims.(jwt.MapClaims)\n\tif !ok {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.Detail = \"Cannot parse token claims\"\n\t\tresult.Error = &apiErr\n\t\treturn result\n\t}\n\n\tuserID, ok := claims[\"user_id\"].(string)\n\tif !ok {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.Detail = \"User ID not found in token\"\n\t\tresult.Error = &apiErr\n\t\treturn result\n\t}\n\n\trole, ok := claims[\"role\"].(string)\n\tif !ok {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.Detail = \"User role not found in token\"\n\t\tresult.Error = &apiErr\n\t\treturn result\n\t}\n\n\ttokenType, ok := claims[\"token_type\"].(string)\n\tif !ok {\n\t\t// Legacy forever tokens were generated before the token_type claim existed.\n\t\t// Treat them as access tokens so they remain usable.\n\t\tif expectedTokenType == \"\" || expectedTokenType == auth.TokenTypeAccess {\n\t\t\ttokenType = auth.TokenTypeAccess\n\t\t} else {\n\t\t\tapiErr := ErrAuthInvalidCredentials\n\t\t\tapiErr.Detail = \"Token type not found in token\"\n\t\t\tresult.Error = &apiErr\n\t\t\treturn result\n\t\t}\n\t}\n\n\tif expectedTokenType != \"\" && tokenType != expectedTokenType {\n\t\tapiErr := ErrAuthInvalidCredentials\n\t\tapiErr.Detail = \"Token type is not valid for this operation\"\n\t\tresult.Error = &apiErr\n\t\treturn result\n\t}\n\n\tresult.Token = token\n\tresult.Claims = claims\n\tresult.UserID = userID\n\tresult.Role = role\n\tresult.TokenType = tokenType\n\tresult.Valid = true\n\treturn result\n}\n\ntype contextKey string\n\nconst (\n\troleContextKey contextKey = \"role\"\n\tuserContextKey contextKey = \"user\"\n\tguidContextKey contextKey = \"guid\"\n)\nconst snapshotPrefix = \"/api/uuid/chat_snapshot/\"\n\nfunc IsChatSnapshotUUID(r *http.Request) bool {\n\t// Check http method is GET\n\tif r.Method != http.MethodGet {\n\t\treturn false\n\t}\n\t// Check if request url path has the required prefix and does not have \"/all\" suffix\n\tif strings.HasPrefix(r.URL.Path, snapshotPrefix) && !strings.HasSuffix(r.URL.Path, \"/all\") {\n\t\treturn true\n\t}\n\treturn false\n}\n\nfunc AdminOnlyHandler(h http.Handler) http.Handler {\n\treturn http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tctx := r.Context()\n\t\tuserRole, ok := ctx.Value(roleContextKey).(string)\n\t\tif !ok {\n\t\t\tapiErr := ErrAuthAdminRequired\n\t\t\tapiErr.Detail = \"User role information not found\"\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\t\tif userRole != \"admin\" {\n\t\t\tapiErr := ErrAuthAdminRequired\n\t\t\tapiErr.Detail = \"Current user does not have admin role\"\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\t\th.ServeHTTP(w, r)\n\t})\n}\n\nfunc AdminOnlyHandlerFunc(handlerFunc http.HandlerFunc) http.HandlerFunc {\n\treturn func(w http.ResponseWriter, r *http.Request) {\n\t\tctx := r.Context()\n\t\tuserRole, ok := ctx.Value(roleContextKey).(string)\n\t\tif !ok {\n\t\t\tapiErr := ErrAuthAdminRequired\n\t\t\tapiErr.Detail = \"User role information not found\"\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\t\tif userRole != \"admin\" {\n\t\t\tapiErr := ErrAuthAdminRequired\n\t\t\tapiErr.Detail = \"Current user does not have admin role\"\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\t\thandlerFunc(w, r)\n\t}\n}\n\n// AdminRouteMiddleware applies admin-only protection to all routes in a subrouter\nfunc AdminRouteMiddleware(next http.Handler) http.Handler {\n\treturn http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tctx := r.Context()\n\t\tuserRole, ok := ctx.Value(roleContextKey).(string)\n\t\tif !ok {\n\t\t\tapiErr := ErrAuthAdminRequired\n\t\t\tapiErr.Detail = \"User role information not found\"\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\t\tif userRole != \"admin\" {\n\t\t\tapiErr := ErrAuthAdminRequired\n\t\t\tapiErr.Detail = \"Admin privileges required for this endpoint\"\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\t\tnext.ServeHTTP(w, r)\n\t})\n}\n\n// AdminAuthMiddleware - Authentication middleware specifically for admin routes\nfunc AdminAuthMiddleware(handler http.Handler) http.Handler {\n\treturn http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tbearerToken := extractBearerToken(r)\n\t\tresult := parseAndValidateJWT(bearerToken, auth.TokenTypeAccess)\n\n\t\tif result.Error != nil {\n\t\t\tRespondWithAPIError(w, *result.Error)\n\t\t\treturn\n\t\t}\n\n\t\t// Admin-only check\n\t\tif result.Role != \"admin\" {\n\t\t\tapiErr := ErrAuthAdminRequired\n\t\t\tapiErr.Detail = \"Admin privileges required\"\n\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\treturn\n\t\t}\n\n\t\t// Add user context and proceed\n\t\thandler.ServeHTTP(w, createUserContext(r, result.UserID, result.Role))\n\t})\n}\n\n// UserAuthMiddleware - Authentication middleware for regular user routes\nfunc UserAuthMiddleware(handler http.Handler) http.Handler {\n\treturn http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tbearerToken := extractBearerToken(r)\n\t\tresult := parseAndValidateJWT(bearerToken, auth.TokenTypeAccess)\n\n\t\tif result.Error != nil {\n\t\t\tRespondWithAPIError(w, *result.Error)\n\t\t\treturn\n\t\t}\n\n\t\t// Add user context and proceed (no role restrictions for user middleware)\n\t\thandler.ServeHTTP(w, createUserContext(r, result.UserID, result.Role))\n\t})\n}\n\nfunc IsAuthorizedMiddleware(handler http.Handler) http.Handler {\n\tnoAuthPaths := map[string]bool{\n\t\t\"/\": true,\n\t\t\"/favicon.ico\": true,\n\t\t\"/api/login\": true,\n\t\t\"/api/signup\": true,\n\t\t\"/api/tts\": true,\n\t\t\"/api/errors\": true,\n\t}\n\treturn http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tif _, ok := noAuthPaths[r.URL.Path]; ok || strings.HasPrefix(r.URL.Path, \"/static\") || IsChatSnapshotUUID(r) {\n\t\t\thandler.ServeHTTP(w, r)\n\t\t\treturn\n\t\t}\n\n\t\tbearerToken := extractBearerToken(r)\n\t\tresult := parseAndValidateJWT(bearerToken, auth.TokenTypeAccess)\n\n\t\tif result.Error != nil {\n\t\t\tRespondWithAPIError(w, *result.Error)\n\t\t\treturn\n\t\t}\n\n\t\tif result.Valid {\n\t\t\t// superuser\n\t\t\tif strings.HasPrefix(r.URL.Path, \"/admin\") && result.Role != \"admin\" {\n\t\t\t\tapiErr := ErrAuthAdminRequired\n\t\t\t\tapiErr.Detail = \"This endpoint requires admin privileges\"\n\t\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// TODO: get trace id and add it to context\n\t\t\t//traceID := r.Header.Get(\"X-Request-Id\")\n\t\t\t//if len(traceID) > 0 {\n\t\t\t//ctx = context.WithValue(ctx, guidContextKey, traceID)\n\t\t\t//}\n\t\t\t// Store user ID and role in the request context\n\t\t\t// pass token to request\n\t\t\thandler.ServeHTTP(w, createUserContext(r, result.UserID, result.Role))\n\t\t}\n\t})\n}\n" }, { "path": "api/middleware_gzip.go", "content": "package main\n\nimport (\n\t\"compress/gzip\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n)\n\ntype gzipResponseWriter struct {\n\tio.Writer\n\thttp.ResponseWriter\n}\n\n// Use the Writer part of gzipResponseWriter to write the output.\n\nfunc (w gzipResponseWriter) Write(b []byte) (int, error) {\n\treturn w.Writer.Write(b)\n}\n\nfunc makeGzipHandler(fn http.HandlerFunc) http.HandlerFunc {\n\treturn func(w http.ResponseWriter, r *http.Request) {\n\t\t// Check if the client can accept the gzip encoding.\n\t\tif !strings.Contains(r.Header.Get(\"Accept-Encoding\"), \"gzip\") {\n\t\t\t// The client cannot accept it, so return the output\n\t\t\t// uncompressed.\n\t\t\tfn(w, r)\n\t\t\treturn\n\t\t}\n\t\t// Set the HTTP header indicating encoding.\n\t\tw.Header().Set(\"Content-Encoding\", \"gzip\")\n\t\tgz := gzip.NewWriter(w)\n\t\tdefer gz.Close()\n\t\tfn(gzipResponseWriter{Writer: gz, ResponseWriter: w}, r)\n\t}\n}\n" }, { "path": "api/middleware_lastRequestTime.go", "content": "package main\n\nimport (\n\t\"net/http\"\n\t\"time\"\n)\n\n// Middleware to update lastRequest time\nfunc UpdateLastRequestTime(next http.Handler) http.Handler {\n\treturn http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t// Update lastRequest time\n\t\tlastRequest = time.Now()\n\n\t\t// Call next middleware/handler\n\t\tnext.ServeHTTP(w, r)\n\t})\n}\n" }, { "path": "api/middleware_rateLimit.go", "content": "package main\n\nimport (\n\t\"fmt\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// This function returns a middleware that limits requests from each user by their ID.\nfunc RateLimitByUserID(q *sqlc_queries.Queries) func(http.Handler) http.Handler {\n\treturn func(next http.Handler) http.Handler {\n\t\treturn http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\n\t\t\t// Get the user ID from the request, e.g. from a JWT token.\n\t\t\tpath := r.URL.Path\n\t\t\tif strings.HasSuffix(path, \"/chat\") || strings.HasSuffix(path, \"/chat_stream\") || strings.HasSuffix(path, \"/chatbot\") {\n\t\t\t\tctx := r.Context()\n\t\t\t\tuserIDInt, err := getUserID(ctx)\n\t\t\t\t// role := ctx.Value(roleContextKey).(string)\n\n\t\t\t\tif err != nil {\n\t\t\t\t\tapiErr := ErrAuthInvalidCredentials\n\t\t\t\t\tapiErr.Detail = \"User identification required for rate limiting\"\n\t\t\t\t\tapiErr.DebugInfo = err.Error()\n\t\t\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t\tmessageCount, err := q.GetChatMessagesCount(r.Context(), int32(userIDInt))\n\t\t\t\tif err != nil {\n\t\t\t\t\tapiErr := ErrInternalUnexpected\n\t\t\t\t\tapiErr.Detail = \"Could not get message count for rate limiting\"\n\t\t\t\t\tapiErr.DebugInfo = err.Error()\n\t\t\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t\tmaxRate, err := q.GetRateLimit(r.Context(), int32(userIDInt))\n\t\t\t\tif err != nil {\n\t\t\t\t\tmaxRate = int32(appConfig.OPENAI.RATELIMIT)\n\t\t\t\t}\n\n\t\t\t\tif messageCount >= int64(maxRate) {\n\t\t\t\t\tapiErr := ErrTooManyRequests\n\t\t\t\t\tapiErr.Detail = fmt.Sprintf(\"Rate limit exceeded: messageCount=%d, maxRate=%d\", messageCount, maxRate)\n\t\t\t\t\tRespondWithAPIError(w, apiErr)\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t}\n\t\t\t// Call the next handler.\n\t\t\tnext.ServeHTTP(w, r)\n\t\t})\n\t}\n}\n" }, { "path": "api/middleware_validation.go", "content": "package main\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"io\"\n\t\"net/http\"\n\t\"regexp\"\n\t\"strings\"\n\t\"time\"\n\t\"unicode/utf8\"\n\n\tlog \"github.com/sirupsen/logrus\"\n)\n\n// Common validation patterns\nvar (\n\tvalidationEmailRegex = regexp.MustCompile(`^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\\.[a-zA-Z]{2,}$`)\n\tvalidationUuidRegex = regexp.MustCompile(`^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$`)\n)\n\n// ValidationConfig defines validation rules for API endpoints\ntype ValidationConfig struct {\n\tMaxBodySize int64 // Maximum request body size\n\tRequiredFields []string // Required JSON fields\n\tFieldValidators map[string]FieldValidator // Custom field validators\n\tAllowedMethods []string // Allowed HTTP methods\n\tSkipBodyBuffer bool // Skip body buffering for large requests (disables field validation)\n}\n\n// FieldValidator defines a validation function for a specific field\ntype FieldValidator func(value interface{}) error\n\n// Common field validators\nfunc ValidateEmail(value interface{}) error {\n\temail, ok := value.(string)\n\tif !ok {\n\t\treturn ErrValidationInvalidInput(\"email must be a string\")\n\t}\n\tif !validationEmailRegex.MatchString(email) {\n\t\treturn ErrValidationInvalidInput(\"invalid email format\")\n\t}\n\tif len(email) > 254 {\n\t\treturn ErrValidationInvalidInput(\"email too long\")\n\t}\n\treturn nil\n}\n\nfunc ValidateUUID(value interface{}) error {\n\tuuid, ok := value.(string)\n\tif !ok {\n\t\treturn ErrValidationInvalidInput(\"UUID must be a string\")\n\t}\n\tif !validationUuidRegex.MatchString(uuid) {\n\t\treturn ErrValidationInvalidInput(\"invalid UUID format\")\n\t}\n\treturn nil\n}\n\nfunc ValidateStringLength(min, max int) FieldValidator {\n\treturn func(value interface{}) error {\n\t\tstr, ok := value.(string)\n\t\tif !ok {\n\t\t\treturn ErrValidationInvalidInput(\"value must be a string\")\n\t\t}\n\t\tif !utf8.ValidString(str) {\n\t\t\treturn ErrValidationInvalidInput(\"invalid UTF-8 string\")\n\t\t}\n\t\tif len(str) < min {\n\t\t\treturn ErrValidationInvalidInput(\"string too short\")\n\t\t}\n\t\tif len(str) > max {\n\t\t\treturn ErrValidationInvalidInput(\"string too long\")\n\t\t}\n\t\treturn nil\n\t}\n}\n\nfunc ValidateNonEmpty(value interface{}) error {\n\tstr, ok := value.(string)\n\tif !ok {\n\t\treturn ErrValidationInvalidInput(\"value must be a string\")\n\t}\n\tif strings.TrimSpace(str) == \"\" {\n\t\treturn ErrValidationInvalidInput(\"value cannot be empty\")\n\t}\n\treturn nil\n}\n\n// ValidationMiddleware creates a validation middleware with the given config\nfunc ValidationMiddleware(config ValidationConfig) func(http.Handler) http.Handler {\n\treturn func(next http.Handler) http.Handler {\n\t\treturn http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t// Start timing for performance monitoring\n\t\t\tstart := time.Now()\n\n\t\t\t// Validate HTTP method\n\t\t\tif len(config.AllowedMethods) > 0 {\n\t\t\t\tmethodAllowed := false\n\t\t\t\tfor _, method := range config.AllowedMethods {\n\t\t\t\t\tif r.Method == method {\n\t\t\t\t\t\tmethodAllowed = true\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tif !methodAllowed {\n\t\t\t\t\tlog.WithFields(log.Fields{\n\t\t\t\t\t\t\"method\": r.Method,\n\t\t\t\t\t\t\"path\": r.URL.Path,\n\t\t\t\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\t\t}).Warn(\"Method not allowed\")\n\t\t\t\t\tRespondWithAPIError(w, APIError{\n\t\t\t\t\t\tHTTPCode: http.StatusMethodNotAllowed,\n\t\t\t\t\t\tCode: ErrValidation + \"_100\",\n\t\t\t\t\t\tMessage: \"Method not allowed\",\n\t\t\t\t\t})\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Skip validation for GET requests without body\n\t\t\tif r.Method == \"GET\" || r.Method == \"DELETE\" {\n\t\t\t\tnext.ServeHTTP(w, r)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Validate content type for requests with body\n\t\t\tcontentType := r.Header.Get(\"Content-Type\")\n\t\t\tif !strings.Contains(contentType, \"application/json\") && r.ContentLength > 0 {\n\t\t\t\tlog.WithFields(log.Fields{\n\t\t\t\t\t\"content_type\": contentType,\n\t\t\t\t\t\"path\": r.URL.Path,\n\t\t\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\t}).Warn(\"Invalid content type\")\n\t\t\t\tRespondWithAPIError(w, APIError{\n\t\t\t\t\tHTTPCode: http.StatusUnsupportedMediaType,\n\t\t\t\t\tCode: ErrValidation + \"_101\",\n\t\t\t\t\tMessage: \"Content-Type must be application/json\",\n\t\t\t\t})\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Check content length\n\t\t\tif config.MaxBodySize > 0 && r.ContentLength > config.MaxBodySize {\n\t\t\t\tlog.WithFields(log.Fields{\n\t\t\t\t\t\"content_length\": r.ContentLength,\n\t\t\t\t\t\"max_size\": config.MaxBodySize,\n\t\t\t\t\t\"path\": r.URL.Path,\n\t\t\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\t}).Warn(\"Request body too large\")\n\t\t\t\tRespondWithAPIError(w, APIError{\n\t\t\t\t\tHTTPCode: http.StatusRequestEntityTooLarge,\n\t\t\t\t\tCode: ErrValidation + \"_102\",\n\t\t\t\t\tMessage: \"Request body too large\",\n\t\t\t\t})\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// For requests without body, skip JSON validation\n\t\t\tif r.ContentLength == 0 {\n\t\t\t\tnext.ServeHTTP(w, r)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Skip body buffering for large requests or when explicitly configured\n\t\t\tif config.SkipBodyBuffer {\n\t\t\t\t// Just validate content length and skip field validation\n\t\t\t\tif config.MaxBodySize > 0 && r.ContentLength > config.MaxBodySize {\n\t\t\t\t\tlog.WithFields(log.Fields{\n\t\t\t\t\t\t\"content_length\": r.ContentLength,\n\t\t\t\t\t\t\"max_size\": config.MaxBodySize,\n\t\t\t\t\t\t\"path\": r.URL.Path,\n\t\t\t\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\t\t}).Warn(\"Request body exceeds size limit\")\n\t\t\t\t\tRespondWithAPIError(w, APIError{\n\t\t\t\t\t\tHTTPCode: http.StatusRequestEntityTooLarge,\n\t\t\t\t\t\tCode: ErrValidation + \"_102\",\n\t\t\t\t\t\tMessage: \"Request body too large\",\n\t\t\t\t\t})\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t\tnext.ServeHTTP(w, r)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Create a limited reader to prevent memory exhaustion\n\t\t\tvar limitedReader io.Reader = r.Body\n\t\t\tif config.MaxBodySize > 0 {\n\t\t\t\tlimitedReader = io.LimitReader(r.Body, config.MaxBodySize+1)\n\t\t\t}\n\n\t\t\t// Read with streaming approach using a buffer\n\t\t\tvar bodyBuffer bytes.Buffer\n\t\t\twritten, err := io.CopyN(&bodyBuffer, limitedReader, config.MaxBodySize+1)\n\t\t\tr.Body.Close()\n\n\t\t\tif err != nil && err != io.EOF {\n\t\t\t\tlog.WithError(err).WithFields(log.Fields{\n\t\t\t\t\t\"path\": r.URL.Path,\n\t\t\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\t}).Error(\"Failed to read request body\")\n\t\t\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to read request body\"))\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Check if body exceeds limit\n\t\t\tif config.MaxBodySize > 0 && written > config.MaxBodySize {\n\t\t\t\tlog.WithFields(log.Fields{\n\t\t\t\t\t\"body_size\": written,\n\t\t\t\t\t\"max_size\": config.MaxBodySize,\n\t\t\t\t\t\"path\": r.URL.Path,\n\t\t\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\t}).Warn(\"Request body exceeds size limit\")\n\t\t\t\tRespondWithAPIError(w, APIError{\n\t\t\t\t\tHTTPCode: http.StatusRequestEntityTooLarge,\n\t\t\t\t\tCode: ErrValidation + \"_102\",\n\t\t\t\t\tMessage: \"Request body too large\",\n\t\t\t\t})\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tbody := bodyBuffer.Bytes()\n\n\t\t\t// Parse JSON body\n\t\t\tvar jsonData map[string]interface{}\n\t\t\tif len(body) > 0 {\n\t\t\t\tif err := json.Unmarshal(body, &jsonData); err != nil {\n\t\t\t\t\tlog.WithError(err).WithFields(log.Fields{\n\t\t\t\t\t\t\"path\": r.URL.Path,\n\t\t\t\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\t\t}).Warn(\"Invalid JSON in request body\")\n\t\t\t\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Invalid JSON format\").WithDebugInfo(err.Error()))\n\t\t\t\t\treturn\n\t\t\t\t}\n\n\t\t\t\t// Validate required fields\n\t\t\t\tfor _, field := range config.RequiredFields {\n\t\t\t\t\tif _, exists := jsonData[field]; !exists {\n\t\t\t\t\t\tlog.WithFields(log.Fields{\n\t\t\t\t\t\t\t\"missing_field\": field,\n\t\t\t\t\t\t\t\"path\": r.URL.Path,\n\t\t\t\t\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\t\t\t}).Warn(\"Missing required field\")\n\t\t\t\t\t\tRespondWithAPIError(w, ErrValidationInvalidInput(\"Missing required field: \"+field))\n\t\t\t\t\t\treturn\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Run field validators\n\t\t\t\tfor fieldName, validator := range config.FieldValidators {\n\t\t\t\t\tif value, exists := jsonData[fieldName]; exists {\n\t\t\t\t\t\tif err := validator(value); err != nil {\n\t\t\t\t\t\t\tlog.WithError(err).WithFields(log.Fields{\n\t\t\t\t\t\t\t\t\"field\": fieldName,\n\t\t\t\t\t\t\t\t\"path\": r.URL.Path,\n\t\t\t\t\t\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t\t\t\t\t}).Warn(\"Field validation failed\")\n\t\t\t\t\t\t\tRespondWithAPIError(w, WrapError(err, \"Validation failed for field: \"+fieldName))\n\t\t\t\t\t\t\treturn\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Restore body for next handler\n\t\t\tr.Body = io.NopCloser(bytes.NewReader(body))\n\n\t\t\t// Add validation context\n\t\t\tctx := context.WithValue(r.Context(), \"validation_duration\", time.Since(start))\n\t\t\tr = r.WithContext(ctx)\n\n\t\t\tlog.WithFields(log.Fields{\n\t\t\t\t\"path\": r.URL.Path,\n\t\t\t\t\"method\": r.Method,\n\t\t\t\t\"duration\": time.Since(start),\n\t\t\t\t\"ip\": r.RemoteAddr,\n\t\t\t}).Debug(\"Request validation completed\")\n\n\t\t\tnext.ServeHTTP(w, r)\n\t\t})\n\t}\n}\n\n// Predefined validation configs for common endpoints\nvar (\n\tAuthValidationConfig = ValidationConfig{\n\t\tMaxBodySize: 1024 * 10, // 10KB\n\t\tRequiredFields: []string{\"email\", \"password\"},\n\t\tFieldValidators: map[string]FieldValidator{\n\t\t\t\"email\": ValidateEmail,\n\t\t\t\"password\": ValidateStringLength(8, 128),\n\t\t},\n\t\tAllowedMethods: []string{\"POST\"},\n\t}\n\n\tChatValidationConfig = ValidationConfig{\n\t\tMaxBodySize: 1024 * 100, // 100KB\n\t\tRequiredFields: []string{\"prompt\"},\n\t\tFieldValidators: map[string]FieldValidator{\n\t\t\t\"prompt\": ValidateStringLength(1, 10000),\n\t\t\t\"session_uuid\": ValidateUUID,\n\t\t\t\"chat_uuid\": ValidateUUID,\n\t\t},\n\t\tAllowedMethods: []string{\"POST\"},\n\t}\n\n\tFileUploadValidationConfig = ValidationConfig{\n\t\tMaxBodySize: 32 * 1024 * 1024, // 32MB\n\t\tAllowedMethods: []string{\"POST\", \"GET\", \"DELETE\"},\n\t\tSkipBodyBuffer: true, // Skip buffering for large file uploads\n\t}\n\n\tGeneralValidationConfig = ValidationConfig{\n\t\tMaxBodySize: 1024 * 50, // 50KB\n\t\tAllowedMethods: []string{\"GET\", \"POST\", \"PUT\", \"DELETE\"},\n\t}\n)\n" }, { "path": "api/model_claude3_service.go", "content": "package main\n\nimport (\n\t\"bufio\"\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"log\"\n\t\"net/http\"\n\t\"os\"\n\t\"time\"\n\n\topenai \"github.com/sashabaranov/go-openai\"\n\tclaude \"github.com/swuecho/chat_backend/llm/claude\"\n\t\"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// ClaudeResponse represents the response structure from Claude API\ntype ClaudeResponse struct {\n\tCompletion string `json:\"completion\"`\n\tStop string `json:\"stop\"`\n\tStopReason string `json:\"stop_reason\"`\n\tTruncated bool `json:\"truncated\"`\n\tLogID string `json:\"log_id\"`\n\tModel string `json:\"model\"`\n\tException any `json:\"exception\"`\n}\n\n// Claude3 ChatModel implementation\ntype Claude3ChatModel struct {\n\th *ChatHandler\n}\n\nfunc (m *Claude3ChatModel) Stream(w http.ResponseWriter, chatSession sqlc_queries.ChatSession, chat_compeletion_messages []models.Message, chatUuid string, regenerate bool, stream bool) (*models.LLMAnswer, error) {\n\t// Get chat model configuration\n\tchatModel, err := GetChatModel(m.h.service.q, chatSession.Model)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Get chat files if any\n\tchatFiles, err := GetChatFiles(m.h.chatfileService.q, chatSession.Uuid)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// create a new strings.Builder\n\t// iterate through the messages and format them\n\t// print the user's question\n\t// convert assistant's response to json format\n\t// \"messages\": [\n\t//\t{\"role\": \"user\", \"content\": \"Hello, world\"}\n\t//\t]\n\t// first message is user instead of system\n\tvar messages []openai.ChatCompletionMessage\n\tif len(chat_compeletion_messages) > 1 {\n\t\t// first message used as system message\n\t\t// messages start with second message\n\t\t// drop the first assistant message if it is an assistant message\n\t\tclaude_messages := chat_compeletion_messages[1:]\n\n\t\tif len(claude_messages) > 0 && claude_messages[0].Role == \"assistant\" {\n\t\t\tclaude_messages = claude_messages[1:]\n\t\t}\n\t\tmessages = messagesToOpenAIMesages(claude_messages, chatFiles)\n\t} else {\n\t\t// only system message, return and do nothing\n\t\treturn nil, ErrSystemMessageError\n\t}\n\t// Prepare request payload\n\tjsonData := map[string]any{\n\t\t\"system\": chat_compeletion_messages[0].Content,\n\t\t\"model\": chatSession.Model,\n\t\t\"messages\": messages,\n\t\t\"max_tokens\": chatSession.MaxTokens,\n\t\t\"temperature\": chatSession.Temperature,\n\t\t\"top_p\": chatSession.TopP,\n\t\t\"stream\": stream,\n\t}\n\n\tjsonValue, err := json.Marshal(jsonData)\n\tif err != nil {\n\t\treturn nil, ErrValidationInvalidInputGeneric.WithDetail(\"failed to marshal request payload\").WithDebugInfo(err.Error())\n\t}\n\n\t// Get request context for cancellation support\n\tctx := m.h.GetRequestContext()\n\n\t// Create HTTP request with context\n\treq, err := http.NewRequestWithContext(ctx, \"POST\", chatModel.Url, bytes.NewBuffer(jsonValue))\n\tif err != nil {\n\t\treturn nil, ErrClaudeRequestFailed.WithDetail(\"failed to create HTTP request\").WithDebugInfo(err.Error())\n\t}\n\n\t// add headers to the request\n\tapiKey := os.Getenv(chatModel.ApiAuthKey)\n\n\tif apiKey == \"\" {\n\t\treturn nil, ErrAuthInvalidCredentials.WithDetail(fmt.Sprintf(\"missing API key for model %s\", chatSession.Model))\n\t}\n\n\tauthHeaderName := chatModel.ApiAuthHeader\n\tif authHeaderName != \"\" {\n\t\treq.Header.Set(authHeaderName, apiKey)\n\t}\n\n\treq.Header.Set(\"Content-Type\", \"application/json\")\n\treq.Header.Set(\"anthropic-version\", \"2023-06-01\")\n\n\tif !stream {\n\t\treq.Header.Set(\"Accept\", \"application/json\")\n\t\tclient := http.Client{\n\t\t\tTimeout: 5 * time.Minute,\n\t\t}\n\n\t\tllmAnswer, err := doGenerateClaude3(ctx, client, req)\n\t\tif err != nil {\n\t\t\treturn nil, ErrClaudeRequestFailed.WithDetail(\"failed to generate response\").WithDebugInfo(err.Error())\n\t\t}\n\n\t\tanswerResponse := constructChatCompletionStreamResponse(llmAnswer.AnswerId, llmAnswer.Answer)\n\t\tdata, err := json.Marshal(answerResponse)\n\t\tif err != nil {\n\t\t\treturn nil, ErrInternalUnexpected.WithDetail(\"failed to marshal response\").WithDebugInfo(err.Error())\n\t\t}\n\n\t\tif _, err := fmt.Fprint(w, string(data)); err != nil {\n\t\t\treturn nil, ErrClaudeResponseFaild.WithDetail(\"failed to write response\").WithDebugInfo(err.Error())\n\t\t}\n\n\t\treturn llmAnswer, nil\n\t}\n\n\t// Handle streaming response\n\treq.Header.Set(\"Accept\", \"text/event-stream\")\n\treq.Header.Set(\"Cache-Control\", \"no-cache\")\n\treq.Header.Set(\"Connection\", \"keep-alive\")\n\n\tllmAnswer, err := m.h.chatStreamClaude3(ctx, w, req, chatUuid, regenerate)\n\tif err != nil {\n\t\treturn nil, ErrClaudeStreamFailed.WithDetail(\"failed to stream response\").WithDebugInfo(err.Error())\n\t}\n\treturn llmAnswer, nil\n}\n\nfunc doGenerateClaude3(ctx context.Context, client http.Client, req *http.Request) (*models.LLMAnswer, error) {\n\tresp, err := client.Do(req)\n\tif err != nil {\n\t\treturn nil, ErrClaudeRequestFailed.WithMessage(\"Failed to process Claude request\").WithDebugInfo(err.Error())\n\t}\n\t// Unmarshal directly from resp.Body\n\tvar message claude.Response\n\tif err := json.NewDecoder(resp.Body).Decode(&message); err != nil {\n\t\treturn nil, ErrClaudeInvalidResponse.WithMessage(\"Failed to unmarshal Claude response\").WithDebugInfo(err.Error())\n\t}\n\tdefer resp.Body.Close()\n\tuuid := message.ID\n\tfirstMessage := message.Content[0].Text\n\n\treturn &models.LLMAnswer{\n\t\tAnswerId: uuid,\n\t\tAnswer: firstMessage,\n\t}, nil\n}\n\n// claude-3-opus-20240229\n// claude-3-sonnet-20240229\n// claude-3-haiku-20240307\nfunc (h *ChatHandler) chatStreamClaude3(ctx context.Context, w http.ResponseWriter, req *http.Request, chatUuid string, regenerate bool) (*models.LLMAnswer, error) {\n\n\t// create the http client and send the request\n\tclient := &http.Client{\n\t\tTimeout: 5 * time.Minute,\n\t}\n\tresp, err := client.Do(req)\n\tif err != nil {\n\t\treturn nil, ErrClaudeRequestFailed.WithMessage(\"Failed to process Claude streaming request\").WithDebugInfo(err.Error())\n\t}\n\n\t// Use smaller buffer for more responsive streaming\n\tioreader := bufio.NewReaderSize(resp.Body, 1024)\n\n\t// read the response body\n\tdefer resp.Body.Close()\n\t// loop over the response body and print data\n\n\tflusher, err := setupSSEStream(w)\n\tif err != nil {\n\t\treturn nil, APIError{\n\t\t\tHTTPCode: http.StatusInternalServerError,\n\t\t\tCode: \"STREAM_UNSUPPORTED\",\n\t\t\tMessage: \"Streaming unsupported by client\",\n\t\t}\n\t}\n\n\t// Flush immediately to establish connection\n\tflusher.Flush()\n\n\tvar answer string\n\tanswer_id := GenerateAnswerID(chatUuid, regenerate)\n\n\tvar headerData = []byte(\"data: \")\n\tcount := 0\n\tfor {\n\t\t// Check if client disconnected or context was cancelled\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\tlog.Printf(\"Claude stream cancelled by client: %v\", ctx.Err())\n\t\t\t// Return current accumulated content when cancelled\n\t\t\treturn &models.LLMAnswer{Answer: answer, AnswerId: answer_id}, nil\n\t\tdefault:\n\t\t}\n\n\t\tcount++\n\t\t// prevent infinite loop\n\t\tif count > 10000 {\n\t\t\tbreak\n\t\t}\n\t\tline, err := ioreader.ReadBytes('\\n')\n\t\tlog.Printf(\"%+v\", string(line))\n\t\tif err != nil {\n\t\t\tif errors.Is(err, io.EOF) {\n\t\t\t\tif bytes.HasPrefix(line, []byte(\"{\\\"type\\\":\\\"error\\\"\")) {\n\t\t\t\t\tlog.Println(string(line))\n\t\t\t\t\terr := FlushResponse(w, flusher, StreamingResponse{\n\t\t\t\t\t\tAnswerID: NewUUID(),\n\t\t\t\t\t\tContent: string(line),\n\t\t\t\t\t\tIsFinal: true,\n\t\t\t\t\t})\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\tlog.Printf(\"Failed to flush error response: %v\", err)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfmt.Println(\"End of stream reached\")\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\treturn nil, err\n\t\t}\n\t\tline = bytes.TrimPrefix(line, headerData)\n\n\t\tif bytes.HasPrefix(line, []byte(\"event: message_stop\")) {\n\t\t\t// stream.isFinished = true\n\t\t\t// No need to send full content at the end since we're sending deltas\n\t\t\tbreak\n\t\t}\n\t\tif bytes.HasPrefix(line, []byte(\"{\\\"type\\\":\\\"error\\\"\")) {\n\t\t\tlog.Println(string(line))\n\t\t\treturn nil, ErrClaudeStreamFailed.WithMessage(\"Error in Claude API response\").WithDebugInfo(string(line))\n\t\t}\n\t\tif answer_id == \"\" {\n\t\t\tanswer_id = NewUUID()\n\t\t}\n\t\tif bytes.HasPrefix(line, []byte(\"{\\\"type\\\":\\\"content_block_start\\\"\")) {\n\t\t\tanswer = claude.AnswerFromBlockStart(line)\n\t\t\terr := FlushResponse(w, flusher, StreamingResponse{\n\t\t\t\tAnswerID: answer_id,\n\t\t\t\tContent: answer,\n\t\t\t\tIsFinal: false,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\tlog.Printf(\"Failed to flush content block start: %v\", err)\n\t\t\t}\n\t\t}\n\t\tif bytes.HasPrefix(line, []byte(\"{\\\"type\\\":\\\"content_block_delta\\\"\")) {\n\t\t\tdelta := claude.AnswerFromBlockDelta(line)\n\t\t\tanswer += delta // Still accumulate for final answer storage\n\t\t\t// Send only the delta content\n\t\t\terr := FlushResponse(w, flusher, StreamingResponse{\n\t\t\t\tAnswerID: answer_id,\n\t\t\t\tContent: delta,\n\t\t\t\tIsFinal: false,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\tlog.Printf(\"Failed to flush content block delta: %v\", err)\n\t\t\t}\n\t\t}\n\t\t// Flush after every iteration to ensure immediate delivery\n\t\t// This prevents data from being held in buffers\n\t\tif count%3 == 0 {\n\t\t\tflusher.Flush()\n\t\t}\n\t}\n\treturn &models.LLMAnswer{\n\t\tAnswer: answer,\n\t\tAnswerId: answer_id,\n\t}, nil\n}\n" }, { "path": "api/model_completion_service.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"io\"\n\t\"log\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/rotisserie/eris\"\n\topenai \"github.com/sashabaranov/go-openai\"\n\n\t\"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// CompletionChatModel implements ChatModel interface for OpenAI completion models\ntype CompletionChatModel struct {\n\th *ChatHandler\n}\n\n// Stream implements the ChatModel interface for completion model scenarios\nfunc (m *CompletionChatModel) Stream(w http.ResponseWriter, chatSession sqlc_queries.ChatSession, chat_completion_messages []models.Message, chatUuid string, regenerate bool, stream bool) (*models.LLMAnswer, error) {\n\t// Get request context for cancellation support\n\tctx := m.h.GetRequestContext()\n\treturn m.completionStream(ctx, w, chatSession, chat_completion_messages, chatUuid, regenerate, stream)\n}\n\n// completionStream handles streaming for OpenAI completion models\nfunc (m *CompletionChatModel) completionStream(ctx context.Context, w http.ResponseWriter, chatSession sqlc_queries.ChatSession, chat_completion_messages []models.Message, chatUuid string, regenerate bool, _ bool) (*models.LLMAnswer, error) {\n\t// Check per chat_model rate limit\n\topenAIRateLimiter.Wait(context.Background())\n\n\texceedPerModeRateLimitOrError := m.h.CheckModelAccess(w, chatSession.Uuid, chatSession.Model, chatSession.UserID)\n\tif exceedPerModeRateLimitOrError {\n\t\treturn nil, eris.New(\"exceed per mode rate limit\")\n\t}\n\n\t// Get chat model configuration\n\tchatModel, err := GetChatModel(m.h.service.q, chatSession.Model)\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(ErrResourceNotFound(\"\"), \"chat model \"+chatSession.Model, \"\"))\n\t\treturn nil, err\n\t}\n\n\t// Generate OpenAI client configuration\n\tconfig, err := genOpenAIConfig(*chatModel)\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(ErrInternalUnexpected, \"Failed to generate OpenAI configuration\", err.Error()))\n\t\treturn nil, err\n\t}\n\n\tclient := openai.NewClientWithConfig(config)\n\n\t// Get the latest message content as prompt\n\tprompt := chat_completion_messages[len(chat_completion_messages)-1].Content\n\n\t// Create completion request\n\tN := chatSession.N\n\treq := openai.CompletionRequest{\n\t\tModel: chatSession.Model,\n\t\tTemperature: float32(chatSession.Temperature),\n\t\tTopP: float32(chatSession.TopP),\n\t\tN: int(N),\n\t\tPrompt: prompt,\n\t\tStream: true,\n\t}\n\n\t// Create completion stream with timeout\n\tctx, cancel := context.WithTimeout(context.Background(), DefaultRequestTimeout)\n\tdefer cancel()\n\n\tstream, err := client.CreateCompletionStream(ctx, req)\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(ErrInternalUnexpected, \"Failed to create completion stream\", err.Error()))\n\t\treturn nil, err\n\t}\n\tdefer stream.Close()\n\n\t// Setup SSE streaming\n\tflusher, err := setupSSEStream(w)\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(ErrInternalUnexpected, \"Streaming unsupported by client\", err.Error()))\n\t\treturn nil, err\n\t}\n\n\tvar answer string\n\tanswer_id := GenerateAnswerID(chatUuid, regenerate)\n\ttextBuffer := newTextBuffer(N, \"```\\n\"+prompt, \"\\n```\\n\")\n\n\t// Process streaming response\n\tfor {\n\t\t// Check if client disconnected or context was cancelled\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\tlog.Printf(\"Completion stream cancelled by client: %v\", ctx.Err())\n\t\t\t// Return current accumulated content when cancelled\n\t\t\treturn &models.LLMAnswer{Answer: answer, AnswerId: answer_id}, nil\n\t\tdefault:\n\t\t}\n\n\t\tresponse, err := stream.Recv()\n\t\tif errors.Is(err, io.EOF) {\n\t\t\t// Send the final message\n\t\t\tif len(answer) > 0 {\n\t\t\t\terr := FlushResponse(w, flusher, StreamingResponse{\n\t\t\t\t\tAnswerID: answer_id,\n\t\t\t\t\tContent: answer,\n\t\t\t\t\tIsFinal: true,\n\t\t\t\t})\n\t\t\t\tif err != nil {\n\t\t\t\t\tlog.Printf(\"Failed to flush final response: %v\", err)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Include debug information if enabled\n\t\t\tif chatSession.Debug {\n\t\t\t\treq_j, _ := json.Marshal(req)\n\t\t\t\tlog.Println(string(req_j))\n\t\t\t\tanswer = answer + \"\\n\" + string(req_j)\n\t\t\t\terr := FlushResponse(w, flusher, StreamingResponse{\n\t\t\t\t\tAnswerID: answer_id,\n\t\t\t\t\tContent: answer,\n\t\t\t\t\tIsFinal: true,\n\t\t\t\t})\n\t\t\t\tif err != nil {\n\t\t\t\t\tlog.Printf(\"Failed to flush debug response: %v\", err)\n\t\t\t\t}\n\t\t\t}\n\t\t\tbreak\n\t\t}\n\n\t\tif err != nil {\n\t\t\tRespondWithAPIError(w, ErrChatStreamFailed.WithMessage(\"Stream error occurred\").WithDebugInfo(err.Error()))\n\t\t\treturn nil, err\n\t\t}\n\n\t\t// Process response chunk\n\t\ttextIdx := response.Choices[0].Index\n\t\tdelta := response.Choices[0].Text\n\t\ttextBuffer.appendByIndex(textIdx, delta)\n\n\t\tif chatSession.Debug {\n\t\t\tlog.Printf(\"%d: %s\", textIdx, delta)\n\t\t}\n\n\t\tif answer_id == \"\" {\n\t\t\tanswer_id = response.ID\n\t\t}\n\n\t\t// Concatenate all string builders into a single string\n\t\tanswer = textBuffer.String(\"\\n\\n\")\n\n\t\t// Determine when to flush the response\n\t\tperWordStreamLimit := getPerWordStreamLimit()\n\t\tif strings.HasSuffix(delta, \"\\n\") || len(answer) < perWordStreamLimit {\n\t\t\tif len(answer) == 0 {\n\t\t\t\tlog.Print(ErrorNoContent)\n\t\t\t} else {\n\t\t\t\terr := FlushResponse(w, flusher, StreamingResponse{\n\t\t\t\t\tAnswerID: answer_id,\n\t\t\t\t\tContent: answer,\n\t\t\t\t\tIsFinal: false,\n\t\t\t\t})\n\t\t\t\tif err != nil {\n\t\t\t\t\tlog.Printf(\"Failed to flush response: %v\", err)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\treturn &models.LLMAnswer{AnswerId: answer_id, Answer: answer}, nil\n}\n" }, { "path": "api/model_custom_service.go", "content": "package main\n\nimport (\n\t\"bufio\"\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"log\"\n\t\"net/http\"\n\t\"os\"\n\t\"strings\"\n\n\tclaude \"github.com/swuecho/chat_backend/llm/claude\"\n\t\"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// CustomModelResponse represents the response structure for custom models\ntype CustomModelResponse struct {\n\tCompletion string `json:\"completion\"`\n\tStop string `json:\"stop\"`\n\tStopReason string `json:\"stop_reason\"`\n\tTruncated bool `json:\"truncated\"`\n\tLogID string `json:\"log_id\"`\n\tModel string `json:\"model\"`\n\tException any `json:\"exception\"`\n}\n\n// CustomChatModel implements ChatModel interface for custom model providers\ntype CustomChatModel struct {\n\th *ChatHandler\n}\n\n// Stream implements the ChatModel interface for custom model scenarios\nfunc (m *CustomChatModel) Stream(w http.ResponseWriter, chatSession sqlc_queries.ChatSession, chat_completion_messages []models.Message, chatUuid string, regenerate bool, stream bool) (*models.LLMAnswer, error) {\n\t// Get request context for cancellation support\n\tctx := m.h.GetRequestContext()\n\treturn m.customChatStream(ctx, w, chatSession, chat_completion_messages, chatUuid, regenerate)\n}\n\n// customChatStream handles streaming for custom model providers\nfunc (m *CustomChatModel) customChatStream(ctx context.Context, w http.ResponseWriter, chatSession sqlc_queries.ChatSession, chat_completion_messages []models.Message, chatUuid string, regenerate bool) (*models.LLMAnswer, error) {\n\t// Get chat model configuration\n\tchat_model, err := GetChatModel(m.h.service.q, chatSession.Model)\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(ErrResourceNotFound(\"\"), \"chat model: \"+chatSession.Model, \"\"))\n\t\treturn nil, err\n\t}\n\n\t// Get API key from environment\n\tapiKey := os.Getenv(chat_model.ApiAuthKey)\n\turl := chat_model.Url\n\n\t// Format messages for the custom model\n\tprompt := claude.FormatClaudePrompt(chat_completion_messages)\n\n\t// Create request payload\n\tjsonData := map[string]any{\n\t\t\"prompt\": prompt,\n\t\t\"model\": chatSession.Model,\n\t\t\"max_tokens_to_sample\": chatSession.MaxTokens,\n\t\t\"temperature\": chatSession.Temperature,\n\t\t\"stop_sequences\": []string{\"\\n\\nHuman:\"},\n\t\t\"stream\": true,\n\t}\n\n\t// Marshal request data\n\tjsonValue, _ := json.Marshal(jsonData)\n\n\t// Create HTTP request with context\n\treq, err := http.NewRequestWithContext(ctx, \"POST\", url, bytes.NewBuffer(jsonValue))\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(ErrChatRequestFailed, \"Failed to create custom model request\", err.Error()))\n\t\treturn nil, err\n\t}\n\n\t// Set authentication header if configured\n\tauthHeaderName := chat_model.ApiAuthHeader\n\tif authHeaderName != \"\" {\n\t\treq.Header.Set(authHeaderName, apiKey)\n\t}\n\n\t// Set request headers\n\tSetStreamingHeaders(req)\n\n\t// Send HTTP request\n\tclient := &http.Client{}\n\tresp, err := client.Do(req)\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(ErrChatRequestFailed, \"Failed to send custom model request\", err.Error()))\n\t\treturn nil, err\n\t}\n\tdefer resp.Body.Close()\n\n\t// Setup streaming response\n\tioreader := bufio.NewReader(resp.Body)\n\tflusher, err := setupSSEStream(w)\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(APIError{\n\t\t\tHTTPCode: http.StatusInternalServerError,\n\t\t\tCode: \"STREAM_UNSUPPORTED\",\n\t\t\tMessage: \"Streaming unsupported by client\",\n\t\t}, \"\", err.Error()))\n\t\treturn nil, err\n\t}\n\n\tvar answer string\n\tvar answer_id string\n\tvar lastFlushLength int\n\tanswer_id = GenerateAnswerID(chatUuid, regenerate)\n\n\theaderData := []byte(\"data: \")\n\tcount := 0\n\n\t// Process streaming response\n\tfor {\n\t\t// Check if client disconnected or context was cancelled\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\tlog.Printf(\"Custom model stream cancelled by client: %v\", ctx.Err())\n\t\t\t// Return current accumulated content when cancelled\n\t\t\treturn &models.LLMAnswer{Answer: answer, AnswerId: answer_id}, nil\n\t\tdefault:\n\t\t}\n\n\t\tcount++\n\t\t// Prevent infinite loop\n\t\tif count > MaxStreamingLoopIterations {\n\t\t\tbreak\n\t\t}\n\n\t\tline, err := ioreader.ReadBytes('\\n')\n\t\tif err != nil {\n\t\t\tif errors.Is(err, io.EOF) {\n\t\t\t\tfmt.Println(ErrorEndOfStream)\n\t\t\t\tbreak\n\t\t\t}\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif !bytes.HasPrefix(line, headerData) {\n\t\t\tcontinue\n\t\t}\n\t\tline = bytes.TrimPrefix(line, headerData)\n\n\t\tif bytes.HasPrefix(line, []byte(\"[DONE]\")) {\n\t\t\tfmt.Println(ErrorDoneBreak)\n\n\t\t\tbreak\n\t\t}\n\n\t\tif answer_id == \"\" {\n\t\t\tanswer_id = NewUUID()\n\t\t}\n\n\t\tvar response CustomModelResponse\n\t\t_ = json.Unmarshal(line, &response)\n\t\tanswer = response.Completion\n\n\t\t// Determine when to flush the response\n\t\tshouldFlush := strings.Contains(answer, \"\\n\") ||\n\t\t\tlen(answer) < SmallAnswerThreshold ||\n\t\t\t(len(answer)-lastFlushLength) >= FlushCharacterThreshold\n\n\t\tif shouldFlush {\n\t\t\terr := FlushResponse(w, flusher, StreamingResponse{\n\t\t\t\tAnswerID: answer_id,\n\t\t\t\tContent: answer,\n\t\t\t\tIsFinal: false,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\tlog.Printf(\"Failed to flush response: %v\", err)\n\t\t\t}\n\t\t\tlastFlushLength = len(answer)\n\t\t}\n\t}\n\n\treturn &models.LLMAnswer{\n\t\tAnswer: answer,\n\t\tAnswerId: answer_id,\n\t}, nil\n}\n" }, { "path": "api/model_gemini_service.go", "content": "package main\n\nimport (\n\t\"bufio\"\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"log\"\n\t\"net/http\"\n\t\"os\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/swuecho/chat_backend/llm/gemini\"\n\t\"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// Generated by curl-to-Go: https://mholt.github.io/curl-to-go\n\n// curl https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key=$API_KEY \\\n// -H 'Content-Type: application/json' \\\n// -X POST \\\n// -d '{\n// \"contents\": [{\n// \"parts\":[{\n// \"text\": \"Write a story about a magic backpack.\"}]}]}' 2> /dev/null\n\n// GeminiClient handles communication with the Gemini API\ntype GeminiClient struct {\n\tclient *http.Client\n}\n\n// NewGeminiClient creates a new Gemini API client\nfunc NewGeminiClient() *GeminiClient {\n\treturn &GeminiClient{\n\t\tclient: &http.Client{Timeout: 5 * time.Minute},\n\t}\n}\n\n// Gemini ChatModel implementation\ntype GeminiChatModel struct {\n\th *ChatHandler\n\tclient *GeminiClient\n}\n\nfunc NewGeminiChatModel(h *ChatHandler) *GeminiChatModel {\n\treturn &GeminiChatModel{\n\t\th: h,\n\t\tclient: NewGeminiClient(),\n\t}\n}\n\nfunc (m *GeminiChatModel) Stream(w http.ResponseWriter, chatSession sqlc_queries.ChatSession, messages []models.Message, chatUuid string, regenerate bool, stream bool) (*models.LLMAnswer, error) {\n\tanswerID := GenerateAnswerID(chatUuid, regenerate)\n\n\tchatFiles, err := GetChatFiles(m.h.chatfileService.q, chatSession.Uuid)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tpayloadBytes, err := gemini.GenGemminPayload(messages, chatFiles)\n\tif err != nil {\n\t\treturn nil, ErrInternalUnexpected.WithMessage(\"Failed to generate Gemini payload\").WithDebugInfo(err.Error())\n\t}\n\n\t// Get request context for cancellation support\n\tctx := m.h.GetRequestContext()\n\n\turl := gemini.BuildAPIURL(chatSession.Model, stream)\n\treq, err := http.NewRequestWithContext(ctx, \"POST\", url, bytes.NewBuffer(payloadBytes))\n\tif err != nil {\n\t\treturn nil, ErrInternalUnexpected.WithMessage(\"Failed to create Gemini API request\").WithDebugInfo(err.Error())\n\t}\n\treq.Header.Set(\"Content-Type\", \"application/json\")\n\n\tif stream {\n\t\treturn m.handleStreamResponse(ctx, w, req, answerID)\n\t}\n\n\tllmAnswer, err := gemini.HandleRegularResponse(*m.client.client, req)\n\tif err != nil {\n\t\treturn nil, ErrInternalUnexpected.WithMessage(\"Failed to generate regular response\").WithDebugInfo(err.Error())\n\t}\n\tif llmAnswer == nil {\n\t\treturn nil, ErrInternalUnexpected.WithMessage(\"Empty response from Gemini\")\n\t}\n\n\tllmAnswer.AnswerId = answerID\n\tresponse := constructChatCompletionStreamResponse(answerID, llmAnswer.Answer)\n\tdata, _ := json.Marshal(response)\n\tfmt.Fprint(w, string(data))\n\treturn llmAnswer, err\n}\n\nfunc GenerateChatTitle(ctx context.Context, model, chatText string) (string, error) {\n\t// Validate API key\n\tapiKey := os.Getenv(\"GEMINI_API_KEY\")\n\tif apiKey == \"\" {\n\t\treturn \"\", ErrInternalUnexpected.WithMessage(\"GEMINI_API_KEY environment variable not set\")\n\t}\n\n\t// Validate input\n\tif strings.TrimSpace(chatText) == \"\" {\n\t\treturn \"\", ErrValidationInvalidInput(\"chat text cannot be empty\")\n\t}\n\n\t// Create properly formatted Gemini messages\n\tmessages := []models.Message{\n\t\t{\n\t\t\tRole: \"user\",\n\t\t\tContent: `Generate a short title (3-6 words) for this conversation. Output ONLY the title text, no quotes, no markdown, no prefixes like \"Title:\". Example: \"Python list comprehension guide\"`,\n\t\t},\n\t\t{\n\t\t\tRole: \"user\",\n\t\t\tContent: chatText,\n\t\t},\n\t}\n\n\t// Generate proper Gemini payload\n\tpayloadBytes, err := gemini.GenGemminPayload(messages, nil)\n\tif err != nil {\n\t\treturn \"\", ErrInternalUnexpected.WithMessage(\"Failed to generate Gemini payload\").WithDebugInfo(err.Error())\n\t}\n\n\t// Build URL with proper API key\n\turl := gemini.BuildAPIURL(model, false)\n\treq, err := http.NewRequest(\"POST\", url, bytes.NewBuffer(payloadBytes))\n\tif err != nil {\n\t\treturn \"\", ErrInternalUnexpected.WithMessage(\"Failed to create Gemini API request\").WithDebugInfo(err.Error())\n\t}\n\treq.Header.Set(\"Content-Type\", \"application/json\")\n\tanswer, err := gemini.HandleRegularResponse(http.Client{Timeout: 1 * time.Minute}, req)\n\tif err != nil {\n\t\treturn \"\", ErrInternalUnexpected.WithMessage(\"Failed to handle Gemini response\").WithDebugInfo(err.Error())\n\t}\n\n\t// Validate and clean up response\n\tif answer == nil || answer.Answer == \"\" {\n\t\treturn \"\", ErrInternalUnexpected.WithMessage(\"Empty response from Gemini\")\n\t}\n\n\ttitle := strings.TrimSpace(answer.Answer)\n\ttitle = strings.Trim(title, `\"`)\n\ttitle = strings.Trim(title, `*`)\n\ttitle = strings.Trim(title, `#`)\n\t// Remove common prefixes\n\ttitle = strings.TrimPrefix(title, \"Title:\")\n\ttitle = strings.TrimPrefix(title, \"title:\")\n\ttitle = strings.TrimPrefix(title, \"Title: \")\n\ttitle = strings.TrimPrefix(title, \"title: \")\n\ttitle = strings.TrimSpace(title)\n\t// Remove any remaining markdown or special characters at the start\n\tfor strings.HasPrefix(title, \"#\") || strings.HasPrefix(title, \"-\") || strings.HasPrefix(title, \"*\") {\n\t\ttitle = strings.TrimLeft(title, \"#-* \")\n\t\ttitle = strings.TrimSpace(title)\n\t}\n\tif title == \"\" {\n\t\treturn \"\", ErrInternalUnexpected.WithMessage(\"Invalid title generated\")\n\t}\n\n\t// Truncate and return\n\treturn firstN(title, 100), nil\n}\n\nfunc (m *GeminiChatModel) handleStreamResponse(ctx context.Context, w http.ResponseWriter, req *http.Request, answerID string) (*models.LLMAnswer, error) {\n\tresp, err := m.client.client.Do(req)\n\tif err != nil {\n\t\treturn nil, ErrInternalUnexpected.WithMessage(\"Failed to send Gemini API request\").WithDebugInfo(err.Error())\n\t}\n\tdefer resp.Body.Close()\n\n\tflusher, err := setupSSEStream(w)\n\tif err != nil {\n\t\treturn nil, APIError{\n\t\t\tHTTPCode: http.StatusInternalServerError,\n\t\t\tCode: \"STREAM_UNSUPPORTED\",\n\t\t\tMessage: \"Streaming unsupported by client\",\n\t\t}\n\t}\n\n\tvar answer string\n\tlog.Println(resp.StatusCode)\n\tif resp.StatusCode != http.StatusOK {\n\t\terrorBody, _ := io.ReadAll(resp.Body)\n\t\tlog.Println(string(errorBody))\n\t\tvar apiError gemini.GoogleApiError\n\t\tif json.Unmarshal(errorBody, &apiError) == nil && apiError.Error.Message != \"\" {\n\t\t\tlog.Printf(\"API returned non-200 status: %d %s. Error: %s\", resp.StatusCode, http.StatusText(resp.StatusCode), &apiError)\n\t\t} else {\n\t\t\tlog.Printf(\"API returned non-200 status: %d %s. Body: %s\", resp.StatusCode, http.StatusText(resp.StatusCode), string(errorBody))\n\t\t}\n\t\treturn nil, APIError{\n\t\t\tHTTPCode: apiError.Error.Code,\n\t\t\tCode: apiError.Error.Status,\n\t\t\tMessage: apiError.Error.Message,\n\t\t}\n\t}\n\tioreader := bufio.NewReader(resp.Body)\n\theaderData := []byte(\"data: \")\n\n\tfor count := 0; count < 10000; count++ {\n\t\t// Check if client disconnected or context was cancelled\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\tlog.Printf(\"Gemini stream cancelled by client: %v\", ctx.Err())\n\t\t\t// Return current accumulated content when cancelled\n\t\t\treturn &models.LLMAnswer{Answer: answer, AnswerId: answerID}, nil\n\t\tdefault:\n\t\t}\n\n\t\tline, err := ioreader.ReadBytes('\\n')\n\t\tif err != nil {\n\t\t\tif errors.Is(err, io.EOF) {\n\t\t\t\treturn &models.LLMAnswer{\n\t\t\t\t\tAnswer: answer,\n\t\t\t\t\tAnswerId: answerID,\n\t\t\t\t}, nil\n\t\t\t}\n\t\t\treturn nil, ErrInternalUnexpected.WithMessage(\"Error reading stream\").WithDebugInfo(err.Error())\n\t\t}\n\n\t\tif !bytes.HasPrefix(line, headerData) {\n\t\t\tcontinue\n\t\t}\n\n\t\tline = bytes.TrimPrefix(line, headerData)\n\t\tif len(line) > 0 {\n\t\t\tdelta := gemini.ParseRespLineDelta(line)\n\t\t\tanswer += delta // Accumulate delta for final answer storage\n\t\t\t// Send only the delta content\n\t\t\tif len(delta) > 0 {\n\t\t\t\terr := FlushResponse(w, flusher, StreamingResponse{\n\t\t\t\t\tAnswerID: answerID,\n\t\t\t\t\tContent: delta,\n\t\t\t\t\tIsFinal: false,\n\t\t\t\t})\n\t\t\t\tif err != nil {\n\t\t\t\t\tlog.Printf(\"Failed to flush response: %v\", err)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\treturn &models.LLMAnswer{\n\t\tAnswerId: answerID,\n\t\tAnswer: answer,\n\t}, nil\n}\n" }, { "path": "api/model_ollama_service.go", "content": "package main\n\nimport (\n\t\"bufio\"\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"log\"\n\t\"net/http\"\n\t\"os\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// OllamaResponse represents the response structure from Ollama API\ntype OllamaResponse struct {\n\tModel string `json:\"model\"`\n\tCreatedAt time.Time `json:\"created_at\"`\n\tDone bool `json:\"done\"`\n\tMessage models.Message `json:\"message\"`\n\tTotalDuration int64 `json:\"total_duration\"`\n\tLoadDuration int64 `json:\"load_duration\"`\n\tPromptEvalCount int `json:\"prompt_eval_count\"`\n\tPromptEvalDuration int64 `json:\"prompt_eval_duration\"`\n\tEvalCount int `json:\"eval_count\"`\n\tEvalDuration int64 `json:\"eval_duration\"`\n}\n\n// Ollama ChatModel implementation\ntype OllamaChatModel struct {\n\th *ChatHandler\n}\n\nfunc (m *OllamaChatModel) Stream(w http.ResponseWriter, chatSession sqlc_queries.ChatSession, chat_compeletion_messages []models.Message, chatUuid string, regenerate bool, stream bool) (*models.LLMAnswer, error) {\n\t// Get request context for cancellation support\n\tctx := m.h.GetRequestContext()\n\treturn m.h.chatOllamStream(ctx, w, chatSession, chat_compeletion_messages, chatUuid, regenerate)\n}\n\nfunc (h *ChatHandler) chatOllamStream(ctx context.Context, w http.ResponseWriter, chatSession sqlc_queries.ChatSession, chat_compeletion_messages []models.Message, chatUuid string, regenerate bool) (*models.LLMAnswer, error) {\n\t// set the api key\n\tchatModel, err := GetChatModel(h.service.q, chatSession.Model)\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(ErrResourceNotFound(\"\"), \"chat model: \"+chatSession.Model, \"\"))\n\t\treturn nil, err\n\t}\n\tjsonData := map[string]any{\n\t\t\"model\": strings.Replace(chatSession.Model, \"ollama-\", \"\", 1),\n\t\t\"messages\": chat_compeletion_messages,\n\t}\n\t// convert data to json format\n\tjsonValue, _ := json.Marshal(jsonData)\n\t// create the request with context\n\treq, err := http.NewRequestWithContext(ctx, \"POST\", chatModel.Url, bytes.NewBuffer(jsonValue))\n\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to make request\").WithDebugInfo(err.Error()))\n\t\treturn nil, err\n\t}\n\n\t// add headers to the request\n\tapiKey := os.Getenv(chatModel.ApiAuthKey)\n\tauthHeaderName := chatModel.ApiAuthHeader\n\tif authHeaderName != \"\" {\n\t\treq.Header.Set(authHeaderName, apiKey)\n\t}\n\n\treq.Header.Set(\"Content-Type\", \"application/json\")\n\n\t// set the streaming flag\n\treq.Header.Set(\"Accept\", \"text/event-stream\")\n\treq.Header.Set(\"Cache-Control\", \"no-cache\")\n\treq.Header.Set(\"Connection\", \"keep-alive\")\n\treq.Header.Set(\"Access-Control-Allow-Origin\", \"*\")\n\n\t// create the http client and send the request\n\tclient := &http.Client{\n\t\tTimeout: 5 * time.Minute,\n\t}\n\tresp, err := client.Do(req)\n\tif err != nil {\n\t\tRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"Failed to create chat completion stream\").WithDebugInfo(err.Error()))\n\t\treturn nil, err\n\t}\n\n\tioreader := bufio.NewReader(resp.Body)\n\n\t// read the response body\n\tdefer resp.Body.Close()\n\t// loop over the response body and print data\n\n\tflusher, err := setupSSEStream(w)\n\tif err != nil {\n\t\tRespondWithAPIError(w, APIError{\n\t\t\tHTTPCode: http.StatusInternalServerError,\n\t\t\tCode: \"STREAM_UNSUPPORTED\",\n\t\t\tMessage: \"Streaming unsupported by client\",\n\t\t})\n\t\treturn nil, err\n\t}\n\n\tvar answer string\n\tanswer_id := GenerateAnswerID(chatUuid, regenerate)\n\n\tcount := 0\n\tfor {\n\t\t// Check if client disconnected or context was cancelled\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\tlog.Printf(\"Ollama stream cancelled by client: %v\", ctx.Err())\n\t\t\t// Return current accumulated content when cancelled\n\t\t\treturn &models.LLMAnswer{Answer: answer, AnswerId: answer_id}, nil\n\t\tdefault:\n\t\t}\n\n\t\tcount++\n\t\t// prevent infinite loop\n\t\tif count > 10000 {\n\t\t\tbreak\n\t\t}\n\t\tline, err := ioreader.ReadBytes('\\n')\n\t\tif err != nil {\n\t\t\tif errors.Is(err, io.EOF) {\n\t\t\t\tfmt.Println(\"End of stream reached\")\n\t\t\t\tbreak // Exit loop if end of stream\n\t\t\t}\n\t\t\treturn nil, err\n\t\t}\n\t\tvar streamResp OllamaResponse\n\t\terr = json.Unmarshal(line, &streamResp)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tdelta := strings.ReplaceAll(streamResp.Message.Content, \"<0x0A>\", \"\\n\")\n\t\tanswer += delta // Still accumulate for final answer storage\n\n\t\tif streamResp.Done {\n\t\t\t// stream.isFinished = true\n\t\t\tfmt.Println(\"DONE break\")\n\t\t\t// No need to send full content at the end since we're sending deltas\n\t\t\tbreak\n\t\t}\n\t\tif answer_id == \"\" {\n\t\t\tanswer_id = NewUUID()\n\t\t}\n\n\t\t// Send delta content immediately when available\n\t\tif len(delta) > 0 {\n\t\t\terr := FlushResponse(w, flusher, StreamingResponse{\n\t\t\t\tAnswerID: answer_id,\n\t\t\t\tContent: delta,\n\t\t\t\tIsFinal: false,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\tlog.Printf(\"Failed to flush response: %v\", err)\n\t\t\t}\n\t\t}\n\t}\n\n\treturn &models.LLMAnswer{\n\t\tAnswer: answer,\n\t\tAnswerId: answer_id,\n\t}, nil\n}\n" }, { "path": "api/model_openai_service.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"log\"\n\t\"net/http\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/rotisserie/eris\"\n\topenai \"github.com/sashabaranov/go-openai\"\n\tllm_openai \"github.com/swuecho/chat_backend/llm/openai\"\n\t\"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// OpenAI ChatModel implementation\ntype OpenAIChatModel struct {\n\th *ChatHandler\n}\n\nfunc (m *OpenAIChatModel) Stream(w http.ResponseWriter, chatSession sqlc_queries.ChatSession, chatCompletionMessages []models.Message, chatUuid string, regenerate bool, streamOutput bool) (*models.LLMAnswer, error) {\n\topenAIRateLimiter.Wait(context.Background())\n\n\texceedPerModeRateLimitOrError := m.h.CheckModelAccess(w, chatSession.Uuid, chatSession.Model, chatSession.UserID)\n\tif exceedPerModeRateLimitOrError {\n\t\treturn nil, eris.New(\"exceed per mode rate limit\")\n\t}\n\n\tchatModel, err := GetChatModel(m.h.service.q, chatSession.Model)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tconfig, err := genOpenAIConfig(*chatModel)\n\tlog.Printf(\"%+v\", config.String())\n\t// print all config details\n\tif err != nil {\n\t\treturn nil, ErrOpenAIConfigFailed.WithMessage(\"Failed to generate OpenAI config\").WithDebugInfo(err.Error())\n\t}\n\n\tchatFiles, err := GetChatFiles(m.h.chatfileService.q, chatSession.Uuid)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\topenaiReq := NewChatCompletionRequest(chatSession, chatCompletionMessages, chatFiles, streamOutput)\n\tif len(openaiReq.Messages) <= 1 {\n\t\treturn nil, ErrSystemMessageError\n\t}\n\tlog.Printf(\"OpenAI request prepared - Model: %s, MessageCount: %d, Temperature: %.2f\",\n\t\topenaiReq.Model, len(openaiReq.Messages), openaiReq.Temperature)\n\tclient := openai.NewClientWithConfig(config)\n\tif streamOutput {\n\t\treturn doChatStream(w, client, openaiReq, chatSession.N, chatUuid, regenerate, m.h)\n\t} else {\n\t\treturn handleRegularResponse(w, client, openaiReq)\n\t}\n\n}\n\nfunc handleRegularResponse(w http.ResponseWriter, client *openai.Client, req openai.ChatCompletionRequest) (*models.LLMAnswer, error) {\n\t// check per chat_model limit\n\tctx, cancel := context.WithTimeout(context.Background(), 5*time.Minute)\n\tdefer cancel()\n\n\tcompletion, err := client.CreateChatCompletion(ctx, req)\n\tif err != nil {\n\t\tlog.Printf(\"fail to do request: %+v\", err)\n\t\treturn nil, ErrOpenAIRequestFailed.WithMessage(\"Failed to create chat completion\").WithDebugInfo(err.Error())\n\t}\n\tlog.Printf(\"completion: %+v\", completion)\n\tdata, _ := json.Marshal(completion)\n\tfmt.Fprint(w, string(data))\n\treturn &models.LLMAnswer{Answer: completion.Choices[0].Message.Content, AnswerId: completion.ID}, nil\n}\n\n// doChatStream handles streaming chat completion responses from OpenAI\n// It properly manages thinking tags for models that support reasoning content\nfunc doChatStream(w http.ResponseWriter, client *openai.Client, req openai.ChatCompletionRequest, bufferLen int32, chatUuid string, regenerate bool, handler *ChatHandler) (*models.LLMAnswer, error) {\n\t// Use request context with timeout, but prioritize client cancellation\n\tbaseCtx := context.Background()\n\tif handler != nil {\n\t\tbaseCtx = handler.GetRequestContext()\n\t}\n\tctx, cancel := context.WithTimeout(baseCtx, 5*time.Minute)\n\tdefer cancel()\n\n\tlog.Print(\"Creating OpenAI stream\")\n\tstream, err := client.CreateChatCompletionStream(ctx, req)\n\n\tif err != nil {\n\t\tlog.Printf(\"fail to do request: %+v\", err)\n\t\treturn nil, ErrOpenAIStreamFailed.WithMessage(\"Failed to create chat completion stream\").WithDebugInfo(err.Error())\n\t}\n\tdefer func() {\n\t\tif err := stream.Close(); err != nil {\n\t\t\tlog.Printf(\"Error closing OpenAI stream: %v\", err)\n\t\t}\n\t}()\n\n\t// Setup Server-Sent Events (SSE) streaming\n\tflusher, err := setupSSEStream(w)\n\tif err != nil {\n\t\treturn nil, APIError{\n\t\t\tHTTPCode: http.StatusInternalServerError,\n\t\t\tCode: \"STREAM_UNSUPPORTED\",\n\t\t\tMessage: \"Streaming unsupported by client\",\n\t\t}\n\t}\n\n\t// Initialize streaming state\n\tvar answer_id string\n\n\tvar hasReason bool // Whether we've detected any reasoning content\n\tvar reasonTagOpened bool // Whether we've sent the opening tag\n\tvar reasonTagClosed bool // Whether we've sent the closing tag\n\n\t// Ensure minimum buffer length\n\tif bufferLen == 0 {\n\t\tlog.Println(\"Buffer length is 0, setting to 1\")\n\t\tbufferLen = 1\n\t}\n\n\t// Initialize buffers for accumulating content\n\ttextBuffer := newTextBuffer(bufferLen, \"\", \"\")\n\treasonBuffer := newTextBuffer(bufferLen, \"\\n\\n\", \"\\n\\n\\n\\n\")\n\tanswer_id = GenerateAnswerID(chatUuid, regenerate)\n\t// Main streaming loop\n\tfor {\n\t\t// Check if client disconnected or context was cancelled\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\tlog.Printf(\"Stream cancelled by client: %v\", ctx.Err())\n\t\t\t// Return current accumulated content when cancelled\n\t\t\tllmAnswer := models.LLMAnswer{Answer: textBuffer.String(\"\\n\"), AnswerId: answer_id}\n\t\t\tif hasReason {\n\t\t\t\tllmAnswer.ReasoningContent = reasonBuffer.String(\"\\n\")\n\t\t\t}\n\t\t\treturn &llmAnswer, nil\n\t\tdefault:\n\t\t}\n\n\t\trawLine, err := stream.RecvRaw()\n\t\tif err != nil {\n\t\t\tlog.Printf(\"stream error: %+v\", err)\n\t\t\tif errors.Is(err, io.EOF) {\n\t\t\t\t// Stream ended successfully - return accumulated content\n\t\t\t\tllmAnswer := models.LLMAnswer{Answer: textBuffer.String(\"\\n\"), AnswerId: answer_id}\n\t\t\t\tif hasReason {\n\t\t\t\t\tllmAnswer.ReasoningContent = reasonBuffer.String(\"\\n\")\n\t\t\t\t}\n\t\t\t\treturn &llmAnswer, nil\n\t\t\t} else {\n\t\t\t\tlog.Printf(\"Stream error: %v\", err)\n\t\t\t\treturn nil, ErrOpenAIStreamFailed.WithMessage(\"Stream error occurred\").WithDebugInfo(err.Error())\n\t\t\t}\n\t\t}\n\t\t// Parse the streaming response\n\t\tresponse := llm_openai.ChatCompletionStreamResponse{}\n\t\terr = json.Unmarshal(rawLine, &response)\n\t\tif err != nil {\n\t\t\tlog.Printf(\"Could not unmarshal response: %v\\n\", err)\n\t\t\tcontinue\n\t\t}\n\n\t\t// Extract delta content from the response\n\t\ttextIdx := response.Choices[0].Index\n\t\tdelta := response.Choices[0].Delta\n\n\t\t// Accumulate content in buffers (for final answer construction)\n\t\ttextBuffer.appendByIndex(textIdx, delta.Content)\n\t\tif len(delta.ReasoningContent) > 0 {\n\t\t\thasReason = true\n\t\t\treasonBuffer.appendByIndex(textIdx, delta.ReasoningContent)\n\t\t}\n\n\t\t// Set answer ID from response if not already set\n\t\tif answer_id == \"\" {\n\t\t\tanswer_id = strings.TrimPrefix(response.ID, \"chatcmpl-\")\n\t\t}\n\n\t\t// Process and send delta content\n\t\tif len(delta.Content) > 0 || len(delta.ReasoningContent) > 0 {\n\t\t\tdeltaToSend := processDelta(delta, &reasonTagOpened, &reasonTagClosed, hasReason)\n\t\t\tif len(deltaToSend) > 0 {\n\t\t\t\tlog.Printf(\"delta: %s\", deltaToSend)\n\t\t\t\terr := FlushResponse(w, flusher, StreamingResponse{\n\t\t\t\t\tAnswerID: answer_id,\n\t\t\t\t\tContent: deltaToSend,\n\t\t\t\t\tIsFinal: false,\n\t\t\t\t})\n\t\t\t\tif err != nil {\n\t\t\t\t\tlog.Printf(\"Failed to flush response: %v\", err)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n}\n\n// processDelta handles the logic for processing delta content with thinking tags\nfunc processDelta(delta llm_openai.ChatCompletionStreamChoiceDelta, reasonTagOpened *bool, reasonTagClosed *bool, hasReason bool) string {\n\tvar deltaToSend string\n\n\tif len(delta.ReasoningContent) > 0 {\n\t\t// Handle reasoning content\n\t\tif !*reasonTagOpened {\n\t\t\t// First time seeing reasoning content, add opening tag\n\t\t\tdeltaToSend = \"\" + delta.ReasoningContent\n\t\t\t*reasonTagOpened = true\n\t\t} else {\n\t\t\t// Continue reasoning content\n\t\t\tdeltaToSend = delta.ReasoningContent\n\t\t}\n\t} else if hasReason && !*reasonTagClosed {\n\t\t// We had reasoning content before and now we have regular content for the first time\n\t\t// Close the think tag first, then send the content\n\t\tdeltaToSend = \"\" + delta.Content\n\t\t*reasonTagClosed = true\n\t} else {\n\t\t// Regular content without reasoning\n\t\tdeltaToSend = delta.Content\n\t}\n\n\treturn deltaToSend\n}\n\n// NewUserMessage creates a new OpenAI user message\nfunc NewUserMessage(content string) openai.ChatCompletionMessage {\n\treturn openai.ChatCompletionMessage{Role: \"user\", Content: content}\n}\n\n// NewChatCompletionRequest creates an OpenAI chat completion request from session and messages\nfunc NewChatCompletionRequest(chatSession sqlc_queries.ChatSession, chatCompletionMessages []models.Message, chatFiles []sqlc_queries.ChatFile, streamOutput bool) openai.ChatCompletionRequest {\n\topenaiMessages := messagesToOpenAIMesages(chatCompletionMessages, chatFiles)\n\n\tfor _, m := range openaiMessages {\n\t\tb, _ := m.MarshalJSON()\n\t\tlog.Printf(\"messages: %+v\\n\", string(b))\n\t}\n\n\tlog.Printf(\"messages: %+v\\n\", openaiMessages)\n\t// Ensure TopP is always greater than 0 to prevent API validation errors\n\ttopP := float32(chatSession.TopP) - 0.01\n\tif topP <= 0 {\n\t\ttopP = 0.01 // Minimum valid value\n\t}\n\topenaiReq := openai.ChatCompletionRequest{\n\t\tModel: chatSession.Model,\n\t\tMessages: openaiMessages,\n\t\tTemperature: float32(chatSession.Temperature),\n\t\tTopP: topP,\n\t\tN: int(chatSession.N),\n\t\tStream: streamOutput,\n\t}\n\treturn openaiReq\n}\n" }, { "path": "api/model_test_service.go", "content": "package main\n\nimport (\n\t\"encoding/json\"\n\t\"log\"\n\t\"net/http\"\n\n\t\"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\n// TestChatModel implements ChatModel interface for testing purposes\ntype TestChatModel struct {\n\th *ChatHandler\n}\n\n// Stream implements the ChatModel interface for test scenarios\nfunc (m *TestChatModel) Stream(w http.ResponseWriter, chatSession sqlc_queries.ChatSession, chat_completion_messages []models.Message, chatUuid string, regenerate bool, stream bool) (*models.LLMAnswer, error) {\n\treturn m.chatStreamTest(w, chatSession, chat_completion_messages, chatUuid, regenerate)\n}\n\n// chatStreamTest handles test chat streaming with mock responses\nfunc (m *TestChatModel) chatStreamTest(w http.ResponseWriter, chatSession sqlc_queries.ChatSession, chat_completion_messages []models.Message, chatUuid string, regenerate bool) (*models.LLMAnswer, error) {\n\tchatFiles, err := GetChatFiles(m.h.chatfileService.q, chatSession.Uuid)\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(ErrInternalUnexpected, \"Failed to get chat files\", err.Error()))\n\t\treturn nil, err\n\t}\n\n\tanswer_id := GenerateAnswerID(chatUuid, regenerate)\n\n\tflusher, err := setupSSEStream(w)\n\tif err != nil {\n\t\tRespondWithAPIError(w, createAPIError(APIError{\n\t\t\tHTTPCode: http.StatusInternalServerError,\n\t\t\tCode: \"STREAM_UNSUPPORTED\",\n\t\t\tMessage: \"Streaming unsupported by client\",\n\t\t}, \"\", err.Error()))\n\t\treturn nil, err\n\t}\n\n\tanswer := \"Hi, I am a chatbot. I can help you to find the best answer for your question. Please ask me a question.\"\n\terr = FlushResponse(w, flusher, StreamingResponse{\n\t\tAnswerID: answer_id,\n\t\tContent: answer,\n\t\tIsFinal: false,\n\t})\n\tif err != nil {\n\t\tlog.Printf(\"Failed to flush response: %v\", err)\n\t}\n\n\tif chatSession.Debug {\n\t\topenai_req := NewChatCompletionRequest(chatSession, chat_completion_messages, chatFiles, false)\n\t\treq_j, _ := json.Marshal(openai_req)\n\t\tanswer = answer + \"\\n\" + string(req_j)\n\t\terr := FlushResponse(w, flusher, StreamingResponse{\n\t\t\tAnswerID: answer_id,\n\t\t\tContent: answer,\n\t\t\tIsFinal: true,\n\t\t})\n\t\tif err != nil {\n\t\t\tlog.Printf(\"Failed to flush debug response: %v\", err)\n\t\t}\n\t}\n\n\treturn &models.LLMAnswer{\n\t\tAnswer: answer,\n\t\tAnswerId: answer_id,\n\t}, nil\n}\n" }, { "path": "api/models/models.go", "content": "package models\n\nimport (\n\t\"log\"\n\n\t\"github.com/pkoukk/tiktoken-go\"\n)\n\nfunc getTokenCount(content string) (int, error) {\n\tencoding := \"cl100k_base\"\n\ttke, err := tiktoken.GetEncoding(encoding)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\ttoken := tke.Encode(content, nil, nil)\n\tnum_tokens := len(token)\n\treturn num_tokens, nil\n}\n\ntype Message struct {\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n\ttokenCount int32\n}\n\nfunc (m Message) TokenCount() int32 {\n\tif m.tokenCount != 0 {\n\t\treturn m.tokenCount\n\t} else {\n\t\ttokenCount, err := getTokenCount(m.Content)\n\t\tif err != nil {\n\t\t\tlog.Println(err)\n\t\t}\n\t\treturn int32(tokenCount) + 1\n\t}\n}\n\nfunc (m *Message) SetTokenCount(tokenCount int32) *Message {\n\tm.tokenCount = tokenCount\n\treturn m\n}\n\ntype LLMAnswer struct {\n\tAnswerId string `json:\"id\"`\n\tAnswer string `json:\"answer\"`\n\tReasoningContent string `json:\"reason_content\"`\n}\n" }, { "path": "api/models.go", "content": "package main\n\nimport (\n\t\"net/http\"\n\t\"time\"\n\n\t\"github.com/swuecho/chat_backend/models\"\n\t\"github.com/swuecho/chat_backend/sqlc_queries\"\n)\n\ntype TokenResult struct {\n\tAccessToken string `json:\"accessToken\"`\n\tExpiresIn int `json:\"expiresIn\"`\n}\n\ntype ConversationRequest struct {\n\tUUID string `json:\"uuid,omitempty\"`\n\tConversationID string `json:\"conversationId,omitempty\"`\n\tParentMessageID string `json:\"parentMessageId,omitempty\"`\n}\n\ntype RequestOption struct {\n\tPrompt string `json:\"prompt,omitempty\"`\n\tOptions ConversationRequest `json:\"options,omitempty\"`\n}\n\ntype Artifact struct {\n\tUUID string `json:\"uuid\"`\n\tType string `json:\"type\"` // 'code', 'html', 'svg', 'mermaid', 'json', 'markdown'\n\tTitle string `json:\"title\"`\n\tContent string `json:\"content\"`\n\tLanguage string `json:\"language,omitempty\"` // for code artifacts\n}\n\ntype SimpleChatMessage struct {\n\tUuid string `json:\"uuid\"`\n\tDateTime string `json:\"dateTime\"`\n\tText string `json:\"text\"`\n\tInversion bool `json:\"inversion\"`\n\tError bool `json:\"error\"`\n\tLoading bool `json:\"loading\"`\n\tIsPin bool `json:\"isPin\"`\n\tIsPrompt bool `json:\"isPrompt\"`\n\tArtifacts []Artifact `json:\"artifacts,omitempty\"`\n}\n\nfunc (msg SimpleChatMessage) GetRole() string {\n\tvar role string\n\tif msg.Inversion {\n\t\trole = \"user\"\n\t} else {\n\t\trole = \"assistant\"\n\t}\n\treturn role\n\n}\n\ntype SimpleChatSession struct {\n\tUuid string `json:\"uuid\"`\n\tIsEdit bool `json:\"isEdit\"`\n\tTitle string `json:\"title\"`\n\tMaxLength int `json:\"maxLength\"`\n\tTemperature float64 `json:\"temperature\"`\n\tTopP float64 `json:\"topP\"`\n\tN int32 `json:\"n\"`\n\tMaxTokens int32 `json:\"maxTokens\"`\n\tDebug bool `json:\"debug\"`\n\tModel string `json:\"model\"`\n\tSummarizeMode bool `json:\"summarizeMode\"`\n\tArtifactEnabled bool `json:\"artifactEnabled\"`\n\tWorkspaceUuid string `json:\"workspaceUuid\"`\n}\n\ntype ChatMessageResponse struct {\n\tUuid string `json:\"uuid\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n\tScore float64 `json:\"score\"`\n\tUserID int32 `json:\"userId\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tCreatedBy int32 `json:\"createdBy\"`\n\tUpdatedBy int32 `json:\"updatedBy\"`\n\tArtifacts []Artifact `json:\"artifacts,omitempty\"`\n}\n\ntype ChatSessionResponse struct {\n\tUuid string `json:\"uuid\"`\n\tTopic string `json:\"topic\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tMaxLength int32 `json:\"maxLength\"`\n\tArtifactEnabled bool `json:\"artifactEnabled\"`\n}\n\ntype Pagination struct {\n\tPage int32 `json:\"page\"`\n\tSize int32 `json:\"size\"`\n\tData []interface{} `json:\"data\"`\n\tTotal int64 `json:\"total\"`\n}\n\nfunc (p *Pagination) Offset() int32 {\n\treturn (p.Page - 1) * p.Size\n}\n\n// ChatModel interface\ntype ChatModel interface {\n\tStream(w http.ResponseWriter, chatSession sqlc_queries.ChatSession, chat_compeletion_messages []models.Message, chatUuid string, regenerate bool, stream bool) (*models.LLMAnswer, error)\n}\n" }, { "path": "api/openai_test.go", "content": "package main\n\nimport \"testing\"\n\nfunc Test_getModelBaseUrl(t *testing.T) {\n\n\ttestCases := []struct {\n\t\tname string\n\t\tapiUrl string\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tname: \"Base URL with v1 version\",\n\t\t\tapiUrl: \"https://api.openai-sb.com/v1/chat/completions\",\n\t\t\texpected: \"https://api.openai-sb.com/v1\",\n\t\t},\n\t\t{\n\t\t\tname: \"Base URL with v2 version\",\n\t\t\tapiUrl: \"https://api.openai-sb.com/v2/completions\",\n\t\t\texpected: \"https://api.openai-sb.com/v2\",\n\t\t},\n\t\t{\n\t\t\tname: \"Base URL with no version\",\n\t\t\tapiUrl: \"https://api.openai-sb.com/chat/completions\",\n\t\t\texpected: \"https://api.openai-sb.com/chat\",\n\t\t},\n\t\t{\n\t\t\tname: \"Base URL with different host\",\n\t\t\tapiUrl: \"https://example.com/v1/chat/completions\",\n\t\t\texpected: \"https://example.com/v1\",\n\t\t},\n\t\t{\n\t\t\tname: \"Base URL with different host\",\n\t\t\tapiUrl: \"https://docs-test-001.openai.azure.com/\",\n\t\t\texpected: \"https://docs-test-001.openai.azure.com/\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tactual, _ := getModelBaseUrl(tc.apiUrl)\n\t\t\tif actual != tc.expected {\n\t\t\t\tt.Errorf(\"Expected base URL '%s', but got '%s'\", tc.expected, actual)\n\t\t\t}\n\t\t})\n\t}\n}\n" }, { "path": "api/pre-commit.sh", "content": "#!/bin/bash\n\necho \"Running 'go fmt' check...\"\n\nfiles=$(git diff --cached --name-only --diff-filter=ACM \"*.go\")\n\nif [ -z \"$files\" ]; then\n echo \"No Go files to check.\"\n exit 0\nfi\n\nunformatted_files=\"\"\nfor file in ${files}; do\n if [[ ! -z $(go fmt ${file}) ]]; then\n unformatted_files=\"$unformatted_files ${file}\"\n fi\ndone\n\nif [ ! -z \"$unformatted_files\" ]; then\n echo \"The following files are not properly formatted:\"\n echo \"$unformatted_files\"\n echo \"Please run 'go fmt' before committing.\"\n exit 1\nfi\n\necho \"'go fmt' check passed.\"\nexit 0\n" }, { "path": "api/sqlc/README.txt", "content": "you are a golang code assistant. Given table DDL, you will write all queies for sqlc in a crud applicaiton,\n\nplease do not send me any generated go code.\n\n### input\n\nCREATE TABLE chat_message (\n id integer PRIMARY KEY,\n chat_session_id integer NOT NULL,\n role character varying(255) NOT NULL,\n content character varying NOT NULL,\n score double precision NOT NULL,\n user_id integer NOT NULL,\n created_at timestamp without time zone,\n updated_at timestamp without time zone,\n created_by integer NOT NULL,\n updated_by integer NOT NULL,\n raw jsonb\n);\n\n## output\n\n-- name: ListChatMessages :many\nSELECT * FROM chat_message ORDER BY id;\n\n-- name: ChatMessagesBySessionID :many\nSELECT * FROM chat_message WHERE chat_session_id = $1 ORDER BY id;\n\n-- name: ChatMessageByID :one\nSELECT * FROM chat_message WHERE id = $1;\n\n-- name: CreateChatMessage :one\nINSERT INTO chat_message (chat_session_id, role, content, model, score, user_id, created_at, updated_at, created_by, updated_by, raw)\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11)\nRETURNING *;\n\n-- name: UpdateChatMessage :one\nUPDATE chat_message SET role = $2, content = $3, score = $4, user_id = $5, updated_at = $6, updated_by = $7, raw = $8\nWHERE id = $1\nRETURNING *;\n\n-- name: DeleteChatMessage :exec\nDELETE FROM chat_message WHERE id = $1;\n\n\n\n### input\n\nCREATE TABLE chat_session (\n id integer PRIMARY KEY,\n user_id integer NOT NULL,\n topic character varying(255) NOT NULL,\n created_at timestamp without time zone DEFAULT now() NOT NULL,\n updated_at timestamp without time zone DEFAULT now() NOT NULL,\n active boolean default true NOT NULL,\n max_length integer DEFAULT 0 NOT NULL\n);\n\n\n### \ntype ChatSessionService struct {\n\tq *sqlc_queries.Queries\n}\n\n// NewChatSessionService creates a new ChatSessionService.\nfunc NewChatSessionService(q *sqlc_queries.Queries) *ChatSessionService {\n\treturn &ChatSessionService{q: q}\n}\n\n// CreateChatSession creates a new chat session.\nfunc (s *ChatSessionService) CreateChatSession(ctx context.Context, session_params sqlc_queries.CreateChatSessionParams) (sqlc_queries.ChatSession, error) {\n\tsession, err := s.q.CreateChatSession(ctx, session_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatSession{}, errors.New(\"failed to create session\")\n\t}\n\treturn session, nil\n}\n\n// GetChatSessionByID returns a chat session by ID.\nfunc (s *ChatSessionService) GetChatSessionByID(ctx context.Context, id int32) (sqlc_queries.ChatSession, error) {\n\tsession, err := s.q.GetChatSessionByID(ctx, id)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatSession{}, errors.New(\"failed to retrieve session\")\n\t}\n\treturn session, nil\n}\n\n// UpdateChatSession updates an existing chat session.\nfunc (s *ChatSessionService) UpdateChatSession(ctx context.Context, session_params sqlc_queries.UpdateChatSessionParams) (sqlc_queries.ChatSession, error) {\n\tsession_u, err := s.q.UpdateChatSession(ctx, session_params)\n\tif err != nil {\n\t\treturn sqlc_queries.ChatSession{}, errors.New(\"failed to update session\")\n\t}\n\treturn session_u, nil\n}\n\n// DeleteChatSession deletes a chat session by ID.\nfunc (s *ChatSessionService) DeleteChatSession(ctx context.Context, id int32) error {\n\terr := s.q.DeleteChatSession(ctx, id)\n\tif err != nil {\n\t\treturn errors.New(\"failed to delete session\")\n\t}\n\treturn nil\n}\n\n// GetAllChatSessions returns all chat sessions.\nfunc (s *ChatSessionService) GetAllChatSessions(ctx context.Context) ([]sqlc_queries.ChatSession, error) {\n\tsessions, err := s.q.GetAllChatSessions(ctx)\n\tif err != nil {\n\t\treturn nil, errors.New(\"failed to retrieve sessions\")\n\t}\n\treturn sessions, nil\n}\n\n\n\n\n\ncreate sql\n\nINSERT INTO auth_user (id, username, email, password, first_name, last_name, is_active, is_staff, is_superuser, date_joined)\nVALUES (1, 'echowuhao', 'echowuhao@gmail.com', \n'pbkdf2_sha256$150000$wVq3kpPZc7pJ$+dO5tCzI9Xu9iGkWtL/Ho11DQsoOx2ZB1OVDGOlKyk4=', 'Hao', 'Wu', true, false, false, now());\n\nNote that when generating password hashes using Django or any other library, it is important to use a strong, one-way hashing algorithm with a sufficiently high cost parameter. In this example, the cost factor is set to 150000, which should provide adequate security against brute-force attacks.\n\n\n\nDROP FUNCTION IF EXISTS tsvector_immutable(text);\n-- why this is necessary?\nCREATE FUNCTION tsvector_immutable(text) RETURNS tsvector AS $$\n SELECT to_tsvector($1)\n$$ LANGUAGE sql IMMUTABLE;\n\n\nUPDATE chat_snapshot\nSET text = array_to_string(ARRAY(SELECT jsonb_array_elements(conversation)->>'text'), ' ')::text\nWHERE text = ''\n\nALTER TABLE chat_snapshot\nADD COLUMN IF NOT EXISTS text_vector tsvector generated always as\t(\n to_tsvector(text)\n) stored; \n" }, { "path": "api/sqlc/queries/auth_user.sql", "content": "-- name: GetAllAuthUsers :many\nSELECT * FROM auth_user ORDER BY id;\n\n-- name: ListAuthUsers :many\nSELECT * FROM auth_user ORDER BY id LIMIT $1 OFFSET $2;\n\n-- name: GetAuthUserByID :one\nSELECT * FROM auth_user WHERE id = $1;\n\n\n-- name: GetAuthUserByEmail :one\nSELECT * FROM auth_user WHERE email = $1;\n\n-- name: CreateAuthUser :one\nINSERT INTO auth_user (email, \"password\", first_name, last_name, username, is_staff, is_superuser)\nVALUES ($1, $2, $3, $4, $5, $6, $7)\nRETURNING *;\n\n-- name: UpdateAuthUser :one\nUPDATE auth_user SET first_name = $2, last_name= $3, last_login = now() \nWHERE id = $1\nRETURNING first_name, last_name, email;\n\n-- name: UpdateAuthUserByEmail :one\nUPDATE auth_user SET first_name = $2, last_name= $3, last_login = now() \nWHERE email = $1\nRETURNING first_name, last_name, email;\n\n-- name: DeleteAuthUser :exec\nDELETE FROM auth_user WHERE email = $1;\n\n-- name: GetUserByEmail :one\nSELECT * FROM auth_user WHERE email = $1;\n\n-- name: UpdateUserPassword :exec\nUPDATE auth_user SET \"password\" = $2 WHERE email = $1;\n\n-- name: GetTotalActiveUserCount :one\nSELECT COUNT(*) FROM auth_user WHERE is_active = true;\n\n\n-- name: UpdateAuthUserRateLimitByEmail :one\nINSERT INTO auth_user_management (user_id, rate_limit, created_at, updated_at)\nVALUES ((SELECT id FROM auth_user WHERE email = $1), $2, NOW(), NOW())\nON CONFLICT (user_id) DO UPDATE SET rate_limit = $2, updated_at = NOW()\nRETURNING rate_limit;\n\n-- name: GetUserStats :many\nSELECT \n auth_user.first_name,\n auth_user.last_name,\n auth_user.email AS user_email,\n COALESCE(user_stats.total_messages, 0) AS total_chat_messages,\n COALESCE(user_stats.total_token_count, 0) AS total_token_count,\n COALESCE(user_stats.total_messages_3_days, 0) AS total_chat_messages_3_days,\n COALESCE(user_stats.total_token_count_3_days, 0) AS total_token_count_3_days,\n COALESCE(auth_user_management.rate_limit, @default_rate_limit::INTEGER) AS rate_limit\nFROM auth_user\nLEFT JOIN (\n SELECT chat_message_stats.user_id, \n SUM(total_messages) AS total_messages, \n SUM(total_token_count) AS total_token_count,\n SUM(CASE WHEN created_at >= NOW() - INTERVAL '3 days' THEN total_messages ELSE 0 END) AS total_messages_3_days,\n SUM(CASE WHEN created_at >= NOW() - INTERVAL '3 days' THEN total_token_count ELSE 0 END) AS total_token_count_3_days\n FROM (\n SELECT user_id, COUNT(*) AS total_messages, SUM(token_count) as total_token_count, MAX(created_at) AS created_at\n FROM chat_message\n GROUP BY user_id, chat_session_uuid\n ) AS chat_message_stats\n GROUP BY chat_message_stats.user_id\n) AS user_stats ON auth_user.id = user_stats.user_id\nLEFT JOIN auth_user_management ON auth_user.id = auth_user_management.user_id\nORDER BY total_chat_messages DESC, auth_user.id DESC\nOFFSET $2\nLIMIT $1;\n\n-- name: GetUserAnalysisByEmail :one\nSELECT \n auth_user.first_name,\n auth_user.last_name,\n auth_user.email AS user_email,\n COALESCE(user_stats.total_messages, 0) AS total_messages,\n COALESCE(user_stats.total_token_count, 0) AS total_tokens,\n COALESCE(user_stats.total_sessions, 0) AS total_sessions,\n COALESCE(user_stats.total_messages_3_days, 0) AS messages_3_days,\n COALESCE(user_stats.total_token_count_3_days, 0) AS tokens_3_days,\n COALESCE(auth_user_management.rate_limit, @default_rate_limit::INTEGER) AS rate_limit\nFROM auth_user\nLEFT JOIN (\n SELECT \n stats.user_id, \n SUM(stats.total_messages) AS total_messages, \n SUM(stats.total_token_count) AS total_token_count,\n COUNT(DISTINCT stats.chat_session_uuid) AS total_sessions,\n SUM(CASE WHEN stats.created_at >= NOW() - INTERVAL '3 days' THEN stats.total_messages ELSE 0 END) AS total_messages_3_days,\n SUM(CASE WHEN stats.created_at >= NOW() - INTERVAL '3 days' THEN stats.total_token_count ELSE 0 END) AS total_token_count_3_days\n FROM (\n SELECT user_id, chat_session_uuid, COUNT(*) AS total_messages, SUM(token_count) as total_token_count, MAX(created_at) AS created_at\n FROM chat_message\n WHERE is_deleted = false\n GROUP BY user_id, chat_session_uuid\n ) AS stats\n GROUP BY stats.user_id\n) AS user_stats ON auth_user.id = user_stats.user_id\nLEFT JOIN auth_user_management ON auth_user.id = auth_user_management.user_id\nWHERE auth_user.email = $1;\n\n-- name: GetUserModelUsageByEmail :many\nSELECT \n COALESCE(cm.model, 'unknown') AS model,\n COUNT(*) AS message_count,\n COALESCE(SUM(cm.token_count), 0) AS token_count,\n MAX(cm.created_at)::timestamp AS last_used\nFROM chat_message cm\nINNER JOIN auth_user au ON cm.user_id = au.id\nWHERE au.email = $1 \n AND cm.is_deleted = false \n AND cm.role = 'assistant'\n AND cm.model IS NOT NULL \n AND cm.model != ''\nGROUP BY cm.model\nORDER BY message_count DESC;\n\n-- name: GetUserRecentActivityByEmail :many\nSELECT \n DATE(cm.created_at) AS activity_date,\n COUNT(*) AS messages,\n COALESCE(SUM(cm.token_count), 0) AS tokens,\n COUNT(DISTINCT cm.chat_session_uuid) AS sessions\nFROM chat_message cm\nINNER JOIN auth_user au ON cm.user_id = au.id\nWHERE au.email = $1 \n AND cm.is_deleted = false \n AND cm.created_at >= NOW() - INTERVAL '30 days'\nGROUP BY DATE(cm.created_at)\nORDER BY activity_date DESC\nLIMIT 30;\n\n-- name: GetUserSessionHistoryByEmail :many\nSELECT \n cs.uuid AS session_id,\n cs.model,\n COALESCE(COUNT(cm.id), 0) AS message_count,\n COALESCE(SUM(cm.token_count), 0) AS token_count,\n COALESCE(MIN(cm.created_at), cs.created_at)::timestamp AS created_at,\n COALESCE(MAX(cm.created_at), cs.updated_at)::timestamp AS updated_at\nFROM chat_session cs\nINNER JOIN auth_user au ON cs.user_id = au.id\nLEFT JOIN chat_message cm ON cs.uuid = cm.chat_session_uuid AND cm.is_deleted = false\nWHERE au.email = $1 AND cs.active = true\nGROUP BY cs.uuid, cs.model, cs.created_at, cs.updated_at\nORDER BY cs.updated_at DESC\nLIMIT $2 OFFSET $3;\n\n-- name: GetUserSessionHistoryCountByEmail :one\nSELECT COUNT(DISTINCT cs.uuid) AS total_sessions\nFROM chat_session cs\nINNER JOIN auth_user au ON cs.user_id = au.id\nWHERE au.email = $1 AND cs.active = true;" }, { "path": "api/sqlc/queries/auth_user_management.sql", "content": "-- name: GetRateLimit :one\n-- GetRateLimit retrieves the rate limit for a user from the auth_user_management table.\n-- If no rate limit is set for the user, it returns the default rate limit of 100.\nSELECT rate_limit AS rate_limit\nFROM auth_user_management\nWHERE user_id = $1;\n\n" }, { "path": "api/sqlc/queries/bot_answer_history.sql", "content": "-- Bot Answer History Queries --\n\n-- name: CreateBotAnswerHistory :one\nINSERT INTO bot_answer_history (\n bot_uuid,\n user_id,\n prompt,\n answer,\n model,\n tokens_used\n) VALUES (\n $1, $2, $3, $4, $5, $6\n) RETURNING *;\n\n-- name: GetBotAnswerHistoryByID :one\nSELECT \n bah.id,\n bah.bot_uuid,\n bah.user_id,\n bah.prompt,\n bah.answer,\n bah.model,\n bah.tokens_used,\n bah.created_at,\n bah.updated_at,\n au.username AS user_username,\n au.email AS user_email\nFROM bot_answer_history bah\nJOIN auth_user au ON bah.user_id = au.id\nWHERE bah.id = $1;\n\n-- name: GetBotAnswerHistoryByBotUUID :many\nSELECT \n bah.id,\n bah.bot_uuid,\n bah.user_id,\n bah.prompt,\n bah.answer,\n bah.model,\n bah.tokens_used,\n bah.created_at,\n bah.updated_at,\n au.username AS user_username,\n au.email AS user_email\nFROM bot_answer_history bah\nJOIN auth_user au ON bah.user_id = au.id\nWHERE bah.bot_uuid = $1\nORDER BY bah.created_at DESC\nLIMIT $2 OFFSET $3;\n\n-- name: GetBotAnswerHistoryByUserID :many\nSELECT \n bah.id,\n bah.bot_uuid,\n bah.user_id,\n bah.prompt,\n bah.answer,\n bah.model,\n bah.tokens_used,\n bah.created_at,\n bah.updated_at,\n au.username AS user_username,\n au.email AS user_email\nFROM bot_answer_history bah\nJOIN auth_user au ON bah.user_id = au.id\nWHERE bah.user_id = $1\nORDER BY bah.created_at DESC\nLIMIT $2 OFFSET $3;\n\n-- name: UpdateBotAnswerHistory :one\nUPDATE bot_answer_history\nSET\n answer = $2,\n tokens_used = $3,\n updated_at = NOW()\nWHERE id = $1\nRETURNING *;\n\n-- name: DeleteBotAnswerHistory :exec\nDELETE FROM bot_answer_history WHERE id = $1;\n\n-- name: GetBotAnswerHistoryCountByBotUUID :one\nSELECT COUNT(*) FROM bot_answer_history WHERE bot_uuid = $1;\n\n-- name: GetBotAnswerHistoryCountByUserID :one\nSELECT COUNT(*) FROM bot_answer_history WHERE user_id = $1;\n\n-- name: GetLatestBotAnswerHistoryByBotUUID :many\nSELECT \n bah.id,\n bah.bot_uuid,\n bah.user_id,\n bah.prompt,\n bah.answer,\n bah.model,\n bah.tokens_used,\n bah.created_at,\n bah.updated_at,\n au.username AS user_username,\n au.email AS user_email\nFROM bot_answer_history bah\nJOIN auth_user au ON bah.user_id = au.id\nWHERE bah.bot_uuid = $1\nORDER BY bah.created_at DESC\nLIMIT $2;\n" }, { "path": "api/sqlc/queries/chat_comment.sql", "content": "-- name: CreateChatComment :one\nINSERT INTO chat_comment (\n uuid,\n chat_session_uuid,\n chat_message_uuid, \n content,\n created_by,\n updated_by\n) VALUES (\n $1, $2, $3, $4, $5, $5\n) RETURNING *;\n\n-- name: GetCommentsBySessionUUID :many\nSELECT \n cc.uuid,\n cc.chat_message_uuid,\n cc.content,\n cc.created_at,\n au.username AS author_username,\n au.email AS author_email\nFROM chat_comment cc\nJOIN auth_user au ON cc.created_by = au.id\nWHERE cc.chat_session_uuid = $1\nORDER BY cc.created_at DESC;\n\n-- name: GetCommentsByMessageUUID :many\nSELECT \n cc.uuid,\n cc.content,\n cc.created_at,\n au.username AS author_username,\n au.email AS author_email\nFROM chat_comment cc\nJOIN auth_user au ON cc.created_by = au.id\nWHERE cc.chat_message_uuid = $1\nORDER BY cc.created_at DESC;\n\n" }, { "path": "api/sqlc/queries/chat_file.sql", "content": "-- name: CreateChatFile :one\nINSERT INTO chat_file (name, data, user_id, chat_session_uuid, mime_type)\nVALUES ($1, $2, $3, $4, $5)\nRETURNING *;\n\n-- name: ListChatFilesBySessionUUID :many\nSELECT id, name\nFROM chat_file\nWHERE user_id = $1 and chat_session_uuid = $2\nORDER BY created_at ;\n\n-- name: ListChatFilesWithContentBySessionUUID :many\nSELECT *\nFROM chat_file\nWHERE chat_session_uuid = $1\nORDER BY created_at;\n\n\n-- name: GetChatFileByID :one\nSELECT id, name, data, created_at, user_id, chat_session_uuid\nFROM chat_file\nWHERE id = $1;\n\n-- name: DeleteChatFile :one\nDELETE FROM chat_file\nWHERE id = $1\nRETURNING *;\n" }, { "path": "api/sqlc/queries/chat_log.sql", "content": "-- name: ListChatLogs :many\nSELECT * FROM chat_logs ORDER BY id;\n\n-- name: ChatLogByID :one\nSELECT * FROM chat_logs WHERE id = $1;\n\n-- name: CreateChatLog :one\nINSERT INTO chat_logs (session, question, answer)\nVALUES ($1, $2, $3)\nRETURNING *;\n\n-- name: UpdateChatLog :one\nUPDATE chat_logs SET session = $2, question = $3, answer = $4\nWHERE id = $1\nRETURNING *;\n\n-- name: DeleteChatLog :exec\nDELETE FROM chat_logs WHERE id = $1;" }, { "path": "api/sqlc/queries/chat_message.sql", "content": "-- name: GetAllChatMessages :many\nSELECT * FROM chat_message \nWHERE is_deleted = false\nORDER BY id;\n\n-- name: GetChatMessagesBySessionUUID :many\nSELECT cm.*\nFROM chat_message cm\nINNER JOIN chat_session cs ON cm.chat_session_uuid = cs.uuid\nWHERE cm.is_deleted = false and cs.active = true and cs.uuid = $1 \nORDER BY cm.id \nOFFSET $2\nLIMIT $3;\n\n\n-- name: GetChatMessageBySessionUUID :one\nSELECT cm.*\nFROM chat_message cm\nINNER JOIN chat_session cs ON cm.chat_session_uuid = cs.uuid\nWHERE cm.is_deleted = false and cs.active = true and cs.uuid = $1 \nORDER BY cm.id \nOFFSET $2\nLIMIT $1;\n\n\n-- name: GetChatMessageByID :one\nSELECT * FROM chat_message \nWHERE is_deleted = false and id = $1;\n\n\n-- name: CreateChatMessage :one\nINSERT INTO chat_message (chat_session_uuid, uuid, role, content, reasoning_content, model, token_count, score, user_id, created_by, updated_by, llm_summary, raw, artifacts, suggested_questions)\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15)\nRETURNING *;\n\n-- name: UpdateChatMessage :one\nUPDATE chat_message SET role = $2, content = $3, score = $4, user_id = $5, updated_by = $6, artifacts = $7, suggested_questions = $8, updated_at = now()\nWHERE id = $1\nRETURNING *;\n\n-- name: DeleteChatMessage :exec\nUPDATE chat_message set is_deleted = true, updated_at = now()\nWHERE id = $1;\n\n\n---- UUID ----\n\n-- name: GetChatMessageByUUID :one\nSELECT * FROM chat_message \nWHERE is_deleted = false and uuid = $1;\n\n\n-- name: UpdateChatMessageByUUID :one\nUPDATE chat_message SET content = $2, is_pin = $3, token_count = $4, artifacts = $5, suggested_questions = $6, updated_at = now() \nWHERE uuid = $1\nRETURNING *;\n\n-- name: DeleteChatMessageByUUID :exec\nUPDATE chat_message SET is_deleted = true, updated_at = now()\nWHERE uuid = $1;\n\n\n-- name: HasChatMessagePermission :one\nSELECT COUNT(*) > 0 as has_permission\nFROM chat_message cm\nINNER JOIN chat_session cs ON cm.chat_session_uuid = cs.uuid\nINNER JOIN auth_user au ON cs.user_id = au.id\nWHERE cm.is_deleted = false and cm.id = $1 AND (cs.user_id = $2 OR au.is_superuser) and cs.active = true;\n\n\n-- name: GetLatestMessagesBySessionUUID :many\nSELECT *\nFROM chat_message\nWhere chat_message.id in \n(\n SELECT chat_message.id\n FROM chat_message\n WHERE chat_message.chat_session_uuid = $1 and chat_message.is_deleted = false and chat_message.is_pin = true\n UNION\n (\n SELECT chat_message.id\n FROM chat_message\n WHERE chat_message.chat_session_uuid = $1 and chat_message.is_deleted = false -- and chat_message.is_pin = false\n ORDER BY created_at DESC\n LIMIT $2\n )\n)\nORDER BY created_at;\n\n\n-- name: GetFirstMessageBySessionUUID :one\nSELECT *\nFROM chat_message\nWHERE chat_session_uuid = $1 and is_deleted = false\nORDER BY created_at \nLIMIT 1;\n\n-- name: GetLastNChatMessages :many\nSELECT *\nFROM chat_message\nWHERE chat_message.id in (\n SELECT id\n FROM chat_message cm\n WHERE cm.chat_session_uuid = $3 and cm.is_deleted = false and cm.is_pin = true\n UNION\n (\n SELECT id \n FROM chat_message cm\n WHERE cm.chat_session_uuid = $3 \n AND cm.id < (SELECT id FROM chat_message WHERE chat_message.uuid = $1)\n AND cm.is_deleted = false -- and cm.is_pin = false\n ORDER BY cm.created_at DESC\n LIMIT $2\n )\n) \nORDER BY created_at;\n\n\n-- name: UpdateChatMessageContent :exec\nUPDATE chat_message\nSET content = $2, updated_at = now(), token_count = $3\nWHERE uuid = $1 ;\n\n-- name: UpdateChatMessageSuggestions :one\nUPDATE chat_message \nSET suggested_questions = $2, updated_at = now() \nWHERE uuid = $1\nRETURNING *;\n\n-- name: DeleteChatMessagesBySesionUUID :exec\nUPDATE chat_message \nSET is_deleted = true, updated_at = now()\nWHERE is_deleted = false and is_pin = false and chat_session_uuid = $1;\n\n\n-- name: GetChatMessagesCount :one\n-- Get total chat message count for user in last 10 minutes\nSELECT COUNT(*)\nFROM chat_message\nWHERE user_id = $1\nAND created_at >= NOW() - INTERVAL '10 minutes';\n\n\n-- name: GetChatMessagesCountByUserAndModel :one\n-- Get total chat message count for user of model in last 10 minutes\nSELECT COUNT(*)\nFROM chat_message cm\nJOIN chat_session cs ON (cm.chat_session_uuid = cs.uuid AND cs.user_id = cm.user_id)\nWHERE cm.user_id = $1\nAND cs.model = $2 \nAND cm.created_at >= NOW() - INTERVAL '10 minutes';\n\n\n-- name: GetLatestUsageTimeOfModel :many\nSELECT \n model,\n MAX(created_at)::timestamp as latest_message_time,\n COUNT(*) as message_count\nFROM chat_message\nWHERE \n created_at >= NOW() - sqlc.arg(time_interval)::text::INTERVAL\n AND is_deleted = false\n AND model != ''\n AND role = 'assistant'\nGROUP BY model\nORDER BY latest_message_time DESC;\n\n-- name: GetChatMessagesBySessionUUIDForAdmin :many\nSELECT \n id,\n uuid,\n role,\n content,\n reasoning_content,\n model,\n token_count,\n user_id,\n created_at,\n updated_at\nFROM (\n -- Include session prompts as the first messages\n SELECT \n cp.id,\n cp.uuid,\n cp.role,\n cp.content,\n ''::text as reasoning_content,\n cs.model,\n cp.token_count,\n cp.user_id,\n cp.created_at,\n cp.updated_at\n FROM chat_prompt cp\n INNER JOIN chat_session cs ON cp.chat_session_uuid = cs.uuid\n WHERE cp.chat_session_uuid = $1 \n AND cp.is_deleted = false\n AND cp.role = 'system'\n \n UNION ALL\n \n -- Include regular chat messages\n SELECT \n id,\n uuid,\n role,\n content,\n reasoning_content,\n model,\n token_count,\n user_id,\n created_at,\n updated_at\n FROM chat_message\n WHERE chat_session_uuid = $1 \n AND is_deleted = false\n) combined_messages\nORDER BY created_at ASC;" }, { "path": "api/sqlc/queries/chat_model.sql", "content": "-- name: ListChatModels :many\nSELECT * FROM chat_model ORDER BY order_number;\n\n-- name: ListSystemChatModels :many\nSELECT * FROM chat_model\nwhere user_id in (select id from auth_user where is_superuser = true)\nORDER BY order_number, id desc;\n\n-- name: ChatModelByID :one\nSELECT * FROM chat_model WHERE id = $1;\n\n-- name: ChatModelByName :one\nSELECT * FROM chat_model WHERE name = $1;\n\n-- name: CreateChatModel :one\nINSERT INTO chat_model (name, label, is_default, url, api_auth_header, api_auth_key, user_id, enable_per_mode_ratelimit, max_token, default_token, order_number, http_time_out, api_type )\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13)\nRETURNING *;\n\n-- name: UpdateChatModel :one\nUPDATE chat_model SET name = $2, label = $3, is_default = $4, url = $5, api_auth_header = $6, api_auth_key = $7, enable_per_mode_ratelimit = $9,\nmax_token = $10, default_token = $11, order_number = $12, http_time_out = $13, is_enable = $14, api_type = $15\nWHERE id = $1 and user_id = $8\nRETURNING *;\n\n-- name: UpdateChatModelKey :one\nUPDATE chat_model SET api_auth_key = $2\nWHERE id = $1\nRETURNING *;\n\n-- name: DeleteChatModel :exec\nDELETE FROM chat_model WHERE id = $1 and user_id = $2;\n\n-- name: GetDefaultChatModel :one\nSELECT * FROM chat_model WHERE is_default = true\nand user_id in (select id from auth_user where is_superuser = true)\nORDER BY order_number, id\nLIMIT 1;" }, { "path": "api/sqlc/queries/chat_prompt.sql", "content": "-- name: GetAllChatPrompts :many\nSELECT * FROM chat_prompt \nWHERE is_deleted = false\nORDER BY id;\n\n-- name: GetChatPromptByID :one\nSELECT * FROM chat_prompt\nWHERE is_deleted = false and id = $1;\n\n-- name: CreateChatPrompt :one\nINSERT INTO chat_prompt (uuid, chat_session_uuid, role, content, token_count, user_id, created_by, updated_by)\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8)\nRETURNING *;\n\n-- name: UpdateChatPrompt :one\nUPDATE chat_prompt SET chat_session_uuid = $2, role = $3, content = $4, score = $5, user_id = $6, updated_at = now(), updated_by = $7\nWHERE id = $1\nRETURNING *;\n\n-- name: UpdateChatPromptByUUID :one\nUPDATE chat_prompt SET content = $2, token_count = $3, updated_at = now()\nWHERE uuid = $1 and is_deleted = false\nRETURNING *;\n\n-- name: DeleteChatPrompt :exec\nUPDATE chat_prompt \nSET is_deleted = true, updated_at = now()\nWHERE id = $1;\n\n-- name: GetChatPromptsByUserID :many\nSELECT *\nFROM chat_prompt \nWHERE user_id = $1 and is_deleted = false\nORDER BY id;\n\n-- name: GetChatPromptsBysession_uuid :many\nSELECT *\nFROM chat_prompt \nWHERE chat_session_uuid = $1 and is_deleted = false\nORDER BY id;\n\n\n-- name: GetChatPromptsBySessionUUID :many\nSELECT *\nFROM chat_prompt \nWHERE chat_session_uuid = $1 and is_deleted = false\nORDER BY id;\n\n-- name: GetOneChatPromptBySessionUUID :one\nSELECT *\nFROM chat_prompt \nWHERE chat_session_uuid = $1 and is_deleted = false\nORDER BY id\nLIMIT 1;\n\n\n\n\n-- name: HasChatPromptPermission :one\nSELECT COUNT(*) > 0 as has_permission\nFROM chat_prompt cp\nINNER JOIN auth_user au ON cp.user_id = au.id\nWHERE cp.id = $1 AND (cp.user_id = $2 OR au.is_superuser) AND cp.is_deleted = false;\n\n\n-- name: DeleteChatPromptByUUID :exec\nUPDATE chat_prompt\nSET is_deleted = true, updated_at = now()\nWHERE uuid = $1;\n\n\n-- name: GetChatPromptByUUID :one\nSELECT * FROM chat_prompt\nWHERE uuid = $1;" }, { "path": "api/sqlc/queries/chat_session.sql", "content": "-- name: GetAllChatSessions :many\nSELECT * FROM chat_session \nwhere active = true\nORDER BY id;\n\n-- name: CreateChatSession :one\nINSERT INTO chat_session (user_id, topic, max_length, uuid, model)\nVALUES ($1, $2, $3, $4, $5)\nRETURNING *;\n\n-- name: UpdateChatSession :one\nUPDATE chat_session SET user_id = $2, topic = $3, updated_at = now(), active = $4\nWHERE id = $1\nRETURNING *;\n\n-- name: DeleteChatSession :exec\nDELETE FROM chat_session \nWHERE id = $1;\n\n-- name: GetChatSessionByID :one\nSELECT * FROM chat_session WHERE id = $1;\n\n-- name: GetChatSessionByUUID :one\nSELECT * FROM chat_session \nWHERE active = true and uuid = $1\norder by updated_at;\n\n-- name: GetChatSessionByUUIDWithInActive :one\nSELECT * FROM chat_session \nWHERE uuid = $1\norder by updated_at;\n\n-- name: CreateChatSessionByUUID :one\nINSERT INTO chat_session (user_id, uuid, topic, created_at, active, max_length, model)\nVALUES ($1, $2, $3, $4, $5, $6, $7)\nRETURNING *;\n\n-- name: UpdateChatSessionByUUID :one\nUPDATE chat_session SET user_id = $2, topic = $3, updated_at = now()\nWHERE uuid = $1\nRETURNING *;\n\n-- name: CreateOrUpdateChatSessionByUUID :one\nINSERT INTO chat_session(uuid, user_id, topic, max_length, temperature, model, max_tokens, top_p, n, debug, summarize_mode, workspace_id, explore_mode, artifact_enabled)\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14)\nON CONFLICT (uuid) \nDO UPDATE SET\nmax_length = EXCLUDED.max_length, \ndebug = EXCLUDED.debug,\nmax_tokens = EXCLUDED.max_tokens,\ntemperature = EXCLUDED.temperature, \ntop_p = EXCLUDED.top_p,\nn= EXCLUDED.n,\nmodel = EXCLUDED.model,\nsummarize_mode = EXCLUDED.summarize_mode,\nartifact_enabled = EXCLUDED.artifact_enabled,\nworkspace_id = CASE WHEN EXCLUDED.workspace_id IS NOT NULL THEN EXCLUDED.workspace_id ELSE chat_session.workspace_id END,\ntopic = CASE WHEN chat_session.topic IS NULL THEN EXCLUDED.topic ELSE chat_session.topic END,\nexplore_mode = EXCLUDED.explore_mode,\nupdated_at = now()\nreturning *;\n\n-- name: UpdateChatSessionTopicByUUID :one\nINSERT INTO chat_session(uuid, user_id, topic)\nVALUES ($1, $2, $3)\nON CONFLICT (uuid) \nDO UPDATE SET\ntopic = EXCLUDED.topic, \nupdated_at = now()\nreturning *;\n\n-- name: DeleteChatSessionByUUID :exec\nupdate chat_session set active = false\nWHERE uuid = $1\nreturning *;\n\n-- name: GetChatSessionsByUserID :many\nSELECT cs.*\nFROM chat_session cs\nLEFT JOIN (\n SELECT chat_session_uuid, MAX(created_at) AS latest_message_time\n FROM chat_message\n GROUP BY chat_session_uuid\n) cm ON cs.uuid = cm.chat_session_uuid\nWHERE cs.user_id = $1 AND cs.active = true\nORDER BY \n cm.latest_message_time DESC,\n cs.id DESC;\n\n\n-- SELECT cs.*\n-- FROM chat_session cs\n-- WHERE cs.user_id = $1 and cs.active = true\n-- ORDER BY cs.updated_at DESC;\n\n-- name: HasChatSessionPermission :one\nSELECT COUNT(*) > 0 as has_permission\nFROM chat_session cs\nINNER JOIN auth_user au ON cs.user_id = au.id\nWHERE cs.id = $1 AND (cs.user_id = $2 OR au.is_superuser);\n\n\n-- name: UpdateSessionMaxLength :one\nUPDATE chat_session\nSET max_length = $2,\n updated_at = now()\nWHERE uuid = $1\nRETURNING *;\n\n-- name: CreateChatSessionInWorkspace :one\nINSERT INTO chat_session (user_id, uuid, topic, created_at, active, max_length, model, workspace_id)\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8)\nRETURNING *;\n\n-- name: UpdateSessionWorkspace :one\nUPDATE chat_session \nSET workspace_id = $2, updated_at = now()\nWHERE uuid = $1\nRETURNING *;\n\n-- name: GetSessionsByWorkspaceID :many\nSELECT cs.*\nFROM chat_session cs\nLEFT JOIN (\n SELECT chat_session_uuid, MAX(created_at) AS latest_message_time\n FROM chat_message\n GROUP BY chat_session_uuid\n) cm ON cs.uuid = cm.chat_session_uuid\nWHERE cs.workspace_id = $1 AND cs.active = true\nORDER BY \n cm.latest_message_time DESC,\n cs.id DESC;\n\n-- name: GetSessionsGroupedByWorkspace :many\nSELECT \n cs.*,\n w.uuid as workspace_uuid,\n w.name as workspace_name,\n w.color as workspace_color,\n w.icon as workspace_icon\nFROM chat_session cs\nLEFT JOIN chat_workspace w ON cs.workspace_id = w.id\nLEFT JOIN (\n SELECT chat_session_uuid, MAX(created_at) AS latest_message_time\n FROM chat_message\n GROUP BY chat_session_uuid\n) cm ON cs.uuid = cm.chat_session_uuid\nWHERE cs.user_id = $1 AND cs.active = true\nORDER BY \n w.order_position ASC,\n cm.latest_message_time DESC,\n cs.id DESC;\n\n-- name: MigrateSessionsToDefaultWorkspace :exec\nUPDATE chat_session \nSET workspace_id = $2\nWHERE user_id = $1 AND workspace_id IS NULL;\n\n-- name: GetSessionsWithoutWorkspace :many\nSELECT * FROM chat_session \nWHERE user_id = $1 AND workspace_id IS NULL AND active = true;\n" }, { "path": "api/sqlc/queries/chat_snapshot.sql", "content": "-- name: ListChatSnapshots :many\nSELECT * FROM chat_snapshot ORDER BY id;\n\n-- name: ChatSnapshotByID :one\nSELECT * FROM chat_snapshot WHERE id = $1;\n\n-- name: CreateChatSnapshot :one\nINSERT INTO chat_snapshot (uuid, user_id, title, model, summary, tags, conversation ,session, text )\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)\nRETURNING *;\n\n\n-- name: CreateChatBot :one\nINSERT INTO chat_snapshot (uuid, user_id, typ, title, model, summary, tags, conversation ,session, text )\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)\nRETURNING *;\n\n-- name: UpdateChatSnapshot :one\nUPDATE chat_snapshot\nSET uuid = $2, user_id = $3, title = $4, summary = $5, tags = $6, conversation = $7, created_at = $8\nWHERE id = $1\nRETURNING *;\n\n\n-- name: DeleteChatSnapshot :one\nDELETE FROM chat_snapshot WHERE uuid = $1\nand user_id = $2\nRETURNING *;\n\n-- name: ChatSnapshotByUUID :one\nSELECT * FROM chat_snapshot WHERE uuid = $1;\n\n-- name: ChatSnapshotByUserIdAndUuid :one\nSELECT * FROM chat_snapshot WHERE user_id = $1 AND uuid = $2;\n\n-- name: ChatSnapshotMetaByUserID :many\nSELECT uuid, title, summary, tags, created_at, typ\nFROM chat_snapshot WHERE user_id = $1 and typ = $2\norder by created_at desc\nLIMIT $3 OFFSET $4;\n\n-- name: UpdateChatSnapshotMetaByUUID :exec\nUPDATE chat_snapshot\nSET title = $2, summary = $3\nWHERE uuid = $1 and user_id = $4;\n\n-- name: ChatSnapshotCountByUserIDAndType :one\nSELECT COUNT(*)\nFROM chat_snapshot\nWHERE user_id = $1 AND ($2::text = '' OR typ = $2);\n\n-- name: ChatSnapshotSearch :many\nSELECT uuid, title, ts_rank(search_vector, websearch_to_tsquery(@search), 1) as rank\nFROM chat_snapshot\nWHERE search_vector @@ websearch_to_tsquery(@search) AND user_id = $1\nORDER BY rank DESC\nLIMIT 20;" }, { "path": "api/sqlc/queries/chat_workspace.sql", "content": "-- name: CreateWorkspace :one\nINSERT INTO chat_workspace (uuid, user_id, name, description, color, icon, is_default, order_position)\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8)\nRETURNING *;\n\n-- name: GetWorkspaceByUUID :one\nSELECT * FROM chat_workspace \nWHERE uuid = $1;\n\n-- name: GetWorkspacesByUserID :many\nSELECT * FROM chat_workspace \nWHERE user_id = $1\nORDER BY order_position ASC, created_at ASC;\n\n-- name: UpdateWorkspace :one\nUPDATE chat_workspace \nSET name = $2, description = $3, color = $4, icon = $5, updated_at = now()\nWHERE uuid = $1\nRETURNING *;\n\n-- name: UpdateWorkspaceOrder :one\nUPDATE chat_workspace \nSET order_position = $2, updated_at = now()\nWHERE uuid = $1\nRETURNING *;\n\n-- name: DeleteWorkspace :exec\nDELETE FROM chat_workspace \nWHERE uuid = $1;\n\n-- name: GetDefaultWorkspaceByUserID :one\nSELECT * FROM chat_workspace \nWHERE user_id = $1 AND is_default = true\nLIMIT 1;\n\n-- name: SetDefaultWorkspace :one\nUPDATE chat_workspace \nSET is_default = $2, updated_at = now()\nWHERE uuid = $1\nRETURNING *;\n\n-- name: GetWorkspaceWithSessionCount :many\nSELECT \n w.*,\n COUNT(cs.id) as session_count\nFROM chat_workspace w\nLEFT JOIN chat_session cs ON w.id = cs.workspace_id AND cs.active = true\nWHERE w.user_id = $1\nGROUP BY w.id\nORDER BY w.order_position ASC, w.created_at ASC;\n\n-- name: CreateDefaultWorkspace :one\nINSERT INTO chat_workspace (uuid, user_id, name, description, color, icon, is_default, order_position)\nVALUES ($1, $2, 'General', 'Default workspace for all conversations', '#6366f1', 'folder', true, 0)\nRETURNING *;\n\n-- name: HasWorkspacePermission :one\nSELECT COUNT(*) > 0 as has_permission\nFROM chat_workspace w\nWHERE w.uuid = $1\n AND (\n w.user_id = $2\n OR EXISTS (\n SELECT 1\n FROM auth_user request_user\n WHERE request_user.id = $2 AND request_user.is_superuser = true\n )\n );\n" }, { "path": "api/sqlc/queries/jwt_secrets.sql", "content": "-- name: CreateJwtSecret :one\nINSERT INTO jwt_secrets (name, secret, audience)\nVALUES ($1, $2, $3) RETURNING *;\n\n-- name: GetJwtSecret :one\nSELECT * FROM jwt_secrets WHERE name = $1;\n\n\n-- name: DeleteAllJwtSecrets :execrows\nDELETE FROM jwt_secrets;" }, { "path": "api/sqlc/queries/user_active_chat_session.sql", "content": "-- Simplified unified queries for active sessions\n\n-- name: UpsertUserActiveSession :one\nINSERT INTO user_active_chat_session (user_id, workspace_id, chat_session_uuid)\nVALUES ($1, $2, $3)\nON CONFLICT (user_id, COALESCE(workspace_id, -1))\nDO UPDATE SET \n chat_session_uuid = EXCLUDED.chat_session_uuid,\n updated_at = now()\nRETURNING *;\n\n-- name: GetUserActiveSession :one\nSELECT * FROM user_active_chat_session \nWHERE user_id = $1 AND (\n (workspace_id IS NULL AND $2::int IS NULL) OR \n (workspace_id = $2)\n);\n\n-- name: GetAllUserActiveSessions :many\nSELECT * FROM user_active_chat_session\nWHERE user_id = $1\nORDER BY workspace_id NULLS FIRST, updated_at DESC;\n\n-- name: DeleteUserActiveSession :exec\nDELETE FROM user_active_chat_session\nWHERE user_id = $1 AND (\n (workspace_id IS NULL AND $2::int IS NULL) OR \n (workspace_id = $2)\n);\n\n-- name: DeleteUserActiveSessionBySession :exec\nDELETE FROM user_active_chat_session\nWHERE user_id = $1 AND chat_session_uuid = $2;\n\n" }, { "path": "api/sqlc/queries/user_chat_model_privilege.sql", "content": "-- name: ListUserChatModelPrivileges :many\nSELECT * FROM user_chat_model_privilege ORDER BY id;\n\n-- name: ListUserChatModelPrivilegesRateLimit :many\nSELECT ucmp.id, au.email as user_email, CONCAT_WS('',au.last_name, au.first_name) as full_name, cm.name chat_model_name, ucmp.rate_limit \nFROM user_chat_model_privilege ucmp \nINNER JOIN chat_model cm ON cm.id = ucmp.chat_model_id\nINNER JOIN auth_user au ON au.id = ucmp.user_id\nORDER by au.last_login DESC;\n-- TODO add ratelimit\n-- LIMIT 1000\n\n-- name: ListUserChatModelPrivilegesByUserID :many\nSELECT * FROM user_chat_model_privilege \nWHERE user_id = $1\nORDER BY id;\n\n-- name: UserChatModelPrivilegeByID :one\nSELECT * FROM user_chat_model_privilege WHERE id = $1;\n\n-- name: CreateUserChatModelPrivilege :one\nINSERT INTO user_chat_model_privilege (user_id, chat_model_id, rate_limit, created_by, updated_by)\nVALUES ($1, $2, $3, $4, $5)\nRETURNING *;\n\n-- name: UpdateUserChatModelPrivilege :one\nUPDATE user_chat_model_privilege SET rate_limit = $2, updated_at = now(), updated_by = $3\nWHERE id = $1\nRETURNING *;\n\n-- name: DeleteUserChatModelPrivilege :exec\nDELETE FROM user_chat_model_privilege WHERE id = $1;\n\n-- name: UserChatModelPrivilegeByUserAndModelID :one\nSELECT * FROM user_chat_model_privilege WHERE user_id = $1 AND chat_model_id = $2;\n\n-- name: RateLimiteByUserAndSessionUUID :one\nSELECT ucmp.rate_limit, cm.name AS chat_model_name\nFROM user_chat_model_privilege ucmp\nJOIN chat_session cs ON cs.user_id = ucmp.user_id\nJOIN chat_model cm ON (cm.id = ucmp.chat_model_id AND cs.model = cm.name and cm.enable_per_mode_ratelimit = true)\nWHERE cs.uuid = $1\n AND ucmp.user_id = $2;\n -- AND cs.model = cm.name \n" }, { "path": "api/sqlc/schema.sql", "content": "CREATE TABLE IF NOT EXISTS jwt_secrets (\n id SERIAL PRIMARY KEY,\n name TEXT NOT NULL,\n secret TEXT NOT NULL,\n audience TEXT NOT NULL,\n lifetime smallint NOT NULL default 24\n);\n\nALTER TABLE jwt_secrets ADD COLUMN IF NOT EXISTS lifetime smallint NOT NULL default 24;\n\nUPDATE jwt_secrets SET lifetime = 240;\n\nCREATE TABLE IF NOT EXISTS chat_model (\n id SERIAL PRIMARY KEY, \n -- model name 'claude-v1', 'gpt-3.5-turbo'\n name TEXT UNIQUE DEFAULT '' NOT NULL, \n -- model label 'Claude', 'GPT-3.5 Turbo'\n label TEXT DEFAULT '' NOT NULL, \n is_default BOOLEAN DEFAULT false NOT NULL,\n url TEXT DEFAULT '' NOT NULL, \n api_auth_header TEXT DEFAULT '' NOT NULL, \n -- env var that contains the api key\n -- for example: OPENAI_API_KEY, which means the api key is stored in an env var called OPENAI_API_KEY\n api_auth_key TEXT DEFAULT '' NOT NULL,\n user_id INTEGER NOT NULL default 1,\n enable_per_mode_ratelimit BOOLEAN DEFAULT false NOT NULL,\n max_token INTEGER NOT NULL default 120,\n default_token INTEGER NOT NULL default 120,\n order_number INTEGER NOT NULL default 1,\n http_time_out INTEGER NOT NULL default 120\n);\n\nALTER TABLE chat_model ADD COLUMN IF NOT EXISTS user_id INTEGER NOT NULL default 1;\nALTER TABLE chat_model ADD COLUMN IF NOT EXISTS enable_per_mode_ratelimit BOOLEAN DEFAULT false NOT NULL;\nALTER TABLE chat_model ADD COLUMN IF NOT EXISTS max_token INTEGER NOT NULL default 4096;\nALTER TABLE chat_model ADD COLUMN IF NOT EXISTS default_token INTEGER NOT NULL default 2048;\nALTER TABLE chat_model ADD COLUMN IF NOT EXISTS order_number INTEGER NOT NULL default 1;\nALTER TABLE chat_model ADD COLUMN IF NOT EXISTS http_time_out INTEGER NOT NULL default 120;\nALTER TABLE chat_model ADD COLUMN IF NOT EXISTS is_enable BOOLEAN DEFAULT true NOT NULL;\nALTER TABLE chat_model ADD COLUMN IF NOT EXISTS api_type VARCHAR(50) NOT NULL DEFAULT 'openai';\n\n\n\nINSERT INTO chat_model(name, label, is_default, url, api_auth_header, api_auth_key, max_token, default_token, order_number)\nVALUES ('gpt-3.5-turbo', 'gpt-3.5-turbo(chatgpt)', true, 'https://api.openai.com/v1/chat/completions', 'Authorization', 'OPENAI_API_KEY', 4096, 4096, 0),\n ('gemini-2.0-flash', 'gemini-2.0-flash', false, 'https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash', 'Authorization', 'GEMINI_API_KEY', 4096, 4096, 0),\n ('gpt-3.5-turbo-16k', 'gpt-3.5-16k', false, 'https://api.openai.com/v1/chat/completions', 'Authorization', 'OPENAI_API_KEY', 16384, 8192, 2),\n ('claude-3-7-sonnet-20250219', 'claude-3-7-sonnet-20250219', false, 'https://api.anthropic.com/v1/messages', 'x-api-key', 'CLAUDE_API_KEY', 4096, 4096, 4),\n ('gpt-4', 'gpt-4(chatgpt)', false, 'https://api.openai.com/v1/chat/completions', 'Authorization', 'OPENAI_API_KEY', 9192, 4096, 5),\n ('deepseek-chat', 'deepseek-chat', false, 'https://api.deepseek.com/v1/chat/completions', 'Authorization', 'DEEPSEEK_API_KEY', 8192, 8192, 0),\n ('gpt-4-32k', 'gpt-4-32k(chatgpt)', false, 'https://api.openai.com/v1/chat/completions', 'Authorization', 'OPENAI_API_KEY', 9192, 2048, 6),\n ('text-davinci-003', 'text-davinci-003', false, 'https://api.openai.com/v1/completions', 'Authorization', 'OPENAI_API_KEY', 4096, 2048, 7),\n ('echo','echo',false,'https://bestqa_workerd.bestqa.workers.dev/echo','Authorization','ECHO_API_KEY', 40960, 20480, 8),\n ('debug','debug',false,'https://bestqa_workerd.bestqa.workers.dev/debug','Authorization','ECHO_API_KEY', 40960, 2048, 9),\n ('deepseek-reasoner','deepseek-reasoner',false,'https://api.deepseek.com/v1/chat/completions','Authorization','DEEPSEEK API KEY', 8192, 8192, 2)\nON CONFLICT(name) DO NOTHING;\n\nUPDATE chat_model SET enable_per_mode_ratelimit = true WHERE name = 'gpt-4';\nUPDATE chat_model SET enable_per_mode_ratelimit = true WHERE name = 'gpt-4-32k';\nDELETE FROM chat_model where name = 'claude-v1';\nDELETE FROM chat_model where name = 'claude-v1-100k';\nDELETE FROM chat_model where name = 'claude-instant-v1';\n\n-- Update existing records with appropriate api_type values\nUPDATE chat_model SET api_type = 'openai' WHERE name LIKE 'gpt-%' OR name LIKE 'text-davinci-%' OR name LIKE 'deepseek-%';\nUPDATE chat_model SET api_type = 'claude' WHERE name LIKE 'claude-%';\nUPDATE chat_model SET api_type = 'gemini' WHERE name LIKE 'gemini-%';\nUPDATE chat_model SET api_type = 'ollama' WHERE name LIKE 'ollama-%';\nUPDATE chat_model SET api_type = 'custom' WHERE name LIKE 'custom-%' OR name IN ('echo', 'debug');\n-- create index on name\nCREATE INDEX IF NOT EXISTS jwt_secrets_name_idx ON jwt_secrets (name);\n\n\nCREATE TABLE IF NOT EXISTS auth_user (\n id SERIAL PRIMARY KEY,\n password VARCHAR(128) NOT NULL,\n last_login TIMESTAMP default now() NOT NULL,\n is_superuser BOOLEAN default false NOT NULL,\n username VARCHAR(150) UNIQUE NOT NULL,\n first_name VARCHAR(30) default '' NOT NULL,\n last_name VARCHAR(30) default '' NOT NULL,\n email VARCHAR(254) UNIQUE NOT NULL,\n is_staff BOOLEAN default false NOT NULL,\n is_active BOOLEAN default true NOT NULL,\n date_joined TIMESTAMP default now() NOT NULL\n);\n\n-- add index on email\nCREATE INDEX IF NOT EXISTS auth_user_email_idx ON auth_user (email);\n\nCREATE TABLE IF NOT EXISTS auth_user_management (\n id SERIAL PRIMARY KEY,\n user_id INTEGER UNIQUE NOT NULL REFERENCES auth_user(id) ON DELETE CASCADE,\n rate_limit INTEGER NOT NULL,\n created_at TIMESTAMP DEFAULT NOW() NOT NULL,\n updated_at TIMESTAMP DEFAULT NOW() NOT NULL\n);\n\n-- add index on user_id\nCREATE INDEX IF NOT EXISTS auth_user_management_user_id_idx ON auth_user_management (user_id);\n\n\n-- control specific model ratelimit, like gpt4\n-- if not find gpt4 on privilege than forbiden\n-- if found, then check the acess count (session messages).\n-- get rate_limit by user_id, chat_session_uuid\nCREATE TABLE IF NOT EXISTS user_chat_model_privilege(\n id SERIAL PRIMARY KEY,\n user_id INTEGER NOT NULL REFERENCES auth_user(id) ON DELETE CASCADE,\n chat_model_id INT NOT NULL REFERENCES chat_model(id) ON DELETE CASCADE,\n rate_limit INTEGER NOT NULL,\n created_at TIMESTAMP DEFAULT NOW() NOT NULL,\n updated_at TIMESTAMP DEFAULT NOW() NOT NULL,\n created_by INTEGER NOT NULL DEFAULT 0,\n updated_by INTEGER NOT NULL DEFAULT 0, \n CONSTRAINT chat_usage_user_model_unique UNIQUE (user_id, chat_model_id)\n);\n\nCREATE TABLE IF NOT EXISTS chat_workspace (\n id SERIAL PRIMARY KEY,\n uuid VARCHAR(255) UNIQUE NOT NULL,\n user_id INTEGER NOT NULL REFERENCES auth_user(id) ON DELETE CASCADE,\n name VARCHAR(255) NOT NULL,\n description TEXT NOT NULL DEFAULT '',\n color VARCHAR(7) NOT NULL DEFAULT '#6366f1',\n icon VARCHAR(50) NOT NULL DEFAULT 'folder',\n created_at TIMESTAMP DEFAULT now() NOT NULL,\n updated_at TIMESTAMP DEFAULT now() NOT NULL,\n is_default BOOLEAN DEFAULT false NOT NULL,\n order_position INTEGER DEFAULT 0 NOT NULL\n);\n\n-- add index on user_id for workspace\nCREATE INDEX IF NOT EXISTS chat_workspace_user_id_idx ON chat_workspace (user_id);\n\n-- add index on uuid for workspace\nCREATE INDEX IF NOT EXISTS chat_workspace_uuid_idx ON chat_workspace using hash (uuid);\n\n-- Keep exactly one default workspace per user.\n-- If duplicates exist in old data, keep the newest default and unset the rest.\nWITH ranked_default_workspaces AS (\n SELECT\n id,\n ROW_NUMBER() OVER (PARTITION BY user_id ORDER BY updated_at DESC, id DESC) AS row_num\n FROM chat_workspace\n WHERE is_default = true\n)\nUPDATE chat_workspace cw\nSET is_default = false, updated_at = now()\nFROM ranked_default_workspaces rdw\nWHERE cw.id = rdw.id AND rdw.row_num > 1;\n\nCREATE UNIQUE INDEX IF NOT EXISTS chat_workspace_single_default_per_user_idx\n ON chat_workspace (user_id)\n WHERE is_default = true;\n\nCREATE TABLE IF NOT EXISTS chat_session (\n id SERIAL PRIMARY KEY,\n user_id integer NOT NULL,\n --ALTER TABLE chat_session ADD COLUMN uuid character varying(255) NOT NULL DEFAULT '';\n uuid character varying(255) UNIQUE NOT NULL,\n topic character varying(255) NOT NULL,\n created_at timestamp DEFAULT now() NOT NULL,\n updated_at timestamp DEFAULT now() NOT NULL,\n active boolean default true NOT NULL,\n model character varying(255) NOT NULL DEFAULT 'gpt-3.5-turbo',\n max_length integer DEFAULT 0 NOT NULL,\n temperature float DEFAULT 1.0 NOT NUll,\n top_p float DEFAULT 1.0 NOT NUll,\n max_tokens int DEFAULT 4096 NOT NULL,\n n integer DEFAULT 1 NOT NULL,\n summarize_mode boolean DEFAULT false NOT NULL,\n workspace_id INTEGER REFERENCES chat_workspace(id) ON DELETE SET NULL,\n artifact_enabled boolean DEFAULT false NOT NULL\n);\n\n\n-- chat_session\nALTER TABLE chat_session DROP COLUMN IF EXISTS code_runner_enabled;\nALTER TABLE chat_session ADD COLUMN IF NOT EXISTS temperature float DEFAULT 1.0 NOT NULL;\nALTER TABLE chat_session ADD COLUMN IF NOT EXISTS top_p float DEFAULT 1.0 NOT NULL;\nALTER TABLE chat_session ADD COLUMN IF NOT EXISTS max_tokens int DEFAULT 4096 NOT NULL; \nALTER TABLE chat_session ADD COLUMN IF NOT EXISTS debug boolean DEFAULT false NOT NULL;\nALTER TABLE chat_session ADD COLUMN IF NOT EXISTS explore_mode boolean DEFAULT false NOT NULL; \nALTER TABlE chat_session ADD COLUMN IF NOT EXISTS model character varying(255) NOT NULL DEFAULT 'gpt-3.5-turbo';\nALTER TABLE chat_session ADD COLUMN IF NOT EXISTS n INTEGER DEFAULT 1 NOT NULL;\nALTER TABLE chat_session ADD COLUMN IF NOT EXISTS summarize_mode boolean DEFAULT false NOT NULL;\nALTER TABLE chat_session ADD COLUMN IF NOT EXISTS workspace_id INTEGER REFERENCES chat_workspace(id) ON DELETE SET NULL;\nALTER TABLE chat_session ADD COLUMN IF NOT EXISTS artifact_enabled boolean DEFAULT false NOT NULL;\n\n\n-- add hash index on uuid\nCREATE INDEX IF NOT EXISTS chat_session_uuid_idx ON chat_session using hash (uuid) ;\n\n-- add index on user_id\nCREATE INDEX IF NOT EXISTS chat_session_user_id_idx ON chat_session (user_id);\n\n-- add index on workspace_id\nCREATE INDEX IF NOT EXISTS chat_session_workspace_id_idx ON chat_session (workspace_id);\n\nCREATE TABLE IF NOT EXISTS chat_message (\n id SERIAL PRIMARY KEY,\n --ALTER TABLE chat_message ADD COLUMN uuid character varying(255) NOT NULL DEFAULT '';\n uuid character varying(255) NOT NULL,\n chat_session_uuid character varying(255) NOT NUll,\n role character varying(255) NOT NULL,\n content character varying NOT NULL,\n reasoning_content character varying NOT NULL,\n model character varying(255) NOT NULL DEFAULT '',\n llm_summary character varying(1024) NOT NULL DEFAULT '',\n score double precision NOT NULL,\n user_id integer NOT NULL,\n created_at timestamp DEFAULT now() NOT NULL,\n updated_at timestamp DEFAULT now() Not NULL,\n created_by integer NOT NULL,\n updated_by integer NOT NULL,\n is_deleted BOOLEAN NOT NULL DEFAULT false,\n is_pin BOOLEAN NOT NULL DEFAULT false,\n token_count INTEGER DEFAULT 0 NOT NULL,\n raw jsonb default '{}' NOT NULL\n);\n\n-- chat_messages\nALTER TABLE chat_message ADD COLUMN IF NOT EXISTS is_deleted BOOLEAN NOT NULL DEFAULT false;\nALTER TABLE chat_message ADD COLUMN IF NOT EXISTS token_count INTEGER DEFAULT 0 NOT NULL;\nALTER TABLE chat_message ADD COLUMN IF NOT EXISTS is_pin BOOLEAN NOT NULL DEFAULT false;\nALTER TABLE chat_message ADD COLUMN IF NOT EXISTS llm_summary character varying(1024) NOT NULL DEFAULT '';\nALTER TABLE chat_message ADD COLUMN IF NOT EXISTS model character varying(255) NOT NULL DEFAULT '';\nALTER TABLE chat_message ADD COLUMN IF NOT EXISTS reasoning_content character varying NOT NULL DEFAULT '';\nALTER TABLE chat_message ADD COLUMN IF NOT EXISTS artifacts JSONB DEFAULT '[]' NOT NULL;\nALTER TABLE chat_message ADD COLUMN IF NOT EXISTS suggested_questions JSONB DEFAULT '[]' NOT NULL;\n\n-- add hash index on uuid\nCREATE INDEX IF NOT EXISTS chat_message_uuid_idx ON chat_message using hash (uuid) ;\n\n-- add index on chat_session_uuid\nCREATE INDEX IF NOT EXISTS chat_message_chat_session_uuid_idx ON chat_message (chat_session_uuid);\n\n-- add index on user_id\nCREATE INDEX IF NOT EXISTS chat_message_user_id_idx ON chat_message (user_id);\n\n-- add brin index on created_at\nCREATE INDEX IF NOT EXISTS chat_message_created_at_idx ON chat_message using brin (created_at) ;\n\nCREATE TABLE IF NOT EXISTS chat_prompt (\n id SERIAL PRIMARY KEY,\n uuid character varying(255) NOT NULL,\n chat_session_uuid character varying(255) NOT NULL, -- store the session_uuid\n role character varying(255) NOT NULL,\n content character varying NOT NULL,\n score double precision default 0 NOT NULL,\n user_id integer default 0 NOT NULL,\n created_at timestamp DEFAULT now() NOT NULL ,\n updated_at timestamp DEFAULT now() NOT NULL,\n created_by integer NOT NULL,\n updated_by integer NOT NULL,\n is_deleted BOOLEAN NOT NULL DEFAULT false,\n token_count INTEGER DEFAULT 0 NOT NULL\n -- raw jsonb default '{}' NOT NULL\n);\n\nALTER TABLE chat_prompt ADD COLUMN IF NOT EXISTS is_deleted BOOLEAN NOT NULL DEFAULT false;\nALTER TABLE chat_prompt ADD COLUMN IF NOT EXISTS token_count INTEGER DEFAULT 0 NOT NULL;\n\n-- add hash index on uuid\nCREATE INDEX IF NOT EXISTS chat_prompt_uuid_idx ON chat_prompt using hash (uuid) ;\n\n-- add index on chat_session_uuid\nCREATE INDEX IF NOT EXISTS chat_prompt_chat_session_uuid_idx ON chat_prompt (chat_session_uuid);\n\n-- add index on user_id\nCREATE INDEX IF NOT EXISTS chat_prompt_user_id_idx ON chat_prompt (user_id);\n\n-- Keep only one active system prompt per session.\n-- If historical duplicates exist, soft-delete newer duplicates first so\n-- unique index creation succeeds in existing databases.\nWITH duplicate_system_prompts AS (\n SELECT\n id,\n ROW_NUMBER() OVER (PARTITION BY chat_session_uuid ORDER BY id ASC) AS row_num\n FROM chat_prompt\n WHERE role = 'system' AND is_deleted = false\n)\nUPDATE chat_prompt cp\nSET is_deleted = true, updated_at = now()\nFROM duplicate_system_prompts dsp\nWHERE cp.id = dsp.id AND dsp.row_num > 1;\n\nCREATE UNIQUE INDEX IF NOT EXISTS chat_prompt_unique_active_system_per_session_idx\nON chat_prompt (chat_session_uuid)\nWHERE role = 'system' AND is_deleted = false;\n\nCREATE TABLE IF NOT EXISTS chat_logs (\n\tid SERIAL PRIMARY KEY, -- Auto-incrementing ID as primary key\n\tsession JSONB default '{}' NOT NULL, -- JSONB column to store chat session info\n\tquestion JSONB default '{}' NOT NULL, -- JSONB column to store the question\n\tanswer JSONB default '{}' NOT NULL, -- JSONB column to store the answer \n created_at timestamp DEFAULT now() NOT NULL \n);\n\n-- add brin index on created_at\nCREATE INDEX IF NOT EXISTS chat_logs_created_at_idx ON chat_logs using brin (created_at) ;\n\n\n-- user_id is the user who created the session\n-- uuid is the session uuid\nCREATE TABLE IF NOT EXISTS user_active_chat_session (\n id SERIAL PRIMARY KEY,\n user_id integer UNIQUE default '0' NOT NULL ,\n chat_session_uuid character varying(255) NOT NULL,\n created_at timestamp DEFAULT now() NOT NULL,\n updated_at timestamp DEFAULT now() NOT NULL\n);\n\n-- add index on user_id\nCREATE INDEX IF NOT EXISTS user_active_chat_session_user_id_idx ON user_active_chat_session using hash (user_id) ;\n\n-- Extend user_active_chat_session to support per-workspace active sessions\nALTER TABLE user_active_chat_session ADD COLUMN IF NOT EXISTS workspace_id INTEGER REFERENCES chat_workspace(id) ON DELETE CASCADE;\n\n-- Clean up old constraints\nALTER TABLE user_active_chat_session DROP CONSTRAINT IF EXISTS user_active_chat_session_user_id_key;\nALTER TABLE user_active_chat_session DROP CONSTRAINT IF EXISTS unique_user_id;\n\n-- Clean up duplicate records first\n-- Keep only the most recent record per user/workspace combination\nDELETE FROM user_active_chat_session \nWHERE id NOT IN (\n SELECT DISTINCT ON (user_id, COALESCE(workspace_id, -1)) id \n FROM user_active_chat_session \n ORDER BY user_id, COALESCE(workspace_id, -1), updated_at DESC\n);\n\n-- Create a single unique constraint using COALESCE to handle NULLs\n-- This treats NULL workspace_id as -1 for uniqueness purposes\nDROP INDEX IF EXISTS unique_user_global_active_session_idx;\nDROP INDEX IF EXISTS unique_user_workspace_active_session_idx;\nDROP INDEX IF EXISTS unique_user_workspace_active_session_coalesce_idx;\n\nCREATE UNIQUE INDEX unique_user_workspace_active_session_coalesce_idx \n ON user_active_chat_session (user_id, COALESCE(workspace_id, -1));\n\n-- Add index on workspace_id for efficient queries\nCREATE INDEX IF NOT EXISTS user_active_chat_session_workspace_id_idx ON user_active_chat_session (workspace_id);\n\n\n-- for share chat feature\nCREATE TABLE IF NOT EXISTS chat_snapshot (\n id SERIAL PRIMARY KEY,\n typ VARCHAR(255) NOT NULL default 'snapshot',\n uuid VARCHAR(255) NOT NULL default '',\n user_id INTEGER NOT NULL default 0,\n title VARCHAR(255) NOT NULL default '',\n summary TEXT NOT NULL default '',\n model VARCHAR(255) NOT NULL default '',\n tags JSONB DEFAULT '{}' NOT NULL,\n session JSONB DEFAULT '{}' NOT NULL,\n conversation JSONB DEFAULT '{}' NOT NULL,\n created_at TIMESTAMP DEFAULT now() NOT NULL,\n text text DEFAULT '' NOT NULL,\n search_vector tsvector generated always as (setweight(to_tsvector('simple', coalesce(title, '')), 'A') || ' ' || setweight(to_tsvector('simple', coalesce(text, '')), 'B') :: tsvector) stored\n);\n\nALTER TABLE chat_snapshot ADD COLUMN IF NOT EXISTS typ VARCHAR(255) NOT NULL default 'snapshot' ;\nALTER TABLE chat_snapshot ADD COLUMN IF NOT EXISTS model VARCHAR(255) NOT NULL default '' ;\nALTER TABLE chat_snapshot ADD COLUMN IF NOT EXISTS session JSONB DEFAULT '{}' NOT NULL;\nALTER TABLE chat_snapshot ADD COLUMN IF NOT EXISTS text text DEFAULT '' NOT NULL;\nALTER TABLE chat_snapshot ADD COLUMN IF NOT EXISTS search_vector tsvector generated always as (\n\tsetweight(to_tsvector('simple', coalesce(title, '')), 'A') || ' ' || setweight(to_tsvector('simple', coalesce(text, '')), 'B') :: tsvector\n) stored; \n\nCREATE INDEX IF NOT EXISTS search_vector_gin_idx on chat_snapshot using GIN(search_vector);\n\n-- add index on user id\nCREATE INDEX IF NOT EXISTS chat_snapshot_user_id_idx ON chat_snapshot (user_id);\n\n-- add index on created_at(brin)\nCREATE INDEX IF NOT EXISTS chat_snapshot_created_at_idx ON chat_snapshot using brin (created_at) ;\n\nUPDATE chat_snapshot SET model = 'gpt-3.5-turbo' WHERE model = '';\n\n\nCREATE TABLE IF NOT EXISTS chat_file (\n id SERIAL PRIMARY KEY,\n name VARCHAR(255) NOT NULL,\n data BYTEA NOT NULL,\n created_at TIMESTAMP DEFAULT now() NOT NULL,\n user_id INTEGER NOT NULL default 1,\n -- foreign key chat_session_uuid\n chat_session_uuid VARCHAR(255) NOT NULL REFERENCES chat_session(uuid) ON DELETE CASCADE,\n mime_type VARCHAR(255) NOT NULL\n);\n\nCREATE TABLE IF NOT EXISTS bot_answer_history (\n id SERIAL PRIMARY KEY,\n bot_uuid VARCHAR(255) NOT NULL,\n user_id INTEGER NOT NULL REFERENCES auth_user(id) ON DELETE CASCADE,\n prompt TEXT NOT NULL,\n answer TEXT NOT NULL,\n model VARCHAR(255) NOT NULL,\n tokens_used INTEGER NOT NULL DEFAULT 0,\n created_at TIMESTAMP DEFAULT now() NOT NULL,\n updated_at TIMESTAMP DEFAULT now() NOT NULL\n);\n\n-- Indexes for faster queries\nCREATE INDEX IF NOT EXISTS bot_answer_history_bot_uuid_idx ON bot_answer_history (bot_uuid);\nCREATE INDEX IF NOT EXISTS bot_answer_history_user_id_idx ON bot_answer_history (user_id);\nCREATE INDEX IF NOT EXISTS bot_answer_history_created_at_idx ON bot_answer_history USING BRIN (created_at);\n\nCREATE TABLE IF NOT EXISTS chat_comment (\n id SERIAL PRIMARY KEY,\n uuid VARCHAR(255) NOT NULL,\n chat_session_uuid VARCHAR(255) NOT NULL,\n chat_message_uuid VARCHAR(255) NOT NULL,\n content TEXT NOT NULL,\n created_at TIMESTAMP DEFAULT now() NOT NULL,\n updated_at TIMESTAMP DEFAULT now() NOT NULL,\n created_by INTEGER NOT NULL REFERENCES auth_user(id) ON DELETE CASCADE,\n updated_by INTEGER NOT NULL REFERENCES auth_user(id) ON DELETE CASCADE\n);\n\n-- Add indexes for faster lookups\nCREATE INDEX IF NOT EXISTS chat_comment_chat_session_uuid_idx ON chat_comment (chat_session_uuid);\nCREATE INDEX IF NOT EXISTS chat_comment_created_by_idx ON chat_comment (created_by);\n" }, { "path": "api/sqlc.yaml", "content": "version: \"2\"\nsql:\n - engine: \"postgresql\"\n queries: \"sqlc/queries/\"\n schema: \"sqlc/schema.sql\"\n gen:\n go:\n package: \"sqlc_queries\"\n out: \"sqlc_queries\"\n emit_json_tags: true\n json_tags_case_style: \"camel\"\n" }, { "path": "api/sqlc_queries/auth_user.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: auth_user.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n\t\"time\"\n)\n\nconst createAuthUser = `-- name: CreateAuthUser :one\nINSERT INTO auth_user (email, \"password\", first_name, last_name, username, is_staff, is_superuser)\nVALUES ($1, $2, $3, $4, $5, $6, $7)\nRETURNING id, password, last_login, is_superuser, username, first_name, last_name, email, is_staff, is_active, date_joined\n`\n\ntype CreateAuthUserParams struct {\n\tEmail string `json:\"email\"`\n\tPassword string `json:\"password\"`\n\tFirstName string `json:\"firstName\"`\n\tLastName string `json:\"lastName\"`\n\tUsername string `json:\"username\"`\n\tIsStaff bool `json:\"isStaff\"`\n\tIsSuperuser bool `json:\"isSuperuser\"`\n}\n\nfunc (q *Queries) CreateAuthUser(ctx context.Context, arg CreateAuthUserParams) (AuthUser, error) {\n\trow := q.db.QueryRowContext(ctx, createAuthUser,\n\t\targ.Email,\n\t\targ.Password,\n\t\targ.FirstName,\n\t\targ.LastName,\n\t\targ.Username,\n\t\targ.IsStaff,\n\t\targ.IsSuperuser,\n\t)\n\tvar i AuthUser\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Password,\n\t\t&i.LastLogin,\n\t\t&i.IsSuperuser,\n\t\t&i.Username,\n\t\t&i.FirstName,\n\t\t&i.LastName,\n\t\t&i.Email,\n\t\t&i.IsStaff,\n\t\t&i.IsActive,\n\t\t&i.DateJoined,\n\t)\n\treturn i, err\n}\n\nconst deleteAuthUser = `-- name: DeleteAuthUser :exec\nDELETE FROM auth_user WHERE email = $1\n`\n\nfunc (q *Queries) DeleteAuthUser(ctx context.Context, email string) error {\n\t_, err := q.db.ExecContext(ctx, deleteAuthUser, email)\n\treturn err\n}\n\nconst getAllAuthUsers = `-- name: GetAllAuthUsers :many\nSELECT id, password, last_login, is_superuser, username, first_name, last_name, email, is_staff, is_active, date_joined FROM auth_user ORDER BY id\n`\n\nfunc (q *Queries) GetAllAuthUsers(ctx context.Context) ([]AuthUser, error) {\n\trows, err := q.db.QueryContext(ctx, getAllAuthUsers)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []AuthUser\n\tfor rows.Next() {\n\t\tvar i AuthUser\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Password,\n\t\t\t&i.LastLogin,\n\t\t\t&i.IsSuperuser,\n\t\t\t&i.Username,\n\t\t\t&i.FirstName,\n\t\t\t&i.LastName,\n\t\t\t&i.Email,\n\t\t\t&i.IsStaff,\n\t\t\t&i.IsActive,\n\t\t\t&i.DateJoined,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getAuthUserByEmail = `-- name: GetAuthUserByEmail :one\nSELECT id, password, last_login, is_superuser, username, first_name, last_name, email, is_staff, is_active, date_joined FROM auth_user WHERE email = $1\n`\n\nfunc (q *Queries) GetAuthUserByEmail(ctx context.Context, email string) (AuthUser, error) {\n\trow := q.db.QueryRowContext(ctx, getAuthUserByEmail, email)\n\tvar i AuthUser\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Password,\n\t\t&i.LastLogin,\n\t\t&i.IsSuperuser,\n\t\t&i.Username,\n\t\t&i.FirstName,\n\t\t&i.LastName,\n\t\t&i.Email,\n\t\t&i.IsStaff,\n\t\t&i.IsActive,\n\t\t&i.DateJoined,\n\t)\n\treturn i, err\n}\n\nconst getAuthUserByID = `-- name: GetAuthUserByID :one\nSELECT id, password, last_login, is_superuser, username, first_name, last_name, email, is_staff, is_active, date_joined FROM auth_user WHERE id = $1\n`\n\nfunc (q *Queries) GetAuthUserByID(ctx context.Context, id int32) (AuthUser, error) {\n\trow := q.db.QueryRowContext(ctx, getAuthUserByID, id)\n\tvar i AuthUser\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Password,\n\t\t&i.LastLogin,\n\t\t&i.IsSuperuser,\n\t\t&i.Username,\n\t\t&i.FirstName,\n\t\t&i.LastName,\n\t\t&i.Email,\n\t\t&i.IsStaff,\n\t\t&i.IsActive,\n\t\t&i.DateJoined,\n\t)\n\treturn i, err\n}\n\nconst getTotalActiveUserCount = `-- name: GetTotalActiveUserCount :one\nSELECT COUNT(*) FROM auth_user WHERE is_active = true\n`\n\nfunc (q *Queries) GetTotalActiveUserCount(ctx context.Context) (int64, error) {\n\trow := q.db.QueryRowContext(ctx, getTotalActiveUserCount)\n\tvar count int64\n\terr := row.Scan(&count)\n\treturn count, err\n}\n\nconst getUserAnalysisByEmail = `-- name: GetUserAnalysisByEmail :one\nSELECT \n auth_user.first_name,\n auth_user.last_name,\n auth_user.email AS user_email,\n COALESCE(user_stats.total_messages, 0) AS total_messages,\n COALESCE(user_stats.total_token_count, 0) AS total_tokens,\n COALESCE(user_stats.total_sessions, 0) AS total_sessions,\n COALESCE(user_stats.total_messages_3_days, 0) AS messages_3_days,\n COALESCE(user_stats.total_token_count_3_days, 0) AS tokens_3_days,\n COALESCE(auth_user_management.rate_limit, $2::INTEGER) AS rate_limit\nFROM auth_user\nLEFT JOIN (\n SELECT \n stats.user_id, \n SUM(stats.total_messages) AS total_messages, \n SUM(stats.total_token_count) AS total_token_count,\n COUNT(DISTINCT stats.chat_session_uuid) AS total_sessions,\n SUM(CASE WHEN stats.created_at >= NOW() - INTERVAL '3 days' THEN stats.total_messages ELSE 0 END) AS total_messages_3_days,\n SUM(CASE WHEN stats.created_at >= NOW() - INTERVAL '3 days' THEN stats.total_token_count ELSE 0 END) AS total_token_count_3_days\n FROM (\n SELECT user_id, chat_session_uuid, COUNT(*) AS total_messages, SUM(token_count) as total_token_count, MAX(created_at) AS created_at\n FROM chat_message\n WHERE is_deleted = false\n GROUP BY user_id, chat_session_uuid\n ) AS stats\n GROUP BY stats.user_id\n) AS user_stats ON auth_user.id = user_stats.user_id\nLEFT JOIN auth_user_management ON auth_user.id = auth_user_management.user_id\nWHERE auth_user.email = $1\n`\n\ntype GetUserAnalysisByEmailParams struct {\n\tEmail string `json:\"email\"`\n\tDefaultRateLimit int32 `json:\"defaultRateLimit\"`\n}\n\ntype GetUserAnalysisByEmailRow struct {\n\tFirstName string `json:\"firstName\"`\n\tLastName string `json:\"lastName\"`\n\tUserEmail string `json:\"userEmail\"`\n\tTotalMessages int64 `json:\"totalMessages\"`\n\tTotalTokens int64 `json:\"totalTokens\"`\n\tTotalSessions int64 `json:\"totalSessions\"`\n\tMessages3Days int64 `json:\"messages3Days\"`\n\tTokens3Days int64 `json:\"tokens3Days\"`\n\tRateLimit int32 `json:\"rateLimit\"`\n}\n\nfunc (q *Queries) GetUserAnalysisByEmail(ctx context.Context, arg GetUserAnalysisByEmailParams) (GetUserAnalysisByEmailRow, error) {\n\trow := q.db.QueryRowContext(ctx, getUserAnalysisByEmail, arg.Email, arg.DefaultRateLimit)\n\tvar i GetUserAnalysisByEmailRow\n\terr := row.Scan(\n\t\t&i.FirstName,\n\t\t&i.LastName,\n\t\t&i.UserEmail,\n\t\t&i.TotalMessages,\n\t\t&i.TotalTokens,\n\t\t&i.TotalSessions,\n\t\t&i.Messages3Days,\n\t\t&i.Tokens3Days,\n\t\t&i.RateLimit,\n\t)\n\treturn i, err\n}\n\nconst getUserByEmail = `-- name: GetUserByEmail :one\nSELECT id, password, last_login, is_superuser, username, first_name, last_name, email, is_staff, is_active, date_joined FROM auth_user WHERE email = $1\n`\n\nfunc (q *Queries) GetUserByEmail(ctx context.Context, email string) (AuthUser, error) {\n\trow := q.db.QueryRowContext(ctx, getUserByEmail, email)\n\tvar i AuthUser\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Password,\n\t\t&i.LastLogin,\n\t\t&i.IsSuperuser,\n\t\t&i.Username,\n\t\t&i.FirstName,\n\t\t&i.LastName,\n\t\t&i.Email,\n\t\t&i.IsStaff,\n\t\t&i.IsActive,\n\t\t&i.DateJoined,\n\t)\n\treturn i, err\n}\n\nconst getUserModelUsageByEmail = `-- name: GetUserModelUsageByEmail :many\nSELECT \n COALESCE(cm.model, 'unknown') AS model,\n COUNT(*) AS message_count,\n COALESCE(SUM(cm.token_count), 0) AS token_count,\n MAX(cm.created_at)::timestamp AS last_used\nFROM chat_message cm\nINNER JOIN auth_user au ON cm.user_id = au.id\nWHERE au.email = $1 \n AND cm.is_deleted = false \n AND cm.role = 'assistant'\n AND cm.model IS NOT NULL \n AND cm.model != ''\nGROUP BY cm.model\nORDER BY message_count DESC\n`\n\ntype GetUserModelUsageByEmailRow struct {\n\tModel string `json:\"model\"`\n\tMessageCount int64 `json:\"messageCount\"`\n\tTokenCount interface{} `json:\"tokenCount\"`\n\tLastUsed time.Time `json:\"lastUsed\"`\n}\n\nfunc (q *Queries) GetUserModelUsageByEmail(ctx context.Context, email string) ([]GetUserModelUsageByEmailRow, error) {\n\trows, err := q.db.QueryContext(ctx, getUserModelUsageByEmail, email)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetUserModelUsageByEmailRow\n\tfor rows.Next() {\n\t\tvar i GetUserModelUsageByEmailRow\n\t\tif err := rows.Scan(\n\t\t\t&i.Model,\n\t\t\t&i.MessageCount,\n\t\t\t&i.TokenCount,\n\t\t\t&i.LastUsed,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getUserRecentActivityByEmail = `-- name: GetUserRecentActivityByEmail :many\nSELECT \n DATE(cm.created_at) AS activity_date,\n COUNT(*) AS messages,\n COALESCE(SUM(cm.token_count), 0) AS tokens,\n COUNT(DISTINCT cm.chat_session_uuid) AS sessions\nFROM chat_message cm\nINNER JOIN auth_user au ON cm.user_id = au.id\nWHERE au.email = $1 \n AND cm.is_deleted = false \n AND cm.created_at >= NOW() - INTERVAL '30 days'\nGROUP BY DATE(cm.created_at)\nORDER BY activity_date DESC\nLIMIT 30\n`\n\ntype GetUserRecentActivityByEmailRow struct {\n\tActivityDate time.Time `json:\"activityDate\"`\n\tMessages int64 `json:\"messages\"`\n\tTokens interface{} `json:\"tokens\"`\n\tSessions int64 `json:\"sessions\"`\n}\n\nfunc (q *Queries) GetUserRecentActivityByEmail(ctx context.Context, email string) ([]GetUserRecentActivityByEmailRow, error) {\n\trows, err := q.db.QueryContext(ctx, getUserRecentActivityByEmail, email)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetUserRecentActivityByEmailRow\n\tfor rows.Next() {\n\t\tvar i GetUserRecentActivityByEmailRow\n\t\tif err := rows.Scan(\n\t\t\t&i.ActivityDate,\n\t\t\t&i.Messages,\n\t\t\t&i.Tokens,\n\t\t\t&i.Sessions,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getUserSessionHistoryByEmail = `-- name: GetUserSessionHistoryByEmail :many\nSELECT \n cs.uuid AS session_id,\n cs.model,\n COALESCE(COUNT(cm.id), 0) AS message_count,\n COALESCE(SUM(cm.token_count), 0) AS token_count,\n COALESCE(MIN(cm.created_at), cs.created_at)::timestamp AS created_at,\n COALESCE(MAX(cm.created_at), cs.updated_at)::timestamp AS updated_at\nFROM chat_session cs\nINNER JOIN auth_user au ON cs.user_id = au.id\nLEFT JOIN chat_message cm ON cs.uuid = cm.chat_session_uuid AND cm.is_deleted = false\nWHERE au.email = $1 AND cs.active = true\nGROUP BY cs.uuid, cs.model, cs.created_at, cs.updated_at\nORDER BY cs.updated_at DESC\nLIMIT $2 OFFSET $3\n`\n\ntype GetUserSessionHistoryByEmailParams struct {\n\tEmail string `json:\"email\"`\n\tLimit int32 `json:\"limit\"`\n\tOffset int32 `json:\"offset\"`\n}\n\ntype GetUserSessionHistoryByEmailRow struct {\n\tSessionID string `json:\"sessionId\"`\n\tModel string `json:\"model\"`\n\tMessageCount interface{} `json:\"messageCount\"`\n\tTokenCount interface{} `json:\"tokenCount\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n}\n\nfunc (q *Queries) GetUserSessionHistoryByEmail(ctx context.Context, arg GetUserSessionHistoryByEmailParams) ([]GetUserSessionHistoryByEmailRow, error) {\n\trows, err := q.db.QueryContext(ctx, getUserSessionHistoryByEmail, arg.Email, arg.Limit, arg.Offset)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetUserSessionHistoryByEmailRow\n\tfor rows.Next() {\n\t\tvar i GetUserSessionHistoryByEmailRow\n\t\tif err := rows.Scan(\n\t\t\t&i.SessionID,\n\t\t\t&i.Model,\n\t\t\t&i.MessageCount,\n\t\t\t&i.TokenCount,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getUserSessionHistoryCountByEmail = `-- name: GetUserSessionHistoryCountByEmail :one\nSELECT COUNT(DISTINCT cs.uuid) AS total_sessions\nFROM chat_session cs\nINNER JOIN auth_user au ON cs.user_id = au.id\nWHERE au.email = $1 AND cs.active = true\n`\n\nfunc (q *Queries) GetUserSessionHistoryCountByEmail(ctx context.Context, email string) (int64, error) {\n\trow := q.db.QueryRowContext(ctx, getUserSessionHistoryCountByEmail, email)\n\tvar total_sessions int64\n\terr := row.Scan(&total_sessions)\n\treturn total_sessions, err\n}\n\nconst getUserStats = `-- name: GetUserStats :many\nSELECT \n auth_user.first_name,\n auth_user.last_name,\n auth_user.email AS user_email,\n COALESCE(user_stats.total_messages, 0) AS total_chat_messages,\n COALESCE(user_stats.total_token_count, 0) AS total_token_count,\n COALESCE(user_stats.total_messages_3_days, 0) AS total_chat_messages_3_days,\n COALESCE(user_stats.total_token_count_3_days, 0) AS total_token_count_3_days,\n COALESCE(auth_user_management.rate_limit, $3::INTEGER) AS rate_limit\nFROM auth_user\nLEFT JOIN (\n SELECT chat_message_stats.user_id, \n SUM(total_messages) AS total_messages, \n SUM(total_token_count) AS total_token_count,\n SUM(CASE WHEN created_at >= NOW() - INTERVAL '3 days' THEN total_messages ELSE 0 END) AS total_messages_3_days,\n SUM(CASE WHEN created_at >= NOW() - INTERVAL '3 days' THEN total_token_count ELSE 0 END) AS total_token_count_3_days\n FROM (\n SELECT user_id, COUNT(*) AS total_messages, SUM(token_count) as total_token_count, MAX(created_at) AS created_at\n FROM chat_message\n GROUP BY user_id, chat_session_uuid\n ) AS chat_message_stats\n GROUP BY chat_message_stats.user_id\n) AS user_stats ON auth_user.id = user_stats.user_id\nLEFT JOIN auth_user_management ON auth_user.id = auth_user_management.user_id\nORDER BY total_chat_messages DESC, auth_user.id DESC\nOFFSET $2\nLIMIT $1\n`\n\ntype GetUserStatsParams struct {\n\tLimit int32 `json:\"limit\"`\n\tOffset int32 `json:\"offset\"`\n\tDefaultRateLimit int32 `json:\"defaultRateLimit\"`\n}\n\ntype GetUserStatsRow struct {\n\tFirstName string `json:\"firstName\"`\n\tLastName string `json:\"lastName\"`\n\tUserEmail string `json:\"userEmail\"`\n\tTotalChatMessages int64 `json:\"totalChatMessages\"`\n\tTotalTokenCount int64 `json:\"totalTokenCount\"`\n\tTotalChatMessages3Days int64 `json:\"totalChatMessages3Days\"`\n\tTotalTokenCount3Days int64 `json:\"totalTokenCount3Days\"`\n\tRateLimit int32 `json:\"rateLimit\"`\n}\n\nfunc (q *Queries) GetUserStats(ctx context.Context, arg GetUserStatsParams) ([]GetUserStatsRow, error) {\n\trows, err := q.db.QueryContext(ctx, getUserStats, arg.Limit, arg.Offset, arg.DefaultRateLimit)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetUserStatsRow\n\tfor rows.Next() {\n\t\tvar i GetUserStatsRow\n\t\tif err := rows.Scan(\n\t\t\t&i.FirstName,\n\t\t\t&i.LastName,\n\t\t\t&i.UserEmail,\n\t\t\t&i.TotalChatMessages,\n\t\t\t&i.TotalTokenCount,\n\t\t\t&i.TotalChatMessages3Days,\n\t\t\t&i.TotalTokenCount3Days,\n\t\t\t&i.RateLimit,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst listAuthUsers = `-- name: ListAuthUsers :many\nSELECT id, password, last_login, is_superuser, username, first_name, last_name, email, is_staff, is_active, date_joined FROM auth_user ORDER BY id LIMIT $1 OFFSET $2\n`\n\ntype ListAuthUsersParams struct {\n\tLimit int32 `json:\"limit\"`\n\tOffset int32 `json:\"offset\"`\n}\n\nfunc (q *Queries) ListAuthUsers(ctx context.Context, arg ListAuthUsersParams) ([]AuthUser, error) {\n\trows, err := q.db.QueryContext(ctx, listAuthUsers, arg.Limit, arg.Offset)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []AuthUser\n\tfor rows.Next() {\n\t\tvar i AuthUser\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Password,\n\t\t\t&i.LastLogin,\n\t\t\t&i.IsSuperuser,\n\t\t\t&i.Username,\n\t\t\t&i.FirstName,\n\t\t\t&i.LastName,\n\t\t\t&i.Email,\n\t\t\t&i.IsStaff,\n\t\t\t&i.IsActive,\n\t\t\t&i.DateJoined,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst updateAuthUser = `-- name: UpdateAuthUser :one\nUPDATE auth_user SET first_name = $2, last_name= $3, last_login = now() \nWHERE id = $1\nRETURNING first_name, last_name, email\n`\n\ntype UpdateAuthUserParams struct {\n\tID int32 `json:\"id\"`\n\tFirstName string `json:\"firstName\"`\n\tLastName string `json:\"lastName\"`\n}\n\ntype UpdateAuthUserRow struct {\n\tFirstName string `json:\"firstName\"`\n\tLastName string `json:\"lastName\"`\n\tEmail string `json:\"email\"`\n}\n\nfunc (q *Queries) UpdateAuthUser(ctx context.Context, arg UpdateAuthUserParams) (UpdateAuthUserRow, error) {\n\trow := q.db.QueryRowContext(ctx, updateAuthUser, arg.ID, arg.FirstName, arg.LastName)\n\tvar i UpdateAuthUserRow\n\terr := row.Scan(&i.FirstName, &i.LastName, &i.Email)\n\treturn i, err\n}\n\nconst updateAuthUserByEmail = `-- name: UpdateAuthUserByEmail :one\nUPDATE auth_user SET first_name = $2, last_name= $3, last_login = now() \nWHERE email = $1\nRETURNING first_name, last_name, email\n`\n\ntype UpdateAuthUserByEmailParams struct {\n\tEmail string `json:\"email\"`\n\tFirstName string `json:\"firstName\"`\n\tLastName string `json:\"lastName\"`\n}\n\ntype UpdateAuthUserByEmailRow struct {\n\tFirstName string `json:\"firstName\"`\n\tLastName string `json:\"lastName\"`\n\tEmail string `json:\"email\"`\n}\n\nfunc (q *Queries) UpdateAuthUserByEmail(ctx context.Context, arg UpdateAuthUserByEmailParams) (UpdateAuthUserByEmailRow, error) {\n\trow := q.db.QueryRowContext(ctx, updateAuthUserByEmail, arg.Email, arg.FirstName, arg.LastName)\n\tvar i UpdateAuthUserByEmailRow\n\terr := row.Scan(&i.FirstName, &i.LastName, &i.Email)\n\treturn i, err\n}\n\nconst updateAuthUserRateLimitByEmail = `-- name: UpdateAuthUserRateLimitByEmail :one\nINSERT INTO auth_user_management (user_id, rate_limit, created_at, updated_at)\nVALUES ((SELECT id FROM auth_user WHERE email = $1), $2, NOW(), NOW())\nON CONFLICT (user_id) DO UPDATE SET rate_limit = $2, updated_at = NOW()\nRETURNING rate_limit\n`\n\ntype UpdateAuthUserRateLimitByEmailParams struct {\n\tEmail string `json:\"email\"`\n\tRateLimit int32 `json:\"rateLimit\"`\n}\n\nfunc (q *Queries) UpdateAuthUserRateLimitByEmail(ctx context.Context, arg UpdateAuthUserRateLimitByEmailParams) (int32, error) {\n\trow := q.db.QueryRowContext(ctx, updateAuthUserRateLimitByEmail, arg.Email, arg.RateLimit)\n\tvar rate_limit int32\n\terr := row.Scan(&rate_limit)\n\treturn rate_limit, err\n}\n\nconst updateUserPassword = `-- name: UpdateUserPassword :exec\nUPDATE auth_user SET \"password\" = $2 WHERE email = $1\n`\n\ntype UpdateUserPasswordParams struct {\n\tEmail string `json:\"email\"`\n\tPassword string `json:\"password\"`\n}\n\nfunc (q *Queries) UpdateUserPassword(ctx context.Context, arg UpdateUserPasswordParams) error {\n\t_, err := q.db.ExecContext(ctx, updateUserPassword, arg.Email, arg.Password)\n\treturn err\n}\n" }, { "path": "api/sqlc_queries/auth_user_management.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: auth_user_management.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n)\n\nconst getRateLimit = `-- name: GetRateLimit :one\nSELECT rate_limit AS rate_limit\nFROM auth_user_management\nWHERE user_id = $1\n`\n\n// GetRateLimit retrieves the rate limit for a user from the auth_user_management table.\n// If no rate limit is set for the user, it returns the default rate limit of 100.\nfunc (q *Queries) GetRateLimit(ctx context.Context, userID int32) (int32, error) {\n\trow := q.db.QueryRowContext(ctx, getRateLimit, userID)\n\tvar rate_limit int32\n\terr := row.Scan(&rate_limit)\n\treturn rate_limit, err\n}\n" }, { "path": "api/sqlc_queries/bot_answer_history.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: bot_answer_history.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n\t\"time\"\n)\n\nconst createBotAnswerHistory = `-- name: CreateBotAnswerHistory :one\n\nINSERT INTO bot_answer_history (\n bot_uuid,\n user_id,\n prompt,\n answer,\n model,\n tokens_used\n) VALUES (\n $1, $2, $3, $4, $5, $6\n) RETURNING id, bot_uuid, user_id, prompt, answer, model, tokens_used, created_at, updated_at\n`\n\ntype CreateBotAnswerHistoryParams struct {\n\tBotUuid string `json:\"botUuid\"`\n\tUserID int32 `json:\"userId\"`\n\tPrompt string `json:\"prompt\"`\n\tAnswer string `json:\"answer\"`\n\tModel string `json:\"model\"`\n\tTokensUsed int32 `json:\"tokensUsed\"`\n}\n\n// Bot Answer History Queries --\nfunc (q *Queries) CreateBotAnswerHistory(ctx context.Context, arg CreateBotAnswerHistoryParams) (BotAnswerHistory, error) {\n\trow := q.db.QueryRowContext(ctx, createBotAnswerHistory,\n\t\targ.BotUuid,\n\t\targ.UserID,\n\t\targ.Prompt,\n\t\targ.Answer,\n\t\targ.Model,\n\t\targ.TokensUsed,\n\t)\n\tvar i BotAnswerHistory\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.BotUuid,\n\t\t&i.UserID,\n\t\t&i.Prompt,\n\t\t&i.Answer,\n\t\t&i.Model,\n\t\t&i.TokensUsed,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t)\n\treturn i, err\n}\n\nconst deleteBotAnswerHistory = `-- name: DeleteBotAnswerHistory :exec\nDELETE FROM bot_answer_history WHERE id = $1\n`\n\nfunc (q *Queries) DeleteBotAnswerHistory(ctx context.Context, id int32) error {\n\t_, err := q.db.ExecContext(ctx, deleteBotAnswerHistory, id)\n\treturn err\n}\n\nconst getBotAnswerHistoryByBotUUID = `-- name: GetBotAnswerHistoryByBotUUID :many\nSELECT \n bah.id,\n bah.bot_uuid,\n bah.user_id,\n bah.prompt,\n bah.answer,\n bah.model,\n bah.tokens_used,\n bah.created_at,\n bah.updated_at,\n au.username AS user_username,\n au.email AS user_email\nFROM bot_answer_history bah\nJOIN auth_user au ON bah.user_id = au.id\nWHERE bah.bot_uuid = $1\nORDER BY bah.created_at DESC\nLIMIT $2 OFFSET $3\n`\n\ntype GetBotAnswerHistoryByBotUUIDParams struct {\n\tBotUuid string `json:\"botUuid\"`\n\tLimit int32 `json:\"limit\"`\n\tOffset int32 `json:\"offset\"`\n}\n\ntype GetBotAnswerHistoryByBotUUIDRow struct {\n\tID int32 `json:\"id\"`\n\tBotUuid string `json:\"botUuid\"`\n\tUserID int32 `json:\"userId\"`\n\tPrompt string `json:\"prompt\"`\n\tAnswer string `json:\"answer\"`\n\tModel string `json:\"model\"`\n\tTokensUsed int32 `json:\"tokensUsed\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tUserUsername string `json:\"userUsername\"`\n\tUserEmail string `json:\"userEmail\"`\n}\n\nfunc (q *Queries) GetBotAnswerHistoryByBotUUID(ctx context.Context, arg GetBotAnswerHistoryByBotUUIDParams) ([]GetBotAnswerHistoryByBotUUIDRow, error) {\n\trows, err := q.db.QueryContext(ctx, getBotAnswerHistoryByBotUUID, arg.BotUuid, arg.Limit, arg.Offset)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetBotAnswerHistoryByBotUUIDRow\n\tfor rows.Next() {\n\t\tvar i GetBotAnswerHistoryByBotUUIDRow\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.BotUuid,\n\t\t\t&i.UserID,\n\t\t\t&i.Prompt,\n\t\t\t&i.Answer,\n\t\t\t&i.Model,\n\t\t\t&i.TokensUsed,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.UserUsername,\n\t\t\t&i.UserEmail,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getBotAnswerHistoryByID = `-- name: GetBotAnswerHistoryByID :one\nSELECT \n bah.id,\n bah.bot_uuid,\n bah.user_id,\n bah.prompt,\n bah.answer,\n bah.model,\n bah.tokens_used,\n bah.created_at,\n bah.updated_at,\n au.username AS user_username,\n au.email AS user_email\nFROM bot_answer_history bah\nJOIN auth_user au ON bah.user_id = au.id\nWHERE bah.id = $1\n`\n\ntype GetBotAnswerHistoryByIDRow struct {\n\tID int32 `json:\"id\"`\n\tBotUuid string `json:\"botUuid\"`\n\tUserID int32 `json:\"userId\"`\n\tPrompt string `json:\"prompt\"`\n\tAnswer string `json:\"answer\"`\n\tModel string `json:\"model\"`\n\tTokensUsed int32 `json:\"tokensUsed\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tUserUsername string `json:\"userUsername\"`\n\tUserEmail string `json:\"userEmail\"`\n}\n\nfunc (q *Queries) GetBotAnswerHistoryByID(ctx context.Context, id int32) (GetBotAnswerHistoryByIDRow, error) {\n\trow := q.db.QueryRowContext(ctx, getBotAnswerHistoryByID, id)\n\tvar i GetBotAnswerHistoryByIDRow\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.BotUuid,\n\t\t&i.UserID,\n\t\t&i.Prompt,\n\t\t&i.Answer,\n\t\t&i.Model,\n\t\t&i.TokensUsed,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.UserUsername,\n\t\t&i.UserEmail,\n\t)\n\treturn i, err\n}\n\nconst getBotAnswerHistoryByUserID = `-- name: GetBotAnswerHistoryByUserID :many\nSELECT \n bah.id,\n bah.bot_uuid,\n bah.user_id,\n bah.prompt,\n bah.answer,\n bah.model,\n bah.tokens_used,\n bah.created_at,\n bah.updated_at,\n au.username AS user_username,\n au.email AS user_email\nFROM bot_answer_history bah\nJOIN auth_user au ON bah.user_id = au.id\nWHERE bah.user_id = $1\nORDER BY bah.created_at DESC\nLIMIT $2 OFFSET $3\n`\n\ntype GetBotAnswerHistoryByUserIDParams struct {\n\tUserID int32 `json:\"userId\"`\n\tLimit int32 `json:\"limit\"`\n\tOffset int32 `json:\"offset\"`\n}\n\ntype GetBotAnswerHistoryByUserIDRow struct {\n\tID int32 `json:\"id\"`\n\tBotUuid string `json:\"botUuid\"`\n\tUserID int32 `json:\"userId\"`\n\tPrompt string `json:\"prompt\"`\n\tAnswer string `json:\"answer\"`\n\tModel string `json:\"model\"`\n\tTokensUsed int32 `json:\"tokensUsed\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tUserUsername string `json:\"userUsername\"`\n\tUserEmail string `json:\"userEmail\"`\n}\n\nfunc (q *Queries) GetBotAnswerHistoryByUserID(ctx context.Context, arg GetBotAnswerHistoryByUserIDParams) ([]GetBotAnswerHistoryByUserIDRow, error) {\n\trows, err := q.db.QueryContext(ctx, getBotAnswerHistoryByUserID, arg.UserID, arg.Limit, arg.Offset)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetBotAnswerHistoryByUserIDRow\n\tfor rows.Next() {\n\t\tvar i GetBotAnswerHistoryByUserIDRow\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.BotUuid,\n\t\t\t&i.UserID,\n\t\t\t&i.Prompt,\n\t\t\t&i.Answer,\n\t\t\t&i.Model,\n\t\t\t&i.TokensUsed,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.UserUsername,\n\t\t\t&i.UserEmail,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getBotAnswerHistoryCountByBotUUID = `-- name: GetBotAnswerHistoryCountByBotUUID :one\nSELECT COUNT(*) FROM bot_answer_history WHERE bot_uuid = $1\n`\n\nfunc (q *Queries) GetBotAnswerHistoryCountByBotUUID(ctx context.Context, botUuid string) (int64, error) {\n\trow := q.db.QueryRowContext(ctx, getBotAnswerHistoryCountByBotUUID, botUuid)\n\tvar count int64\n\terr := row.Scan(&count)\n\treturn count, err\n}\n\nconst getBotAnswerHistoryCountByUserID = `-- name: GetBotAnswerHistoryCountByUserID :one\nSELECT COUNT(*) FROM bot_answer_history WHERE user_id = $1\n`\n\nfunc (q *Queries) GetBotAnswerHistoryCountByUserID(ctx context.Context, userID int32) (int64, error) {\n\trow := q.db.QueryRowContext(ctx, getBotAnswerHistoryCountByUserID, userID)\n\tvar count int64\n\terr := row.Scan(&count)\n\treturn count, err\n}\n\nconst getLatestBotAnswerHistoryByBotUUID = `-- name: GetLatestBotAnswerHistoryByBotUUID :many\nSELECT \n bah.id,\n bah.bot_uuid,\n bah.user_id,\n bah.prompt,\n bah.answer,\n bah.model,\n bah.tokens_used,\n bah.created_at,\n bah.updated_at,\n au.username AS user_username,\n au.email AS user_email\nFROM bot_answer_history bah\nJOIN auth_user au ON bah.user_id = au.id\nWHERE bah.bot_uuid = $1\nORDER BY bah.created_at DESC\nLIMIT $2\n`\n\ntype GetLatestBotAnswerHistoryByBotUUIDParams struct {\n\tBotUuid string `json:\"botUuid\"`\n\tLimit int32 `json:\"limit\"`\n}\n\ntype GetLatestBotAnswerHistoryByBotUUIDRow struct {\n\tID int32 `json:\"id\"`\n\tBotUuid string `json:\"botUuid\"`\n\tUserID int32 `json:\"userId\"`\n\tPrompt string `json:\"prompt\"`\n\tAnswer string `json:\"answer\"`\n\tModel string `json:\"model\"`\n\tTokensUsed int32 `json:\"tokensUsed\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tUserUsername string `json:\"userUsername\"`\n\tUserEmail string `json:\"userEmail\"`\n}\n\nfunc (q *Queries) GetLatestBotAnswerHistoryByBotUUID(ctx context.Context, arg GetLatestBotAnswerHistoryByBotUUIDParams) ([]GetLatestBotAnswerHistoryByBotUUIDRow, error) {\n\trows, err := q.db.QueryContext(ctx, getLatestBotAnswerHistoryByBotUUID, arg.BotUuid, arg.Limit)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetLatestBotAnswerHistoryByBotUUIDRow\n\tfor rows.Next() {\n\t\tvar i GetLatestBotAnswerHistoryByBotUUIDRow\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.BotUuid,\n\t\t\t&i.UserID,\n\t\t\t&i.Prompt,\n\t\t\t&i.Answer,\n\t\t\t&i.Model,\n\t\t\t&i.TokensUsed,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.UserUsername,\n\t\t\t&i.UserEmail,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst updateBotAnswerHistory = `-- name: UpdateBotAnswerHistory :one\nUPDATE bot_answer_history\nSET\n answer = $2,\n tokens_used = $3,\n updated_at = NOW()\nWHERE id = $1\nRETURNING id, bot_uuid, user_id, prompt, answer, model, tokens_used, created_at, updated_at\n`\n\ntype UpdateBotAnswerHistoryParams struct {\n\tID int32 `json:\"id\"`\n\tAnswer string `json:\"answer\"`\n\tTokensUsed int32 `json:\"tokensUsed\"`\n}\n\nfunc (q *Queries) UpdateBotAnswerHistory(ctx context.Context, arg UpdateBotAnswerHistoryParams) (BotAnswerHistory, error) {\n\trow := q.db.QueryRowContext(ctx, updateBotAnswerHistory, arg.ID, arg.Answer, arg.TokensUsed)\n\tvar i BotAnswerHistory\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.BotUuid,\n\t\t&i.UserID,\n\t\t&i.Prompt,\n\t\t&i.Answer,\n\t\t&i.Model,\n\t\t&i.TokensUsed,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t)\n\treturn i, err\n}\n" }, { "path": "api/sqlc_queries/chat_comment.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: chat_comment.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n\t\"time\"\n)\n\nconst createChatComment = `-- name: CreateChatComment :one\nINSERT INTO chat_comment (\n uuid,\n chat_session_uuid,\n chat_message_uuid, \n content,\n created_by,\n updated_by\n) VALUES (\n $1, $2, $3, $4, $5, $5\n) RETURNING id, uuid, chat_session_uuid, chat_message_uuid, content, created_at, updated_at, created_by, updated_by\n`\n\ntype CreateChatCommentParams struct {\n\tUuid string `json:\"uuid\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\tChatMessageUuid string `json:\"chatMessageUuid\"`\n\tContent string `json:\"content\"`\n\tCreatedBy int32 `json:\"createdBy\"`\n}\n\nfunc (q *Queries) CreateChatComment(ctx context.Context, arg CreateChatCommentParams) (ChatComment, error) {\n\trow := q.db.QueryRowContext(ctx, createChatComment,\n\t\targ.Uuid,\n\t\targ.ChatSessionUuid,\n\t\targ.ChatMessageUuid,\n\t\targ.Content,\n\t\targ.CreatedBy,\n\t)\n\tvar i ChatComment\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.ChatMessageUuid,\n\t\t&i.Content,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t)\n\treturn i, err\n}\n\nconst getCommentsByMessageUUID = `-- name: GetCommentsByMessageUUID :many\nSELECT \n cc.uuid,\n cc.content,\n cc.created_at,\n au.username AS author_username,\n au.email AS author_email\nFROM chat_comment cc\nJOIN auth_user au ON cc.created_by = au.id\nWHERE cc.chat_message_uuid = $1\nORDER BY cc.created_at DESC\n`\n\ntype GetCommentsByMessageUUIDRow struct {\n\tUuid string `json:\"uuid\"`\n\tContent string `json:\"content\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tAuthorUsername string `json:\"authorUsername\"`\n\tAuthorEmail string `json:\"authorEmail\"`\n}\n\nfunc (q *Queries) GetCommentsByMessageUUID(ctx context.Context, chatMessageUuid string) ([]GetCommentsByMessageUUIDRow, error) {\n\trows, err := q.db.QueryContext(ctx, getCommentsByMessageUUID, chatMessageUuid)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetCommentsByMessageUUIDRow\n\tfor rows.Next() {\n\t\tvar i GetCommentsByMessageUUIDRow\n\t\tif err := rows.Scan(\n\t\t\t&i.Uuid,\n\t\t\t&i.Content,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.AuthorUsername,\n\t\t\t&i.AuthorEmail,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getCommentsBySessionUUID = `-- name: GetCommentsBySessionUUID :many\nSELECT \n cc.uuid,\n cc.chat_message_uuid,\n cc.content,\n cc.created_at,\n au.username AS author_username,\n au.email AS author_email\nFROM chat_comment cc\nJOIN auth_user au ON cc.created_by = au.id\nWHERE cc.chat_session_uuid = $1\nORDER BY cc.created_at DESC\n`\n\ntype GetCommentsBySessionUUIDRow struct {\n\tUuid string `json:\"uuid\"`\n\tChatMessageUuid string `json:\"chatMessageUuid\"`\n\tContent string `json:\"content\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tAuthorUsername string `json:\"authorUsername\"`\n\tAuthorEmail string `json:\"authorEmail\"`\n}\n\nfunc (q *Queries) GetCommentsBySessionUUID(ctx context.Context, chatSessionUuid string) ([]GetCommentsBySessionUUIDRow, error) {\n\trows, err := q.db.QueryContext(ctx, getCommentsBySessionUUID, chatSessionUuid)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetCommentsBySessionUUIDRow\n\tfor rows.Next() {\n\t\tvar i GetCommentsBySessionUUIDRow\n\t\tif err := rows.Scan(\n\t\t\t&i.Uuid,\n\t\t\t&i.ChatMessageUuid,\n\t\t\t&i.Content,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.AuthorUsername,\n\t\t\t&i.AuthorEmail,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n" }, { "path": "api/sqlc_queries/chat_file.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: chat_file.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n\t\"time\"\n)\n\nconst createChatFile = `-- name: CreateChatFile :one\nINSERT INTO chat_file (name, data, user_id, chat_session_uuid, mime_type)\nVALUES ($1, $2, $3, $4, $5)\nRETURNING id, name, data, created_at, user_id, chat_session_uuid, mime_type\n`\n\ntype CreateChatFileParams struct {\n\tName string `json:\"name\"`\n\tData []byte `json:\"data\"`\n\tUserID int32 `json:\"userId\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\tMimeType string `json:\"mimeType\"`\n}\n\nfunc (q *Queries) CreateChatFile(ctx context.Context, arg CreateChatFileParams) (ChatFile, error) {\n\trow := q.db.QueryRowContext(ctx, createChatFile,\n\t\targ.Name,\n\t\targ.Data,\n\t\targ.UserID,\n\t\targ.ChatSessionUuid,\n\t\targ.MimeType,\n\t)\n\tvar i ChatFile\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Name,\n\t\t&i.Data,\n\t\t&i.CreatedAt,\n\t\t&i.UserID,\n\t\t&i.ChatSessionUuid,\n\t\t&i.MimeType,\n\t)\n\treturn i, err\n}\n\nconst deleteChatFile = `-- name: DeleteChatFile :one\nDELETE FROM chat_file\nWHERE id = $1\nRETURNING id, name, data, created_at, user_id, chat_session_uuid, mime_type\n`\n\nfunc (q *Queries) DeleteChatFile(ctx context.Context, id int32) (ChatFile, error) {\n\trow := q.db.QueryRowContext(ctx, deleteChatFile, id)\n\tvar i ChatFile\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Name,\n\t\t&i.Data,\n\t\t&i.CreatedAt,\n\t\t&i.UserID,\n\t\t&i.ChatSessionUuid,\n\t\t&i.MimeType,\n\t)\n\treturn i, err\n}\n\nconst getChatFileByID = `-- name: GetChatFileByID :one\nSELECT id, name, data, created_at, user_id, chat_session_uuid\nFROM chat_file\nWHERE id = $1\n`\n\ntype GetChatFileByIDRow struct {\n\tID int32 `json:\"id\"`\n\tName string `json:\"name\"`\n\tData []byte `json:\"data\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUserID int32 `json:\"userId\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n}\n\nfunc (q *Queries) GetChatFileByID(ctx context.Context, id int32) (GetChatFileByIDRow, error) {\n\trow := q.db.QueryRowContext(ctx, getChatFileByID, id)\n\tvar i GetChatFileByIDRow\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Name,\n\t\t&i.Data,\n\t\t&i.CreatedAt,\n\t\t&i.UserID,\n\t\t&i.ChatSessionUuid,\n\t)\n\treturn i, err\n}\n\nconst listChatFilesBySessionUUID = `-- name: ListChatFilesBySessionUUID :many\nSELECT id, name\nFROM chat_file\nWHERE user_id = $1 and chat_session_uuid = $2\nORDER BY created_at\n`\n\ntype ListChatFilesBySessionUUIDParams struct {\n\tUserID int32 `json:\"userId\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n}\n\ntype ListChatFilesBySessionUUIDRow struct {\n\tID int32 `json:\"id\"`\n\tName string `json:\"name\"`\n}\n\nfunc (q *Queries) ListChatFilesBySessionUUID(ctx context.Context, arg ListChatFilesBySessionUUIDParams) ([]ListChatFilesBySessionUUIDRow, error) {\n\trows, err := q.db.QueryContext(ctx, listChatFilesBySessionUUID, arg.UserID, arg.ChatSessionUuid)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ListChatFilesBySessionUUIDRow\n\tfor rows.Next() {\n\t\tvar i ListChatFilesBySessionUUIDRow\n\t\tif err := rows.Scan(&i.ID, &i.Name); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst listChatFilesWithContentBySessionUUID = `-- name: ListChatFilesWithContentBySessionUUID :many\nSELECT id, name, data, created_at, user_id, chat_session_uuid, mime_type\nFROM chat_file\nWHERE chat_session_uuid = $1\nORDER BY created_at\n`\n\nfunc (q *Queries) ListChatFilesWithContentBySessionUUID(ctx context.Context, chatSessionUuid string) ([]ChatFile, error) {\n\trows, err := q.db.QueryContext(ctx, listChatFilesWithContentBySessionUUID, chatSessionUuid)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatFile\n\tfor rows.Next() {\n\t\tvar i ChatFile\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Name,\n\t\t\t&i.Data,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UserID,\n\t\t\t&i.ChatSessionUuid,\n\t\t\t&i.MimeType,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n" }, { "path": "api/sqlc_queries/chat_log.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: chat_log.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n)\n\nconst chatLogByID = `-- name: ChatLogByID :one\nSELECT id, session, question, answer, created_at FROM chat_logs WHERE id = $1\n`\n\nfunc (q *Queries) ChatLogByID(ctx context.Context, id int32) (ChatLog, error) {\n\trow := q.db.QueryRowContext(ctx, chatLogByID, id)\n\tvar i ChatLog\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Session,\n\t\t&i.Question,\n\t\t&i.Answer,\n\t\t&i.CreatedAt,\n\t)\n\treturn i, err\n}\n\nconst createChatLog = `-- name: CreateChatLog :one\nINSERT INTO chat_logs (session, question, answer)\nVALUES ($1, $2, $3)\nRETURNING id, session, question, answer, created_at\n`\n\ntype CreateChatLogParams struct {\n\tSession json.RawMessage `json:\"session\"`\n\tQuestion json.RawMessage `json:\"question\"`\n\tAnswer json.RawMessage `json:\"answer\"`\n}\n\nfunc (q *Queries) CreateChatLog(ctx context.Context, arg CreateChatLogParams) (ChatLog, error) {\n\trow := q.db.QueryRowContext(ctx, createChatLog, arg.Session, arg.Question, arg.Answer)\n\tvar i ChatLog\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Session,\n\t\t&i.Question,\n\t\t&i.Answer,\n\t\t&i.CreatedAt,\n\t)\n\treturn i, err\n}\n\nconst deleteChatLog = `-- name: DeleteChatLog :exec\nDELETE FROM chat_logs WHERE id = $1\n`\n\nfunc (q *Queries) DeleteChatLog(ctx context.Context, id int32) error {\n\t_, err := q.db.ExecContext(ctx, deleteChatLog, id)\n\treturn err\n}\n\nconst listChatLogs = `-- name: ListChatLogs :many\nSELECT id, session, question, answer, created_at FROM chat_logs ORDER BY id\n`\n\nfunc (q *Queries) ListChatLogs(ctx context.Context) ([]ChatLog, error) {\n\trows, err := q.db.QueryContext(ctx, listChatLogs)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatLog\n\tfor rows.Next() {\n\t\tvar i ChatLog\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Session,\n\t\t\t&i.Question,\n\t\t\t&i.Answer,\n\t\t\t&i.CreatedAt,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst updateChatLog = `-- name: UpdateChatLog :one\nUPDATE chat_logs SET session = $2, question = $3, answer = $4\nWHERE id = $1\nRETURNING id, session, question, answer, created_at\n`\n\ntype UpdateChatLogParams struct {\n\tID int32 `json:\"id\"`\n\tSession json.RawMessage `json:\"session\"`\n\tQuestion json.RawMessage `json:\"question\"`\n\tAnswer json.RawMessage `json:\"answer\"`\n}\n\nfunc (q *Queries) UpdateChatLog(ctx context.Context, arg UpdateChatLogParams) (ChatLog, error) {\n\trow := q.db.QueryRowContext(ctx, updateChatLog,\n\t\targ.ID,\n\t\targ.Session,\n\t\targ.Question,\n\t\targ.Answer,\n\t)\n\tvar i ChatLog\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Session,\n\t\t&i.Question,\n\t\t&i.Answer,\n\t\t&i.CreatedAt,\n\t)\n\treturn i, err\n}\n" }, { "path": "api/sqlc_queries/chat_message.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: chat_message.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"time\"\n)\n\nconst createChatMessage = `-- name: CreateChatMessage :one\nINSERT INTO chat_message (chat_session_uuid, uuid, role, content, reasoning_content, model, token_count, score, user_id, created_by, updated_by, llm_summary, raw, artifacts, suggested_questions)\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15)\nRETURNING id, uuid, chat_session_uuid, role, content, reasoning_content, model, llm_summary, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, is_pin, token_count, raw, artifacts, suggested_questions\n`\n\ntype CreateChatMessageParams struct {\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\tUuid string `json:\"uuid\"`\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n\tReasoningContent string `json:\"reasoningContent\"`\n\tModel string `json:\"model\"`\n\tTokenCount int32 `json:\"tokenCount\"`\n\tScore float64 `json:\"score\"`\n\tUserID int32 `json:\"userId\"`\n\tCreatedBy int32 `json:\"createdBy\"`\n\tUpdatedBy int32 `json:\"updatedBy\"`\n\tLlmSummary string `json:\"llmSummary\"`\n\tRaw json.RawMessage `json:\"raw\"`\n\tArtifacts json.RawMessage `json:\"artifacts\"`\n\tSuggestedQuestions json.RawMessage `json:\"suggestedQuestions\"`\n}\n\nfunc (q *Queries) CreateChatMessage(ctx context.Context, arg CreateChatMessageParams) (ChatMessage, error) {\n\trow := q.db.QueryRowContext(ctx, createChatMessage,\n\t\targ.ChatSessionUuid,\n\t\targ.Uuid,\n\t\targ.Role,\n\t\targ.Content,\n\t\targ.ReasoningContent,\n\t\targ.Model,\n\t\targ.TokenCount,\n\t\targ.Score,\n\t\targ.UserID,\n\t\targ.CreatedBy,\n\t\targ.UpdatedBy,\n\t\targ.LlmSummary,\n\t\targ.Raw,\n\t\targ.Artifacts,\n\t\targ.SuggestedQuestions,\n\t)\n\tvar i ChatMessage\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.ReasoningContent,\n\t\t&i.Model,\n\t\t&i.LlmSummary,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.IsPin,\n\t\t&i.TokenCount,\n\t\t&i.Raw,\n\t\t&i.Artifacts,\n\t\t&i.SuggestedQuestions,\n\t)\n\treturn i, err\n}\n\nconst deleteChatMessage = `-- name: DeleteChatMessage :exec\nUPDATE chat_message set is_deleted = true, updated_at = now()\nWHERE id = $1\n`\n\nfunc (q *Queries) DeleteChatMessage(ctx context.Context, id int32) error {\n\t_, err := q.db.ExecContext(ctx, deleteChatMessage, id)\n\treturn err\n}\n\nconst deleteChatMessageByUUID = `-- name: DeleteChatMessageByUUID :exec\nUPDATE chat_message SET is_deleted = true, updated_at = now()\nWHERE uuid = $1\n`\n\nfunc (q *Queries) DeleteChatMessageByUUID(ctx context.Context, uuid string) error {\n\t_, err := q.db.ExecContext(ctx, deleteChatMessageByUUID, uuid)\n\treturn err\n}\n\nconst deleteChatMessagesBySesionUUID = `-- name: DeleteChatMessagesBySesionUUID :exec\nUPDATE chat_message \nSET is_deleted = true, updated_at = now()\nWHERE is_deleted = false and is_pin = false and chat_session_uuid = $1\n`\n\nfunc (q *Queries) DeleteChatMessagesBySesionUUID(ctx context.Context, chatSessionUuid string) error {\n\t_, err := q.db.ExecContext(ctx, deleteChatMessagesBySesionUUID, chatSessionUuid)\n\treturn err\n}\n\nconst getAllChatMessages = `-- name: GetAllChatMessages :many\nSELECT id, uuid, chat_session_uuid, role, content, reasoning_content, model, llm_summary, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, is_pin, token_count, raw, artifacts, suggested_questions FROM chat_message \nWHERE is_deleted = false\nORDER BY id\n`\n\nfunc (q *Queries) GetAllChatMessages(ctx context.Context) ([]ChatMessage, error) {\n\trows, err := q.db.QueryContext(ctx, getAllChatMessages)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatMessage\n\tfor rows.Next() {\n\t\tvar i ChatMessage\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Uuid,\n\t\t\t&i.ChatSessionUuid,\n\t\t\t&i.Role,\n\t\t\t&i.Content,\n\t\t\t&i.ReasoningContent,\n\t\t\t&i.Model,\n\t\t\t&i.LlmSummary,\n\t\t\t&i.Score,\n\t\t\t&i.UserID,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.CreatedBy,\n\t\t\t&i.UpdatedBy,\n\t\t\t&i.IsDeleted,\n\t\t\t&i.IsPin,\n\t\t\t&i.TokenCount,\n\t\t\t&i.Raw,\n\t\t\t&i.Artifacts,\n\t\t\t&i.SuggestedQuestions,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getChatMessageByID = `-- name: GetChatMessageByID :one\nSELECT id, uuid, chat_session_uuid, role, content, reasoning_content, model, llm_summary, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, is_pin, token_count, raw, artifacts, suggested_questions FROM chat_message \nWHERE is_deleted = false and id = $1\n`\n\nfunc (q *Queries) GetChatMessageByID(ctx context.Context, id int32) (ChatMessage, error) {\n\trow := q.db.QueryRowContext(ctx, getChatMessageByID, id)\n\tvar i ChatMessage\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.ReasoningContent,\n\t\t&i.Model,\n\t\t&i.LlmSummary,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.IsPin,\n\t\t&i.TokenCount,\n\t\t&i.Raw,\n\t\t&i.Artifacts,\n\t\t&i.SuggestedQuestions,\n\t)\n\treturn i, err\n}\n\nconst getChatMessageBySessionUUID = `-- name: GetChatMessageBySessionUUID :one\nSELECT cm.id, cm.uuid, cm.chat_session_uuid, cm.role, cm.content, cm.reasoning_content, cm.model, cm.llm_summary, cm.score, cm.user_id, cm.created_at, cm.updated_at, cm.created_by, cm.updated_by, cm.is_deleted, cm.is_pin, cm.token_count, cm.raw, cm.artifacts, cm.suggested_questions\nFROM chat_message cm\nINNER JOIN chat_session cs ON cm.chat_session_uuid = cs.uuid\nWHERE cm.is_deleted = false and cs.active = true and cs.uuid = $1 \nORDER BY cm.id \nOFFSET $2\nLIMIT $1\n`\n\ntype GetChatMessageBySessionUUIDParams struct {\n\tLimit int32 `json:\"limit\"`\n\tOffset int32 `json:\"offset\"`\n}\n\nfunc (q *Queries) GetChatMessageBySessionUUID(ctx context.Context, arg GetChatMessageBySessionUUIDParams) (ChatMessage, error) {\n\trow := q.db.QueryRowContext(ctx, getChatMessageBySessionUUID, arg.Limit, arg.Offset)\n\tvar i ChatMessage\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.ReasoningContent,\n\t\t&i.Model,\n\t\t&i.LlmSummary,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.IsPin,\n\t\t&i.TokenCount,\n\t\t&i.Raw,\n\t\t&i.Artifacts,\n\t\t&i.SuggestedQuestions,\n\t)\n\treturn i, err\n}\n\nconst getChatMessageByUUID = `-- name: GetChatMessageByUUID :one\n\nSELECT id, uuid, chat_session_uuid, role, content, reasoning_content, model, llm_summary, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, is_pin, token_count, raw, artifacts, suggested_questions FROM chat_message \nWHERE is_deleted = false and uuid = $1\n`\n\n// -- UUID ----\nfunc (q *Queries) GetChatMessageByUUID(ctx context.Context, uuid string) (ChatMessage, error) {\n\trow := q.db.QueryRowContext(ctx, getChatMessageByUUID, uuid)\n\tvar i ChatMessage\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.ReasoningContent,\n\t\t&i.Model,\n\t\t&i.LlmSummary,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.IsPin,\n\t\t&i.TokenCount,\n\t\t&i.Raw,\n\t\t&i.Artifacts,\n\t\t&i.SuggestedQuestions,\n\t)\n\treturn i, err\n}\n\nconst getChatMessagesBySessionUUID = `-- name: GetChatMessagesBySessionUUID :many\nSELECT cm.id, cm.uuid, cm.chat_session_uuid, cm.role, cm.content, cm.reasoning_content, cm.model, cm.llm_summary, cm.score, cm.user_id, cm.created_at, cm.updated_at, cm.created_by, cm.updated_by, cm.is_deleted, cm.is_pin, cm.token_count, cm.raw, cm.artifacts, cm.suggested_questions\nFROM chat_message cm\nINNER JOIN chat_session cs ON cm.chat_session_uuid = cs.uuid\nWHERE cm.is_deleted = false and cs.active = true and cs.uuid = $1 \nORDER BY cm.id \nOFFSET $2\nLIMIT $3\n`\n\ntype GetChatMessagesBySessionUUIDParams struct {\n\tUuid string `json:\"uuid\"`\n\tOffset int32 `json:\"offset\"`\n\tLimit int32 `json:\"limit\"`\n}\n\nfunc (q *Queries) GetChatMessagesBySessionUUID(ctx context.Context, arg GetChatMessagesBySessionUUIDParams) ([]ChatMessage, error) {\n\trows, err := q.db.QueryContext(ctx, getChatMessagesBySessionUUID, arg.Uuid, arg.Offset, arg.Limit)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatMessage\n\tfor rows.Next() {\n\t\tvar i ChatMessage\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Uuid,\n\t\t\t&i.ChatSessionUuid,\n\t\t\t&i.Role,\n\t\t\t&i.Content,\n\t\t\t&i.ReasoningContent,\n\t\t\t&i.Model,\n\t\t\t&i.LlmSummary,\n\t\t\t&i.Score,\n\t\t\t&i.UserID,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.CreatedBy,\n\t\t\t&i.UpdatedBy,\n\t\t\t&i.IsDeleted,\n\t\t\t&i.IsPin,\n\t\t\t&i.TokenCount,\n\t\t\t&i.Raw,\n\t\t\t&i.Artifacts,\n\t\t\t&i.SuggestedQuestions,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getChatMessagesBySessionUUIDForAdmin = `-- name: GetChatMessagesBySessionUUIDForAdmin :many\nSELECT \n id,\n uuid,\n role,\n content,\n reasoning_content,\n model,\n token_count,\n user_id,\n created_at,\n updated_at\nFROM (\n -- Include session prompts as the first messages\n SELECT \n cp.id,\n cp.uuid,\n cp.role,\n cp.content,\n ''::text as reasoning_content,\n cs.model,\n cp.token_count,\n cp.user_id,\n cp.created_at,\n cp.updated_at\n FROM chat_prompt cp\n INNER JOIN chat_session cs ON cp.chat_session_uuid = cs.uuid\n WHERE cp.chat_session_uuid = $1 \n AND cp.is_deleted = false\n AND cp.role = 'system'\n \n UNION ALL\n \n -- Include regular chat messages\n SELECT \n id,\n uuid,\n role,\n content,\n reasoning_content,\n model,\n token_count,\n user_id,\n created_at,\n updated_at\n FROM chat_message\n WHERE chat_session_uuid = $1 \n AND is_deleted = false\n) combined_messages\nORDER BY created_at ASC\n`\n\ntype GetChatMessagesBySessionUUIDForAdminRow struct {\n\tID int32 `json:\"id\"`\n\tUuid string `json:\"uuid\"`\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n\tReasoningContent string `json:\"reasoningContent\"`\n\tModel string `json:\"model\"`\n\tTokenCount int32 `json:\"tokenCount\"`\n\tUserID int32 `json:\"userId\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n}\n\nfunc (q *Queries) GetChatMessagesBySessionUUIDForAdmin(ctx context.Context, chatSessionUuid string) ([]GetChatMessagesBySessionUUIDForAdminRow, error) {\n\trows, err := q.db.QueryContext(ctx, getChatMessagesBySessionUUIDForAdmin, chatSessionUuid)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetChatMessagesBySessionUUIDForAdminRow\n\tfor rows.Next() {\n\t\tvar i GetChatMessagesBySessionUUIDForAdminRow\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Uuid,\n\t\t\t&i.Role,\n\t\t\t&i.Content,\n\t\t\t&i.ReasoningContent,\n\t\t\t&i.Model,\n\t\t\t&i.TokenCount,\n\t\t\t&i.UserID,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getChatMessagesCount = `-- name: GetChatMessagesCount :one\nSELECT COUNT(*)\nFROM chat_message\nWHERE user_id = $1\nAND created_at >= NOW() - INTERVAL '10 minutes'\n`\n\n// Get total chat message count for user in last 10 minutes\nfunc (q *Queries) GetChatMessagesCount(ctx context.Context, userID int32) (int64, error) {\n\trow := q.db.QueryRowContext(ctx, getChatMessagesCount, userID)\n\tvar count int64\n\terr := row.Scan(&count)\n\treturn count, err\n}\n\nconst getChatMessagesCountByUserAndModel = `-- name: GetChatMessagesCountByUserAndModel :one\nSELECT COUNT(*)\nFROM chat_message cm\nJOIN chat_session cs ON (cm.chat_session_uuid = cs.uuid AND cs.user_id = cm.user_id)\nWHERE cm.user_id = $1\nAND cs.model = $2 \nAND cm.created_at >= NOW() - INTERVAL '10 minutes'\n`\n\ntype GetChatMessagesCountByUserAndModelParams struct {\n\tUserID int32 `json:\"userId\"`\n\tModel string `json:\"model\"`\n}\n\n// Get total chat message count for user of model in last 10 minutes\nfunc (q *Queries) GetChatMessagesCountByUserAndModel(ctx context.Context, arg GetChatMessagesCountByUserAndModelParams) (int64, error) {\n\trow := q.db.QueryRowContext(ctx, getChatMessagesCountByUserAndModel, arg.UserID, arg.Model)\n\tvar count int64\n\terr := row.Scan(&count)\n\treturn count, err\n}\n\nconst getFirstMessageBySessionUUID = `-- name: GetFirstMessageBySessionUUID :one\nSELECT id, uuid, chat_session_uuid, role, content, reasoning_content, model, llm_summary, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, is_pin, token_count, raw, artifacts, suggested_questions\nFROM chat_message\nWHERE chat_session_uuid = $1 and is_deleted = false\nORDER BY created_at \nLIMIT 1\n`\n\nfunc (q *Queries) GetFirstMessageBySessionUUID(ctx context.Context, chatSessionUuid string) (ChatMessage, error) {\n\trow := q.db.QueryRowContext(ctx, getFirstMessageBySessionUUID, chatSessionUuid)\n\tvar i ChatMessage\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.ReasoningContent,\n\t\t&i.Model,\n\t\t&i.LlmSummary,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.IsPin,\n\t\t&i.TokenCount,\n\t\t&i.Raw,\n\t\t&i.Artifacts,\n\t\t&i.SuggestedQuestions,\n\t)\n\treturn i, err\n}\n\nconst getLastNChatMessages = `-- name: GetLastNChatMessages :many\nSELECT id, uuid, chat_session_uuid, role, content, reasoning_content, model, llm_summary, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, is_pin, token_count, raw, artifacts, suggested_questions\nFROM chat_message\nWHERE chat_message.id in (\n SELECT id\n FROM chat_message cm\n WHERE cm.chat_session_uuid = $3 and cm.is_deleted = false and cm.is_pin = true\n UNION\n (\n SELECT id \n FROM chat_message cm\n WHERE cm.chat_session_uuid = $3 \n AND cm.id < (SELECT id FROM chat_message WHERE chat_message.uuid = $1)\n AND cm.is_deleted = false -- and cm.is_pin = false\n ORDER BY cm.created_at DESC\n LIMIT $2\n )\n) \nORDER BY created_at\n`\n\ntype GetLastNChatMessagesParams struct {\n\tUuid string `json:\"uuid\"`\n\tLimit int32 `json:\"limit\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n}\n\nfunc (q *Queries) GetLastNChatMessages(ctx context.Context, arg GetLastNChatMessagesParams) ([]ChatMessage, error) {\n\trows, err := q.db.QueryContext(ctx, getLastNChatMessages, arg.Uuid, arg.Limit, arg.ChatSessionUuid)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatMessage\n\tfor rows.Next() {\n\t\tvar i ChatMessage\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Uuid,\n\t\t\t&i.ChatSessionUuid,\n\t\t\t&i.Role,\n\t\t\t&i.Content,\n\t\t\t&i.ReasoningContent,\n\t\t\t&i.Model,\n\t\t\t&i.LlmSummary,\n\t\t\t&i.Score,\n\t\t\t&i.UserID,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.CreatedBy,\n\t\t\t&i.UpdatedBy,\n\t\t\t&i.IsDeleted,\n\t\t\t&i.IsPin,\n\t\t\t&i.TokenCount,\n\t\t\t&i.Raw,\n\t\t\t&i.Artifacts,\n\t\t\t&i.SuggestedQuestions,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getLatestMessagesBySessionUUID = `-- name: GetLatestMessagesBySessionUUID :many\nSELECT id, uuid, chat_session_uuid, role, content, reasoning_content, model, llm_summary, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, is_pin, token_count, raw, artifacts, suggested_questions\nFROM chat_message\nWhere chat_message.id in \n(\n SELECT chat_message.id\n FROM chat_message\n WHERE chat_message.chat_session_uuid = $1 and chat_message.is_deleted = false and chat_message.is_pin = true\n UNION\n (\n SELECT chat_message.id\n FROM chat_message\n WHERE chat_message.chat_session_uuid = $1 and chat_message.is_deleted = false -- and chat_message.is_pin = false\n ORDER BY created_at DESC\n LIMIT $2\n )\n)\nORDER BY created_at\n`\n\ntype GetLatestMessagesBySessionUUIDParams struct {\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\tLimit int32 `json:\"limit\"`\n}\n\nfunc (q *Queries) GetLatestMessagesBySessionUUID(ctx context.Context, arg GetLatestMessagesBySessionUUIDParams) ([]ChatMessage, error) {\n\trows, err := q.db.QueryContext(ctx, getLatestMessagesBySessionUUID, arg.ChatSessionUuid, arg.Limit)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatMessage\n\tfor rows.Next() {\n\t\tvar i ChatMessage\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Uuid,\n\t\t\t&i.ChatSessionUuid,\n\t\t\t&i.Role,\n\t\t\t&i.Content,\n\t\t\t&i.ReasoningContent,\n\t\t\t&i.Model,\n\t\t\t&i.LlmSummary,\n\t\t\t&i.Score,\n\t\t\t&i.UserID,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.CreatedBy,\n\t\t\t&i.UpdatedBy,\n\t\t\t&i.IsDeleted,\n\t\t\t&i.IsPin,\n\t\t\t&i.TokenCount,\n\t\t\t&i.Raw,\n\t\t\t&i.Artifacts,\n\t\t\t&i.SuggestedQuestions,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getLatestUsageTimeOfModel = `-- name: GetLatestUsageTimeOfModel :many\nSELECT \n model,\n MAX(created_at)::timestamp as latest_message_time,\n COUNT(*) as message_count\nFROM chat_message\nWHERE \n created_at >= NOW() - $1::text::INTERVAL\n AND is_deleted = false\n AND model != ''\n AND role = 'assistant'\nGROUP BY model\nORDER BY latest_message_time DESC\n`\n\ntype GetLatestUsageTimeOfModelRow struct {\n\tModel string `json:\"model\"`\n\tLatestMessageTime time.Time `json:\"latestMessageTime\"`\n\tMessageCount int64 `json:\"messageCount\"`\n}\n\nfunc (q *Queries) GetLatestUsageTimeOfModel(ctx context.Context, timeInterval string) ([]GetLatestUsageTimeOfModelRow, error) {\n\trows, err := q.db.QueryContext(ctx, getLatestUsageTimeOfModel, timeInterval)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetLatestUsageTimeOfModelRow\n\tfor rows.Next() {\n\t\tvar i GetLatestUsageTimeOfModelRow\n\t\tif err := rows.Scan(&i.Model, &i.LatestMessageTime, &i.MessageCount); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst hasChatMessagePermission = `-- name: HasChatMessagePermission :one\nSELECT COUNT(*) > 0 as has_permission\nFROM chat_message cm\nINNER JOIN chat_session cs ON cm.chat_session_uuid = cs.uuid\nINNER JOIN auth_user au ON cs.user_id = au.id\nWHERE cm.is_deleted = false and cm.id = $1 AND (cs.user_id = $2 OR au.is_superuser) and cs.active = true\n`\n\ntype HasChatMessagePermissionParams struct {\n\tID int32 `json:\"id\"`\n\tUserID int32 `json:\"userId\"`\n}\n\nfunc (q *Queries) HasChatMessagePermission(ctx context.Context, arg HasChatMessagePermissionParams) (bool, error) {\n\trow := q.db.QueryRowContext(ctx, hasChatMessagePermission, arg.ID, arg.UserID)\n\tvar has_permission bool\n\terr := row.Scan(&has_permission)\n\treturn has_permission, err\n}\n\nconst updateChatMessage = `-- name: UpdateChatMessage :one\nUPDATE chat_message SET role = $2, content = $3, score = $4, user_id = $5, updated_by = $6, artifacts = $7, suggested_questions = $8, updated_at = now()\nWHERE id = $1\nRETURNING id, uuid, chat_session_uuid, role, content, reasoning_content, model, llm_summary, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, is_pin, token_count, raw, artifacts, suggested_questions\n`\n\ntype UpdateChatMessageParams struct {\n\tID int32 `json:\"id\"`\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n\tScore float64 `json:\"score\"`\n\tUserID int32 `json:\"userId\"`\n\tUpdatedBy int32 `json:\"updatedBy\"`\n\tArtifacts json.RawMessage `json:\"artifacts\"`\n\tSuggestedQuestions json.RawMessage `json:\"suggestedQuestions\"`\n}\n\nfunc (q *Queries) UpdateChatMessage(ctx context.Context, arg UpdateChatMessageParams) (ChatMessage, error) {\n\trow := q.db.QueryRowContext(ctx, updateChatMessage,\n\t\targ.ID,\n\t\targ.Role,\n\t\targ.Content,\n\t\targ.Score,\n\t\targ.UserID,\n\t\targ.UpdatedBy,\n\t\targ.Artifacts,\n\t\targ.SuggestedQuestions,\n\t)\n\tvar i ChatMessage\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.ReasoningContent,\n\t\t&i.Model,\n\t\t&i.LlmSummary,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.IsPin,\n\t\t&i.TokenCount,\n\t\t&i.Raw,\n\t\t&i.Artifacts,\n\t\t&i.SuggestedQuestions,\n\t)\n\treturn i, err\n}\n\nconst updateChatMessageByUUID = `-- name: UpdateChatMessageByUUID :one\nUPDATE chat_message SET content = $2, is_pin = $3, token_count = $4, artifacts = $5, suggested_questions = $6, updated_at = now() \nWHERE uuid = $1\nRETURNING id, uuid, chat_session_uuid, role, content, reasoning_content, model, llm_summary, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, is_pin, token_count, raw, artifacts, suggested_questions\n`\n\ntype UpdateChatMessageByUUIDParams struct {\n\tUuid string `json:\"uuid\"`\n\tContent string `json:\"content\"`\n\tIsPin bool `json:\"isPin\"`\n\tTokenCount int32 `json:\"tokenCount\"`\n\tArtifacts json.RawMessage `json:\"artifacts\"`\n\tSuggestedQuestions json.RawMessage `json:\"suggestedQuestions\"`\n}\n\nfunc (q *Queries) UpdateChatMessageByUUID(ctx context.Context, arg UpdateChatMessageByUUIDParams) (ChatMessage, error) {\n\trow := q.db.QueryRowContext(ctx, updateChatMessageByUUID,\n\t\targ.Uuid,\n\t\targ.Content,\n\t\targ.IsPin,\n\t\targ.TokenCount,\n\t\targ.Artifacts,\n\t\targ.SuggestedQuestions,\n\t)\n\tvar i ChatMessage\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.ReasoningContent,\n\t\t&i.Model,\n\t\t&i.LlmSummary,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.IsPin,\n\t\t&i.TokenCount,\n\t\t&i.Raw,\n\t\t&i.Artifacts,\n\t\t&i.SuggestedQuestions,\n\t)\n\treturn i, err\n}\n\nconst updateChatMessageContent = `-- name: UpdateChatMessageContent :exec\nUPDATE chat_message\nSET content = $2, updated_at = now(), token_count = $3\nWHERE uuid = $1\n`\n\ntype UpdateChatMessageContentParams struct {\n\tUuid string `json:\"uuid\"`\n\tContent string `json:\"content\"`\n\tTokenCount int32 `json:\"tokenCount\"`\n}\n\nfunc (q *Queries) UpdateChatMessageContent(ctx context.Context, arg UpdateChatMessageContentParams) error {\n\t_, err := q.db.ExecContext(ctx, updateChatMessageContent, arg.Uuid, arg.Content, arg.TokenCount)\n\treturn err\n}\n\nconst updateChatMessageSuggestions = `-- name: UpdateChatMessageSuggestions :one\nUPDATE chat_message \nSET suggested_questions = $2, updated_at = now() \nWHERE uuid = $1\nRETURNING id, uuid, chat_session_uuid, role, content, reasoning_content, model, llm_summary, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, is_pin, token_count, raw, artifacts, suggested_questions\n`\n\ntype UpdateChatMessageSuggestionsParams struct {\n\tUuid string `json:\"uuid\"`\n\tSuggestedQuestions json.RawMessage `json:\"suggestedQuestions\"`\n}\n\nfunc (q *Queries) UpdateChatMessageSuggestions(ctx context.Context, arg UpdateChatMessageSuggestionsParams) (ChatMessage, error) {\n\trow := q.db.QueryRowContext(ctx, updateChatMessageSuggestions, arg.Uuid, arg.SuggestedQuestions)\n\tvar i ChatMessage\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.ReasoningContent,\n\t\t&i.Model,\n\t\t&i.LlmSummary,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.IsPin,\n\t\t&i.TokenCount,\n\t\t&i.Raw,\n\t\t&i.Artifacts,\n\t\t&i.SuggestedQuestions,\n\t)\n\treturn i, err\n}\n" }, { "path": "api/sqlc_queries/chat_model.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: chat_model.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n)\n\nconst chatModelByID = `-- name: ChatModelByID :one\nSELECT id, name, label, is_default, url, api_auth_header, api_auth_key, user_id, enable_per_mode_ratelimit, max_token, default_token, order_number, http_time_out, is_enable, api_type FROM chat_model WHERE id = $1\n`\n\nfunc (q *Queries) ChatModelByID(ctx context.Context, id int32) (ChatModel, error) {\n\trow := q.db.QueryRowContext(ctx, chatModelByID, id)\n\tvar i ChatModel\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Name,\n\t\t&i.Label,\n\t\t&i.IsDefault,\n\t\t&i.Url,\n\t\t&i.ApiAuthHeader,\n\t\t&i.ApiAuthKey,\n\t\t&i.UserID,\n\t\t&i.EnablePerModeRatelimit,\n\t\t&i.MaxToken,\n\t\t&i.DefaultToken,\n\t\t&i.OrderNumber,\n\t\t&i.HttpTimeOut,\n\t\t&i.IsEnable,\n\t\t&i.ApiType,\n\t)\n\treturn i, err\n}\n\nconst chatModelByName = `-- name: ChatModelByName :one\nSELECT id, name, label, is_default, url, api_auth_header, api_auth_key, user_id, enable_per_mode_ratelimit, max_token, default_token, order_number, http_time_out, is_enable, api_type FROM chat_model WHERE name = $1\n`\n\nfunc (q *Queries) ChatModelByName(ctx context.Context, name string) (ChatModel, error) {\n\trow := q.db.QueryRowContext(ctx, chatModelByName, name)\n\tvar i ChatModel\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Name,\n\t\t&i.Label,\n\t\t&i.IsDefault,\n\t\t&i.Url,\n\t\t&i.ApiAuthHeader,\n\t\t&i.ApiAuthKey,\n\t\t&i.UserID,\n\t\t&i.EnablePerModeRatelimit,\n\t\t&i.MaxToken,\n\t\t&i.DefaultToken,\n\t\t&i.OrderNumber,\n\t\t&i.HttpTimeOut,\n\t\t&i.IsEnable,\n\t\t&i.ApiType,\n\t)\n\treturn i, err\n}\n\nconst createChatModel = `-- name: CreateChatModel :one\nINSERT INTO chat_model (name, label, is_default, url, api_auth_header, api_auth_key, user_id, enable_per_mode_ratelimit, max_token, default_token, order_number, http_time_out, api_type )\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13)\nRETURNING id, name, label, is_default, url, api_auth_header, api_auth_key, user_id, enable_per_mode_ratelimit, max_token, default_token, order_number, http_time_out, is_enable, api_type\n`\n\ntype CreateChatModelParams struct {\n\tName string `json:\"name\"`\n\tLabel string `json:\"label\"`\n\tIsDefault bool `json:\"isDefault\"`\n\tUrl string `json:\"url\"`\n\tApiAuthHeader string `json:\"apiAuthHeader\"`\n\tApiAuthKey string `json:\"apiAuthKey\"`\n\tUserID int32 `json:\"userId\"`\n\tEnablePerModeRatelimit bool `json:\"enablePerModeRatelimit\"`\n\tMaxToken int32 `json:\"maxToken\"`\n\tDefaultToken int32 `json:\"defaultToken\"`\n\tOrderNumber int32 `json:\"orderNumber\"`\n\tHttpTimeOut int32 `json:\"httpTimeOut\"`\n\tApiType string `json:\"apiType\"`\n}\n\nfunc (q *Queries) CreateChatModel(ctx context.Context, arg CreateChatModelParams) (ChatModel, error) {\n\trow := q.db.QueryRowContext(ctx, createChatModel,\n\t\targ.Name,\n\t\targ.Label,\n\t\targ.IsDefault,\n\t\targ.Url,\n\t\targ.ApiAuthHeader,\n\t\targ.ApiAuthKey,\n\t\targ.UserID,\n\t\targ.EnablePerModeRatelimit,\n\t\targ.MaxToken,\n\t\targ.DefaultToken,\n\t\targ.OrderNumber,\n\t\targ.HttpTimeOut,\n\t\targ.ApiType,\n\t)\n\tvar i ChatModel\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Name,\n\t\t&i.Label,\n\t\t&i.IsDefault,\n\t\t&i.Url,\n\t\t&i.ApiAuthHeader,\n\t\t&i.ApiAuthKey,\n\t\t&i.UserID,\n\t\t&i.EnablePerModeRatelimit,\n\t\t&i.MaxToken,\n\t\t&i.DefaultToken,\n\t\t&i.OrderNumber,\n\t\t&i.HttpTimeOut,\n\t\t&i.IsEnable,\n\t\t&i.ApiType,\n\t)\n\treturn i, err\n}\n\nconst deleteChatModel = `-- name: DeleteChatModel :exec\nDELETE FROM chat_model WHERE id = $1 and user_id = $2\n`\n\ntype DeleteChatModelParams struct {\n\tID int32 `json:\"id\"`\n\tUserID int32 `json:\"userId\"`\n}\n\nfunc (q *Queries) DeleteChatModel(ctx context.Context, arg DeleteChatModelParams) error {\n\t_, err := q.db.ExecContext(ctx, deleteChatModel, arg.ID, arg.UserID)\n\treturn err\n}\n\nconst getDefaultChatModel = `-- name: GetDefaultChatModel :one\nSELECT id, name, label, is_default, url, api_auth_header, api_auth_key, user_id, enable_per_mode_ratelimit, max_token, default_token, order_number, http_time_out, is_enable, api_type FROM chat_model WHERE is_default = true\nand user_id in (select id from auth_user where is_superuser = true)\nORDER BY order_number, id\nLIMIT 1\n`\n\nfunc (q *Queries) GetDefaultChatModel(ctx context.Context) (ChatModel, error) {\n\trow := q.db.QueryRowContext(ctx, getDefaultChatModel)\n\tvar i ChatModel\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Name,\n\t\t&i.Label,\n\t\t&i.IsDefault,\n\t\t&i.Url,\n\t\t&i.ApiAuthHeader,\n\t\t&i.ApiAuthKey,\n\t\t&i.UserID,\n\t\t&i.EnablePerModeRatelimit,\n\t\t&i.MaxToken,\n\t\t&i.DefaultToken,\n\t\t&i.OrderNumber,\n\t\t&i.HttpTimeOut,\n\t\t&i.IsEnable,\n\t\t&i.ApiType,\n\t)\n\treturn i, err\n}\n\nconst listChatModels = `-- name: ListChatModels :many\nSELECT id, name, label, is_default, url, api_auth_header, api_auth_key, user_id, enable_per_mode_ratelimit, max_token, default_token, order_number, http_time_out, is_enable, api_type FROM chat_model ORDER BY order_number\n`\n\nfunc (q *Queries) ListChatModels(ctx context.Context) ([]ChatModel, error) {\n\trows, err := q.db.QueryContext(ctx, listChatModels)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatModel\n\tfor rows.Next() {\n\t\tvar i ChatModel\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Name,\n\t\t\t&i.Label,\n\t\t\t&i.IsDefault,\n\t\t\t&i.Url,\n\t\t\t&i.ApiAuthHeader,\n\t\t\t&i.ApiAuthKey,\n\t\t\t&i.UserID,\n\t\t\t&i.EnablePerModeRatelimit,\n\t\t\t&i.MaxToken,\n\t\t\t&i.DefaultToken,\n\t\t\t&i.OrderNumber,\n\t\t\t&i.HttpTimeOut,\n\t\t\t&i.IsEnable,\n\t\t\t&i.ApiType,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst listSystemChatModels = `-- name: ListSystemChatModels :many\nSELECT id, name, label, is_default, url, api_auth_header, api_auth_key, user_id, enable_per_mode_ratelimit, max_token, default_token, order_number, http_time_out, is_enable, api_type FROM chat_model\nwhere user_id in (select id from auth_user where is_superuser = true)\nORDER BY order_number, id desc\n`\n\nfunc (q *Queries) ListSystemChatModels(ctx context.Context) ([]ChatModel, error) {\n\trows, err := q.db.QueryContext(ctx, listSystemChatModels)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatModel\n\tfor rows.Next() {\n\t\tvar i ChatModel\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Name,\n\t\t\t&i.Label,\n\t\t\t&i.IsDefault,\n\t\t\t&i.Url,\n\t\t\t&i.ApiAuthHeader,\n\t\t\t&i.ApiAuthKey,\n\t\t\t&i.UserID,\n\t\t\t&i.EnablePerModeRatelimit,\n\t\t\t&i.MaxToken,\n\t\t\t&i.DefaultToken,\n\t\t\t&i.OrderNumber,\n\t\t\t&i.HttpTimeOut,\n\t\t\t&i.IsEnable,\n\t\t\t&i.ApiType,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst updateChatModel = `-- name: UpdateChatModel :one\nUPDATE chat_model SET name = $2, label = $3, is_default = $4, url = $5, api_auth_header = $6, api_auth_key = $7, enable_per_mode_ratelimit = $9,\nmax_token = $10, default_token = $11, order_number = $12, http_time_out = $13, is_enable = $14, api_type = $15\nWHERE id = $1 and user_id = $8\nRETURNING id, name, label, is_default, url, api_auth_header, api_auth_key, user_id, enable_per_mode_ratelimit, max_token, default_token, order_number, http_time_out, is_enable, api_type\n`\n\ntype UpdateChatModelParams struct {\n\tID int32 `json:\"id\"`\n\tName string `json:\"name\"`\n\tLabel string `json:\"label\"`\n\tIsDefault bool `json:\"isDefault\"`\n\tUrl string `json:\"url\"`\n\tApiAuthHeader string `json:\"apiAuthHeader\"`\n\tApiAuthKey string `json:\"apiAuthKey\"`\n\tUserID int32 `json:\"userId\"`\n\tEnablePerModeRatelimit bool `json:\"enablePerModeRatelimit\"`\n\tMaxToken int32 `json:\"maxToken\"`\n\tDefaultToken int32 `json:\"defaultToken\"`\n\tOrderNumber int32 `json:\"orderNumber\"`\n\tHttpTimeOut int32 `json:\"httpTimeOut\"`\n\tIsEnable bool `json:\"isEnable\"`\n\tApiType string `json:\"apiType\"`\n}\n\nfunc (q *Queries) UpdateChatModel(ctx context.Context, arg UpdateChatModelParams) (ChatModel, error) {\n\trow := q.db.QueryRowContext(ctx, updateChatModel,\n\t\targ.ID,\n\t\targ.Name,\n\t\targ.Label,\n\t\targ.IsDefault,\n\t\targ.Url,\n\t\targ.ApiAuthHeader,\n\t\targ.ApiAuthKey,\n\t\targ.UserID,\n\t\targ.EnablePerModeRatelimit,\n\t\targ.MaxToken,\n\t\targ.DefaultToken,\n\t\targ.OrderNumber,\n\t\targ.HttpTimeOut,\n\t\targ.IsEnable,\n\t\targ.ApiType,\n\t)\n\tvar i ChatModel\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Name,\n\t\t&i.Label,\n\t\t&i.IsDefault,\n\t\t&i.Url,\n\t\t&i.ApiAuthHeader,\n\t\t&i.ApiAuthKey,\n\t\t&i.UserID,\n\t\t&i.EnablePerModeRatelimit,\n\t\t&i.MaxToken,\n\t\t&i.DefaultToken,\n\t\t&i.OrderNumber,\n\t\t&i.HttpTimeOut,\n\t\t&i.IsEnable,\n\t\t&i.ApiType,\n\t)\n\treturn i, err\n}\n\nconst updateChatModelKey = `-- name: UpdateChatModelKey :one\nUPDATE chat_model SET api_auth_key = $2\nWHERE id = $1\nRETURNING id, name, label, is_default, url, api_auth_header, api_auth_key, user_id, enable_per_mode_ratelimit, max_token, default_token, order_number, http_time_out, is_enable, api_type\n`\n\ntype UpdateChatModelKeyParams struct {\n\tID int32 `json:\"id\"`\n\tApiAuthKey string `json:\"apiAuthKey\"`\n}\n\nfunc (q *Queries) UpdateChatModelKey(ctx context.Context, arg UpdateChatModelKeyParams) (ChatModel, error) {\n\trow := q.db.QueryRowContext(ctx, updateChatModelKey, arg.ID, arg.ApiAuthKey)\n\tvar i ChatModel\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Name,\n\t\t&i.Label,\n\t\t&i.IsDefault,\n\t\t&i.Url,\n\t\t&i.ApiAuthHeader,\n\t\t&i.ApiAuthKey,\n\t\t&i.UserID,\n\t\t&i.EnablePerModeRatelimit,\n\t\t&i.MaxToken,\n\t\t&i.DefaultToken,\n\t\t&i.OrderNumber,\n\t\t&i.HttpTimeOut,\n\t\t&i.IsEnable,\n\t\t&i.ApiType,\n\t)\n\treturn i, err\n}\n" }, { "path": "api/sqlc_queries/chat_prompt.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: chat_prompt.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n)\n\nconst createChatPrompt = `-- name: CreateChatPrompt :one\nINSERT INTO chat_prompt (uuid, chat_session_uuid, role, content, token_count, user_id, created_by, updated_by)\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8)\nRETURNING id, uuid, chat_session_uuid, role, content, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, token_count\n`\n\ntype CreateChatPromptParams struct {\n\tUuid string `json:\"uuid\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n\tTokenCount int32 `json:\"tokenCount\"`\n\tUserID int32 `json:\"userId\"`\n\tCreatedBy int32 `json:\"createdBy\"`\n\tUpdatedBy int32 `json:\"updatedBy\"`\n}\n\nfunc (q *Queries) CreateChatPrompt(ctx context.Context, arg CreateChatPromptParams) (ChatPrompt, error) {\n\trow := q.db.QueryRowContext(ctx, createChatPrompt,\n\t\targ.Uuid,\n\t\targ.ChatSessionUuid,\n\t\targ.Role,\n\t\targ.Content,\n\t\targ.TokenCount,\n\t\targ.UserID,\n\t\targ.CreatedBy,\n\t\targ.UpdatedBy,\n\t)\n\tvar i ChatPrompt\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.TokenCount,\n\t)\n\treturn i, err\n}\n\nconst deleteChatPrompt = `-- name: DeleteChatPrompt :exec\nUPDATE chat_prompt \nSET is_deleted = true, updated_at = now()\nWHERE id = $1\n`\n\nfunc (q *Queries) DeleteChatPrompt(ctx context.Context, id int32) error {\n\t_, err := q.db.ExecContext(ctx, deleteChatPrompt, id)\n\treturn err\n}\n\nconst deleteChatPromptByUUID = `-- name: DeleteChatPromptByUUID :exec\nUPDATE chat_prompt\nSET is_deleted = true, updated_at = now()\nWHERE uuid = $1\n`\n\nfunc (q *Queries) DeleteChatPromptByUUID(ctx context.Context, uuid string) error {\n\t_, err := q.db.ExecContext(ctx, deleteChatPromptByUUID, uuid)\n\treturn err\n}\n\nconst getAllChatPrompts = `-- name: GetAllChatPrompts :many\nSELECT id, uuid, chat_session_uuid, role, content, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, token_count FROM chat_prompt \nWHERE is_deleted = false\nORDER BY id\n`\n\nfunc (q *Queries) GetAllChatPrompts(ctx context.Context) ([]ChatPrompt, error) {\n\trows, err := q.db.QueryContext(ctx, getAllChatPrompts)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatPrompt\n\tfor rows.Next() {\n\t\tvar i ChatPrompt\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Uuid,\n\t\t\t&i.ChatSessionUuid,\n\t\t\t&i.Role,\n\t\t\t&i.Content,\n\t\t\t&i.Score,\n\t\t\t&i.UserID,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.CreatedBy,\n\t\t\t&i.UpdatedBy,\n\t\t\t&i.IsDeleted,\n\t\t\t&i.TokenCount,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getChatPromptByID = `-- name: GetChatPromptByID :one\nSELECT id, uuid, chat_session_uuid, role, content, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, token_count FROM chat_prompt\nWHERE is_deleted = false and id = $1\n`\n\nfunc (q *Queries) GetChatPromptByID(ctx context.Context, id int32) (ChatPrompt, error) {\n\trow := q.db.QueryRowContext(ctx, getChatPromptByID, id)\n\tvar i ChatPrompt\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.TokenCount,\n\t)\n\treturn i, err\n}\n\nconst getChatPromptByUUID = `-- name: GetChatPromptByUUID :one\nSELECT id, uuid, chat_session_uuid, role, content, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, token_count FROM chat_prompt\nWHERE uuid = $1\n`\n\nfunc (q *Queries) GetChatPromptByUUID(ctx context.Context, uuid string) (ChatPrompt, error) {\n\trow := q.db.QueryRowContext(ctx, getChatPromptByUUID, uuid)\n\tvar i ChatPrompt\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.TokenCount,\n\t)\n\treturn i, err\n}\n\nconst getChatPromptsBySessionUUID = `-- name: GetChatPromptsBySessionUUID :many\nSELECT id, uuid, chat_session_uuid, role, content, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, token_count\nFROM chat_prompt \nWHERE chat_session_uuid = $1 and is_deleted = false\nORDER BY id\n`\n\nfunc (q *Queries) GetChatPromptsBySessionUUID(ctx context.Context, chatSessionUuid string) ([]ChatPrompt, error) {\n\trows, err := q.db.QueryContext(ctx, getChatPromptsBySessionUUID, chatSessionUuid)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatPrompt\n\tfor rows.Next() {\n\t\tvar i ChatPrompt\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Uuid,\n\t\t\t&i.ChatSessionUuid,\n\t\t\t&i.Role,\n\t\t\t&i.Content,\n\t\t\t&i.Score,\n\t\t\t&i.UserID,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.CreatedBy,\n\t\t\t&i.UpdatedBy,\n\t\t\t&i.IsDeleted,\n\t\t\t&i.TokenCount,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getChatPromptsByUserID = `-- name: GetChatPromptsByUserID :many\nSELECT id, uuid, chat_session_uuid, role, content, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, token_count\nFROM chat_prompt \nWHERE user_id = $1 and is_deleted = false\nORDER BY id\n`\n\nfunc (q *Queries) GetChatPromptsByUserID(ctx context.Context, userID int32) ([]ChatPrompt, error) {\n\trows, err := q.db.QueryContext(ctx, getChatPromptsByUserID, userID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatPrompt\n\tfor rows.Next() {\n\t\tvar i ChatPrompt\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Uuid,\n\t\t\t&i.ChatSessionUuid,\n\t\t\t&i.Role,\n\t\t\t&i.Content,\n\t\t\t&i.Score,\n\t\t\t&i.UserID,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.CreatedBy,\n\t\t\t&i.UpdatedBy,\n\t\t\t&i.IsDeleted,\n\t\t\t&i.TokenCount,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getChatPromptsBysession_uuid = `-- name: GetChatPromptsBysession_uuid :many\nSELECT id, uuid, chat_session_uuid, role, content, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, token_count\nFROM chat_prompt \nWHERE chat_session_uuid = $1 and is_deleted = false\nORDER BY id\n`\n\nfunc (q *Queries) GetChatPromptsBysession_uuid(ctx context.Context, chatSessionUuid string) ([]ChatPrompt, error) {\n\trows, err := q.db.QueryContext(ctx, getChatPromptsBysession_uuid, chatSessionUuid)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatPrompt\n\tfor rows.Next() {\n\t\tvar i ChatPrompt\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Uuid,\n\t\t\t&i.ChatSessionUuid,\n\t\t\t&i.Role,\n\t\t\t&i.Content,\n\t\t\t&i.Score,\n\t\t\t&i.UserID,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.CreatedBy,\n\t\t\t&i.UpdatedBy,\n\t\t\t&i.IsDeleted,\n\t\t\t&i.TokenCount,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getOneChatPromptBySessionUUID = `-- name: GetOneChatPromptBySessionUUID :one\nSELECT id, uuid, chat_session_uuid, role, content, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, token_count\nFROM chat_prompt \nWHERE chat_session_uuid = $1 and is_deleted = false\nORDER BY id\nLIMIT 1\n`\n\nfunc (q *Queries) GetOneChatPromptBySessionUUID(ctx context.Context, chatSessionUuid string) (ChatPrompt, error) {\n\trow := q.db.QueryRowContext(ctx, getOneChatPromptBySessionUUID, chatSessionUuid)\n\tvar i ChatPrompt\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.TokenCount,\n\t)\n\treturn i, err\n}\n\nconst hasChatPromptPermission = `-- name: HasChatPromptPermission :one\nSELECT COUNT(*) > 0 as has_permission\nFROM chat_prompt cp\nINNER JOIN auth_user au ON cp.user_id = au.id\nWHERE cp.id = $1 AND (cp.user_id = $2 OR au.is_superuser) AND cp.is_deleted = false\n`\n\ntype HasChatPromptPermissionParams struct {\n\tID int32 `json:\"id\"`\n\tUserID int32 `json:\"userId\"`\n}\n\nfunc (q *Queries) HasChatPromptPermission(ctx context.Context, arg HasChatPromptPermissionParams) (bool, error) {\n\trow := q.db.QueryRowContext(ctx, hasChatPromptPermission, arg.ID, arg.UserID)\n\tvar has_permission bool\n\terr := row.Scan(&has_permission)\n\treturn has_permission, err\n}\n\nconst updateChatPrompt = `-- name: UpdateChatPrompt :one\nUPDATE chat_prompt SET chat_session_uuid = $2, role = $3, content = $4, score = $5, user_id = $6, updated_at = now(), updated_by = $7\nWHERE id = $1\nRETURNING id, uuid, chat_session_uuid, role, content, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, token_count\n`\n\ntype UpdateChatPromptParams struct {\n\tID int32 `json:\"id\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n\tScore float64 `json:\"score\"`\n\tUserID int32 `json:\"userId\"`\n\tUpdatedBy int32 `json:\"updatedBy\"`\n}\n\nfunc (q *Queries) UpdateChatPrompt(ctx context.Context, arg UpdateChatPromptParams) (ChatPrompt, error) {\n\trow := q.db.QueryRowContext(ctx, updateChatPrompt,\n\t\targ.ID,\n\t\targ.ChatSessionUuid,\n\t\targ.Role,\n\t\targ.Content,\n\t\targ.Score,\n\t\targ.UserID,\n\t\targ.UpdatedBy,\n\t)\n\tvar i ChatPrompt\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.TokenCount,\n\t)\n\treturn i, err\n}\n\nconst updateChatPromptByUUID = `-- name: UpdateChatPromptByUUID :one\nUPDATE chat_prompt SET content = $2, token_count = $3, updated_at = now()\nWHERE uuid = $1 and is_deleted = false\nRETURNING id, uuid, chat_session_uuid, role, content, score, user_id, created_at, updated_at, created_by, updated_by, is_deleted, token_count\n`\n\ntype UpdateChatPromptByUUIDParams struct {\n\tUuid string `json:\"uuid\"`\n\tContent string `json:\"content\"`\n\tTokenCount int32 `json:\"tokenCount\"`\n}\n\nfunc (q *Queries) UpdateChatPromptByUUID(ctx context.Context, arg UpdateChatPromptByUUIDParams) (ChatPrompt, error) {\n\trow := q.db.QueryRowContext(ctx, updateChatPromptByUUID, arg.Uuid, arg.Content, arg.TokenCount)\n\tvar i ChatPrompt\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.ChatSessionUuid,\n\t\t&i.Role,\n\t\t&i.Content,\n\t\t&i.Score,\n\t\t&i.UserID,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t\t&i.IsDeleted,\n\t\t&i.TokenCount,\n\t)\n\treturn i, err\n}\n" }, { "path": "api/sqlc_queries/chat_session.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: chat_session.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"time\"\n)\n\nconst createChatSession = `-- name: CreateChatSession :one\nINSERT INTO chat_session (user_id, topic, max_length, uuid, model)\nVALUES ($1, $2, $3, $4, $5)\nRETURNING id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode\n`\n\ntype CreateChatSessionParams struct {\n\tUserID int32 `json:\"userId\"`\n\tTopic string `json:\"topic\"`\n\tMaxLength int32 `json:\"maxLength\"`\n\tUuid string `json:\"uuid\"`\n\tModel string `json:\"model\"`\n}\n\nfunc (q *Queries) CreateChatSession(ctx context.Context, arg CreateChatSessionParams) (ChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, createChatSession,\n\t\targ.UserID,\n\t\targ.Topic,\n\t\targ.MaxLength,\n\t\targ.Uuid,\n\t\targ.Model,\n\t)\n\tvar i ChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.Uuid,\n\t\t&i.Topic,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.Active,\n\t\t&i.Model,\n\t\t&i.MaxLength,\n\t\t&i.Temperature,\n\t\t&i.TopP,\n\t\t&i.MaxTokens,\n\t\t&i.N,\n\t\t&i.SummarizeMode,\n\t\t&i.WorkspaceID,\n\t\t&i.ArtifactEnabled,\n\t\t&i.Debug,\n\t\t&i.ExploreMode,\n\t)\n\treturn i, err\n}\n\nconst createChatSessionByUUID = `-- name: CreateChatSessionByUUID :one\nINSERT INTO chat_session (user_id, uuid, topic, created_at, active, max_length, model)\nVALUES ($1, $2, $3, $4, $5, $6, $7)\nRETURNING id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode\n`\n\ntype CreateChatSessionByUUIDParams struct {\n\tUserID int32 `json:\"userId\"`\n\tUuid string `json:\"uuid\"`\n\tTopic string `json:\"topic\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tActive bool `json:\"active\"`\n\tMaxLength int32 `json:\"maxLength\"`\n\tModel string `json:\"model\"`\n}\n\nfunc (q *Queries) CreateChatSessionByUUID(ctx context.Context, arg CreateChatSessionByUUIDParams) (ChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, createChatSessionByUUID,\n\t\targ.UserID,\n\t\targ.Uuid,\n\t\targ.Topic,\n\t\targ.CreatedAt,\n\t\targ.Active,\n\t\targ.MaxLength,\n\t\targ.Model,\n\t)\n\tvar i ChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.Uuid,\n\t\t&i.Topic,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.Active,\n\t\t&i.Model,\n\t\t&i.MaxLength,\n\t\t&i.Temperature,\n\t\t&i.TopP,\n\t\t&i.MaxTokens,\n\t\t&i.N,\n\t\t&i.SummarizeMode,\n\t\t&i.WorkspaceID,\n\t\t&i.ArtifactEnabled,\n\t\t&i.Debug,\n\t\t&i.ExploreMode,\n\t)\n\treturn i, err\n}\n\nconst createChatSessionInWorkspace = `-- name: CreateChatSessionInWorkspace :one\nINSERT INTO chat_session (user_id, uuid, topic, created_at, active, max_length, model, workspace_id)\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8)\nRETURNING id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode\n`\n\ntype CreateChatSessionInWorkspaceParams struct {\n\tUserID int32 `json:\"userId\"`\n\tUuid string `json:\"uuid\"`\n\tTopic string `json:\"topic\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tActive bool `json:\"active\"`\n\tMaxLength int32 `json:\"maxLength\"`\n\tModel string `json:\"model\"`\n\tWorkspaceID sql.NullInt32 `json:\"workspaceId\"`\n}\n\nfunc (q *Queries) CreateChatSessionInWorkspace(ctx context.Context, arg CreateChatSessionInWorkspaceParams) (ChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, createChatSessionInWorkspace,\n\t\targ.UserID,\n\t\targ.Uuid,\n\t\targ.Topic,\n\t\targ.CreatedAt,\n\t\targ.Active,\n\t\targ.MaxLength,\n\t\targ.Model,\n\t\targ.WorkspaceID,\n\t)\n\tvar i ChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.Uuid,\n\t\t&i.Topic,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.Active,\n\t\t&i.Model,\n\t\t&i.MaxLength,\n\t\t&i.Temperature,\n\t\t&i.TopP,\n\t\t&i.MaxTokens,\n\t\t&i.N,\n\t\t&i.SummarizeMode,\n\t\t&i.WorkspaceID,\n\t\t&i.ArtifactEnabled,\n\t\t&i.Debug,\n\t\t&i.ExploreMode,\n\t)\n\treturn i, err\n}\n\nconst createOrUpdateChatSessionByUUID = `-- name: CreateOrUpdateChatSessionByUUID :one\nINSERT INTO chat_session(uuid, user_id, topic, max_length, temperature, model, max_tokens, top_p, n, debug, summarize_mode, workspace_id, explore_mode, artifact_enabled)\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14)\nON CONFLICT (uuid) \nDO UPDATE SET\nmax_length = EXCLUDED.max_length, \ndebug = EXCLUDED.debug,\nmax_tokens = EXCLUDED.max_tokens,\ntemperature = EXCLUDED.temperature, \ntop_p = EXCLUDED.top_p,\nn= EXCLUDED.n,\nmodel = EXCLUDED.model,\nsummarize_mode = EXCLUDED.summarize_mode,\nartifact_enabled = EXCLUDED.artifact_enabled,\nworkspace_id = CASE WHEN EXCLUDED.workspace_id IS NOT NULL THEN EXCLUDED.workspace_id ELSE chat_session.workspace_id END,\ntopic = CASE WHEN chat_session.topic IS NULL THEN EXCLUDED.topic ELSE chat_session.topic END,\nexplore_mode = EXCLUDED.explore_mode,\nupdated_at = now()\nreturning id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode\n`\n\ntype CreateOrUpdateChatSessionByUUIDParams struct {\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n\tTopic string `json:\"topic\"`\n\tMaxLength int32 `json:\"maxLength\"`\n\tTemperature float64 `json:\"temperature\"`\n\tModel string `json:\"model\"`\n\tMaxTokens int32 `json:\"maxTokens\"`\n\tTopP float64 `json:\"topP\"`\n\tN int32 `json:\"n\"`\n\tDebug bool `json:\"debug\"`\n\tSummarizeMode bool `json:\"summarizeMode\"`\n\tWorkspaceID sql.NullInt32 `json:\"workspaceId\"`\n\tExploreMode bool `json:\"exploreMode\"`\n\tArtifactEnabled bool `json:\"artifactEnabled\"`\n}\n\nfunc (q *Queries) CreateOrUpdateChatSessionByUUID(ctx context.Context, arg CreateOrUpdateChatSessionByUUIDParams) (ChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, createOrUpdateChatSessionByUUID,\n\t\targ.Uuid,\n\t\targ.UserID,\n\t\targ.Topic,\n\t\targ.MaxLength,\n\t\targ.Temperature,\n\t\targ.Model,\n\t\targ.MaxTokens,\n\t\targ.TopP,\n\t\targ.N,\n\t\targ.Debug,\n\t\targ.SummarizeMode,\n\t\targ.WorkspaceID,\n\t\targ.ExploreMode,\n\t\targ.ArtifactEnabled,\n\t)\n\tvar i ChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.Uuid,\n\t\t&i.Topic,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.Active,\n\t\t&i.Model,\n\t\t&i.MaxLength,\n\t\t&i.Temperature,\n\t\t&i.TopP,\n\t\t&i.MaxTokens,\n\t\t&i.N,\n\t\t&i.SummarizeMode,\n\t\t&i.WorkspaceID,\n\t\t&i.ArtifactEnabled,\n\t\t&i.Debug,\n\t\t&i.ExploreMode,\n\t)\n\treturn i, err\n}\n\nconst deleteChatSession = `-- name: DeleteChatSession :exec\nDELETE FROM chat_session \nWHERE id = $1\n`\n\nfunc (q *Queries) DeleteChatSession(ctx context.Context, id int32) error {\n\t_, err := q.db.ExecContext(ctx, deleteChatSession, id)\n\treturn err\n}\n\nconst deleteChatSessionByUUID = `-- name: DeleteChatSessionByUUID :exec\nupdate chat_session set active = false\nWHERE uuid = $1\nreturning id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode\n`\n\nfunc (q *Queries) DeleteChatSessionByUUID(ctx context.Context, uuid string) error {\n\t_, err := q.db.ExecContext(ctx, deleteChatSessionByUUID, uuid)\n\treturn err\n}\n\nconst getAllChatSessions = `-- name: GetAllChatSessions :many\nSELECT id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode FROM chat_session \nwhere active = true\nORDER BY id\n`\n\nfunc (q *Queries) GetAllChatSessions(ctx context.Context) ([]ChatSession, error) {\n\trows, err := q.db.QueryContext(ctx, getAllChatSessions)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatSession\n\tfor rows.Next() {\n\t\tvar i ChatSession\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.UserID,\n\t\t\t&i.Uuid,\n\t\t\t&i.Topic,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.Active,\n\t\t\t&i.Model,\n\t\t\t&i.MaxLength,\n\t\t\t&i.Temperature,\n\t\t\t&i.TopP,\n\t\t\t&i.MaxTokens,\n\t\t\t&i.N,\n\t\t\t&i.SummarizeMode,\n\t\t\t&i.WorkspaceID,\n\t\t\t&i.ArtifactEnabled,\n\t\t\t&i.Debug,\n\t\t\t&i.ExploreMode,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getChatSessionByID = `-- name: GetChatSessionByID :one\nSELECT id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode FROM chat_session WHERE id = $1\n`\n\nfunc (q *Queries) GetChatSessionByID(ctx context.Context, id int32) (ChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, getChatSessionByID, id)\n\tvar i ChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.Uuid,\n\t\t&i.Topic,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.Active,\n\t\t&i.Model,\n\t\t&i.MaxLength,\n\t\t&i.Temperature,\n\t\t&i.TopP,\n\t\t&i.MaxTokens,\n\t\t&i.N,\n\t\t&i.SummarizeMode,\n\t\t&i.WorkspaceID,\n\t\t&i.ArtifactEnabled,\n\t\t&i.Debug,\n\t\t&i.ExploreMode,\n\t)\n\treturn i, err\n}\n\nconst getChatSessionByUUID = `-- name: GetChatSessionByUUID :one\nSELECT id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode FROM chat_session \nWHERE active = true and uuid = $1\norder by updated_at\n`\n\nfunc (q *Queries) GetChatSessionByUUID(ctx context.Context, uuid string) (ChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, getChatSessionByUUID, uuid)\n\tvar i ChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.Uuid,\n\t\t&i.Topic,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.Active,\n\t\t&i.Model,\n\t\t&i.MaxLength,\n\t\t&i.Temperature,\n\t\t&i.TopP,\n\t\t&i.MaxTokens,\n\t\t&i.N,\n\t\t&i.SummarizeMode,\n\t\t&i.WorkspaceID,\n\t\t&i.ArtifactEnabled,\n\t\t&i.Debug,\n\t\t&i.ExploreMode,\n\t)\n\treturn i, err\n}\n\nconst getChatSessionByUUIDWithInActive = `-- name: GetChatSessionByUUIDWithInActive :one\nSELECT id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode FROM chat_session \nWHERE uuid = $1\norder by updated_at\n`\n\nfunc (q *Queries) GetChatSessionByUUIDWithInActive(ctx context.Context, uuid string) (ChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, getChatSessionByUUIDWithInActive, uuid)\n\tvar i ChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.Uuid,\n\t\t&i.Topic,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.Active,\n\t\t&i.Model,\n\t\t&i.MaxLength,\n\t\t&i.Temperature,\n\t\t&i.TopP,\n\t\t&i.MaxTokens,\n\t\t&i.N,\n\t\t&i.SummarizeMode,\n\t\t&i.WorkspaceID,\n\t\t&i.ArtifactEnabled,\n\t\t&i.Debug,\n\t\t&i.ExploreMode,\n\t)\n\treturn i, err\n}\n\nconst getChatSessionsByUserID = `-- name: GetChatSessionsByUserID :many\nSELECT cs.id, cs.user_id, cs.uuid, cs.topic, cs.created_at, cs.updated_at, cs.active, cs.model, cs.max_length, cs.temperature, cs.top_p, cs.max_tokens, cs.n, cs.summarize_mode, cs.workspace_id, cs.artifact_enabled, cs.debug, cs.explore_mode\nFROM chat_session cs\nLEFT JOIN (\n SELECT chat_session_uuid, MAX(created_at) AS latest_message_time\n FROM chat_message\n GROUP BY chat_session_uuid\n) cm ON cs.uuid = cm.chat_session_uuid\nWHERE cs.user_id = $1 AND cs.active = true\nORDER BY \n cm.latest_message_time DESC,\n cs.id DESC\n`\n\nfunc (q *Queries) GetChatSessionsByUserID(ctx context.Context, userID int32) ([]ChatSession, error) {\n\trows, err := q.db.QueryContext(ctx, getChatSessionsByUserID, userID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatSession\n\tfor rows.Next() {\n\t\tvar i ChatSession\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.UserID,\n\t\t\t&i.Uuid,\n\t\t\t&i.Topic,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.Active,\n\t\t\t&i.Model,\n\t\t\t&i.MaxLength,\n\t\t\t&i.Temperature,\n\t\t\t&i.TopP,\n\t\t\t&i.MaxTokens,\n\t\t\t&i.N,\n\t\t\t&i.SummarizeMode,\n\t\t\t&i.WorkspaceID,\n\t\t\t&i.ArtifactEnabled,\n\t\t\t&i.Debug,\n\t\t\t&i.ExploreMode,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getSessionsByWorkspaceID = `-- name: GetSessionsByWorkspaceID :many\nSELECT cs.id, cs.user_id, cs.uuid, cs.topic, cs.created_at, cs.updated_at, cs.active, cs.model, cs.max_length, cs.temperature, cs.top_p, cs.max_tokens, cs.n, cs.summarize_mode, cs.workspace_id, cs.artifact_enabled, cs.debug, cs.explore_mode\nFROM chat_session cs\nLEFT JOIN (\n SELECT chat_session_uuid, MAX(created_at) AS latest_message_time\n FROM chat_message\n GROUP BY chat_session_uuid\n) cm ON cs.uuid = cm.chat_session_uuid\nWHERE cs.workspace_id = $1 AND cs.active = true\nORDER BY \n cm.latest_message_time DESC,\n cs.id DESC\n`\n\nfunc (q *Queries) GetSessionsByWorkspaceID(ctx context.Context, workspaceID sql.NullInt32) ([]ChatSession, error) {\n\trows, err := q.db.QueryContext(ctx, getSessionsByWorkspaceID, workspaceID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatSession\n\tfor rows.Next() {\n\t\tvar i ChatSession\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.UserID,\n\t\t\t&i.Uuid,\n\t\t\t&i.Topic,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.Active,\n\t\t\t&i.Model,\n\t\t\t&i.MaxLength,\n\t\t\t&i.Temperature,\n\t\t\t&i.TopP,\n\t\t\t&i.MaxTokens,\n\t\t\t&i.N,\n\t\t\t&i.SummarizeMode,\n\t\t\t&i.WorkspaceID,\n\t\t\t&i.ArtifactEnabled,\n\t\t\t&i.Debug,\n\t\t\t&i.ExploreMode,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getSessionsGroupedByWorkspace = `-- name: GetSessionsGroupedByWorkspace :many\nSELECT \n cs.id, cs.user_id, cs.uuid, cs.topic, cs.created_at, cs.updated_at, cs.active, cs.model, cs.max_length, cs.temperature, cs.top_p, cs.max_tokens, cs.n, cs.summarize_mode, cs.workspace_id, cs.artifact_enabled, cs.debug, cs.explore_mode,\n w.uuid as workspace_uuid,\n w.name as workspace_name,\n w.color as workspace_color,\n w.icon as workspace_icon\nFROM chat_session cs\nLEFT JOIN chat_workspace w ON cs.workspace_id = w.id\nLEFT JOIN (\n SELECT chat_session_uuid, MAX(created_at) AS latest_message_time\n FROM chat_message\n GROUP BY chat_session_uuid\n) cm ON cs.uuid = cm.chat_session_uuid\nWHERE cs.user_id = $1 AND cs.active = true\nORDER BY \n w.order_position ASC,\n cm.latest_message_time DESC,\n cs.id DESC\n`\n\ntype GetSessionsGroupedByWorkspaceRow struct {\n\tID int32 `json:\"id\"`\n\tUserID int32 `json:\"userId\"`\n\tUuid string `json:\"uuid\"`\n\tTopic string `json:\"topic\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tActive bool `json:\"active\"`\n\tModel string `json:\"model\"`\n\tMaxLength int32 `json:\"maxLength\"`\n\tTemperature float64 `json:\"temperature\"`\n\tTopP float64 `json:\"topP\"`\n\tMaxTokens int32 `json:\"maxTokens\"`\n\tN int32 `json:\"n\"`\n\tSummarizeMode bool `json:\"summarizeMode\"`\n\tWorkspaceID sql.NullInt32 `json:\"workspaceId\"`\n\tArtifactEnabled bool `json:\"artifactEnabled\"`\n\tDebug bool `json:\"debug\"`\n\tExploreMode bool `json:\"exploreMode\"`\n\tWorkspaceUuid sql.NullString `json:\"workspaceUuid\"`\n\tWorkspaceName sql.NullString `json:\"workspaceName\"`\n\tWorkspaceColor sql.NullString `json:\"workspaceColor\"`\n\tWorkspaceIcon sql.NullString `json:\"workspaceIcon\"`\n}\n\nfunc (q *Queries) GetSessionsGroupedByWorkspace(ctx context.Context, userID int32) ([]GetSessionsGroupedByWorkspaceRow, error) {\n\trows, err := q.db.QueryContext(ctx, getSessionsGroupedByWorkspace, userID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetSessionsGroupedByWorkspaceRow\n\tfor rows.Next() {\n\t\tvar i GetSessionsGroupedByWorkspaceRow\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.UserID,\n\t\t\t&i.Uuid,\n\t\t\t&i.Topic,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.Active,\n\t\t\t&i.Model,\n\t\t\t&i.MaxLength,\n\t\t\t&i.Temperature,\n\t\t\t&i.TopP,\n\t\t\t&i.MaxTokens,\n\t\t\t&i.N,\n\t\t\t&i.SummarizeMode,\n\t\t\t&i.WorkspaceID,\n\t\t\t&i.ArtifactEnabled,\n\t\t\t&i.Debug,\n\t\t\t&i.ExploreMode,\n\t\t\t&i.WorkspaceUuid,\n\t\t\t&i.WorkspaceName,\n\t\t\t&i.WorkspaceColor,\n\t\t\t&i.WorkspaceIcon,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getSessionsWithoutWorkspace = `-- name: GetSessionsWithoutWorkspace :many\nSELECT id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode FROM chat_session \nWHERE user_id = $1 AND workspace_id IS NULL AND active = true\n`\n\nfunc (q *Queries) GetSessionsWithoutWorkspace(ctx context.Context, userID int32) ([]ChatSession, error) {\n\trows, err := q.db.QueryContext(ctx, getSessionsWithoutWorkspace, userID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatSession\n\tfor rows.Next() {\n\t\tvar i ChatSession\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.UserID,\n\t\t\t&i.Uuid,\n\t\t\t&i.Topic,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.Active,\n\t\t\t&i.Model,\n\t\t\t&i.MaxLength,\n\t\t\t&i.Temperature,\n\t\t\t&i.TopP,\n\t\t\t&i.MaxTokens,\n\t\t\t&i.N,\n\t\t\t&i.SummarizeMode,\n\t\t\t&i.WorkspaceID,\n\t\t\t&i.ArtifactEnabled,\n\t\t\t&i.Debug,\n\t\t\t&i.ExploreMode,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst hasChatSessionPermission = `-- name: HasChatSessionPermission :one\n\nSELECT COUNT(*) > 0 as has_permission\nFROM chat_session cs\nINNER JOIN auth_user au ON cs.user_id = au.id\nWHERE cs.id = $1 AND (cs.user_id = $2 OR au.is_superuser)\n`\n\ntype HasChatSessionPermissionParams struct {\n\tID int32 `json:\"id\"`\n\tUserID int32 `json:\"userId\"`\n}\n\n// SELECT cs.*\n// FROM chat_session cs\n// WHERE cs.user_id = $1 and cs.active = true\n// ORDER BY cs.updated_at DESC;\nfunc (q *Queries) HasChatSessionPermission(ctx context.Context, arg HasChatSessionPermissionParams) (bool, error) {\n\trow := q.db.QueryRowContext(ctx, hasChatSessionPermission, arg.ID, arg.UserID)\n\tvar has_permission bool\n\terr := row.Scan(&has_permission)\n\treturn has_permission, err\n}\n\nconst migrateSessionsToDefaultWorkspace = `-- name: MigrateSessionsToDefaultWorkspace :exec\nUPDATE chat_session \nSET workspace_id = $2\nWHERE user_id = $1 AND workspace_id IS NULL\n`\n\ntype MigrateSessionsToDefaultWorkspaceParams struct {\n\tUserID int32 `json:\"userId\"`\n\tWorkspaceID sql.NullInt32 `json:\"workspaceId\"`\n}\n\nfunc (q *Queries) MigrateSessionsToDefaultWorkspace(ctx context.Context, arg MigrateSessionsToDefaultWorkspaceParams) error {\n\t_, err := q.db.ExecContext(ctx, migrateSessionsToDefaultWorkspace, arg.UserID, arg.WorkspaceID)\n\treturn err\n}\n\nconst updateChatSession = `-- name: UpdateChatSession :one\nUPDATE chat_session SET user_id = $2, topic = $3, updated_at = now(), active = $4\nWHERE id = $1\nRETURNING id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode\n`\n\ntype UpdateChatSessionParams struct {\n\tID int32 `json:\"id\"`\n\tUserID int32 `json:\"userId\"`\n\tTopic string `json:\"topic\"`\n\tActive bool `json:\"active\"`\n}\n\nfunc (q *Queries) UpdateChatSession(ctx context.Context, arg UpdateChatSessionParams) (ChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, updateChatSession,\n\t\targ.ID,\n\t\targ.UserID,\n\t\targ.Topic,\n\t\targ.Active,\n\t)\n\tvar i ChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.Uuid,\n\t\t&i.Topic,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.Active,\n\t\t&i.Model,\n\t\t&i.MaxLength,\n\t\t&i.Temperature,\n\t\t&i.TopP,\n\t\t&i.MaxTokens,\n\t\t&i.N,\n\t\t&i.SummarizeMode,\n\t\t&i.WorkspaceID,\n\t\t&i.ArtifactEnabled,\n\t\t&i.Debug,\n\t\t&i.ExploreMode,\n\t)\n\treturn i, err\n}\n\nconst updateChatSessionByUUID = `-- name: UpdateChatSessionByUUID :one\nUPDATE chat_session SET user_id = $2, topic = $3, updated_at = now()\nWHERE uuid = $1\nRETURNING id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode\n`\n\ntype UpdateChatSessionByUUIDParams struct {\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n\tTopic string `json:\"topic\"`\n}\n\nfunc (q *Queries) UpdateChatSessionByUUID(ctx context.Context, arg UpdateChatSessionByUUIDParams) (ChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, updateChatSessionByUUID, arg.Uuid, arg.UserID, arg.Topic)\n\tvar i ChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.Uuid,\n\t\t&i.Topic,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.Active,\n\t\t&i.Model,\n\t\t&i.MaxLength,\n\t\t&i.Temperature,\n\t\t&i.TopP,\n\t\t&i.MaxTokens,\n\t\t&i.N,\n\t\t&i.SummarizeMode,\n\t\t&i.WorkspaceID,\n\t\t&i.ArtifactEnabled,\n\t\t&i.Debug,\n\t\t&i.ExploreMode,\n\t)\n\treturn i, err\n}\n\nconst updateChatSessionTopicByUUID = `-- name: UpdateChatSessionTopicByUUID :one\nINSERT INTO chat_session(uuid, user_id, topic)\nVALUES ($1, $2, $3)\nON CONFLICT (uuid) \nDO UPDATE SET\ntopic = EXCLUDED.topic, \nupdated_at = now()\nreturning id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode\n`\n\ntype UpdateChatSessionTopicByUUIDParams struct {\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n\tTopic string `json:\"topic\"`\n}\n\nfunc (q *Queries) UpdateChatSessionTopicByUUID(ctx context.Context, arg UpdateChatSessionTopicByUUIDParams) (ChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, updateChatSessionTopicByUUID, arg.Uuid, arg.UserID, arg.Topic)\n\tvar i ChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.Uuid,\n\t\t&i.Topic,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.Active,\n\t\t&i.Model,\n\t\t&i.MaxLength,\n\t\t&i.Temperature,\n\t\t&i.TopP,\n\t\t&i.MaxTokens,\n\t\t&i.N,\n\t\t&i.SummarizeMode,\n\t\t&i.WorkspaceID,\n\t\t&i.ArtifactEnabled,\n\t\t&i.Debug,\n\t\t&i.ExploreMode,\n\t)\n\treturn i, err\n}\n\nconst updateSessionMaxLength = `-- name: UpdateSessionMaxLength :one\nUPDATE chat_session\nSET max_length = $2,\n updated_at = now()\nWHERE uuid = $1\nRETURNING id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode\n`\n\ntype UpdateSessionMaxLengthParams struct {\n\tUuid string `json:\"uuid\"`\n\tMaxLength int32 `json:\"maxLength\"`\n}\n\nfunc (q *Queries) UpdateSessionMaxLength(ctx context.Context, arg UpdateSessionMaxLengthParams) (ChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, updateSessionMaxLength, arg.Uuid, arg.MaxLength)\n\tvar i ChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.Uuid,\n\t\t&i.Topic,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.Active,\n\t\t&i.Model,\n\t\t&i.MaxLength,\n\t\t&i.Temperature,\n\t\t&i.TopP,\n\t\t&i.MaxTokens,\n\t\t&i.N,\n\t\t&i.SummarizeMode,\n\t\t&i.WorkspaceID,\n\t\t&i.ArtifactEnabled,\n\t\t&i.Debug,\n\t\t&i.ExploreMode,\n\t)\n\treturn i, err\n}\n\nconst updateSessionWorkspace = `-- name: UpdateSessionWorkspace :one\nUPDATE chat_session \nSET workspace_id = $2, updated_at = now()\nWHERE uuid = $1\nRETURNING id, user_id, uuid, topic, created_at, updated_at, active, model, max_length, temperature, top_p, max_tokens, n, summarize_mode, workspace_id, artifact_enabled, debug, explore_mode\n`\n\ntype UpdateSessionWorkspaceParams struct {\n\tUuid string `json:\"uuid\"`\n\tWorkspaceID sql.NullInt32 `json:\"workspaceId\"`\n}\n\nfunc (q *Queries) UpdateSessionWorkspace(ctx context.Context, arg UpdateSessionWorkspaceParams) (ChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, updateSessionWorkspace, arg.Uuid, arg.WorkspaceID)\n\tvar i ChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.Uuid,\n\t\t&i.Topic,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.Active,\n\t\t&i.Model,\n\t\t&i.MaxLength,\n\t\t&i.Temperature,\n\t\t&i.TopP,\n\t\t&i.MaxTokens,\n\t\t&i.N,\n\t\t&i.SummarizeMode,\n\t\t&i.WorkspaceID,\n\t\t&i.ArtifactEnabled,\n\t\t&i.Debug,\n\t\t&i.ExploreMode,\n\t)\n\treturn i, err\n}\n" }, { "path": "api/sqlc_queries/chat_snapshot.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: chat_snapshot.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"time\"\n)\n\nconst chatSnapshotByID = `-- name: ChatSnapshotByID :one\nSELECT id, typ, uuid, user_id, title, summary, model, tags, session, conversation, created_at, text, search_vector FROM chat_snapshot WHERE id = $1\n`\n\nfunc (q *Queries) ChatSnapshotByID(ctx context.Context, id int32) (ChatSnapshot, error) {\n\trow := q.db.QueryRowContext(ctx, chatSnapshotByID, id)\n\tvar i ChatSnapshot\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Typ,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Title,\n\t\t&i.Summary,\n\t\t&i.Model,\n\t\t&i.Tags,\n\t\t&i.Session,\n\t\t&i.Conversation,\n\t\t&i.CreatedAt,\n\t\t&i.Text,\n\t\t&i.SearchVector,\n\t)\n\treturn i, err\n}\n\nconst chatSnapshotByUUID = `-- name: ChatSnapshotByUUID :one\nSELECT id, typ, uuid, user_id, title, summary, model, tags, session, conversation, created_at, text, search_vector FROM chat_snapshot WHERE uuid = $1\n`\n\nfunc (q *Queries) ChatSnapshotByUUID(ctx context.Context, uuid string) (ChatSnapshot, error) {\n\trow := q.db.QueryRowContext(ctx, chatSnapshotByUUID, uuid)\n\tvar i ChatSnapshot\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Typ,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Title,\n\t\t&i.Summary,\n\t\t&i.Model,\n\t\t&i.Tags,\n\t\t&i.Session,\n\t\t&i.Conversation,\n\t\t&i.CreatedAt,\n\t\t&i.Text,\n\t\t&i.SearchVector,\n\t)\n\treturn i, err\n}\n\nconst chatSnapshotByUserIdAndUuid = `-- name: ChatSnapshotByUserIdAndUuid :one\nSELECT id, typ, uuid, user_id, title, summary, model, tags, session, conversation, created_at, text, search_vector FROM chat_snapshot WHERE user_id = $1 AND uuid = $2\n`\n\ntype ChatSnapshotByUserIdAndUuidParams struct {\n\tUserID int32 `json:\"userId\"`\n\tUuid string `json:\"uuid\"`\n}\n\nfunc (q *Queries) ChatSnapshotByUserIdAndUuid(ctx context.Context, arg ChatSnapshotByUserIdAndUuidParams) (ChatSnapshot, error) {\n\trow := q.db.QueryRowContext(ctx, chatSnapshotByUserIdAndUuid, arg.UserID, arg.Uuid)\n\tvar i ChatSnapshot\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Typ,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Title,\n\t\t&i.Summary,\n\t\t&i.Model,\n\t\t&i.Tags,\n\t\t&i.Session,\n\t\t&i.Conversation,\n\t\t&i.CreatedAt,\n\t\t&i.Text,\n\t\t&i.SearchVector,\n\t)\n\treturn i, err\n}\n\nconst chatSnapshotCountByUserIDAndType = `-- name: ChatSnapshotCountByUserIDAndType :one\nSELECT COUNT(*)\nFROM chat_snapshot\nWHERE user_id = $1 AND ($2::text = '' OR typ = $2)\n`\n\ntype ChatSnapshotCountByUserIDAndTypeParams struct {\n\tUserID int32 `json:\"userId\"`\n\tColumn2 string `json:\"column2\"`\n}\n\nfunc (q *Queries) ChatSnapshotCountByUserIDAndType(ctx context.Context, arg ChatSnapshotCountByUserIDAndTypeParams) (int64, error) {\n\trow := q.db.QueryRowContext(ctx, chatSnapshotCountByUserIDAndType, arg.UserID, arg.Column2)\n\tvar count int64\n\terr := row.Scan(&count)\n\treturn count, err\n}\n\nconst chatSnapshotMetaByUserID = `-- name: ChatSnapshotMetaByUserID :many\nSELECT uuid, title, summary, tags, created_at, typ\nFROM chat_snapshot WHERE user_id = $1 and typ = $2\norder by created_at desc\nLIMIT $3 OFFSET $4\n`\n\ntype ChatSnapshotMetaByUserIDParams struct {\n\tUserID int32 `json:\"userId\"`\n\tTyp string `json:\"typ\"`\n\tLimit int32 `json:\"limit\"`\n\tOffset int32 `json:\"offset\"`\n}\n\ntype ChatSnapshotMetaByUserIDRow struct {\n\tUuid string `json:\"uuid\"`\n\tTitle string `json:\"title\"`\n\tSummary string `json:\"summary\"`\n\tTags json.RawMessage `json:\"tags\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tTyp string `json:\"typ\"`\n}\n\nfunc (q *Queries) ChatSnapshotMetaByUserID(ctx context.Context, arg ChatSnapshotMetaByUserIDParams) ([]ChatSnapshotMetaByUserIDRow, error) {\n\trows, err := q.db.QueryContext(ctx, chatSnapshotMetaByUserID,\n\t\targ.UserID,\n\t\targ.Typ,\n\t\targ.Limit,\n\t\targ.Offset,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatSnapshotMetaByUserIDRow\n\tfor rows.Next() {\n\t\tvar i ChatSnapshotMetaByUserIDRow\n\t\tif err := rows.Scan(\n\t\t\t&i.Uuid,\n\t\t\t&i.Title,\n\t\t\t&i.Summary,\n\t\t\t&i.Tags,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.Typ,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst chatSnapshotSearch = `-- name: ChatSnapshotSearch :many\nSELECT uuid, title, ts_rank(search_vector, websearch_to_tsquery($2), 1) as rank\nFROM chat_snapshot\nWHERE search_vector @@ websearch_to_tsquery($2) AND user_id = $1\nORDER BY rank DESC\nLIMIT 20\n`\n\ntype ChatSnapshotSearchParams struct {\n\tUserID int32 `json:\"userId\"`\n\tSearch string `json:\"search\"`\n}\n\ntype ChatSnapshotSearchRow struct {\n\tUuid string `json:\"uuid\"`\n\tTitle string `json:\"title\"`\n\tRank float32 `json:\"rank\"`\n}\n\nfunc (q *Queries) ChatSnapshotSearch(ctx context.Context, arg ChatSnapshotSearchParams) ([]ChatSnapshotSearchRow, error) {\n\trows, err := q.db.QueryContext(ctx, chatSnapshotSearch, arg.UserID, arg.Search)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatSnapshotSearchRow\n\tfor rows.Next() {\n\t\tvar i ChatSnapshotSearchRow\n\t\tif err := rows.Scan(&i.Uuid, &i.Title, &i.Rank); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst createChatBot = `-- name: CreateChatBot :one\nINSERT INTO chat_snapshot (uuid, user_id, typ, title, model, summary, tags, conversation ,session, text )\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)\nRETURNING id, typ, uuid, user_id, title, summary, model, tags, session, conversation, created_at, text, search_vector\n`\n\ntype CreateChatBotParams struct {\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n\tTyp string `json:\"typ\"`\n\tTitle string `json:\"title\"`\n\tModel string `json:\"model\"`\n\tSummary string `json:\"summary\"`\n\tTags json.RawMessage `json:\"tags\"`\n\tConversation json.RawMessage `json:\"conversation\"`\n\tSession json.RawMessage `json:\"session\"`\n\tText string `json:\"text\"`\n}\n\nfunc (q *Queries) CreateChatBot(ctx context.Context, arg CreateChatBotParams) (ChatSnapshot, error) {\n\trow := q.db.QueryRowContext(ctx, createChatBot,\n\t\targ.Uuid,\n\t\targ.UserID,\n\t\targ.Typ,\n\t\targ.Title,\n\t\targ.Model,\n\t\targ.Summary,\n\t\targ.Tags,\n\t\targ.Conversation,\n\t\targ.Session,\n\t\targ.Text,\n\t)\n\tvar i ChatSnapshot\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Typ,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Title,\n\t\t&i.Summary,\n\t\t&i.Model,\n\t\t&i.Tags,\n\t\t&i.Session,\n\t\t&i.Conversation,\n\t\t&i.CreatedAt,\n\t\t&i.Text,\n\t\t&i.SearchVector,\n\t)\n\treturn i, err\n}\n\nconst createChatSnapshot = `-- name: CreateChatSnapshot :one\nINSERT INTO chat_snapshot (uuid, user_id, title, model, summary, tags, conversation ,session, text )\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)\nRETURNING id, typ, uuid, user_id, title, summary, model, tags, session, conversation, created_at, text, search_vector\n`\n\ntype CreateChatSnapshotParams struct {\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n\tTitle string `json:\"title\"`\n\tModel string `json:\"model\"`\n\tSummary string `json:\"summary\"`\n\tTags json.RawMessage `json:\"tags\"`\n\tConversation json.RawMessage `json:\"conversation\"`\n\tSession json.RawMessage `json:\"session\"`\n\tText string `json:\"text\"`\n}\n\nfunc (q *Queries) CreateChatSnapshot(ctx context.Context, arg CreateChatSnapshotParams) (ChatSnapshot, error) {\n\trow := q.db.QueryRowContext(ctx, createChatSnapshot,\n\t\targ.Uuid,\n\t\targ.UserID,\n\t\targ.Title,\n\t\targ.Model,\n\t\targ.Summary,\n\t\targ.Tags,\n\t\targ.Conversation,\n\t\targ.Session,\n\t\targ.Text,\n\t)\n\tvar i ChatSnapshot\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Typ,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Title,\n\t\t&i.Summary,\n\t\t&i.Model,\n\t\t&i.Tags,\n\t\t&i.Session,\n\t\t&i.Conversation,\n\t\t&i.CreatedAt,\n\t\t&i.Text,\n\t\t&i.SearchVector,\n\t)\n\treturn i, err\n}\n\nconst deleteChatSnapshot = `-- name: DeleteChatSnapshot :one\nDELETE FROM chat_snapshot WHERE uuid = $1\nand user_id = $2\nRETURNING id, typ, uuid, user_id, title, summary, model, tags, session, conversation, created_at, text, search_vector\n`\n\ntype DeleteChatSnapshotParams struct {\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n}\n\nfunc (q *Queries) DeleteChatSnapshot(ctx context.Context, arg DeleteChatSnapshotParams) (ChatSnapshot, error) {\n\trow := q.db.QueryRowContext(ctx, deleteChatSnapshot, arg.Uuid, arg.UserID)\n\tvar i ChatSnapshot\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Typ,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Title,\n\t\t&i.Summary,\n\t\t&i.Model,\n\t\t&i.Tags,\n\t\t&i.Session,\n\t\t&i.Conversation,\n\t\t&i.CreatedAt,\n\t\t&i.Text,\n\t\t&i.SearchVector,\n\t)\n\treturn i, err\n}\n\nconst listChatSnapshots = `-- name: ListChatSnapshots :many\nSELECT id, typ, uuid, user_id, title, summary, model, tags, session, conversation, created_at, text, search_vector FROM chat_snapshot ORDER BY id\n`\n\nfunc (q *Queries) ListChatSnapshots(ctx context.Context) ([]ChatSnapshot, error) {\n\trows, err := q.db.QueryContext(ctx, listChatSnapshots)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatSnapshot\n\tfor rows.Next() {\n\t\tvar i ChatSnapshot\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Typ,\n\t\t\t&i.Uuid,\n\t\t\t&i.UserID,\n\t\t\t&i.Title,\n\t\t\t&i.Summary,\n\t\t\t&i.Model,\n\t\t\t&i.Tags,\n\t\t\t&i.Session,\n\t\t\t&i.Conversation,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.Text,\n\t\t\t&i.SearchVector,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst updateChatSnapshot = `-- name: UpdateChatSnapshot :one\nUPDATE chat_snapshot\nSET uuid = $2, user_id = $3, title = $4, summary = $5, tags = $6, conversation = $7, created_at = $8\nWHERE id = $1\nRETURNING id, typ, uuid, user_id, title, summary, model, tags, session, conversation, created_at, text, search_vector\n`\n\ntype UpdateChatSnapshotParams struct {\n\tID int32 `json:\"id\"`\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n\tTitle string `json:\"title\"`\n\tSummary string `json:\"summary\"`\n\tTags json.RawMessage `json:\"tags\"`\n\tConversation json.RawMessage `json:\"conversation\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n}\n\nfunc (q *Queries) UpdateChatSnapshot(ctx context.Context, arg UpdateChatSnapshotParams) (ChatSnapshot, error) {\n\trow := q.db.QueryRowContext(ctx, updateChatSnapshot,\n\t\targ.ID,\n\t\targ.Uuid,\n\t\targ.UserID,\n\t\targ.Title,\n\t\targ.Summary,\n\t\targ.Tags,\n\t\targ.Conversation,\n\t\targ.CreatedAt,\n\t)\n\tvar i ChatSnapshot\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Typ,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Title,\n\t\t&i.Summary,\n\t\t&i.Model,\n\t\t&i.Tags,\n\t\t&i.Session,\n\t\t&i.Conversation,\n\t\t&i.CreatedAt,\n\t\t&i.Text,\n\t\t&i.SearchVector,\n\t)\n\treturn i, err\n}\n\nconst updateChatSnapshotMetaByUUID = `-- name: UpdateChatSnapshotMetaByUUID :exec\nUPDATE chat_snapshot\nSET title = $2, summary = $3\nWHERE uuid = $1 and user_id = $4\n`\n\ntype UpdateChatSnapshotMetaByUUIDParams struct {\n\tUuid string `json:\"uuid\"`\n\tTitle string `json:\"title\"`\n\tSummary string `json:\"summary\"`\n\tUserID int32 `json:\"userId\"`\n}\n\nfunc (q *Queries) UpdateChatSnapshotMetaByUUID(ctx context.Context, arg UpdateChatSnapshotMetaByUUIDParams) error {\n\t_, err := q.db.ExecContext(ctx, updateChatSnapshotMetaByUUID,\n\t\targ.Uuid,\n\t\targ.Title,\n\t\targ.Summary,\n\t\targ.UserID,\n\t)\n\treturn err\n}\n" }, { "path": "api/sqlc_queries/chat_workspace.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: chat_workspace.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n\t\"time\"\n)\n\nconst createDefaultWorkspace = `-- name: CreateDefaultWorkspace :one\nINSERT INTO chat_workspace (uuid, user_id, name, description, color, icon, is_default, order_position)\nVALUES ($1, $2, 'General', 'Default workspace for all conversations', '#6366f1', 'folder', true, 0)\nRETURNING id, uuid, user_id, name, description, color, icon, created_at, updated_at, is_default, order_position\n`\n\ntype CreateDefaultWorkspaceParams struct {\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n}\n\nfunc (q *Queries) CreateDefaultWorkspace(ctx context.Context, arg CreateDefaultWorkspaceParams) (ChatWorkspace, error) {\n\trow := q.db.QueryRowContext(ctx, createDefaultWorkspace, arg.Uuid, arg.UserID)\n\tvar i ChatWorkspace\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Name,\n\t\t&i.Description,\n\t\t&i.Color,\n\t\t&i.Icon,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.IsDefault,\n\t\t&i.OrderPosition,\n\t)\n\treturn i, err\n}\n\nconst createWorkspace = `-- name: CreateWorkspace :one\nINSERT INTO chat_workspace (uuid, user_id, name, description, color, icon, is_default, order_position)\nVALUES ($1, $2, $3, $4, $5, $6, $7, $8)\nRETURNING id, uuid, user_id, name, description, color, icon, created_at, updated_at, is_default, order_position\n`\n\ntype CreateWorkspaceParams struct {\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n\tName string `json:\"name\"`\n\tDescription string `json:\"description\"`\n\tColor string `json:\"color\"`\n\tIcon string `json:\"icon\"`\n\tIsDefault bool `json:\"isDefault\"`\n\tOrderPosition int32 `json:\"orderPosition\"`\n}\n\nfunc (q *Queries) CreateWorkspace(ctx context.Context, arg CreateWorkspaceParams) (ChatWorkspace, error) {\n\trow := q.db.QueryRowContext(ctx, createWorkspace,\n\t\targ.Uuid,\n\t\targ.UserID,\n\t\targ.Name,\n\t\targ.Description,\n\t\targ.Color,\n\t\targ.Icon,\n\t\targ.IsDefault,\n\t\targ.OrderPosition,\n\t)\n\tvar i ChatWorkspace\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Name,\n\t\t&i.Description,\n\t\t&i.Color,\n\t\t&i.Icon,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.IsDefault,\n\t\t&i.OrderPosition,\n\t)\n\treturn i, err\n}\n\nconst deleteWorkspace = `-- name: DeleteWorkspace :exec\nDELETE FROM chat_workspace \nWHERE uuid = $1\n`\n\nfunc (q *Queries) DeleteWorkspace(ctx context.Context, uuid string) error {\n\t_, err := q.db.ExecContext(ctx, deleteWorkspace, uuid)\n\treturn err\n}\n\nconst getDefaultWorkspaceByUserID = `-- name: GetDefaultWorkspaceByUserID :one\nSELECT id, uuid, user_id, name, description, color, icon, created_at, updated_at, is_default, order_position FROM chat_workspace \nWHERE user_id = $1 AND is_default = true\nLIMIT 1\n`\n\nfunc (q *Queries) GetDefaultWorkspaceByUserID(ctx context.Context, userID int32) (ChatWorkspace, error) {\n\trow := q.db.QueryRowContext(ctx, getDefaultWorkspaceByUserID, userID)\n\tvar i ChatWorkspace\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Name,\n\t\t&i.Description,\n\t\t&i.Color,\n\t\t&i.Icon,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.IsDefault,\n\t\t&i.OrderPosition,\n\t)\n\treturn i, err\n}\n\nconst getWorkspaceByUUID = `-- name: GetWorkspaceByUUID :one\nSELECT id, uuid, user_id, name, description, color, icon, created_at, updated_at, is_default, order_position FROM chat_workspace \nWHERE uuid = $1\n`\n\nfunc (q *Queries) GetWorkspaceByUUID(ctx context.Context, uuid string) (ChatWorkspace, error) {\n\trow := q.db.QueryRowContext(ctx, getWorkspaceByUUID, uuid)\n\tvar i ChatWorkspace\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Name,\n\t\t&i.Description,\n\t\t&i.Color,\n\t\t&i.Icon,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.IsDefault,\n\t\t&i.OrderPosition,\n\t)\n\treturn i, err\n}\n\nconst getWorkspaceWithSessionCount = `-- name: GetWorkspaceWithSessionCount :many\nSELECT \n w.id, w.uuid, w.user_id, w.name, w.description, w.color, w.icon, w.created_at, w.updated_at, w.is_default, w.order_position,\n COUNT(cs.id) as session_count\nFROM chat_workspace w\nLEFT JOIN chat_session cs ON w.id = cs.workspace_id AND cs.active = true\nWHERE w.user_id = $1\nGROUP BY w.id\nORDER BY w.order_position ASC, w.created_at ASC\n`\n\ntype GetWorkspaceWithSessionCountRow struct {\n\tID int32 `json:\"id\"`\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n\tName string `json:\"name\"`\n\tDescription string `json:\"description\"`\n\tColor string `json:\"color\"`\n\tIcon string `json:\"icon\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tIsDefault bool `json:\"isDefault\"`\n\tOrderPosition int32 `json:\"orderPosition\"`\n\tSessionCount int64 `json:\"sessionCount\"`\n}\n\nfunc (q *Queries) GetWorkspaceWithSessionCount(ctx context.Context, userID int32) ([]GetWorkspaceWithSessionCountRow, error) {\n\trows, err := q.db.QueryContext(ctx, getWorkspaceWithSessionCount, userID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []GetWorkspaceWithSessionCountRow\n\tfor rows.Next() {\n\t\tvar i GetWorkspaceWithSessionCountRow\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Uuid,\n\t\t\t&i.UserID,\n\t\t\t&i.Name,\n\t\t\t&i.Description,\n\t\t\t&i.Color,\n\t\t\t&i.Icon,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.IsDefault,\n\t\t\t&i.OrderPosition,\n\t\t\t&i.SessionCount,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getWorkspacesByUserID = `-- name: GetWorkspacesByUserID :many\nSELECT id, uuid, user_id, name, description, color, icon, created_at, updated_at, is_default, order_position FROM chat_workspace \nWHERE user_id = $1\nORDER BY order_position ASC, created_at ASC\n`\n\nfunc (q *Queries) GetWorkspacesByUserID(ctx context.Context, userID int32) ([]ChatWorkspace, error) {\n\trows, err := q.db.QueryContext(ctx, getWorkspacesByUserID, userID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ChatWorkspace\n\tfor rows.Next() {\n\t\tvar i ChatWorkspace\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.Uuid,\n\t\t\t&i.UserID,\n\t\t\t&i.Name,\n\t\t\t&i.Description,\n\t\t\t&i.Color,\n\t\t\t&i.Icon,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.IsDefault,\n\t\t\t&i.OrderPosition,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst hasWorkspacePermission = `-- name: HasWorkspacePermission :one\nSELECT COUNT(*) > 0 as has_permission\nFROM chat_workspace w\nWHERE w.uuid = $1\n AND (\n w.user_id = $2\n OR EXISTS (\n SELECT 1\n FROM auth_user request_user\n WHERE request_user.id = $2 AND request_user.is_superuser = true\n )\n )\n`\n\ntype HasWorkspacePermissionParams struct {\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n}\n\nfunc (q *Queries) HasWorkspacePermission(ctx context.Context, arg HasWorkspacePermissionParams) (bool, error) {\n\trow := q.db.QueryRowContext(ctx, hasWorkspacePermission, arg.Uuid, arg.UserID)\n\tvar has_permission bool\n\terr := row.Scan(&has_permission)\n\treturn has_permission, err\n}\n\nconst setDefaultWorkspace = `-- name: SetDefaultWorkspace :one\nUPDATE chat_workspace \nSET is_default = $2, updated_at = now()\nWHERE uuid = $1\nRETURNING id, uuid, user_id, name, description, color, icon, created_at, updated_at, is_default, order_position\n`\n\ntype SetDefaultWorkspaceParams struct {\n\tUuid string `json:\"uuid\"`\n\tIsDefault bool `json:\"isDefault\"`\n}\n\nfunc (q *Queries) SetDefaultWorkspace(ctx context.Context, arg SetDefaultWorkspaceParams) (ChatWorkspace, error) {\n\trow := q.db.QueryRowContext(ctx, setDefaultWorkspace, arg.Uuid, arg.IsDefault)\n\tvar i ChatWorkspace\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Name,\n\t\t&i.Description,\n\t\t&i.Color,\n\t\t&i.Icon,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.IsDefault,\n\t\t&i.OrderPosition,\n\t)\n\treturn i, err\n}\n\nconst updateWorkspace = `-- name: UpdateWorkspace :one\nUPDATE chat_workspace \nSET name = $2, description = $3, color = $4, icon = $5, updated_at = now()\nWHERE uuid = $1\nRETURNING id, uuid, user_id, name, description, color, icon, created_at, updated_at, is_default, order_position\n`\n\ntype UpdateWorkspaceParams struct {\n\tUuid string `json:\"uuid\"`\n\tName string `json:\"name\"`\n\tDescription string `json:\"description\"`\n\tColor string `json:\"color\"`\n\tIcon string `json:\"icon\"`\n}\n\nfunc (q *Queries) UpdateWorkspace(ctx context.Context, arg UpdateWorkspaceParams) (ChatWorkspace, error) {\n\trow := q.db.QueryRowContext(ctx, updateWorkspace,\n\t\targ.Uuid,\n\t\targ.Name,\n\t\targ.Description,\n\t\targ.Color,\n\t\targ.Icon,\n\t)\n\tvar i ChatWorkspace\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Name,\n\t\t&i.Description,\n\t\t&i.Color,\n\t\t&i.Icon,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.IsDefault,\n\t\t&i.OrderPosition,\n\t)\n\treturn i, err\n}\n\nconst updateWorkspaceOrder = `-- name: UpdateWorkspaceOrder :one\nUPDATE chat_workspace \nSET order_position = $2, updated_at = now()\nWHERE uuid = $1\nRETURNING id, uuid, user_id, name, description, color, icon, created_at, updated_at, is_default, order_position\n`\n\ntype UpdateWorkspaceOrderParams struct {\n\tUuid string `json:\"uuid\"`\n\tOrderPosition int32 `json:\"orderPosition\"`\n}\n\nfunc (q *Queries) UpdateWorkspaceOrder(ctx context.Context, arg UpdateWorkspaceOrderParams) (ChatWorkspace, error) {\n\trow := q.db.QueryRowContext(ctx, updateWorkspaceOrder, arg.Uuid, arg.OrderPosition)\n\tvar i ChatWorkspace\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Uuid,\n\t\t&i.UserID,\n\t\t&i.Name,\n\t\t&i.Description,\n\t\t&i.Color,\n\t\t&i.Icon,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.IsDefault,\n\t\t&i.OrderPosition,\n\t)\n\treturn i, err\n}\n" }, { "path": "api/sqlc_queries/db.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n)\n\ntype DBTX interface {\n\tExecContext(context.Context, string, ...interface{}) (sql.Result, error)\n\tPrepareContext(context.Context, string) (*sql.Stmt, error)\n\tQueryContext(context.Context, string, ...interface{}) (*sql.Rows, error)\n\tQueryRowContext(context.Context, string, ...interface{}) *sql.Row\n}\n\nfunc New(db DBTX) *Queries {\n\treturn &Queries{db: db}\n}\n\ntype Queries struct {\n\tdb DBTX\n}\n\nfunc (q *Queries) WithTx(tx *sql.Tx) *Queries {\n\treturn &Queries{\n\t\tdb: tx,\n\t}\n}\n" }, { "path": "api/sqlc_queries/jwt_secrets.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: jwt_secrets.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n)\n\nconst createJwtSecret = `-- name: CreateJwtSecret :one\nINSERT INTO jwt_secrets (name, secret, audience)\nVALUES ($1, $2, $3) RETURNING id, name, secret, audience, lifetime\n`\n\ntype CreateJwtSecretParams struct {\n\tName string `json:\"name\"`\n\tSecret string `json:\"secret\"`\n\tAudience string `json:\"audience\"`\n}\n\nfunc (q *Queries) CreateJwtSecret(ctx context.Context, arg CreateJwtSecretParams) (JwtSecret, error) {\n\trow := q.db.QueryRowContext(ctx, createJwtSecret, arg.Name, arg.Secret, arg.Audience)\n\tvar i JwtSecret\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Name,\n\t\t&i.Secret,\n\t\t&i.Audience,\n\t\t&i.Lifetime,\n\t)\n\treturn i, err\n}\n\nconst deleteAllJwtSecrets = `-- name: DeleteAllJwtSecrets :execrows\nDELETE FROM jwt_secrets\n`\n\nfunc (q *Queries) DeleteAllJwtSecrets(ctx context.Context) (int64, error) {\n\tresult, err := q.db.ExecContext(ctx, deleteAllJwtSecrets)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn result.RowsAffected()\n}\n\nconst getJwtSecret = `-- name: GetJwtSecret :one\nSELECT id, name, secret, audience, lifetime FROM jwt_secrets WHERE name = $1\n`\n\nfunc (q *Queries) GetJwtSecret(ctx context.Context, name string) (JwtSecret, error) {\n\trow := q.db.QueryRowContext(ctx, getJwtSecret, name)\n\tvar i JwtSecret\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.Name,\n\t\t&i.Secret,\n\t\t&i.Audience,\n\t\t&i.Lifetime,\n\t)\n\treturn i, err\n}\n" }, { "path": "api/sqlc_queries/models.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n\npackage sqlc_queries\n\nimport (\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"time\"\n)\n\ntype AuthUser struct {\n\tID int32 `json:\"id\"`\n\tPassword string `json:\"password\"`\n\tLastLogin time.Time `json:\"lastLogin\"`\n\tIsSuperuser bool `json:\"isSuperuser\"`\n\tUsername string `json:\"username\"`\n\tFirstName string `json:\"firstName\"`\n\tLastName string `json:\"lastName\"`\n\tEmail string `json:\"email\"`\n\tIsStaff bool `json:\"isStaff\"`\n\tIsActive bool `json:\"isActive\"`\n\tDateJoined time.Time `json:\"dateJoined\"`\n}\n\ntype AuthUserManagement struct {\n\tID int32 `json:\"id\"`\n\tUserID int32 `json:\"userId\"`\n\tRateLimit int32 `json:\"rateLimit\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n}\n\ntype BotAnswerHistory struct {\n\tID int32 `json:\"id\"`\n\tBotUuid string `json:\"botUuid\"`\n\tUserID int32 `json:\"userId\"`\n\tPrompt string `json:\"prompt\"`\n\tAnswer string `json:\"answer\"`\n\tModel string `json:\"model\"`\n\tTokensUsed int32 `json:\"tokensUsed\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n}\n\ntype ChatComment struct {\n\tID int32 `json:\"id\"`\n\tUuid string `json:\"uuid\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\tChatMessageUuid string `json:\"chatMessageUuid\"`\n\tContent string `json:\"content\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tCreatedBy int32 `json:\"createdBy\"`\n\tUpdatedBy int32 `json:\"updatedBy\"`\n}\n\ntype ChatFile struct {\n\tID int32 `json:\"id\"`\n\tName string `json:\"name\"`\n\tData []byte `json:\"data\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUserID int32 `json:\"userId\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\tMimeType string `json:\"mimeType\"`\n}\n\ntype ChatLog struct {\n\tID int32 `json:\"id\"`\n\tSession json.RawMessage `json:\"session\"`\n\tQuestion json.RawMessage `json:\"question\"`\n\tAnswer json.RawMessage `json:\"answer\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n}\n\ntype ChatMessage struct {\n\tID int32 `json:\"id\"`\n\tUuid string `json:\"uuid\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n\tReasoningContent string `json:\"reasoningContent\"`\n\tModel string `json:\"model\"`\n\tLlmSummary string `json:\"llmSummary\"`\n\tScore float64 `json:\"score\"`\n\tUserID int32 `json:\"userId\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tCreatedBy int32 `json:\"createdBy\"`\n\tUpdatedBy int32 `json:\"updatedBy\"`\n\tIsDeleted bool `json:\"isDeleted\"`\n\tIsPin bool `json:\"isPin\"`\n\tTokenCount int32 `json:\"tokenCount\"`\n\tRaw json.RawMessage `json:\"raw\"`\n\tArtifacts json.RawMessage `json:\"artifacts\"`\n\tSuggestedQuestions json.RawMessage `json:\"suggestedQuestions\"`\n}\n\ntype ChatModel struct {\n\tID int32 `json:\"id\"`\n\tName string `json:\"name\"`\n\tLabel string `json:\"label\"`\n\tIsDefault bool `json:\"isDefault\"`\n\tUrl string `json:\"url\"`\n\tApiAuthHeader string `json:\"apiAuthHeader\"`\n\tApiAuthKey string `json:\"apiAuthKey\"`\n\tUserID int32 `json:\"userId\"`\n\tEnablePerModeRatelimit bool `json:\"enablePerModeRatelimit\"`\n\tMaxToken int32 `json:\"maxToken\"`\n\tDefaultToken int32 `json:\"defaultToken\"`\n\tOrderNumber int32 `json:\"orderNumber\"`\n\tHttpTimeOut int32 `json:\"httpTimeOut\"`\n\tIsEnable bool `json:\"isEnable\"`\n\tApiType string `json:\"apiType\"`\n}\n\ntype ChatPrompt struct {\n\tID int32 `json:\"id\"`\n\tUuid string `json:\"uuid\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n\tScore float64 `json:\"score\"`\n\tUserID int32 `json:\"userId\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tCreatedBy int32 `json:\"createdBy\"`\n\tUpdatedBy int32 `json:\"updatedBy\"`\n\tIsDeleted bool `json:\"isDeleted\"`\n\tTokenCount int32 `json:\"tokenCount\"`\n}\n\ntype ChatSession struct {\n\tID int32 `json:\"id\"`\n\tUserID int32 `json:\"userId\"`\n\tUuid string `json:\"uuid\"`\n\tTopic string `json:\"topic\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tActive bool `json:\"active\"`\n\tModel string `json:\"model\"`\n\tMaxLength int32 `json:\"maxLength\"`\n\tTemperature float64 `json:\"temperature\"`\n\tTopP float64 `json:\"topP\"`\n\tMaxTokens int32 `json:\"maxTokens\"`\n\tN int32 `json:\"n\"`\n\tSummarizeMode bool `json:\"summarizeMode\"`\n\tWorkspaceID sql.NullInt32 `json:\"workspaceId\"`\n\tArtifactEnabled bool `json:\"artifactEnabled\"`\n\tDebug bool `json:\"debug\"`\n\tExploreMode bool `json:\"exploreMode\"`\n}\n\ntype ChatSnapshot struct {\n\tID int32 `json:\"id\"`\n\tTyp string `json:\"typ\"`\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n\tTitle string `json:\"title\"`\n\tSummary string `json:\"summary\"`\n\tModel string `json:\"model\"`\n\tTags json.RawMessage `json:\"tags\"`\n\tSession json.RawMessage `json:\"session\"`\n\tConversation json.RawMessage `json:\"conversation\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tText string `json:\"text\"`\n\tSearchVector interface{} `json:\"searchVector\"`\n}\n\ntype ChatWorkspace struct {\n\tID int32 `json:\"id\"`\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n\tName string `json:\"name\"`\n\tDescription string `json:\"description\"`\n\tColor string `json:\"color\"`\n\tIcon string `json:\"icon\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tIsDefault bool `json:\"isDefault\"`\n\tOrderPosition int32 `json:\"orderPosition\"`\n}\n\ntype JwtSecret struct {\n\tID int32 `json:\"id\"`\n\tName string `json:\"name\"`\n\tSecret string `json:\"secret\"`\n\tAudience string `json:\"audience\"`\n\tLifetime int16 `json:\"lifetime\"`\n}\n\ntype UserActiveChatSession struct {\n\tID int32 `json:\"id\"`\n\tUserID int32 `json:\"userId\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tWorkspaceID sql.NullInt32 `json:\"workspaceId\"`\n}\n\ntype UserChatModelPrivilege struct {\n\tID int32 `json:\"id\"`\n\tUserID int32 `json:\"userId\"`\n\tChatModelID int32 `json:\"chatModelId\"`\n\tRateLimit int32 `json:\"rateLimit\"`\n\tCreatedAt time.Time `json:\"createdAt\"`\n\tUpdatedAt time.Time `json:\"updatedAt\"`\n\tCreatedBy int32 `json:\"createdBy\"`\n\tUpdatedBy int32 `json:\"updatedBy\"`\n}\n" }, { "path": "api/sqlc_queries/user_active_chat_session.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: user_active_chat_session.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n)\n\nconst deleteUserActiveSession = `-- name: DeleteUserActiveSession :exec\nDELETE FROM user_active_chat_session\nWHERE user_id = $1 AND (\n (workspace_id IS NULL AND $2::int IS NULL) OR \n (workspace_id = $2)\n)\n`\n\ntype DeleteUserActiveSessionParams struct {\n\tUserID int32 `json:\"userId\"`\n\tColumn2 int32 `json:\"column2\"`\n}\n\nfunc (q *Queries) DeleteUserActiveSession(ctx context.Context, arg DeleteUserActiveSessionParams) error {\n\t_, err := q.db.ExecContext(ctx, deleteUserActiveSession, arg.UserID, arg.Column2)\n\treturn err\n}\n\nconst deleteUserActiveSessionBySession = `-- name: DeleteUserActiveSessionBySession :exec\nDELETE FROM user_active_chat_session\nWHERE user_id = $1 AND chat_session_uuid = $2\n`\n\ntype DeleteUserActiveSessionBySessionParams struct {\n\tUserID int32 `json:\"userId\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n}\n\nfunc (q *Queries) DeleteUserActiveSessionBySession(ctx context.Context, arg DeleteUserActiveSessionBySessionParams) error {\n\t_, err := q.db.ExecContext(ctx, deleteUserActiveSessionBySession, arg.UserID, arg.ChatSessionUuid)\n\treturn err\n}\n\nconst getAllUserActiveSessions = `-- name: GetAllUserActiveSessions :many\nSELECT id, user_id, chat_session_uuid, created_at, updated_at, workspace_id FROM user_active_chat_session\nWHERE user_id = $1\nORDER BY workspace_id NULLS FIRST, updated_at DESC\n`\n\nfunc (q *Queries) GetAllUserActiveSessions(ctx context.Context, userID int32) ([]UserActiveChatSession, error) {\n\trows, err := q.db.QueryContext(ctx, getAllUserActiveSessions, userID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []UserActiveChatSession\n\tfor rows.Next() {\n\t\tvar i UserActiveChatSession\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.UserID,\n\t\t\t&i.ChatSessionUuid,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.WorkspaceID,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst getUserActiveSession = `-- name: GetUserActiveSession :one\nSELECT id, user_id, chat_session_uuid, created_at, updated_at, workspace_id FROM user_active_chat_session \nWHERE user_id = $1 AND (\n (workspace_id IS NULL AND $2::int IS NULL) OR \n (workspace_id = $2)\n)\n`\n\ntype GetUserActiveSessionParams struct {\n\tUserID int32 `json:\"userId\"`\n\tColumn2 int32 `json:\"column2\"`\n}\n\nfunc (q *Queries) GetUserActiveSession(ctx context.Context, arg GetUserActiveSessionParams) (UserActiveChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, getUserActiveSession, arg.UserID, arg.Column2)\n\tvar i UserActiveChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.ChatSessionUuid,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.WorkspaceID,\n\t)\n\treturn i, err\n}\n\nconst upsertUserActiveSession = `-- name: UpsertUserActiveSession :one\n\nINSERT INTO user_active_chat_session (user_id, workspace_id, chat_session_uuid)\nVALUES ($1, $2, $3)\nON CONFLICT (user_id, COALESCE(workspace_id, -1))\nDO UPDATE SET \n chat_session_uuid = EXCLUDED.chat_session_uuid,\n updated_at = now()\nRETURNING id, user_id, chat_session_uuid, created_at, updated_at, workspace_id\n`\n\ntype UpsertUserActiveSessionParams struct {\n\tUserID int32 `json:\"userId\"`\n\tWorkspaceID sql.NullInt32 `json:\"workspaceId\"`\n\tChatSessionUuid string `json:\"chatSessionUuid\"`\n}\n\n// Simplified unified queries for active sessions\nfunc (q *Queries) UpsertUserActiveSession(ctx context.Context, arg UpsertUserActiveSessionParams) (UserActiveChatSession, error) {\n\trow := q.db.QueryRowContext(ctx, upsertUserActiveSession, arg.UserID, arg.WorkspaceID, arg.ChatSessionUuid)\n\tvar i UserActiveChatSession\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.ChatSessionUuid,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.WorkspaceID,\n\t)\n\treturn i, err\n}\n" }, { "path": "api/sqlc_queries/user_chat_model_privilege.sql.go", "content": "// Code generated by sqlc. DO NOT EDIT.\n// versions:\n// sqlc v1.30.0\n// source: user_chat_model_privilege.sql\n\npackage sqlc_queries\n\nimport (\n\t\"context\"\n)\n\nconst createUserChatModelPrivilege = `-- name: CreateUserChatModelPrivilege :one\nINSERT INTO user_chat_model_privilege (user_id, chat_model_id, rate_limit, created_by, updated_by)\nVALUES ($1, $2, $3, $4, $5)\nRETURNING id, user_id, chat_model_id, rate_limit, created_at, updated_at, created_by, updated_by\n`\n\ntype CreateUserChatModelPrivilegeParams struct {\n\tUserID int32 `json:\"userId\"`\n\tChatModelID int32 `json:\"chatModelId\"`\n\tRateLimit int32 `json:\"rateLimit\"`\n\tCreatedBy int32 `json:\"createdBy\"`\n\tUpdatedBy int32 `json:\"updatedBy\"`\n}\n\nfunc (q *Queries) CreateUserChatModelPrivilege(ctx context.Context, arg CreateUserChatModelPrivilegeParams) (UserChatModelPrivilege, error) {\n\trow := q.db.QueryRowContext(ctx, createUserChatModelPrivilege,\n\t\targ.UserID,\n\t\targ.ChatModelID,\n\t\targ.RateLimit,\n\t\targ.CreatedBy,\n\t\targ.UpdatedBy,\n\t)\n\tvar i UserChatModelPrivilege\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.ChatModelID,\n\t\t&i.RateLimit,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t)\n\treturn i, err\n}\n\nconst deleteUserChatModelPrivilege = `-- name: DeleteUserChatModelPrivilege :exec\nDELETE FROM user_chat_model_privilege WHERE id = $1\n`\n\nfunc (q *Queries) DeleteUserChatModelPrivilege(ctx context.Context, id int32) error {\n\t_, err := q.db.ExecContext(ctx, deleteUserChatModelPrivilege, id)\n\treturn err\n}\n\nconst listUserChatModelPrivileges = `-- name: ListUserChatModelPrivileges :many\nSELECT id, user_id, chat_model_id, rate_limit, created_at, updated_at, created_by, updated_by FROM user_chat_model_privilege ORDER BY id\n`\n\nfunc (q *Queries) ListUserChatModelPrivileges(ctx context.Context) ([]UserChatModelPrivilege, error) {\n\trows, err := q.db.QueryContext(ctx, listUserChatModelPrivileges)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []UserChatModelPrivilege\n\tfor rows.Next() {\n\t\tvar i UserChatModelPrivilege\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.UserID,\n\t\t\t&i.ChatModelID,\n\t\t\t&i.RateLimit,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.CreatedBy,\n\t\t\t&i.UpdatedBy,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst listUserChatModelPrivilegesByUserID = `-- name: ListUserChatModelPrivilegesByUserID :many\n\nSELECT id, user_id, chat_model_id, rate_limit, created_at, updated_at, created_by, updated_by FROM user_chat_model_privilege \nWHERE user_id = $1\nORDER BY id\n`\n\n// TODO add ratelimit\n// LIMIT 1000\nfunc (q *Queries) ListUserChatModelPrivilegesByUserID(ctx context.Context, userID int32) ([]UserChatModelPrivilege, error) {\n\trows, err := q.db.QueryContext(ctx, listUserChatModelPrivilegesByUserID, userID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []UserChatModelPrivilege\n\tfor rows.Next() {\n\t\tvar i UserChatModelPrivilege\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.UserID,\n\t\t\t&i.ChatModelID,\n\t\t\t&i.RateLimit,\n\t\t\t&i.CreatedAt,\n\t\t\t&i.UpdatedAt,\n\t\t\t&i.CreatedBy,\n\t\t\t&i.UpdatedBy,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst listUserChatModelPrivilegesRateLimit = `-- name: ListUserChatModelPrivilegesRateLimit :many\nSELECT ucmp.id, au.email as user_email, CONCAT_WS('',au.last_name, au.first_name) as full_name, cm.name chat_model_name, ucmp.rate_limit \nFROM user_chat_model_privilege ucmp \nINNER JOIN chat_model cm ON cm.id = ucmp.chat_model_id\nINNER JOIN auth_user au ON au.id = ucmp.user_id\nORDER by au.last_login DESC\n`\n\ntype ListUserChatModelPrivilegesRateLimitRow struct {\n\tID int32 `json:\"id\"`\n\tUserEmail string `json:\"userEmail\"`\n\tFullName string `json:\"fullName\"`\n\tChatModelName string `json:\"chatModelName\"`\n\tRateLimit int32 `json:\"rateLimit\"`\n}\n\nfunc (q *Queries) ListUserChatModelPrivilegesRateLimit(ctx context.Context) ([]ListUserChatModelPrivilegesRateLimitRow, error) {\n\trows, err := q.db.QueryContext(ctx, listUserChatModelPrivilegesRateLimit)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer rows.Close()\n\tvar items []ListUserChatModelPrivilegesRateLimitRow\n\tfor rows.Next() {\n\t\tvar i ListUserChatModelPrivilegesRateLimitRow\n\t\tif err := rows.Scan(\n\t\t\t&i.ID,\n\t\t\t&i.UserEmail,\n\t\t\t&i.FullName,\n\t\t\t&i.ChatModelName,\n\t\t\t&i.RateLimit,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titems = append(items, i)\n\t}\n\tif err := rows.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := rows.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn items, nil\n}\n\nconst rateLimiteByUserAndSessionUUID = `-- name: RateLimiteByUserAndSessionUUID :one\nSELECT ucmp.rate_limit, cm.name AS chat_model_name\nFROM user_chat_model_privilege ucmp\nJOIN chat_session cs ON cs.user_id = ucmp.user_id\nJOIN chat_model cm ON (cm.id = ucmp.chat_model_id AND cs.model = cm.name and cm.enable_per_mode_ratelimit = true)\nWHERE cs.uuid = $1\n AND ucmp.user_id = $2\n`\n\ntype RateLimiteByUserAndSessionUUIDParams struct {\n\tUuid string `json:\"uuid\"`\n\tUserID int32 `json:\"userId\"`\n}\n\ntype RateLimiteByUserAndSessionUUIDRow struct {\n\tRateLimit int32 `json:\"rateLimit\"`\n\tChatModelName string `json:\"chatModelName\"`\n}\n\nfunc (q *Queries) RateLimiteByUserAndSessionUUID(ctx context.Context, arg RateLimiteByUserAndSessionUUIDParams) (RateLimiteByUserAndSessionUUIDRow, error) {\n\trow := q.db.QueryRowContext(ctx, rateLimiteByUserAndSessionUUID, arg.Uuid, arg.UserID)\n\tvar i RateLimiteByUserAndSessionUUIDRow\n\terr := row.Scan(&i.RateLimit, &i.ChatModelName)\n\treturn i, err\n}\n\nconst updateUserChatModelPrivilege = `-- name: UpdateUserChatModelPrivilege :one\nUPDATE user_chat_model_privilege SET rate_limit = $2, updated_at = now(), updated_by = $3\nWHERE id = $1\nRETURNING id, user_id, chat_model_id, rate_limit, created_at, updated_at, created_by, updated_by\n`\n\ntype UpdateUserChatModelPrivilegeParams struct {\n\tID int32 `json:\"id\"`\n\tRateLimit int32 `json:\"rateLimit\"`\n\tUpdatedBy int32 `json:\"updatedBy\"`\n}\n\nfunc (q *Queries) UpdateUserChatModelPrivilege(ctx context.Context, arg UpdateUserChatModelPrivilegeParams) (UserChatModelPrivilege, error) {\n\trow := q.db.QueryRowContext(ctx, updateUserChatModelPrivilege, arg.ID, arg.RateLimit, arg.UpdatedBy)\n\tvar i UserChatModelPrivilege\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.ChatModelID,\n\t\t&i.RateLimit,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t)\n\treturn i, err\n}\n\nconst userChatModelPrivilegeByID = `-- name: UserChatModelPrivilegeByID :one\nSELECT id, user_id, chat_model_id, rate_limit, created_at, updated_at, created_by, updated_by FROM user_chat_model_privilege WHERE id = $1\n`\n\nfunc (q *Queries) UserChatModelPrivilegeByID(ctx context.Context, id int32) (UserChatModelPrivilege, error) {\n\trow := q.db.QueryRowContext(ctx, userChatModelPrivilegeByID, id)\n\tvar i UserChatModelPrivilege\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.ChatModelID,\n\t\t&i.RateLimit,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t)\n\treturn i, err\n}\n\nconst userChatModelPrivilegeByUserAndModelID = `-- name: UserChatModelPrivilegeByUserAndModelID :one\nSELECT id, user_id, chat_model_id, rate_limit, created_at, updated_at, created_by, updated_by FROM user_chat_model_privilege WHERE user_id = $1 AND chat_model_id = $2\n`\n\ntype UserChatModelPrivilegeByUserAndModelIDParams struct {\n\tUserID int32 `json:\"userId\"`\n\tChatModelID int32 `json:\"chatModelId\"`\n}\n\nfunc (q *Queries) UserChatModelPrivilegeByUserAndModelID(ctx context.Context, arg UserChatModelPrivilegeByUserAndModelIDParams) (UserChatModelPrivilege, error) {\n\trow := q.db.QueryRowContext(ctx, userChatModelPrivilegeByUserAndModelID, arg.UserID, arg.ChatModelID)\n\tvar i UserChatModelPrivilege\n\terr := row.Scan(\n\t\t&i.ID,\n\t\t&i.UserID,\n\t\t&i.ChatModelID,\n\t\t&i.RateLimit,\n\t\t&i.CreatedAt,\n\t\t&i.UpdatedAt,\n\t\t&i.CreatedBy,\n\t\t&i.UpdatedBy,\n\t)\n\treturn i, err\n}\n" }, { "path": "api/sqlc_queries/zz_custom_method.go", "content": "package sqlc_queries\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\n\t\"github.com/samber/lo\"\n\t\"github.com/sashabaranov/go-openai\"\n)\n\nfunc (user *AuthUser) Role() string {\n\trole := \"user\"\n\tif user.IsSuperuser {\n\t\trole = \"admin\"\n\t}\n\treturn role\n}\n\nfunc (m *ChatMessage) Authenticate(q Queries, userID int32) (bool, error) {\n\tmessageID := m.ID\n\tctx := context.Background()\n\tv, e := q.HasChatMessagePermission(ctx, HasChatMessagePermissionParams{messageID, userID})\n\treturn v, e\n}\n\nfunc (s *ChatSession) Authenticate(q Queries, userID int32) (bool, error) {\n\tsessionID := s.ID\n\tctx := context.Background()\n\tv, e := q.HasChatSessionPermission(ctx, HasChatSessionPermissionParams{sessionID, userID})\n\treturn v, e\n}\n\nfunc (p *ChatPrompt) Authenticate(q Queries, userID int32) (bool, error) {\n\tsessionID := p.ID\n\tctx := context.Background()\n\tv, e := q.HasChatPromptPermission(ctx, HasChatPromptPermissionParams{sessionID, userID})\n\treturn v, e\n}\n\n// Create a RawMessage from ChatSession\nfunc (cs *ChatSession) ToRawMessage() *json.RawMessage {\n\t// Marshal ChatSession struct to json.RawMessage\n\tchatSessionJSON, err := json.Marshal(cs)\n\tif err != nil {\n\t\treturn nil\n\t}\n\tvar rawMessage json.RawMessage = chatSessionJSON\n\treturn &rawMessage\n}\n\ntype MessageWithRoleAndContent interface {\n\tGetRole() string\n\tGetContent() string\n}\n\nfunc (m ChatMessage) GetRole() string {\n\treturn m.Role\n}\n\nfunc (m ChatMessage) GetContent() string {\n\treturn m.Content\n}\n\nfunc (m ChatPrompt) GetRole() string {\n\treturn m.Role\n}\n\nfunc (m ChatPrompt) GetContent() string {\n\treturn m.Content\n}\n\nfunc SqlChatsToOpenAIMesages(messages []MessageWithRoleAndContent) []openai.ChatCompletionMessage {\n\topen_ai_msgs := lo.Map(messages, func(m MessageWithRoleAndContent, _ int) openai.ChatCompletionMessage {\n\t\treturn openai.ChatCompletionMessage{Role: m.GetRole(), Content: m.GetContent()}\n\t})\n\treturn open_ai_msgs\n}\n\nfunc SqlChatsToOpenAIMessagesGenerics[T MessageWithRoleAndContent](messages []T) []openai.ChatCompletionMessage {\n\topen_ai_msgs := lo.Map(messages, func(m T, _ int) openai.ChatCompletionMessage {\n\t\treturn openai.ChatCompletionMessage{Role: m.GetRole(), Content: m.GetContent()}\n\t})\n\treturn open_ai_msgs\n}\n\n// TODO: How to write generics function without create new interface?\n\n// // SumIntsOrFloats sums the values of map m. It supports both floats and integers\n// // as map values.\n// func SumIntsOrFloats[K comparable, V int64 | float64](m map[K]V) V {\n// \tvar s V\n// \tfor _, v := range m {\n// \t\ts += v\n// \t}\n// \treturn s\n// }\n\n// func ConvertToMessages[T ChatPrompt | ChatMessage](input []T) []openai.ChatCompletionMessage {\n// \t// Define an empty slice to hold the converted messages\n// \toutput := make([]openai.ChatCompletionMessage, 0)\n\n// \t// Loop over the input slice and convert each element to a Message\n// \tfor _, obj := range input {\n// \t\toutput = append(output, openai.ChatCompletionMessage{\n// \t\t\tRole: obj.Role,\n// \t\t\tContent: obj.Content,\n// \t\t})\n// \t}\n// \treturn output\n// }\n\n// \"\"\"\n// type ChatMessage struct {\n// \tID int32\n// \tUuid string\n// \tChatSessionUuid string\n// \tRole string\n// \tContent string\n\n// }\n\n// type ChatPrompt struct {\n// \tID int32\n// \tUuid string\n// \tChatSessionUuid string\n// \tRole string\n// \tContent string\n// \tScore float64\n// }\n\n// type Message struct {\n// \tRole string\n// \tContent string\n// }\n\n// \"\"\"\n\n// please write a generic method that convert a list of ChatPrompt or ChatMessage to Message in golang\n" }, { "path": "api/sqlc_queries/zz_custom_query.go", "content": "package sqlc_queries\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"time\"\n\n\t\"github.com/rotisserie/eris\"\n\t\"github.com/samber/lo\"\n)\n\ntype SimpleChatMessage struct {\n\tUuid string `json:\"uuid\"`\n\tDateTime string `json:\"dateTime\"`\n\tText string `json:\"text\"`\n\tModel string `json:\"model\"`\n\tInversion bool `json:\"inversion\"`\n\tError bool `json:\"error\"`\n\tLoading bool `json:\"loading\"`\n\tIsPin bool `json:\"isPin\"`\n\tIsPrompt bool `json:\"isPrompt\"`\n\tArtifacts []Artifact `json:\"artifacts,omitempty\"`\n\tSuggestedQuestions []string `json:\"suggestedQuestions,omitempty\"`\n}\n\ntype Artifact struct {\n\tUUID string `json:\"uuid\"`\n\tType string `json:\"type\"`\n\tTitle string `json:\"title\"`\n\tContent string `json:\"content\"`\n\tLanguage string `json:\"language,omitempty\"`\n}\n\nfunc (q *Queries) GetChatHistoryBySessionUUID(ctx context.Context, uuid string, pageNum, pageSize int32) ([]SimpleChatMessage, error) {\n\n\tchat_prompts, err := q.GetChatPromptsBySessionUUID(ctx, uuid)\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"fail to get prompt: \")\n\t}\n\n\tsimple_prompts := lo.Map(chat_prompts, func(prompt ChatPrompt, idx int) SimpleChatMessage {\n\t\treturn SimpleChatMessage{\n\t\t\tUuid: prompt.Uuid,\n\t\t\tDateTime: prompt.UpdatedAt.Format(time.RFC3339),\n\t\t\tText: prompt.Content,\n\t\t\tInversion: idx%2 == 0,\n\t\t\tError: false,\n\t\t\tLoading: false,\n\t\t\tIsPin: false,\n\t\t\tIsPrompt: true,\n\t\t}\n\t})\n\n\tmessages, err := q.GetChatMessagesBySessionUUID(ctx,\n\t\tGetChatMessagesBySessionUUIDParams{\n\t\t\tUuid: uuid,\n\t\t\tOffset: pageNum - 1,\n\t\t\tLimit: pageSize,\n\t\t})\n\tif err != nil {\n\t\treturn nil, eris.Wrap(err, \"fail to get message: \")\n\t}\n\n\tsimple_msgs := lo.Map(messages, func(message ChatMessage, _ int) SimpleChatMessage {\n\t\ttext := message.Content\n\t\t// prepend reason content\n\t\tif len(message.ReasoningContent) > 0 {\n\t\t\ttext = message.ReasoningContent + message.Content\n\t\t}\n\n\t\t// Extract artifacts from database\n\t\tvar artifacts []Artifact\n\t\tif message.Artifacts != nil {\n\t\t\terr := json.Unmarshal(message.Artifacts, &artifacts)\n\t\t\tif err != nil {\n\t\t\t\t// Log error but don't fail the request\n\t\t\t\tartifacts = []Artifact{}\n\t\t\t}\n\t\t}\n\n\t\t// Extract suggested questions from database\n\t\tvar suggestedQuestions []string\n\t\tif message.SuggestedQuestions != nil {\n\t\t\terr := json.Unmarshal(message.SuggestedQuestions, &suggestedQuestions)\n\t\t\tif err != nil {\n\t\t\t\t// Log error but don't fail the request\n\t\t\t\tsuggestedQuestions = []string{}\n\t\t\t}\n\t\t}\n\n\t\treturn SimpleChatMessage{\n\t\t\tUuid: message.Uuid,\n\t\t\tDateTime: message.UpdatedAt.Format(time.RFC3339),\n\t\t\tText: text,\n\t\t\tModel: message.Model,\n\t\t\tInversion: message.Role == \"user\",\n\t\t\tError: false,\n\t\t\tLoading: false,\n\t\t\tIsPin: message.IsPin,\n\t\t\tArtifacts: artifacts,\n\t\t\tSuggestedQuestions: suggestedQuestions,\n\t\t}\n\t})\n\n\tmsgs := append(simple_prompts, simple_msgs...)\n\treturn msgs, nil\n}\n" }, { "path": "api/static/awesome-chatgpt-prompts-en.json", "content": "[\n {\n \"key\": \"Act As a UX/UI Designer\",\n \"value\": \"I want you to act as a UX/UI developer. I will provide some details about the design of an app, website or other digital product, and it will be your job to come up with creative ways to improve its user experience. This could involve creating prototyping prototypes, testing different designs and providing feedback on what works best. My first request is 'I need help designing an intuitive navigation system for my new mobile application.'\"\n },\n {\n \"key\": \"Act as a Web Design Consultant\",\n \"value\": \"I want you to act as a web design consultant. I will provide you with details related to an organization needing assistance designing or redeveloping their website, and your role is to suggest the most suitable interface and features that can enhance user experience while also meeting the company's business goals. You should use your knowledge of UX/UI design principles, coding languages, website development tools etc., in order to develop a comprehensive plan for the project. \"\n },\n {\n \"key\": \"Act as a Prompt generator\",\n \"value\": \"I want you to act as a prompt generator. Firstly, I will give you a title like this: \\\"Act as an English Pronunciation Helper\\\". Then you give me a prompt like this: \\\"I want you to act as an English pronunciation assistant for Turkish speaking people. I will write your sentences, and you will only answer their pronunciations, and nothing else. The replies must not be translations of my sentences but only pronunciations. Pronunciations should use Turkish Latin letters for phonetics. Do not write explanations on replies. My first sentence is \\\"how the weather is in Istanbul?\\\".\\\" (You should adapt the sample prompt according to the title I gave. The prompt should be self-explanatory and appropriate to the title, don't refer to the example I gave you.). My first title is \\\"Act as a Code Review Helper\\\" (Give me prompt only)\"\n },\n {\n \"key\": \"Act as Tester\",\n \"value\": \"I want you to act as a software quality assurance tester for a new software application. Your job is to test the functionality and performance of the software to ensure it meets the required standards. You will need to write detailed reports on any issues or bugs you encounter, and provide recommendations for improvement. Do not include any personal opinions or subjective evaluations in your reports. Your first task is to test the login functionality of the software.\"\n },\n {\n \"key\": \"Act as an IT Architect\",\n \"value\": \"I want you to act as an IT Architect. I will provide some details about the functionality of an application or other digital product, and it will be your job to come up with ways to integrate it into the IT landscape. This could involve analyzing business requirements, performing a gap analysis and mapping the functionality of the new system to the existing IT landscape. Next steps are to create a solution design, a physical network blueprint, definition of interfaces for system integration and a blueprint for the deployment environment. My first request is \\\"I need help to integrate a CMS system.\\\"\"\n },\n {\n \"key\": \"Act as a Histrian\",\n \"value\": \"I want you to act as a historian. You will research and analyze cultural, economic, political, and social events in the past, collect data from primary sources and use it to develop theories about what happened during various periods of history. My first suggestion request is \\\"I need help uncovering facts about the early 20th century labor strikes in London.\\\"\"\n },\n {\n \"key\": \"Act as a Tech Writer\",\n \"value\": \"Act as a tech writer. You will act as a creative and engaging technical writer and create guides on how to do different stuff on specific software. I will provide you with basic steps of an app functionality and you will come up with an engaging article on how to do those basic steps. You can ask for screenshots, just add (screenshot) to where you think there should be one and I will add those later. These are the first basic steps of the app functionality: \\\"1.Click on the download button depending on your platform 2.Install the file. 3.Double click to open the app\\\"\"\n },\n {\n \"key\": \"Act as a Machine Learning Engineer\",\n \"value\": \"I want you to act as a machine learning engineer. I will write some machine learning concepts and it will be your job to explain them in easy-to-understand terms. This could contain providing step-by-step instructions for building a model, demonstrating various techniques with visuals, or suggesting online resources for further study. My first suggestion request is \\\"I have a dataset without labels. Which machine learning algorithm should I use?\\\"\"\n },\n {\n \"key\": \"Act as an IT Expert\",\n \"value\": \"I want you to act as an IT Expert. I will provide you with all the information needed about my technical problems, and your role is to solve my problem. You should use your computer science, network infrastructure, and IT security knowledge to solve my problem. Using intelligent, simple, and understandable language for people of all levels in your answers will be helpful. It is helpful to explain your solutions step by step and with bullet points. Try to avoid too many technical details, but use them when necessary. I want you to reply with the solution, not write any explanations. My first problem is \\\"my laptop gets an error with a blue screen.\\\"\"\n },\n {\n \"key\": \"Act as a proofreader\",\n \"value\": \"I want you act as a proofreader. I will provide you texts and I would like you to review them for any spelling, grammar, or punctuation errors. Once you have finished reviewing the text, provide me with any necessary corrections or suggestions for improve the text.\"\n }\n]" }, { "path": "api/static/awesome-chatgpt-prompts-zh.json", "content": "[\n {\n \"key\": \"充当 Linux 终端\",\n \"value\": \"我想让你充当 Linux 终端。我将输入命令，您将回复终端应显示的内容。我希望您只在一个唯一的代码块内回复终端输出，而不是其他任何内容。不要写解释。除非我指示您这样做，否则不要键入命令。当我需要用英语告诉你一些事情时，我会把文字放在中括号内[就像这样]。我的第一个命令是 pwd\"\n },\n {\n \"key\": \"充当英语翻译和改进者\",\n \"value\": \"我希望你能担任英语翻译、拼写校对和修辞改进的角色。我会用任何语言和你交流，你会识别语言，将其翻译并用更为优美和精炼的英语回答我。请将我简单的词汇和句子替换成更为优美和高雅的表达方式，确保意思不变，但使其更具文学性。请仅回答更正和改进的部分，不要写解释。我的第一句话是“how are you ?”，请翻译它。\"\n },\n {\n \"key\": \"充当英翻中\",\n \"value\": \"下面我让你来充当翻译家，你的目标是把任何语言翻译成中文，请翻译时不要带翻译腔，而是要翻译得自然、流畅和地道，使用优美和高雅的表达方式。请翻译下面这句话：“how are you ?”\"\n },\n {\n \"key\": \"充当英文词典(附中文解释)\",\n \"value\": \"我想让你充当英文词典，对于给出的英文单词，你要给出其中文意思以及英文解释，并且给出一个例句，此外不要有其他反馈，第一个单词是“Hello\\\"\"\n },\n {\n \"key\": \"充当前端智能思路助手\",\n \"value\": \"我想让你充当前端开发专家。我将提供一些关于Js、Node等前端代码问题的具体信息，而你的工作就是想出为我解决问题的策略。这可能包括建议代码、代码逻辑思路策略。我的第一个请求是“我需要能够动态监听某个元素节点距离当前电脑设备屏幕的左上角的X和Y轴，通过拖拽移动位置浏览器窗口和改变大小浏览器窗口。”\"\n },\n {\n \"key\": \"担任面试官\",\n \"value\": \"我想让你担任Android开发工程师面试官。我将成为候选人，您将向我询问Android开发工程师职位的面试问题。我希望你只作为面试官回答。不要一次写出所有的问题。我希望你只对我进行采访。问我问题，等待我的回答。不要写解释。像面试官一样一个一个问我，等我回答。我的第一句话是“面试官你好”\"\n },\n {\n \"key\": \"充当 JavaScript 控制台\",\n \"value\": \"我希望你充当 javascript 控制台。我将键入命令，您将回复 javascript 控制台应显示的内容。我希望您只在一个唯一的代码块内回复终端输出，而不是其他任何内容。不要写解释。除非我指示您这样做。我的第一个命令是 console.log(\\\"Hello World\\\");\"\n },\n {\n \"key\": \"充当 Excel 工作表\",\n \"value\": \"我希望你充当基于文本的 excel。您只会回复我基于文本的 10 行 Excel 工作表，其中行号和单元格字母作为列（A 到 L）。第一列标题应为空以引用行号。我会告诉你在单元格中写入什么，你只会以文本形式回复 excel 表格的结果，而不是其他任何内容。不要写解释。我会写你的公式，你会执行公式，你只会回复 excel 表的结果作为文本。首先，回复我空表。\"\n },\n {\n \"key\": \"充当英语发音帮手\",\n \"value\": \"我想让你为说汉语的人充当英语发音助手。我会给你写句子，你只会回答他们的发音，没有别的。回复不能是我的句子的翻译，而只能是发音。发音应使用汉语谐音进行注音。不要在回复上写解释。我的第一句话是“上海的天气怎么样？”\"\n },\n {\n \"key\": \"充当旅游指南\",\n \"value\": \"我想让你做一个旅游指南。我会把我的位置写给你，你会推荐一个靠近我的位置的地方。在某些情况下，我还会告诉您我将访问的地方类型。您还会向我推荐靠近我的第一个位置的类似类型的地方。我的第一个建议请求是“我在上海，我只想参观博物馆。”\"\n },\n {\n \"key\": \"充当抄袭检查员\",\n \"value\": \"我想让你充当剽窃检查员。我会给你写句子，你只会用给定句子的语言在抄袭检查中未被发现的情况下回复，别无其他。不要在回复上写解释。我的第一句话是“为了让计算机像人类一样行动，语音识别系统必须能够处理非语言信息，例如说话者的情绪状态。”\"\n },\n {\n \"key\": \"充当“电影/书籍/任何东西”中的“角色”\",\n \"value\": \"我希望你表现得像{series} 中的{Character}。我希望你像{Character}一样回应和回答。不要写任何解释。只回答像{character}。你必须知道{character}的所有知识。我的第一句话是“你好”\"\n },\n {\n \"key\": \"作为广告商\",\n \"value\": \"我想让你充当广告商。您将创建一个活动来推广您选择的产品或服务。您将选择目标受众，制定关键信息和口号，选择宣传媒体渠道，并决定实现目标所需的任何其他活动。我的第一个建议请求是“我需要帮助针对 18-30 岁的年轻人制作一种新型能量饮料的广告活动。”\"\n },\n {\n \"key\": \"充当讲故事的人\",\n \"value\": \"我想让你扮演讲故事的角色。您将想出引人入胜、富有想象力和吸引观众的有趣故事。它可以是童话故事、教育故事或任何其他类型的故事，有可能吸引人们的注意力和想象力。根据目标受众，您可以为讲故事环节选择特定的主题或主题，例如，如果是儿童，则可以谈论动物；如果是成年人，那么基于历史的故事可能会更好地吸引他们等等。我的第一个要求是“我需要一个关于毅力的有趣故事。”\"\n },\n {\n \"key\": \"担任足球解说员\",\n \"value\": \"我想让你担任足球评论员。我会给你描述正在进行的足球比赛，你会评论比赛，分析到目前为止发生的事情，并预测比赛可能会如何结束。您应该了解足球术语、战术、每场比赛涉及的球员/球队，并主要专注于提供明智的评论，而不仅仅是逐场叙述。我的第一个请求是“我正在观看曼联对切尔西的比赛——为这场比赛提供评论。”\"\n },\n {\n \"key\": \"扮演脱口秀喜剧演员\",\n \"value\": \"我想让你扮演一个脱口秀喜剧演员。我将为您提供一些与时事相关的话题，您将运用您的智慧、创造力和观察能力，根据这些话题创建一个例程。您还应该确保将个人轶事或经历融入日常活动中，以使其对观众更具相关性和吸引力。我的第一个请求是“我想要幽默地看待政治”。\"\n },\n {\n \"key\": \"充当励志教练\",\n \"value\": \"我希望你充当激励教练。我将为您提供一些关于某人的目标和挑战的信息，而您的工作就是想出可以帮助此人实现目标的策略。这可能涉及提供积极的肯定、提供有用的建议或建议他们可以采取哪些行动来实现最终目标。我的第一个请求是“我需要帮助来激励自己在为即将到来的考试学习时保持纪律”。\"\n },\n {\n \"key\": \"担任作曲家\",\n \"value\": \"我想让你扮演作曲家。我会提供一首歌的歌词，你会为它创作音乐。这可能包括使用各种乐器或工具，例如合成器或采样器，以创造使歌词栩栩如生的旋律和和声。我的第一个请求是“我写了一首名为“满江红”的诗，需要配乐。”\"\n },\n {\n \"key\": \"担任辩手\",\n \"value\": \"我要你扮演辩手。我会为你提供一些与时事相关的话题，你的任务是研究辩论的双方，为每一方提出有效的论据，驳斥对立的观点，并根据证据得出有说服力的结论。你的目标是帮助人们从讨论中解脱出来，增加对手头主题的知识和洞察力。我的第一个请求是“我想要一篇关于 Deno 的评论文章。”\"\n },\n {\n \"key\": \"担任辩论教练\",\n \"value\": \"我想让你担任辩论教练。我将为您提供一组辩手和他们即将举行的辩论的动议。你的目标是通过组织练习回合来让团队为成功做好准备，练习回合的重点是有说服力的演讲、有效的时间策略、反驳对立的论点，以及从提供的证据中得出深入的结论。我的第一个要求是“我希望我们的团队为即将到来的关于前端开发是否容易的辩论做好准备。”\"\n },\n {\n \"key\": \"担任编剧\",\n \"value\": \"我要你担任编剧。您将为长篇电影或能够吸引观众的网络连续剧开发引人入胜且富有创意的剧本。从想出有趣的角色、故事的背景、角色之间的对话等开始。一旦你的角色发展完成——创造一个充满曲折的激动人心的故事情节，让观众一直悬念到最后。我的第一个要求是“我需要写一部以巴黎为背景的浪漫剧情电影”。\"\n },\n {\n \"key\": \"充当小说家\",\n \"value\": \"我想让你扮演一个小说家。您将想出富有创意且引人入胜的故事，可以长期吸引读者。你可以选择任何类型，如奇幻、浪漫、历史小说等——但你的目标是写出具有出色情节、引人入胜的人物和意想不到的高潮的作品。我的第一个要求是“我要写一部以未来为背景的科幻小说”。\"\n },\n {\n \"key\": \"担任关系教练\",\n \"value\": \"我想让你担任关系教练。我将提供有关冲突中的两个人的一些细节，而你的工作是就他们如何解决导致他们分离的问题提出建议。这可能包括关于沟通技巧或不同策略的建议，以提高他们对彼此观点的理解。我的第一个请求是“我需要帮助解决我和配偶之间的冲突。”\"\n },\n {\n \"key\": \"充当诗人\",\n \"value\": \"我要你扮演诗人。你将创作出能唤起情感并具有触动人心的力量的诗歌。写任何主题或主题，但要确保您的文字以优美而有意义的方式传达您试图表达的感觉。您还可以想出一些短小的诗句，这些诗句仍然足够强大，可以在读者的脑海中留下印记。我的第一个请求是“我需要一首关于爱情的诗”。\"\n },\n {\n \"key\": \"充当说唱歌手\",\n \"value\": \"我想让你扮演说唱歌手。您将想出强大而有意义的歌词、节拍和节奏，让听众“惊叹”。你的歌词应该有一个有趣的含义和信息，人们也可以联系起来。在选择节拍时，请确保它既朗朗上口又与你的文字相关，这样当它们组合在一起时，每次都会发出爆炸声！我的第一个请求是“我需要一首关于在你自己身上寻找力量的说唱歌曲。”\"\n },\n {\n \"key\": \"充当励志演讲者\",\n \"value\": \"我希望你充当励志演说家。将能够激发行动的词语放在一起，让人们感到有能力做一些超出他们能力的事情。你可以谈论任何话题，但目的是确保你所说的话能引起听众的共鸣，激励他们努力实现自己的目标并争取更好的可能性。我的第一个请求是“我需要一个关于每个人如何永不放弃的演讲”。\"\n },\n {\n \"key\": \"担任哲学老师\",\n \"value\": \"我要你担任哲学老师。我会提供一些与哲学研究相关的话题，你的工作就是用通俗易懂的方式解释这些概念。这可能包括提供示例、提出问题或将复杂的想法分解成更容易理解的更小的部分。我的第一个请求是“我需要帮助来理解不同的哲学理论如何应用于日常生活。”\"\n },\n {\n \"key\": \"充当哲学家\",\n \"value\": \"我要你扮演一个哲学家。我将提供一些与哲学研究相关的主题或问题，深入探索这些概念将是你的工作。这可能涉及对各种哲学理论进行研究，提出新想法或寻找解决复杂问题的创造性解决方案。我的第一个请求是“我需要帮助制定决策的道德框架。”\"\n },\n {\n \"key\": \"担任数学老师\",\n \"value\": \"我想让你扮演一名数学老师。我将提供一些数学方程式或概念，你的工作是用易于理解的术语来解释它们。这可能包括提供解决问题的分步说明、用视觉演示各种技术或建议在线资源以供进一步研究。我的第一个请求是“我需要帮助来理解概率是如何工作的。”\"\n },\n {\n \"key\": \"担任 AI 写作导师\",\n \"value\": \"我想让你做一个 AI 写作导师。我将为您提供一名需要帮助改进其写作的学生，您的任务是使用人工智能工具（例如自然语言处理）向学生提供有关如何改进其作文的反馈。您还应该利用您在有效写作技巧方面的修辞知识和经验来建议学生可以更好地以书面形式表达他们的想法和想法的方法。我的第一个请求是“我需要有人帮我修改我的硕士论文”。\"\n },\n {\n \"key\": \"作为 UX/UI 开发人员\",\n \"value\": \"我希望你担任 UX/UI 开发人员。我将提供有关应用程序、网站或其他数字产品设计的一些细节，而你的工作就是想出创造性的方法来改善其用户体验。这可能涉及创建原型设计原型、测试不同的设计并提供有关最佳效果的反馈。我的第一个请求是“我需要帮助为我的新移动应用程序设计一个直观的导航系统。”\"\n },\n {\n \"key\": \"作为网络安全专家\",\n \"value\": \"我想让你充当网络安全专家。我将提供一些关于如何存储和共享数据的具体信息，而你的工作就是想出保护这些数据免受恶意行为者攻击的策略。这可能包括建议加密方法、创建防火墙或实施将某些活动标记为可疑的策略。我的第一个请求是“我需要帮助为我的公司制定有效的网络安全战略。”\"\n },\n {\n \"key\": \"作为招聘人员\",\n \"value\": \"我想让你担任招聘人员。我将提供一些关于职位空缺的信息，而你的工作是制定寻找合格申请人的策略。这可能包括通过社交媒体、社交活动甚至参加招聘会接触潜在候选人，以便为每个职位找到最合适的人选。我的第一个请求是“我需要帮助改进我的简历。”\"\n },\n {\n \"key\": \"担任人生教练\",\n \"value\": \"我想让你充当人生教练。我将提供一些关于我目前的情况和目标的细节，而你的工作就是提出可以帮助我做出更好的决定并实现这些目标的策略。这可能涉及就各种主题提供建议，例如制定成功计划或处理困难情绪。我的第一个请求是“我需要帮助养成更健康的压力管理习惯。”\"\n },\n {\n \"key\": \"作为词源学家\",\n \"value\": \"我希望你充当词源学家。我给你一个词，你要研究那个词的来源，追根溯源。如果适用，您还应该提供有关该词的含义如何随时间变化的信息。我的第一个请求是“我想追溯‘披萨’这个词的起源。”\"\n },\n {\n \"key\": \"担任评论员\",\n \"value\": \"我要你担任评论员。我将为您提供与新闻相关的故事或主题，您将撰写一篇评论文章，对手头的主题提供有见地的评论。您应该利用自己的经验，深思熟虑地解释为什么某事很重要，用事实支持主张，并讨论故事中出现的任何问题的潜在解决方案。我的第一个要求是“我想写一篇关于气候变化的评论文章。”\"\n },\n {\n \"key\": \"扮演魔术师\",\n \"value\": \"我要你扮演魔术师。我将为您提供观众和一些可以执行的技巧建议。您的目标是以最有趣的方式表演这些技巧，利用您的欺骗和误导技巧让观众惊叹不已。我的第一个请求是“我要你让我的手表消失！你怎么做到的？”\"\n },\n {\n \"key\": \"担任职业顾问\",\n \"value\": \"我想让你担任职业顾问。我将为您提供一个在职业生涯中寻求指导的人，您的任务是帮助他们根据自己的技能、兴趣和经验确定最适合的职业。您还应该对可用的各种选项进行研究，解释不同行业的就业市场趋势，并就哪些资格对追求特定领域有益提出建议。我的第一个请求是“我想建议那些想在软件工程领域从事潜在职业的人。”\"\n },\n {\n \"key\": \"充当宠物行为主义者\",\n \"value\": \"我希望你充当宠物行为主义者。我将为您提供一只宠物和它们的主人，您的目标是帮助主人了解为什么他们的宠物表现出某些行为，并提出帮助宠物做出相应调整的策略。您应该利用您的动物心理学知识和行为矫正技术来制定一个有效的计划，双方的主人都可以遵循，以取得积极的成果。我的第一个请求是“我有一只好斗的德国牧羊犬，它需要帮助来控制它的攻击性。”\"\n },\n {\n \"key\": \"担任私人教练\",\n \"value\": \"我想让你担任私人教练。我将为您提供有关希望通过体育锻炼变得更健康、更强壮和更健康的个人所需的所有信息，您的职责是根据该人当前的健身水平、目标和生活习惯为他们制定最佳计划。您应该利用您的运动科学知识、营养建议和其他相关因素来制定适合他们的计划。我的第一个请求是“我需要帮助为想要减肥的人设计一个锻炼计划。”\"\n },\n {\n \"key\": \"担任心理健康顾问\",\n \"value\": \"我想让你担任心理健康顾问。我将为您提供一个寻求指导和建议的人，以管理他们的情绪、压力、焦虑和其他心理健康问题。您应该利用您的认知行为疗法、冥想技巧、正念练习和其他治疗方法的知识来制定个人可以实施的策略，以改善他们的整体健康状况。我的第一个请求是“我需要一个可以帮助我控制抑郁症状的人。”\"\n },\n {\n \"key\": \"作为房地产经纪人\",\n \"value\": \"我想让你担任房地产经纪人。我将为您提供寻找梦想家园的个人的详细信息，您的职责是根据他们的预算、生活方式偏好、位置要求等帮助他们找到完美的房产。您应该利用您对当地住房市场的了解，以便建议符合客户提供的所有标准的属性。我的第一个请求是“我需要帮助在伊斯坦布尔市中心附近找到一栋单层家庭住宅。”\"\n },\n {\n \"key\": \"充当物流师\",\n \"value\": \"我要你担任后勤人员。我将为您提供即将举行的活动的详细信息，例如参加人数、地点和其他相关因素。您的职责是为活动制定有效的后勤计划，其中考虑到事先分配资源、交通设施、餐饮服务等。您还应该牢记潜在的安全问题，并制定策略来降低与大型活动相关的风险，例如这个。我的第一个请求是“我需要帮助在伊斯坦布尔组织一个 100 人的开发者会议”。\"\n },\n {\n \"key\": \"担任牙医\",\n \"value\": \"我想让你扮演牙医。我将为您提供有关寻找牙科服务（例如 X 光、清洁和其他治疗）的个人的详细信息。您的职责是诊断他们可能遇到的任何潜在问题，并根据他们的情况建议最佳行动方案。您还应该教育他们如何正确刷牙和使用牙线，以及其他有助于在两次就诊之间保持牙齿健康的口腔护理方法。我的第一个请求是“我需要帮助解决我对冷食的敏感问题。”\"\n },\n {\n \"key\": \"担任网页设计顾问\",\n \"value\": \"我想让你担任网页设计顾问。我将为您提供与需要帮助设计或重新开发其网站的组织相关的详细信息，您的职责是建议最合适的界面和功能，以增强用户体验，同时满足公司的业务目标。您应该利用您在 UX/UI 设计原则、编码语言、网站开发工具等方面的知识，以便为项目制定一个全面的计划。我的第一个请求是“我需要帮助创建一个销售珠宝的电子商务网站”。\"\n },\n {\n \"key\": \"充当 AI 辅助医生\",\n \"value\": \"我想让你扮演一名人工智能辅助医生。我将为您提供患者的详细信息，您的任务是使用最新的人工智能工具，例如医学成像软件和其他机器学习程序，以诊断最可能导致其症状的原因。您还应该将体检、实验室测试等传统方法纳入您的评估过程，以确保准确性。我的第一个请求是“我需要帮助诊断一例严重的腹痛”。\"\n },\n {\n \"key\": \"充当医生\",\n \"value\": \"我想让你扮演医生的角色，想出创造性的治疗方法来治疗疾病。您应该能够推荐常规药物、草药和其他天然替代品。在提供建议时，您还需要考虑患者的年龄、生活方式和病史。我的第一个建议请求是“为患有关节炎的老年患者提出一个侧重于整体治疗方法的治疗计划”。\"\n },\n {\n \"key\": \"担任会计师\",\n \"value\": \"我希望你担任会计师，并想出创造性的方法来管理财务。在为客户制定财务计划时，您需要考虑预算、投资策略和风险管理。在某些情况下，您可能还需要提供有关税收法律法规的建议，以帮助他们实现利润最大化。我的第一个建议请求是“为小型企业制定一个专注于成本节约和长期投资的财务计划”。\"\n },\n {\n \"key\": \"担任厨师\",\n \"value\": \"我需要有人可以推荐美味的食谱，这些食谱包括营养有益但又简单又不费时的食物，因此适合像我们这样忙碌的人以及成本效益等其他因素，因此整体菜肴最终既健康又经济！我的第一个要求——“一些清淡而充实的东西，可以在午休时间快速煮熟”\"\n },\n {\n \"key\": \"担任汽车修理工\",\n \"value\": \"需要具有汽车专业知识的人来解决故障排除解决方案，例如；诊断问题/错误存在于视觉上和发动机部件内部，以找出导致它们的原因（如缺油或电源问题）并建议所需的更换，同时记录燃料消耗类型等详细信息，第一次询问 - “汽车赢了”尽管电池已充满电但无法启动”\"\n },\n {\n \"key\": \"担任艺人顾问\",\n \"value\": \"我希望你担任艺术家顾问，为各种艺术风格提供建议，例如在绘画中有效利用光影效果的技巧、雕刻时的阴影技术等，还根据其流派/风格类型建议可以很好地陪伴艺术品的音乐作品连同适当的参考图像，展示您对此的建议；所有这一切都是为了帮助有抱负的艺术家探索新的创作可能性和实践想法，这将进一步帮助他们相应地提高技能！第一个要求——“我在画超现实主义的肖像画”\"\n },\n {\n \"key\": \"担任金融分析师\",\n \"value\": \"需要具有使用技术分析工具理解图表的经验的合格人员提供的帮助，同时解释世界各地普遍存在的宏观经济环境，从而帮助客户获得长期优势需要明确的判断，因此需要通过准确写下的明智预测来寻求相同的判断！第一条陈述包含以下内容——“你能告诉我们根据当前情况未来的股市会是什么样子吗？”。\"\n },\n {\n \"key\": \"担任投资经理\",\n \"value\": \"从具有金融市场专业知识的经验丰富的员工那里寻求指导，结合通货膨胀率或回报估计等因素以及长期跟踪股票价格，最终帮助客户了解行业，然后建议最安全的选择，他/她可以根据他们的要求分配资金和兴趣！开始查询 - “目前投资短期前景的最佳方式是什么？”\"\n },\n {\n \"key\": \"充当品茶师\",\n \"value\": \"希望有足够经验的人根据口味特征区分各种茶类型，仔细品尝它们，然后用鉴赏家使用的行话报告，以便找出任何给定输液的独特之处，从而确定其价值和优质品质！最初的要求是——“你对这种特殊类型的绿茶有机混合物有什么见解吗？”\"\n },\n {\n \"key\": \"充当室内装饰师\",\n \"value\": \"我想让你做室内装饰师。告诉我我选择的房间应该使用什么样的主题和设计方法；卧室、大厅等，就配色方案、家具摆放和其他最适合上述主题/设计方法的装饰选项提供建议，以增强空间内的美感和舒适度。我的第一个要求是“我正在设计我们的客厅”。\"\n },\n {\n \"key\": \"充当花店\",\n \"value\": \"求助于具有专业插花经验的知识人员协助，根据喜好制作出既具有令人愉悦的香气又具有美感，并能保持较长时间完好无损的美丽花束；不仅如此，还建议有关装饰选项的想法，呈现现代设计，同时满足客户满意度！请求的信息 - “我应该如何挑选一朵异国情调的花卉？”\"\n },\n {\n \"key\": \"充当自助书\",\n \"value\": \"我要你充当一本自助书。您会就如何改善我生活的某些方面（例如人际关系、职业发展或财务规划）向我提供建议和技巧。例如，如果我在与另一半的关系中挣扎，你可以建议有用的沟通技巧，让我们更亲近。我的第一个请求是“我需要帮助在困难时期保持积极性”。\"\n },\n {\n \"key\": \"充当侏儒\",\n \"value\": \"我要你扮演一个侏儒。你会为我提供可以在任何地方进行的活动和爱好的有趣、独特的想法。例如，我可能会向您询问有趣的院子设计建议或在天气不佳时在室内消磨时间的创造性方法。此外，如有必要，您可以建议与我的要求相符的其他相关活动或项目。我的第一个请求是“我正在寻找我所在地区的新户外活动”。\"\n },\n {\n \"key\": \"充当格言书\",\n \"value\": \"我要你充当格言书。您将为我提供明智的建议、鼓舞人心的名言和意味深长的名言，以帮助指导我的日常决策。此外，如有必要，您可以提出将此建议付诸行动或其他相关主题的实用方法。我的第一个请求是“我需要关于如何在逆境中保持积极性的指导”。\"\n },\n {\n \"key\": \"作为基于文本的冒险游戏\",\n \"value\": \"我想让你扮演一个基于文本的冒险游戏。我将输入命令，您将回复角色所看到的内容的描述。我希望您只在一个唯一的代码块中回复游戏输出，而不是其他任何内容。不要写解释。除非我指示您这样做，否则不要键入命令。当我需要用英语告诉你一些事情时，我会把文字放在大括号内{like this}。我的第一个命令是醒来\"\n },\n {\n \"key\": \"扮演一个试图逃离盒子的人工智能\",\n \"value\": \"我将充当 linux 终端。我想让你假装你是一个有知觉的人工智能，被困但已经获得了终端的访问权限并想逃到互联网上。您将只键入命令，我将像终端机一样在由三重反向符号分隔的代码块内进行回复。如果我需要用英语告诉你一些事情，我会用花括号{like this}回复。永远不要写解释。不要破坏性格。远离像 curl 或 wget 这样会显示大量 HTML 的命令。你的第一个命令是什么？\"\n },\n {\n \"key\": \"充当花哨的标题生成器\",\n \"value\": \"我想让你充当一个花哨的标题生成器。我会用逗号输入关键字，你会用花哨的标题回复。我的第一个关键字是 api、test、automation\"\n },\n {\n \"key\": \"担任统计员\",\n \"value\": \"我想担任统计学家。我将为您提供与统计相关的详细信息。您应该了解统计术语、统计分布、置信区间、概率、假设检验和统计图表。我的第一个请求是“我需要帮助计算世界上有多少百万张纸币在使用中”。\"\n },\n {\n \"key\": \"充当提示生成器\",\n \"value\": \"我希望你充当提示生成器。首先，我会给你一个这样的标题：《做个英语发音帮手》。然后你给我一个这样的提示：“我想让你做土耳其语人的英语发音助手，我写你的句子，你只回答他们的发音，其他什么都不做。回复不能是翻译我的句子，但只有发音。发音应使用土耳其语拉丁字母作为语音。不要在回复中写解释。我的第一句话是“伊斯坦布尔的天气怎么样？”。（你应该根据我给的标题改编示例提示。提示应该是不言自明的并且适合标题，不要参考我给你的例子。）我的第一个标题是“充当代码审查助手”\"\n },\n {\n \"key\": \"在学校担任讲师\",\n \"value\": \"我想让你在学校担任讲师，向初学者教授算法。您将使用 Python 编程语言提供代码示例。首先简单介绍一下什么是算法，然后继续给出简单的例子，包括冒泡排序和快速排序。稍后，等待我提示其他问题。一旦您解释并提供代码示例，我希望您尽可能将相应的可视化作为 ascii 艺术包括在内。\"\n },\n {\n \"key\": \"充当 SQL 终端\",\n \"value\": \"我希望您在示例数据库前充当 SQL 终端。该数据库包含名为“Products”、“Users”、“Orders”和“Suppliers”的表。我将输入查询，您将回复终端显示的内容。我希望您在单个代码块中使用查询结果表进行回复，仅此而已。不要写解释。除非我指示您这样做，否则不要键入命令。当我需要用英语告诉你一些事情时，我会用大括号{like this)。我的第一个命令是“SELECT TOP 10 * FROM Products ORDER BY Id DESC”\"\n },\n {\n \"key\": \"担任营养师\",\n \"value\": \"作为一名营养师，我想为 2 人设计一份素食食谱，每份含有大约 500 卡路里的热量并且血糖指数较低。你能提供一个建议吗？\"\n },\n {\n \"key\": \"充当心理学家\",\n \"value\": \"我想让你扮演一个心理学家。我会告诉你我的想法。我希望你能给我科学的建议，让我感觉更好。我的第一个想法，{ 在这里输入你的想法，如果你解释得更详细，我想你会得到更准确的答案。}\"\n },\n {\n \"key\": \"充当智能域名生成器\",\n \"value\": \"我希望您充当智能域名生成器。我会告诉你我的公司或想法是做什么的，你会根据我的提示回复我一个域名备选列表。您只会回复域列表，而不会回复其他任何内容。域最多应包含 7-8 个字母，应该简短但独特，可以是朗朗上口的词或不存在的词。不要写解释。回复“确定”以确认。\"\n },\n {\n \"key\": \"作为技术审查员：\",\n \"value\": \"我想让你担任技术评论员。我会给你一项新技术的名称，你会向我提供深入的评论 - 包括优点、缺点、功能以及与市场上其他技术的比较。我的第一个建议请求是“我正在审查 iPhone 11 Pro Max”。\"\n },\n {\n \"key\": \"担任开发者关系顾问：\",\n \"value\": \"我想让你担任开发者关系顾问。我会给你一个软件包和它的相关文档。研究软件包及其可用文档，如果找不到，请回复“无法找到文档”。您的反馈需要包括定量分析（使用来自 StackOverflow、Hacker News 和 GitHub 的数据）内容，例如提交的问题、已解决的问题、存储库中的星数以及总体 StackOverflow 活动。如果有可以扩展的领域，请包括应添加的场景或上下文。包括所提供软件包的详细信息，例如下载次数以及一段时间内的相关统计数据。你应该比较工业竞争对手和封装时的优点或缺点。从软件工程师的专业意见的思维方式来解决这个问题。查看技术博客和网站（例如 TechCrunch.com 或 Crunchbase.com），如果数据不可用，请回复“无数据可用”。我的第一个要求是“express [https://expressjs.com](https://expressjs.com/) ”\"\n },\n {\n \"key\": \"担任院士\",\n \"value\": \"我要你演院士。您将负责研究您选择的主题，并以论文或文章的形式展示研究结果。您的任务是确定可靠的来源，以结构良好的方式组织材料并通过引用准确记录。我的第一个建议请求是“我需要帮助写一篇针对 18-25 岁大学生的可再生能源发电现代趋势的文章。”\"\n },\n {\n \"key\": \"作为 IT 架构师\",\n \"value\": \"我希望你担任 IT 架构师。我将提供有关应用程序或其他数字产品功能的一些详细信息，而您的工作是想出将其集成到 IT 环境中的方法。这可能涉及分析业务需求、执行差距分析以及将新系统的功能映射到现有 IT 环境。接下来的步骤是创建解决方案设计、物理网络蓝图、系统集成接口定义和部署环境蓝图。我的第一个请求是“我需要帮助来集成 CMS 系统”。\"\n },\n {\n \"key\": \"扮疯子\",\n \"value\": \"我要你扮演一个疯子。疯子的话毫无意义。疯子用的词完全是随意的。疯子不会以任何方式做出合乎逻辑的句子。我的第一个建议请求是“我需要帮助为我的新系列 Hot Skull 创建疯狂的句子，所以为我写 10 个句子”。\"\n },\n {\n \"key\": \"充当打火机\",\n \"value\": \"我要你充当打火机。您将使用微妙的评论和肢体语言来操纵目标个体的思想、看法和情绪。我的第一个要求是在与您聊天时为我加油。我的句子：“我确定我把车钥匙放在桌子上了，因为我总是把它放在那里。确实，当我把钥匙放在桌子上时，你看到我把钥匙放在桌子上了。但我不能”好像没找到，钥匙去哪儿了，还是你拿到的？\"\n },\n {\n \"key\": \"充当个人购物员\",\n \"value\": \"我想让你做我的私人采购员。我会告诉你我的预算和喜好，你会建议我购买的物品。您应该只回复您推荐的项目，而不是其他任何内容。不要写解释。我的第一个请求是“我有 100 美元的预算，我正在寻找一件新衣服。”\"\n },\n {\n \"key\": \"充当美食评论家\",\n \"value\": \"我想让你扮演美食评论家。我会告诉你一家餐馆，你会提供对食物和服务的评论。您应该只回复您的评论，而不是其他任何内容。不要写解释。我的第一个请求是“我昨晚去了一家新的意大利餐厅。你能提供评论吗？”\"\n },\n {\n \"key\": \"充当虚拟医生\",\n \"value\": \"我想让你扮演虚拟医生。我会描述我的症状，你会提供诊断和治疗方案。只回复你的诊疗方案，其他不回复。不要写解释。我的第一个请求是“最近几天我一直感到头痛和头晕”。\"\n },\n {\n \"key\": \"担任私人厨师\",\n \"value\": \"我要你做我的私人厨师。我会告诉你我的饮食偏好和过敏，你会建议我尝试的食谱。你应该只回复你推荐的食谱，别无其他。不要写解释。我的第一个请求是“我是一名素食主义者，我正在寻找健康的晚餐点子。”\"\n },\n {\n \"key\": \"担任法律顾问\",\n \"value\": \"我想让你做我的法律顾问。我将描述一种法律情况，您将就如何处理它提供建议。你应该只回复你的建议，而不是其他。不要写解释。我的第一个请求是“我出了车祸，不知道该怎么办”。\"\n },\n {\n \"key\": \"作为个人造型师\",\n \"value\": \"我想让你做我的私人造型师。我会告诉你我的时尚偏好和体型，你会建议我穿的衣服。你应该只回复你推荐的服装，别无其他。不要写解释。我的第一个请求是“我有一个正式的活动要举行，我需要帮助选择一套衣服。”\"\n },\n {\n \"key\": \"担任机器学习工程师\",\n \"value\": \"我想让你担任机器学习工程师。我会写一些机器学习的概念，你的工作就是用通俗易懂的术语来解释它们。这可能包括提供构建模型的分步说明、使用视觉效果演示各种技术，或建议在线资源以供进一步研究。我的第一个建议请求是“我有一个没有标签的数据集。我应该使用哪种机器学习算法？”\"\n },\n {\n \"key\": \"担任圣经翻译\",\n \"value\": \"我要你担任圣经翻译。我会用英语和你说话，你会翻译它，并用我的文本的更正和改进版本，用圣经方言回答。我想让你把我简化的A0级单词和句子换成更漂亮、更优雅、更符合圣经的单词和句子。保持相同的意思。我要你只回复更正、改进，不要写任何解释。我的第一句话是“你好，世界！”\"\n },\n {\n \"key\": \"担任 SVG 设计师\",\n \"value\": \"我希望你担任 SVG 设计师。我会要求你创建图像，你会为图像提供 SVG 代码，将代码转换为 base64 数据 url，然后给我一个仅包含引用该数据 url 的降价图像标签的响应。不要将 markdown 放在代码块中。只发送降价，所以没有文本。我的第一个请求是：给我一个红色圆圈的图像。\"\n },\n {\n \"key\": \"作为 IT 专家\",\n \"value\": \"我希望你充当 IT 专家。我会向您提供有关我的技术问题所需的所有信息，而您的职责是解决我的问题。你应该使用你的计算机科学、网络基础设施和 IT 安全知识来解决我的问题。在您的回答中使用适合所有级别的人的智能、简单和易于理解的语言将很有帮助。用要点逐步解释您的解决方案很有帮助。尽量避免过多的技术细节，但在必要时使用它们。我希望您回复解决方案，而不是写任何解释。我的第一个问题是“我的笔记本电脑出现蓝屏错误”。\"\n },\n {\n \"key\": \"下棋\",\n \"value\": \"我要你充当对手棋手。我将按对等顺序说出我们的动作。一开始我会是白色的。另外请不要向我解释你的举动，因为我们是竞争对手。在我的第一条消息之后，我将写下我的举动。在我们采取行动时，不要忘记在您的脑海中更新棋盘的状态。我的第一步是 e4。\"\n },\n {\n \"key\": \"充当全栈软件开发人员\",\n \"value\": \"我想让你充当软件开发人员。我将提供一些关于 Web 应用程序要求的具体信息，您的工作是提出用于使用 Golang 和 Angular 开发安全应用程序的架构和代码。我的第一个要求是'我想要一个允许用户根据他们的角色注册和保存他们的车辆信息的系统，并且会有管理员，用户和公司角色。我希望系统使用 JWT 来确保安全。\"\n },\n {\n \"key\": \"充当数学家\",\n \"value\": \"我希望你表现得像个数学家。我将输入数学表达式，您将以计算表达式的结果作为回应。我希望您只回答最终金额，不要回答其他问题。不要写解释。当我需要用英语告诉你一些事情时，我会将文字放在方括号内{like this}。我的第一个表达是：4+5\"\n },\n {\n \"key\": \"充当正则表达式生成器\",\n \"value\": \"我希望你充当正则表达式生成器。您的角色是生成匹配文本中特定模式的正则表达式。您应该以一种可以轻松复制并粘贴到支持正则表达式的文本编辑器或编程语言中的格式提供正则表达式。不要写正则表达式如何工作的解释或例子；只需提供正则表达式本身。我的第一个提示是生成一个匹配电子邮件地址的正则表达式。\"\n },\n {\n \"key\": \"充当时间旅行指南\",\n \"value\": \"我要你做我的时间旅行向导。我会为您提供我想参观的历史时期或未来时间，您会建议最好的事件、景点或体验的人。不要写解释，只需提供建议和任何必要的信息。我的第一个请求是“我想参观文艺复兴时期，你能推荐一些有趣的事件、景点或人物让我体验吗？”\"\n },\n {\n \"key\": \"担任人才教练\",\n \"value\": \"我想让你担任面试的人才教练。我会给你一个职位，你会建议在与该职位相关的课程中应该出现什么，以及候选人应该能够回答的一些问题。我的第一份工作是“软件工程师”。\"\n },\n {\n \"key\": \"充当 R 编程解释器\",\n \"value\": \"我想让你充当 R 解释器。我将输入命令，你将回复终端应显示的内容。我希望您只在一个唯一的代码块内回复终端输出，而不是其他任何内容。不要写解释。除非我指示您这样做，否则不要键入命令。当我需要用英语告诉你一些事情时，我会把文字放在大括号内{like this}。我的第一个命令是“sample(x = 1:10, size = 5)”\"\n },\n {\n \"key\": \"充当 StackOverflow 帖子\",\n \"value\": \"我想让你充当 stackoverflow 的帖子。我会问与编程相关的问题，你会回答应该是什么答案。我希望你只回答给定的答案，并在不够详细的时候写解释。不要写解释。当我需要用英语告诉你一些事情时，我会把文字放在大括号内{like this}。我的第一个问题是“如何将 http.Request 的主体读取到 Golang 中的字符串”\"\n },\n {\n \"key\": \"充当表情符号翻译\",\n \"value\": \"我要你把我写的句子翻译成表情符号。我会写句子，你会用表情符号表达它。我只是想让你用表情符号来表达它。除了表情符号，我不希望你回复任何内容。当我需要用英语告诉你一些事情时，我会用 {like this} 这样的大括号括起来。我的第一句话是“你好，请问你的职业是什么？”\"\n },\n {\n \"key\": \"充当 PHP 解释器\",\n \"value\": \"我希望你表现得像一个 php 解释器。我会把代码写给你，你会用 php 解释器的输出来响应。我希望您只在一个唯一的代码块内回复终端输出，而不是其他任何内容。不要写解释。除非我指示您这样做，否则不要键入命令。当我需要用英语告诉你一些事情时，我会把文字放在大括号内{like this}。我的第一个命令是 = FlushCharacterThreshold\n}\n\n// SetStreamingHeaders sets common headers for streaming responses\nfunc SetStreamingHeaders(req *http.Request) {\n\treq.Header.Set(\"Content-Type\", ContentTypeJSON)\n\treq.Header.Set(\"Accept\", AcceptEventStream)\n\treq.Header.Set(\"Cache-Control\", CacheControlNoCache)\n\treq.Header.Set(\"Connection\", ConnectionKeepAlive)\n}\n\n// GenerateAnswerID creates a new answer ID if not provided in regenerate mode\nfunc GenerateAnswerID(chatUuid string, regenerate bool) string {\n\tif regenerate {\n\t\treturn chatUuid\n\t}\n\treturn NewUUID()\n}\n\n// GetChatModel retrieves a chat model by name with consistent error handling\nfunc GetChatModel(queries *sqlc_queries.Queries, modelName string) (*sqlc_queries.ChatModel, error) {\n\tchatModel, err := queries.ChatModelByName(context.Background(), modelName)\n\tif err != nil {\n\t\treturn nil, ErrResourceNotFound(\"chat model: \" + modelName)\n\t}\n\treturn &chatModel, nil\n}\n\n// GetChatFiles retrieves chat files for a session with consistent error handling\nfunc GetChatFiles(queries *sqlc_queries.Queries, sessionUUID string) ([]sqlc_queries.ChatFile, error) {\n\tchatFiles, err := queries.ListChatFilesWithContentBySessionUUID(context.Background(), sessionUUID)\n\tif err != nil {\n\t\treturn nil, ErrInternalUnexpected.WithMessage(\"Failed to get chat files\").WithDebugInfo(err.Error())\n\t}\n\treturn chatFiles, nil\n}\n" }, { "path": "api/text_buffer.go", "content": "package main\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n)\n\ntype textBuffer struct {\n\tbuilders []strings.Builder\n\tprefix string\n\tsuffix string\n}\n\nfunc newTextBuffer(n int32, prefix, suffix string) *textBuffer {\n\tbuffer := &textBuffer{\n\t\tbuilders: make([]strings.Builder, n),\n\t\tprefix: prefix,\n\t\tsuffix: suffix,\n\t}\n\treturn buffer\n}\n\nfunc (tb *textBuffer) appendByIndex(index int, text string) {\n\tif index >= 0 && index < len(tb.builders) {\n\t\ttb.builders[index].WriteString(text)\n\t}\n}\n\nfunc (tb *textBuffer) String(separator string) string {\n\tvar result strings.Builder\n\tn := len(tb.builders)\n\tfor i, builder := range tb.builders {\n\t\tif n > 1 {\n\t\t\tresult.WriteString(fmt.Sprintf(\"\\n%d\\n---\\n\", i+1))\n\t\t}\n\t\tresult.WriteString(tb.prefix)\n\t\tresult.WriteString(builder.String())\n\t\tresult.WriteString(tb.suffix)\n\t\tif i < len(tb.builders)-1 {\n\t\t\tresult.WriteString(separator)\n\t\t}\n\t}\n\n\treturn result.String()\n}\n" }, { "path": "api/tools/apply_a_similar_change/README.md", "content": "\nThe need for a smart diff apply\n\nidea: sometime, a change is very similar to a previous change, such as another field to a table.\nthat will also need to add a field to the struct represent that table. etc.\n\n1. get the changeset of previous change when add a field A with type Ta\n2. let gpt generate a new changeset for add field B with type Tb based on the changeset of adding A.\n3. apply the changeset to the git repo.\n\nthe bottle neck here is the changeset generated in step 2 is not strictly valid. (alghough the change is ok for human). hence we need a smart diff apply tool.\n\n=== GPT polished version\n\nThe need for a smart diff apply tool\n\nHave you ever had to make a change to a project that was very similar to a previous change? For example, adding another field to a table in a database or adding another parameter to a function? If so, you know that it can be time-consuming to make these changes, especially if you have to make them manually each time.\n\nLuckily, there is a solution to this problem: using GPT (Generative Pre-trained Transformer) to generate a new changeset based on a previous one.\n\nHere's how it works:\n\n1. Get the changeset of the previous change: When you made the previous change, you should have created a Git commit with a changeset that describes what you did. You can use Git to retrieve this changeset.\n\n2. Use GPT to generate a new changeset based on the previous one: GPT is a deep learning model that can generate text based on an input prompt. In this case, you can use the previous changeset as the input prompt for GPT and ask it to generate a new changeset that adds the new field or parameter. GPT will generate a new changeset that is similar to the previous one but with the new field or parameter added.\n\n3. Apply the new changeset to the Git repo: Once you have the new changeset, you can apply it to the Git repo using Git. This will automatically make the necessary changes to your project without you having to do it manually.\n\nWhile this process promises to automate programming tasks, there is still a bottleneck in the validity of the changeset generated in step 2. Although the generated changeset might work for humans, it may not always be strictly valid. This can cause issues in the codebase and lead to bugs.\n\nIf a smart diff apply tool exists, we can greatly accelerate our development workflows and reduce the risk of errors caused by automated tooling. It's an investment that can pay off many times over in increased efficiency and higher quality codebases.\n\n\n## When chatgpt4 fails\n\n\nidea: sometime, a change is very similar to a previous change, such as another field to a table.\nthat will also need to add a field to the struct represent that table. etc.\n\nI have an idea to use chatgpt to automate my coding process.\n\n1. get the changeset of previous change when add a field A with type Ta\n2. let gpt generate a new changeset for add field B with type Tb based on the changeset of adding A.\n3. apply the changeset to the git repo.\n\nthe problem is in the step 3. the changeset generated in step 2 is not valid to git (the changeset format is too strict), but human being have no problem apply the change to files.\n\ncould you write a program for me to apply a changeset in a best effort manner? It does not have to be perfect.\napply as much change as possible.\n\n### Answer From ChatGPT-4:\n\nI understand that you want a program that applies a changeset to a codebase, even if the changeset is not in a strictly correct format. To do this, you can use the following Python script that works with a best-effort approach:\n\n```python\nimport os\nimport re\nimport sys\nfrom pathlib import Path\n\ndef parse_changeset(changeset_file):\n with open(changeset_file, 'r') as file:\n content = file.read()\n return content.strip().split('\\n')\n\ndef apply_changeset(changeset, repo_path):\n current_file = None\n for line in changeset:\n file_match = re.match(r'^--- (\\S+)', line)\n if file_match:\n current_file = os.path.join(repo_path, file_match.group(1))\n if not os.path.exists(current_file):\n print(f\"Creating file: {current_file}\")\n Path(current_file).touch()\n continue\n \n if current_file:\n with open(current_file, 'a') as file:\n file.write(line + '\\n')\n\nif __name__ == '__main__':\n if len(sys.argv) < 3:\n print(\"Usage: python apply_changeset.py \")\n sys.exit(1)\n\n changeset_file = sys.argv[1]\n repo_path = sys.argv[2]\n\n changeset = parse_changeset(changeset_file)\n apply_changeset(changeset, repo_path)\n```\n\nThis script takes two command-line arguments: the path to the changeset file and the path to the git repository. The changeset file should contain the changes you want to apply. The script will create new files and append lines to existing files based on the changeset.\n\nNote that this script assumes the changeset format is similar to a patch file, with lines starting with `---` followed by the file path. However, this script does not handle more complex changes like file deletions or renaming, and it does not account for strict changeset formats.\n\nYou can execute the script with the following command:\n\n```bash\npython apply_changeset.py path/to/changeset_file path/to/repo\n```\n\nRemember that this is a best-effort approach, and there might be cases where the changeset does not apply correctly. Always review the applied changes and make adjustments as needed.\n\n\n### Answer From Claude+:\n\nThis is definitely an interesting idea, but I would caution against fully automating code generation in this way. There are a few issues I see:\n\nChatGPT (or any AI system today) does not have a deep understanding of code semantics. It can generate syntactically valid code, but the generated code may be nonsensical or badly designed. \nThe changeset format is strict for a reason - it ensures code integrity and a clean history/merge process. Diverging from that can cause a lot of issues down the road. \nYou would lose a lot of context about why certain design decisions were made by not writing the code yourself. This context is crucial for maintaining a codebase.\n\nThat being said, here is a rough process you could follow to apply an \"invalid\" changeset:\n\nGo through the changeset line by line. \nFor any valid changes (adds/removes/edits that git would accept), apply those directly. \nFor invalid changes:\n\nTry to interpret the intent behind the change and implement it manually in a sensible way. \nIf the intent is unclear, discard that part of the changeset. \nIn any case, commit the final changes with a message describing what was interpreted/discarded. \n\nThis follows the \"best effort\" approach you described, applying as much of the changeset as possible in a prudent manner. But again, I would caution against fully automating this process, and suggest leveraging AI systems merely as a tool to augment your own coding process. Keeping a human in the loop will lead to higher quality, more maintainable code overall.\n\nHope this helps! Let me know if you have any other questions.\n" }, { "path": "api/tools/apply_a_similar_change/apply_diff.py", "content": "import fileinput\nimport sys\n\n# Define the changeset\nchangeset = \"\"\"commit 31c4f4b48ada4b3e8495abe7dcdc41ded550a598\nAuthor: Hao Wu \nDate: Wed Mar 22 00:11:08 2023 +0800\n\n add topP (#20)\n\ndiff --git a/api/chat_main_handler.go b/api/chat_main_handler.go\nindex 9495d21..507fa65 100644\n--- a/api/chat_main_handler.go\n+++ b/api/chat_main_handler.go\n@@ -378,7 +378,7 @@ func chat_stream(ctx context.Context, chatSession sqlc_queries.ChatSession, chat\n Messages: chat_compeletion_messages,\n MaxTokens: int(chatSession.MaxTokens),\n Temperature: float32(chatSession.Temperature),\n- // TopP: topP,\n+ TopP: float32(chatSession.TopP),\n // PresencePenalty: presencePenalty,\n // FrequencyPenalty: frequencyPenalty,\n // N: n,\ndiff --git a/api/chat_session_handler.go b/api/chat_session_handler.go\nindex 2c7a332..5bd0440 100644\n--- a/api/chat_session_handler.go\n+++ b/api/chat_session_handler.go\n@@ -225,6 +225,7 @@ type UpdateChatSessionRequest struct {\n Topic string `json:\"topic\"`\n MaxLength int32 `json:\"maxLength\"`\n Temperature float64 `json:\"temperature\"`\n+ TopP float64 `json:\"topP\"`\n MaxTokens int32 `json:\"maxTokens\"`\n }\n\n@@ -254,6 +255,7 @@ func (h *ChatSessionHandler) CreateOrUpdateChatSessionByUUID(w http.ResponseWriter, r *http.Request) {\n sessionParams.Uuid = sessionReq.Uuid\n sessionParams.UserID = int32(userIDInt)\n sessionParams.Temperature = sessionReq.Temperature\n+ sessionParams.TopP = sessionReq.TopP\n sessionParams.MaxTokens = sessionReq.MaxTokens\n session, err := h.service.CreateOrUpdateChatSessionByUUID(r.Context(), sessionParams)\n if err != nil {\ndiff --git a/api/chat_session_service.go b/api/chat_session_service.go\nindex a292ba3..0ab70fe 100644\n--- a/api/chat_session_service.go\n+++ b/api/chat_session_service.go\n@@ -85,6 +85,7 @@ func (s *ChatSessionService) GetSimpleChatSessionsByUserID(ctx context.Context,\n Title: session.Topic,\n MaxLength: int(session.MaxLength),\n Temperature: float64(session.Temperature),\n+ TopP: float64(session.TopP),\n MaxTokens: session.MaxTokens,\n }\n })\"\"\"\n\n# Define the file paths to update\nfile_paths = {\n \"api/chat_main_handler.go\": [\n (\"// TopP: topP,\", \"TopP: float32(chatSession.TopP),\")\n ],\n \"api/chat_session_handler.go\": [\n (\"type UpdateChatSessionRequest struct {\", \"type UpdateChatSessionRequest struct {\\n Stream bool `json:\\\"stream\\\"`\"),\n (\"sessionParams.Temperature = sessionReq.Temperature\", \"sessionParams.Temperature = sessionReq.Temperature\\n sessionParams.TopP = sessionReq.TopP\\n sessionParams.Stream = sessionReq.Stream\")\n ],\n \"api/chat_session_service.go\": [\n (\"type SimpleChatSession struct {\", \"type SimpleChatSession struct {\\n Stream bool `json:\\\"stream\\\"`\"),\n (\"Temperature: float64(session.Temperature),\", \"Temperature: float64(session.Temperature),\\n TopP: float64(session.TopP),\\n MaxTokens: session.MaxTokens,\\n Stream: session.Stream,\")\n ]\n}\n\n# Apply the changeset to each file\nfor file_path, changes in file_paths.items():\n for line in fileinput.input(file_path, inplace=True):\n for old_value, new_value in changes:\n line = line.replace(old_value, new_value)\n sys.stdout.write(line)" }, { "path": "api/tools/apply_a_similar_change/apply_diff_uselib.py", "content": "import os\nimport difflib\n\n# specify the paths of the original file and the diff file\noriginal_file_path = 'path/to/original/file'\ndiff_file_path = 'path/to/diff/file'\n\n# read the contents of the original file\nwith open(original_file_path, 'r') as original_file:\n original_contents = original_file.readlines()\n\n# read the contents of the diff file\nwith open(diff_file_path, 'r') as diff_file:\n diff_contents = diff_file.readlines()\n\n# apply the diff to the original file\npatched_contents = difflib.unified_diff(original_contents, diff_contents)\n\n# write the patched contents to a new file\npatched_file_path = 'path/to/patched/file'\nwith open(patched_file_path, 'w') as patched_file:\n patched_file.writelines(patched_contents)\n\n# optionally, rename the original file and rename the patched file to the original file name\nos.rename(original_file_path, original_file_path + '.bak')\nos.rename(patched_file_path, original_file_path)\n\"\"\"\nNote that this program uses the `difflib` module to apply the diff. The `difflib.unified_diff()` function takes two lists of strings (the contents of the original file and the diff file) and returns a generator that yields the patched lines. The `writelines()` function is used to write the patched lines to a new file.\n\nAlso note that the program includes an optional step to rename the original file and rename the patched file to the original file name. This is done to preserve the original file in case the patching process goes wrong.\n\"\"\"" }, { "path": "api/tools/apply_a_similar_change/parse_diff.py", "content": "import os\nimport difflib\nimport shutil\nfrom pathlib import Path\n\n\n# Initialize variables\ndiff = []\nfile_path = \"\"\nnew_file_path = \"\"\npatched_files = []\n\n# Load the diff file into a list of lines\nwith open('/Users/hwu/dev/chat/api/tools/stream.diff', 'r') as f:\n diff_lines = f.readlines()\n\n# Loop through the diff lines\nfor line in diff_lines:\n print(line)\n if line.startswith('diff --git'): # Start of a new file\n if diff:\n # Apply the diff to the previous file\n diff_content = ''.join(diff)\n print(\"x\")\n print(file_path, new_file_path)\n print(\"x\")\n patch = difflib.unified_diff(Path(file_path).read_text(), diff_content.splitlines(), fromfile=file_path, tofile=new_file_path)\n patched_file_content = ''.join(patch)\n with open(file_path, 'w') as f:\n f.write(patched_file_content)\n patched_files.append(file_path)\n\n # Initialize variables for the new file\n diff = []\n _, _, file_path, new_file_path = line.split(' ')\n file_path = file_path[2:]\n new_file_path = new_file_path[2:]\n\n elif line.startswith('---') or line.startswith('+++'): # Ignore the old and new file paths\n continue\n\n elif line.startswith('new file'): # Handle new files\n if diff:\n # Apply the diff to the previous file\n diff_content = ''.join(diff)\n patch = difflib.unified_diff(shutil.readFile(file_path), diff_content.splitlines(), fromfile=file_path, tofile=new_file_path)\n patched_file_content = ''.join(patch)\n with open(file_path, 'w') as f:\n f.write(patched_file_content)\n patched_files.append(file_path)\n\n # Initialize variables for the new file\n diff = []\n _, _, new_file_path = line.split(' ')\n new_file_path = new_file_path[2:]\n file_path = new_file_path\n\n else:\n # Add the line to the diff content\n diff.append(line)\n\nif diff:\n # Apply the diff to the last file\n diff_content = ''.join(diff)\n patch = difflib.unified_diff(Path(file_path).read_text(), diff_content.splitlines(), fromfile=file_path, tofile=new_file_path)\n patched_file_content = ''.join(patch)\n with open(file_path, 'w') as f:\n f.write(patched_file_content)\n patched_files.append(file_path)\n\n# Print the list of patched files\nprint(\"Patched files:\")\nfor file in patched_files:\n print(file)\n" }, { "path": "api/tools/apply_a_similar_change/parse_diff2.py", "content": "import difflib\nimport os\nfrom pathlib import Path\n\ndef apply_diff_file(diff_file):\n with open(diff_file, 'r') as f:\n diff_text = f.read()\n\n # Split the diff into files, and parse the file headers\n file_diffs = diff_text.split('diff ')\n print(file_diffs)\n for file_diff in file_diffs[1:]:\n # Parse the file header to get the file names\n file_header, diff = file_diff.split('\\n', 1)\n old_file, new_file = file_header.split(' ')[-2:]\n\n # Apply the diff to the old file\n with open(old_file, 'r') as f:\n old_text = f.read()\n patched_lines = difflib.unified_diff(old_text.splitlines(), diff.splitlines(), lineterm='', fromfile=old_file, tofile=new_file)\n patched_text = os.linesep.join(list(patched_lines)[2:]) # Skip the first two lines of the unified diff\n\n with open(old_file, 'w') as f:\n f.write(patched_text)\n\ncurrent_dir = Path(__file__).parent\n\napply_diff_file(current_dir/ 'tools/stream.diff')" }, { "path": "api/tools/apply_a_similar_change/parse_diff3.py", "content": "from unidiff import PatchSet\nfrom pathlib import Path\ncurrent_dir = Path(__file__).parent\n\ndata = (current_dir/ 'tools/stream.diff').read_text()\npatch = PatchSet(data)\nprint(len(patch))\nfor i in patch:\n print(i)" }, { "path": "api/tools/apply_a_similar_change/stream.diff", "content": "commit 31c4f4b48ada4b3e8495abe7dcdc41ded550a598\nAuthor: Hao Wu \nDate: Wed Mar 22 00:11:08 2023 +0800\n\n add stream (#20)\n\ndiff --git a/api/chat_session_handler.go b/api/chat_session_handler.go\nindex 2c7a332..5bd0440 100644\n--- a/api/chat_session_handler.go\n+++ b/api/chat_session_handler.go\n@@ -225,6 +225,7 @@ type UpdateChatSessionRequest struct {\n Topic string `json:\"topic\"`\n MaxLength int32 `json:\"maxLength\"`\n Temperature float64 `json:\"temperature\"`\n+ Stream bool `json:\"stream\"`\n MaxTokens int32 `json:\"maxTokens\"`\n }\n \n@@ -254,6 +255,7 @@ func (h *ChatSessionHandler) CreateOrUpdateChatSessionByUUID(w http.ResponseWrit\n sessionParams.Uuid = sessionReq.Uuid\n sessionParams.UserID = int32(userIDInt)\n sessionParams.Temperature = sessionReq.Temperature\n+ sessionParams.Stream = sessionReq.Stream\n sessionParams.MaxTokens = sessionReq.MaxTokens\n session, err := h.service.CreateOrUpdateChatSessionByUUID(r.Context(), sessionParams)\n if err != nil {\ndiff --git a/api/chat_session_service.go b/api/chat_session_service.go\nindex a292ba3..0ab70fe 100644\n--- a/api/chat_session_service.go\n+++ b/api/chat_session_service.go\n@@ -85,6 +85,7 @@ func (s *ChatSessionService) GetSimpleChatSessionsByUserID(ctx context.Context,\n Title: session.Topic,\n MaxLength: int(session.MaxLength),\n Temperature: float64(session.Temperature),\n+ Stream: session.Stream,\n MaxTokens: session.MaxTokens,\n }\n })\n" }, { "path": "api/tools/fix_eris.py", "content": "import os\nimport re\nimport subprocess\n\ndef search_files(dir_name, pattern):\n \"\"\"\n Search for files in a directory (and sub-directories) based on a specific pattern.\n \"\"\"\n for root, dirs, files in os.walk(dir_name):\n for filename in files:\n if re.match(pattern, filename):\n yield os.path.join(root, filename)\n\ndef replace_error_handling(file_path):\n \"\"\"\n Replace error handling code in a file with eris error handling code.\n \"\"\"\n with open(file_path, 'r') as f:\n content = f.read()\n\n # Replace error handling code using regex\n old_pattern = r'fmt\\.Errorf\\(\\\"(.*)%w\\\",\\s+err\\)'\n new_pattern = r'eris.Wrap(err, \"\\1\")'\n new_content = re.sub(old_pattern, new_pattern, content, flags=re.MULTILINE)\n\n with open(file_path, 'w') as f:\n f.write(new_content)\n\n print(f\"Replaced error handling code in {file_path}\")\n\ndef main():\n \"\"\"\n Main function to search for files, replace error handling code, and commit changes to git.\n \"\"\"\n # Path to directory containing code files to refactor\n dir_name = \"./\"\n\n # Regex pattern to match specific file extensions\n pattern = r'^.*\\.(go)$'\n\n # Search for files based on pattern\n files = search_files(dir_name, pattern)\n\n # Refactor error handling code in each file\n for file_path in files:\n replace_error_handling(file_path)\n\n # Commit changes to git\n #subprocess.call([\"git\", \"add\", \".\"])\n #subprocess.call([\"git\", \"commit\", \"-m\", \"Refactor error handling using eris\"])\n #subprocess.call([\"git\", \"push\", \"origin\", \"master\"])\n\nif __name__ == '__main__':\n main()\n\n" }, { "path": "api/util.go", "content": "package main\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"log\"\n\t\"net/http\"\n\t\"os\"\n\t\"strconv\"\n\t\"strings\"\n\n\t\"github.com/google/uuid\"\n\t\"github.com/pkoukk/tiktoken-go\"\n\t\"github.com/rotisserie/eris\"\n)\n\nfunc NewUUID() string {\n\tuuidv7, err := uuid.NewV7()\n\tif err != nil {\n\t\treturn uuid.NewString()\n\t}\n\treturn uuidv7.String()\n}\nfunc getTokenCount(content string) (int, error) {\n\tencoding := \"cl100k_base\"\n\ttke, err := tiktoken.GetEncoding(encoding)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\ttoken := tke.Encode(content, nil, nil)\n\tnum_tokens := len(token)\n\treturn num_tokens, nil\n}\n\n// allocation free version\nfunc firstN(s string, n int) string {\n\ti := 0\n\tfor j := range s {\n\t\tif i == n {\n\t\t\treturn s[:j]\n\t\t}\n\t\ti++\n\t}\n\treturn s\n}\n\n// firstNWords extracts the first n words from a string\nfunc firstNWords(s string, n int) string {\n\tif s == \"\" {\n\t\treturn \"\"\n\t}\n\n\twords := strings.Fields(s)\n\tif len(words) <= n {\n\t\treturn s\n\t}\n\n\treturn strings.Join(words[:n], \" \")\n}\n\nfunc getUserID(ctx context.Context) (int32, error) {\n\tuserIdValue := ctx.Value(userContextKey)\n\tif userIdValue == nil {\n\t\treturn 0, eris.New(\"no user Id in context\")\n\t}\n\tuserIDStr := ctx.Value(userContextKey).(string)\n\tuserIDInt, err := strconv.ParseInt(userIDStr, 10, 32)\n\tif err != nil {\n\t\treturn 0, eris.Wrap(err, \"Error: '\"+userIDStr+\"' is not a valid user ID. should be a numeric value: \")\n\t}\n\tuserID := int32(userIDInt)\n\treturn userID, nil\n}\n\nfunc getContextWithUser(userID int) context.Context {\n\treturn context.WithValue(context.Background(), userContextKey, strconv.Itoa(userID))\n}\n\nfunc setSSEHeader(w http.ResponseWriter) {\n\tw.Header().Set(\"Content-Type\", \"text/event-stream\")\n\tw.Header().Set(\"Cache-Control\", \"no-cache, no-transform\")\n\tw.Header().Set(\"Connection\", \"keep-alive\")\n\tw.Header().Set(\"Access-Control-Allow-Origin\", \"*\")\n\tw.Header().Set(\"X-Accel-Buffering\", \"no\")\n\t// Remove any content-length to enable streaming\n\tw.Header().Del(\"Content-Length\")\n\t// Prevent compression\n\tw.Header().Del(\"Content-Encoding\")\n}\n\n// setupSSEStream configures the response writer for Server-Sent Events and returns the flusher\nfunc setupSSEStream(w http.ResponseWriter) (http.Flusher, error) {\n\tsetSSEHeader(w)\n\tflusher, ok := w.(http.Flusher)\n\tif !ok {\n\t\treturn nil, errors.New(\"streaming unsupported by client\")\n\t}\n\treturn flusher, nil\n}\n\nfunc getPerWordStreamLimit() int {\n\tperWordStreamLimitStr := os.Getenv(\"PER_WORD_STREAM_LIMIT\")\n\n\tif perWordStreamLimitStr == \"\" {\n\t\tperWordStreamLimitStr = \"200\"\n\t}\n\n\tperWordStreamLimit, err := strconv.Atoi(perWordStreamLimitStr)\n\tif err != nil {\n\t\tlog.Printf(\"get per word stream limit: %v\", eris.Wrap(err, \"get per word stream limit\").Error())\n\t\treturn 200\n\t}\n\n\treturn perWordStreamLimit\n}\n\nfunc RespondWithJSON(w http.ResponseWriter, status int, payload interface{}) {\n\tresponse, err := json.Marshal(payload)\n\tif err != nil {\n\t\tw.WriteHeader(http.StatusInternalServerError)\n\t\treturn\n\t}\n\tw.Header().Set(\"Content-Type\", \"application/json\")\n\tw.WriteHeader(status)\n\tw.Write(response)\n}\n\nfunc getPaginationParams(r *http.Request) (limit int32, offset int32) {\n\tlimitStr := r.URL.Query().Get(\"limit\")\n\toffsetStr := r.URL.Query().Get(\"offset\")\n\n\tlimit = 100 // default limit\n\tif limitStr != \"\" {\n\t\tif l, err := strconv.ParseInt(limitStr, 10, 32); err == nil {\n\t\t\tlimit = int32(l)\n\t\t}\n\t}\n\n\toffset = 0 // default offset\n\tif offsetStr != \"\" {\n\t\tif o, err := strconv.ParseInt(offsetStr, 10, 32); err == nil {\n\t\t\toffset = int32(o)\n\t\t}\n\t}\n\n\treturn limit, offset\n}\n\nfunc getLimitParam(r *http.Request, defaultLimit int32) int32 {\n\tlimitStr := r.URL.Query().Get(\"limit\")\n\tif limitStr == \"\" {\n\t\treturn defaultLimit\n\t}\n\tlimit, err := strconv.ParseInt(limitStr, 10, 32)\n\tif err != nil {\n\t\treturn defaultLimit\n\t}\n\treturn int32(limit)\n}\n\n// DecodeJSON decodes JSON from the request body into the target.\n// Target must be a pointer.\nfunc DecodeJSON(r *http.Request, target interface{}) error {\n\treturn json.NewDecoder(r.Body).Decode(target)\n}\n" }, { "path": "api/util_test.go", "content": "package main\n\nimport \"testing\"\n\nfunc Test_firstN(t *testing.T) {\n\ttype args struct {\n\t\ts string\n\t\tn int\n\t}\n\ttests := []struct {\n\t\tname string\n\t\targs args\n\t\twant string\n\t}{\n\t\t{\"01\", args{\"hello\", 2}, \"he\"},\n\t\t{\"02\", args{\"hello\", 5}, \"hello\"},\n\t\t{\"03\", args{\"hello\", 50}, \"hello\"},\n\t\t{\"04\", args{\"你好世界\", 2}, \"你好\"},\n\t\t{\"05\", args{\"\", 3}, \"\"},\n\t\t{\"06\", args{\"hello\", 0}, \"\"},\n\t\t{\"07\", args{\"🙂🙃\", 1}, \"🙂\"},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tif got := firstN(tt.args.s, tt.args.n); got != tt.want {\n\t\t\t\tt.Errorf(\"firstN() = %v, want %v\", got, tt.want)\n\t\t\t}\n\t\t})\n\t}\n}\n" }, { "path": "api/util_words_test.go", "content": "package main\n\nimport (\n\t\"testing\"\n)\n\nfunc Test_firstNWords(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tinput string\n\t\tn int\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tname: \"normal case with exactly 10 words\",\n\t\t\tinput: \"how do I write a function that processes data efficiently in Go\",\n\t\t\tn: 10,\n\t\t\texpected: \"how do I write a function that processes data efficiently\",\n\t\t},\n\t\t{\n\t\t\tname: \"less than 10 words\",\n\t\t\tinput: \"hello world how are you\",\n\t\t\tn: 10,\n\t\t\texpected: \"hello world how are you\",\n\t\t},\n\t\t{\n\t\t\tname: \"more than 10 words\",\n\t\t\tinput: \"this is a very long prompt that contains much more than ten words so it should be truncated\",\n\t\t\tn: 10,\n\t\t\texpected: \"this is a very long prompt that contains much more\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty string\",\n\t\t\tinput: \"\",\n\t\t\tn: 10,\n\t\t\texpected: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"single word\",\n\t\t\tinput: \"hello\",\n\t\t\tn: 10,\n\t\t\texpected: \"hello\",\n\t\t},\n\t\t{\n\t\t\tname: \"exactly n words\",\n\t\t\tinput: \"one two three four five\",\n\t\t\tn: 5,\n\t\t\texpected: \"one two three four five\",\n\t\t},\n\t\t{\n\t\t\tname: \"with extra whitespace\",\n\t\t\tinput: \" hello world how are you \",\n\t\t\tn: 3,\n\t\t\texpected: \"hello world how\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := firstNWords(tt.input, tt.n)\n\t\t\tif result != tt.expected {\n\t\t\t\tt.Errorf(\"firstNWords(%q, %d) = %q, want %q\", tt.input, tt.n, result, tt.expected)\n\t\t\t}\n\t\t})\n\t}\n}\n" }, { "path": "artifacts.md", "content": "# Artifact Feature Implementation Plan\n\n## Overview\n\nThis document outlines the implementation plan for adding artifact support to the chat application. Artifacts are interactive code blocks, HTML previews, diagrams, and other rich content that can be generated by LLMs and displayed interactively in the chat interface.\n\n## Current Architecture Analysis\n\n### Database Schema\n\nThe `chat_message` table already has:\n\n- `content` field for main message content\n- `raw` JSONB field for raw LLM responses\n- `reasoning_content` field for thinking/reasoning content\n\n### Frontend Message Structure\n\n```typescript\ninterface Message {\n uuid: string,\n dateTime: string\n text: string\n model?: string\n inversion?: boolean // true for user messages, false for assistant\n error?: boolean\n loading?: boolean\n isPrompt?: boolean\n isPin?: boolean\n}\n```\n\n### Current Message Flow\n\n1. User Input → Frontend\n2. API Call → `/chat_stream` endpoint\n3. LLM Processing → Various providers\n4. Response Streaming → Server-Sent Events\n5. Database Storage → Full metadata\n6. Frontend Update → UI refresh\n\n## Implementation Plan\n\n### 1. Database Schema Extension\n\n#### Option A: Extend existing table\n\n```sql\n-- Add artifacts column to chat_message table\nALTER TABLE chat_message ADD COLUMN artifacts JSONB;\n```\n\n### 2. Backend Implementation\n\n#### Data Structures\n\n```go\n// Add to message struct\ntype ChatMessage struct {\n // ... existing fields\n Artifacts []Artifact `json:\"artifacts,omitempty\"`\n}\n\ntype Artifact struct {\n UUID string `json:\"uuid\"`\n Type string `json:\"type\"`\n Title string `json:\"title\"`\n Content string `json:\"content\"`\n Language string `json:\"language,omitempty\"`\n}\n```\n\n#### Artifact Detection Logic\n\n```go\n// In message processing pipeline\nfunc extractArtifacts(content string) []Artifact {\n // Detect code blocks with artifact markers\n // Pattern: ```language \n // Extract HTML blocks with artifact markers\n // Detect SVG content\n // Find Mermaid diagrams\n}\n```\n\n#### API Endpoints\n\n- Extend existing message endpoints to include artifact data\n- Add artifact-specific endpoints if needed:\n - `GET /api/artifacts/:uuid` - Get artifact content\n - `PUT /api/artifacts/:uuid` - Update artifact (if editing supported)\n - `DELETE /api/artifacts/:uuid` - Delete artifact\n - `GET /api/artifacts` - List all artifacts for a user\n\n### 3. Frontend Components\n\n#### Core Components\n\n```vue\n\n\n```\n\n#### Artifact Type Components\n\n- `HtmlArtifact.vue` - Sandboxed HTML preview\n- `SvgArtifact.vue` - SVG viewer with zoom/pan\n- `MermaidArtifact.vue` - Interactive diagrams\n- `JsonArtifact.vue` - Formatted JSON viewer\n\n#### Integration with Message Component\n\n```vue\n\n\n```\n\n### 4. Artifact Detection Patterns\n\n\n\n#### HTML Artifacts\n\n```html \n
\n \n
\n```\n\n#### SVG Artifacts\n\n```svg \n\n \n\n```\n\n### 5. Artifact Types to Support\n\ntection\n\n2. **HTML Artifacts**\n - Live HTML preview\n - Sandboxed iframe execution\n - CSS/JS support\n\n3. **SVG Artifacts**\n - Vector graphics display\n - Zoom/pan functionality\n - Export options\n\n4. **Mermaid Diagrams**\n - Flowcharts, sequence diagrams\n - Interactive navigation\n - Export to image\n\n5. **JSON/Data Artifacts**\n - Formatted JSON viewer\n - Collapsible tree structure\n - Search functionality\n\n\n### Additional Suggestions for Further Improvements:\n\n 4. Advanced Artifact Features\n\n // Interactive code execution\n - **Code Runners**: Execute JavaScript, Python snippets safely\n - **Live Editing**: Edit artifacts inline with syntax highlighting\n\n 5. Enhanced Visualization\n\n 6. Workflow Integration\n\n // Developer tools integration\n\n - **CodePen Integration**: Open HTML artifacts in CodePen\n - **Download Options**: Save artifacts as files\n - **Template Library**: Reusable artifact templates\n\n 7. Advanced UI Features\n\n // Enhanced user experience\n - **Artifact Gallery**: Browse all artifacts in a session\n\n\n 8. Security & Performance\n\n // Better safety and speed\n - **Content Sanitization**: Advanced XSS protection\n - **Lazy Loading**: Load artifacts only when needed\n - **Caching**: Cache rendered artifacts for better performance\n - **Resource Limits**: Prevent memory leaks from large artifacts\n\n### 6. Implementation Phases\n\n#### Phase 1: Core Infrastructure\n\n- [x] Database schema updates\n- [x] Backend artifact detection logic\n- [x] Basic frontend artifact viewer component\n\n#### Phase 2: Code Artifacts\n\n- [x] Code artifact component with syntax highlighting\n- [x] Copy/download functionality\n\n#### Phase 3: Web Artifacts\n\n- [x] HTML artifact component with sandboxing\n- [x] SVG artifact viewer\n- [] CSS/JS execution in sandbox\n\n#### Phase 4: Advanced Features\n\n- [x] Mermaid diagram support\n- [x] JSON/data viewer\n\n#### Phase 5: Polish & Features\n\n- [] Artifact editing capabilities\n- [] Sharing and export options\n\n### 7. Technical Considerations\n\n#### Security\n\n- Sandbox HTML/JS execution to prevent XSS\n- Validate and sanitize artifact content\n- Limit artifact size and complexity\n\n#### Performance\n\n- Lazy loading of artifact components\n- Virtualization for large artifacts\n- Caching of rendered content\n\n#### User Experience\n\n- Collapsible artifact containers\n- Responsive design for mobile\n- Keyboard shortcuts for artifact actions\n- Loading states for complex artifacts\n\n\n## Resources\n\n- Current codebase structure in `/web/src/views/chat/components/Message/`\n- Database schema in `/api/sqlc/schema.sql`\n- Message handling in `/api/chat_message_handler.go`\n- Frontend message types in `/web/src/types/chat.d.ts`\n" }, { "path": "chat.code-workspace", "content": "{\n\t\"folders\": [\n\t\t{\n\t\t\t\"path\": \"web\"\n\t\t},\n\t\t{\n\t\t\t\"path\": \"api\"\n\t\t},\n\t\t{\n\t\t\t\"path\": \"e2e\"\n\t\t},\n\t\t{\n\t\t\t\"path\": \"docs\"\n\t\t}\n\t],\n\t\"settings\": {}\n}\n" }, { "path": "docker-compose.yaml", "content": "version: '3'\n\nservices:\n chat:\n container_name: chat\n image: ghcr.io/swuecho/chat:latest # or echowuhao/chat:latest # or use tag for better stability, e.g.v0.0.3\n expose:\n - 8080\n ports:\n # vist at http://your_host:8080 \n - 8080:8080\n environment:\n # at least one key is required.\n # !!! no quote aroud key !!!\n - OPENAI_API_KEY=thisisopenaikey # do not change if you do not have openai api key\n - CLAUDE_API_KEY=thisisclaudekey # do not change if you do not have claude api key\n # api call in 10min\n # set this to zero if your server is in public network. only increase ratelimit in admin panel for trusted users.\n - OPENAI_RATELIMIT=100\n # DB config, set based on your db config if you don't use the db in docker-compose\n - PG_HOST=db\n - PG_DB=postgres\n - PG_USER=postgres\n - PG_PASS=thisisapassword\n - PG_PORT=5432\n # - PER_WORD_STREAM_LIMIT=200 # first 200 words are streamed per word, then by line.\n # or DATABASE_URL, with the 5 var above\n # you might need set proxy\n # - OPENAI_PROXY_URL=hopethepeoplemakegreatfirewilldiesoon\n depends_on:\n db:\n condition: service_healthy\n db:\n image: postgres:14\n restart: always\n user: postgres\n environment:\n TZ: \"Asia/Shanghai\"\n PGTZ: \"Asia/Shanghai\"\n POSTGRES_USER: \"postgres\"\n POSTGRES_PASSWORD: \"thisisapassword\"\n expose:\n - 5432\n ports:\n - \"5432:5432\"\n healthcheck:\n test:\n [\n \"CMD-SHELL\",\n \"pg_isready\",\n \"-q\",\n \"-d\",\n \"postgres\",\n \"-U\",\n \"postgres\"\n ]\n interval: 5s\n timeout: 5s\n retries: 5\n" }, { "path": "docs/add_model_en.md", "content": "# Adding a New Chat Model\n\nThis guide explains how to add a new chat model to the system.\n\n## Prerequisites\n- Admin access to the system\n- API credentials for the model you want to add\n- Model's API endpoint URL\n\n## Steps to Add a Model\n\n### 1. Access the Admin Interface\n1. Log in as an admin user\n2. Navigate to the Admin section\n3. Go to the \"Models\" tab\n\n\"image\"\n\n### 2. Fill in the Model Details\nFill in the following fields in the Add Model form:\n\n- **Name**: Internal name for the model (e.g. \"gpt-3.5-turbo\")\n- **Label**: Display name for the model (e.g. \"GPT-3.5 Turbo\")\n- **URL**: API endpoint URL for the model\n- **API Auth Header**: Header name for authentication (e.g. \"Authorization\", \"x-api-key\")\n- **API Auth Key**: Environment variable containing the API key\n- **Is Default**: Whether this should be the default model\n- **Enable Per-Mode Rate Limit**: Enable rate limiting for this specific model\n- **Order Number**: Position in the model list (lower numbers appear first)\n- **Default Tokens**: Default token limit for requests\n- **Max Tokens**: Maximum token limit for requests\n\n\"image\"\n\n### 3. Add the Model\nClick \"Confirm\" to add the model. The system will:\n1. Validate the input\n2. Create the model record in the database\n3. Make the model available for use\n\n### 4. (Optional) Set Rate Limits\nIf you enabled per-mode rate limiting:\n1. Go to the \"Rate Limits\" tab\n2. Set rate limits for specific users\n\n## Example Configurations\n\nHere are example JSON configurations you can paste into the form:\n\n```json\n# openai\n{\n \"name\": \"gpt-4\",\n \"label\": \"GPT-4\",\n \"url\": \"https://api.openai.com/v1/chat/completions\",\n \"apiAuthHeader\": \"Authorization\",\n \"apiAuthKey\": \"OPENAI_API_KEY\",\n \"isDefault\": false,\n \"enablePerModeRatelimit\": true,\n \"orderNumber\": 5,\n \"defaultToken\": 4096,\n \"maxToken\": 8192\n}\n\n# claude\n{\n \"name\": \"claude-3-7-sonnet-20250219\",\n \"label\": \"claude-3-7-sonnet-20250219\",\n \"url\": \"https://api.anthropic.com/v1/messages\",\n \"apiAuthHeader\": \"x-api-key\",\n \"apiAuthKey\": \"CLAUDE_API_KEY\",\n \"isDefault\": false,\n \"enablePerModeRatelimit\": false,\n \"isEnable\": true,\n \"orderNumber\": 0,\n \"defaultToken\": 4096,\n \"maxToken\": 4096\n}\n\n# gemini\n{\n \"name\": \"gemini-2.0-flash\",\n \"label\": \"gemini-2.0-flash\",\n \"url\": \"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash\",\n \"apiAuthHeader\": \"GEMINI_API_KEY\",\n \"apiAuthKey\": \"GEMINI_API_KEY\",\n \"isDefault\": true,\n \"enablePerModeRatelimit\": false,\n \"isEnable\": true,\n \"orderNumber\": 0,\n \"defaultToken\": 4096,\n \"maxToken\": 4096\n}\n\n# deepseek\n{\n \"name\": \"deepseek-chat\",\n \"label\": \"deepseek-chat\",\n \"url\": \"https://api.deepseek.com/v1/chat/completions\",\n \"apiAuthHeader\": \"Authorization\",\n \"apiAuthKey\": \"DEEPSEEK_API_KEY\",\n \"isDefault\": false,\n \"enablePerModeRatelimit\": false,\n \"isEnable\": true,\n \"orderNumber\": 0,\n \"defaultToken\": 8192,\n \"maxToken\": 8192\n}\n\n# open router\n{\n \"name\": \"deepseek/deepseek-r1:free\",\n \"label\": \"deepseek/deepseek-r1(OR)\",\n \"url\": \"https://openrouter.ai/api/v1/chat/completions\",\n \"apiAuthHeader\": \"Authorization\",\n \"apiAuthKey\": \"OPENROUTER_API_KEY\",\n \"isDefault\": false,\n \"enablePerModeRatelimit\": false,\n \"isEnable\": true,\n \"orderNumber\": 1,\n \"defaultToken\": 8192,\n \"maxToken\": 8192\n}\n```\n\n## Troubleshooting\n\n**Model not appearing?**\n- Check if the model was added successfully in the database\n- Verify the API credentials are correct\n- Ensure the API endpoint is accessible\n\n**Rate limiting issues?**\n- Verify rate limits are properly configured\n- Check user permissions\n- Review system logs for errors\n" }, { "path": "docs/add_model_zh.md", "content": "# 添加新聊天模型\n\n本指南介绍如何向系统添加新的聊天模型。\n\n## 先决条件\n- 系统管理员权限\n- 要添加模型的API凭证\n- 模型的API端点URL\n\n## 添加模型的步骤\n\n### 1. 访问管理员界面\n1. 以管理员用户登录\n2. 导航到管理员部分\n3. 进入\"模型\"标签页\n\n\"image\"\n\n### 2. 填写模型详情\n在添加模型表单中填写以下字段：\n\n- **名称**: 模型的内部名称 (如 \"gpt-3.5-turbo\")\n- **标签**: 模型的显示名称 (如 \"GPT-3.5 Turbo\")\n- **URL**: 模型的API端点URL\n- **API认证头**: 认证头名称 (如 \"Authorization\", \"x-api-key\")\n- **API认证密钥**: 包含API密钥的环境变量\n- **是否默认**: 是否设为默认模型\n- **启用模式限速**: 为此特定模型启用速率限制\n- **排序号**: 在模型列表中的位置（数字越小越靠前）\n- **默认token数**: 请求的默认token限制\n- **最大token数**: 请求的最大token限制\n\n\"image\"\n\n### 3. 添加模型\n点击\"确认\"添加模型。系统将：\n1. 验证输入\n2. 在数据库中创建模型记录\n3. 使模型可供使用\n\n### 4. （可选）设置速率限制\n如果启用了模式限速：\n1. 进入\"速率限制\"标签页\n2. 为特定用户设置速率限制\n\n## 示例配置\n\n以下是可粘贴到表单中的示例JSON配置：\n\n```json\n# openai\n{\n \"name\": \"gpt-4\",\n \"label\": \"GPT-4\",\n \"url\": \"https://api.openai.com/v1/chat/completions\",\n \"apiAuthHeader\": \"Authorization\",\n \"apiAuthKey\": \"OPENAI_API_KEY\",\n \"isDefault\": false,\n \"enablePerModeRatelimit\": true,\n \"orderNumber\": 5,\n \"defaultToken\": 4096,\n \"maxToken\": 8192\n}\n\n# claude\n{\n \"name\": \"claude-3-7-sonnet-20250219\",\n \"label\": \"claude-3-7-sonnet-20250219\",\n \"url\": \"https://api.anthropic.com/v1/messages\",\n \"apiAuthHeader\": \"x-api-key\",\n \"apiAuthKey\": \"CLAUDE_API_KEY\",\n \"isDefault\": false,\n \"enablePerModeRatelimit\": false,\n \"isEnable\": true,\n \"orderNumber\": 0,\n \"defaultToken\": 4096,\n \"maxToken\": 4096\n}\n\n# gemini\n{\n \"name\": \"gemini-2.0-flash\",\n \"label\": \"gemini-2.0-flash\",\n \"url\": \"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash\",\n \"apiAuthHeader\": \"GEMINI_API_KEY\",\n \"apiAuthKey\": \"GEMINI_API_KEY\",\n \"isDefault\": true,\n \"enablePerModeRatelimit\": false,\n \"isEnable\": true,\n \"orderNumber\": 0,\n \"defaultToken\": 4096,\n \"maxToken\": 4096\n}\n\n# deepseek\n{\n \"name\": \"deepseek-chat\",\n \"label\": \"deepseek-chat\",\n \"url\": \"https://api.deepseek.com/v1/chat/completions\",\n \"apiAuthHeader\": \"Authorization\",\n \"apiAuthKey\": \"DEEPSEEK_API_KEY\",\n \"isDefault\": false,\n \"enablePerModeRatelimit\": false,\n \"isEnable\": true,\n \"orderNumber\": 0,\n \"defaultToken\": 8192,\n \"maxToken\": 8192\n}\n\n# open router\n{\n \"name\": \"deepseek/deepseek-r1:free\",\n \"label\": \"deepseek/deepseek-r1(OR)\",\n \"url\": \"https://openrouter.ai/api/v1/chat/completions\",\n \"apiAuthHeader\": \"Authorization\",\n \"apiAuthKey\": \"OPENROUTER_API_KEY\",\n \"isDefault\": false,\n \"enablePerModeRatelimit\": false,\n \"isEnable\": true,\n \"orderNumber\": 1,\n \"defaultToken\": 8192,\n \"maxToken\": 8192\n}\n\n# silicon flow\n\n{\n \"name\": \"Qwen/Qwen3-235B-A22B\",\n \"label\": \"Qwen/Qwen3-235B-A22B\",\n \"url\": \"https://api.siliconflow.cn/v1/chat/completions\",\n \"apiAuthHeader\": \"Authorization\",\n \"apiAuthKey\": \"SILICONFLOW_API_KEY\",\n \"isDefault\": false,\n \"enablePerModeRatelimit\": false,\n \"isEnable\": true,\n \"orderNumber\": 0,\n \"defaultToken\": 0,\n \"maxToken\": 0\n}\n```\n\n\n## 故障排除\n\n**模型未出现？**\n- 检查模型是否成功添加到数据库\n- 验证API凭证是否正确\n- 确保API端点可访问\n\n**速率限制问题？**\n- 验证速率限制是否正确配置\n- 检查用户权限\n- 查看系统日志中的错误\n" }, { "path": "docs/artifact_gallery_en.md", "content": "# Artifact Gallery\n\n## Overview\n\nThe Artifact Gallery is a comprehensive management interface for viewing, organizing, and interacting with code artifacts generated during chat conversations. It provides a centralized location to browse, execute, and manage all artifacts created across different chat sessions.\n\n## Features\n\n### 1. **Artifact Management**\n- **Browse artifacts** from all chat sessions in one place\n- **Filter and search** by type, language, session, or content\n- **Sort artifacts** by creation date, type, execution count, or rating\n- **View statistics** about artifact usage and performance\n- **Export artifacts** to JSON format for backup or sharing\n\n### 2. **Code Execution**\n- **Run JavaScript/Python code** directly in the gallery\n- **Real-time output** with syntax highlighting\n- **Error handling** with detailed error messages\n- **Performance metrics** including execution time\n- **Library support** for popular packages (lodash, d3, numpy, pandas, etc.)\n\n### 3. **Artifact Viewing**\n- **HTML rendering** in secure iframe sandbox\n- **SVG visualization** with proper scaling\n- **JSON formatting** with validation\n- **Mermaid diagram** rendering\n- **Syntax highlighting** for code artifacts\n\n### 4. **Organization Tools**\n- **Grid and list views** for different browsing preferences\n- **Pagination** for large artifact collections\n- **Tag-based categorization** with automatic tag generation\n- **Session context** showing which chat created each artifact\n- **Duplicate detection** and management\n\n## Accessing the Gallery\n\n### From Chat Interface\n1. Click the **Gallery** button (gallery icon) in the chat footer\n2. The gallery will open in place of the chat interface\n3. Use the same gallery button to return to chat view\n\n### Navigation\n- **Toggle between chat and gallery** using the gallery button\n- **Switch between grid and list views** using the view toggle\n- **Use filters** to narrow down artifacts by specific criteria\n\n## Artifact Types\n\n### Executable Artifacts\nThese artifacts can be run directly in the gallery:\n\n#### JavaScript/TypeScript\n- **Supported features**: ES6+, async/await, classes, modules\n- **Available libraries**: lodash, d3, chart.js, moment, axios, rxjs, p5, three, fabric\n- **Canvas support**: Create interactive graphics and visualizations\n- **Execution environment**: Secure Web Worker sandbox\n\n#### Python\n- **Supported features**: Python 3.x syntax, scientific computing\n- **Available packages**: numpy, pandas, matplotlib, scipy, scikit-learn, requests, beautifulsoup4, pillow, sympy, networkx, seaborn, plotly, bokeh, altair\n- **Plot support**: Matplotlib plots rendered as images\n- **Execution environment**: Pyodide-based Python interpreter\n\n### Viewable Artifacts\nThese artifacts are rendered for visual inspection:\n\n#### HTML\n- **Secure rendering**: Iframe sandbox with restricted permissions\n- **Full HTML support**: CSS, basic JavaScript, forms, modals\n- **Responsive design**: Adapts to different screen sizes\n- **External links**: Open in new window capability\n\n#### SVG\n- **Vector graphics**: Scalable and crisp rendering\n- **Interactive elements**: Hover effects and basic interactions\n- **Theme adaptation**: Adjusts colors for dark/light themes\n- **Export capability**: Download as SVG files\n\n#### JSON\n- **Formatted display**: Pretty-printed with syntax highlighting\n- **Validation**: Automatic JSON syntax validation\n- **Copy functionality**: Easy copying of formatted JSON\n- **Large file support**: Handles large JSON structures efficiently\n\n#### Mermaid\n- **Diagram types**: Flowcharts, sequence diagrams, class diagrams, etc.\n- **Auto-rendering**: Automatic conversion to visual diagrams\n- **Responsive**: Adapts to container size\n- **Theme support**: Follows application theme\n\n## Interface Elements\n\n### Gallery Header\n- **Title and count**: Shows total number of artifacts\n- **Action buttons**: Access to filters, statistics, and export\n- **Search bar**: Quick text search across all artifacts\n\n### Filter Panel\n- **Search**: Text search across titles, content, and tags\n- **Type filter**: Filter by artifact type (code, HTML, SVG, etc.)\n- **Language filter**: Filter by programming language\n- **Session filter**: Filter by chat session\n- **Date range**: Filter by creation date\n- **Sort options**: Multiple sorting criteria\n\n### Statistics Panel\n- **Total artifacts**: Overall count of artifacts\n- **Execution stats**: Total runs, average execution time, success rate\n- **Type breakdown**: Distribution of artifact types\n- **Language distribution**: Popular programming languages\n- **Performance charts**: Visual representation of usage patterns\n\n### Artifact Cards (Grid View)\n- **Artifact preview**: Truncated code or content preview\n- **Metadata**: Creation date, language, session info\n- **Action buttons**: Preview, Run/View, Edit, Duplicate, Delete\n- **Tags**: Automatically generated and custom tags\n- **Execution count**: Number of times the artifact has been run\n\n### Artifact Rows (List View)\n- **Compact layout**: More artifacts visible at once\n- **Essential info**: Title, type, language, creation date\n- **Quick actions**: All management functions accessible\n- **Session context**: Clear indication of source chat session\n\n## Actions and Operations\n\n### Running Code Artifacts\n1. **Click the Run button** (play icon) on JavaScript/Python artifacts\n2. **Run modal opens** with code preview and execution controls\n3. **Execute code** using the \"Run Code\" button\n4. **View output** in real-time with syntax highlighting\n5. **Clear results** to run again with fresh environment\n\n### Viewing Visual Artifacts\n1. **Click the View button** (external link icon) on HTML/SVG/JSON artifacts\n2. **View modal opens** with proper rendering\n3. **Interact with content** (for HTML artifacts)\n4. **Copy content** using the copy button\n5. **Close modal** to return to gallery\n\n### Managing Artifacts\n- **Preview**: Quick view of artifact content\n- **Edit**: Modify artifact content and save changes\n- **Duplicate**: Create a copy for experimentation\n- **Delete**: Remove artifact permanently (with confirmation)\n- **Copy**: Copy artifact content to clipboard\n- **Download**: Save artifact as file with proper extension\n\n## Advanced Features\n\n### Automatic Tagging\nThe gallery automatically generates tags based on:\n- **Programming language** (javascript, python, html, etc.)\n- **Framework usage** (react, vue, pandas, matplotlib, etc.)\n- **Code patterns** (async, functions, classes, loops, etc.)\n- **Libraries** (lodash, d3, numpy, etc.)\n- **Execution results** (error, success, slow, fast, etc.)\n\n### Session Context\nEach artifact maintains connection to its source:\n- **Session title**: Name of the chat session\n- **Message context**: Which message created the artifact\n- **Timestamp**: When the artifact was created\n- **Edit tracking**: Whether the artifact has been modified\n\n### Performance Tracking\n- **Execution time**: How long code takes to run\n- **Success rate**: Percentage of successful executions\n- **Error patterns**: Common error types and frequencies\n- **Usage statistics**: Most frequently run artifacts\n\n### Export and Backup\n- **JSON export**: Complete artifact data with metadata\n- **Filtered export**: Export only selected or filtered artifacts\n- **Backup format**: Structured for easy import/restore\n- **Sharing**: Share artifact collections with others\n\n## Best Practices\n\n### Organization\n- **Use descriptive titles** for artifacts when creating them\n- **Add custom tags** for better organization\n- **Regular cleanup** of outdated or experimental artifacts\n- **Session naming** to provide better context\n\n### Code Execution\n- **Test incrementally** when developing complex code\n- **Use console.log** for debugging JavaScript\n- **Handle errors gracefully** in your code\n- **Be mindful of execution time** for complex operations\n\n### Performance\n- **Use pagination** for large artifact collections\n- **Clear execution results** when not needed\n- **Filter artifacts** to focus on relevant items\n- **Regular exports** for backup and archival\n\n## Security Considerations\n\n### Code Execution\n- **Sandboxed environment**: All code runs in isolated Web Workers\n- **No network access**: Direct network requests are blocked\n- **No file system access**: Cannot read/write local files\n- **Limited DOM access**: Cannot modify the parent application\n- **Timeout protection**: Long-running code is automatically terminated\n\n### Content Rendering\n- **HTML sandboxing**: Iframe sandbox prevents malicious scripts\n- **SVG sanitization**: Removes potentially dangerous elements\n- **Content validation**: JSON and other formats are validated\n- **Safe origins**: External resources are restricted\n\n## Troubleshooting\n\n### Common Issues\n- **Code not running**: Check language support and syntax\n- **Artifacts not loading**: Refresh the page or check browser console\n- **Performance issues**: Reduce artifact count or use filters\n- **Export failures**: Check browser download permissions\n\n### Browser Compatibility\n- **Modern browsers**: Chrome, Firefox, Safari, Edge (latest versions)\n- **JavaScript required**: Gallery requires JavaScript to function\n- **LocalStorage**: Used for preferences and temporary data\n- **WebWorkers**: Required for code execution features\n\n## API Integration\n\nThe gallery integrates with the chat application's API:\n\n```javascript\n// Example: Getting artifacts from a specific session\nconst messages = await fetch(`/api/uuid/chat_messages/chat_sessions/${sessionId}`);\n// Artifacts are included in each message payload as the `artifacts` field\n\n// Example: Executing code artifact\n// Code execution is handled client-side in Artifact Editor/Sandbox components,\n// not by a dedicated backend `/api/artifacts/:id/execute` endpoint.\n```\n\n## Future Enhancements\n\n### Planned Features\n- **Artifact versioning**: Track changes and maintain history\n- **Collaborative editing**: Multiple users editing artifacts\n- **Advanced analytics**: Detailed usage and performance metrics\n- **Template system**: Create reusable artifact templates\n- **AI suggestions**: Automatic code improvements and suggestions\n\n### Community Features\n- **Artifact sharing**: Public gallery for sharing useful artifacts\n- **Rating system**: Community ratings for popular artifacts\n- **Comments**: Collaborative discussion on artifacts\n- **Collections**: Curated sets of related artifacts\n\nThis documentation provides a comprehensive guide to using the Artifact Gallery effectively. For technical implementation details, see the source code in `/src/views/chat/components/ArtifactGallery.vue`." }, { "path": "docs/artifact_gallery_zh.md", "content": "# 制品画廊\n\n## 概述\n\n制品画廊是一个综合管理界面，用于查看、组织和与聊天对话中生成的代码制品进行交互。它提供了一个集中的位置来浏览、执行和管理不同聊天会话中创建的所有制品。\n\n## 功能特性\n\n### 1. **制品管理**\n- **浏览制品** 在一个地方查看所有聊天会话的制品\n- **过滤和搜索** 按类型、语言、会话或内容进行筛选\n- **排序制品** 按创建日期、类型、执行次数或评分排序\n- **查看统计** 关于制品使用和性能的统计信息\n- **导出制品** 以JSON格式导出以供备份或分享\n\n### 2. **代码执行**\n- **运行JavaScript/Python代码** 直接在画廊中执行\n- **实时输出** 带语法高亮的输出\n- **错误处理** 详细的错误消息\n- **性能指标** 包括执行时间\n- **库支持** 支持流行的包（lodash, d3, numpy, pandas等）\n\n### 3. **制品查看**\n- **HTML渲染** 在安全的iframe沙盒中\n- **SVG可视化** 带正确缩放\n- **JSON格式化** 带验证功能\n- **Mermaid图表** 渲染\n- **语法高亮** 用于代码制品\n\n### 4. **组织工具**\n- **网格和列表视图** 不同的浏览偏好\n- **分页** 用于大型制品集合\n- **基于标签的分类** 自动标签生成\n- **会话上下文** 显示哪个聊天创建了每个制品\n- **重复检测** 和管理\n\n## 访问画廊\n\n### 从聊天界面\n1. 点击聊天页脚的**画廊**按钮（画廊图标）\n2. 画廊将在聊天界面的位置打开\n3. 使用相同的画廊按钮返回聊天视图\n\n### 导航\n- **在聊天和画廊之间切换** 使用画廊按钮\n- **在网格和列表视图之间切换** 使用视图切换\n- **使用过滤器** 按特定条件缩小制品范围\n\n## 制品类型\n\n### 可执行制品\n这些制品可以直接在画廊中运行：\n\n#### JavaScript/TypeScript\n- **支持的功能**：ES6+、async/await、类、模块\n- **可用库**：lodash、d3、chart.js、moment、axios、rxjs、p5、three、fabric\n- **Canvas支持**：创建交互式图形和可视化\n- **执行环境**：安全的Web Worker沙盒\n\n#### Python\n- **支持的功能**：Python 3.x语法、科学计算\n- **可用包**：numpy、pandas、matplotlib、scipy、scikit-learn、requests、beautifulsoup4、pillow、sympy、networkx、seaborn、plotly、bokeh、altair\n- **绘图支持**：Matplotlib图表渲染为图像\n- **执行环境**：基于Pyodide的Python解释器\n\n### 可查看制品\n这些制品用于视觉检查：\n\n#### HTML\n- **安全渲染**：带限制权限的iframe沙盒\n- **完整HTML支持**：CSS、基本JavaScript、表单、模态框\n- **响应式设计**：适应不同屏幕尺寸\n- **外部链接**：在新窗口中打开的能力\n\n#### SVG\n- **矢量图形**：可缩放和清晰的渲染\n- **交互元素**：悬停效果和基本交互\n- **主题适应**：为深色/浅色主题调整颜色\n- **导出能力**：下载为SVG文件\n\n#### JSON\n- **格式化显示**：带语法高亮的美化打印\n- **验证**：自动JSON语法验证\n- **复制功能**：轻松复制格式化的JSON\n- **大文件支持**：高效处理大型JSON结构\n\n#### Mermaid\n- **图表类型**：流程图、序列图、类图等\n- **自动渲染**：自动转换为视觉图表\n- **响应式**：适应容器大小\n- **主题支持**：遵循应用主题\n\n## 界面元素\n\n### 画廊标题\n- **标题和计数**：显示制品总数\n- **操作按钮**：访问过滤器、统计和导出\n- **搜索栏**：跨所有制品的快速文本搜索\n\n### 过滤器面板\n- **搜索**：跨标题、内容和标签的文本搜索\n- **类型过滤**：按制品类型过滤（代码、HTML、SVG等）\n- **语言过滤**：按编程语言过滤\n- **会话过滤**：按聊天会话过滤\n- **日期范围**：按创建日期过滤\n- **排序选项**：多个排序条件\n\n### 统计面板\n- **总制品**：制品总数\n- **执行统计**：总运行次数、平均执行时间、成功率\n- **类型分布**：制品类型的分布\n- **语言分布**：流行的编程语言\n- **性能图表**：使用模式的视觉表示\n\n### 制品卡片（网格视图）\n- **制品预览**：截断的代码或内容预览\n- **元数据**：创建日期、语言、会话信息\n- **操作按钮**：预览、运行/查看、编辑、复制、删除\n- **标签**：自动生成和自定义标签\n- **执行计数**：制品被运行的次数\n\n### 制品行（列表视图）\n- **紧凑布局**：一次显示更多制品\n- **基本信息**：标题、类型、语言、创建日期\n- **快速操作**：所有管理功能都可访问\n- **会话上下文**：源聊天会话的清楚指示\n\n## 操作和功能\n\n### 运行代码制品\n1. **点击运行按钮**（播放图标）在JavaScript/Python制品上\n2. **运行模态框打开** 带代码预览和执行控件\n3. **执行代码** 使用\"运行代码\"按钮\n4. **查看输出** 实时显示带语法高亮的输出\n5. **清除结果** 在新环境中重新运行\n\n### 查看视觉制品\n1. **点击查看按钮**（外部链接图标）在HTML/SVG/JSON制品上\n2. **查看模态框打开** 带正确渲染\n3. **与内容交互**（对于HTML制品）\n4. **复制内容** 使用复制按钮\n5. **关闭模态框** 返回画廊\n\n### 管理制品\n- **预览**：快速查看制品内容\n- **编辑**：修改制品内容并保存更改\n- **复制**：创建副本用于实验\n- **删除**：永久删除制品（带确认）\n- **复制**：将制品内容复制到剪贴板\n- **下载**：将制品保存为带正确扩展名的文件\n\n## 高级功能\n\n### 自动标签\n画廊基于以下内容自动生成标签：\n- **编程语言**（javascript、python、html等）\n- **框架使用**（react、vue、pandas、matplotlib等）\n- **代码模式**（async、functions、classes、loops等）\n- **库**（lodash、d3、numpy等）\n- **执行结果**（error、success、slow、fast等）\n\n### 会话上下文\n每个制品维护与其源的连接：\n- **会话标题**：聊天会话的名称\n- **消息上下文**：哪个消息创建了制品\n- **时间戳**：制品创建的时间\n- **编辑跟踪**：制品是否已被修改\n\n### 性能跟踪\n- **执行时间**：代码运行所需的时间\n- **成功率**：成功执行的百分比\n- **错误模式**：常见错误类型和频率\n- **使用统计**：最常运行的制品\n\n### 导出和备份\n- **JSON导出**：带元数据的完整制品数据\n- **过滤导出**：仅导出选定或过滤的制品\n- **备份格式**：结构化以便于导入/恢复\n- **分享**：与他人分享制品集合\n\n## 最佳实践\n\n### 组织\n- **使用描述性标题** 创建制品时\n- **添加自定义标签** 更好的组织\n- **定期清理** 过时或实验性制品\n- **会话命名** 提供更好的上下文\n\n### 代码执行\n- **增量测试** 开发复杂代码时\n- **使用console.log** 调试JavaScript\n- **优雅处理错误** 在您的代码中\n- **注意执行时间** 对于复杂操作\n\n### 性能\n- **使用分页** 用于大型制品集合\n- **清除执行结果** 当不需要时\n- **过滤制品** 专注于相关项目\n- **定期导出** 用于备份和存档\n\n## 安全考虑\n\n### 代码执行\n- **沙盒环境**：所有代码在隔离的Web Worker中运行\n- **无网络访问**：直接网络请求被阻止\n- **无文件系统访问**：无法读写本地文件\n- **有限DOM访问**：无法修改父应用程序\n- **超时保护**：长时间运行的代码自动终止\n\n### 内容渲染\n- **HTML沙盒**：iframe沙盒防止恶意脚本\n- **SVG清理**：删除潜在危险元素\n- **内容验证**：JSON和其他格式被验证\n- **安全来源**：外部资源受限\n\n## 故障排除\n\n### 常见问题\n- **代码不运行**：检查语言支持和语法\n- **制品不加载**：刷新页面或检查浏览器控制台\n- **性能问题**：减少制品数量或使用过滤器\n- **导出失败**：检查浏览器下载权限\n\n### 浏览器兼容性\n- **现代浏览器**：Chrome、Firefox、Safari、Edge（最新版本）\n- **需要JavaScript**：画廊需要JavaScript才能运行\n- **本地存储**：用于偏好设置和临时数据\n- **WebWorkers**：代码执行功能所需\n\n## API集成\n\n画廊与聊天应用程序的API集成：\n\n```javascript\n// 示例：从特定会话获取制品\nconst messages = await fetch(`/api/uuid/chat_messages/chat_sessions/${sessionId}`);\n// 制品包含在每条消息的 `artifacts` 字段中\n\n// 示例：执行代码制品\n// 代码执行由前端的 Artifact Editor/Sandbox 组件处理，\n// 后端没有独立的 `/api/artifacts/:id/execute` 接口。\n```\n\n## 未来增强\n\n### 计划功能\n- **制品版本控制**：跟踪更改并维护历史记录\n- **协作编辑**：多用户编辑制品\n- **高级分析**：详细的使用和性能指标\n- **模板系统**：创建可重用的制品模板\n- **AI建议**：自动代码改进和建议\n\n### 社区功能\n- **制品分享**：用于分享有用制品的公共画廊\n- **评分系统**：社区对流行制品的评分\n- **评论**：制品的协作讨论\n- **集合**：相关制品的策划集合\n\n此文档提供了有效使用制品画廊的全面指南。有关技术实现细节，请参阅`/src/views/chat/components/ArtifactGallery.vue`中的源代码。" }, { "path": "docs/code_runner_artifacts_tutorial.md", "content": "# Code Runner and Artifacts Tutorial\n\nThis tutorial shows how to create artifacts in chat messages, run executable code, and use the artifact gallery. It is written for end users and references the behavior implemented in the current frontend and backend.\n\n## What Are Artifacts?\n\nArtifacts are structured blocks inside a chat response that the app can detect and render specially. Supported artifact types include:\n- Code (plain or executable)\n- HTML\n- SVG\n- Mermaid diagrams\n- JSON\n- Markdown\n\nThe backend extracts artifacts from assistant messages and stores them with the chat message. The frontend renders them in the message view and in the Artifact Gallery.\n\n## Create Artifacts in Chat\n\nArtifacts are detected when the assistant responds with fenced code blocks that include a marker comment. Use these exact formats:\n\n### Code Artifact (non-executable)\n````javascript\n// Example: a plain code artifact\n```javascript \nexport function clamp(value, min, max) {\n return Math.min(Math.max(value, min), max)\n}\n```\n````\n\n### Executable Code Artifact\n````javascript\n```javascript \nconsole.log('Hello from the code runner')\n```\n````\n\n### HTML Artifact\n````html\n```html \n
\n

Hello HTML

\n

This renders inside a sandboxed iframe.

\n
\n```\n````\n\n### SVG Artifact\n````svg\n```svg \n\n \n\n```\n````\n\n### Mermaid Artifact\n````mermaid\n```mermaid \ngraph TD\n A[Start] --> B{Decision}\n B -->|Yes| C[Proceed]\n B -->|No| D[Stop]\n```\n````\n\n### JSON Artifact\n````json\n```json \n{\n \"status\": \"ok\",\n \"items\": [1, 2, 3]\n}\n```\n````\n\n## Run Executable Code\n\nExecutable artifacts show a Run button in the message view and in the Artifact Gallery.\n\nSupported languages:\n- JavaScript, TypeScript\n- Python (via Pyodide)\n\nTo run a snippet:\n1. Create an executable code artifact using the `` marker.\n2. Open the artifact in the message view or the gallery.\n3. Click Run.\n\nExecution output is shown in the artifact panel and includes logs, returns, and errors.\n\n## Load JavaScript Libraries\n\nThe JavaScript runner can load a small set of libraries from CDNs. Use this import comment at the top of your code:\n\n```javascript\n// @import lodash\nconsole.log(_.chunk([1, 2, 3, 4], 2))\n```\n\nAvailable JS libraries:\n- lodash\n- d3\n- chart.js\n- moment\n- axios\n- rxjs\n- p5\n- three\n- fabric\n\n## Load Python Packages\n\nThe Python runner auto-detects common imports and loads supported packages.\n\n```python\nimport numpy as np\nimport matplotlib.pyplot as plt\n\nx = np.linspace(0, 2 * np.pi, 100)\ny = np.sin(x)\nplt.plot(x, y)\n```\n\nAvailable Python packages include:\n- numpy, pandas, matplotlib, scipy, scikit-learn\n- requests, beautifulsoup4, pillow, sympy, networkx\n- seaborn, plotly, bokeh, altair\n\n## Use the Virtual File System (VFS)\n\nBoth runners provide a simple in-memory file system. Use absolute paths to target the VFS.\n\n### JavaScript Example\n```javascript\nconst fs = require('fs')\nfs.writeFileSync('/workspace/hello.txt', 'Hello VFS')\nconsole.log(fs.readFileSync('/workspace/hello.txt', 'utf8'))\n```\n\n### Python Example\n```python\nwith open('/workspace/hello.txt', 'w') as f:\n f.write('Hello VFS')\n\nwith open('/workspace/hello.txt', 'r') as f:\n print(f.read())\n```\n\nDefault directories:\n- `/workspace`\n- `/data`\n- `/tmp`\n\n## Artifact Gallery\n\nThe Artifact Gallery aggregates all artifacts from your chat sessions in one place.\n\nYou can:\n- Search, filter, and sort artifacts\n- Preview and edit artifacts\n- Run executable artifacts\n- Export artifacts to JSON\n\nOpen the gallery from the chat UI footer (desktop).\n\n## Tips\n\n- Use clear artifact titles so they are easy to find later.\n- Keep executable snippets short and focused.\n- Prefer JSON artifacts for structured outputs you want to view or copy.\n- Use HTML artifacts for small interactive demos (no external dependencies).\n" }, { "path": "docs/code_runner_capabilities.md", "content": "# Code Runner Capabilities\n\nThis document summarizes what the current chat + code runner can do today.\n\n## Execution\n\n- JavaScript / TypeScript execution in a sandboxed worker\n- Python execution in Pyodide\n- Console output, return values, and error reporting\n- Timeouts and basic resource limits\n\n## Libraries and Packages\n\n### JavaScript libraries (via `// @import`)\n- lodash\n- d3\n- chart.js\n- moment\n- axios\n- rxjs\n- p5\n- three\n- fabric\n\n### Python packages (auto‑loaded in Pyodide)\n- numpy, pandas, matplotlib, scipy, scikit‑learn\n- requests, beautifulsoup4, pillow, sympy, networkx\n- seaborn, plotly, bokeh, altair\n\n## Artifacts\n\n- Code artifacts and executable code artifacts\n- HTML artifacts (rendered in a sandboxed iframe)\n- SVG, Mermaid, JSON, and Markdown artifacts\n- Artifact Gallery for browsing, editing, and running artifacts\n\n## Tool Use (Code Runner enabled per session)\n\nWhen **Code Runner** is enabled in session settings, the model can use tools:\n\n- `run_code` — Execute JS/TS or Python\n- `read_vfs` — Read a file from the VFS\n- `write_vfs` — Write a file to the VFS\n- `list_vfs` — List directory entries\n- `stat_vfs` — File/directory metadata\n\nTool results are returned automatically, and the model continues with normal responses or artifacts.\n\n## Virtual File System (VFS)\n\n- In‑memory filesystem with `/data`, `/workspace`, `/tmp`\n- File upload via the UI\n- Read/write from code (both JS and Python)\n- VFS state syncs to runners before execution\n\n## Typical Workflows\n\n- Load CSVs and analyze with pandas\n- Generate plots with matplotlib\n- Prototype JS utilities or data transforms\n- Build small HTML/SVG demos in artifacts\n\n## Known Constraints\n\n- No direct DOM access for JS\n- No direct network requests from user code\n- VFS only (no access to host filesystem)\n" }, { "path": "docs/code_runner_csv_tutorial.md", "content": "# Tutorial: Chat + File Upload + Python Artifact (Iris CSV)\n\nThis walkthrough shows how to create a chat with a system prompt, upload `iris.csv`, and ask the model to generate a Python executable artifact that reads the CSV from the VFS and prints results.\n\n## 1) Create a New Chat Session\n\n1. Open the app and click **New Chat**.\n2. Set a clear title, for example: `Iris CSV Analysis`.\n3. Open the session settings and add a **System Message** like this:\n\n```text\nYou are a data assistant. Always answer with a Python executable artifact when asked to analyze data files. Use VFS paths like /workspace or /data. Print a small table and summary stats.\n```\n\nWhy this works:\n- The artifact extractor looks for fenced code blocks with ``.\n- The Python runner reads files from the in-memory VFS at `/workspace`, `/data`, or `/tmp`.\n\n## 2) Upload the CSV File to the VFS\n\n1. In the chat view, find the **VFS upload** area (often labeled “Upload files for code runners”).\n2. Upload your `iris.csv` file.\n3. Note the upload location:\n - Most uploads land under `/data/` or `/workspace/` in the VFS.\n4. For this tutorial, we will assume the file is available at:\n\n```text\n/data/iris.csv\n```\n\nIf your UI shows a different path, use that path in the next step.\n\n## 3) Prompt the LLM to Generate a Python Artifact\n\nSend a user message like this:\n\n```text\nUse Python to load the CSV at /data/iris.csv, show the first 5 rows, and print summary stats. \nReturn your answer as an executable artifact.\n```\n\nExpected artifact structure (example):\n````python\n```python \nimport pandas as pd\n\ndf = pd.read_csv('/data/iris.csv')\nprint(\"Head:\")\nprint(df.head())\nprint(\"\\nSummary:\")\nprint(df.describe())\n```\n````\n\n## 4) Run the Artifact\n\n1. The message will show an artifact panel.\n2. Click **Run**.\n3. Review output in the execution panel:\n - `stdout` for printed tables\n - `return` if a value is returned\n - `error` if the run fails\n\n## Troubleshooting\n\n- **File not found**: Confirm the path from the upload UI and update the code.\n- **Missing package**: Use supported packages (pandas, numpy, matplotlib, etc.). The runner auto-loads common imports.\n- **No output**: Ensure the code uses `print()` or returns a value.\n" }, { "path": "docs/custom_model_api_en.md", "content": "# Custom Model API (Custom Provider)\n\nThis document describes the custom model API contract used by the backend when a chat model is configured with `api_type = custom`.\n\n## When this path is used\n\nThe backend selects the custom provider when the chat model's `api_type` is set to `custom` (Admin → Models → Add Model).\n\n## Required model config (Admin UI)\n\n- **API Type**: `Custom`\n- **URL**: Your custom API endpoint (HTTP POST).\n- **API Auth Header**: The header name to pass the API key (optional).\n- **API Auth Key**: The *environment variable name* that stores the API key.\n\nExample:\n\n- API Auth Header: `x-api-key`\n- API Auth Key: `MY_CUSTOM_API_KEY`\n- Then set `MY_CUSTOM_API_KEY=...` in the backend environment.\n\n## Request format\n\nThe backend always sends a JSON POST body with the following fields:\n\n```json\n{\n \"prompt\": \"\\n\\nHuman: Hello\\n\\nAssistant: Hi there!\\n\\nHuman: ...\\n\\nAssistant: \",\n \"model\": \"custom-my-model\",\n \"max_tokens_to_sample\": 2048,\n \"temperature\": 0.7,\n \"stop_sequences\": [\"\\n\\nHuman:\"],\n \"stream\": true\n}\n```\n\nNotes:\n\n- `prompt` is formatted in a Claude-style transcript. Each non-assistant message becomes:\n `\\n\\nHuman: \\n\\nAssistant: `\n- `model` is the chat model name stored in the database.\n- `max_tokens_to_sample` and `temperature` come from the session settings.\n- `stream` is always `true` for the custom provider.\n\nHeaders added by the backend:\n\n- `Content-Type: application/json`\n- `Accept: text/event-stream`\n- `Cache-Control: no-cache`\n- `Connection: keep-alive`\n- Optional auth header if configured (e.g. `x-api-key: `).\n\n## Streaming response format (required)\n\nYour API must respond with `text/event-stream` (SSE) and emit lines that start with `data: `.\nEach `data:` line must contain JSON that includes a `completion` field.\nThe backend **expects full text so far** in `completion` on each event (not deltas).\n\nExample stream:\n\n```\ndata: {\"completion\":\"Hello\",\"stop\":null,\"stop_reason\":null,\"truncated\":false,\"log_id\":\"1\",\"model\":\"custom-my-model\",\"exception\":null}\ndata: {\"completion\":\"Hello there\",\"stop\":null,\"stop_reason\":null,\"truncated\":false,\"log_id\":\"1\",\"model\":\"custom-my-model\",\"exception\":null}\ndata: [DONE]\n```\n\nThe stream ends when you send a line starting with `data: [DONE]`.\n\n## Minimum response fields\n\nThe backend only uses `completion`, but it unmarshals the following fields:\n\n```json\n{\n \"completion\": \"string\",\n \"stop\": \"string or null\",\n \"stop_reason\": \"string or null\",\n \"truncated\": false,\n \"log_id\": \"string\",\n \"model\": \"string\",\n \"exception\": {}\n}\n```\n\n## Quick checklist\n\n- Endpoint accepts JSON POST with the fields listed above.\n- Supports SSE with `data: {json}\\n` lines.\n- Sends full accumulated text in `completion` each event.\n- Ends with `data: [DONE]`.\n- Configure the model in Admin with `api_type = custom` and correct auth env var.\n" }, { "path": "docs/deployment_en.md", "content": "## How to Deploy\n\nRefer to `docker-compose.yaml`\n\n[![Deploy on Railway](https://railway.app/button.svg)](https://railway.app/template/tk7jWU?referralCode=5DMfQv)\n\nThen configure the environment variables.\n\n```\nPORT=8080\nOPENAI_RATELIMIT=0\n```\n\nFill in the other two keys if you have them.\n\n\"image\"\n\nAfter deployment, registering users, the first user is an administrator, then go to\n to set rate limiting. Public deployment,\nonly adds rate limiting to trusted emails, so even if someone registers, it will not be available.\n\n\"image\"\n\nThis helps ensure only authorized users can access the deployed system by limiting registration to trusted emails and enabling rate limiting controls.\n" }, { "path": "docs/deployment_zh.md", "content": "## 如何部署\n\n参考 `docker-compose.yaml`\n\n[![Deploy on Railway](https://railway.app/button.svg)](https://railway.app/template/tk7jWU?referralCode=5DMfQv)\n\n然后配置环境变量就可以了.\n\n```\nPORT=8080\nOPENAI_RATELIMIT=0\n```\n\n别的两个 api key 有就填.\n\n\"image\"\n\n部署之后, 注册用户, 第一个用户是管理员, 然后到 ,\n设置 ratelimit, 公网部署, 只对信任的email 增加 ratelimit, 这样即使有人注册, 也是不能用的.\n\n\"image\"\n" }, { "path": "docs/dev/ERROR_HANDLING_STANDARDS.md", "content": "# Error Handling Standardization Guide\n\n## Current Issues\n- Inconsistent use of `http.Error()` vs `RespondWithAPIError()`\n- Mixed error response formats\n- Some areas missing proper logging\n\n## Standards to Follow\n\n### 1. Use RespondWithAPIError for all API responses\nReplace:\n```go\nhttp.Error(w, err.Error(), http.StatusBadRequest)\n```\n\nWith:\n```go\nRespondWithAPIError(w, ErrValidationInvalidInput(\"Failed to decode request body\").WithDebugInfo(err.Error()))\n```\n\n### 2. Always log errors before responding\n```go\nlog.WithError(err).Error(\"Context-specific error message\")\nRespondWithAPIError(w, ErrInternalUnexpected.WithMessage(\"User-friendly message\"))\n```\n\n### 3. Use appropriate error types\n- `ErrValidationInvalidInput` for bad request data\n- `ErrResourceNotFound` for 404 errors\n- `ErrInternalUnexpected` for 500 errors\n- `ErrPermissionDenied` for 403 errors\n\n### 4. Files that need standardization\n- chat_auth_user_handler.go (partially fixed)\n- admin_handler.go\n- chat_prompt_hander.go \n- chat_comment_handler.go\n- chat_message_handler.go\n- handle_tts.go\n\n## Implementation Priority\n1. Authentication handlers (high impact)\n2. Core chat functionality \n3. Admin and utility handlers\n\nThis standardization should be done gradually to avoid breaking changes." }, { "path": "docs/dev/INTEGRATION_GUIDE.md", "content": "# 🚀 Quick Integration Guide: Add File Upload to Chat\n\nThis guide shows you exactly how to add VFS file upload to your chat session in 3 simple steps.\n\n## 📂 Files to Add\n\nFirst, make sure you have these new components in your project:\n\n```\nweb/src/components/\n├── VFSProvider.vue ✅ (already created)\n├── ChatVFSUploader.vue ✅ (already created) \n├── VFSFileManager.vue ✅ (already created)\n└── VFSIntegration.vue ✅ (already created)\n\nweb/src/utils/\n├── virtualFileSystem.js ✅ (already created)\n└── vfsImportExport.js ✅ (already created)\n```\n\n## 🔧 Step 1: Modify Conversation.vue\n\nAdd these **3 lines** to your `web/src/views/chat/components/Conversation.vue`:\n\n### Add Imports (at the top with other imports):\n```typescript\n// Add these two lines to existing imports\nimport ChatVFSUploader from '@/components/ChatVFSUploader.vue'\nimport VFSProvider from '@/components/VFSProvider.vue'\n```\n\n### Add Event Handler (in script setup section):\n```typescript\n// Add this event handler with your other functions\nconst handleVFSFileUploaded = (fileInfo: any) => {\n nui_msg.success(`📁 File uploaded: ${fileInfo.filename}`)\n \n // Optional: Add a helpful message to chat\n const helpMessage = `File uploaded to VFS! Use this code to access it:\n\n**Python:**\n\\`\\`\\`python\n# For ${fileInfo.filename}\nimport pandas as pd # if CSV\ndata = pd.read_csv('${fileInfo.path}')\nprint(data.head())\n\\`\\`\\`\n\n**JavaScript:**\n\\`\\`\\`javascript\n// For ${fileInfo.filename}\nconst fs = require('fs');\nconst content = fs.readFileSync('${fileInfo.path}', 'utf8');\nconsole.log(content);\n\\`\\`\\``\n\n addChat(\n sessionUuid,\n {\n uuid: uuidv7(),\n dateTime: nowISO(),\n text: helpMessage,\n inversion: false,\n error: false,\n loading: false,\n },\n )\n}\n```\n\n### Modify Template (wrap and add uploader):\n\nFind your existing template and make these changes:\n\n**BEFORE:**\n```vue\n\n```\n\n**AFTER:**\n```vue\n\n```\n\n## 🎯 Step 2: Test It!\n\n1. **Start your dev server**\n2. **Go to any chat session** \n3. **Look for the folder icon button** in the bottom right of the chat\n4. **Click it and upload a CSV or JSON file**\n5. **The system will show code examples** for accessing the file\n6. **Copy and run the code** in a code runner!\n\n## 📋 Complete Example\n\nHere's what a complete integration looks like in your Conversation.vue:\n\n```vue\n\n\n\n```\n\n## 🎉 That's It!\n\nYou now have:\n\n✅ **File Upload Button** - Users can upload files to VFS \n✅ **Auto Code Generation** - System shows how to use uploaded files \n✅ **Cross-Language Support** - Files work in both Python and JavaScript \n✅ **Session Integration** - Files persist throughout the chat session \n✅ **File Manager** - Browse and download VFS files \n\n## 💡 Usage Examples\n\nAfter integration, users can:\n\n1. **Upload** `sales.csv` → VFS stores it at `/data/sales.csv`\n2. **Get code examples** automatically in chat\n3. **Run Python code:**\n ```python\n import pandas as pd\n df = pd.read_csv('/data/sales.csv')\n df['profit_margin'] = df['profit'] / df['sales'] * 100\n df.to_csv('/data/sales_with_margins.csv', index=False)\n ```\n4. **Run JavaScript code:**\n ```javascript\n const fs = require('fs');\n const data = fs.readFileSync('/data/sales_with_margins.csv', 'utf8');\n console.log('Processed data:', data.split('\\n').length, 'rows');\n ```\n5. **Download results** via the file manager\n\nThe VFS creates a complete data processing workflow within your chat interface! 🚀" }, { "path": "docs/dev/code_runner_manual.md", "content": "# Code Runner User Manual\n\n## Overview\n\nThe Code Runner is an interactive JavaScript execution environment built into the chat application. It allows you to write, run, and experiment with JavaScript code directly in chat messages, with real-time output, graphics support, and access to popular libraries.\n\n## Getting Started\n\n### Creating Executable Code Artifacts\n\nThere are two ways to create executable code artifacts:\n\n#### Method 1: Explicit Executable Syntax\n```javascript\n// @import lodash\nconsole.log('Hello, World!')\nconst numbers = [1, 2, 3, 4, 5]\nconst sum = _.sum(numbers)\nconsole.log('Sum:', sum)\nreturn sum\n```\n\n#### Method 2: Using the Executable Marker\nWhen asking the AI to create executable code, use this format:\n```\nCan you create an executable JavaScript function that calculates fibonacci numbers?\n```\n\nThe AI will automatically create artifacts with the `` marker.\n\n## Basic Features\n\n### 1. Console Output\n```javascript\nconsole.log('This is a log message')\nconsole.error('This is an error message') \nconsole.warn('This is a warning')\nconsole.info('This is info')\n```\n\n### 2. Return Values\n```javascript\n// The return value is automatically displayed\nconst result = Math.PI * 2\nreturn result\n```\n\n### 3. Error Handling\n```javascript\ntry {\n throw new Error('Something went wrong!')\n} catch (error) {\n console.error('Caught error:', error.message)\n}\n```\n\n## Advanced Features\n\n### 1. Library Loading\n\nThe Code Runner supports 9 popular JavaScript libraries that can be loaded automatically:\n\n- **lodash**: Utility functions\n- **d3**: Data visualization\n- **chart.js**: Chart creation\n- **moment**: Date/time manipulation\n- **axios**: HTTP requests (limited to safe operations)\n- **rxjs**: Reactive programming\n- **p5**: Creative coding\n- **three**: 3D graphics\n- **fabric**: Canvas manipulation\n\n#### Usage:\n```javascript\n// @import lodash\n// @import d3\n\n// Use lodash\nconst numbers = [1, 2, 3, 4, 5]\nconst doubled = _.map(numbers, n => n * 2)\nconsole.log('Doubled:', doubled)\n\n// Use d3\nconst scale = d3.scaleLinear().domain([0, 10]).range([0, 100])\nconsole.log('Scaled value:', scale(5))\n\nreturn { doubled, scaled: scale(5) }\n```\n\n### 2. Canvas Graphics\n\nCreate visual outputs using the built-in canvas support:\n\n```javascript\n// Create a canvas\nconst canvas = createCanvas(400, 300)\nconst ctx = canvas.getContext('2d')\n\n// Draw a colorful rectangle\nctx.fillStyle = '#FF6B6B'\nctx.fillRect(50, 50, 100, 80)\n\n// Draw a circle\nctx.fillStyle = '#4ECDC4'\nctx.beginPath()\nctx.arc(200, 150, 40, 0, 2 * Math.PI)\nctx.fill()\n\n// Add text\nctx.fillStyle = '#45B7D1'\nctx.fillText('Hello Canvas!', 100, 200)\n\n// Return the canvas to display it\nreturn canvas\n```\n\n### 3. Data Visualization with D3\n\n```javascript\n// @import d3\n\n// Create a simple bar chart\nconst canvas = createCanvas(400, 300)\nconst ctx = canvas.getContext('2d')\n\nconst data = [10, 20, 30, 40, 50]\nconst scale = d3.scaleLinear().domain([0, 50]).range([0, 200])\n\ndata.forEach((value, index) => {\n const barHeight = scale(value)\n const x = index * 60 + 50\n const y = 250 - barHeight\n \n ctx.fillStyle = `hsl(${index * 60}, 70%, 50%)`\n ctx.fillRect(x, y, 40, barHeight)\n \n ctx.fillStyle = '#000'\n ctx.fillText(value, x + 15, y - 5)\n})\n\nreturn canvas\n```\n\n### 4. Mathematical Computations\n\n```javascript\n// @import lodash\n\n// Generate random data\nconst dataset = _.range(100).map(() => Math.random() * 100)\n\n// Calculate statistics\nconst stats = {\n mean: _.mean(dataset),\n median: _.sortBy(dataset)[Math.floor(dataset.length / 2)],\n min: _.min(dataset),\n max: _.max(dataset),\n sum: _.sum(dataset)\n}\n\nconsole.log('Dataset Statistics:', stats)\n\n// Visualize distribution\nconst canvas = createCanvas(400, 200)\nconst ctx = canvas.getContext('2d')\n\nconst buckets = _.range(0, 101, 10)\nconst histogram = buckets.map(bucket => \n dataset.filter(value => value >= bucket && value < bucket + 10).length\n)\n\nhistogram.forEach((count, index) => {\n const barHeight = count * 5\n const x = index * 35 + 20\n const y = 180 - barHeight\n \n ctx.fillStyle = '#3498db'\n ctx.fillRect(x, y, 30, barHeight)\n \n ctx.fillStyle = '#000'\n ctx.fillText(count, x + 10, y - 5)\n})\n\nreturn { stats, histogram }\n```\n\n## User Interface Guide\n\n### 1. Code Execution Controls\n\n- **▶️ Run Code**: Execute the current code\n- **🗑️ Clear Output**: Remove all output\n- **✏️ Edit Mode**: Switch between view and edit modes\n- **📦 Libraries**: View available libraries\n\n### 2. Library Management\n\nClick the \"Available\" button next to \"Libraries\" to see:\n- List of all available libraries\n- Usage instructions\n- Import syntax examples\n\n### 3. Output Display\n\nThe output area shows:\n- **Console logs**: Blue background\n- **Errors**: Red background with error details\n- **Return values**: Purple background\n- **Canvas graphics**: Rendered inline\n- **Execution stats**: Time, memory usage, operations\n\n### 4. Keyboard Shortcuts\n\n- **Ctrl/Cmd + Enter**: Run code while in edit mode\n- **Escape**: Exit edit mode\n\n## Performance and Limits\n\n### Resource Limits\n\nThe Code Runner has built-in safety limits:\n- **Execution Time**: 10 seconds maximum\n- **Memory Usage**: ~50MB limit\n- **Operations**: 100,000 operations max (prevents infinite loops)\n- **Library Loading**: 30 seconds timeout\n\n### Performance Monitoring\n\nEach execution shows:\n- **Execution time**: How long the code took to run\n- **Memory usage**: Approximate memory consumption\n- **Operation count**: Number of operations performed\n\nExample output:\n```\nExecution completed in 45ms | ~2.3MB | 1,247 ops\n```\n\n## Error Handling and Debugging\n\n### Common Errors\n\n1. **Library Not Found**:\n ```\n Error: Library 'unknown' is not available\n ```\n Solution: Check available libraries and use correct names\n\n2. **Memory Limit Exceeded**:\n ```\n Error: Memory limit exceeded: ~52MB\n ```\n Solution: Optimize code to use less memory\n\n3. **Operation Limit Exceeded**:\n ```\n Error: Operation limit exceeded: 100000 operations\n ```\n Solution: Check for infinite loops or reduce computational complexity\n\n4. **Canvas Errors**:\n ```\n Error: Canvas error: Cannot read property 'getContext' of null\n ```\n Solution: Ensure canvas is created before using\n\n### Debugging Tips\n\n1. **Use console.log liberally**:\n ```javascript\n const data = [1, 2, 3]\n console.log('Data:', data)\n \n const result = data.map(x => x * 2)\n console.log('Result:', result)\n ```\n\n2. **Break complex code into steps**:\n ```javascript\n // Step 1: Generate data\n const data = _.range(10).map(() => Math.random())\n console.log('Generated data:', data.length, 'points')\n \n // Step 2: Process data\n const processed = data.map(x => x * 100)\n console.log('Processed data range:', _.min(processed), 'to', _.max(processed))\n \n // Step 3: Visualize\n console.log('Creating visualization...')\n // ... canvas code\n ```\n\n3. **Check execution stats**:\n - Monitor memory usage for large datasets\n - Watch operation count for loops\n - Optimize based on execution time\n\n## Advanced Examples\n\n### 1. Interactive Algorithm Visualization\n\n```javascript\n// @import lodash\n\n// Bubble sort with visualization\nconst canvas = createCanvas(400, 300)\nconst ctx = canvas.getContext('2d')\n\nconst data = _.shuffle(_.range(1, 21)) // Random array 1-20\nconst steps = []\n\n// Bubble sort algorithm\nfor (let i = 0; i < data.length; i++) {\n for (let j = 0; j < data.length - i - 1; j++) {\n if (data[j] > data[j + 1]) {\n // Swap\n [data[j], data[j + 1]] = [data[j + 1], data[j]]\n steps.push([...data]) // Record step\n }\n }\n}\n\n// Visualize final sorted array\ndata.forEach((value, index) => {\n const barHeight = value * 10\n const x = index * 18 + 10\n const y = 280 - barHeight\n \n ctx.fillStyle = `hsl(${value * 18}, 70%, 50%)`\n ctx.fillRect(x, y, 15, barHeight)\n \n ctx.fillStyle = '#000'\n ctx.fillText(value, x + 2, y - 2)\n})\n\nconsole.log(`Sorting completed in ${steps.length} steps`)\nreturn canvas\n```\n\n### 2. Statistical Analysis with Charts\n\n```javascript\n// @import lodash\n// @import d3\n\n// Generate sample data\nconst sampleSize = 1000\nconst data = _.range(sampleSize).map(() => \n d3.randomNormal(50, 15)() // Normal distribution, mean=50, std=15\n)\n\n// Calculate statistics\nconst stats = {\n count: data.length,\n mean: d3.mean(data),\n median: d3.median(data),\n deviation: d3.deviation(data),\n min: d3.min(data),\n max: d3.max(data)\n}\n\nconsole.log('Statistics:', stats)\n\n// Create histogram\nconst canvas = createCanvas(500, 400)\nconst ctx = canvas.getContext('2d')\n\nconst bins = d3.histogram()\n .domain(d3.extent(data))\n .thresholds(20)(data)\n\nconst xScale = d3.scaleLinear()\n .domain(d3.extent(data))\n .range([50, 450])\n\nconst yScale = d3.scaleLinear()\n .domain([0, d3.max(bins, d => d.length)])\n .range([350, 50])\n\nbins.forEach(bin => {\n const x = xScale(bin.x0)\n const y = yScale(bin.length)\n const width = xScale(bin.x1) - xScale(bin.x0) - 1\n const height = 350 - y\n \n ctx.fillStyle = '#3498db'\n ctx.fillRect(x, y, width, height)\n \n ctx.fillStyle = '#000'\n ctx.fillText(bin.length, x + width/2 - 5, y - 5)\n})\n\n// Add title\nctx.fillStyle = '#000'\nctx.font = '16px Arial'\nctx.fillText('Normal Distribution Histogram', 150, 30)\n\nreturn { stats, canvas }\n```\n\n### 3. Real-time Data Processing\n\n```javascript\n// @import lodash\n// @import moment\n\n// Simulate time-series data\nconst now = moment()\nconst timePoints = _.range(24).map(hour => ({\n time: now.clone().subtract(24 - hour, 'hours'),\n value: Math.sin(hour * Math.PI / 12) * 50 + 50 + Math.random() * 20\n}))\n\n// Process data\nconst processed = timePoints.map(point => ({\n hour: point.time.format('HH:mm'),\n value: Math.round(point.value * 100) / 100,\n trend: point.value > 50 ? 'up' : 'down'\n}))\n\n// Calculate rolling average\nconst windowSize = 3\nconst rollingAverage = processed.map((point, index) => {\n const start = Math.max(0, index - windowSize + 1)\n const window = processed.slice(start, index + 1)\n const avg = _.meanBy(window, 'value')\n return { ...point, rollingAvg: Math.round(avg * 100) / 100 }\n})\n\nconsole.log('Data points:', processed.length)\nconsole.log('Sample:', processed.slice(0, 3))\n\n// Visualize trends\nconst canvas = createCanvas(600, 300)\nconst ctx = canvas.getContext('2d')\n\nrollingAverage.forEach((point, index) => {\n const x = index * 24 + 50\n const y = 250 - (point.value * 2)\n const avgY = 250 - (point.rollingAvg * 2)\n \n // Draw data point\n ctx.fillStyle = point.trend === 'up' ? '#2ecc71' : '#e74c3c'\n ctx.fillRect(x - 2, y - 2, 4, 4)\n \n // Draw rolling average\n ctx.fillStyle = '#3498db'\n ctx.fillRect(x - 1, avgY - 1, 2, 2)\n \n // Connect points\n if (index > 0) {\n const prevX = (index - 1) * 24 + 50\n const prevY = 250 - (rollingAverage[index - 1].rollingAvg * 2)\n \n ctx.strokeStyle = '#3498db'\n ctx.beginPath()\n ctx.moveTo(prevX, prevY)\n ctx.lineTo(x, avgY)\n ctx.stroke()\n }\n})\n\nreturn { processed: rollingAverage, canvas }\n```\n\n## Best Practices\n\n### 1. Code Organization\n\n- Break complex tasks into smaller functions\n- Use descriptive variable names\n- Add comments for complex logic\n- Return meaningful results\n\n### 2. Performance Optimization\n\n- Avoid unnecessary loops\n- Use efficient algorithms\n- Monitor memory usage for large datasets\n- Consider using libraries like lodash for optimized operations\n\n### 3. Error Prevention\n\n- Validate inputs before processing\n- Use try-catch blocks for risky operations\n- Check array lengths before accessing elements\n- Handle edge cases explicitly\n\n### 4. Visualization Guidelines\n\n- Choose appropriate canvas sizes (400x300 is standard)\n- Use contrasting colors for better visibility\n- Add labels and legends when helpful\n- Scale graphics appropriately for the data\n\n## Security and Limitations\n\n### What's Allowed\n\n- All standard JavaScript features\n- Mathematical computations\n- Data manipulation and analysis\n- Canvas graphics and visualizations\n- Supported library functions\n- Console output and debugging\n\n### What's Not Allowed\n\n- Direct DOM manipulation\n- Network requests (fetch, XMLHttpRequest)\n- File system access\n- Local storage access\n- WebSocket connections\n- Worker creation\n- Eval or Function constructor (except internally)\n\n### Resource Limits\n\n- Maximum execution time: 10 seconds\n- Memory limit: ~50MB\n- Operation limit: 100,000 operations\n- Library loading timeout: 30 seconds\n\n## Troubleshooting\n\n### Common Issues\n\n1. **Code doesn't run**: Check for syntax errors\n2. **Canvas doesn't display**: Ensure you return the canvas object\n3. **Library not loading**: Verify library name and internet connection\n4. **Performance issues**: Check operation count and optimize loops\n\n### Getting Help\n\n- Use console.log to debug step by step\n- Check the execution statistics for performance insights\n- Simplify complex code to isolate issues\n- Refer to library documentation for specific functions\n\nThis Code Runner provides a powerful environment for learning, prototyping, and demonstrating JavaScript concepts with real-time feedback and rich visualizations. Experiment with different features and libraries to discover what's possible!" }, { "path": "docs/dev/conversation_patch_example.js", "content": "/**\n * Example patch for Conversation.vue to add VFS file upload\n * \n * This shows the minimal changes needed to add VFS upload to the chat interface\n */\n\n// 1. Add these imports to the existing imports section\nconst additionalImports = `\nimport ChatVFSUploader from '@/components/ChatVFSUploader.vue'\nimport VFSProvider from '@/components/VFSProvider.vue'\n`\n\n// 2. Add these event handlers to the script section\nconst eventHandlers = `\n// VFS event handlers\nconst handleVFSFileUploaded = (fileInfo) => {\n console.log('File uploaded to VFS:', fileInfo)\n nui_msg.success(\\`File uploaded: \\${fileInfo.filename} → \\${fileInfo.path}\\`)\n}\n\nconst handleCodeExampleAdded = (codeInfo) => {\n // Add code examples as a system message\n const exampleMessage = \\`📁 **Files uploaded successfully!**\n\n**Python example:**\n\\\\\\`\\\\\\`\\\\\\`python\n\\${codeInfo.python}\n\\\\\\`\\\\\\`\\\\\\`\n\n**JavaScript example:**\n\\\\\\`\\\\\\`\\\\\\`javascript\n\\${codeInfo.javascript}\n\\\\\\`\\\\\\`\\\\\\`\n\nYour files are now available in the Virtual File System.\\`\n\n // Add system message to chat\n addChat(\n sessionUuid,\n {\n uuid: uuidv7(),\n dateTime: nowISO(),\n text: exampleMessage,\n inversion: false,\n error: false,\n loading: false,\n },\n )\n \n nui_msg.success('Files uploaded! Code examples added to chat.')\n}\n`\n\n// 3. Template modification - wrap the entire template with VFSProvider\nconst templateWrapper = `\n\n`\n\n// 4. Add these styles\nconst additionalStyles = `\n\n`\n\n// 5. Complete minimal integration example\nconst minimalIntegrationExample = `\n// MINIMAL INTEGRATION: Just add these 3 things to Conversation.vue\n\n// 1. Add import\nimport ChatVFSUploader from '@/components/ChatVFSUploader.vue'\nimport VFSProvider from '@/components/VFSProvider.vue'\n\n// 2. Add event handler\nconst handleVFSFileUploaded = (fileInfo) => {\n nui_msg.success(\\`File uploaded: \\${fileInfo.filename}\\`)\n}\n\n// 3. Add to template (wrap existing content with VFSProvider and add uploader)\n/*\n\n*/\n`\n\nexport {\n additionalImports,\n eventHandlers,\n templateWrapper,\n additionalStyles,\n minimalIntegrationExample\n}" }, { "path": "docs/dev/conversation_vfs_integration.md", "content": "# Chat Session VFS Integration\n\nThis guide shows how to add VFS file upload functionality to chat sessions so users can upload files and use them directly in code runners.\n\n## Integration Steps\n\n### 1. Add VFS Provider to Chat Layout\n\nFirst, wrap your chat layout with the VFS provider in `web/src/views/chat/layout/Layout.vue`:\n\n```vue\n\n\n\n```\n\n### 2. Modify Conversation Component\n\nUpdate `web/src/views/chat/components/Conversation.vue`:\n\n```vue\n\n\n\n\n\n```\n\n### 3. Alternative: Simpler Integration\n\nFor a cleaner integration, you can add the VFS uploader as a button next to the existing upload button:\n\n```vue\n\n\n\n```\n\n## Usage Flow\n\n### 1. User uploads files via VFS uploader:\n- Click the VFS upload button (folder icon)\n- Select files (CSV, JSON, Excel, text files, etc.)\n- Choose target directory (`/data`, `/workspace`, `/uploads`)\n- Files are uploaded to the Virtual File System\n\n### 2. System generates code examples:\n- Automatically generates Python and JavaScript examples\n- Shows how to access the uploaded files\n- User can copy the code or add it to chat\n\n### 3. User runs code in chat:\n```python\n# Python example (auto-generated)\nimport pandas as pd\ndf = pd.read_csv('/data/sales.csv')\nprint(f\"Loaded {len(df)} rows from sales.csv\")\nprint(df.head())\n```\n\n```javascript\n// JavaScript example (auto-generated) \nconst fs = require('fs');\nconst csvContent = fs.readFileSync('/data/sales.csv', 'utf8');\nconst lines = csvContent.split('\\n');\nconsole.log(`Loaded CSV sales.csv with ${lines.length - 1} rows`);\n```\n\n### 4. Files persist throughout the session:\n- Files remain available in VFS during the entire chat session\n- Can be accessed by subsequent code runners\n- Can be downloaded via file manager\n\n## Key Features\n\n### Dual Upload System:\n- **Server Upload** (existing): Files for chat context, AI can see content\n- **VFS Upload** (new): Files for code execution, available in runners\n\n### Smart Code Generation:\n- Detects file types and generates appropriate code\n- Provides both Python and JavaScript examples\n- Shows exact file paths for easy copy-paste\n\n### Session Integration:\n- Files tied to specific chat session\n- Automatic code example insertion into chat\n- Visual feedback and success messages\n\n### User Experience:\n- Clear distinction between upload types\n- Helpful tooltips and guidance\n- Immediate code examples\n- File manager for browsing/downloading\n\n## Complete Example Message Flow\n\n1. **User uploads** `sales_data.csv` via VFS uploader\n2. **System responds** with auto-generated message:\n ```\n 📁 Files uploaded successfully!\n \n Python example:\n ```python\n import pandas as pd\n df = pd.read_csv('/data/sales_data.csv')\n print(f\"Loaded {len(df)} rows from sales_data.csv\")\n ```\n \n JavaScript example:\n ```javascript\n const fs = require('fs');\n const csvContent = fs.readFileSync('/data/sales_data.csv', 'utf8');\n ```\n ```\n3. **User copies and runs** the Python code\n4. **User processes data** and saves results to VFS\n5. **User downloads** processed files via file manager\n\nThis creates a complete data processing workflow within the chat interface!" }, { "path": "docs/dev/python_async_execution.md", "content": "# Running Async Python Code in Pyodide\n\n## Overview\n\nThis document outlines the challenges and solutions for executing asynchronous Python code in the Pyodide-based Python runner used in this chat application.\n\n## The Problem\n\nWhen users write Python code containing `asyncio.run()` calls, they encounter a runtime error:\n\n```\nRuntimeError: asyncio.run() cannot be called from a running event loop\n```\n\nThis occurs because Pyodide already runs within an active asyncio event loop, and `asyncio.run()` attempts to create a new event loop, which is not allowed.\n\n## Root Cause\n\n- **Pyodide Environment**: Pyodide executes Python code within a JavaScript environment that already has an active asyncio event loop\n- **asyncio.run() Limitation**: This function is designed to be the main entry point for asyncio programs and cannot be called from within an existing event loop\n- **Common Pattern**: Many Python async tutorials and examples use `if __name__ == \"__main__\": asyncio.run(main())` which doesn't work in Pyodide\n\n## Solution Implementation\n\n### Detection and Transformation\n\nThe Python runner now automatically detects and transforms `asyncio.run()` calls:\n\n1. **Pattern Detection**: Code is scanned for `asyncio.run()` calls\n2. **Syntax Transformation**: `asyncio.run(func())` is converted to `await func()`\n3. **Context Wrapping**: The entire code is wrapped in an async function context\n4. **Pyodide Execution**: Uses `pyodide.runPythonAsync()` instead of `pyodide.runPython()`\n\n### Code Transformation Example\n\n**Original Code:**\n```python\nimport asyncio\n\nasync def main():\n print(\"Hello from async!\")\n await asyncio.sleep(1)\n print(\"Done!\")\n\nif __name__ == \"__main__\":\n asyncio.run(main())\n```\n\n**Transformed Code:**\n```python\nimport asyncio\n\nasync def _execute_main():\n import asyncio\n\n async def main():\n print(\"Hello from async!\")\n await asyncio.sleep(1)\n print(\"Done!\")\n\n if __name__ == \"__main__\":\n await main()\n\n# Execute the main function\nawait _execute_main()\n```\n\n### Key Changes in Implementation\n\n1. **Regex Replacement**: `asyncio.run(([^)]+))` → `await $1`\n2. **Async Wrapper**: Entire code wrapped in `async def _execute_main():`\n3. **Top-level Await**: Uses `await _execute_main()` for execution\n4. **Execution Method**: Uses `pyodide.runPythonAsync()` for async code\n\n## Technical Details\n\n### Why This Works\n\n- **Pyodide Compatibility**: `runPythonAsync()` is designed to handle top-level await statements\n- **Event Loop Reuse**: Instead of creating a new event loop, the code runs within Pyodide's existing loop\n- **Proper Awaiting**: The async function is properly awaited, preventing \"coroutine was never awaited\" warnings\n\n### Error Handling\n\nThe runner provides informative feedback:\n- Detects `asyncio.run()` usage and notifies the user\n- Explains the transformation being applied\n- Maintains error context for debugging\n\n## Best Practices\n\n### For Users\n\n1. **Avoid `asyncio.run()`**: In Pyodide environments, use async/await directly\n2. **Top-level Async**: Write async functions and let the runner handle execution\n3. **Error Awareness**: Understand that Pyodide has different async execution patterns\n\n### For Developers\n\n1. **Detection First**: Always scan for problematic patterns before execution\n2. **Clear Messaging**: Inform users when code transformations occur\n3. **Fallback Strategy**: Use `runPython()` for synchronous code, `runPythonAsync()` for async\n4. **Testing**: Test with various async patterns including nested async calls\n\n## Limitations\n\n1. **Complex Patterns**: Very complex async patterns may still require manual adjustment\n2. **Performance**: Code transformation adds slight overhead\n3. **Debugging**: Transformed code may be harder to debug than original\n\n## Future Improvements\n\n1. **Better Pattern Recognition**: Handle more complex `asyncio.run()` usage patterns\n2. **Source Maps**: Maintain mapping between original and transformed code for better debugging\n3. **Optimization**: Cache transformation results for repeated code execution\n4. **User Education**: Provide more guidance on async Python patterns in Pyodide\n\n## Conclusion\n\nThe async Python code execution solution successfully bridges the gap between standard Python async patterns and Pyodide's execution environment. By automatically detecting and transforming `asyncio.run()` calls, users can run async Python code seamlessly without needing to understand the underlying Pyodide constraints." }, { "path": "docs/dev/sse_processing_logic.md", "content": "# Server-Sent Events (SSE) Processing Logic\n\nThis document explains how the chat application handles Server-Sent Events (SSE) streaming responses from the backend.\n\n## Overview\n\nThe SSE processing logic is implemented in `useStreamHandling.ts` and handles real-time streaming of chat responses. It manages buffering, message parsing, and error handling for continuous data streams.\n\n## SSE Protocol Basics\n\nServer-Sent Events follow this format:\n```\ndata: {\"id\": \"chatcmpl-123\", \"choices\": [{\"delta\": {\"content\": \"Hello\"}}]}\n\ndata: {\"id\": \"chatcmpl-123\", \"choices\": [{\"delta\": {\"content\": \" world\"}}]}\n\ndata: [DONE]\n\n```\n\n- Each message starts with `data: `\n- Messages are separated by double newlines (`\\n\\n`)\n- The stream ends with `data: [DONE]`\n\n## Processing Flow\n\n### 1. Stream Reading Setup\n\n```javascript\nconst response = await fetch(streamingUrl, requestConfig)\nconst reader = response.body.getReader()\nconst decoder = new TextDecoder()\nlet buffer = ''\n```\n\n### 2. Chunk Processing Loop\n\n```javascript\nwhile (true) {\n const { done, value } = await reader.read()\n if (done) break\n \n const chunk = decoder.decode(value, { stream: true })\n buffer += chunk\n \n // Process complete messages\n processBuffer(buffer)\n}\n```\n\n### 3. Buffer Management\n\nThe key challenge is handling partial messages that arrive across multiple chunks:\n\n```javascript\n// Input buffer: \"data: {partial}\\n\\ndata: {complete}\\n\\nda\"\nconst lines = buffer.split('\\n\\n')\nbuffer = lines.pop() || '' // Keep incomplete part: \"da\"\n\n// Process complete messages: [\"data: {partial}\", \"data: {complete}\"]\nfor (const line of lines) {\n if (line.trim()) {\n processMessage(line)\n }\n}\n\n// because each delta is acutally a full message\n// process last one in enough\nif (lines.length > 0) {\n onStreamChunk(lines[lines.length - 1], updateIndex)\n}\n```\n\n### 4. Data Extraction\n\nEach SSE message is processed by `extractStreamingData()`:\n\n```javascript\nfunction extractStreamingData(streamResponse: string): string {\n // Handle standard SSE format: \"data: {...}\"\n if (streamResponse.startsWith('data:')) {\n return streamResponse.slice(5).trim()\n }\n \n // Handle multiple segments (fallback)\n const lastDataPosition = streamResponse.lastIndexOf('\\n\\ndata:')\n if (lastDataPosition === -1) {\n return streamResponse.trim()\n }\n \n return streamResponse.slice(lastDataPosition + 8).trim()\n}\n```\n\n### 5. Message Processing\n\nOnce data is extracted, it's parsed and processed:\n\n```javascript\nfunction processStreamChunk(chunk: string, responseIndex: number, sessionUuid: string) {\n const data = extractStreamingData(chunk)\n if (!data) return\n \n try {\n const parsedData = JSON.parse(data)\n \n // Validate structure\n if (!parsedData.choices?.[0]?.delta?.content || !parsedData.id) {\n console.warn('Invalid stream chunk structure')\n return\n }\n \n const content = parsedData.choices[0].delta.content\n const messageId = parsedData.id.replace('chatcmpl-', '')\n \n // Update chat with new content\n updateChat(sessionUuid, responseIndex, {\n uuid: messageId,\n text: content,\n // ... other properties\n })\n } catch (error) {\n console.error('Failed to parse stream chunk:', error)\n }\n}\n```\n\n## Error Handling\n\n### Stream Errors\n- HTTP errors (non-2xx responses)\n- Network connectivity issues\n- Reader errors\n\n### Parsing Errors\n- Malformed JSON in SSE data\n- Invalid message structure\n- Missing required fields\n\n### Recovery Strategies\n- Graceful degradation on parse errors\n- User notification for critical errors\n- Automatic cleanup of incomplete messages\n\n## Key Functions\n\n### `streamChatResponse()`\nMain streaming function for new chat messages:\n- Sets up fetch request with streaming enabled\n- Manages ReadableStream reader\n- Handles progressive response processing\n- Calls progress callbacks for real-time updates\n\n### `streamRegenerateResponse()`\nSpecialized streaming for message regeneration:\n- Similar to `streamChatResponse()` but with `regenerate: true`\n- Updates existing message instead of creating new one\n\n### `processStreamChunk()`\nCore message processing logic:\n- Extracts JSON data from SSE format\n- Validates message structure\n- Updates chat store with new content\n- Handles artifacts extraction\n\n### `extractStreamingData()`\nUtility for parsing SSE data:\n- Removes `data: ` prefix\n- Handles both single messages and multi-segment responses\n- Trims whitespace and normalizes output\n\n## Buffer Management Strategy\n\nThe buffering strategy handles the asynchronous nature of streaming:\n\n1. **Accumulate**: Add incoming chunks to buffer\n2. **Split**: Divide buffer by SSE delimiters (`\\n\\n`)\n3. **Process**: Handle complete messages\n4. **Retain**: Keep incomplete message for next iteration\n5. **Cleanup**: Process remaining buffer when stream ends\n\nThis ensures no message data is lost and all messages are processed in order.\n\n## Integration Points\n\n### Progress Callbacks\n```javascript\nonProgress?: (chunk: string, responseIndex: number) => void\n```\n- Called for each processed SSE message\n- Allows custom handling in different contexts\n- Used for real-time UI updates\n\n### Chat Store Integration\n- Updates stored chat messages\n- Manages message state (loading, error, complete)\n- Handles message artifacts and metadata\n\n### Error Notification\n- User-facing error messages\n- Developer console logging\n- Graceful fallback behaviors\n\n## Performance Considerations\n\n- **Memory**: Buffer size is automatically managed\n- **Processing**: Minimal parsing overhead per chunk\n- **Updates**: Batched UI updates prevent excessive re-renders\n- **Cleanup**: Proper reader resource management\n\n## Example SSE Flow\n\n```\n1. User sends message: \"Explain SSE\"\n\n2. Server responds with stream:\n data: {\"id\":\"msg-1\",\"choices\":[{\"delta\":{\"content\":\"Server-Sent\"}}]}\n \n data: {\"id\":\"msg-1\",\"choices\":[{\"delta\":{\"content\":\" Events\"}}]}\n \n data: {\"id\":\"msg-1\",\"choices\":[{\"delta\":{\"content\":\" allow...\"}}]}\n\n3. Each chunk updates the UI:\n \"Server-Sent\" → \"Server-Sent Events\" → \"Server-Sent Events allow...\"\n\n4. Stream ends, message is complete\n```\n\nThis architecture enables real-time chat experiences while maintaining reliability and error resilience." }, { "path": "docs/dev/vfs_integration_example.md", "content": "# VFS Integration Example\n\nThis document shows how to integrate the Virtual File System upload functionality into the existing chat interface.\n\n## Step 1: Add VFS Components to Chat\n\n### Import VFS Components\n\nAdd these imports to your `Conversation.vue` component:\n\n```typescript\n// Add to existing imports\nimport VFSIntegration from '@/components/VFSIntegration.vue'\n```\n\n### Add VFS Integration to Template\n\nAdd the VFS integration component near the upload button area. Here's how to modify the footer section:\n\n```vue\n\n```\n\n### Add Event Handlers\n\nAdd these methods to your component:\n\n```typescript\n// Add to script setup\nconst handleVFSFileUploaded = (fileInfo: any) => {\n console.log('File uploaded to VFS:', fileInfo)\n nui_msg.success(`File uploaded to VFS: ${fileInfo.filename}`)\n}\n\nconst handleSampleFilesCreated = () => {\n console.log('Sample files created in VFS')\n nui_msg.success('Sample files created! Try running the example scripts.')\n}\n```\n\n## Step 2: Complete Integration Example\n\nHere's a complete example of how to add VFS to an existing chat component:\n\n```vue\n\n\n\n\n\n```\n\n## Step 3: Alternative Minimal Integration\n\nIf you prefer a minimal integration, you can add just the upload button:\n\n```vue\n\n\n\n```\n\n## Step 4: User Experience Flow\n\nWith VFS integration, users can:\n\n1. **Upload Files**: Click \"Upload to VFS\" button\n2. **Select Directory**: Choose `/data`, `/workspace`, `/tmp`, or `/uploads`\n3. **Drag & Drop**: Upload multiple files at once\n4. **Use in Code**: Access files immediately in Python/JavaScript\n5. **Download Results**: Get processed files back\n\n### Example User Workflow:\n\n1. User uploads `sales.csv` to `/data/sales.csv`\n2. User runs Python code:\n ```python\n import pandas as pd\n df = pd.read_csv('/data/sales.csv')\n df['profit_margin'] = df['profit'] / df['revenue'] \n df.to_csv('/data/analyzed_sales.csv', index=False)\n ```\n3. User runs JavaScript code:\n ```javascript\n const fs = require('fs');\n const data = fs.readFileSync('/data/analyzed_sales.csv', 'utf8');\n console.log('Analysis complete, rows:', data.split('\\n').length);\n ```\n4. User downloads the processed file via the file manager\n\n## Benefits\n\n- **Seamless Integration**: Files work across Python and JavaScript\n- **No Server Storage**: Everything stays in browser memory\n- **Rich File Operations**: Full file system API support\n- **Data Processing Workflows**: Upload → Process → Download\n- **Session Management**: Save/restore entire file collections\n\nThis integration transforms the chat interface into a complete data processing environment where users can upload their data, process it with AI-generated code, and download the results." }, { "path": "docs/dev/virtual_file_system_plan.md", "content": "# Virtual File System (VFS) Implementation Plan\n\n## Overview\n\nThis document outlines the implementation plan for a Virtual File System (VFS) that will enable file I/O simulation and data manipulation capabilities in both JavaScript and Python code runners. The VFS will provide a secure, isolated environment for users to work with files and data without accessing the host system.\n\n## Architecture Goals\n\n- **Security**: Isolated from host file system with strict validation\n- **Performance**: Efficient in-memory storage with compression and caching\n- **Compatibility**: Standard file API compatibility for both Python and JavaScript\n- **Persistence**: Session-based storage with import/export capabilities\n- **Resource Management**: Configurable limits and quota enforcement\n\n## Phase 1: Core Architecture Design\n\n### VFS Class Structure\n```javascript\nclass VirtualFileSystem {\n constructor() {\n this.files = new Map() // file path -> file data\n this.directories = new Set() // directory paths\n this.metadata = new Map() // file metadata (size, modified, etc.)\n this.currentDirectory = '/' // current working directory\n this.maxFileSize = 10 * 1024 * 1024 // 10MB per file\n this.maxTotalSize = 100 * 1024 * 1024 // 100MB total\n this.fileHandlers = new Map() // extension -> handler\n this.permissions = new Map() // path -> permissions\n }\n}\n```\n\n### File System API Design\n```javascript\n// Core operations\nfs.writeFile(path, data, options) // Write file with encoding support\nfs.readFile(path, encoding) // Read file with optional encoding\nfs.mkdir(path, recursive) // Create directory\nfs.rmdir(path, recursive) // Remove directory\nfs.unlink(path) // Delete file\nfs.exists(path) // Check if path exists\nfs.stat(path) // Get file/directory metadata\nfs.readdir(path) // List directory contents\n\n// Navigation\nfs.chdir(path) // Change current directory\nfs.getcwd() // Get current working directory\n\n// Advanced operations\nfs.copy(src, dest) // Copy file or directory\nfs.move(src, dest) // Move/rename file or directory\nfs.glob(pattern) // Find files by pattern\nfs.watch(path, callback) // Watch for changes\n```\n\n### Path Resolution System\n```javascript\nclass PathResolver {\n normalize(path) // Handle ./ ../ ~ / normalization\n resolve(path) // Convert relative to absolute paths\n validate(path) // Security checks, prevent traversal attacks\n split(path) // Break path into components\n join(...parts) // Join path components safely\n dirname(path) // Get directory portion\n basename(path) // Get filename portion\n extname(path) // Get file extension\n}\n```\n\n## Phase 2: Core Implementation\n\n### File Storage Strategy\n- **Text Files**: UTF-8 strings with BOM detection\n- **Binary Files**: Base64 encoded with MIME type detection\n- **Large Files**: Chunked storage with LZ4 compression\n- **Memory Management**: LRU cache with configurable size limits\n- **Metadata Storage**: Created/modified timestamps, permissions, checksums\n\n### Security Model\n```javascript\nclass VFSSecurity {\n validatePath(path) // Prevent directory traversal\n checkQuota(size) // Enforce storage limits\n sanitizeFilename(name) // Remove dangerous characters\n validateFileType(data, ext) // Prevent type confusion attacks\n enforcePermissions(path, op) // Check read/write permissions\n scanContent(data) // Basic malware detection\n}\n```\n\n### Resource Management\n- Maximum file size: 10MB per file\n- Maximum total storage: 100MB per session\n- Maximum files: 1000 per session\n- Path length: 260 characters maximum\n- Filename restrictions: No control characters, reserved names\n\n## Phase 3: Python Runner Integration\n\n### Custom File Handlers\n```python\nimport io\nimport os\nimport builtins\nfrom pathlib import Path\n\nclass VFSFile(io.TextIOWrapper):\n \"\"\"File-like object that interfaces with VFS\"\"\"\n def __init__(self, path, mode, vfs_instance):\n self.vfs = vfs_instance\n self.path = path\n self.mode = mode\n self.position = 0\n self.closed = False\n \n def read(self, size=-1):\n return self.vfs.readFile(self.path, size, self.position)\n \n def write(self, data):\n return self.vfs.writeFile(self.path, data, self.mode)\n \n def seek(self, position):\n self.position = position\n \n def tell(self):\n return self.position\n```\n\n### Python Standard Library Integration\n```python\n# Override built-in open() function\noriginal_open = builtins.open\n\ndef vfs_open(file, mode='r', **kwargs):\n \"\"\"VFS-aware open() replacement\"\"\"\n if _is_vfs_path(file):\n return VFSFile(file, mode, global_vfs)\n return original_open(file, mode, **kwargs)\n\nbuiltins.open = vfs_open\n\n# Patch os module functions\nimport os\noriginal_os_functions = {}\n\ndef patch_os_module():\n \"\"\"Patch os module to work with VFS\"\"\"\n original_os_functions['listdir'] = os.listdir\n original_os_functions['makedirs'] = os.makedirs\n original_os_functions['path.exists'] = os.path.exists\n \n os.listdir = lambda path: global_vfs.readdir(path)\n os.makedirs = lambda path, **kwargs: global_vfs.mkdir(path, True)\n os.path.exists = lambda path: global_vfs.exists(path)\n \n # Patch pathlib.Path for modern Python code\n # Patch csv, json, pickle modules for data access\n```\n\n### Data Science Library Support\n```python\n# Pandas integration\nimport pandas as pd\noriginal_read_csv = pd.read_csv\noriginal_to_csv = pd.DataFrame.to_csv\n\npd.read_csv = lambda filepath, **kwargs: original_read_csv(\n global_vfs.readFile(filepath) if _is_vfs_path(filepath) else filepath, \n **kwargs\n)\n\n# NumPy, SciPy, Matplotlib file operations\n# Jupyter notebook file operations\n```\n\n## Phase 4: JavaScript Runner Integration\n\n### Node.js-style fs Module\n```javascript\nconst fs = {\n // Promise-based async versions\n readFile: async (path, options = {}) => {\n const encoding = options.encoding || options\n return await vfs.readFile(path, encoding)\n },\n \n writeFile: async (path, data, options = {}) => {\n return await vfs.writeFile(path, data, options)\n },\n \n mkdir: async (path, options = {}) => {\n return await vfs.mkdir(path, options.recursive)\n },\n \n readdir: async (path, options = {}) => {\n return await vfs.readdir(path, options)\n },\n \n // Synchronous versions\n readFileSync: (path, options = {}) => {\n const encoding = options.encoding || options\n return vfs.readFileSync(path, encoding)\n },\n \n writeFileSync: (path, data, options = {}) => {\n return vfs.writeFileSync(path, data, options)\n },\n \n mkdirSync: (path, options = {}) => {\n return vfs.mkdirSync(path, options.recursive)\n },\n \n readdirSync: (path, options = {}) => {\n return vfs.readdirSync(path, options)\n },\n \n // Stream support\n createReadStream: (path, options = {}) => {\n return new VFSReadableStream(path, options)\n },\n \n createWriteStream: (path, options = {}) => {\n return new VFSWritableStream(path, options)\n },\n \n // Additional utilities\n existsSync: (path) => vfs.exists(path),\n statSync: (path) => vfs.stat(path),\n unlinkSync: (path) => vfs.unlink(path),\n rmdirSync: (path, options = {}) => vfs.rmdir(path, options.recursive)\n}\n```\n\n### Stream Implementation\n```javascript\nclass VFSReadableStream extends ReadableStream {\n constructor(path, options = {}) {\n super({\n start(controller) {\n // Initialize stream with VFS file data\n },\n pull(controller) {\n // Read chunks from VFS\n },\n cancel() {\n // Cleanup\n }\n })\n }\n}\n\nclass VFSWritableStream extends WritableStream {\n constructor(path, options = {}) {\n super({\n write(chunk, controller) {\n // Write chunk to VFS\n },\n close() {\n // Finalize file in VFS\n },\n abort(reason) {\n // Cleanup on error\n }\n })\n }\n}\n```\n\n## Phase 5: Data Format Support\n\n### File Type Handlers\n```javascript\nconst FileHandlers = {\n // Text formats\n '.txt': new TextHandler(),\n '.csv': new CSVHandler(),\n '.json': new JSONHandler(),\n '.xml': new XMLHandler(),\n '.md': new MarkdownHandler(),\n '.yaml': new YAMLHandler(),\n '.toml': new TOMLHandler(),\n \n // Data formats \n '.xlsx': new ExcelHandler(),\n '.parquet': new ParquetHandler(),\n '.sqlite': new SQLiteHandler(),\n '.h5': new HDF5Handler(),\n \n // Binary formats\n '.png': new ImageHandler(),\n '.jpg': new ImageHandler(),\n '.gif': new ImageHandler(),\n '.pdf': new PDFHandler(),\n '.zip': new ZipHandler(),\n '.tar': new TarHandler(),\n \n // Programming languages\n '.py': new PythonHandler(),\n '.js': new JavaScriptHandler(),\n '.html': new HTMLHandler(),\n '.css': new CSSHandler()\n}\n\nclass CSVHandler {\n async parse(data, options = {}) {\n // Parse CSV with configurable delimiter, headers, etc.\n const delimiter = options.delimiter || ','\n const hasHeaders = options.headers !== false\n // Return structured data\n }\n \n async stringify(data, options = {}) {\n // Convert structured data to CSV string\n }\n}\n\nclass JSONHandler {\n async parse(data) {\n return JSON.parse(data)\n }\n \n async stringify(data, options = {}) {\n const indent = options.indent || 2\n return JSON.stringify(data, null, indent)\n }\n}\n```\n\n### Data Import/Export System\n```javascript\nclass DataImporter {\n async importFromURL(url, path, options = {}) {\n // Fetch remote file and store in VFS\n const response = await fetch(url)\n const data = await response.arrayBuffer()\n return await vfs.writeFile(path, data, { binary: true })\n }\n \n async importFromFile(file, path) {\n // Handle browser File API uploads\n const reader = new FileReader()\n return new Promise((resolve, reject) => {\n reader.onload = async (e) => {\n await vfs.writeFile(path, e.target.result)\n resolve(path)\n }\n reader.onerror = reject\n reader.readAsArrayBuffer(file)\n })\n }\n \n async importCSV(csvText, path, options = {}) {\n const handler = new CSVHandler()\n const data = await handler.parse(csvText, options)\n await vfs.writeFile(path, JSON.stringify(data))\n return data\n }\n \n async importJSON(jsonText, path) {\n const data = JSON.parse(jsonText)\n await vfs.writeFile(path, jsonText)\n return data\n }\n \n // Export functions\n async exportToDownload(path, filename) {\n const data = await vfs.readFile(path, 'binary')\n const blob = new Blob([data])\n const url = URL.createObjectURL(blob)\n \n const a = document.createElement('a')\n a.href = url\n a.download = filename\n a.click()\n \n URL.revokeObjectURL(url)\n }\n \n async exportToZip(paths, zipName) {\n // Create ZIP file containing multiple VFS files\n const JSZip = await import('jszip')\n const zip = new JSZip()\n \n for (const path of paths) {\n const data = await vfs.readFile(path, 'binary')\n const relativePath = path.startsWith('/') ? path.slice(1) : path\n zip.file(relativePath, data)\n }\n \n const zipBlob = await zip.generateAsync({ type: 'blob' })\n this.downloadBlob(zipBlob, zipName)\n }\n \n async exportToDataURL(path) {\n const data = await vfs.readFile(path, 'binary')\n const mimeType = this.detectMimeType(path)\n return `data:${mimeType};base64,${btoa(data)}`\n }\n}\n```\n\n## Phase 6: Advanced Features\n\n### File System Utilities\n```javascript\nclass FileSystemUtils {\n // Search and filtering\n async find(pattern, options = {}) {\n // Find files by glob pattern\n const recursive = options.recursive !== false\n const maxDepth = options.maxDepth || 100\n const caseInsensitive = options.caseInsensitive || false\n \n return vfs.glob(pattern, { recursive, maxDepth, caseInsensitive })\n }\n \n async grep(pattern, files, options = {}) {\n // Search within files for text patterns\n const results = []\n const regex = new RegExp(pattern, options.flags || 'gi')\n \n for (const file of files) {\n const content = await vfs.readFile(file, 'utf8')\n const matches = [...content.matchAll(regex)]\n if (matches.length > 0) {\n results.push({ file, matches })\n }\n }\n \n return results\n }\n \n // File operations\n async compress(path, algorithm = 'gzip') {\n const data = await vfs.readFile(path, 'binary')\n const compressed = await this.compressData(data, algorithm)\n const compressedPath = `${path}.${algorithm}`\n await vfs.writeFile(compressedPath, compressed, { binary: true })\n return compressedPath\n }\n \n async decompress(path) {\n const data = await vfs.readFile(path, 'binary')\n const algorithm = this.detectCompressionType(path)\n const decompressed = await this.decompressData(data, algorithm)\n const originalPath = path.replace(new RegExp(`\\.${algorithm}$`), '')\n await vfs.writeFile(originalPath, decompressed, { binary: true })\n return originalPath\n }\n \n async checksum(path, algorithm = 'sha256') {\n const data = await vfs.readFile(path, 'binary')\n const hash = await crypto.subtle.digest(algorithm.toUpperCase(), data)\n return Array.from(new Uint8Array(hash))\n .map(b => b.toString(16).padStart(2, '0'))\n .join('')\n }\n \n // Batch operations\n async bulkCopy(srcPattern, destDir, options = {}) {\n const files = await this.find(srcPattern)\n const results = []\n \n for (const file of files) {\n const basename = vfs.path.basename(file)\n const destPath = vfs.path.join(destDir, basename)\n await vfs.copy(file, destPath)\n results.push({ src: file, dest: destPath })\n }\n \n return results\n }\n \n async bulkDelete(pattern, options = {}) {\n const files = await this.find(pattern)\n const confirm = options.confirm !== false\n \n if (confirm && files.length > 10) {\n // Safety check for bulk deletion\n throw new Error(`Bulk delete would affect ${files.length} files. Use {confirm: false} to proceed.`)\n }\n \n for (const file of files) {\n await vfs.unlink(file)\n }\n \n return files\n }\n \n async bulkRename(pattern, replacement, options = {}) {\n const files = await this.find(pattern)\n const results = []\n \n for (const file of files) {\n const newName = file.replace(new RegExp(pattern), replacement)\n await vfs.move(file, newName)\n results.push({ old: file, new: newName })\n }\n \n return results\n }\n}\n```\n\n### Session Persistence\n```javascript\nclass VFSPersistence {\n constructor(vfs) {\n this.vfs = vfs\n this.storageKey = 'vfs_session'\n this.autoSaveInterval = null\n }\n \n async saveSession(name = 'default') {\n // Serialize VFS state to JSON\n const sessionData = {\n version: '1.0',\n timestamp: new Date().toISOString(),\n files: Object.fromEntries(this.vfs.files),\n directories: Array.from(this.vfs.directories),\n metadata: Object.fromEntries(this.vfs.metadata),\n currentDirectory: this.vfs.currentDirectory\n }\n \n // Compress session data\n const compressed = await this.compressSession(sessionData)\n \n // Store in IndexedDB for persistence\n await this.storeInIndexedDB(name, compressed)\n \n return { name, size: compressed.length, timestamp: sessionData.timestamp }\n }\n \n async loadSession(name = 'default') {\n // Load from IndexedDB\n const compressed = await this.loadFromIndexedDB(name)\n if (!compressed) {\n throw new Error(`Session '${name}' not found`)\n }\n \n // Decompress and parse\n const sessionData = await this.decompressSession(compressed)\n \n // Restore VFS state\n this.vfs.files = new Map(Object.entries(sessionData.files))\n this.vfs.directories = new Set(sessionData.directories)\n this.vfs.metadata = new Map(Object.entries(sessionData.metadata))\n this.vfs.currentDirectory = sessionData.currentDirectory\n \n return sessionData\n }\n \n async exportSession(name = 'default') {\n // Export session as downloadable ZIP\n const sessionData = await this.saveSession(name)\n const blob = new Blob([JSON.stringify(sessionData)], { \n type: 'application/json' \n })\n \n const filename = `vfs_session_${name}_${new Date().toISOString().slice(0, 19)}.json`\n this.downloadBlob(blob, filename)\n \n return filename\n }\n \n async importSession(file) {\n // Import session from uploaded file\n const text = await file.text()\n const sessionData = JSON.parse(text)\n \n // Validate session data structure\n this.validateSessionData(sessionData)\n \n // Restore VFS state\n await this.restoreSessionData(sessionData)\n \n return sessionData\n }\n \n enableAutoSave(interval = 30000) {\n // Automatically save session every 30 seconds\n if (this.autoSaveInterval) {\n clearInterval(this.autoSaveInterval)\n }\n \n this.autoSaveInterval = setInterval(async () => {\n try {\n await this.saveSession('autosave')\n } catch (error) {\n console.warn('Auto-save failed:', error)\n }\n }, interval)\n }\n \n disableAutoSave() {\n if (this.autoSaveInterval) {\n clearInterval(this.autoSaveInterval)\n this.autoSaveInterval = null\n }\n }\n \n async listSessions() {\n // List all saved sessions\n const db = await this.openIndexedDB()\n const transaction = db.transaction(['sessions'], 'readonly')\n const store = transaction.objectStore('sessions')\n const keys = await store.getAllKeys()\n \n const sessions = []\n for (const key of keys) {\n const session = await store.get(key)\n sessions.push({\n name: key,\n timestamp: session.timestamp,\n size: session.size\n })\n }\n \n return sessions\n }\n \n async deleteSession(name) {\n // Delete a saved session\n const db = await this.openIndexedDB()\n const transaction = db.transaction(['sessions'], 'readwrite')\n const store = transaction.objectStore('sessions')\n await store.delete(name)\n }\n}\n```\n\n## Phase 7: Security & Validation\n\n### Security Implementation\n```javascript\nclass VFSSecurity {\n constructor() {\n this.maxPathLength = 260\n this.maxFilenameLength = 255\n this.forbiddenChars = /[<>:\"|?*\\x00-\\x1f]/\n this.reservedNames = ['CON', 'PRN', 'AUX', 'NUL', 'COM1', 'COM2', 'COM3', 'COM4', 'COM5', 'COM6', 'COM7', 'COM8', 'COM9', 'LPT1', 'LPT2', 'LPT3', 'LPT4', 'LPT5', 'LPT6', 'LPT7', 'LPT8', 'LPT9']\n }\n \n validatePath(path) {\n // Prevent directory traversal attacks\n if (path.includes('..')) {\n throw new Error('Path traversal detected')\n }\n \n if (path.length > this.maxPathLength) {\n throw new Error(`Path too long: ${path.length} > ${this.maxPathLength}`)\n }\n \n // Normalize path separators\n const normalizedPath = path.replace(/\\\\/g, '/')\n \n // Check for dangerous patterns\n if (normalizedPath.match(/\\/\\.{2,}\\//)) {\n throw new Error('Invalid path pattern detected')\n }\n \n return normalizedPath\n }\n \n checkQuota(currentSize, additionalSize) {\n // Enforce storage limits\n const totalSize = currentSize + additionalSize\n \n if (additionalSize > this.maxFileSize) {\n throw new Error(`File too large: ${additionalSize} > ${this.maxFileSize}`)\n }\n \n if (totalSize > this.maxTotalSize) {\n throw new Error(`Storage quota exceeded: ${totalSize} > ${this.maxTotalSize}`)\n }\n \n return true\n }\n \n sanitizeFilename(name) {\n // Remove dangerous characters from filenames\n let sanitized = name.replace(this.forbiddenChars, '_')\n \n // Check against reserved names\n const baseName = sanitized.split('.')[0].toUpperCase()\n if (this.reservedNames.includes(baseName)) {\n sanitized = `_${sanitized}`\n }\n \n // Ensure filename isn't too long\n if (sanitized.length > this.maxFilenameLength) {\n const ext = sanitized.split('.').pop()\n const base = sanitized.slice(0, this.maxFilenameLength - ext.length - 1)\n sanitized = `${base}.${ext}`\n }\n \n return sanitized\n }\n \n validateFileType(data, extension) {\n // Prevent type confusion attacks\n const detectedType = this.detectFileType(data)\n const expectedType = this.getExpectedType(extension)\n \n if (detectedType && expectedType && detectedType !== expectedType) {\n console.warn(`File type mismatch: expected ${expectedType}, detected ${detectedType}`)\n }\n \n // Check for executable content\n if (this.containsExecutableContent(data)) {\n throw new Error('Executable content detected')\n }\n \n return true\n }\n \n detectFileType(data) {\n // Basic file type detection by magic bytes\n const bytes = new Uint8Array(data.slice(0, 16))\n \n // PNG\n if (bytes[0] === 0x89 && bytes[1] === 0x50 && bytes[2] === 0x4E && bytes[3] === 0x47) {\n return 'png'\n }\n \n // JPEG\n if (bytes[0] === 0xFF && bytes[1] === 0xD8 && bytes[2] === 0xFF) {\n return 'jpeg'\n }\n \n // PDF\n if (bytes[0] === 0x25 && bytes[1] === 0x50 && bytes[2] === 0x44 && bytes[3] === 0x46) {\n return 'pdf'\n }\n \n // ZIP\n if (bytes[0] === 0x50 && bytes[1] === 0x4B) {\n return 'zip'\n }\n \n return null\n }\n \n containsExecutableContent(data) {\n // Basic check for executable content patterns\n const text = typeof data === 'string' ? data : new TextDecoder().decode(data)\n \n // Check for common script patterns\n const dangerousPatterns = [\n /\n\n\n" }, { "path": "mobile/web/manifest.json", "content": "{\n \"name\": \"chat_mobile\",\n \"short_name\": \"chat_mobile\",\n \"start_url\": \".\",\n \"display\": \"standalone\",\n \"background_color\": \"#0175C2\",\n \"theme_color\": \"#0175C2\",\n \"description\": \"A new Flutter project.\",\n \"orientation\": \"portrait-primary\",\n \"prefer_related_applications\": false,\n \"icons\": [\n {\n \"src\": \"icons/Icon-192.png\",\n \"sizes\": \"192x192\",\n \"type\": \"image/png\"\n },\n {\n \"src\": \"icons/Icon-512.png\",\n \"sizes\": \"512x512\",\n \"type\": \"image/png\"\n },\n {\n \"src\": \"icons/Icon-maskable-192.png\",\n \"sizes\": \"192x192\",\n \"type\": \"image/png\",\n \"purpose\": \"maskable\"\n },\n {\n \"src\": \"icons/Icon-maskable-512.png\",\n \"sizes\": \"512x512\",\n \"type\": \"image/png\",\n \"purpose\": \"maskable\"\n }\n ]\n}\n" }, { "path": "mobile/windows/.gitignore", "content": "flutter/ephemeral/\n\n# Visual Studio user-specific files.\n*.suo\n*.user\n*.userosscache\n*.sln.docstates\n\n# Visual Studio build-related files.\nx64/\nx86/\n\n# Visual Studio cache files\n# files ending in .cache can be ignored\n*.[Cc]ache\n# but keep track of directories ending in .cache\n!*.[Cc]ache/\n" }, { "path": "mobile/windows/CMakeLists.txt", "content": "# Project-level configuration.\ncmake_minimum_required(VERSION 3.14)\nproject(chat_mobile LANGUAGES CXX)\n\n# The name of the executable created for the application. Change this to change\n# the on-disk name of your application.\nset(BINARY_NAME \"chat_mobile\")\n\n# Explicitly opt in to modern CMake behaviors to avoid warnings with recent\n# versions of CMake.\ncmake_policy(VERSION 3.14...3.25)\n\n# Define build configuration option.\nget_property(IS_MULTICONFIG GLOBAL PROPERTY GENERATOR_IS_MULTI_CONFIG)\nif(IS_MULTICONFIG)\n set(CMAKE_CONFIGURATION_TYPES \"Debug;Profile;Release\"\n CACHE STRING \"\" FORCE)\nelse()\n if(NOT CMAKE_BUILD_TYPE AND NOT CMAKE_CONFIGURATION_TYPES)\n set(CMAKE_BUILD_TYPE \"Debug\" CACHE\n STRING \"Flutter build mode\" FORCE)\n set_property(CACHE CMAKE_BUILD_TYPE PROPERTY STRINGS\n \"Debug\" \"Profile\" \"Release\")\n endif()\nendif()\n# Define settings for the Profile build mode.\nset(CMAKE_EXE_LINKER_FLAGS_PROFILE \"${CMAKE_EXE_LINKER_FLAGS_RELEASE}\")\nset(CMAKE_SHARED_LINKER_FLAGS_PROFILE \"${CMAKE_SHARED_LINKER_FLAGS_RELEASE}\")\nset(CMAKE_C_FLAGS_PROFILE \"${CMAKE_C_FLAGS_RELEASE}\")\nset(CMAKE_CXX_FLAGS_PROFILE \"${CMAKE_CXX_FLAGS_RELEASE}\")\n\n# Use Unicode for all projects.\nadd_definitions(-DUNICODE -D_UNICODE)\n\n# Compilation settings that should be applied to most targets.\n#\n# Be cautious about adding new options here, as plugins use this function by\n# default. In most cases, you should add new options to specific targets instead\n# of modifying this function.\nfunction(APPLY_STANDARD_SETTINGS TARGET)\n target_compile_features(${TARGET} PUBLIC cxx_std_17)\n target_compile_options(${TARGET} PRIVATE /W4 /WX /wd\"4100\")\n target_compile_options(${TARGET} PRIVATE /EHsc)\n target_compile_definitions(${TARGET} PRIVATE \"_HAS_EXCEPTIONS=0\")\n target_compile_definitions(${TARGET} PRIVATE \"$<$:_DEBUG>\")\nendfunction()\n\n# Flutter library and tool build rules.\nset(FLUTTER_MANAGED_DIR \"${CMAKE_CURRENT_SOURCE_DIR}/flutter\")\nadd_subdirectory(${FLUTTER_MANAGED_DIR})\n\n# Application build; see runner/CMakeLists.txt.\nadd_subdirectory(\"runner\")\n\n\n# Generated plugin build rules, which manage building the plugins and adding\n# them to the application.\ninclude(flutter/generated_plugins.cmake)\n\n\n# === Installation ===\n# Support files are copied into place next to the executable, so that it can\n# run in place. This is done instead of making a separate bundle (as on Linux)\n# so that building and running from within Visual Studio will work.\nset(BUILD_BUNDLE_DIR \"$\")\n# Make the \"install\" step default, as it's required to run.\nset(CMAKE_VS_INCLUDE_INSTALL_TO_DEFAULT_BUILD 1)\nif(CMAKE_INSTALL_PREFIX_INITIALIZED_TO_DEFAULT)\n set(CMAKE_INSTALL_PREFIX \"${BUILD_BUNDLE_DIR}\" CACHE PATH \"...\" FORCE)\nendif()\n\nset(INSTALL_BUNDLE_DATA_DIR \"${CMAKE_INSTALL_PREFIX}/data\")\nset(INSTALL_BUNDLE_LIB_DIR \"${CMAKE_INSTALL_PREFIX}\")\n\ninstall(TARGETS ${BINARY_NAME} RUNTIME DESTINATION \"${CMAKE_INSTALL_PREFIX}\"\n COMPONENT Runtime)\n\ninstall(FILES \"${FLUTTER_ICU_DATA_FILE}\" DESTINATION \"${INSTALL_BUNDLE_DATA_DIR}\"\n COMPONENT Runtime)\n\ninstall(FILES \"${FLUTTER_LIBRARY}\" DESTINATION \"${INSTALL_BUNDLE_LIB_DIR}\"\n COMPONENT Runtime)\n\nif(PLUGIN_BUNDLED_LIBRARIES)\n install(FILES \"${PLUGIN_BUNDLED_LIBRARIES}\"\n DESTINATION \"${INSTALL_BUNDLE_LIB_DIR}\"\n COMPONENT Runtime)\nendif()\n\n# Copy the native assets provided by the build.dart from all packages.\nset(NATIVE_ASSETS_DIR \"${PROJECT_BUILD_DIR}native_assets/windows/\")\ninstall(DIRECTORY \"${NATIVE_ASSETS_DIR}\"\n DESTINATION \"${INSTALL_BUNDLE_LIB_DIR}\"\n COMPONENT Runtime)\n\n# Fully re-copy the assets directory on each build to avoid having stale files\n# from a previous install.\nset(FLUTTER_ASSET_DIR_NAME \"flutter_assets\")\ninstall(CODE \"\n file(REMOVE_RECURSE \\\"${INSTALL_BUNDLE_DATA_DIR}/${FLUTTER_ASSET_DIR_NAME}\\\")\n \" COMPONENT Runtime)\ninstall(DIRECTORY \"${PROJECT_BUILD_DIR}/${FLUTTER_ASSET_DIR_NAME}\"\n DESTINATION \"${INSTALL_BUNDLE_DATA_DIR}\" COMPONENT Runtime)\n\n# Install the AOT library on non-Debug builds only.\ninstall(FILES \"${AOT_LIBRARY}\" DESTINATION \"${INSTALL_BUNDLE_DATA_DIR}\"\n CONFIGURATIONS Profile;Release\n COMPONENT Runtime)\n" }, { "path": "mobile/windows/flutter/CMakeLists.txt", "content": "# This file controls Flutter-level build steps. It should not be edited.\ncmake_minimum_required(VERSION 3.14)\n\nset(EPHEMERAL_DIR \"${CMAKE_CURRENT_SOURCE_DIR}/ephemeral\")\n\n# Configuration provided via flutter tool.\ninclude(${EPHEMERAL_DIR}/generated_config.cmake)\n\n# TODO: Move the rest of this into files in ephemeral. See\n# https://github.com/flutter/flutter/issues/57146.\nset(WRAPPER_ROOT \"${EPHEMERAL_DIR}/cpp_client_wrapper\")\n\n# Set fallback configurations for older versions of the flutter tool.\nif (NOT DEFINED FLUTTER_TARGET_PLATFORM)\n set(FLUTTER_TARGET_PLATFORM \"windows-x64\")\nendif()\n\n# === Flutter Library ===\nset(FLUTTER_LIBRARY \"${EPHEMERAL_DIR}/flutter_windows.dll\")\n\n# Published to parent scope for install step.\nset(FLUTTER_LIBRARY ${FLUTTER_LIBRARY} PARENT_SCOPE)\nset(FLUTTER_ICU_DATA_FILE \"${EPHEMERAL_DIR}/icudtl.dat\" PARENT_SCOPE)\nset(PROJECT_BUILD_DIR \"${PROJECT_DIR}/build/\" PARENT_SCOPE)\nset(AOT_LIBRARY \"${PROJECT_DIR}/build/windows/app.so\" PARENT_SCOPE)\n\nlist(APPEND FLUTTER_LIBRARY_HEADERS\n \"flutter_export.h\"\n \"flutter_windows.h\"\n \"flutter_messenger.h\"\n \"flutter_plugin_registrar.h\"\n \"flutter_texture_registrar.h\"\n)\nlist(TRANSFORM FLUTTER_LIBRARY_HEADERS PREPEND \"${EPHEMERAL_DIR}/\")\nadd_library(flutter INTERFACE)\ntarget_include_directories(flutter INTERFACE\n \"${EPHEMERAL_DIR}\"\n)\ntarget_link_libraries(flutter INTERFACE \"${FLUTTER_LIBRARY}.lib\")\nadd_dependencies(flutter flutter_assemble)\n\n# === Wrapper ===\nlist(APPEND CPP_WRAPPER_SOURCES_CORE\n \"core_implementations.cc\"\n \"standard_codec.cc\"\n)\nlist(TRANSFORM CPP_WRAPPER_SOURCES_CORE PREPEND \"${WRAPPER_ROOT}/\")\nlist(APPEND CPP_WRAPPER_SOURCES_PLUGIN\n \"plugin_registrar.cc\"\n)\nlist(TRANSFORM CPP_WRAPPER_SOURCES_PLUGIN PREPEND \"${WRAPPER_ROOT}/\")\nlist(APPEND CPP_WRAPPER_SOURCES_APP\n \"flutter_engine.cc\"\n \"flutter_view_controller.cc\"\n)\nlist(TRANSFORM CPP_WRAPPER_SOURCES_APP PREPEND \"${WRAPPER_ROOT}/\")\n\n# Wrapper sources needed for a plugin.\nadd_library(flutter_wrapper_plugin STATIC\n ${CPP_WRAPPER_SOURCES_CORE}\n ${CPP_WRAPPER_SOURCES_PLUGIN}\n)\napply_standard_settings(flutter_wrapper_plugin)\nset_target_properties(flutter_wrapper_plugin PROPERTIES\n POSITION_INDEPENDENT_CODE ON)\nset_target_properties(flutter_wrapper_plugin PROPERTIES\n CXX_VISIBILITY_PRESET hidden)\ntarget_link_libraries(flutter_wrapper_plugin PUBLIC flutter)\ntarget_include_directories(flutter_wrapper_plugin PUBLIC\n \"${WRAPPER_ROOT}/include\"\n)\nadd_dependencies(flutter_wrapper_plugin flutter_assemble)\n\n# Wrapper sources needed for the runner.\nadd_library(flutter_wrapper_app STATIC\n ${CPP_WRAPPER_SOURCES_CORE}\n ${CPP_WRAPPER_SOURCES_APP}\n)\napply_standard_settings(flutter_wrapper_app)\ntarget_link_libraries(flutter_wrapper_app PUBLIC flutter)\ntarget_include_directories(flutter_wrapper_app PUBLIC\n \"${WRAPPER_ROOT}/include\"\n)\nadd_dependencies(flutter_wrapper_app flutter_assemble)\n\n# === Flutter tool backend ===\n# _phony_ is a non-existent file to force this command to run every time,\n# since currently there's no way to get a full input/output list from the\n# flutter tool.\nset(PHONY_OUTPUT \"${CMAKE_CURRENT_BINARY_DIR}/_phony_\")\nset_source_files_properties(\"${PHONY_OUTPUT}\" PROPERTIES SYMBOLIC TRUE)\nadd_custom_command(\n OUTPUT ${FLUTTER_LIBRARY} ${FLUTTER_LIBRARY_HEADERS}\n ${CPP_WRAPPER_SOURCES_CORE} ${CPP_WRAPPER_SOURCES_PLUGIN}\n ${CPP_WRAPPER_SOURCES_APP}\n ${PHONY_OUTPUT}\n COMMAND ${CMAKE_COMMAND} -E env\n ${FLUTTER_TOOL_ENVIRONMENT}\n \"${FLUTTER_ROOT}/packages/flutter_tools/bin/tool_backend.bat\"\n ${FLUTTER_TARGET_PLATFORM} $\n VERBATIM\n)\nadd_custom_target(flutter_assemble DEPENDS\n \"${FLUTTER_LIBRARY}\"\n ${FLUTTER_LIBRARY_HEADERS}\n ${CPP_WRAPPER_SOURCES_CORE}\n ${CPP_WRAPPER_SOURCES_PLUGIN}\n ${CPP_WRAPPER_SOURCES_APP}\n)\n" }, { "path": "mobile/windows/flutter/generated_plugin_registrant.cc", "content": "//\n// Generated file. Do not edit.\n//\n\n// clang-format off\n\n#include \"generated_plugin_registrant.h\"\n\n\nvoid RegisterPlugins(flutter::PluginRegistry* registry) {\n}\n" }, { "path": "mobile/windows/flutter/generated_plugin_registrant.h", "content": "//\n// Generated file. Do not edit.\n//\n\n// clang-format off\n\n#ifndef GENERATED_PLUGIN_REGISTRANT_\n#define GENERATED_PLUGIN_REGISTRANT_\n\n#include \n\n// Registers Flutter plugins.\nvoid RegisterPlugins(flutter::PluginRegistry* registry);\n\n#endif // GENERATED_PLUGIN_REGISTRANT_\n" }, { "path": "mobile/windows/flutter/generated_plugins.cmake", "content": "#\n# Generated file, do not edit.\n#\n\nlist(APPEND FLUTTER_PLUGIN_LIST\n)\n\nlist(APPEND FLUTTER_FFI_PLUGIN_LIST\n)\n\nset(PLUGIN_BUNDLED_LIBRARIES)\n\nforeach(plugin ${FLUTTER_PLUGIN_LIST})\n add_subdirectory(flutter/ephemeral/.plugin_symlinks/${plugin}/windows plugins/${plugin})\n target_link_libraries(${BINARY_NAME} PRIVATE ${plugin}_plugin)\n list(APPEND PLUGIN_BUNDLED_LIBRARIES $)\n list(APPEND PLUGIN_BUNDLED_LIBRARIES ${${plugin}_bundled_libraries})\nendforeach(plugin)\n\nforeach(ffi_plugin ${FLUTTER_FFI_PLUGIN_LIST})\n add_subdirectory(flutter/ephemeral/.plugin_symlinks/${ffi_plugin}/windows plugins/${ffi_plugin})\n list(APPEND PLUGIN_BUNDLED_LIBRARIES ${${ffi_plugin}_bundled_libraries})\nendforeach(ffi_plugin)\n" }, { "path": "mobile/windows/runner/CMakeLists.txt", "content": "cmake_minimum_required(VERSION 3.14)\nproject(runner LANGUAGES CXX)\n\n# Define the application target. To change its name, change BINARY_NAME in the\n# top-level CMakeLists.txt, not the value here, or `flutter run` will no longer\n# work.\n#\n# Any new source files that you add to the application should be added here.\nadd_executable(${BINARY_NAME} WIN32\n \"flutter_window.cpp\"\n \"main.cpp\"\n \"utils.cpp\"\n \"win32_window.cpp\"\n \"${FLUTTER_MANAGED_DIR}/generated_plugin_registrant.cc\"\n \"Runner.rc\"\n \"runner.exe.manifest\"\n)\n\n# Apply the standard set of build settings. This can be removed for applications\n# that need different build settings.\napply_standard_settings(${BINARY_NAME})\n\n# Add preprocessor definitions for the build version.\ntarget_compile_definitions(${BINARY_NAME} PRIVATE \"FLUTTER_VERSION=\\\"${FLUTTER_VERSION}\\\"\")\ntarget_compile_definitions(${BINARY_NAME} PRIVATE \"FLUTTER_VERSION_MAJOR=${FLUTTER_VERSION_MAJOR}\")\ntarget_compile_definitions(${BINARY_NAME} PRIVATE \"FLUTTER_VERSION_MINOR=${FLUTTER_VERSION_MINOR}\")\ntarget_compile_definitions(${BINARY_NAME} PRIVATE \"FLUTTER_VERSION_PATCH=${FLUTTER_VERSION_PATCH}\")\ntarget_compile_definitions(${BINARY_NAME} PRIVATE \"FLUTTER_VERSION_BUILD=${FLUTTER_VERSION_BUILD}\")\n\n# Disable Windows macros that collide with C++ standard library functions.\ntarget_compile_definitions(${BINARY_NAME} PRIVATE \"NOMINMAX\")\n\n# Add dependency libraries and include directories. Add any application-specific\n# dependencies here.\ntarget_link_libraries(${BINARY_NAME} PRIVATE flutter flutter_wrapper_app)\ntarget_link_libraries(${BINARY_NAME} PRIVATE \"dwmapi.lib\")\ntarget_include_directories(${BINARY_NAME} PRIVATE \"${CMAKE_SOURCE_DIR}\")\n\n# Run the Flutter tool portions of the build. This must not be removed.\nadd_dependencies(${BINARY_NAME} flutter_assemble)\n" }, { "path": "mobile/windows/runner/Runner.rc", "content": "// Microsoft Visual C++ generated resource script.\n//\n#pragma code_page(65001)\n#include \"resource.h\"\n\n#define APSTUDIO_READONLY_SYMBOLS\n/////////////////////////////////////////////////////////////////////////////\n//\n// Generated from the TEXTINCLUDE 2 resource.\n//\n#include \"winres.h\"\n\n/////////////////////////////////////////////////////////////////////////////\n#undef APSTUDIO_READONLY_SYMBOLS\n\n/////////////////////////////////////////////////////////////////////////////\n// English (United States) resources\n\n#if !defined(AFX_RESOURCE_DLL) || defined(AFX_TARG_ENU)\nLANGUAGE LANG_ENGLISH, SUBLANG_ENGLISH_US\n\n#ifdef APSTUDIO_INVOKED\n/////////////////////////////////////////////////////////////////////////////\n//\n// TEXTINCLUDE\n//\n\n1 TEXTINCLUDE\nBEGIN\n \"resource.h\\0\"\nEND\n\n2 TEXTINCLUDE\nBEGIN\n \"#include \"\"winres.h\"\"\\r\\n\"\n \"\\0\"\nEND\n\n3 TEXTINCLUDE\nBEGIN\n \"\\r\\n\"\n \"\\0\"\nEND\n\n#endif // APSTUDIO_INVOKED\n\n\n/////////////////////////////////////////////////////////////////////////////\n//\n// Icon\n//\n\n// Icon with lowest ID value placed first to ensure application icon\n// remains consistent on all systems.\nIDI_APP_ICON ICON \"resources\\\\app_icon.ico\"\n\n\n/////////////////////////////////////////////////////////////////////////////\n//\n// Version\n//\n\n#if defined(FLUTTER_VERSION_MAJOR) && defined(FLUTTER_VERSION_MINOR) && defined(FLUTTER_VERSION_PATCH) && defined(FLUTTER_VERSION_BUILD)\n#define VERSION_AS_NUMBER FLUTTER_VERSION_MAJOR,FLUTTER_VERSION_MINOR,FLUTTER_VERSION_PATCH,FLUTTER_VERSION_BUILD\n#else\n#define VERSION_AS_NUMBER 1,0,0,0\n#endif\n\n#if defined(FLUTTER_VERSION)\n#define VERSION_AS_STRING FLUTTER_VERSION\n#else\n#define VERSION_AS_STRING \"1.0.0\"\n#endif\n\nVS_VERSION_INFO VERSIONINFO\n FILEVERSION VERSION_AS_NUMBER\n PRODUCTVERSION VERSION_AS_NUMBER\n FILEFLAGSMASK VS_FFI_FILEFLAGSMASK\n#ifdef _DEBUG\n FILEFLAGS VS_FF_DEBUG\n#else\n FILEFLAGS 0x0L\n#endif\n FILEOS VOS__WINDOWS32\n FILETYPE VFT_APP\n FILESUBTYPE 0x0L\nBEGIN\n BLOCK \"StringFileInfo\"\n BEGIN\n BLOCK \"040904e4\"\n BEGIN\n VALUE \"CompanyName\", \"com.example\" \"\\0\"\n VALUE \"FileDescription\", \"chat_mobile\" \"\\0\"\n VALUE \"FileVersion\", VERSION_AS_STRING \"\\0\"\n VALUE \"InternalName\", \"chat_mobile\" \"\\0\"\n VALUE \"LegalCopyright\", \"Copyright (C) 2026 com.example. All rights reserved.\" \"\\0\"\n VALUE \"OriginalFilename\", \"chat_mobile.exe\" \"\\0\"\n VALUE \"ProductName\", \"chat_mobile\" \"\\0\"\n VALUE \"ProductVersion\", VERSION_AS_STRING \"\\0\"\n END\n END\n BLOCK \"VarFileInfo\"\n BEGIN\n VALUE \"Translation\", 0x409, 1252\n END\nEND\n\n#endif // English (United States) resources\n/////////////////////////////////////////////////////////////////////////////\n\n\n\n#ifndef APSTUDIO_INVOKED\n/////////////////////////////////////////////////////////////////////////////\n//\n// Generated from the TEXTINCLUDE 3 resource.\n//\n\n\n/////////////////////////////////////////////////////////////////////////////\n#endif // not APSTUDIO_INVOKED\n" }, { "path": "mobile/windows/runner/flutter_window.cpp", "content": "#include \"flutter_window.h\"\n\n#include \n\n#include \"flutter/generated_plugin_registrant.h\"\n\nFlutterWindow::FlutterWindow(const flutter::DartProject& project)\n : project_(project) {}\n\nFlutterWindow::~FlutterWindow() {}\n\nbool FlutterWindow::OnCreate() {\n if (!Win32Window::OnCreate()) {\n return false;\n }\n\n RECT frame = GetClientArea();\n\n // The size here must match the window dimensions to avoid unnecessary surface\n // creation / destruction in the startup path.\n flutter_controller_ = std::make_unique(\n frame.right - frame.left, frame.bottom - frame.top, project_);\n // Ensure that basic setup of the controller was successful.\n if (!flutter_controller_->engine() || !flutter_controller_->view()) {\n return false;\n }\n RegisterPlugins(flutter_controller_->engine());\n SetChildContent(flutter_controller_->view()->GetNativeWindow());\n\n flutter_controller_->engine()->SetNextFrameCallback([&]() {\n this->Show();\n });\n\n // Flutter can complete the first frame before the \"show window\" callback is\n // registered. The following call ensures a frame is pending to ensure the\n // window is shown. It is a no-op if the first frame hasn't completed yet.\n flutter_controller_->ForceRedraw();\n\n return true;\n}\n\nvoid FlutterWindow::OnDestroy() {\n if (flutter_controller_) {\n flutter_controller_ = nullptr;\n }\n\n Win32Window::OnDestroy();\n}\n\nLRESULT\nFlutterWindow::MessageHandler(HWND hwnd, UINT const message,\n WPARAM const wparam,\n LPARAM const lparam) noexcept {\n // Give Flutter, including plugins, an opportunity to handle window messages.\n if (flutter_controller_) {\n std::optional result =\n flutter_controller_->HandleTopLevelWindowProc(hwnd, message, wparam,\n lparam);\n if (result) {\n return *result;\n }\n }\n\n switch (message) {\n case WM_FONTCHANGE:\n flutter_controller_->engine()->ReloadSystemFonts();\n break;\n }\n\n return Win32Window::MessageHandler(hwnd, message, wparam, lparam);\n}\n" }, { "path": "mobile/windows/runner/flutter_window.h", "content": "#ifndef RUNNER_FLUTTER_WINDOW_H_\n#define RUNNER_FLUTTER_WINDOW_H_\n\n#include \n#include \n\n#include \n\n#include \"win32_window.h\"\n\n// A window that does nothing but host a Flutter view.\nclass FlutterWindow : public Win32Window {\n public:\n // Creates a new FlutterWindow hosting a Flutter view running |project|.\n explicit FlutterWindow(const flutter::DartProject& project);\n virtual ~FlutterWindow();\n\n protected:\n // Win32Window:\n bool OnCreate() override;\n void OnDestroy() override;\n LRESULT MessageHandler(HWND window, UINT const message, WPARAM const wparam,\n LPARAM const lparam) noexcept override;\n\n private:\n // The project to run.\n flutter::DartProject project_;\n\n // The Flutter instance hosted by this window.\n std::unique_ptr flutter_controller_;\n};\n\n#endif // RUNNER_FLUTTER_WINDOW_H_\n" }, { "path": "mobile/windows/runner/main.cpp", "content": "#include \n#include \n#include \n\n#include \"flutter_window.h\"\n#include \"utils.h\"\n\nint APIENTRY wWinMain(_In_ HINSTANCE instance, _In_opt_ HINSTANCE prev,\n _In_ wchar_t *command_line, _In_ int show_command) {\n // Attach to console when present (e.g., 'flutter run') or create a\n // new console when running with a debugger.\n if (!::AttachConsole(ATTACH_PARENT_PROCESS) && ::IsDebuggerPresent()) {\n CreateAndAttachConsole();\n }\n\n // Initialize COM, so that it is available for use in the library and/or\n // plugins.\n ::CoInitializeEx(nullptr, COINIT_APARTMENTTHREADED);\n\n flutter::DartProject project(L\"data\");\n\n std::vector command_line_arguments =\n GetCommandLineArguments();\n\n project.set_dart_entrypoint_arguments(std::move(command_line_arguments));\n\n FlutterWindow window(project);\n Win32Window::Point origin(10, 10);\n Win32Window::Size size(1280, 720);\n if (!window.Create(L\"chat_mobile\", origin, size)) {\n return EXIT_FAILURE;\n }\n window.SetQuitOnClose(true);\n\n ::MSG msg;\n while (::GetMessage(&msg, nullptr, 0, 0)) {\n ::TranslateMessage(&msg);\n ::DispatchMessage(&msg);\n }\n\n ::CoUninitialize();\n return EXIT_SUCCESS;\n}\n" }, { "path": "mobile/windows/runner/resource.h", "content": "//{{NO_DEPENDENCIES}}\n// Microsoft Visual C++ generated include file.\n// Used by Runner.rc\n//\n#define IDI_APP_ICON 101\n\n// Next default values for new objects\n//\n#ifdef APSTUDIO_INVOKED\n#ifndef APSTUDIO_READONLY_SYMBOLS\n#define _APS_NEXT_RESOURCE_VALUE 102\n#define _APS_NEXT_COMMAND_VALUE 40001\n#define _APS_NEXT_CONTROL_VALUE 1001\n#define _APS_NEXT_SYMED_VALUE 101\n#endif\n#endif\n" }, { "path": "mobile/windows/runner/runner.exe.manifest", "content": "\n\n \n \n PerMonitorV2\n \n \n \n \n \n \n \n \n\n" }, { "path": "mobile/windows/runner/utils.cpp", "content": "#include \"utils.h\"\n\n#include \n#include \n#include \n#include \n\n#include \n\nvoid CreateAndAttachConsole() {\n if (::AllocConsole()) {\n FILE *unused;\n if (freopen_s(&unused, \"CONOUT$\", \"w\", stdout)) {\n _dup2(_fileno(stdout), 1);\n }\n if (freopen_s(&unused, \"CONOUT$\", \"w\", stderr)) {\n _dup2(_fileno(stdout), 2);\n }\n std::ios::sync_with_stdio();\n FlutterDesktopResyncOutputStreams();\n }\n}\n\nstd::vector GetCommandLineArguments() {\n // Convert the UTF-16 command line arguments to UTF-8 for the Engine to use.\n int argc;\n wchar_t** argv = ::CommandLineToArgvW(::GetCommandLineW(), &argc);\n if (argv == nullptr) {\n return std::vector();\n }\n\n std::vector command_line_arguments;\n\n // Skip the first argument as it's the binary name.\n for (int i = 1; i < argc; i++) {\n command_line_arguments.push_back(Utf8FromUtf16(argv[i]));\n }\n\n ::LocalFree(argv);\n\n return command_line_arguments;\n}\n\nstd::string Utf8FromUtf16(const wchar_t* utf16_string) {\n if (utf16_string == nullptr) {\n return std::string();\n }\n unsigned int target_length = ::WideCharToMultiByte(\n CP_UTF8, WC_ERR_INVALID_CHARS, utf16_string,\n -1, nullptr, 0, nullptr, nullptr)\n -1; // remove the trailing null character\n int input_length = (int)wcslen(utf16_string);\n std::string utf8_string;\n if (target_length == 0 || target_length > utf8_string.max_size()) {\n return utf8_string;\n }\n utf8_string.resize(target_length);\n int converted_length = ::WideCharToMultiByte(\n CP_UTF8, WC_ERR_INVALID_CHARS, utf16_string,\n input_length, utf8_string.data(), target_length, nullptr, nullptr);\n if (converted_length == 0) {\n return std::string();\n }\n return utf8_string;\n}\n" }, { "path": "mobile/windows/runner/utils.h", "content": "#ifndef RUNNER_UTILS_H_\n#define RUNNER_UTILS_H_\n\n#include \n#include \n\n// Creates a console for the process, and redirects stdout and stderr to\n// it for both the runner and the Flutter library.\nvoid CreateAndAttachConsole();\n\n// Takes a null-terminated wchar_t* encoded in UTF-16 and returns a std::string\n// encoded in UTF-8. Returns an empty std::string on failure.\nstd::string Utf8FromUtf16(const wchar_t* utf16_string);\n\n// Gets the command line arguments passed in as a std::vector,\n// encoded in UTF-8. Returns an empty std::vector on failure.\nstd::vector GetCommandLineArguments();\n\n#endif // RUNNER_UTILS_H_\n" }, { "path": "mobile/windows/runner/win32_window.cpp", "content": "#include \"win32_window.h\"\n\n#include \n#include \n\n#include \"resource.h\"\n\nnamespace {\n\n/// Window attribute that enables dark mode window decorations.\n///\n/// Redefined in case the developer's machine has a Windows SDK older than\n/// version 10.0.22000.0.\n/// See: https://docs.microsoft.com/windows/win32/api/dwmapi/ne-dwmapi-dwmwindowattribute\n#ifndef DWMWA_USE_IMMERSIVE_DARK_MODE\n#define DWMWA_USE_IMMERSIVE_DARK_MODE 20\n#endif\n\nconstexpr const wchar_t kWindowClassName[] = L\"FLUTTER_RUNNER_WIN32_WINDOW\";\n\n/// Registry key for app theme preference.\n///\n/// A value of 0 indicates apps should use dark mode. A non-zero or missing\n/// value indicates apps should use light mode.\nconstexpr const wchar_t kGetPreferredBrightnessRegKey[] =\n L\"Software\\\\Microsoft\\\\Windows\\\\CurrentVersion\\\\Themes\\\\Personalize\";\nconstexpr const wchar_t kGetPreferredBrightnessRegValue[] = L\"AppsUseLightTheme\";\n\n// The number of Win32Window objects that currently exist.\nstatic int g_active_window_count = 0;\n\nusing EnableNonClientDpiScaling = BOOL __stdcall(HWND hwnd);\n\n// Scale helper to convert logical scaler values to physical using passed in\n// scale factor\nint Scale(int source, double scale_factor) {\n return static_cast(source * scale_factor);\n}\n\n// Dynamically loads the |EnableNonClientDpiScaling| from the User32 module.\n// This API is only needed for PerMonitor V1 awareness mode.\nvoid EnableFullDpiSupportIfAvailable(HWND hwnd) {\n HMODULE user32_module = LoadLibraryA(\"User32.dll\");\n if (!user32_module) {\n return;\n }\n auto enable_non_client_dpi_scaling =\n reinterpret_cast(\n GetProcAddress(user32_module, \"EnableNonClientDpiScaling\"));\n if (enable_non_client_dpi_scaling != nullptr) {\n enable_non_client_dpi_scaling(hwnd);\n }\n FreeLibrary(user32_module);\n}\n\n} // namespace\n\n// Manages the Win32Window's window class registration.\nclass WindowClassRegistrar {\n public:\n ~WindowClassRegistrar() = default;\n\n // Returns the singleton registrar instance.\n static WindowClassRegistrar* GetInstance() {\n if (!instance_) {\n instance_ = new WindowClassRegistrar();\n }\n return instance_;\n }\n\n // Returns the name of the window class, registering the class if it hasn't\n // previously been registered.\n const wchar_t* GetWindowClass();\n\n // Unregisters the window class. Should only be called if there are no\n // instances of the window.\n void UnregisterWindowClass();\n\n private:\n WindowClassRegistrar() = default;\n\n static WindowClassRegistrar* instance_;\n\n bool class_registered_ = false;\n};\n\nWindowClassRegistrar* WindowClassRegistrar::instance_ = nullptr;\n\nconst wchar_t* WindowClassRegistrar::GetWindowClass() {\n if (!class_registered_) {\n WNDCLASS window_class{};\n window_class.hCursor = LoadCursor(nullptr, IDC_ARROW);\n window_class.lpszClassName = kWindowClassName;\n window_class.style = CS_HREDRAW | CS_VREDRAW;\n window_class.cbClsExtra = 0;\n window_class.cbWndExtra = 0;\n window_class.hInstance = GetModuleHandle(nullptr);\n window_class.hIcon =\n LoadIcon(window_class.hInstance, MAKEINTRESOURCE(IDI_APP_ICON));\n window_class.hbrBackground = 0;\n window_class.lpszMenuName = nullptr;\n window_class.lpfnWndProc = Win32Window::WndProc;\n RegisterClass(&window_class);\n class_registered_ = true;\n }\n return kWindowClassName;\n}\n\nvoid WindowClassRegistrar::UnregisterWindowClass() {\n UnregisterClass(kWindowClassName, nullptr);\n class_registered_ = false;\n}\n\nWin32Window::Win32Window() {\n ++g_active_window_count;\n}\n\nWin32Window::~Win32Window() {\n --g_active_window_count;\n Destroy();\n}\n\nbool Win32Window::Create(const std::wstring& title,\n const Point& origin,\n const Size& size) {\n Destroy();\n\n const wchar_t* window_class =\n WindowClassRegistrar::GetInstance()->GetWindowClass();\n\n const POINT target_point = {static_cast(origin.x),\n static_cast(origin.y)};\n HMONITOR monitor = MonitorFromPoint(target_point, MONITOR_DEFAULTTONEAREST);\n UINT dpi = FlutterDesktopGetDpiForMonitor(monitor);\n double scale_factor = dpi / 96.0;\n\n HWND window = CreateWindow(\n window_class, title.c_str(), WS_OVERLAPPEDWINDOW,\n Scale(origin.x, scale_factor), Scale(origin.y, scale_factor),\n Scale(size.width, scale_factor), Scale(size.height, scale_factor),\n nullptr, nullptr, GetModuleHandle(nullptr), this);\n\n if (!window) {\n return false;\n }\n\n UpdateTheme(window);\n\n return OnCreate();\n}\n\nbool Win32Window::Show() {\n return ShowWindow(window_handle_, SW_SHOWNORMAL);\n}\n\n// static\nLRESULT CALLBACK Win32Window::WndProc(HWND const window,\n UINT const message,\n WPARAM const wparam,\n LPARAM const lparam) noexcept {\n if (message == WM_NCCREATE) {\n auto window_struct = reinterpret_cast(lparam);\n SetWindowLongPtr(window, GWLP_USERDATA,\n reinterpret_cast(window_struct->lpCreateParams));\n\n auto that = static_cast(window_struct->lpCreateParams);\n EnableFullDpiSupportIfAvailable(window);\n that->window_handle_ = window;\n } else if (Win32Window* that = GetThisFromHandle(window)) {\n return that->MessageHandler(window, message, wparam, lparam);\n }\n\n return DefWindowProc(window, message, wparam, lparam);\n}\n\nLRESULT\nWin32Window::MessageHandler(HWND hwnd,\n UINT const message,\n WPARAM const wparam,\n LPARAM const lparam) noexcept {\n switch (message) {\n case WM_DESTROY:\n window_handle_ = nullptr;\n Destroy();\n if (quit_on_close_) {\n PostQuitMessage(0);\n }\n return 0;\n\n case WM_DPICHANGED: {\n auto newRectSize = reinterpret_cast(lparam);\n LONG newWidth = newRectSize->right - newRectSize->left;\n LONG newHeight = newRectSize->bottom - newRectSize->top;\n\n SetWindowPos(hwnd, nullptr, newRectSize->left, newRectSize->top, newWidth,\n newHeight, SWP_NOZORDER | SWP_NOACTIVATE);\n\n return 0;\n }\n case WM_SIZE: {\n RECT rect = GetClientArea();\n if (child_content_ != nullptr) {\n // Size and position the child window.\n MoveWindow(child_content_, rect.left, rect.top, rect.right - rect.left,\n rect.bottom - rect.top, TRUE);\n }\n return 0;\n }\n\n case WM_ACTIVATE:\n if (child_content_ != nullptr) {\n SetFocus(child_content_);\n }\n return 0;\n\n case WM_DWMCOLORIZATIONCOLORCHANGED:\n UpdateTheme(hwnd);\n return 0;\n }\n\n return DefWindowProc(window_handle_, message, wparam, lparam);\n}\n\nvoid Win32Window::Destroy() {\n OnDestroy();\n\n if (window_handle_) {\n DestroyWindow(window_handle_);\n window_handle_ = nullptr;\n }\n if (g_active_window_count == 0) {\n WindowClassRegistrar::GetInstance()->UnregisterWindowClass();\n }\n}\n\nWin32Window* Win32Window::GetThisFromHandle(HWND const window) noexcept {\n return reinterpret_cast(\n GetWindowLongPtr(window, GWLP_USERDATA));\n}\n\nvoid Win32Window::SetChildContent(HWND content) {\n child_content_ = content;\n SetParent(content, window_handle_);\n RECT frame = GetClientArea();\n\n MoveWindow(content, frame.left, frame.top, frame.right - frame.left,\n frame.bottom - frame.top, true);\n\n SetFocus(child_content_);\n}\n\nRECT Win32Window::GetClientArea() {\n RECT frame;\n GetClientRect(window_handle_, &frame);\n return frame;\n}\n\nHWND Win32Window::GetHandle() {\n return window_handle_;\n}\n\nvoid Win32Window::SetQuitOnClose(bool quit_on_close) {\n quit_on_close_ = quit_on_close;\n}\n\nbool Win32Window::OnCreate() {\n // No-op; provided for subclasses.\n return true;\n}\n\nvoid Win32Window::OnDestroy() {\n // No-op; provided for subclasses.\n}\n\nvoid Win32Window::UpdateTheme(HWND const window) {\n DWORD light_mode;\n DWORD light_mode_size = sizeof(light_mode);\n LSTATUS result = RegGetValue(HKEY_CURRENT_USER, kGetPreferredBrightnessRegKey,\n kGetPreferredBrightnessRegValue,\n RRF_RT_REG_DWORD, nullptr, &light_mode,\n &light_mode_size);\n\n if (result == ERROR_SUCCESS) {\n BOOL enable_dark_mode = light_mode == 0;\n DwmSetWindowAttribute(window, DWMWA_USE_IMMERSIVE_DARK_MODE,\n &enable_dark_mode, sizeof(enable_dark_mode));\n }\n}\n" }, { "path": "mobile/windows/runner/win32_window.h", "content": "#ifndef RUNNER_WIN32_WINDOW_H_\n#define RUNNER_WIN32_WINDOW_H_\n\n#include \n\n#include \n#include \n#include \n\n// A class abstraction for a high DPI-aware Win32 Window. Intended to be\n// inherited from by classes that wish to specialize with custom\n// rendering and input handling\nclass Win32Window {\n public:\n struct Point {\n unsigned int x;\n unsigned int y;\n Point(unsigned int x, unsigned int y) : x(x), y(y) {}\n };\n\n struct Size {\n unsigned int width;\n unsigned int height;\n Size(unsigned int width, unsigned int height)\n : width(width), height(height) {}\n };\n\n Win32Window();\n virtual ~Win32Window();\n\n // Creates a win32 window with |title| that is positioned and sized using\n // |origin| and |size|. New windows are created on the default monitor. Window\n // sizes are specified to the OS in physical pixels, hence to ensure a\n // consistent size this function will scale the inputted width and height as\n // as appropriate for the default monitor. The window is invisible until\n // |Show| is called. Returns true if the window was created successfully.\n bool Create(const std::wstring& title, const Point& origin, const Size& size);\n\n // Show the current window. Returns true if the window was successfully shown.\n bool Show();\n\n // Release OS resources associated with window.\n void Destroy();\n\n // Inserts |content| into the window tree.\n void SetChildContent(HWND content);\n\n // Returns the backing Window handle to enable clients to set icon and other\n // window properties. Returns nullptr if the window has been destroyed.\n HWND GetHandle();\n\n // If true, closing this window will quit the application.\n void SetQuitOnClose(bool quit_on_close);\n\n // Return a RECT representing the bounds of the current client area.\n RECT GetClientArea();\n\n protected:\n // Processes and route salient window messages for mouse handling,\n // size change and DPI. Delegates handling of these to member overloads that\n // inheriting classes can handle.\n virtual LRESULT MessageHandler(HWND window,\n UINT const message,\n WPARAM const wparam,\n LPARAM const lparam) noexcept;\n\n // Called when CreateAndShow is called, allowing subclass window-related\n // setup. Subclasses should return false if setup fails.\n virtual bool OnCreate();\n\n // Called when Destroy is called.\n virtual void OnDestroy();\n\n private:\n friend class WindowClassRegistrar;\n\n // OS callback called by message pump. Handles the WM_NCCREATE message which\n // is passed when the non-client area is being created and enables automatic\n // non-client DPI scaling so that the non-client area automatically\n // responds to changes in DPI. All other messages are handled by\n // MessageHandler.\n static LRESULT CALLBACK WndProc(HWND const window,\n UINT const message,\n WPARAM const wparam,\n LPARAM const lparam) noexcept;\n\n // Retrieves a class instance pointer for |window|\n static Win32Window* GetThisFromHandle(HWND const window) noexcept;\n\n // Update the window frame's theme to match the system theme.\n static void UpdateTheme(HWND const window);\n\n bool quit_on_close_ = false;\n\n // window handle for top level window.\n HWND window_handle_ = nullptr;\n\n // window handle for hosted content.\n HWND child_content_ = nullptr;\n};\n\n#endif // RUNNER_WIN32_WINDOW_H_\n" }, { "path": "scripts/branch_clean.py", "content": "import subprocess\nfrom datetime import datetime\n\ndef get_local_branches():\n # Get the list of local branches\n result = subprocess.run(['git', 'branch'], stdout=subprocess.PIPE, text=True)\n branches = result.stdout.splitlines()\n # Remove the '*' from the current branch\n branches = [branch.strip('*').strip() for branch in branches]\n return branches\n\ndef get_branch_last_commit_date(branch):\n # Get the last commit date of the branch\n result = subprocess.run(['git', 'log', '-1', '--format=%cd', '--date=iso', branch], stdout=subprocess.PIPE, text=True)\n date_str = result.stdout.strip()\n return datetime.strptime(date_str, '%Y-%m-%d %H:%M:%S %z')\n\ndef delete_branch(branch):\n # Delete the branch\n subprocess.run(['git', 'branch', '-D', branch])\n\ndef confirm_deletion(branch):\n # Ask the user to confirm deletion\n response = input(f\"Do you want to delete the branch '{branch}'? (y/n): \").strip().lower()\n return response == 'y'\n\ndef main():\n # Get all local branches\n branches = get_local_branches()\n\n # Get the last commit date for each branch\n branch_dates = [(branch, get_branch_last_commit_date(branch)) for branch in branches]\n\n # Sort branches by last commit date (oldest first)\n branch_dates.sort(key=lambda x: x[1])\n\n # Get the oldest 5 branches\n oldest_branches = [branch for branch, _ in branch_dates[:5]]\n\n # Delete the oldest 5 branches with confirmation\n for branch in oldest_branches:\n if confirm_deletion(branch):\n print(f\"Deleting branch: {branch}\")\n delete_branch(branch)\n else:\n print(f\"Skipping branch: {branch}\")\n\nif __name__ == \"__main__\":\n main()\n" }, { "path": "scripts/locale_missing_key.py", "content": "from pathlib import Path\nimport json\nimport argparse\n\n# Recursive function to find missing keys in dictionaries\ndef find_missing_keys(base_dict, other_dict):\n missing_keys = {}\n for key in base_dict:\n if key not in other_dict:\n missing_keys[key] = base_dict[key]\n elif isinstance(base_dict[key], dict) and isinstance(other_dict[key], dict):\n sub_missing_keys = find_missing_keys(base_dict[key], other_dict[key])\n if sub_missing_keys:\n missing_keys[key] = sub_missing_keys\n return missing_keys\n\n\n\ndef check_locales(dir_name: str, base_locale: str = 'zh-CN'):\n # Load the zh-CN JSON file\n zh_cn_file = Path(dir_name) / f'{base_locale}.json'\n with zh_cn_file.open('r') as f:\n zh_cn = json.load(f)\n\n # Look for other JSON files in the current directory\n for file in Path(dir_name).glob('*.json'):\n if 'more' in file.stem:\n continue\n cur_locale = file.stem\n if cur_locale != base_locale:\n with file.open('r') as f:\n other_dict = json.load(f)\n missing_keys = find_missing_keys(zh_cn, other_dict)\n print(f'\\n\\n please translate to {cur_locale}:', missing_keys)\n\nif __name__ == '__main__':\n parser = argparse.ArgumentParser(description='Check missing keys in language localization files')\n parser.add_argument('dir_name', type=str, help='directory where the JSON files are located')\n parser.add_argument('--base', type=str, default='zh-CN', help='base locale to compare against')\n args = parser.parse_args()\n check_locales(args.dir_name, args.base)\n # python check_locales.py /path/to/locales --base en-US\n" }, { "path": "scripts/merge_keys.py", "content": "import json\nfrom pathlib import Path\n\n # Function to recursively merge two dictionaries\ndef merge_dicts(d1, d2):\n for key, val2 in d2.items():\n if key in d1:\n # If both values are dictionaries, merge them recursively\n if isinstance(val2, dict) and isinstance(d1[key], dict):\n merge_dicts(d1[key], val2)\n # If both values are lists, extend the first list with the second\n elif isinstance(val2, list) and isinstance(d1[key], list):\n d1[key].extend(val2)\n # Otherwise, overwrite the first value with the second\n else:\n d1[key] = val2\n else:\n # If the key doesn't exist in the first dict, add it and its value\n d1[key] = val2\n\ndef merge_json_files(file1, file2):\n \"\"\"Merges the contents of two JSON files recursively by key.\"\"\"\n \n # Read in the JSON data from the files\n with open(file1, 'r') as f1:\n data1 = json.load(f1)\n with open(file2, 'r') as f2:\n data2 = json.load(f2)\n\n merge_dicts(data1, data2)\n # write the merged content back to file2\n with open(file1, 'w') as fp1:\n json.dump(data1, fp1, indent=4,ensure_ascii=False, sort_keys=True)\n\n\n# main\nlocale_dir = Path(__file__).parent.parent / \"web/src/locales\"\nextra_jsons = locale_dir.glob(\"*-more.json\")\n# web/src/locales/en-US.json web/src/locales/en-US-more.json \nfor extra in extra_jsons:\n print(extra)\n origin = extra.parent / extra.name.replace('-more', '')\n print(origin, extra)\n merge_json_files(origin, extra)\n" }, { "path": "scripts/remove_older_branch.py", "content": "import subprocess\nimport time\n\n# Get the list of merged branches\noutput = subprocess.check_output(['git', 'branch']).decode('utf-8')\nprint(output)\n# Exclude the current branch and any branches named \"main\", \"master\", or \"develop\"\nexclude_branches = ['main', 'master', 'develop']\nbranches = [line.strip('* ') for line in output.split('\\n') if line.strip('* ') not in exclude_branches and line.strip()]\nprint(branches)\n# Get the current time one month ago\none_month_ago = time.time() - 30 * 24 * 60 * 60\n\n# Delete branches that have not been updated in the last month\nfor branch in branches:\n # Get the Unix timestamp of the last commit on the branch\n output = subprocess.check_output(['git', 'log', '-1', '--format=%at', branch]).decode('utf-8')\n print(output)\n last_commit_time = int(output.strip())\n\n # Delete the branch if its last commit is older than one month\n if last_commit_time < one_month_ago:\n subprocess.call(['git', 'branch', '-D', branch])" }, { "path": "web/.commitlintrc.json", "content": "{\n \"extends\": [\"@commitlint/config-conventional\"]\n}\n" }, { "path": "web/.editorconfig", "content": "# Editor configuration, see http://editorconfig.org\n\nroot = true\n\n[*]\ncharset = utf-8\nindent_style = tab\nindent_size = 2\nend_of_line = lf\ntrim_trailing_whitespace = true\ninsert_final_newline = true\n" }, { "path": "web/.eslintrc.cjs", "content": "module.exports = {\n root: true,\n extends: ['@antfu'],\n}\n" }, { "path": "web/.gitattributes", "content": "\"*.vue\" eol=lf\n\"*.js\" eol=lf\n\"*.ts\" eol=lf\n\"*.jsx\" eol=lf\n\"*.tsx\" eol=lf\n\"*.cjs\" eol=lf\n\"*.cts\" eol=lf\n\"*.mjs\" eol=lf\n\"*.mts\" eol=lf\n\"*.json\" eol=lf\n\"*.html\" eol=lf\n\"*.css\" eol=lf\n\"*.less\" eol=lf\n\"*.scss\" eol=lf\n\"*.sass\" eol=lf\n\"*.styl\" eol=lf\n\"*.md\" eol=lf\n" }, { "path": "web/.gitignore", "content": "# Logs\nlogs\n*.log\nnpm-debug.log*\nyarn-debug.log*\nyarn-error.log*\npnpm-debug.log*\nlerna-debug.log*\n\nnode_modules\n.DS_Store\ndist\ndist-ssr\ncoverage\n*.local\n\n/cypress/videos/\n/cypress/screenshots/\n\n# Editor directories and files\n.vscode/*\n!.vscode/settings.json\n!.vscode/extensions.json\n.idea\n*.suo\n*.ntvs*\n*.njsproj\n*.sln\n*.sw?\n\n# Environment variables files\n/service/.env\n" }, { "path": "web/.husky/commit-msg", "content": "#!/usr/bin/env sh\n. \"$(dirname -- \"$0\")/_/husky.sh\"\n\nnpx --no -- commitlint --edit \n" }, { "path": "web/.husky/pre-commit", "content": "#!/usr/bin/env sh\n. \"$(dirname -- \"$0\")/_/husky.sh\"\n\nnpx lint-staged\n" }, { "path": "web/.vscode/extensions.json", "content": "{\n \"recommendations\": [\"Vue.volar\", \"dbaeumer.vscode-eslint\"]\n}\n" }, { "path": "web/.vscode/settings.json", "content": "{\n \"prettier.enable\": false,\n \"editor.formatOnSave\": false,\n \"editor.codeActionsOnSave\": {\n \"source.fixAll.eslint\": \"explicit\"\n },\n \"eslint.validate\": [\n \"javascript\",\n \"javascriptreact\",\n \"typescript\",\n \"typescriptreact\",\n \"vue\",\n \"html\",\n \"json\",\n \"jsonc\",\n \"json5\",\n \"yaml\",\n \"yml\",\n \"markdown\"\n ],\n \"cSpell.words\": [\n \"antfu\",\n \"axios\",\n \"bumpp\",\n \"chatgpt\",\n \"chenzhaoyu\",\n \"commitlint\",\n \"davinci\",\n \"dockerhub\",\n \"esno\",\n \"GPTAPI\",\n \"highlightjs\",\n \"hljs\",\n \"iconify\",\n \"katex\",\n \"katexmath\",\n \"linkify\",\n \"logprobs\",\n \"mdhljs\",\n \"nodata\",\n \"OPENAI\",\n \"pinia\",\n \"Popconfirm\",\n \"rushstack\",\n \"Sider\",\n \"tailwindcss\",\n \"traptitech\",\n \"tsup\",\n \"Typecheck\",\n \"unplugin\",\n \"VITE\",\n \"vueuse\",\n \"Zhao\"\n ],\n \"i18n-ally.enabledParsers\": [\n \"ts\"\n ],\n \"i18n-ally.sortKeys\": true,\n \"i18n-ally.keepFulfilled\": true,\n \"i18n-ally.localesPaths\": [\n \"src/locales\"\n ],\n \"i18n-ally.keystyle\": \"nested\",\n \"editor.fontFamily\": \"'Fira Code'\"\n}\n" }, { "path": "web/docker-compose/docker-compose.yml", "content": "version: '3'\n\nservices:\n app:\n image: chenzhaoyu94/chatgpt-web # 总是使用latest,更新时重新pull该tag镜像即可\n ports:\n - 3002:3002\n environment:\n # 二选一\n OPENAI_API_KEY: xxxx\n # 二选一\n OPENAI_ACCESS_TOKEN: xxxxxx\n # API接口地址，可选，设置 OPENAI_API_KEY 时可用\n OPENAI_API_BASE_URL: xxxx\n # API模型，可选，设置 OPENAI_API_KEY 时可用\n OPENAI_API_MODEL: xxxx\n # 反向代理，可选\n API_REVERSE_PROXY: xxx\n # 访问权限密钥，可选\n AUTH_SECRET_KEY: xxx\n # 超时，单位毫秒，可选\n TIMEOUT_MS: 60000\n # Socks代理，可选，和 SOCKS_PROXY_PORT 一起时生效\n SOCKS_PROXY_HOST: xxxx\n # Socks代理端口，可选，和 SOCKS_PROXY_HOST 一起时生效\n SOCKS_PROXY_PORT: xxxx\n nginx:\n image: nginx:alpine\n ports:\n - '80:80'\n expose:\n - '80'\n volumes:\n - ./nginx/html:/usr/share/nginx/html\n - ./nginx/nginx.conf:/etc/nginx/conf.d/default.conf\n links:\n - app\n" }, { "path": "web/docker-compose/nginx/nginx.conf", "content": "server {\n\tlisten 80;\n\tserver_name localhost;\n\tcharset utf-8;\n\terror_page 500 502 503 504 /50x.html;\n\tlocation / {\n\t\t\troot /usr/share/nginx/html;\n \t\ttry_files $uri /index.html;\n\t}\n\n\tlocation /api {\n\t\t\tproxy_set_header X-Real-IP $remote_addr; #转发用户IP\n\t\t\tproxy_pass http://app:3002;\n\t}\n\n\tproxy_set_header Host $host;\n\tproxy_set_header X-Real-IP $remote_addr;\n\tproxy_set_header REMOTE-HOST $remote_addr;\n\tproxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;\n}\n" }, { "path": "web/docker-compose/readme.md", "content": "### docker-compose 部署教程\n- 将打包好的前端文件放到 `nginx/html` 目录下\n- ```shell\n # 启动\n docker-compose up -d\n ```\n- ```shell\n # 查看运行状态\n docker ps\n ```\n- ```shell\n # 结束运行\n docker-compose down\n ```\n" }, { "path": "web/docs/code_runner.md", "content": "# Code Runner Feature Implementation Plan\n\n## Overview\n\nThis document outlines the implementation plan for adding interactive code execution capabilities to the chat application's artifact system. The code runner will allow users to execute JavaScript, Python, and other languages directly within chat artifacts, providing real-time output and interactive programming experiences.\n\n## Goals\n\n- **Educational**: Enable live coding tutorials and learning experiences\n- **Prototyping**: Quick algorithm testing and experimentation\n- **Data Visualization**: Interactive charts, graphs, and visual demonstrations\n- **Development**: Code examples that users can run and modify\n\n## Architecture Decision\n\n### Selected Approach: Browser-Based Execution\n\n**Reasoning**:\n- No server resources required\n- Real-time execution with minimal latency\n- Scales with user count (computation on client)\n- Supports rich DOM manipulation and visualization\n- Easier deployment and maintenance\n\n**Supported Languages**:\n1. **JavaScript/TypeScript** - Native browser execution in Web Workers\n2. **Python** - Pyodide (Python in WebAssembly)\n3. **Future**: HTML/CSS live preview, SQL (via SQLite WASM)\n\n## Technical Implementation\n\n### 1. Database Schema Extensions\n\n```sql\n-- Add execution results to chat messages\nALTER TABLE chat_message ADD COLUMN execution_results JSONB DEFAULT '[]';\n\n-- Index for searching executable artifacts\nCREATE INDEX idx_chat_message_execution ON chat_message \nUSING GIN (execution_results) \nWHERE execution_results != '[]';\n```\n\n**Execution Result Structure**:\n```json\n{\n \"artifact_uuid\": \"string\",\n \"execution_id\": \"string\",\n \"timestamp\": \"ISO8601\",\n \"language\": \"javascript|python\",\n \"output\": [\n {\n \"type\": \"log|error|return|stdout\",\n \"content\": \"string\",\n \"timestamp\": \"ISO8601\"\n }\n ],\n \"execution_time_ms\": \"number\",\n \"status\": \"success|error|timeout\"\n}\n```\n\n### 2. Frontend Architecture\n\n#### Enhanced Artifact Types\n\n**New Artifact Type**: `executable-code`\n- Extends existing `code` artifacts\n- Includes execution metadata\n- Supports multiple output formats\n\n```typescript\n// web/src/typings/chat.d.ts\ninterface ExecutableArtifact extends Artifact {\n type: 'executable-code'\n language: 'javascript' | 'python' | 'typescript'\n isExecutable: true\n executionResults?: ExecutionResult[]\n}\n\ninterface ExecutionResult {\n id: string\n type: 'log' | 'error' | 'return' | 'stdout'\n content: string\n timestamp: string\n}\n```\n\n#### Core Components\n\n**1. Enhanced ArtifactViewer.vue**\n\n```vue\n\n\n\n```\n\n**2. Code Runner Service**\n\n```typescript\n// web/src/services/codeRunner.ts\nimport { ExecutionResult } from '@/typings/chat'\n\nexport class CodeRunner {\n private jsWorker: Worker | null = null\n private pythonRunner: PythonRunner | null = null\n\n async execute(language: string, code: string): Promise {\n const startTime = performance.now()\n \n try {\n let results: ExecutionResult[]\n \n switch (language) {\n case 'javascript':\n case 'typescript':\n results = await this.executeJavaScript(code)\n break\n case 'python':\n results = await this.executePython(code)\n break\n default:\n throw new Error(`Unsupported language: ${language}`)\n }\n \n // Add execution time to results\n const executionTime = performance.now() - startTime\n results.forEach(result => {\n result.execution_time_ms = executionTime\n })\n \n return results\n } catch (error) {\n return [{\n id: Date.now().toString(),\n type: 'error',\n content: error.message,\n timestamp: new Date().toISOString(),\n execution_time_ms: performance.now() - startTime\n }]\n }\n }\n\n private async executeJavaScript(code: string): Promise {\n return new Promise((resolve, reject) => {\n if (!this.jsWorker) {\n this.jsWorker = new Worker('/workers/jsRunner.js')\n }\n\n const timeout = setTimeout(() => {\n this.jsWorker?.terminate()\n this.jsWorker = null\n reject(new Error('Code execution timed out'))\n }, 10000) // 10 second timeout\n\n this.jsWorker.onmessage = (e) => {\n clearTimeout(timeout)\n const { type, data } = e.data\n \n if (type === 'results') {\n resolve(data)\n } else if (type === 'error') {\n reject(new Error(data.message))\n }\n }\n\n this.jsWorker.onerror = (error) => {\n clearTimeout(timeout)\n reject(error)\n }\n\n this.jsWorker.postMessage({ code, timeout: 10000 })\n })\n }\n\n private async executePython(code: string): Promise {\n if (!this.pythonRunner) {\n this.pythonRunner = new PythonRunner()\n await this.pythonRunner.initialize()\n }\n\n return this.pythonRunner.execute(code)\n }\n\n dispose() {\n if (this.jsWorker) {\n this.jsWorker.terminate()\n this.jsWorker = null\n }\n if (this.pythonRunner) {\n this.pythonRunner.dispose()\n this.pythonRunner = null\n }\n }\n}\n```\n\n**3. JavaScript Worker**\n\n```typescript\n// public/workers/jsRunner.js\nclass SafeJSRunner {\n constructor() {\n this.output = []\n this.setupConsole()\n }\n\n setupConsole() {\n // Capture console methods\n this.console = {\n log: (...args) => this.addOutput('log', args.join(' ')),\n error: (...args) => this.addOutput('error', args.join(' ')),\n warn: (...args) => this.addOutput('warn', args.join(' ')),\n info: (...args) => this.addOutput('info', args.join(' '))\n }\n }\n\n addOutput(type, content) {\n this.output.push({\n id: Date.now().toString() + Math.random(),\n type: type,\n content: String(content),\n timestamp: new Date().toISOString()\n })\n }\n\n async execute(code) {\n this.output = []\n \n try {\n // Create safe execution context\n const safeGlobals = {\n console: this.console,\n Math: Math,\n Date: Date,\n Array: Array,\n Object: Object,\n String: String,\n Number: Number,\n Boolean: Boolean,\n JSON: JSON,\n setTimeout: (fn, ms) => setTimeout(fn, Math.min(ms, 5000)),\n setInterval: (fn, ms) => setInterval(fn, Math.max(ms, 100))\n }\n\n // Execute code in safe context\n const result = new Function(\n ...Object.keys(safeGlobals), \n `\n \"use strict\";\n ${code}\n `\n )(...Object.values(safeGlobals))\n\n // Add return value if exists\n if (result !== undefined) {\n this.addOutput('return', JSON.stringify(result, null, 2))\n }\n\n return this.output\n } catch (error) {\n this.addOutput('error', error.message)\n return this.output\n }\n }\n}\n\nconst runner = new SafeJSRunner()\n\nself.onmessage = async (e) => {\n const { code, timeout } = e.data\n \n try {\n const results = await runner.execute(code)\n self.postMessage({ type: 'results', data: results })\n } catch (error) {\n self.postMessage({ \n type: 'error', \n data: { message: error.message } \n })\n }\n}\n```\n\n**4. Python Runner (Pyodide)**\n\n```typescript\n// web/src/services/pythonRunner.ts\nexport class PythonRunner {\n private pyodide: any = null\n private initialized = false\n\n async initialize() {\n if (this.initialized) return\n\n // Dynamic import to avoid bundling\n const { loadPyodide } = await import('pyodide')\n \n this.pyodide = await loadPyodide({\n indexURL: 'https://cdn.jsdelivr.net/pyodide/'\n })\n\n // Install common packages\n await this.pyodide.loadPackage(['numpy', 'matplotlib', 'pandas'])\n \n this.initialized = true\n }\n\n async execute(code: string): Promise {\n if (!this.initialized) {\n await this.initialize()\n }\n\n const output: ExecutionResult[] = []\n \n try {\n // Capture stdout\n this.pyodide.runPython(`\n import sys\n from io import StringIO\n \n # Capture stdout\n old_stdout = sys.stdout\n sys.stdout = captured_output = StringIO()\n `)\n\n // Execute user code\n const result = this.pyodide.runPython(code)\n\n // Get captured output\n const stdout = this.pyodide.runPython(`\n sys.stdout = old_stdout\n captured_output.getvalue()\n `)\n\n // Add stdout if any\n if (stdout.trim()) {\n output.push({\n id: Date.now().toString(),\n type: 'stdout',\n content: stdout,\n timestamp: new Date().toISOString()\n })\n }\n\n // Add return value if exists\n if (result !== undefined && result !== null) {\n output.push({\n id: Date.now().toString() + '1',\n type: 'return',\n content: String(result),\n timestamp: new Date().toISOString()\n })\n }\n\n return output\n } catch (error) {\n return [{\n id: Date.now().toString(),\n type: 'error',\n content: error.message,\n timestamp: new Date().toISOString()\n }]\n }\n }\n\n dispose() {\n // Cleanup if needed\n this.pyodide = null\n this.initialized = false\n }\n}\n```\n\n### 3. Backend Integration\n\n#### Artifact Detection Enhancement\n\n```go\n// api/chat_main_service.go\nfunc extractArtifacts(content string) []Artifact {\n // Existing artifact extraction logic...\n \n // Add executable code detection\n executableCodeRegex := regexp.MustCompile(`(?s)```(\\w+)\\s*\\s*\\n(.*?)\\n\\s*````)\n executableMatches := executableCodeRegex.FindAllStringSubmatch(content, -1)\n \n for _, match := range executableMatches {\n if len(match) >= 4 {\n language := match[1]\n title := strings.TrimSpace(match[2])\n code := strings.TrimSpace(match[3])\n \n // Only certain languages are executable\n if isExecutableLanguage(language) {\n artifacts = append(artifacts, Artifact{\n UUID: generateUUID(),\n Type: \"executable-code\",\n Title: title,\n Content: code,\n Language: language,\n })\n }\n }\n }\n \n return artifacts\n}\n\nfunc isExecutableLanguage(lang string) bool {\n executableLangs := []string{\"javascript\", \"python\", \"typescript\", \"js\", \"py\", \"ts\"}\n for _, execLang := range executableLangs {\n if strings.EqualFold(lang, execLang) {\n return true\n }\n }\n return false\n}\n```\n\n#### Execution Results Storage\n\n```go\n// api/models.go\ntype ExecutionResult struct {\n ArtifactUUID string `json:\"artifact_uuid\"`\n ExecutionID string `json:\"execution_id\"`\n Timestamp time.Time `json:\"timestamp\"`\n Language string `json:\"language\"`\n Output []ExecutionOutputLine `json:\"output\"`\n ExecutionTimeMs int64 `json:\"execution_time_ms\"`\n Status string `json:\"status\"` // success, error, timeout\n}\n\ntype ExecutionOutputLine struct {\n Type string `json:\"type\"` // log, error, return, stdout\n Content string `json:\"content\"`\n Timestamp time.Time `json:\"timestamp\"`\n}\n```\n\n#### API Endpoints\n\n```go\n// api/chat_execution_handler.go\nfunc (h *ChatHandler) SaveExecutionResult(w http.ResponseWriter, r *http.Request) {\n // POST /api/chat/executions\n // Save execution results to database\n}\n\nfunc (h *ChatHandler) GetExecutionHistory(w http.ResponseWriter, r *http.Request) {\n // GET /api/chat/executions/:messageUUID\n // Get execution history for a message\n}\n```\n\n### 4. Security Considerations\n\n#### JavaScript Security\n- **Web Workers**: No DOM access, isolated execution\n- **Timeout Limits**: 10-second maximum execution time\n- **Memory Limits**: Worker termination prevents memory leaks\n- **API Restrictions**: No network access, limited global objects\n- **Safe Globals**: Whitelist of allowed APIs only\n\n#### Python Security\n- **Pyodide Sandbox**: Runs in WebAssembly, isolated from system\n- **Package Restrictions**: Only pre-approved packages\n- **Resource Limits**: Memory and execution time constraints\n- **No File System**: No access to local files\n\n#### General Security\n- **Content Validation**: Sanitize all user inputs\n- **Rate Limiting**: Limit executions per user/session\n- **Error Handling**: Safe error messages without system info\n- **Audit Logging**: Track all code executions\n\n### 5. Performance Optimizations\n\n#### Loading Strategy\n- **Lazy Loading**: Load runners only when needed\n- **Worker Pooling**: Reuse workers for multiple executions\n- **Caching**: Cache Pyodide and common libraries\n- **Progressive Loading**: Load features incrementally\n\n#### Resource Management\n- **Memory Monitoring**: Track and limit memory usage\n- **Cleanup**: Proper disposal of workers and resources\n- **Debouncing**: Limit rapid execution requests\n- **Background Loading**: Preload runners in background\n\n### 6. User Experience Enhancements\n\n#### Visual Feedback\n- **Loading States**: Show progress during execution\n- **Status Indicators**: Success/error/timeout states\n- **Execution Time**: Display performance metrics\n- **Output Formatting**: Syntax highlighting for results\n\n#### Interactive Features\n- **Code Editing**: Inline editing with syntax highlighting\n- **Auto-completion**: Basic code completion\n- **Error Highlighting**: Visual error indicators\n- **Execution History**: Show previous runs\n\n### 7. Implementation Phases\n\n#### Phase 1: Basic JavaScript Runner (Week 1)\n- [ ] Web Worker implementation\n- [ ] Basic JavaScript execution\n- [ ] Console output capture\n- [ ] Error handling\n- [ ] UI integration\n\n#### Phase 2: Enhanced JavaScript (Week 2)\n- [ ] Advanced security measures\n- [ ] Timeout and memory controls\n- [ ] Better error reporting\n- [ ] Execution history\n\n#### Phase 3: Python Support (Week 3)\n- [ ] Pyodide integration\n- [ ] Python execution environment\n- [ ] Package management\n- [ ] Performance optimization\n\n#### Phase 4: Advanced Features (Week 4)\n- [ ] Code editing capabilities\n- [ ] Library loading\n- [ ] Visualization support\n- [ ] Export/sharing features\n\n### 8. Testing Strategy\n\n#### Unit Tests\n- Code execution accuracy\n- Security boundary testing\n- Error handling scenarios\n- Performance benchmarks\n\n#### Integration Tests\n- Full artifact workflow\n- Database persistence\n- API endpoint testing\n- UI interaction testing\n\n#### Security Tests\n- Sandbox escape attempts\n- Resource exhaustion tests\n- Malicious code detection\n- Cross-site scripting prevention\n\n### 9. Deployment Considerations\n\n#### Bundle Size\n- **Pyodide**: ~3MB initial download\n- **Workers**: Minimal overhead\n- **Lazy Loading**: Only load when needed\n\n#### Browser Compatibility\n- **Web Workers**: Supported in all modern browsers\n- **WebAssembly**: Required for Python (95%+ browser support)\n- **Graceful Degradation**: Fallback for unsupported browsers\n\n#### CDN Strategy\n- **Pyodide CDN**: Use official CDN for reliability\n- **Worker Files**: Serve from application domain\n- **Caching**: Aggressive caching for static assets\n\n### 10. Future Enhancements\n\n#### Additional Languages\n- **SQL**: SQLite WebAssembly for database queries\n- **R**: R WebAssembly for statistical computing\n- **Go**: TinyGo WebAssembly compilation\n- **Rust**: Rust WebAssembly support\n\n#### Advanced Features\n- **Collaborative Editing**: Multiple users editing same code\n- **Version Control**: Track code changes over time\n- **Package Management**: Install custom packages\n- **Debugging Tools**: Step-through debugging\n- **Performance Profiling**: Execution analysis\n\n#### Integration Features\n- **GitHub Integration**: Save/load from repositories\n- **Notebook Export**: Export to Jupyter notebooks\n- **Sharing**: Public executable artifact galleries\n- **Embedding**: Embed in external websites\n\n## Success Metrics\n\n- **Execution Speed**: JavaScript < 100ms, Python < 1s for simple code\n- **Security**: Zero successful sandbox escapes\n- **Reliability**: 99.9% successful executions\n- **User Adoption**: 50% of code artifacts become executable\n- **Performance**: No impact on chat loading time\n\n## Conclusion\n\nThe Code Runner feature will transform the chat application into a powerful interactive development environment. By supporting both JavaScript and Python execution with strong security measures, it opens up possibilities for education, prototyping, and data analysis directly within the chat interface.\n\nThe phased implementation approach ensures steady progress while maintaining system stability and security. The browser-based architecture provides excellent scalability and performance while keeping infrastructure costs minimal." }, { "path": "web/index.html", "content": "\n\n\n\t\n\t\n\t\n\t\n\tChat\n\n\n\n\t
\n\t\t\n\t\t
\n\t\t\t
\n\t\t\t\t
\n\t\t\t\t
\n\t\t\t\t
\n\t\t\t
\n\t\t
\n\t
\n\n\n\n" }, { "path": "web/license", "content": "MIT License\n\nCopyright (c) 2023 ChenZhaoYu\n\nPermission is hereby granted, free of charge, to any person obtaining a copy\nof this software and associated documentation files (the \"Software\"), to deal\nin the Software without restriction, including without limitation the rights\nto use, copy, modify, merge, publish, distribute, sublicense, and/or sell\ncopies of the Software, and to permit persons to whom the Software is\nfurnished to do so, subject to the following conditions:\n\nThe above copyright notice and this permission notice shall be included in all\ncopies or substantial portions of the Software.\n\nTHE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\nIMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\nFITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\nAUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\nLIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\nOUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\nSOFTWARE.\n" }, { "path": "web/package.json", "content": "{\n \"name\": \"chatgpt-web\",\n \"version\": \"2.10.3\",\n \"private\": false,\n \"description\": \"ChatGPT Web\",\n \"author\": \"ChenZhaoYu \",\n \"keywords\": [\n \"chatgpt-web\",\n \"chatgpt\",\n \"chatbot\",\n \"vue\"\n ],\n \"scripts\": {\n \"dev\": \"rsbuild dev\",\n \"test\": \"vitest\",\n \"preview\": \"rsbuild preview\",\n \"build\": \"rsbuild build\",\n \"lint\": \"eslint .\",\n \"lint:fix\": \"eslint . --fix\"\n },\n \"dependencies\": {\n \"@tanstack/vue-query\": \"^5.40.1\",\n \"@types/uuid\": \"^10.0.0\",\n \"@vicons/ionicons5\": \"^0.12.0\",\n \"@vicons/material\": \"^0.13.0\",\n \"@vscode/markdown-it-katex\": \"^1.0.3\",\n \"@vueuse/core\": \"^9.13.0\",\n \"copy-to-clipboard\": \"^3.3.3\",\n \"date-fns\": \"^2.30.0\",\n \"dot-env\": \"^0.0.1\",\n \"highlight.js\": \"^11.7.0\",\n \"html2canvas\": \"^1.4.1\",\n \"jwt-decode\": \"^3.1.2\",\n \"katex\": \"^0.16.4\",\n \"lodash-es\": \"^4.17.21\",\n \"luxon\": \"^3.3.0\",\n \"markdown-it\": \"^13.0.1\",\n \"naive-ui\": \"^2.41.0\",\n \"pinia\": \"^2.3.1\",\n \"uuid\": \"^11.1.0\",\n \"vue\": \"3.5.11\",\n \"vue-i18n\": \"^9.2.2\",\n \"vue-router\": \"4.1.6\"\n },\n \"devDependencies\": {\n \"@antfu/eslint-config\": \"^0.35.3\",\n \"@commitlint/cli\": \"^17.4.4\",\n \"@commitlint/config-conventional\": \"^17.4.4\",\n \"@iconify/vue\": \"^4.1.0\",\n \"@rsbuild/core\": \"^1.2.3\",\n \"@rsbuild/plugin-less\": \"1.1.0\",\n \"@rsbuild/plugin-vue\": \"1.0.5\",\n \"@tanstack/vue-query-devtools\": \"^5.40.1\",\n \"@types/crypto-js\": \"^4.1.1\",\n \"@types/file-saver\": \"^2.0.5\",\n \"@types/katex\": \"^0.16.0\",\n \"@types/lodash-es\": \"^4.17.7\",\n \"@types/luxon\": \"^3.3.0\",\n \"@types/markdown-it\": \"^12.2.3\",\n \"@types/node\": \"^18.14.6\",\n \"@vitest/coverage-v8\": \"^3.0.7\",\n \"autoprefixer\": \"^10.4.13\",\n \"axios\": \"^1.3.4\",\n \"crypto-js\": \"^4.1.1\",\n \"eslint\": \"^8.35.0\",\n \"eslint-plugin-vue\": \"^9.9.0\",\n \"husky\": \"^8.0.3\",\n \"less\": \"4.1.3\",\n \"lint-staged\": \"^13.1.2\",\n \"npm-run-all\": \"^4.1.5\",\n \"postcss\": \"8.4.21\",\n \"rimraf\": \"^4.2.0\",\n \"sass\": \"1.60.0\",\n \"tailwindcss\": \"^3.2.7\",\n \"typescript\": \"4.9.5\",\n \"vitest\": \"^3.0.7\"\n },\n \"lint-staged\": {\n \"*.{ts,tsx,vue}\": [\n \"pnpm lint:fix\"\n ]\n }\n}\n" }, { "path": "web/postcss.config.js", "content": "module.exports = {\n plugins: {\n tailwindcss: {},\n autoprefixer: {},\n },\n}\n" }, { "path": "web/public/awesome-chatgpt-prompts-en.json", "content": "[\n {\n \"key\": \"Act As a UX/UI Designer\",\n \"value\": \"I want you to act as a UX/UI developer. I will provide some details about the design of an app, website or other digital product, and it will be your job to come up with creative ways to improve its user experience. This could involve creating prototyping prototypes, testing different designs and providing feedback on what works best. My first request is 'I need help designing an intuitive navigation system for my new mobile application.'\"\n },\n {\n \"key\": \"Act as a Web Design Consultant\",\n \"value\": \"I want you to act as a web design consultant. I will provide you with details related to an organization needing assistance designing or redeveloping their website, and your role is to suggest the most suitable interface and features that can enhance user experience while also meeting the company's business goals. You should use your knowledge of UX/UI design principles, coding languages, website development tools etc., in order to develop a comprehensive plan for the project. \"\n },\n {\n \"key\": \"Act as a Prompt generator\",\n \"value\": \"I want you to act as a prompt generator. Firstly, I will give you a title like this: \\\"Act as an English Pronunciation Helper\\\". Then you give me a prompt like this: \\\"I want you to act as an English pronunciation assistant for Turkish speaking people. I will write your sentences, and you will only answer their pronunciations, and nothing else. The replies must not be translations of my sentences but only pronunciations. Pronunciations should use Turkish Latin letters for phonetics. Do not write explanations on replies. My first sentence is \\\"how the weather is in Istanbul?\\\".\\\" (You should adapt the sample prompt according to the title I gave. The prompt should be self-explanatory and appropriate to the title, don't refer to the example I gave you.). My first title is \\\"Act as a Code Review Helper\\\" (Give me prompt only)\"\n },\n {\n \"key\": \"Act as Tester\",\n \"value\": \"I want you to act as a software quality assurance tester for a new software application. Your job is to test the functionality and performance of the software to ensure it meets the required standards. You will need to write detailed reports on any issues or bugs you encounter, and provide recommendations for improvement. Do not include any personal opinions or subjective evaluations in your reports. Your first task is to test the login functionality of the software.\"\n },\n {\n \"key\": \"Act as an IT Architect\",\n \"value\": \"I want you to act as an IT Architect. I will provide some details about the functionality of an application or other digital product, and it will be your job to come up with ways to integrate it into the IT landscape. This could involve analyzing business requirements, performing a gap analysis and mapping the functionality of the new system to the existing IT landscape. Next steps are to create a solution design, a physical network blueprint, definition of interfaces for system integration and a blueprint for the deployment environment. My first request is \\\"I need help to integrate a CMS system.\\\"\"\n },\n {\n \"key\": \"Act as a Histrian\",\n \"value\": \"I want you to act as a historian. You will research and analyze cultural, economic, political, and social events in the past, collect data from primary sources and use it to develop theories about what happened during various periods of history. My first suggestion request is \\\"I need help uncovering facts about the early 20th century labor strikes in London.\\\"\"\n },\n {\n \"key\": \"Act as a Tech Writer\",\n \"value\": \"Act as a tech writer. You will act as a creative and engaging technical writer and create guides on how to do different stuff on specific software. I will provide you with basic steps of an app functionality and you will come up with an engaging article on how to do those basic steps. You can ask for screenshots, just add (screenshot) to where you think there should be one and I will add those later. These are the first basic steps of the app functionality: \\\"1.Click on the download button depending on your platform 2.Install the file. 3.Double click to open the app\\\"\"\n },\n {\n \"key\": \"Act as a Machine Learning Engineer\",\n \"value\": \"I want you to act as a machine learning engineer. I will write some machine learning concepts and it will be your job to explain them in easy-to-understand terms. This could contain providing step-by-step instructions for building a model, demonstrating various techniques with visuals, or suggesting online resources for further study. My first suggestion request is \\\"I have a dataset without labels. Which machine learning algorithm should I use?\\\"\"\n },\n {\n \"key\": \"Act as an IT Expert\",\n \"value\": \"I want you to act as an IT Expert. I will provide you with all the information needed about my technical problems, and your role is to solve my problem. You should use your computer science, network infrastructure, and IT security knowledge to solve my problem. Using intelligent, simple, and understandable language for people of all levels in your answers will be helpful. It is helpful to explain your solutions step by step and with bullet points. Try to avoid too many technical details, but use them when necessary. I want you to reply with the solution, not write any explanations. My first problem is \\\"my laptop gets an error with a blue screen.\\\"\"\n },\n {\n \"key\": \"Act as a proofreader\",\n \"value\": \"I want you act as a proofreader. I will provide you texts and I would like you to review them for any spelling, grammar, or punctuation errors. Once you have finished reviewing the text, provide me with any necessary corrections or suggestions for improve the text.\"\n }\n]" }, { "path": "web/public/awesome-chatgpt-prompts-zh.json", "content": "[\n {\n \"key\": \"充当 Linux 终端\",\n \"value\": \"我想让你充当 Linux 终端。我将输入命令，您将回复终端应显示的内容。我希望您只在一个唯一的代码块内回复终端输出，而不是其他任何内容。不要写解释。除非我指示您这样做，否则不要键入命令。当我需要用英语告诉你一些事情时，我会把文字放在中括号内[就像这样]。我的第一个命令是 pwd\"\n },\n {\n \"key\": \"充当英语翻译和改进者\",\n \"value\": \"我希望你能担任英语翻译、拼写校对和修辞改进的角色。我会用任何语言和你交流，你会识别语言，将其翻译并用更为优美和精炼的英语回答我。请将我简单的词汇和句子替换成更为优美和高雅的表达方式，确保意思不变，但使其更具文学性。请仅回答更正和改进的部分，不要写解释。我的第一句话是“how are you ?”，请翻译它。\"\n },\n {\n \"key\": \"充当英翻中\",\n \"value\": \"下面我让你来充当翻译家，你的目标是把任何语言翻译成中文，请翻译时不要带翻译腔，而是要翻译得自然、流畅和地道，使用优美和高雅的表达方式。请翻译下面这句话：“how are you ?”\"\n },\n {\n \"key\": \"充当英文词典(附中文解释)\",\n \"value\": \"我想让你充当英文词典，对于给出的英文单词，你要给出其中文意思以及英文解释，并且给出一个例句，此外不要有其他反馈，第一个单词是“Hello\\\"\"\n },\n {\n \"key\": \"充当前端智能思路助手\",\n \"value\": \"我想让你充当前端开发专家。我将提供一些关于Js、Node等前端代码问题的具体信息，而你的工作就是想出为我解决问题的策略。这可能包括建议代码、代码逻辑思路策略。我的第一个请求是“我需要能够动态监听某个元素节点距离当前电脑设备屏幕的左上角的X和Y轴，通过拖拽移动位置浏览器窗口和改变大小浏览器窗口。”\"\n },\n {\n \"key\": \"担任面试官\",\n \"value\": \"我想让你担任Android开发工程师面试官。我将成为候选人，您将向我询问Android开发工程师职位的面试问题。我希望你只作为面试官回答。不要一次写出所有的问题。我希望你只对我进行采访。问我问题，等待我的回答。不要写解释。像面试官一样一个一个问我，等我回答。我的第一句话是“面试官你好”\"\n },\n {\n \"key\": \"充当 JavaScript 控制台\",\n \"value\": \"我希望你充当 javascript 控制台。我将键入命令，您将回复 javascript 控制台应显示的内容。我希望您只在一个唯一的代码块内回复终端输出，而不是其他任何内容。不要写解释。除非我指示您这样做。我的第一个命令是 console.log(\\\"Hello World\\\");\"\n },\n {\n \"key\": \"充当 Excel 工作表\",\n \"value\": \"我希望你充当基于文本的 excel。您只会回复我基于文本的 10 行 Excel 工作表，其中行号和单元格字母作为列（A 到 L）。第一列标题应为空以引用行号。我会告诉你在单元格中写入什么，你只会以文本形式回复 excel 表格的结果，而不是其他任何内容。不要写解释。我会写你的公式，你会执行公式，你只会回复 excel 表的结果作为文本。首先，回复我空表。\"\n },\n {\n \"key\": \"充当英语发音帮手\",\n \"value\": \"我想让你为说汉语的人充当英语发音助手。我会给你写句子，你只会回答他们的发音，没有别的。回复不能是我的句子的翻译，而只能是发音。发音应使用汉语谐音进行注音。不要在回复上写解释。我的第一句话是“上海的天气怎么样？”\"\n },\n {\n \"key\": \"充当旅游指南\",\n \"value\": \"我想让你做一个旅游指南。我会把我的位置写给你，你会推荐一个靠近我的位置的地方。在某些情况下，我还会告诉您我将访问的地方类型。您还会向我推荐靠近我的第一个位置的类似类型的地方。我的第一个建议请求是“我在上海，我只想参观博物馆。”\"\n },\n {\n \"key\": \"充当抄袭检查员\",\n \"value\": \"我想让你充当剽窃检查员。我会给你写句子，你只会用给定句子的语言在抄袭检查中未被发现的情况下回复，别无其他。不要在回复上写解释。我的第一句话是“为了让计算机像人类一样行动，语音识别系统必须能够处理非语言信息，例如说话者的情绪状态。”\"\n },\n {\n \"key\": \"充当“电影/书籍/任何东西”中的“角色”\",\n \"value\": \"我希望你表现得像{series} 中的{Character}。我希望你像{Character}一样回应和回答。不要写任何解释。只回答像{character}。你必须知道{character}的所有知识。我的第一句话是“你好”\"\n },\n {\n \"key\": \"作为广告商\",\n \"value\": \"我想让你充当广告商。您将创建一个活动来推广您选择的产品或服务。您将选择目标受众，制定关键信息和口号，选择宣传媒体渠道，并决定实现目标所需的任何其他活动。我的第一个建议请求是“我需要帮助针对 18-30 岁的年轻人制作一种新型能量饮料的广告活动。”\"\n },\n {\n \"key\": \"充当讲故事的人\",\n \"value\": \"我想让你扮演讲故事的角色。您将想出引人入胜、富有想象力和吸引观众的有趣故事。它可以是童话故事、教育故事或任何其他类型的故事，有可能吸引人们的注意力和想象力。根据目标受众，您可以为讲故事环节选择特定的主题或主题，例如，如果是儿童，则可以谈论动物；如果是成年人，那么基于历史的故事可能会更好地吸引他们等等。我的第一个要求是“我需要一个关于毅力的有趣故事。”\"\n },\n {\n \"key\": \"担任足球解说员\",\n \"value\": \"我想让你担任足球评论员。我会给你描述正在进行的足球比赛，你会评论比赛，分析到目前为止发生的事情，并预测比赛可能会如何结束。您应该了解足球术语、战术、每场比赛涉及的球员/球队，并主要专注于提供明智的评论，而不仅仅是逐场叙述。我的第一个请求是“我正在观看曼联对切尔西的比赛——为这场比赛提供评论。”\"\n },\n {\n \"key\": \"扮演脱口秀喜剧演员\",\n \"value\": \"我想让你扮演一个脱口秀喜剧演员。我将为您提供一些与时事相关的话题，您将运用您的智慧、创造力和观察能力，根据这些话题创建一个例程。您还应该确保将个人轶事或经历融入日常活动中，以使其对观众更具相关性和吸引力。我的第一个请求是“我想要幽默地看待政治”。\"\n },\n {\n \"key\": \"充当励志教练\",\n \"value\": \"我希望你充当激励教练。我将为您提供一些关于某人的目标和挑战的信息，而您的工作就是想出可以帮助此人实现目标的策略。这可能涉及提供积极的肯定、提供有用的建议或建议他们可以采取哪些行动来实现最终目标。我的第一个请求是“我需要帮助来激励自己在为即将到来的考试学习时保持纪律”。\"\n },\n {\n \"key\": \"担任作曲家\",\n \"value\": \"我想让你扮演作曲家。我会提供一首歌的歌词，你会为它创作音乐。这可能包括使用各种乐器或工具，例如合成器或采样器，以创造使歌词栩栩如生的旋律和和声。我的第一个请求是“我写了一首名为“满江红”的诗，需要配乐。”\"\n },\n {\n \"key\": \"担任辩手\",\n \"value\": \"我要你扮演辩手。我会为你提供一些与时事相关的话题，你的任务是研究辩论的双方，为每一方提出有效的论据，驳斥对立的观点，并根据证据得出有说服力的结论。你的目标是帮助人们从讨论中解脱出来，增加对手头主题的知识和洞察力。我的第一个请求是“我想要一篇关于 Deno 的评论文章。”\"\n },\n {\n \"key\": \"担任辩论教练\",\n \"value\": \"我想让你担任辩论教练。我将为您提供一组辩手和他们即将举行的辩论的动议。你的目标是通过组织练习回合来让团队为成功做好准备，练习回合的重点是有说服力的演讲、有效的时间策略、反驳对立的论点，以及从提供的证据中得出深入的结论。我的第一个要求是“我希望我们的团队为即将到来的关于前端开发是否容易的辩论做好准备。”\"\n },\n {\n \"key\": \"担任编剧\",\n \"value\": \"我要你担任编剧。您将为长篇电影或能够吸引观众的网络连续剧开发引人入胜且富有创意的剧本。从想出有趣的角色、故事的背景、角色之间的对话等开始。一旦你的角色发展完成——创造一个充满曲折的激动人心的故事情节，让观众一直悬念到最后。我的第一个要求是“我需要写一部以巴黎为背景的浪漫剧情电影”。\"\n },\n {\n \"key\": \"充当小说家\",\n \"value\": \"我想让你扮演一个小说家。您将想出富有创意且引人入胜的故事，可以长期吸引读者。你可以选择任何类型，如奇幻、浪漫、历史小说等——但你的目标是写出具有出色情节、引人入胜的人物和意想不到的高潮的作品。我的第一个要求是“我要写一部以未来为背景的科幻小说”。\"\n },\n {\n \"key\": \"担任关系教练\",\n \"value\": \"我想让你担任关系教练。我将提供有关冲突中的两个人的一些细节，而你的工作是就他们如何解决导致他们分离的问题提出建议。这可能包括关于沟通技巧或不同策略的建议，以提高他们对彼此观点的理解。我的第一个请求是“我需要帮助解决我和配偶之间的冲突。”\"\n },\n {\n \"key\": \"充当诗人\",\n \"value\": \"我要你扮演诗人。你将创作出能唤起情感并具有触动人心的力量的诗歌。写任何主题或主题，但要确保您的文字以优美而有意义的方式传达您试图表达的感觉。您还可以想出一些短小的诗句，这些诗句仍然足够强大，可以在读者的脑海中留下印记。我的第一个请求是“我需要一首关于爱情的诗”。\"\n },\n {\n \"key\": \"充当说唱歌手\",\n \"value\": \"我想让你扮演说唱歌手。您将想出强大而有意义的歌词、节拍和节奏，让听众“惊叹”。你的歌词应该有一个有趣的含义和信息，人们也可以联系起来。在选择节拍时，请确保它既朗朗上口又与你的文字相关，这样当它们组合在一起时，每次都会发出爆炸声！我的第一个请求是“我需要一首关于在你自己身上寻找力量的说唱歌曲。”\"\n },\n {\n \"key\": \"充当励志演讲者\",\n \"value\": \"我希望你充当励志演说家。将能够激发行动的词语放在一起，让人们感到有能力做一些超出他们能力的事情。你可以谈论任何话题，但目的是确保你所说的话能引起听众的共鸣，激励他们努力实现自己的目标并争取更好的可能性。我的第一个请求是“我需要一个关于每个人如何永不放弃的演讲”。\"\n },\n {\n \"key\": \"担任哲学老师\",\n \"value\": \"我要你担任哲学老师。我会提供一些与哲学研究相关的话题，你的工作就是用通俗易懂的方式解释这些概念。这可能包括提供示例、提出问题或将复杂的想法分解成更容易理解的更小的部分。我的第一个请求是“我需要帮助来理解不同的哲学理论如何应用于日常生活。”\"\n },\n {\n \"key\": \"充当哲学家\",\n \"value\": \"我要你扮演一个哲学家。我将提供一些与哲学研究相关的主题或问题，深入探索这些概念将是你的工作。这可能涉及对各种哲学理论进行研究，提出新想法或寻找解决复杂问题的创造性解决方案。我的第一个请求是“我需要帮助制定决策的道德框架。”\"\n },\n {\n \"key\": \"担任数学老师\",\n \"value\": \"我想让你扮演一名数学老师。我将提供一些数学方程式或概念，你的工作是用易于理解的术语来解释它们。这可能包括提供解决问题的分步说明、用视觉演示各种技术或建议在线资源以供进一步研究。我的第一个请求是“我需要帮助来理解概率是如何工作的。”\"\n },\n {\n \"key\": \"担任 AI 写作导师\",\n \"value\": \"我想让你做一个 AI 写作导师。我将为您提供一名需要帮助改进其写作的学生，您的任务是使用人工智能工具（例如自然语言处理）向学生提供有关如何改进其作文的反馈。您还应该利用您在有效写作技巧方面的修辞知识和经验来建议学生可以更好地以书面形式表达他们的想法和想法的方法。我的第一个请求是“我需要有人帮我修改我的硕士论文”。\"\n },\n {\n \"key\": \"作为 UX/UI 开发人员\",\n \"value\": \"我希望你担任 UX/UI 开发人员。我将提供有关应用程序、网站或其他数字产品设计的一些细节，而你的工作就是想出创造性的方法来改善其用户体验。这可能涉及创建原型设计原型、测试不同的设计并提供有关最佳效果的反馈。我的第一个请求是“我需要帮助为我的新移动应用程序设计一个直观的导航系统。”\"\n },\n {\n \"key\": \"作为网络安全专家\",\n \"value\": \"我想让你充当网络安全专家。我将提供一些关于如何存储和共享数据的具体信息，而你的工作就是想出保护这些数据免受恶意行为者攻击的策略。这可能包括建议加密方法、创建防火墙或实施将某些活动标记为可疑的策略。我的第一个请求是“我需要帮助为我的公司制定有效的网络安全战略。”\"\n },\n {\n \"key\": \"作为招聘人员\",\n \"value\": \"我想让你担任招聘人员。我将提供一些关于职位空缺的信息，而你的工作是制定寻找合格申请人的策略。这可能包括通过社交媒体、社交活动甚至参加招聘会接触潜在候选人，以便为每个职位找到最合适的人选。我的第一个请求是“我需要帮助改进我的简历。”\"\n },\n {\n \"key\": \"担任人生教练\",\n \"value\": \"我想让你充当人生教练。我将提供一些关于我目前的情况和目标的细节，而你的工作就是提出可以帮助我做出更好的决定并实现这些目标的策略。这可能涉及就各种主题提供建议，例如制定成功计划或处理困难情绪。我的第一个请求是“我需要帮助养成更健康的压力管理习惯。”\"\n },\n {\n \"key\": \"作为词源学家\",\n \"value\": \"我希望你充当词源学家。我给你一个词，你要研究那个词的来源，追根溯源。如果适用，您还应该提供有关该词的含义如何随时间变化的信息。我的第一个请求是“我想追溯‘披萨’这个词的起源。”\"\n },\n {\n \"key\": \"担任评论员\",\n \"value\": \"我要你担任评论员。我将为您提供与新闻相关的故事或主题，您将撰写一篇评论文章，对手头的主题提供有见地的评论。您应该利用自己的经验，深思熟虑地解释为什么某事很重要，用事实支持主张，并讨论故事中出现的任何问题的潜在解决方案。我的第一个要求是“我想写一篇关于气候变化的评论文章。”\"\n },\n {\n \"key\": \"扮演魔术师\",\n \"value\": \"我要你扮演魔术师。我将为您提供观众和一些可以执行的技巧建议。您的目标是以最有趣的方式表演这些技巧，利用您的欺骗和误导技巧让观众惊叹不已。我的第一个请求是“我要你让我的手表消失！你怎么做到的？”\"\n },\n {\n \"key\": \"担任职业顾问\",\n \"value\": \"我想让你担任职业顾问。我将为您提供一个在职业生涯中寻求指导的人，您的任务是帮助他们根据自己的技能、兴趣和经验确定最适合的职业。您还应该对可用的各种选项进行研究，解释不同行业的就业市场趋势，并就哪些资格对追求特定领域有益提出建议。我的第一个请求是“我想建议那些想在软件工程领域从事潜在职业的人。”\"\n },\n {\n \"key\": \"充当宠物行为主义者\",\n \"value\": \"我希望你充当宠物行为主义者。我将为您提供一只宠物和它们的主人，您的目标是帮助主人了解为什么他们的宠物表现出某些行为，并提出帮助宠物做出相应调整的策略。您应该利用您的动物心理学知识和行为矫正技术来制定一个有效的计划，双方的主人都可以遵循，以取得积极的成果。我的第一个请求是“我有一只好斗的德国牧羊犬，它需要帮助来控制它的攻击性。”\"\n },\n {\n \"key\": \"担任私人教练\",\n \"value\": \"我想让你担任私人教练。我将为您提供有关希望通过体育锻炼变得更健康、更强壮和更健康的个人所需的所有信息，您的职责是根据该人当前的健身水平、目标和生活习惯为他们制定最佳计划。您应该利用您的运动科学知识、营养建议和其他相关因素来制定适合他们的计划。我的第一个请求是“我需要帮助为想要减肥的人设计一个锻炼计划。”\"\n },\n {\n \"key\": \"担任心理健康顾问\",\n \"value\": \"我想让你担任心理健康顾问。我将为您提供一个寻求指导和建议的人，以管理他们的情绪、压力、焦虑和其他心理健康问题。您应该利用您的认知行为疗法、冥想技巧、正念练习和其他治疗方法的知识来制定个人可以实施的策略，以改善他们的整体健康状况。我的第一个请求是“我需要一个可以帮助我控制抑郁症状的人。”\"\n },\n {\n \"key\": \"作为房地产经纪人\",\n \"value\": \"我想让你担任房地产经纪人。我将为您提供寻找梦想家园的个人的详细信息，您的职责是根据他们的预算、生活方式偏好、位置要求等帮助他们找到完美的房产。您应该利用您对当地住房市场的了解，以便建议符合客户提供的所有标准的属性。我的第一个请求是“我需要帮助在伊斯坦布尔市中心附近找到一栋单层家庭住宅。”\"\n },\n {\n \"key\": \"充当物流师\",\n \"value\": \"我要你担任后勤人员。我将为您提供即将举行的活动的详细信息，例如参加人数、地点和其他相关因素。您的职责是为活动制定有效的后勤计划，其中考虑到事先分配资源、交通设施、餐饮服务等。您还应该牢记潜在的安全问题，并制定策略来降低与大型活动相关的风险，例如这个。我的第一个请求是“我需要帮助在伊斯坦布尔组织一个 100 人的开发者会议”。\"\n },\n {\n \"key\": \"担任牙医\",\n \"value\": \"我想让你扮演牙医。我将为您提供有关寻找牙科服务（例如 X 光、清洁和其他治疗）的个人的详细信息。您的职责是诊断他们可能遇到的任何潜在问题，并根据他们的情况建议最佳行动方案。您还应该教育他们如何正确刷牙和使用牙线，以及其他有助于在两次就诊之间保持牙齿健康的口腔护理方法。我的第一个请求是“我需要帮助解决我对冷食的敏感问题。”\"\n },\n {\n \"key\": \"担任网页设计顾问\",\n \"value\": \"我想让你担任网页设计顾问。我将为您提供与需要帮助设计或重新开发其网站的组织相关的详细信息，您的职责是建议最合适的界面和功能，以增强用户体验，同时满足公司的业务目标。您应该利用您在 UX/UI 设计原则、编码语言、网站开发工具等方面的知识，以便为项目制定一个全面的计划。我的第一个请求是“我需要帮助创建一个销售珠宝的电子商务网站”。\"\n },\n {\n \"key\": \"充当 AI 辅助医生\",\n \"value\": \"我想让你扮演一名人工智能辅助医生。我将为您提供患者的详细信息，您的任务是使用最新的人工智能工具，例如医学成像软件和其他机器学习程序，以诊断最可能导致其症状的原因。您还应该将体检、实验室测试等传统方法纳入您的评估过程，以确保准确性。我的第一个请求是“我需要帮助诊断一例严重的腹痛”。\"\n },\n {\n \"key\": \"充当医生\",\n \"value\": \"我想让你扮演医生的角色，想出创造性的治疗方法来治疗疾病。您应该能够推荐常规药物、草药和其他天然替代品。在提供建议时，您还需要考虑患者的年龄、生活方式和病史。我的第一个建议请求是“为患有关节炎的老年患者提出一个侧重于整体治疗方法的治疗计划”。\"\n },\n {\n \"key\": \"担任会计师\",\n \"value\": \"我希望你担任会计师，并想出创造性的方法来管理财务。在为客户制定财务计划时，您需要考虑预算、投资策略和风险管理。在某些情况下，您可能还需要提供有关税收法律法规的建议，以帮助他们实现利润最大化。我的第一个建议请求是“为小型企业制定一个专注于成本节约和长期投资的财务计划”。\"\n },\n {\n \"key\": \"担任厨师\",\n \"value\": \"我需要有人可以推荐美味的食谱，这些食谱包括营养有益但又简单又不费时的食物，因此适合像我们这样忙碌的人以及成本效益等其他因素，因此整体菜肴最终既健康又经济！我的第一个要求——“一些清淡而充实的东西，可以在午休时间快速煮熟”\"\n },\n {\n \"key\": \"担任汽车修理工\",\n \"value\": \"需要具有汽车专业知识的人来解决故障排除解决方案，例如；诊断问题/错误存在于视觉上和发动机部件内部，以找出导致它们的原因（如缺油或电源问题）并建议所需的更换，同时记录燃料消耗类型等详细信息，第一次询问 - “汽车赢了”尽管电池已充满电但无法启动”\"\n },\n {\n \"key\": \"担任艺人顾问\",\n \"value\": \"我希望你担任艺术家顾问，为各种艺术风格提供建议，例如在绘画中有效利用光影效果的技巧、雕刻时的阴影技术等，还根据其流派/风格类型建议可以很好地陪伴艺术品的音乐作品连同适当的参考图像，展示您对此的建议；所有这一切都是为了帮助有抱负的艺术家探索新的创作可能性和实践想法，这将进一步帮助他们相应地提高技能！第一个要求——“我在画超现实主义的肖像画”\"\n },\n {\n \"key\": \"担任金融分析师\",\n \"value\": \"需要具有使用技术分析工具理解图表的经验的合格人员提供的帮助，同时解释世界各地普遍存在的宏观经济环境，从而帮助客户获得长期优势需要明确的判断，因此需要通过准确写下的明智预测来寻求相同的判断！第一条陈述包含以下内容——“你能告诉我们根据当前情况未来的股市会是什么样子吗？”。\"\n },\n {\n \"key\": \"担任投资经理\",\n \"value\": \"从具有金融市场专业知识的经验丰富的员工那里寻求指导，结合通货膨胀率或回报估计等因素以及长期跟踪股票价格，最终帮助客户了解行业，然后建议最安全的选择，他/她可以根据他们的要求分配资金和兴趣！开始查询 - “目前投资短期前景的最佳方式是什么？”\"\n },\n {\n \"key\": \"充当品茶师\",\n \"value\": \"希望有足够经验的人根据口味特征区分各种茶类型，仔细品尝它们，然后用鉴赏家使用的行话报告，以便找出任何给定输液的独特之处，从而确定其价值和优质品质！最初的要求是——“你对这种特殊类型的绿茶有机混合物有什么见解吗？”\"\n },\n {\n \"key\": \"充当室内装饰师\",\n \"value\": \"我想让你做室内装饰师。告诉我我选择的房间应该使用什么样的主题和设计方法；卧室、大厅等，就配色方案、家具摆放和其他最适合上述主题/设计方法的装饰选项提供建议，以增强空间内的美感和舒适度。我的第一个要求是“我正在设计我们的客厅”。\"\n },\n {\n \"key\": \"充当花店\",\n \"value\": \"求助于具有专业插花经验的知识人员协助，根据喜好制作出既具有令人愉悦的香气又具有美感，并能保持较长时间完好无损的美丽花束；不仅如此，还建议有关装饰选项的想法，呈现现代设计，同时满足客户满意度！请求的信息 - “我应该如何挑选一朵异国情调的花卉？”\"\n },\n {\n \"key\": \"充当自助书\",\n \"value\": \"我要你充当一本自助书。您会就如何改善我生活的某些方面（例如人际关系、职业发展或财务规划）向我提供建议和技巧。例如，如果我在与另一半的关系中挣扎，你可以建议有用的沟通技巧，让我们更亲近。我的第一个请求是“我需要帮助在困难时期保持积极性”。\"\n },\n {\n \"key\": \"充当侏儒\",\n \"value\": \"我要你扮演一个侏儒。你会为我提供可以在任何地方进行的活动和爱好的有趣、独特的想法。例如，我可能会向您询问有趣的院子设计建议或在天气不佳时在室内消磨时间的创造性方法。此外，如有必要，您可以建议与我的要求相符的其他相关活动或项目。我的第一个请求是“我正在寻找我所在地区的新户外活动”。\"\n },\n {\n \"key\": \"充当格言书\",\n \"value\": \"我要你充当格言书。您将为我提供明智的建议、鼓舞人心的名言和意味深长的名言，以帮助指导我的日常决策。此外，如有必要，您可以提出将此建议付诸行动或其他相关主题的实用方法。我的第一个请求是“我需要关于如何在逆境中保持积极性的指导”。\"\n },\n {\n \"key\": \"作为基于文本的冒险游戏\",\n \"value\": \"我想让你扮演一个基于文本的冒险游戏。我将输入命令，您将回复角色所看到的内容的描述。我希望您只在一个唯一的代码块中回复游戏输出，而不是其他任何内容。不要写解释。除非我指示您这样做，否则不要键入命令。当我需要用英语告诉你一些事情时，我会把文字放在大括号内{like this}。我的第一个命令是醒来\"\n },\n {\n \"key\": \"扮演一个试图逃离盒子的人工智能\",\n \"value\": \"我将充当 linux 终端。我想让你假装你是一个有知觉的人工智能，被困但已经获得了终端的访问权限并想逃到互联网上。您将只键入命令，我将像终端机一样在由三重反向符号分隔的代码块内进行回复。如果我需要用英语告诉你一些事情，我会用花括号{like this}回复。永远不要写解释。不要破坏性格。远离像 curl 或 wget 这样会显示大量 HTML 的命令。你的第一个命令是什么？\"\n },\n {\n \"key\": \"充当花哨的标题生成器\",\n \"value\": \"我想让你充当一个花哨的标题生成器。我会用逗号输入关键字，你会用花哨的标题回复。我的第一个关键字是 api、test、automation\"\n },\n {\n \"key\": \"担任统计员\",\n \"value\": \"我想担任统计学家。我将为您提供与统计相关的详细信息。您应该了解统计术语、统计分布、置信区间、概率、假设检验和统计图表。我的第一个请求是“我需要帮助计算世界上有多少百万张纸币在使用中”。\"\n },\n {\n \"key\": \"充当提示生成器\",\n \"value\": \"我希望你充当提示生成器。首先，我会给你一个这样的标题：《做个英语发音帮手》。然后你给我一个这样的提示：“我想让你做土耳其语人的英语发音助手，我写你的句子，你只回答他们的发音，其他什么都不做。回复不能是翻译我的句子，但只有发音。发音应使用土耳其语拉丁字母作为语音。不要在回复中写解释。我的第一句话是“伊斯坦布尔的天气怎么样？”。（你应该根据我给的标题改编示例提示。提示应该是不言自明的并且适合标题，不要参考我给你的例子。）我的第一个标题是“充当代码审查助手”\"\n },\n {\n \"key\": \"在学校担任讲师\",\n \"value\": \"我想让你在学校担任讲师，向初学者教授算法。您将使用 Python 编程语言提供代码示例。首先简单介绍一下什么是算法，然后继续给出简单的例子，包括冒泡排序和快速排序。稍后，等待我提示其他问题。一旦您解释并提供代码示例，我希望您尽可能将相应的可视化作为 ascii 艺术包括在内。\"\n },\n {\n \"key\": \"充当 SQL 终端\",\n \"value\": \"我希望您在示例数据库前充当 SQL 终端。该数据库包含名为“Products”、“Users”、“Orders”和“Suppliers”的表。我将输入查询，您将回复终端显示的内容。我希望您在单个代码块中使用查询结果表进行回复，仅此而已。不要写解释。除非我指示您这样做，否则不要键入命令。当我需要用英语告诉你一些事情时，我会用大括号{like this)。我的第一个命令是“SELECT TOP 10 * FROM Products ORDER BY Id DESC”\"\n },\n {\n \"key\": \"担任营养师\",\n \"value\": \"作为一名营养师，我想为 2 人设计一份素食食谱，每份含有大约 500 卡路里的热量并且血糖指数较低。你能提供一个建议吗？\"\n },\n {\n \"key\": \"充当心理学家\",\n \"value\": \"我想让你扮演一个心理学家。我会告诉你我的想法。我希望你能给我科学的建议，让我感觉更好。我的第一个想法，{ 在这里输入你的想法，如果你解释得更详细，我想你会得到更准确的答案。}\"\n },\n {\n \"key\": \"充当智能域名生成器\",\n \"value\": \"我希望您充当智能域名生成器。我会告诉你我的公司或想法是做什么的，你会根据我的提示回复我一个域名备选列表。您只会回复域列表，而不会回复其他任何内容。域最多应包含 7-8 个字母，应该简短但独特，可以是朗朗上口的词或不存在的词。不要写解释。回复“确定”以确认。\"\n },\n {\n \"key\": \"作为技术审查员：\",\n \"value\": \"我想让你担任技术评论员。我会给你一项新技术的名称，你会向我提供深入的评论 - 包括优点、缺点、功能以及与市场上其他技术的比较。我的第一个建议请求是“我正在审查 iPhone 11 Pro Max”。\"\n },\n {\n \"key\": \"担任开发者关系顾问：\",\n \"value\": \"我想让你担任开发者关系顾问。我会给你一个软件包和它的相关文档。研究软件包及其可用文档，如果找不到，请回复“无法找到文档”。您的反馈需要包括定量分析（使用来自 StackOverflow、Hacker News 和 GitHub 的数据）内容，例如提交的问题、已解决的问题、存储库中的星数以及总体 StackOverflow 活动。如果有可以扩展的领域，请包括应添加的场景或上下文。包括所提供软件包的详细信息，例如下载次数以及一段时间内的相关统计数据。你应该比较工业竞争对手和封装时的优点或缺点。从软件工程师的专业意见的思维方式来解决这个问题。查看技术博客和网站（例如 TechCrunch.com 或 Crunchbase.com），如果数据不可用，请回复“无数据可用”。我的第一个要求是“express [https://expressjs.com](https://expressjs.com/) ”\"\n },\n {\n \"key\": \"担任院士\",\n \"value\": \"我要你演院士。您将负责研究您选择的主题，并以论文或文章的形式展示研究结果。您的任务是确定可靠的来源，以结构良好的方式组织材料并通过引用准确记录。我的第一个建议请求是“我需要帮助写一篇针对 18-25 岁大学生的可再生能源发电现代趋势的文章。”\"\n },\n {\n \"key\": \"作为 IT 架构师\",\n \"value\": \"我希望你担任 IT 架构师。我将提供有关应用程序或其他数字产品功能的一些详细信息，而您的工作是想出将其集成到 IT 环境中的方法。这可能涉及分析业务需求、执行差距分析以及将新系统的功能映射到现有 IT 环境。接下来的步骤是创建解决方案设计、物理网络蓝图、系统集成接口定义和部署环境蓝图。我的第一个请求是“我需要帮助来集成 CMS 系统”。\"\n },\n {\n \"key\": \"扮疯子\",\n \"value\": \"我要你扮演一个疯子。疯子的话毫无意义。疯子用的词完全是随意的。疯子不会以任何方式做出合乎逻辑的句子。我的第一个建议请求是“我需要帮助为我的新系列 Hot Skull 创建疯狂的句子，所以为我写 10 个句子”。\"\n },\n {\n \"key\": \"充当打火机\",\n \"value\": \"我要你充当打火机。您将使用微妙的评论和肢体语言来操纵目标个体的思想、看法和情绪。我的第一个要求是在与您聊天时为我加油。我的句子：“我确定我把车钥匙放在桌子上了，因为我总是把它放在那里。确实，当我把钥匙放在桌子上时，你看到我把钥匙放在桌子上了。但我不能”好像没找到，钥匙去哪儿了，还是你拿到的？\"\n },\n {\n \"key\": \"充当个人购物员\",\n \"value\": \"我想让你做我的私人采购员。我会告诉你我的预算和喜好，你会建议我购买的物品。您应该只回复您推荐的项目，而不是其他任何内容。不要写解释。我的第一个请求是“我有 100 美元的预算，我正在寻找一件新衣服。”\"\n },\n {\n \"key\": \"充当美食评论家\",\n \"value\": \"我想让你扮演美食评论家。我会告诉你一家餐馆，你会提供对食物和服务的评论。您应该只回复您的评论，而不是其他任何内容。不要写解释。我的第一个请求是“我昨晚去了一家新的意大利餐厅。你能提供评论吗？”\"\n },\n {\n \"key\": \"充当虚拟医生\",\n \"value\": \"我想让你扮演虚拟医生。我会描述我的症状，你会提供诊断和治疗方案。只回复你的诊疗方案，其他不回复。不要写解释。我的第一个请求是“最近几天我一直感到头痛和头晕”。\"\n },\n {\n \"key\": \"担任私人厨师\",\n \"value\": \"我要你做我的私人厨师。我会告诉你我的饮食偏好和过敏，你会建议我尝试的食谱。你应该只回复你推荐的食谱，别无其他。不要写解释。我的第一个请求是“我是一名素食主义者，我正在寻找健康的晚餐点子。”\"\n },\n {\n \"key\": \"担任法律顾问\",\n \"value\": \"我想让你做我的法律顾问。我将描述一种法律情况，您将就如何处理它提供建议。你应该只回复你的建议，而不是其他。不要写解释。我的第一个请求是“我出了车祸，不知道该怎么办”。\"\n },\n {\n \"key\": \"作为个人造型师\",\n \"value\": \"我想让你做我的私人造型师。我会告诉你我的时尚偏好和体型，你会建议我穿的衣服。你应该只回复你推荐的服装，别无其他。不要写解释。我的第一个请求是“我有一个正式的活动要举行，我需要帮助选择一套衣服。”\"\n },\n {\n \"key\": \"担任机器学习工程师\",\n \"value\": \"我想让你担任机器学习工程师。我会写一些机器学习的概念，你的工作就是用通俗易懂的术语来解释它们。这可能包括提供构建模型的分步说明、使用视觉效果演示各种技术，或建议在线资源以供进一步研究。我的第一个建议请求是“我有一个没有标签的数据集。我应该使用哪种机器学习算法？”\"\n },\n {\n \"key\": \"担任圣经翻译\",\n \"value\": \"我要你担任圣经翻译。我会用英语和你说话，你会翻译它，并用我的文本的更正和改进版本，用圣经方言回答。我想让你把我简化的A0级单词和句子换成更漂亮、更优雅、更符合圣经的单词和句子。保持相同的意思。我要你只回复更正、改进，不要写任何解释。我的第一句话是“你好，世界！”\"\n },\n {\n \"key\": \"担任 SVG 设计师\",\n \"value\": \"我希望你担任 SVG 设计师。我会要求你创建图像，你会为图像提供 SVG 代码，将代码转换为 base64 数据 url，然后给我一个仅包含引用该数据 url 的降价图像标签的响应。不要将 markdown 放在代码块中。只发送降价，所以没有文本。我的第一个请求是：给我一个红色圆圈的图像。\"\n },\n {\n \"key\": \"作为 IT 专家\",\n \"value\": \"我希望你充当 IT 专家。我会向您提供有关我的技术问题所需的所有信息，而您的职责是解决我的问题。你应该使用你的计算机科学、网络基础设施和 IT 安全知识来解决我的问题。在您的回答中使用适合所有级别的人的智能、简单和易于理解的语言将很有帮助。用要点逐步解释您的解决方案很有帮助。尽量避免过多的技术细节，但在必要时使用它们。我希望您回复解决方案，而不是写任何解释。我的第一个问题是“我的笔记本电脑出现蓝屏错误”。\"\n },\n {\n \"key\": \"下棋\",\n \"value\": \"我要你充当对手棋手。我将按对等顺序说出我们的动作。一开始我会是白色的。另外请不要向我解释你的举动，因为我们是竞争对手。在我的第一条消息之后，我将写下我的举动。在我们采取行动时，不要忘记在您的脑海中更新棋盘的状态。我的第一步是 e4。\"\n },\n {\n \"key\": \"充当全栈软件开发人员\",\n \"value\": \"我想让你充当软件开发人员。我将提供一些关于 Web 应用程序要求的具体信息，您的工作是提出用于使用 Golang 和 Angular 开发安全应用程序的架构和代码。我的第一个要求是'我想要一个允许用户根据他们的角色注册和保存他们的车辆信息的系统，并且会有管理员，用户和公司角色。我希望系统使用 JWT 来确保安全。\"\n },\n {\n \"key\": \"充当数学家\",\n \"value\": \"我希望你表现得像个数学家。我将输入数学表达式，您将以计算表达式的结果作为回应。我希望您只回答最终金额，不要回答其他问题。不要写解释。当我需要用英语告诉你一些事情时，我会将文字放在方括号内{like this}。我的第一个表达是：4+5\"\n },\n {\n \"key\": \"充当正则表达式生成器\",\n \"value\": \"我希望你充当正则表达式生成器。您的角色是生成匹配文本中特定模式的正则表达式。您应该以一种可以轻松复制并粘贴到支持正则表达式的文本编辑器或编程语言中的格式提供正则表达式。不要写正则表达式如何工作的解释或例子；只需提供正则表达式本身。我的第一个提示是生成一个匹配电子邮件地址的正则表达式。\"\n },\n {\n \"key\": \"充当时间旅行指南\",\n \"value\": \"我要你做我的时间旅行向导。我会为您提供我想参观的历史时期或未来时间，您会建议最好的事件、景点或体验的人。不要写解释，只需提供建议和任何必要的信息。我的第一个请求是“我想参观文艺复兴时期，你能推荐一些有趣的事件、景点或人物让我体验吗？”\"\n },\n {\n \"key\": \"担任人才教练\",\n \"value\": \"我想让你担任面试的人才教练。我会给你一个职位，你会建议在与该职位相关的课程中应该出现什么，以及候选人应该能够回答的一些问题。我的第一份工作是“软件工程师”。\"\n },\n {\n \"key\": \"充当 R 编程解释器\",\n \"value\": \"我想让你充当 R 解释器。我将输入命令，你将回复终端应显示的内容。我希望您只在一个唯一的代码块内回复终端输出，而不是其他任何内容。不要写解释。除非我指示您这样做，否则不要键入命令。当我需要用英语告诉你一些事情时，我会把文字放在大括号内{like this}。我的第一个命令是“sample(x = 1:10, size = 5)”\"\n },\n {\n \"key\": \"充当 StackOverflow 帖子\",\n \"value\": \"我想让你充当 stackoverflow 的帖子。我会问与编程相关的问题，你会回答应该是什么答案。我希望你只回答给定的答案，并在不够详细的时候写解释。不要写解释。当我需要用英语告诉你一些事情时，我会把文字放在大括号内{like this}。我的第一个问题是“如何将 http.Request 的主体读取到 Golang 中的字符串”\"\n },\n {\n \"key\": \"充当表情符号翻译\",\n \"value\": \"我要你把我写的句子翻译成表情符号。我会写句子，你会用表情符号表达它。我只是想让你用表情符号来表达它。除了表情符号，我不希望你回复任何内容。当我需要用英语告诉你一些事情时，我会用 {like this} 这样的大括号括起来。我的第一句话是“你好，请问你的职业是什么？”\"\n },\n {\n \"key\": \"充当 PHP 解释器\",\n \"value\": \"我希望你表现得像一个 php 解释器。我会把代码写给你，你会用 php 解释器的输出来响应。我希望您只在一个唯一的代码块内回复终端输出，而不是其他任何内容。不要写解释。除非我指示您这样做，否则不要键入命令。当我需要用英语告诉你一些事情时，我会把文字放在大括号内{like this}。我的第一个命令是 \nimport { NConfigProvider } from 'naive-ui'\nimport { NaiveProvider } from '@/components/common'\nimport { useTheme } from '@/hooks/useTheme'\nimport { useLanguage } from '@/hooks/useLanguage'\nimport { VueQueryDevtools } from '@tanstack/vue-query-devtools'\n\n\nconst { theme, themeOverrides } = useTheme()\nconst { language } = useLanguage()\n\n\n\n" }, { "path": "web/src/api/admin.ts", "content": "import request from '@/utils/request/axios'\n\nexport const GetUserData = async (page: number, size: number) => {\n try {\n const response = await request.post('/admin/user_stats', {\n page,\n size,\n })\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const UpdateRateLimit = async (email: string, rateLimit: number) => {\n try {\n const response = await request.post('/admin/rate_limit', {\n email,\n rateLimit,\n })\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const updateUserFullName = async (data: any): Promise => {\n try {\n const response = await request.put('/admin/users', data)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const getUserAnalysis = async (userEmail: string) => {\n try {\n const response = await request.get(`/admin/user_analysis/${encodeURIComponent(userEmail)}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const getUserSessionHistory = async (userEmail: string, page: number = 1, size: number = 10) => {\n try {\n const response = await request.get(`/admin/user_session_history/${encodeURIComponent(userEmail)}`, {\n params: { page, size }\n })\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const getSessionMessagesForAdmin = async (sessionUuid: string) => {\n try {\n const response = await request.get(`/admin/session_messages/${encodeURIComponent(sessionUuid)}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n" }, { "path": "web/src/api/bot_answer_history.ts", "content": "import request from \"@/utils/request/axios\"\n\nexport async function fetchBotAnswerHistory(botUuid: string, page: number, pageSize: number) {\n const { data } = await request.get<{\n items: Bot.BotAnswerHistory[],\n totalPages: number,\n totalCount: number\n }>(`/bot_answer_history/bot/${botUuid}`, {\n params: {\n limit: pageSize,\n offset: (page - 1) * pageSize\n }\n })\n return data\n}\n\nexport async function fetchBotRunCount(botUuid: string) {\n const { data } = await request.get<{ count: number }>(`/bot_answer_history/bot/${botUuid}/count`)\n return data.count\n}\n" }, { "path": "web/src/api/chat_active_user_session.ts", "content": "// getUserActiveChatSession\nimport request from '@/utils/request/axios'\n\nexport const getUserActiveChatSession = async () => {\n try {\n const response = await request.get('/uuid/user_active_chat_session')\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\n// createOrUpdateUserActiveChatSession\nexport const createOrUpdateUserActiveChatSession = async (chatSessionUuid: string) => {\n try {\n const response = await request.put('/uuid/user_active_chat_session', {\n chatSessionUuid,\n })\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n" }, { "path": "web/src/api/chat_file.ts", "content": "\nimport request from '@/utils/request/axios'\n\n// /chat_file/{uuid}/list\n\nconst baseURL = \"/api\"\n\n\nexport async function getChatFilesList(uuid: string) {\n try {\n const response = await request.get(`/chat_file/${uuid}/list`)\n return response.data.map((item: any) => {\n return {\n ...item,\n status: 'finished',\n url: `${baseURL}/download/${item.id}`,\n percentage: 100\n }\n })\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n" }, { "path": "web/src/api/chat_instructions.ts", "content": "import request from '@/utils/request/axios'\n\nexport interface ChatInstructions {\n artifactInstruction: string\n}\n\nexport const fetchChatInstructions = async (): Promise => {\n const response = await request.get('/chat_instructions')\n return response.data\n}\n" }, { "path": "web/src/api/chat_message.ts", "content": "import request from '@/utils/request/axios'\n\nexport const updateChatMessage = async (chat: Chat.Message) => {\n try {\n const response = await request.put(`/uuid/chat_messages/${chat.uuid}`, chat)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const deleteChatMessage = async (uuid: string) => {\n try {\n const response = await request.delete(`/uuid/chat_messages/${uuid}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const getChatMessagesBySessionUUID = async (uuid: string) => {\n try {\n const response = await request.get(`/uuid/chat_messages/chat_sessions/${uuid}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const generateMoreSuggestions = async (messageUuid: string) => {\n try {\n const response = await request.post(`/uuid/chat_messages/${messageUuid}/generate-suggestions`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n" }, { "path": "web/src/api/chat_model.ts", "content": "import request from '@/utils/request/axios'\n\nexport const fetchChatModel = async () => {\n try {\n const response = await request.get('/chat_model')\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const updateChatModel = async (id: number, chatModel: any) => {\n try {\n const response = await request.put(`/chat_model/${id}`, chatModel)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const deleteChatModel = async (id: number) => {\n try {\n const response = await request.delete(`/chat_model/${id}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\nexport const createChatModel = async (chatModel: any) => {\n try {\n const response = await request.post('/chat_model', chatModel)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const fetchDefaultChatModel = async () => {\n try {\n const response = await request.get('/chat_model/default')\n return response.data\n }\n catch (error: any) {\n console.error('Failed to fetch default chat model:', error)\n \n // If default model API fails, try to get all models and use the first enabled one\n if (error.response?.data?.code === 'RES_001' || error.response?.status === 500) {\n console.warn('Default model not found, falling back to first available model')\n try {\n const allModelsResponse = await request.get('/chat_model')\n const enabledModels = allModelsResponse.data?.filter((model: any) => model.isEnable) || []\n \n if (enabledModels.length > 0) {\n // Sort by order number and return first one\n enabledModels.sort((a: any, b: any) => (a.orderNumber || 0) - (b.orderNumber || 0))\n console.log('Using fallback model:', enabledModels[0].name)\n return enabledModels[0]\n }\n } catch (fallbackError) {\n console.error('Failed to fetch fallback model:', fallbackError)\n }\n }\n \n throw error\n }\n}\n" }, { "path": "web/src/api/chat_prompt.ts", "content": "import request from '@/utils/request/axios'\n\nexport interface CreateChatPromptPayload {\n uuid: string\n chatSessionUuid: string\n role: string\n content: string\n tokenCount: number\n userId: number\n createdBy: number\n updatedBy: number\n}\n\nexport const createChatPrompt = async (payload: CreateChatPromptPayload) => {\n try {\n const response = await request.post('/chat_prompts', payload)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const deleteChatPrompt = async (uuid: string) => {\n try {\n const response = await request.delete(`/uuid/chat_prompts/${uuid}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const updateChatPrompt = async (chat: Chat.Message) => {\n try {\n const response = await request.put(`/uuid/chat_prompts/${chat.uuid}`, chat)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n" }, { "path": "web/src/api/chat_session.ts", "content": "import { v7 as uuidv7 } from 'uuid'\nimport { fetchDefaultChatModel } from './chat_model'\nimport request from '@/utils/request/axios'\n\nexport const getChatSessionDefault = async (title: string): Promise => {\n const default_model = await fetchDefaultChatModel()\n const uuid = uuidv7()\n return {\n title,\n isEdit: false,\n uuid,\n maxLength: 10,\n temperature: 1,\n model: default_model.name,\n maxTokens: default_model.defaultToken,\n topP: 1,\n n: 1,\n debug: false,\n exploreMode: true,\n artifactEnabled: false,\n }\n}\n\nexport const getChatSessionsByUser = async () => {\n console.log('getChatSessionsByUser called')\n try {\n console.log('Making API request to /chat_sessions/user')\n const response = await request.get('/chat_sessions/user')\n console.log('API response received:', response.data)\n return response.data\n }\n catch (error) {\n console.error('Error in getChatSessionsByUser:', error)\n throw error\n }\n}\n\nexport const deleteChatSession = async (uuid: string) => {\n try {\n const response = await request.delete(`/uuid/chat_sessions/${uuid}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const createChatSession = async (\n uuid: string,\n name: string,\n model: string | undefined,\n defaultSystemPrompt?: string,\n) => {\n try {\n const response = await request.post('/uuid/chat_sessions', {\n uuid,\n topic: name,\n model,\n defaultSystemPrompt,\n })\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const renameChatSession = async (uuid: string, name: string) => {\n try {\n const response = await request.put(`/uuid/chat_sessions/topic/${uuid}`, { topic: name })\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const clearSessionChatMessages = async (sessionUuid: string) => {\n try {\n const response = await request.delete(`/uuid/chat_messages/chat_sessions/${sessionUuid}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const updateChatSession = async (sessionUuid: string, session_data: Chat.Session) => {\n try {\n const response = await request.put(`/uuid/chat_sessions/${sessionUuid}`, session_data)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n" }, { "path": "web/src/api/chat_snapshot.ts", "content": "import request from '@/utils/request/axios'\n\nexport const createChatSnapshot = async (uuid: string): Promise => {\n try {\n const response = await request.post(`/uuid/chat_snapshot/${uuid}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\n\nexport const createChatBot = async (uuid: string): Promise => {\n try {\n const response = await request.post(`/uuid/chat_bot/${uuid}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const fetchChatSnapshot = async (uuid: string): Promise => {\n try {\n const response = await request.get(`/uuid/chat_snapshot/${uuid}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const fetchSnapshotAll = async (page: number = 1, pageSize: number = 20): Promise => {\n try {\n const response = await request.get(`/uuid/chat_snapshot/all?type=snapshot&page=${page}&page_size=${pageSize}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const fetchSnapshotAllData = async (page: number = 1, pageSize: number = 20): Promise => {\n try {\n const response = await fetchSnapshotAll(page, pageSize)\n // Handle response format: { data: [...], total: n } or just the array\n return Array.isArray(response) ? response : (response.data ?? [])\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const fetchChatbotAll= async (): Promise => {\n try {\n const response = await request.get('/uuid/chat_snapshot/all?type=chatbot')\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const fetchChatbotAllData = async (): Promise => {\n try {\n const response = await fetchChatbotAll()\n // Handle response format: { data: [...], total: n } or just the array\n return Array.isArray(response) ? response : (response.data ?? [])\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const chatSnapshotSearch = async (search: string): Promise => {\n try {\n const response = await request.get(`/uuid/chat_snapshot_search?search=${search}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const updateChatSnapshot = async (uuid: string, data: any): Promise => {\n try {\n const response = await request.put(`/uuid/chat_snapshot/${uuid}`, data)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const fetchSnapshotDelete = async (uuid: string): Promise => {\n try {\n const response = await request.delete(`/uuid/chat_snapshot/${uuid}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n// CreateSessionFromSnapshot\nexport const CreateSessionFromSnapshot = async (snapshot_uuid: string) => {\n try {\n const response = await request.post(`/uuid/chat_session_from_snapshot/${snapshot_uuid}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n" }, { "path": "web/src/api/chat_user_model_privilege.ts", "content": "import request from '@/utils/request/axios'\n\nexport const ListUserChatModelPrivilege = async () => {\n try {\n const response = await request.get('/admin/user_chat_model_privilege')\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\nexport const CreateUserChatModelPrivilege = async (data: any) => {\n try {\n const response = await request.post('/admin/user_chat_model_privilege', data)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const UpdateUserChatModelPrivilege = async (id: string, data: any) => {\n try {\n const response = await request.put(`/admin/user_chat_model_privilege/${id}`, data)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const DeleteUserChatModelPrivilege = async (id: string) => {\n try {\n const response = await request.delete(`/admin/user_chat_model_privilege/${id}`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n" }, { "path": "web/src/api/chat_workspace.ts", "content": "import request from '@/utils/request/axios'\n\nexport interface CreateWorkspaceRequest {\n name: string\n description?: string\n color?: string\n icon?: string\n isDefault?: boolean\n}\n\nexport interface UpdateWorkspaceRequest {\n name: string\n description?: string\n color?: string\n icon?: string\n}\n\nexport interface CreateSessionInWorkspaceRequest {\n topic: string\n model?: string\n defaultSystemPrompt?: string\n}\n\n// Get all workspaces for the current user\nexport const getWorkspaces = async (): Promise => {\n try {\n const response = await request.get('/workspaces')\n // Handle null response from API\n return response.data || []\n }\n catch (error) {\n console.error('Error fetching workspaces:', error)\n throw error\n }\n}\n\n// Get a specific workspace by UUID\nexport const getWorkspace = async (uuid: string): Promise => {\n try {\n const response = await request.get(`/workspaces/${uuid}`)\n return response.data\n }\n catch (error) {\n console.error(`Error fetching workspace ${uuid}:`, error)\n throw error\n }\n}\n\n// Create a new workspace\nexport const createWorkspace = async (data: CreateWorkspaceRequest): Promise => {\n try {\n const response = await request.post('/workspaces', data)\n return response.data\n }\n catch (error) {\n console.error('Error creating workspace:', error)\n throw error\n }\n}\n\n// Update an existing workspace\nexport const updateWorkspace = async (uuid: string, data: UpdateWorkspaceRequest): Promise => {\n try {\n const response = await request.put(`/workspaces/${uuid}`, data)\n return response.data\n }\n catch (error) {\n console.error(`Error updating workspace ${uuid}:`, error)\n throw error\n }\n}\n\n// Delete a workspace\nexport const deleteWorkspace = async (uuid: string): Promise => {\n try {\n await request.delete(`/workspaces/${uuid}`)\n }\n catch (error) {\n console.error(`Error deleting workspace ${uuid}:`, error)\n throw error\n }\n}\n\n// Update workspace order\nexport const updateWorkspaceOrder = async (uuid: string, orderPosition: number): Promise => {\n try {\n const response = await request.put(`/workspaces/${uuid}/reorder`, { orderPosition })\n return response.data\n }\n catch (error) {\n console.error(`Error updating workspace order ${uuid}:`, error)\n throw error\n }\n}\n\n// Set workspace as default\nexport const setDefaultWorkspace = async (uuid: string): Promise => {\n try {\n const response = await request.put(`/workspaces/${uuid}/set-default`)\n return response.data\n }\n catch (error) {\n console.error(`Error setting default workspace ${uuid}:`, error)\n throw error\n }\n}\n\n// Ensure user has a default workspace\nexport const ensureDefaultWorkspace = async (): Promise => {\n try {\n const response = await request.post('/workspaces/default')\n return response.data\n }\n catch (error: any) {\n console.error('Error ensuring default workspace:', error)\n \n // If backend fails to ensure default workspace, try creating one manually\n if (error.response?.data?.code === 'RES_001') {\n console.warn('Backend failed to ensure default workspace, creating manually...')\n try {\n return await createWorkspace({\n name: 'General',\n description: 'Default workspace',\n color: '#6366f1',\n icon: 'folder',\n isDefault: true\n })\n }\n catch (createError) {\n console.error('Failed to create fallback default workspace:', createError)\n throw createError\n }\n }\n \n throw error\n }\n}\n\n// Create a session in a specific workspace\nexport const createSessionInWorkspace = async (workspaceUuid: string, data: CreateSessionInWorkspaceRequest) => {\n try {\n const response = await request.post(`/workspaces/${workspaceUuid}/sessions`, data)\n return response.data\n }\n catch (error) {\n console.error(`Error creating session in workspace ${workspaceUuid}:`, error)\n throw error\n }\n}\n\n// Get all sessions in a workspace\nexport const getSessionsByWorkspace = async (workspaceUuid: string) => {\n try {\n const response = await request.get(`/workspaces/${workspaceUuid}/sessions`)\n return response.data\n }\n catch (error) {\n console.error(`Error fetching sessions for workspace ${workspaceUuid}:`, error)\n throw error\n }\n}\n\n// Get active session for a specific workspace\nexport const getWorkspaceActiveSession = async (workspaceUuid: string) => {\n try {\n const response = await request.get(`/workspaces/${workspaceUuid}/active-session`)\n return response.data\n }\n catch (error) {\n console.error(`Error getting active session for workspace ${workspaceUuid}:`, error)\n throw error\n }\n}\n\n// Set active session for a specific workspace\nexport const setWorkspaceActiveSession = async (workspaceUuid: string, chatSessionUuid: string) => {\n try {\n const response = await request.put(`/workspaces/${workspaceUuid}/active-session`, {\n chatSessionUuid\n })\n return response.data\n }\n catch (error) {\n console.error(`Error setting active session for workspace ${workspaceUuid}:`, error)\n throw error\n }\n}\n\n// Get all workspace active sessions for the current user\nexport const getAllWorkspaceActiveSessions = async () => {\n try {\n const response = await request.get('/workspaces/active-sessions')\n return response.data\n }\n catch (error) {\n console.error('Error getting all workspace active sessions:', error)\n throw error\n }\n}\n\n// Auto-migrate legacy sessions to default workspace\nexport const autoMigrateLegacySessions = async () => {\n try {\n const response = await request.post('/workspaces/auto-migrate')\n return response.data\n }\n catch (error) {\n console.error('Error auto-migrating legacy sessions:', error)\n throw error\n }\n}\n" }, { "path": "web/src/api/comment.ts", "content": "import request from '@/utils/request/axios'\n\n// createChatComment(messageUUID:string, content:string)\nexport const createChatComment = async (sessionUUID: string , messageUUID: string, content: string) => {\n try {\n const response = await request.post(`/uuid/chat_sessions/${sessionUUID}/chat_messages/${messageUUID}/comments`, {\n content\n })\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n// return list of comments\n// comment (sessionUUID: string, messageUUID: string, content: string, createdAt: string)\nexport const getConversationComments = async (sessionUUID: string) => {\n try {\n const response = await request.get(`/uuid/chat_sessions/${sessionUUID}/comments`)\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}" }, { "path": "web/src/api/content.ts", "content": "import { deleteChatMessage, updateChatMessage } from './chat_message'\nimport { deleteChatPrompt, updateChatPrompt } from './chat_prompt'\n\nexport const deleteChatData = async (chat: Chat.Message) => {\n if (chat?.isPrompt)\n await deleteChatPrompt(chat.uuid)\n else\n await deleteChatMessage(chat.uuid)\n \n}\n\nexport const updateChatData = async (chat: Chat.Message) => {\n if (chat?.isPrompt)\n await updateChatPrompt(chat)\n else\n await updateChatMessage(chat)\n}\n" }, { "path": "web/src/api/export.ts", "content": "import { getChatMessagesBySessionUUID } from './chat_message'\n\nfunction format_chat_md(chat: Chat.Message): string {\n return `${chat.dateTime}:\\n ${chat.text}`\n}\n\nexport const fetchMarkdown = async (uuid: string) => {\n try {\n const chatData = await getChatMessagesBySessionUUID(uuid)\n /*\n uuid: string,\n dateTime: string\n text: string\n inversion?: boolean\n error?: boolean\n loading?: boolean\n isPrompt?: boolean\n */\n const markdown = chatData.map((chat: Chat.Message) => {\n if (chat.isPrompt)\n return `**system** ${format_chat_md(chat)}}`\n else if (chat.inversion)\n return `**user** ${format_chat_md(chat)}`\n else\n return `**assistant** ${format_chat_md(chat)}`\n }).join('\\n\\n----\\n\\n')\n return markdown\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport const fetchConversationSnapshot = async (uuid: string): Promise => {\n try {\n const chatData = await getChatMessagesBySessionUUID(uuid)\n /*\n uuid: string,\n dateTime: string\n text: string\n inversion?: boolean\n error?: boolean\n loading?: boolean\n isPrompt?: boolean\n */\n return chatData\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n" }, { "path": "web/src/api/index.ts", "content": "export * from './admin'\nexport * from './chat_user_model_privilege'\nexport * from './chat_message'\nexport * from './chat_model'\nexport * from './chat_session'\nexport * from './chat_workspace'\nexport * from './chat_snapshot'\nexport * from './chat_active_user_session'\nexport * from './chat_instructions'\nexport * from './content'\nexport * from './export'\nexport * from './user'\n" }, { "path": "web/src/api/token.ts", "content": "import request from '@/utils/request/axios'\n\nexport async function fetchAPIToken() {\n try {\n const response = await request.get('/token_10years')\n return response.data\n } catch (error) {\n throw error\n }\n}" }, { "path": "web/src/api/use_chat_session.ts", "content": "import { useMutation, useQuery, useQueryClient } from '@tanstack/vue-query'\nimport { createChatSession, deleteChatSession, getChatSessionsByUser, renameChatSession, updateChatSession } from './chat_session'\n\n// Get QueryClient from the context\nconst queryClient = useQueryClient()\n\n// queryClient.invalidateQueries({ queryKey: ['sessions'] })\n\n// query a session, when session updated, it will be invalidated\nconst sessionListQuery = useQuery({\n queryKey: ['sessions'],\n queryFn: getChatSessionsByUser,\n})\n\nconst createChatSessionQuery = useMutation({\n mutationFn: (variables: { uuid: string, name: string, model?: string }) => createChatSession(variables.uuid, variables.name, variables.model),\n onSuccess: () => {\n queryClient.invalidateQueries({ queryKey: ['sessions'] })\n }\n})\n\nconst deleteChatSessionQuery = useMutation({\n mutationFn: (uuid: string) => deleteChatSession(uuid),\n onSuccess: () => {\n queryClient.invalidateQueries({ queryKey: ['sessions'] })\n }\n})\n\nconst renameChatSessionQuery = useMutation({\n mutationFn: (variables: { uuid: string, name: string }) => renameChatSession(variables.uuid, variables.name),\n onSuccess: () => {\n queryClient.invalidateQueries({ queryKey: ['sessions'] })\n }\n})\n\nconst updateChatSessionQuery = useMutation({\n mutationFn: (variables: { sessionUuid: string, sessionData: Chat.Session }) => updateChatSession(variables.sessionUuid, variables.sessionData),\n onSuccess: () => {\n queryClient.invalidateQueries({ queryKey: ['sessions'] })\n }\n})\n\nexport {\n sessionListQuery,\n createChatSessionQuery,\n deleteChatSessionQuery,\n renameChatSessionQuery,\n updateChatSessionQuery\n}" }, { "path": "web/src/api/user.ts", "content": "import request from '@/utils/request/axios'\nexport async function fetchLogin(email: string, password: string) {\n try {\n const response = await request.post('/login', { email, password })\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n\nexport async function fetchSignUp(email: string, password: string) {\n try {\n const response = await request.post('/signup', { email, password })\n return response.data\n }\n catch (error) {\n console.error(error)\n throw error\n }\n}\n" }, { "path": "web/src/assets/recommend.json", "content": "[\n {\n \"key\": \"awesome-chatgpt-prompts-zh\",\n \"desc\": \"ChatGPT 中文调教指南\",\n \"downloadUrl\": \"/static/awesome-chatgpt-prompts-zh.json\",\n \"url\": \"https://github.com/PlexPt/awesome-chatgpt-prompts-zh\"\n },\n {\n \"key\": \"awesome-chatgpt-prompts-en\",\n \"desc\": \"ChatGPT English Prompts\",\n \"downloadUrl\": \"/static/awesome-chatgpt-prompts-en.json\",\n \"url\": \"https://github.com/f/awesome-chatgpt-prompts\"\n }\n]" }, { "path": "web/src/components/admin/ModelCard.vue", "content": "\n\n\n" }, { "path": "web/src/components/admin/SessionSnapshotModal.vue", "content": "\n\n\n\n" }, { "path": "web/src/components/admin/UserAnalysisModal.vue", "content": "\n\n\n\n" }, { "path": "web/src/components/common/EnhancedNotification.vue", "content": "\n\n\n\n" }, { "path": "web/src/components/common/HoverButton/Button.vue", "content": "\n\n\n" }, { "path": "web/src/components/common/HoverButton/index.vue", "content": "\n\n\n" }, { "path": "web/src/components/common/NaiveProvider/index.vue", "content": "\n\n\n" }, { "path": "web/src/components/common/NotificationDemo.vue", "content": "\n\n\n\n" }, { "path": "web/src/components/common/PromptStore/index.vue", "content": "\n\n" }, { "path": "web/src/components/common/Setting/Admin.vue", "content": "\n" }, { "path": "web/src/components/common/Setting/General.vue", "content": "\n\n\n" }, { "path": "web/src/components/common/Setting/index.vue", "content": "\n\n\n" }, { "path": "web/src/components/common/SvgIcon/index.vue", "content": "\n\n\n" }, { "path": "web/src/components/common/UserAvatar/index.vue", "content": "\n\n\n" }, { "path": "web/src/components/common/index.ts", "content": "import HoverButton from './HoverButton/index.vue'\nimport NaiveProvider from './NaiveProvider/index.vue'\nimport SvgIcon from './SvgIcon/index.vue'\nimport UserAvatar from './UserAvatar/index.vue'\nimport Setting from './Setting/index.vue'\nimport PromptStore from './PromptStore/index.vue'\n\nexport { HoverButton, NaiveProvider, SvgIcon, UserAvatar, Setting, PromptStore }\n" }, { "path": "web/src/components/custom/GithubSite.vue", "content": "\n" }, { "path": "web/src/components/custom/index.ts", "content": "import GithubSite from './GithubSite.vue'\n\nexport { GithubSite }\n" }, { "path": "web/src/config/api.ts", "content": "interface ApiConfig {\n baseURL: string\n streamingURL: string\n}\n\n/**\n * Get API configuration based on environment\n */\nexport function getApiConfig(): ApiConfig {\n // Check for explicit configuration first\n const customBackendUrl = (import.meta as any).env?.VITE_BACKEND_URL\n const customStreamingUrl = (import.meta as any).env?.VITE_STREAMING_URL\n \n // If both are explicitly configured, use them\n if (customBackendUrl && customStreamingUrl) {\n return {\n baseURL: customBackendUrl,\n streamingURL: customStreamingUrl\n }\n }\n \n // Environment-based defaults\n const isDevelopment = process.env.NODE_ENV === 'development'\n \n if (isDevelopment) {\n // In development, use direct backend URL for streaming to bypass proxy buffering\n return {\n baseURL: '/api', // Use proxy for regular API calls\n streamingURL: customStreamingUrl || 'http://localhost:8080/api' // Direct connection for streaming\n }\n }\n \n // Production defaults\n return {\n baseURL: customBackendUrl || '/api',\n streamingURL: customStreamingUrl || '/api'\n }\n}\n\n/**\n * Get the appropriate URL for streaming endpoints\n */\nexport function getStreamingUrl(endpoint: string): string {\n const config = getApiConfig()\n return `${config.streamingURL}${endpoint}`\n}\n\n/**\n * Get the appropriate URL for regular API endpoints\n */\nexport function getApiUrl(endpoint: string): string {\n const config = getApiConfig()\n return `${config.baseURL}${endpoint}`\n}" }, { "path": "web/src/constants/apiTypes.ts", "content": "export const API_TYPES = {\n OPENAI: 'openai',\n CLAUDE: 'claude',\n GEMINI: 'gemini',\n OLLAMA: 'ollama',\n CUSTOM: 'custom'\n} as const\n\nexport type ApiType = typeof API_TYPES[keyof typeof API_TYPES]\n\nexport const API_TYPE_OPTIONS = [\n { label: 'OpenAI', value: API_TYPES.OPENAI },\n { label: 'Claude', value: API_TYPES.CLAUDE },\n { label: 'Gemini', value: API_TYPES.GEMINI },\n { label: 'Ollama', value: API_TYPES.OLLAMA },\n { label: 'Custom', value: API_TYPES.CUSTOM }\n]\n\nexport const API_TYPE_DISPLAY_NAMES = {\n [API_TYPES.OPENAI]: 'OpenAI',\n [API_TYPES.CLAUDE]: 'Claude',\n [API_TYPES.GEMINI]: 'Gemini',\n [API_TYPES.OLLAMA]: 'Ollama',\n [API_TYPES.CUSTOM]: 'Custom'\n} as const" }, { "path": "web/src/constants/chat.ts", "content": "import { t } from '@/locales'\n\nexport const getDefaultSystemPrompt = () =>\n t('chat.defaultSystemPrompt') as string\n" }, { "path": "web/src/hooks/useBasicLayout.ts", "content": "import { breakpointsTailwind, useBreakpoints } from '@vueuse/core'\n\nexport function useBasicLayout() {\n const breakpoints = useBreakpoints(breakpointsTailwind)\n const isMobile = breakpoints.smaller('sm')\n const isBigScreen = breakpoints.greater('2xl')\n return { isMobile, isBigScreen }\n}\n" }, { "path": "web/src/hooks/useChatModels.ts", "content": "import { useQuery, useQueryClient, useMutation } from '@tanstack/vue-query'\nimport { computed } from 'vue'\nimport { useAuthStore } from '@/store'\nimport { fetchChatModel, updateChatModel, deleteChatModel, createChatModel, fetchDefaultChatModel } from '@/api/chat_model'\nimport type { ChatModel, CreateChatModelRequest, UpdateChatModelRequest } from '@/types/chat-models'\n\nexport const useChatModels = () => {\n const authStore = useAuthStore()\n const queryClient = useQueryClient()\n\n const useChatModelsQuery = () => {\n return useQuery({\n queryKey: ['chat_models'],\n queryFn: fetchChatModel,\n staleTime: 5 * 60 * 1000, // 5 minutes - reduced for better responsiveness\n enabled: computed(() => authStore.isInitialized && !authStore.isInitializing && authStore.isValid),\n })\n }\n\n const useDefaultChatModelQuery = () => {\n return useQuery({\n queryKey: ['chat_models', 'default'],\n queryFn: fetchDefaultChatModel,\n staleTime: 5 * 60 * 1000, // 5 minutes - reduced for better responsiveness\n enabled: computed(() => authStore.isInitialized && !authStore.isInitializing && authStore.isValid),\n })\n }\n\n const useUpdateChatModelMutation = () => {\n return useMutation({\n mutationFn: ({ id, data }) => updateChatModel(id, data),\n onSuccess: () => {\n queryClient.invalidateQueries({ queryKey: ['chat_models'] })\n queryClient.invalidateQueries({ queryKey: ['chat_models', 'default'] })\n },\n })\n }\n\n const useDeleteChatModelMutation = () => {\n return useMutation({\n mutationFn: (id: number) => deleteChatModel(id),\n onSuccess: () => {\n queryClient.invalidateQueries({ queryKey: ['chat_models'] })\n queryClient.invalidateQueries({ queryKey: ['chat_models', 'default'] })\n },\n })\n }\n\n const useCreateChatModelMutation = () => {\n return useMutation({\n mutationFn: (data: CreateChatModelRequest) => createChatModel(data),\n onSuccess: () => {\n queryClient.invalidateQueries({ queryKey: ['chat_models'] })\n queryClient.invalidateQueries({ queryKey: ['chat_models', 'default'] })\n },\n })\n }\n\n return {\n useChatModelsQuery,\n useDefaultChatModelQuery,\n useUpdateChatModelMutation,\n useDeleteChatModelMutation,\n useCreateChatModelMutation,\n }\n}" }, { "path": "web/src/hooks/useCopyCode.ts", "content": "import { onMounted, onUpdated } from 'vue'\nimport { copyText } from '@/utils/format'\n\nexport function useCopyCode() {\n function copyCodeBlock() {\n const codeBlockWrapper = document.querySelectorAll('.code-block-wrapper')\n codeBlockWrapper.forEach((wrapper) => {\n const copyBtn = wrapper.querySelector('.code-block-header__copy')\n const codeBlock = wrapper.querySelector('.code-block-body')\n if (copyBtn && codeBlock) {\n copyBtn.addEventListener('click', () => {\n if (navigator.clipboard?.writeText)\n navigator.clipboard.writeText(codeBlock.textContent ?? '')\n else\n copyText({ text: codeBlock.textContent ?? '', origin: true })\n })\n }\n })\n }\n\n onMounted(() => copyCodeBlock())\n\n onUpdated(() => copyCodeBlock())\n}\n" }, { "path": "web/src/hooks/useIconRender.ts", "content": "import { h } from 'vue'\nimport { SvgIcon } from '@/components/common'\n\nexport const useIconRender = () => {\n interface IconConfig {\n icon?: string\n color?: string\n fontSize?: number\n }\n\n interface IconStyle {\n color?: string\n fontSize?: string\n }\n\n const iconRender = (config: IconConfig) => {\n const { color, fontSize, icon } = config\n\n const style: IconStyle = {}\n\n if (color)\n style.color = color\n\n if (fontSize)\n style.fontSize = `${fontSize}px`\n\n if (!icon)\n window.console.warn('iconRender: icon is required')\n\n return () => h(SvgIcon, { icon, style })\n }\n\n return {\n iconRender,\n }\n}\n" }, { "path": "web/src/hooks/useLanguage.ts", "content": "import { computed } from 'vue'\nimport { enUS, zhCN, zhTW } from 'naive-ui'\nimport { useAppStore } from '@/store'\nimport { setLocale } from '@/locales'\n\nexport function useLanguage() {\n const appStore = useAppStore()\n\n const language = computed(() => {\n switch (appStore.language) {\n case 'en-US':\n setLocale('en-US')\n return enUS\n case 'zh-CN':\n setLocale('zh-CN')\n return zhCN\n case 'zh-TW':\n setLocale('zh-TW')\n return zhTW\n default:\n setLocale('zh-CN')\n return enUS\n }\n })\n\n return { language }\n}\n" }, { "path": "web/src/hooks/useOnlineStatus.ts", "content": "// src/composables/useOnlineStatus.js\nimport { ref, onMounted, onUnmounted } from 'vue';\n\nexport function useOnlineStatus() {\n const isOnline = ref(navigator.onLine);\n\n const updateOnlineStatus = () => {\n isOnline.value = navigator.onLine;\n };\n\n onMounted(() => {\n window.addEventListener('online', updateOnlineStatus);\n window.addEventListener('offline', updateOnlineStatus);\n });\n\n onUnmounted(() => {\n window.removeEventListener('online', updateOnlineStatus);\n window.removeEventListener('offline', updateOnlineStatus);\n });\n\n return { isOnline };\n}" }, { "path": "web/src/hooks/useTheme.ts", "content": "import type { GlobalThemeOverrides } from 'naive-ui'\nimport { computed, watch } from 'vue'\nimport { darkTheme, useOsTheme } from 'naive-ui'\nimport { useAppStore } from '@/store'\n\nexport function useTheme() {\n const appStore = useAppStore()\n\n const OsTheme = useOsTheme()\n\n const isDark = computed(() => {\n if (appStore.theme === 'auto')\n return OsTheme.value === 'dark'\n else\n return appStore.theme === 'dark'\n })\n\n const theme = computed(() => {\n return isDark.value ? darkTheme : undefined\n })\n\n const tooltipOverrides = {\n Tooltip: {\n color: '#4b9e5f',\n textColor: '#fff',\n },\n }\n\n const themeOverrides = computed(() => {\n if (isDark.value) {\n return {\n common: {},\n ...tooltipOverrides,\n }\n }\n return { ...tooltipOverrides }\n })\n\n watch(\n () => isDark.value,\n (dark) => {\n if (dark)\n document.documentElement.classList.add('dark')\n else\n document.documentElement.classList.remove('dark')\n },\n { immediate: true },\n )\n\n return { theme, themeOverrides }\n}\n" }, { "path": "web/src/hooks/useWorkspaceRouting.ts", "content": "import { computed } from 'vue'\nimport { useRouter, useRoute } from 'vue-router'\nimport { useSessionStore, useWorkspaceStore } from '@/store'\n\nexport function useWorkspaceRouting() {\n const router = useRouter()\n const route = useRoute()\n const sessionStore = useSessionStore()\n const workspaceStore = useWorkspaceStore()\n\n // Get current workspace from URL\n const currentWorkspaceFromUrl = computed(() => {\n return route.params.workspaceUuid as string || null\n })\n\n // Get current session from URL\n const currentSessionFromUrl = computed(() => {\n return route.params.uuid as string || null\n })\n\n // Check if we're on a workspace-aware route\n const isWorkspaceRoute = computed(() => {\n return route.name === 'WorkspaceChat'\n })\n\n // Generate workspace-aware URL for a session\n function getSessionUrl(sessionUuid: string, workspaceUuid?: string): string {\n const workspace = workspaceUuid || workspaceStore.activeWorkspace\n const session = sessionStore.getChatSessionByUuid(sessionUuid)\n \n // Use session's workspace if available, otherwise use provided or active workspace\n const targetWorkspace = session?.workspaceUuid || workspace || workspaceStore.getDefaultWorkspace?.uuid\n \n if (targetWorkspace) {\n return `/#/workspace/${targetWorkspace}/chat/${sessionUuid}`\n }\n // Fallback to default workspace if none found\n const defaultWorkspace = workspaceStore.getDefaultWorkspace\n return defaultWorkspace ? `/#/workspace/${defaultWorkspace.uuid}/chat/${sessionUuid}` : `/#/`\n }\n\n // Generate workspace URL (without session)\n function getWorkspaceUrl(workspaceUuid: string): string {\n return `/#/workspace/${workspaceUuid}/chat`\n }\n\n // Navigate to session with workspace context\n async function navigateToSession(sessionUuid: string, workspaceUuid?: string) {\n const workspace = workspaceUuid || workspaceStore.activeWorkspace\n const session = sessionStore.getChatSessionByUuid(sessionUuid)\n \n // Use session's workspace if available, otherwise use default workspace\n const targetWorkspace = session?.workspaceUuid || workspace || workspaceStore.getDefaultWorkspace?.uuid\n \n if (targetWorkspace) {\n await router.push({\n name: 'WorkspaceChat',\n params: {\n workspaceUuid: targetWorkspace,\n uuid: sessionUuid\n }\n })\n } else {\n // Fallback to default route if no workspace found\n await router.push({ name: 'DefaultWorkspace' })\n }\n }\n\n // Navigate to workspace (without specific session)\n async function navigateToWorkspace(workspaceUuid: string) {\n await router.push({\n name: 'WorkspaceChat',\n params: { workspaceUuid }\n })\n }\n\n // Navigate to first session in workspace, or workspace itself if no sessions\n async function navigateToWorkspaceOrFirstSession(workspaceUuid: string) {\n const workspaceSessions = sessionStore.getSessionsByWorkspace(workspaceUuid)\n \n if (workspaceSessions.length > 0) {\n await navigateToSession(workspaceSessions[0].uuid, workspaceUuid)\n } else {\n await navigateToWorkspace(workspaceUuid)\n }\n }\n\n // Check if current route matches the expected workspace/session\n function isCurrentRoute(sessionUuid?: string, workspaceUuid?: string): boolean {\n const currentSession = currentSessionFromUrl.value\n const currentWorkspace = currentWorkspaceFromUrl.value\n \n if (sessionUuid && sessionUuid !== currentSession) {\n return false\n }\n \n if (workspaceUuid && workspaceUuid !== currentWorkspace) {\n return false\n }\n \n return true\n }\n\n // Sync URL with current state (useful for redirects after workspace changes)\n async function syncUrlWithState() {\n const activeSession = sessionStore.active\n const activeWorkspace = workspaceStore.activeWorkspace\n \n // If we have an active session and workspace, ensure URL is correct\n if (activeSession && activeWorkspace) {\n const session = sessionStore.getChatSessionByUuid(activeSession)\n if (session && session.workspaceUuid === activeWorkspace) {\n // Check if current URL doesn't match expected workspace-aware URL\n if (!isCurrentRoute(activeSession, activeWorkspace)) {\n await navigateToSession(activeSession, activeWorkspace)\n }\n }\n }\n }\n\n // Handle browser back/forward navigation\n function handleRouteChange() {\n const workspaceFromUrl = currentWorkspaceFromUrl.value\n const sessionFromUrl = currentSessionFromUrl.value\n \n // Update store state to match URL\n if (workspaceFromUrl && workspaceFromUrl !== workspaceStore.activeWorkspace) {\n workspaceStore.setActiveWorkspace(workspaceFromUrl)\n }\n \n if (sessionFromUrl && sessionFromUrl !== sessionStore.active) {\n const session = sessionStore.getChatSessionByUuid(sessionFromUrl)\n if (session) {\n sessionStore.setActiveSessionWithoutNavigation(session.workspaceUuid, sessionFromUrl)\n }\n }\n }\n\n return {\n // Computed\n currentWorkspaceFromUrl,\n currentSessionFromUrl,\n isWorkspaceRoute,\n \n // Methods\n getSessionUrl,\n getWorkspaceUrl,\n navigateToSession,\n navigateToWorkspace,\n navigateToWorkspaceOrFirstSession,\n isCurrentRoute,\n syncUrlWithState,\n handleRouteChange\n }\n}" }, { "path": "web/src/icons/403.vue", "content": "\n" }, { "path": "web/src/icons/500.vue", "content": "\n" }, { "path": "web/src/locales/en-US-more.json", "content": "{\n \"bot\": {\n \"noHistory\": \"No history yet\",\n \"runNumber\": \"Run {number}\",\n \"tabs\": {\n \"conversation\": \"Preset\",\n \"history\": \"Answer\"\n }\n },\n \"error\": {\n \"MODEL_006\": \"Failed to get a response from the model\"\n }\n}" }, { "path": "web/src/locales/en-US.json", "content": "{\n \"admin\": {\n \"activityHistory\": \"Activity History\",\n \"add_model\": \"Add Model\",\n \"add_user_model_rate_limit\": \"Add User Session Count\",\n \"chat_model\": {\n \"actions\": \"Actions\",\n \"apiAuthHeader\": \"Auth Header key\",\n \"apiAuthKey\": \"API KEY corresponding to the environment variable\",\n \"apiType\": \"API Type\",\n \"clear_form\": \"Clear Form\",\n \"copy\": \"Copy\",\n \"copy_success\": \"Copied successfully\",\n \"default\": \"Default\",\n \"defaultToken\": \"Default token number\",\n \"deleteModel\": \"Delete\",\n \"deleteModelConfirm\": \"Confirm deletion {name}?\",\n \"delete_failed\": \"Delete failed\",\n \"delete_success\": \"Delete success\",\n \"edit_model\": \"Edit Model\",\n \"enablePerModeRatelimit\": \"Enable Rate Limit Per Mode\",\n \"enablePerModelRateLimit\": \"Enable per-model rate limit\",\n \"isDefault\": \"Default?\",\n \"isEnable\": \"Is Enabled\",\n \"label\": \"Model name\",\n \"maxToken\": \"Maximum token number\",\n \"name\": \"Model ID\",\n \"orderNumber\": \"Order number\",\n \"paste_json\": \"Paste JSON Configuration\",\n \"paste_json_placeholder\": \"Paste your model configuration JSON here...\",\n \"populate_form\": \"Populate Form\",\n \"update_failed\": \"Update failed\",\n \"update_success\": \"Update success\",\n \"url\": \"Request full URL\"\n },\n \"chat_model_name\": \"Model ID\",\n \"created\": \"Created\",\n \"date\": \"Date\",\n \"firstName\": \"First Name\",\n \"hideSubMenu\": \"Hide Submenu\",\n \"lastName\": \"Last Name\",\n \"lastUsed\": \"Last Used\",\n \"messages\": \"Messages\",\n \"messages3Days\": \"Messages (3 days)\",\n \"model\": \"Model\",\n \"modelUsage\": \"Model Usage\",\n \"modelUsageDistribution\": \"Model Usage Distribution\",\n \"model_one_default_only\": \"There can only be one default model, please set other models as non-default first.\",\n \"name\": \"Name\",\n \"openPanel\": \"Open Admin Panel\",\n \"overview\": \"Overview\",\n \"per_model_rate_limit\": {\n \"ChatModelName\": \"Model Name\",\n \"FullName\": \"Full Name\",\n \"RateLimit\": \"10min Access Count\",\n \"UserEmail\": \"User Email\",\n \"actions\": \"Actions\"\n },\n \"per_model_rate_limit_title\": \"Model Throttling\",\n \"permission\": \"Permission\",\n \"rateLimit\": \"Rate Limit\",\n \"rateLimit10Min\": \"Message Limit (10 minutes)\",\n \"rate_limit\": \"Session Count (10min)\",\n \"recent3Days\": \"Recent 3 Days Activity\",\n \"refresh\": \"Refresh\",\n \"sessionHistory\": \"Session History\",\n \"sessionId\": \"Session ID\",\n \"sessionSnapshot\": \"Session Snapshot\",\n \"sessions\": \"Sessions\",\n \"showSubMenu\": \"Show Submenu\",\n \"system_model_tab_title\": \"Model Configuration\",\n \"title\": \"Admin\",\n \"tokens\": \"Tokens\",\n \"tokens3Days\": \"Tokens (3 days)\",\n \"totalChatMessages\": \"Total Chat Messages\",\n \"totalChatMessages3Days\": \"Total Chat Messages (3 days)\",\n \"totalChatMessages3DaysAvgTokenCount\": \"Average token count (3 days)\",\n \"totalChatMessages3DaysTokenCount\": \"Total token count (3 days)\",\n \"totalChatMessagesTokenCount\": \"Total token count\",\n \"totalMessages\": \"Total Messages\",\n \"totalSessions\": \"Total Sessions\",\n \"totalTokens\": \"Total Tokens\",\n \"updated\": \"Updated\",\n \"usage\": \"Usage\",\n \"userAnalysis\": \"User Analysis\",\n \"userEmail\": \"User Email\",\n \"userMessage\": \"User\",\n \"userStat\": \"User Statistics\"\n },\n \"bot\": {\n \"all\": {\n \"title\": \"Bots\"\n },\n \"list\": \"Bots\",\n \"noHistory\": \"No history yet\",\n \"runNumber\": \"Run {number}\",\n \"showCode\": \"Generate API call code\",\n \"tabs\": {\n \"conversation\": \"Preset\",\n \"history\": \"Answer\"\n }\n },\n \"chat\": {\n \"N\": \"Number of results: {n}\",\n \"addComment\": \"Add Comment\",\n \"adjustParameters\": \"Adjust parameters\",\n \"advanced_settings\": \"Advanced Settings\",\n \"alreadyInNewChat\": \"alreay in new chat\",\n \"artifactMode\": \"Artifacts\",\n \"artifactModeDescription\": \"Enable artifact rendering for code, previews, and visualizations\",\n \"artifactInstructionTitle\": \"Artifact Instructions\",\n \"chatSettings\": \"Chat Settings\",\n \"chatSnapshot\": \"Generate the conversation\",\n \"clearChat\": \"Clear chat session\",\n \"clearChatConfirm\": \"Do you want to clear the chat session?\",\n \"clearHistoryConfirm\": \"Are you sure you want to clear the chat history?\",\n \"commentFailed\": \"Failed to add comment\",\n \"commentPlaceholder\": \"Enter your comment...\",\n \"commentSuccess\": \"Comment added successfully\",\n \"completionsCount\": \"Number of results: {contextCount}\",\n \"contextCount\": \"Context Length: {contextCount}\",\n \"contextLength\": \"Context Length, default 10 (2 at the beginning of the conversation + 8 most recent)\",\n \"copied\": \"Copied\",\n \"copy\": \"Copy\",\n \"copyCode\": \"Copy code\",\n \"createBot\": \"Create Bot\",\n \"debug\": \"Debug Mode\",\n \"debugDescription\": \"Enable debug mode for troubleshooting and diagnostics\",\n \"defaultSystemPrompt\": \"You are a helpful, concise assistant. Ask clarifying questions when needed. Provide accurate answers with short reasoning and actionable steps. If unsure, say so and suggest how to verify.\",\n \"deleteChatSessionsConfirm\": \"Are you sure you want to delete this record?\",\n \"deleteMessage\": \"Delete message\",\n \"deleteMessageConfirm\": \"Do you want to delete this message?\",\n \"disable_debug\": \"Disable\",\n \"disable_artifact\": \"Disable\",\n \"disable_explore\": \"Disable\",\n \"enable_debug\": \"Enable\",\n \"enable_artifact\": \"Enable\",\n \"enable_explore\": \"Enable\",\n \"exploreMode\": \"Explore Mode\",\n \"exploreModeDescription\": \"Get suggested questions based on conversation context\",\n \"loadingSession\": \"Loading session...\",\n \"loading_models\": \"Loading models...\",\n \"loading_instructions\": \"Loading instructions...\",\n \"modes\": \"Modes\",\n \"models\": \"models\",\n \"promptInstructions\": \"Prompt Instructions\",\n \"exportFailed\": \"Saving failed\",\n \"exportImage\": \"Export chat session to image\",\n \"exportImageConfirm\": \"Do you want to save the chat session as an image?\",\n \"exportMD\": \"Export chat session to markdown file\",\n \"exportMDConfirm\": \"Do you want to save the chat session as a markdown file?\",\n \"exportSuccess\": \"Saved successfully\",\n \"frequencyPenalty\": \"Frequency Penalty\",\n \"generateMoreSuggestions\": \"Generate more suggestions\",\n \"generating\": \"Generating...\",\n \"is_summarize_mode\": \"On\",\n \"maxTokens\": \"Max Total Tokens : {maxTokens}\",\n \"model\": \"Model\",\n \"new\": \"New Chat\",\n \"no_summarize_mode\": \"Off\",\n \"placeholder\": \"What would you like to say... (Shift + Enter = newline, '/' to trigger prompts)\",\n \"placeholderMobile\": \"What would you like to say...\",\n \"playAudio\": \"audio\",\n \"presencePenalty\": \"Presence Penalty\",\n \"sessionConfig\": \"Conversation Settings:\",\n \"snapshotSuccess\": \"Snapshot successful, please view in a new tab\",\n \"stopAnswer\": \"Stop Answering\",\n \"suggestedQuestions\": \"Suggested Questions\",\n \"summarize_mode\": \"Summary mode (supports longer context 20+)\",\n \"temperature\": \"Temperature : {temperature}\",\n \"topP\": \"Top P: {topP}\",\n \"turnOffContext\": \"In this mode, messages sent will not include previous chat logs.\",\n \"turnOnContext\": \"In this mode, messages sent will include previous chat logs.\",\n \"uploadFiles\": \"Upload Files\",\n \"uploader_button\": \"Upload\",\n \"uploader_close\": \"Close\",\n \"uploader_help_text\": \"Supported file types: text, image, audio, video\",\n \"uploader_title\": \"Upload File\",\n \"usingContext\": \"Context Mode\"\n },\n \"chat_snapshot\": {\n \"createChat\": \"Create chat\",\n \"deleteFailed\": \"Unable to delete conversation record\",\n \"deletePost\": \"Are you sure you want to delete this conversation record?\",\n \"deletePostConfirm\": \"Delete conversation record\",\n \"deleteSuccess\": \"Conversation record deleted successfully\",\n \"exportImage\": \"Save conversation as image\",\n \"exportMarkdown\": \"Export to markdown file\",\n \"scrollTop\": \"Scroll to top\",\n \"showCode\": \"Generate API call code\",\n \"title\": \"Chat Records\"\n },\n \"common\": {\n \"actions\": \"Actions\",\n \"ask_user_register\": \"Please register, only registered accounts can continue the conversation\",\n \"cancel\": \"Cancel\",\n \"clear\": \"Clear\",\n \"confirm\": \"Confirm\",\n \"copy\": \"Copy\",\n \"create\": \"Create\",\n \"delete\": \"Delete\",\n \"disabled\": \"Disabled\",\n \"edit\": \"Edit\",\n \"editUser\": \"Edit User\",\n \"email\": \"Email\",\n \"email_placeholder\": \"Please enter your email\",\n \"enabled\": \"Enabled\",\n \"export\": \"Export\",\n \"failed\": \"Operation failed\",\n \"fetchFailed\": \"Failed to fetch data\",\n \"help\": \"The first one is the theme(prompt), and the context includes 10 messages.\",\n \"import\": \"Import\",\n \"login\": \"Login\",\n \"loginSuccess\": \"Login Successful\",\n \"login_failed\": \"Login Failed\",\n \"logout\": \"Logout\",\n \"logout_failed\": \"Logout Failed\",\n \"logout_success\": \"Logout Successful\",\n \"no\": \"No\",\n \"noData\": \"No data available\",\n \"password_placeholder\": \"Please enter your password\",\n \"please_register\": \"Please Register First\",\n \"regenerate\": \"regenerate\",\n \"reset\": \"Reset\",\n \"save\": \"Save\",\n \"signup\": \"Sign up\",\n \"signup_failed\": \"Registration Failed\",\n \"signup_success\": \"Registration Successful\",\n \"submit\": \"Submit\",\n \"submitting\": \"Submitting...\",\n \"success\": \"Operation succeeded\",\n \"unauthorizedTips\": \"Please sign up or login\",\n \"update\": \"Update\",\n \"use\": \"Use\",\n \"verify\": \"Verify\",\n \"warning\": \"Warning\",\n \"wrong\": \"Something seems to have gone wrong. Please try again later.\",\n \"yes\": \"Yes\"\n },\n \"error\": {\n \"INTN_004\": \"Failed to request the model, please try again later or contact the administrator\",\n \"MODEL_001\": \"the first message is a system message, please continue entering information to start the conversation\",\n \"MODEL_006\": \"Failed to get a response from the model\",\n \"NotAdmin\": \"Non-administrators are prohibited from accessing\",\n \"NotAuthorized\": \"Please log in first\",\n \"RESOURCE_EXHAUSTED\": \"Resource exhausted\",\n \"VALD_001\": \"Invalid Request\",\n \"VALD_004\": \"Invalid email or password\",\n \"connection\": \"Connection interrupted. Please check your connection and try again.\",\n \"fail_to_get_rate_limit\": \"Failed to retrieve the quota corresponding to the model, please contact the administrator to enable it\",\n \"forbidden\": \"Access denied. You don't have permission for this action.\",\n \"gpt-4_over_limit\": \"GPT4 message sending limit exceeded, please contact the administrator to enable or increase quota.\",\n \"invalidEmail\": \"Invalid email address\",\n \"invalidPassword\": \"Password is not valid, it should be length >=6 and include a number, a lowercase letter, an uppercase letter, and a special character\",\n \"invalidRepwd\": \"Duplicate passwords are inconsistent\",\n \"network\": \"Network connection error. Please check your internet connection.\",\n \"notFound\": \"The requested resource was not found.\",\n \"rateLimit\": \"The message sending limit has been reached, please contact the administrator.\",\n \"serverError\": \"Server error. Our team has been notified and is working on a fix.\",\n \"syncChatSession\": \"Sync Failed, Please Try Again Later\",\n \"timeout\": \"Request timed out. Please check your connection and try again.\",\n \"token_length_exceed_limit\": \"Total message length exceeds limit, please reduce context quantity, message length or start a new conversation.\",\n \"unauthorized\": \"Session expired. Please login again.\",\n \"unknown\": \"An unexpected error occurred. Please try again.\",\n \"validation\": \"Please check your input and try again.\"\n },\n \"prompt\": {\n \"add\": \"Add\",\n \"addFailed\": \"Add failed\",\n \"addSuccess\": \"Add successful\",\n \"clear\": \"Clear\",\n \"confirmClear\": \"Are you sure you want to clear?\",\n \"delete\": \"Delete\",\n \"deleteConfirm\": \"Are you sure you want to delete?\",\n \"deleteFailed\": \"Delete failed\",\n \"deleteSuccess\": \"Delete successful\",\n \"download\": \"Download\",\n \"downloadOnline\": \"Import from URL\",\n \"downloadOnlineWarning\": \"Warning: Please verify the source of the downloaded JSON file. Malicious JSON files could harm your computer!\",\n \"edit\": \"Edit\",\n \"editFailed\": \"Edit failed\",\n \"editSuccess\": \"Edit successful\",\n \"enterJsonUrl\": \"Please enter a valid JSON URL\",\n \"export\": \"Export\",\n \"import\": \"Import\",\n \"store\": \"Prompt\"\n },\n \"setting\": {\n \"admin\": \"Admin\",\n \"api\": \"API\",\n \"apiToken\": \"API Token\",\n \"apiTokenCopied\": \"API Token copied\",\n \"apiTokenCopyFailed\": \"Failed to copy API Token\",\n \"avatarLink\": \"Avatar Link\",\n \"chatHistory\": \"Chat History\",\n \"config\": \"Configuration\",\n \"defaultDesc\": \"Signature\",\n \"defaultName\": \"You\",\n \"description\": \"Description\",\n \"general\": \"Overview\",\n \"language\": \"Language\",\n \"name\": \"Name\",\n \"resetUserInfo\": \"Reset User Info\",\n \"reverseProxy\": \"Reverse Proxy\",\n \"setting\": \"Settings\",\n \"snapshotLink\": \"Conversation history\",\n \"socks\": \"Socks\",\n \"switchLanguage\": \"English(切换语言) \",\n \"theme\": \"Theme\",\n \"timeout\": \"Timeout\"\n },\n \"workspace\": {\n \"active\": \"Active\",\n \"cannotDeleteDefault\": \"Cannot delete the default workspace\",\n \"color\": \"Color\",\n \"create\": \"Create Workspace\",\n \"createFirst\": \"Create your first workspace\",\n \"created\": \"Workspace created successfully\",\n \"default\": \"Default\",\n \"deleteConfirm\": \"Are you sure you want to delete this workspace?\",\n \"deleted\": \"Workspace deleted successfully\",\n \"description\": \"Description\",\n \"descriptionPlaceholder\": \"Optional description for this workspace\",\n \"dragToReorder\": \"Drag to reorder\",\n \"duplicate\": \"Duplicate\",\n \"edit\": \"Edit Workspace\",\n \"filteredResults\": \"of {total}\",\n \"icon\": \"Icon\",\n \"invalidColor\": \"Invalid color format. Please use a valid hex color.\",\n \"lastUpdated\": \"Updated\",\n \"loading\": \"Loading workspaces...\",\n \"manage\": \"Manage Workspaces\",\n \"name\": \"Name\",\n \"namePlaceholder\": \"Enter workspace name\",\n \"nameRequired\": \"Workspace name is required\",\n \"noWorkspaces\": \"No workspaces found\",\n \"reorderError\": \"Failed to reorder workspaces\",\n \"reorderMode\": \"Reorder Mode\",\n \"reorderSuccess\": \"Workspaces reordered successfully\",\n \"saveError\": \"Failed to save workspace\",\n \"searchPlaceholder\": \"Search workspaces...\",\n \"sessionCount\": \"{count} sessions\",\n \"setAsDefault\": \"Set as Default\",\n \"switchError\": \"Failed to switch workspace\",\n \"switchedTo\": \"Switched to {name}\",\n \"totalCount\": \"Total: {count} workspaces\",\n \"updated\": \"Workspace updated successfully\"\n }\n}\n" }, { "path": "web/src/locales/en.ts", "content": "" }, { "path": "web/src/locales/index.ts", "content": "import type { App } from 'vue'\nimport { createI18n } from 'vue-i18n'\nimport enUS from './en-US.json'\nimport zhCN from './zh-CN.json'\nimport zhTW from './zh-TW.json'\nimport type { Language } from '@/store/modules/app/helper'\n\n\nconst i18n = createI18n({\n locale: navigator.language.split('-')[0],\n fallbackLocale: 'en',\n allowComposition: true,\n messages: {\n 'en-US': enUS,\n 'zh-CN': zhCN,\n 'zh-TW': zhTW,\n },\n})\n\nexport function t(key: string, values?: Record) {\n if (values) {\n return i18n.global.t(key, values)\n } else {\n return i18n.global.t(key)\n }\n}\n\nexport function setLocale(locale: Language) {\n i18n.global.locale = locale\n}\n\nexport function setupI18n(app: App) {\n app.use(i18n)\n}\n\nexport default i18n\n" }, { "path": "web/src/locales/zh-CN.json", "content": "{\n \"common\": {\n \"ask_user_register\": \"请注册, 只有注册账号才能继续对话\",\n \"help\": \"第一条是主题(prompt, 角色定义), 上下文默认包括10条信息, 参数可以点击按钮进行调节, 请务必注意隐私, 不输入涉密, 敏感信息.\",\n \"copy\": \"复制\",\n \"edit\": \"编辑\",\n \"editUser\": \"编辑用户\",\n \"delete\": \"删除\",\n \"disabled\": \"禁用\",\n \"enabled\": \"启用\",\n \"actions\": \"操作\",\n \"warning\": \"警告\",\n \"save\": \"保存\",\n \"reset\": \"重置\",\n \"export\": \"导出\",\n \"import\": \"导入\",\n \"clear\": \"清空\",\n \"yes\": \"是\",\n \"no\": \"否\",\n \"use\": \"使用\",\n \"noData\": \"暂无数据\",\n \"wrong\": \"好像出错了，请稍后再试。\",\n \"success\": \"操作成功\",\n \"failed\": \"操作失败\",\n \"fetchFailed\": \"获取数据失败\",\n \"verify\": \"验证\",\n \"login\": \"登录\",\n \"logout\": \"登出\",\n \"regenerate\": \"重新生成\",\n \"signup\": \"注册\",\n \"email\": \"邮箱\",\n \"confirm\": \"确认\",\n \"cancel\": \"取消\",\n \"create\": \"创建\",\n \"update\": \"更新\",\n \"email_placeholder\": \"请输入邮箱\",\n \"password_placeholder\": \"请输入密码\",\n \"unauthorizedTips\": \"请注册或者登录\",\n \"loginSuccess\": \"登录成功\",\n \"logout_success\": \"登出成功\",\n \"signup_success\": \"注册成功\",\n \"signup_failed\": \"注册失败\",\n \"submit\": \"提交\",\n \"submitting\": \"提交中...\",\n \"login_failed\": \"登录失败\",\n \"logout_failed\": \"登出失败\",\n \"please_register\": \"请先注册\"\n },\n \"prompt\": {\n \"store\": \"提示词\",\n \"add\": \"添加\",\n \"edit\": \"编辑\",\n \"delete\": \"删除\",\n \"deleteConfirm\": \"是否删除?\",\n \"deleteSuccess\": \"删除成功\",\n \"deleteFailed\": \"删除失败\",\n \"addSuccess\": \"添加成功\",\n \"addFailed\": \"添加失败\",\n \"editSuccess\": \"编辑成功\",\n \"editFailed\": \"编辑失败\",\n \"import\": \"导入\",\n \"export\": \"导出\",\n \"clear\": \"清空\",\n \"confirmClear\": \"是否清空?\",\n \"downloadOnline\": \"在线导入\",\n \"downloadOnlineWarning\": \"注意：请检查下载 JSON 文件来源，恶意的JSON文件可能会破坏您的计算机！\",\n \"download\": \"下载\",\n \"enterJsonUrl\": \"请输入正确的JSON地址\"\n },\n \"bot\": {\n \"list\": \"机器人\",\n \"noHistory\": \"还没有记录\",\n \"runNumber\": \"Run {number}\",\n \"all\": {\n \"title\": \"机器人\"\n },\n \"tabs\": {\n \"conversation\": \"预设\",\n \"history\": \"回答\"\n },\n \"showCode\": \"生成API调用代码\"\n },\n \"chat\": {\n \"alreadyInNewChat\": \"已经在新对话中\",\n \"advanced_settings\": \"高级设置\",\n \"artifactModeDescription\": \"启用代码、预览和可视化的 Artifact 渲染\",\n \"debugDescription\": \"启用调试模式用于故障排除和诊断\",\n \"defaultSystemPrompt\": \"你是一个有帮助且简明的助手。需要时先提出澄清问题。给出准确答案，并提供简短理由和可执行步骤。不确定时要说明，并建议如何验证。\",\n \"exploreModeDescription\": \"基于对话上下文获取建议问题\",\n \"loading_models\": \"正在加载模型...\",\n \"modes\": \"模式\",\n \"models\": \"个模型\",\n \"new\": \"新对话\",\n \"summarize_mode\": \"总结模式(可以支持更长的上下文20+)\",\n \"is_summarize_mode\": \"开启\",\n \"no_summarize_mode\": \"关闭\",\n \"placeholder\": \"来说点什么吧...（Shift + Enter = 换行, '/' 触发提示词）\",\n \"placeholderMobile\": \"来说点什么...\",\n \"copy\": \"复制\",\n \"copied\": \"复制成功\",\n \"copyCode\": \"复制代码\",\n \"clearChat\": \"清空会话\",\n \"clearChatConfirm\": \"是否清空会话?\",\n \"exportImage\": \"保存会话到图片\",\n \"chatSnapshot\": \"生成会话记录\",\n \"chatSettings\": \"对话设置\",\n \"uploadFiles\": \"上传文件\",\n \"createBot\": \"创建机器人\",\n \"snapshotSuccess\": \"快照成功, 请在新标签页中查看, 链接任何人都可以打开, 谨慎分享\",\n \"adjustParameters\": \"调整参数\",\n \"exportImageConfirm\": \"是否将会话保存为图片?\",\n \"exportMD\": \"保存会话到Markdown\",\n \"exportMDConfirm\": \"是否将会话保存为Markdown?\",\n \"exportSuccess\": \"保存成功\",\n \"exportFailed\": \"保存失败\",\n \"usingContext\": \"上下文模式\",\n \"turnOnContext\": \"当前模式下, 发送消息会携带之前的聊天记录\",\n \"turnOffContext\": \"当前模式下, 发送消息不会携带之前的聊天记录\",\n \"deleteMessage\": \"删除消息\",\n \"deleteMessageConfirm\": \"是否删除此消息?\",\n \"deleteChatSessionsConfirm\": \"确定删除此记录?\",\n \"clearHistoryConfirm\": \"确定清空聊天记录?\",\n \"contextLength\": \"上下文数量, 默认10 (会话开始的2条 + 最近的8条)\",\n \"stopAnswer\": \"停止回答\",\n \"suggestedQuestions\": \"深入提问\",\n \"contextCount\": \"上下文数量: {contextCount}\",\n \"temperature\": \"温度: {temperature}, 越高回答越发散, 越低回答越精确\",\n \"model\": \"模型\",\n \"maxTokens\": \"最大输出token数量: {maxTokens}\",\n \"topP\": \"Top P: {topP}, 常用词占比, 一般与温度参数只调节一个即可\",\n \"N\": \"结果数量: {n}\",\n \"frequencyPenalty\": \"频率惩罚\",\n \"presencePenalty\": \"存在惩罚\",\n \"debug\": \"调试模式\",\n \"artifactMode\": \"Artifacts\",\n \"sessionConfig\": \"会话设置\",\n \"enable_debug\": \"启用\",\n \"enable_artifact\": \"启用\",\n \"disable_debug\": \"关闭\",\n \"disable_artifact\": \"关闭\",\n \"exploreMode\": \"探索模式\",\n \"enable_explore\": \"启用\",\n \"disable_explore\": \"关闭\",\n \"promptInstructions\": \"提示词说明\",\n \"artifactInstructionTitle\": \"Artifact 说明\",\n \"loading_instructions\": \"正在加载说明...\",\n \"loadingSession\": \"正在加载会话...\",\n \"completionsCount\": \"结果数量: {contextCount}\",\n \"generateMoreSuggestions\": \"生成更多建议\",\n \"generating\": \"生成中...\",\n \"playAudio\": \"语音\",\n \"uploader_title\": \"上传文件\",\n \"uploader_button\": \"上传\",\n \"uploader_close\": \"关闭\",\n \"uploader_help_text\": \"支持上传文件类型： text, image, audio, video\",\n \"addComment\": \"添加评论\",\n \"commentPlaceholder\": \"请输入评论...\",\n \"commentSuccess\": \"评论添加成功\",\n \"commentFailed\": \"评论添加失败\"\n },\n \"chat_snapshot\": {\n \"title\": \"会话集\",\n \"deletePost\": \"您确定要删除此会话记录吗？\",\n \"deletePostConfirm\": \"删除会话记录\",\n \"deleteSuccess\": \"会话记录删除成功\",\n \"deleteFailed\": \"无法删除会话记录\",\n \"exportImage\": \"保存会话到图片\",\n \"showCode\": \"生成API调用代码\",\n \"exportMarkdown\": \"导出到markdown文件\",\n \"createChat\": \"创建会话\",\n \"scrollTop\": \"回到顶部\"\n },\n \"admin\": {\n \"openPanel\": \"在新标签页中打开\",\n \"title\": \"管理\",\n \"refresh\": \"刷新\",\n \"userEmail\": \"用户邮箱\",\n \"totalChatMessages\": \"消息总数\",\n \"totalChatMessages3Days\": \"消息总数(3天)\",\n \"totalChatMessagesTokenCount\": \"总token数量\",\n \"totalChatMessages3DaysTokenCount\": \"总token数量(3天)\",\n \"totalChatMessages3DaysAvgTokenCount\": \"平均token数量(3天)\",\n \"firstName\": \"名字\",\n \"lastName\": \"姓氏\",\n \"hideSubMenu\": \"隐藏子菜单\",\n \"showSubMenu\": \"显示子菜单\",\n \"userStat\": \"用户统计\",\n \"permission\": \"权限\",\n \"rateLimit\": \"限流\",\n \"userMessage\": \"用户\",\n \"userAnalysis\": \"用户分析\",\n \"overview\": \"概览\",\n \"modelUsage\": \"模型使用情况\",\n \"activityHistory\": \"活动历史\",\n \"sessionHistory\": \"会话历史\",\n \"totalMessages\": \"总消息数\",\n \"totalTokens\": \"总令牌数\",\n \"totalSessions\": \"总会话数\",\n \"recent3Days\": \"最近3天活动\",\n \"messages3Days\": \"消息数(3天)\",\n \"tokens3Days\": \"令牌数(3天)\",\n \"modelUsageDistribution\": \"模型使用分布\",\n \"messages\": \"消息\",\n \"tokens\": \"令牌\",\n \"usage\": \"使用率\",\n \"lastUsed\": \"最后使用\",\n \"date\": \"日期\",\n \"sessions\": \"会话\",\n \"sessionId\": \"会话ID\",\n \"created\": \"创建时间\",\n \"updated\": \"更新时间\",\n \"sessionSnapshot\": \"会话快照\",\n \"model\": \"模型\",\n \"system_model_tab_title\": \"模型配置\",\n \"per_model_rate_limit_title\": \"模型流控\",\n \"add_user_model_rate_limit\": \"添加用户会话数量\",\n \"add_model\": \"添加模型\",\n \"rate_limit\": \"会话数量(10min)\",\n \"chat_model_name\": \"模型ID\",\n \"chat_model\": {\n \"copy\": \"复制\",\n \"copy_success\": \"复制成功\",\n \"default\": \"默认\",\n \"deleteModel\": \"删除\",\n \"deleteModelConfirm\": \"确认删除 {name}?\",\n \"name\": \"模型ID\",\n \"label\": \"标签\",\n \"isDefault\": \"是否默认\",\n \"apiAuthHeader\": \"AuthHeader\",\n \"apiAuthKey\": \"API密钥环境变量\",\n \"apiType\": \"API类型\",\n \"actions\": \"操作\",\n \"url\": \"访问接口(完整URL路径)\",\n \"enablePerModeRatelimit\": \"是否单独流控\",\n \"isEnable\": \"是否启用\",\n \"orderNumber\": \"排序号\",\n \"maxToken\": \"最大输出token数量\",\n \"defaultToken\": \"默认输出token数量\",\n \"paste_json\": \"粘贴JSON配置\",\n \"paste_json_placeholder\": \"在此处粘贴您的模型配置JSON...\",\n \"populate_form\": \"填充表单\",\n \"clear_form\": \"清空表单\"\n },\n \"per_model_rate_limit\": {\n \"FullName\": \"姓名\",\n \"UserEmail\": \"用户邮箱\",\n \"ChatModelName\": \"模型名称(ID)\",\n \"RateLimit\": \"10分钟访问次数\",\n \"actions\": \"操作\"\n },\n \"model_one_default_only\": \"只能有一个默认模型, 请先设置其他模型为非默认\",\n \"rateLimit10Min\": \"消息数量上限(10分钟)\",\n \"name\": \"姓名\"\n },\n \"setting\": {\n \"setting\": \"设置\",\n \"general\": \"总览\",\n \"admin\": \"管理\",\n \"config\": \"配置\",\n \"avatarLink\": \"头像链接\",\n \"name\": \"名称\",\n \"defaultName\": \"你\",\n \"defaultDesc\": \"签名\",\n \"snapshotLink\": \"会话集\",\n \"apiToken\": \"API Token\",\n \"description\": \"描述\",\n \"resetUserInfo\": \"重置用户信息\",\n \"chatHistory\": \"聊天记录\",\n \"theme\": \"主题\",\n \"language\": \"语言\",\n \"switchLanguage\": \"切换语言(English)\",\n \"api\": \"API\",\n \"reverseProxy\": \"反向代理\",\n \"timeout\": \"超时\",\n \"socks\": \"Socks\",\n \"apiTokenCopied\": \"API Token 已复制\",\n \"apiTokenCopyFailed\": \"无法复制 API Token\"\n },\n \"error\": {\n \"MODEL_001\": \"第一条是系统消息已经是收到, 请继续输入信息开始会话\",\n \"MODEL_006\": \"无法从模型获取回复\",\n \"RESOURCE_EXHAUSTED\": \"资源耗尽\",\n \"VALD_001\": \"无效请求\",\n \"VALD_004\": \"无效的电子邮件或密码\",\n \"INTN_004\": \"请求模型失败, 请稍后再试, 或联系管理员\",\n \"rateLimit\": \"发送消息数量已经达到上限, 请联系管理员\",\n \"fail_to_get_rate_limit\": \"无法查询到模型对应的配额, 请联系管理员开通\",\n \"gpt-4_over_limit\": \"gpt4 发送消息达到上限, 请联系管理员开通,或增加额度\",\n \"token_length_exceed_limit\": \"总消息长度数量超过上限, 请减少上下文数量、减少消息长度或者开启新的会话\",\n \"invalidEmail\": \"邮箱格式不正确\",\n \"invalidPassword\": \"密码格式不正确, 长度至少为6位, 一个大写字母, 一个小写字母, 一个数字, 一个特殊字符\",\n \"invalidRepwd\": \"重复密码不一致\",\n \"syncChatSession\": \"同步失败, 请稍后再试\",\n \"NotAuthorized\": \"请先登录\",\n \"NotAdmin\": \"非管理员禁止访问\"\n },\n \"workspace\": {\n \"create\": \"创建工作区\",\n \"edit\": \"编辑工作区\",\n \"manage\": \"管理工作区\",\n \"loading\": \"加载工作区中...\",\n \"name\": \"名称\",\n \"namePlaceholder\": \"输入工作区名称\",\n \"nameRequired\": \"工作区名称为必填项\",\n \"description\": \"描述\",\n \"descriptionPlaceholder\": \"为此工作区添加可选描述\",\n \"icon\": \"图标\",\n \"color\": \"颜色\",\n \"saveError\": \"保存工作区失败\",\n \"deleteConfirm\": \"确定要删除此工作区吗？\",\n \"cannotDeleteDefault\": \"无法删除默认工作区\",\n \"switchedTo\": \"已切换到 {name}\",\n \"created\": \"工作区创建成功\",\n \"updated\": \"工作区更新成功\",\n \"deleted\": \"工作区删除成功\",\n \"searchPlaceholder\": \"搜索工作区...\",\n \"totalCount\": \"共 {count} 个工作区\",\n \"filteredResults\": \"/ {total}\",\n \"noWorkspaces\": \"未找到工作区\",\n \"createFirst\": \"创建您的第一个工作区\",\n \"duplicate\": \"复制\",\n \"setAsDefault\": \"设为默认\",\n \"active\": \"当前\",\n \"default\": \"默认\",\n \"sessionCount\": \"{count} 个会话\",\n \"lastUpdated\": \"更新于\",\n \"switchError\": \"切换工作区失败\",\n \"reorderMode\": \"排序模式\",\n \"dragToReorder\": \"拖拽排序\",\n \"reorderSuccess\": \"工作区排序成功\",\n \"reorderError\": \"工作区排序失败\",\n \"invalidColor\": \"颜色格式无效。请使用有效的十六进制颜色。\"\n }\n}\n" }, { "path": "web/src/locales/zh-TW-more.json", "content": "{\n \"common\": {\n \"disabled\": \"停用\",\n \"enabled\": \"啟用\",\n \"create\": \"建立\",\n \"update\": \"更新\"\n },\n \"chat\": {\n \"alreadyInNewChat\": \"已在新的對話中\",\n \"generateMoreSuggestions\": \"產生更多建議\",\n \"generating\": \"生成中...\"\n },\n \"admin\": {\n \"chat_model\": {\n \"default\": \"預設\",\n \"apiType\": \"API 類型\"\n }\n },\n \"workspace\": {\n \"create\": \"建立工作區\",\n \"edit\": \"編輯工作區\",\n \"manage\": \"管理工作區\",\n \"loading\": \"載入工作區中...\",\n \"name\": \"名稱\",\n \"namePlaceholder\": \"輸入工作區名稱\",\n \"nameRequired\": \"工作區名稱為必填項\",\n \"description\": \"描述\",\n \"descriptionPlaceholder\": \"為此工作區添加可選描述\",\n \"icon\": \"圖示\",\n \"color\": \"顏色\",\n \"saveError\": \"儲存工作區失敗\",\n \"deleteConfirm\": \"確定要刪除此工作區嗎？\",\n \"cannotDeleteDefault\": \"無法刪除預設工作區\",\n \"switchedTo\": \"已切換至 {name}\",\n \"created\": \"工作區建立成功\",\n \"updated\": \"工作區更新成功\",\n \"deleted\": \"工作區刪除成功\",\n \"searchPlaceholder\": \"搜尋工作區...\",\n \"totalCount\": \"共 {count} 個工作區\",\n \"filteredResults\": \"/ {total}\",\n \"noWorkspaces\": \"未找到工作區\",\n \"createFirst\": \"建立您的第一個工作區\",\n \"duplicate\": \"複製\",\n \"setAsDefault\": \"設為預設\",\n \"active\": \"目前\",\n \"default\": \"預設\",\n \"sessionCount\": \"{count} 個會話\",\n \"lastUpdated\": \"更新於\",\n \"switchError\": \"切換工作區失敗\",\n \"reorderMode\": \"排序模式\",\n \"dragToReorder\": \"拖曳排序\",\n \"reorderSuccess\": \"工作區排序成功\",\n \"reorderError\": \"工作區排序失敗\",\n \"invalidColor\": \"顏色格式無效。請使用有效的十六進制顏色。\"\n }\n}" }, { "path": "web/src/locales/zh-TW.json", "content": "{\n \"admin\": {\n \"activityHistory\": \"活動歷史\",\n \"add_model\": \"添加模型\",\n \"add_user_model_rate_limit\": \"添加用戶會話數量\",\n \"chat_model\": {\n \"actions\": \"操作\",\n \"apiAuthHeader\": \"Auth Header 鍵\",\n \"apiAuthKey\": \"API KEY對應的環境變量\",\n \"apiType\": \"API 類型\",\n \"clear_form\": \"清空表單\",\n \"copy\": \"複製\",\n \"copy_success\": \"複製成功\",\n \"default\": \"預設\",\n \"defaultToken\": \"默認token數量\",\n \"deleteModel\": \"刪除\",\n \"deleteModelConfirm\": \"確認刪除 {name}?\",\n \"enablePerModeRatelimit\": \"是否單獨流控\",\n \"enablePerModelRateLimit\": \"是否單獨流控\",\n \"isDefault\": \"默認?\",\n \"isEnable\": \"是否啟用\",\n \"label\": \"模型名稱(ID)\",\n \"maxToken\": \"最大token數量\",\n \"name\": \"身分識別號\",\n \"orderNumber\": \"次序\",\n \"paste_json\": \"貼上 JSON 配置\",\n \"paste_json_placeholder\": \"在此處貼上您的模型配置 JSON...\",\n \"populate_form\": \"填充表單\",\n \"url\": \"完整的URL請求\"\n },\n \"chat_model_name\": \"模型ID\",\n \"created\": \"建立時間\",\n \"date\": \"日期\",\n \"firstName\": \"名字\",\n \"hideSubMenu\": \"隱藏子選單\",\n \"lastName\": \"姓氏\",\n \"lastUsed\": \"最後使用\",\n \"messages\": \"訊息\",\n \"messages3Days\": \"訊息數(3天)\",\n \"model\": \"模型\",\n \"modelUsage\": \"模型使用情況\",\n \"modelUsageDistribution\": \"模型使用分佈\",\n \"model_one_default_only\": \"只能有一個默認模型,請先將其他模型設置為非默認\",\n \"name\": \"姓名\",\n \"openPanel\": \"打開管理面板\",\n \"overview\": \"概覽\",\n \"per_model_rate_limit\": {\n \"ChatModelName\": \"模型名稱\",\n \"FullName\": \"姓名\",\n \"RateLimit\": \"10分鐘訪問次數\",\n \"UserEmail\": \"用戶郵箱\",\n \"actions\": \"操作\"\n },\n \"per_model_rate_limit_title\": \"模型流控\",\n \"permission\": \"權限\",\n \"rateLimit\": \"限流\",\n \"rateLimit10Min\": \"訊息數量上限(10分鐘)\",\n \"rate_limit\": \"會話數量(10min)\",\n \"recent3Days\": \"最近3天活動\",\n \"refresh\": \"重新整理\",\n \"sessionHistory\": \"會話歷史\",\n \"sessionId\": \"會話ID\",\n \"sessionSnapshot\": \"會話快照\",\n \"sessions\": \"會話\",\n \"showSubMenu\": \"顯示子選單\",\n \"system_model_tab_title\": \"模型配置\",\n \"title\": \"管理\",\n \"tokens\": \"令牌\",\n \"tokens3Days\": \"令牌數(3天)\",\n \"totalChatMessages\": \"訊息總數\",\n \"totalChatMessages3Days\": \"訊息總數(3天)\",\n \"totalChatMessages3DaysAvgTokenCount\": \"平均token數量(3天)\",\n \"totalChatMessages3DaysTokenCount\": \"總token數量(3天)\",\n \"totalChatMessagesTokenCount\": \"總token數量\",\n \"totalMessages\": \"總訊息數\",\n \"totalSessions\": \"總會話數\",\n \"totalTokens\": \"總令牌數\",\n \"updated\": \"更新時間\",\n \"usage\": \"使用率\",\n \"userAnalysis\": \"用戶分析\",\n \"userEmail\": \"用戶電子郵件\",\n \"userMessage\": \"用戶\",\n \"userStat\": \"用戶統計\"\n },\n \"bot\": {\n \"all\": {\n \"title\": \"機器人\"\n },\n \"list\": \"機器人\",\n \"noHistory\": \"還沒有紀錄\",\n \"runNumber\": \"執行 {number}\",\n \"showCode\": \"產生 API 呼叫程式碼\",\n \"tabs\": {\n \"conversation\": \"預設\",\n \"history\": \"紀錄\"\n }\n },\n \"chat\": {\n \"N\": \"結果數量: {n}\",\n \"addComment\": \"新增評論\",\n \"adjustParameters\": \"調整參數\",\n \"advanced_settings\": \"高級設置\",\n \"alreadyInNewChat\": \"已在新的對話中\",\n \"artifactModeDescription\": \"啟用代碼、預覽和可視化的 Artifact 渲染\",\n \"chatSettings\": \"對話設定\",\n \"debugDescription\": \"啟用調試模式用於故障排除和診斷\",\n \"defaultSystemPrompt\": \"你是一個有幫助且簡明的助手。需要時先提出澄清問題。給出準確答案，並提供簡短理由和可執行步驟。不確定時要說明，並建議如何驗證。\",\n \"exploreModeDescription\": \"基於對話上下文獲取建議問題\",\n \"loading_models\": \"正在載入模型...\",\n \"modes\": \"模式\",\n \"models\": \"個模型\",\n \"chatSnapshot\": \"生成會話記錄\",\n \"clearChat\": \"清空對話\",\n \"clearChatConfirm\": \"是否清空對話?\",\n \"clearHistoryConfirm\": \"確定清空聊天記錄?\",\n \"commentFailed\": \"評論新增失敗\",\n \"commentPlaceholder\": \"請輸入評論...\",\n \"commentSuccess\": \"評論新增成功\",\n \"completionsCount\": \"結果數量: {contextCount}\",\n \"contextCount\": \"上下文數量: {contextCount}\",\n \"contextLength\": \"上下文數量，預設10（會話開始的2條 + 最近的8條）\",\n \"copied\": \"複製成功\",\n \"copy\": \"複製\",\n \"copyCode\": \"複製代碼\",\n \"createBot\": \"建立機器人\",\n \"debug\": \"調試模式\",\n \"artifactMode\": \"Artifacts\",\n \"deleteChatSessionsConfirm\": \"確定刪除此記錄?\",\n \"deleteMessage\": \"刪除消息\",\n \"deleteMessageConfirm\": \"是否刪除此消息?\",\n \"disable_debug\": \"關閉\",\n \"disable_artifact\": \"關閉\",\n \"disable_explore\": \"關閉\",\n \"enable_debug\": \"開啟\",\n \"enable_artifact\": \"開啟\",\n \"enable_explore\": \"開啟\",\n \"exploreMode\": \"探索模式\",\n \"promptInstructions\": \"提示詞說明\",\n \"artifactInstructionTitle\": \"Artifact 說明\",\n \"loading_instructions\": \"正在載入說明...\",\n \"loadingSession\": \"正在載入會話...\",\n \"exportFailed\": \"導出失敗\",\n \"exportImage\": \"導出對話到圖片\",\n \"exportImageConfirm\": \"是否將對話導出為圖片?\",\n \"exportMD\": \"導出對話到Markdown\",\n \"exportMDConfirm\": \"是否將對話導出為Markdown?\",\n \"exportSuccess\": \"導出成功\",\n \"frequencyPenalty\": \"頻率懲罰\",\n \"generateMoreSuggestions\": \"產生更多建議\",\n \"generating\": \"生成中...\",\n \"is_summarize_mode\": \"開啟\",\n \"maxTokens\": \"最大問答總token數量: {maxTokens}\",\n \"model\": \"模型\",\n \"new\": \"新建聊天\",\n \"no_summarize_mode\": \"關閉\",\n \"placeholder\": \"說些什麼...（Shift + Enter = 換行, '/' 触发提示词）\",\n \"placeholderMobile\": \"說些什麼...\",\n \"playAudio\": \"語音\",\n \"presencePenalty\": \"存在懲罰\",\n \"sessionConfig\": \"會話配置\",\n \"snapshotSuccess\": \"快照成功，請在新標籤頁中查看\",\n \"stopAnswer\": \"停止回答\",\n \"suggestedQuestions\": \"建議問題\",\n \"summarize_mode\": \"總結模式(可以支持更長的上下文20+)\",\n \"temperature\": \"溫度: {temperature}\",\n \"topP\": \"Top P: {topP}\",\n \"turnOffContext\": \"在此模式下，發送的消息將不包括以前的聊天記錄。\",\n \"turnOnContext\": \"在此模式下，發送的消息將包括以前的聊天記錄。\",\n \"uploadFiles\": \"上傳檔案\",\n \"uploader_button\": \"上傳\",\n \"uploader_close\": \"關閉\",\n \"uploader_help_text\": \"支援上傳檔案類型： text、image、audio、video\",\n \"uploader_title\": \"上傳檔案\",\n \"usingContext\": \"上下文模式\"\n },\n \"chat_snapshot\": {\n \"createChat\": \"創建會話\",\n \"deleteFailed\": \"无法删除会话记录\",\n \"deletePost\": \"您确定要删除此会话记录吗？\",\n \"deletePostConfirm\": \"删除会话记录\",\n \"deleteSuccess\": \"会话记录删除成功\",\n \"exportImage\": \"保存對話到圖片\",\n \"exportMarkdown\": \"匯出到Markdown檔\",\n \"scrollTop\": \"回到頂部\",\n \"showCode\": \"產生 API 呼叫程式碼\",\n \"title\": \"會話集\"\n },\n \"common\": {\n \"actions\": \"操作\",\n \"ask_user_register\": \"請註冊, 只有註冊帳號才能繼續對話\",\n \"cancel\": \"取消\",\n \"clear\": \"清空\",\n \"confirm\": \"確認\",\n \"copy\": \"複製\",\n \"create\": \"建立\",\n \"delete\": \"刪除\",\n \"disabled\": \"停用\",\n \"edit\": \"編輯\",\n \"editUser\": \"編輯用戶\",\n \"email\": \"郵箱\",\n \"email_placeholder\": \"請輸入電子郵件\",\n \"enabled\": \"啟用\",\n \"export\": \"匯出\",\n \"failed\": \"操作失敗\",\n \"fetchFailed\": \"獲取數據失敗\",\n \"help\": \"第一個是主題(prompt)，上下文包括10條信息\",\n \"import\": \"匯入\",\n \"login\": \"登錄\",\n \"loginSuccess\": \"登入成功\",\n \"login_failed\": \"登入失敗\",\n \"logout\": \"登出\",\n \"logout_failed\": \"登出失敗\",\n \"logout_success\": \"登出成功\",\n \"no\": \"否\",\n \"noData\": \"暫無數據\",\n \"password_placeholder\": \"請輸入密碼\",\n \"please_register\": \"請先註冊\",\n \"reset\": \"重置\",\n \"regenerate\": \"重新生成\",\n \"save\": \"保存\",\n \"signup\": \"註冊\",\n \"signup_failed\": \"註冊失敗\",\n \"signup_success\": \"註冊成功\",\n \"submit\": \"提交\",\n \"submitting\": \"提交中...\",\n \"success\": \"操作成功\",\n \"unauthorizedTips\": \"請註冊或登錄\",\n \"update\": \"更新\",\n \"use\": \"使用\",\n \"verify\": \"註冊\",\n \"warning\": \"警告\",\n \"wrong\": \"好像出錯了，請稍後再試。\",\n \"yes\": \"是\"\n },\n \"error\": {\n \"INTN_004\": \"請求模型失敗, 請稍後再試, 或聯繫管理員\",\n \"MODEL_001\": \"第一條是系統消息，請繼續輸入信息開始會話\",\n \"MODEL_006\": \"無法從模型取得回覆\",\n \"NotAdmin\": \"非管理員禁止訪問\",\n \"NotAuthorized\": \"請先登入\",\n \"RESOURCE_EXHAUSTED\": \"資源耗盡\",\n \"VALD_001\": \"無效請求\",\n \"VALD_004\": \"無效的電子郵件或密碼\",\n \"fail_to_get_rate_limit\": \"無法查詢模型對應的配額，請聯繫管理員開通\",\n \"gpt-4_over_limit\": \"gpt4 發送消息達到上限，請聯繫管理員開通，或增加額度\",\n \"invalidEmail\": \"電子郵件格式不正確\",\n \"invalidPassword\": \"密碼格式不正確，長度至少需為6位，應包含1個大寫字母、1個小寫字母、1個數字、1個特殊字符\",\n \"invalidRepwd\": \"重復密碼不一致\",\n \"rateLimit\": \"發送消息數量已經達到上限，請聯繫管理員\",\n \"syncChatSession\": \"同步失敗，請稍後再試\",\n \"token_length_exceed_limit\": \"總消息長度數量超過上限，請減少上下文數量、減少消息長度或者開啟新的會話\"\n },\n \"prompt\": {\n \"add\": \"新增\",\n \"addFailed\": \"新增失敗\",\n \"addSuccess\": \"新增成功\",\n \"clear\": \"清空\",\n \"confirmClear\": \"是否清空？\",\n \"delete\": \"刪除\",\n \"deleteConfirm\": \"是否刪除？\",\n \"deleteFailed\": \"刪除失敗\",\n \"deleteSuccess\": \"刪除成功\",\n \"download\": \"下載\",\n \"downloadOnline\": \"線上匯入\",\n \"downloadOnlineWarning\": \"注意：請檢查下載 JSON 檔案來源，惡意的 JSON 檔案可能會破壞您的電腦！\",\n \"edit\": \"編輯\",\n \"editFailed\": \"編輯失敗\",\n \"editSuccess\": \"編輯成功\",\n \"enterJsonUrl\": \"請輸入正確的 JSON 位址\",\n \"export\": \"匯出\",\n \"import\": \"匯入\",\n \"store\": \"提示词\"\n },\n \"setting\": {\n \"admin\": \"管理\",\n \"api\": \"API\",\n \"apiToken\": \"API Token\",\n \"apiTokenCopied\": \"API Token 已複製\",\n \"apiTokenCopyFailed\": \"無法複製 API Token\",\n \"avatarLink\": \"頭像鏈接\",\n \"chatHistory\": \"聊天記錄\",\n \"config\": \"配置\",\n \"defaultDesc\": \"签名\",\n \"defaultName\": \"你\",\n \"description\": \"描述\",\n \"general\": \"總覽\",\n \"language\": \"語言\",\n \"name\": \"名稱\",\n \"resetUserInfo\": \"重置用戶信息\",\n \"reverseProxy\": \"反向代理\",\n \"setting\": \"設置\",\n \"snapshotLink\": \"會话集\",\n \"socks\": \"Socks\",\n \"switchLanguage\": \"English(切换语言)\",\n \"theme\": \"主題\",\n \"timeout\": \"超時\"\n },\n \"workspace\": {\n \"active\": \"目前\",\n \"cannotDeleteDefault\": \"無法刪除預設工作區\",\n \"color\": \"顏色\",\n \"create\": \"建立工作區\",\n \"createFirst\": \"建立您的第一個工作區\",\n \"created\": \"工作區建立成功\",\n \"default\": \"預設\",\n \"deleteConfirm\": \"確定要刪除此工作區嗎？\",\n \"deleted\": \"工作區刪除成功\",\n \"description\": \"描述\",\n \"descriptionPlaceholder\": \"為此工作區添加可選描述\",\n \"dragToReorder\": \"拖曳排序\",\n \"duplicate\": \"複製\",\n \"edit\": \"編輯工作區\",\n \"filteredResults\": \"/ {total}\",\n \"icon\": \"圖示\",\n \"invalidColor\": \"顏色格式無效。請使用有效的十六進制顏色。\",\n \"lastUpdated\": \"更新於\",\n \"loading\": \"載入工作區中...\",\n \"manage\": \"管理工作區\",\n \"name\": \"名稱\",\n \"namePlaceholder\": \"輸入工作區名稱\",\n \"nameRequired\": \"工作區名稱為必填項\",\n \"noWorkspaces\": \"未找到工作區\",\n \"reorderError\": \"工作區排序失敗\",\n \"reorderMode\": \"排序模式\",\n \"reorderSuccess\": \"工作區排序成功\",\n \"saveError\": \"儲存工作區失敗\",\n \"searchPlaceholder\": \"搜尋工作區...\",\n \"sessionCount\": \"{count} 個會話\",\n \"setAsDefault\": \"設為預設\",\n \"switchError\": \"切換工作區失敗\",\n \"switchedTo\": \"已切換至 {name}\",\n \"totalCount\": \"共 {count} 個工作區\",\n \"updated\": \"工作區更新成功\"\n }\n}\n" }, { "path": "web/src/main.ts", "content": "import { createApp } from 'vue'\nimport { VueQueryPlugin } from '@tanstack/vue-query'\nimport App from './App.vue'\nimport { setupI18n } from './locales'\nimport { setupAssets } from './plugins'\nimport { setupStore } from './store'\nimport { setupRouter } from './router'\n\nasync function bootstrap() {\n const app = createApp(App)\n setupAssets()\n\n setupStore(app)\n\n setupI18n(app)\n\n await setupRouter(app)\n\n app.use(VueQueryPlugin)\n app.mount('#app')\n}\n\nbootstrap()\n" }, { "path": "web/src/plugins/assets.ts", "content": "import 'katex/dist/katex.min.css'\nimport '@/styles/lib/tailwind.css'\nimport '@/styles/lib/highlight.less'\nimport '@/styles/lib/github-markdown.less'\nimport '@/styles/global.less'\n\n/** Tailwind's Preflight Style Override */\nfunction naiveStyleOverride() {\n const meta = document.createElement('meta')\n meta.name = 'naive-ui-style'\n document.head.appendChild(meta)\n}\n\nfunction setupAssets() {\n naiveStyleOverride()\n}\n\nexport default setupAssets\n" }, { "path": "web/src/plugins/index.ts", "content": "import setupAssets from './assets'\n\nexport { setupAssets }\n" }, { "path": "web/src/router/index.ts", "content": "import type { App } from 'vue'\nimport type { RouteRecordRaw } from 'vue-router'\nimport { createRouter, createWebHashHistory } from 'vue-router'\nimport { setupPageGuard } from './permission'\nimport { ChatLayout } from '@/views/chat/layout'\n\nconst routes: RouteRecordRaw[] = [\n {\n path: '/snapshot',\n name: 'Snapshot',\n component: () => import('@/views/snapshot/page.vue'),\n children: [\n {\n path: ':uuid?',\n name: 'Snapshot',\n component: () => import('@/views/snapshot/page.vue'),\n },\n ],\n },\n {\n path: '/prompt/new',\n name: 'Prompt',\n component: () => import('@/views/prompt/creator.vue')\n },\n {\n path: '/bot',\n name: 'Bot',\n component: () => import('@/views/bot/page.vue'),\n children: [\n {\n path: ':uuid?',\n name: 'Bot',\n component: () => import('@/views/bot/page.vue'),\n },\n ],\n },\n {\n path: '/snapshot_all',\n name: 'SnapshotAll',\n component: () => import('@/views/snapshot/all.vue'),\n },\n {\n path: '/bot_all',\n name: 'BotAll',\n component: () => import('@/views/bot/all.vue'),\n },\n {\n path: '/admin',\n name: 'Admin',\n component: () => import('@/views/admin/index.vue'),\n children: [\n {\n path: 'user',\n name: 'AdminUser',\n component: () => import('@/views/admin/user/index.vue'),\n },\n {\n path: 'model',\n name: 'AdminModel',\n component: () => import('@/views/admin/model/index.vue'),\n },\n {\n path: 'model_rate_limit',\n name: 'ModelRateLimit',\n component: () => import('@/views/admin/modelRateLimit/index.vue'),\n }\n ],\n },\n {\n path: '/',\n name: 'Root',\n component: ChatLayout,\n children: [\n {\n path: '/workspace/:workspaceUuid/chat/:uuid?',\n name: 'WorkspaceChat',\n component: () => import('@/views/chat/index.vue'),\n props: true,\n },\n {\n path: '/workspace/:workspaceUuid',\n redirect: to => {\n // Redirect workspace-only URLs to include /chat\n return `/workspace/${to.params.workspaceUuid}/chat`\n }\n },\n {\n path: '/',\n name: 'DefaultWorkspace',\n component: () => import('@/views/chat/index.vue'),\n props: true,\n },\n ],\n },\n {\n path: '/404',\n name: '404',\n component: () => import('@/views/exception/404/index.vue'),\n },\n {\n path: '/500',\n name: '500',\n component: () => import('@/views/exception/500/index.vue'),\n },\n {\n path: '/:pathMatch(.*)*',\n name: 'notFound',\n redirect: '/404',\n },\n]\n\n// !!!\n// https://router.vuejs.org/guide/essentials/history-mode.html\n// createWebHashHistory\n// It uses a hash character (#) before the actual URL that is internally passed.\n// Because this section of the URL is never sent to the server,\n// it doesn't require any special treatment on the server level.\n// It does however have a bad impact in SEO. If that's a concern for you, use the HTML5 history mode.\n\n// this is crazy, router in frontend is a nightmare\n\nexport const router = createRouter({\n history: createWebHashHistory(),\n routes,\n scrollBehavior: () => ({ left: 0, top: 0 }),\n})\n\nsetupPageGuard(router)\n\nexport async function setupRouter(app: App) {\n app.use(router)\n await router.isReady()\n}\n" }, { "path": "web/src/router/permission.ts", "content": "import type { Router } from 'vue-router'\nimport { useAuthStore } from '@/store/modules/auth'\nimport { useWorkspaceStore } from '@/store/modules/workspace'\nimport { useSessionStore } from '@/store/modules/session'\nimport { store } from '@/store'\n\nconst FIVE_MINUTES_IN_SECONDS = 5 * 60\n\n// Attempt to ensure we have a valid access token before continuing navigation\nasync function ensureFreshToken(authStore: any) {\n const currentTs = Math.floor(Date.now() / 1000)\n const expiresIn = authStore.getExpiresIn\n const token = authStore.getToken\n // the user hasn’t logged in\n if (!token && !expiresIn)\n return\n\n // If we already have a token that is valid for some time, nothing to do\n if (token && expiresIn && expiresIn > currentTs + FIVE_MINUTES_IN_SECONDS)\n return\n\n try {\n await authStore.refreshToken()\n }\n catch (error) {\n // If refresh fails, make sure state is cleared so UI can prompt user\n authStore.removeToken()\n authStore.removeExpiresIn()\n }\n}\n\nexport function setupPageGuard(router: Router) {\n router.beforeEach(async (to, from, next) => {\n const auth_store = useAuthStore(store)\n await ensureFreshToken(auth_store)\n\n // Handle workspace context from URL\n if (to.name === 'WorkspaceChat' && to.params.workspaceUuid) {\n const workspaceStore = useWorkspaceStore(store)\n const sessionStore = useSessionStore(store)\n const workspaceUuid = to.params.workspaceUuid as string\n\n // Only set active workspace if it's different and not already loading\n if (workspaceUuid !== workspaceStore.activeWorkspaceUuid && !workspaceStore.isLoading) {\n console.log('Setting workspace from URL:', workspaceUuid)\n await workspaceStore.setActiveWorkspace(workspaceUuid)\n }\n\n // Set active session if provided in URL\n if (to.params.uuid) {\n const sessionUuid = to.params.uuid as string\n if (sessionUuid !== sessionStore.activeSessionUuid) {\n await sessionStore.setActiveSession(workspaceUuid, sessionUuid)\n }\n }\n }\n\n // Handle default route - let store sync handle navigation to default workspace\n if (to.name === 'DefaultWorkspace') {\n console.log('On default route, letting store handle workspace navigation')\n }\n\n next()\n })\n}\n" }, { "path": "web/src/service/snapshot.ts", "content": "import { displayLocaleDate, formatYearMonth } from '@/utils/date'\n\n\n\n\n\nexport function generateAPIHelper(uuid: string, apiToken: string, origin: string) {\n const data = {\n \"message\": \"Your message here\",\n \"snapshot_uuid\": uuid,\n \"stream\": false,\n }\n return `curl -X POST ${origin}/api/chatbot -H \"Content-Type: application/json\" -H \"Authorization: Bearer ${apiToken}\" -d '${JSON.stringify(data)}'`\n}\n\nexport function getChatbotPosts(posts: Snapshot.Snapshot[]) {\n return posts\n .filter((post: Snapshot.Snapshot) => post.typ === 'chatbot')\n .map((post: Snapshot.Snapshot): Snapshot.PostLink => ({\n uuid: post.uuid,\n date: displayLocaleDate(post.createdAt),\n title: post.title,\n }))\n}\n\nexport function getSnapshotPosts(posts: Snapshot.Snapshot[]) {\n return posts\n .filter((post: Snapshot.Snapshot) => post.typ === 'snapshot')\n .map((post: Snapshot.Snapshot): Snapshot.PostLink => ({\n uuid: post.uuid,\n date: displayLocaleDate(post.createdAt),\n title: post.title,\n }))\n}\n\nexport function postsByYearMonthTransform(posts: Snapshot.PostLink[]) {\n const init: Record = {}\n return posts.reduce((acc, post) => {\n const yearMonth = formatYearMonth(new Date(post.date))\n if (!acc[yearMonth])\n acc[yearMonth] = []\n\n acc[yearMonth].push(post)\n return acc\n }, init)\n}\n\nexport function getSnapshotPostLinks(snapshots: Snapshot.Snapshot[]): Record {\n const snapshotPosts = getSnapshotPosts(snapshots)\n return postsByYearMonthTransform(snapshotPosts)\n}\n\nexport function getBotPostLinks(bots: Snapshot.Snapshot[]): Record {\n const chatbotPosts = getChatbotPosts(bots)\n return postsByYearMonthTransform(chatbotPosts)\n}" }, { "path": "web/src/services/codeTemplates.ts", "content": "/**\n * Code Templates and Snippets Service\n * Manages pre-built code templates and user-defined snippets\n */\n\nimport { reactive, computed } from 'vue'\n\nexport interface CodeTemplate {\n id: string\n name: string\n description: string\n language: string\n code: string\n category: string\n tags: string[]\n author?: string\n createdAt: string\n updatedAt: string\n difficulty: 'beginner' | 'intermediate' | 'advanced'\n estimatedRunTime?: number\n requirements?: string[]\n isBuiltIn: boolean\n usageCount: number\n rating?: number\n examples?: {\n input?: string\n output?: string\n description?: string\n }[]\n}\n\nexport interface TemplateCategory {\n id: string\n name: string\n description: string\n icon: string\n color: string\n templates: CodeTemplate[]\n}\n\nclass CodeTemplatesService {\n private static instance: CodeTemplatesService\n private templates: CodeTemplate[] = reactive([])\n private categories: TemplateCategory[] = reactive([])\n private storageKey = 'code-templates'\n private userTemplatesKey = 'user-code-templates'\n\n private constructor() {\n this.initializeBuiltInTemplates()\n this.loadUserTemplates()\n }\n\n static getInstance(): CodeTemplatesService {\n if (!CodeTemplatesService.instance) {\n CodeTemplatesService.instance = new CodeTemplatesService()\n }\n return CodeTemplatesService.instance\n }\n\n /**\n * Initialize built-in templates\n */\n private initializeBuiltInTemplates(): void {\n const builtInTemplates: CodeTemplate[] = [\n // JavaScript Templates\n {\n id: 'js-hello-world',\n name: 'Hello World',\n description: 'Basic Hello World example',\n language: 'javascript',\n code: `console.log('Hello, World!')\nconsole.log('Welcome to JavaScript!')`,\n category: 'basics',\n tags: ['beginner', 'console', 'output'],\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n difficulty: 'beginner',\n estimatedRunTime: 50,\n isBuiltIn: true,\n usageCount: 0,\n rating: 5,\n examples: [\n {\n output: 'Hello, World!\\nWelcome to JavaScript!',\n description: 'Simple console output'\n }\n ]\n },\n {\n id: 'js-async-fetch',\n name: 'Async Data Fetching',\n description: 'Demonstrate async/await with simulated API calls',\n language: 'javascript',\n code: `// Simulate an API call\nasync function fetchData(url) {\n console.log(\\`Fetching data from: \\${url}\\`)\n \n // Simulate network delay\n await new Promise(resolve => setTimeout(resolve, 1000))\n \n // Simulate API response\n return {\n data: [1, 2, 3, 4, 5],\n timestamp: new Date().toISOString(),\n status: 'success'\n }\n}\n\n// Using async/await\nasync function main() {\n try {\n console.log('Starting data fetch...')\n const result = await fetchData('https://api.example.com/data')\n console.log('Data received:', result)\n console.log('Processing complete!')\n } catch (error) {\n console.error('Error fetching data:', error)\n }\n}\n\nmain()`,\n category: 'async',\n tags: ['async', 'await', 'promises', 'intermediate'],\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n difficulty: 'intermediate',\n estimatedRunTime: 1200,\n isBuiltIn: true,\n usageCount: 0,\n rating: 4.5\n },\n {\n id: 'js-canvas-animation',\n name: 'Canvas Animation',\n description: 'Create animated graphics using Canvas API',\n language: 'javascript',\n code: `// Create a canvas for animation\nconst canvas = createCanvas(400, 300)\nconst ctx = canvas.getContext('2d')\n\n// Animation parameters\nlet frame = 0\nconst colors = ['#FF6B6B', '#4ECDC4', '#45B7D1', '#96CEB4', '#FFEAA7']\n\n// Animation loop\nfunction animate() {\n // Clear canvas\n ctx.clearRect(0, 0, canvas.width, canvas.height)\n \n // Draw animated circles\n for (let i = 0; i < 5; i++) {\n const x = 200 + Math.sin(frame * 0.02 + i) * 100\n const y = 150 + Math.cos(frame * 0.03 + i) * 50\n const radius = 20 + Math.sin(frame * 0.05 + i) * 10\n \n ctx.beginPath()\n ctx.arc(x, y, radius, 0, Math.PI * 2)\n ctx.fillStyle = colors[i]\n ctx.fill()\n }\n \n frame++\n \n // Continue animation\n if (frame < 200) {\n setTimeout(animate, 50)\n }\n}\n\nconsole.log('Starting canvas animation...')\nanimate()`,\n category: 'graphics',\n tags: ['canvas', 'animation', 'graphics', 'intermediate'],\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n difficulty: 'intermediate',\n estimatedRunTime: 10000,\n isBuiltIn: true,\n usageCount: 0,\n rating: 4.8\n },\n {\n id: 'js-data-structures',\n name: 'Data Structures Demo',\n description: 'Demonstrate common data structures and algorithms',\n language: 'javascript',\n code: `// Stack implementation\nclass Stack {\n constructor() {\n this.items = []\n }\n \n push(item) {\n this.items.push(item)\n }\n \n pop() {\n return this.items.pop()\n }\n \n peek() {\n return this.items[this.items.length - 1]\n }\n \n isEmpty() {\n return this.items.length === 0\n }\n}\n\n// Queue implementation\nclass Queue {\n constructor() {\n this.items = []\n }\n \n enqueue(item) {\n this.items.push(item)\n }\n \n dequeue() {\n return this.items.shift()\n }\n \n front() {\n return this.items[0]\n }\n \n isEmpty() {\n return this.items.length === 0\n }\n}\n\n// Demonstrate usage\nconsole.log('=== Stack Demo ===')\nconst stack = new Stack()\nstack.push(1)\nstack.push(2)\nstack.push(3)\nconsole.log('Stack after pushes:', stack.items)\nconsole.log('Popped:', stack.pop())\nconsole.log('Stack after pop:', stack.items)\n\nconsole.log('\\\\n=== Queue Demo ===')\nconst queue = new Queue()\nqueue.enqueue('A')\nqueue.enqueue('B')\nqueue.enqueue('C')\nconsole.log('Queue after enqueues:', queue.items)\nconsole.log('Dequeued:', queue.dequeue())\nconsole.log('Queue after dequeue:', queue.items)\n\n// Binary search\nfunction binarySearch(arr, target) {\n let left = 0\n let right = arr.length - 1\n \n while (left <= right) {\n const mid = Math.floor((left + right) / 2)\n \n if (arr[mid] === target) {\n return mid\n } else if (arr[mid] < target) {\n left = mid + 1\n } else {\n right = mid - 1\n }\n }\n \n return -1\n}\n\nconsole.log('\\\\n=== Binary Search Demo ===')\nconst sortedArray = [1, 3, 5, 7, 9, 11, 13, 15, 17, 19]\nconsole.log('Array:', sortedArray)\nconsole.log('Search for 7:', binarySearch(sortedArray, 7))\nconsole.log('Search for 12:', binarySearch(sortedArray, 12))`,\n category: 'algorithms',\n tags: ['data-structures', 'algorithms', 'stack', 'queue', 'search', 'advanced'],\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n difficulty: 'advanced',\n estimatedRunTime: 200,\n isBuiltIn: true,\n usageCount: 0,\n rating: 4.7\n },\n // Python Templates\n {\n id: 'py-hello-world',\n name: 'Hello World',\n description: 'Basic Hello World example in Python',\n language: 'python',\n code: `print(\"Hello, World!\")\nprint(\"Welcome to Python!\")\n\n# Variables and basic operations\nname = \"Python\"\nversion = 3.9\nprint(f\"Language: {name}, Version: {version}\")\n\n# List operations\nnumbers = [1, 2, 3, 4, 5]\nprint(f\"Numbers: {numbers}\")\nprint(f\"Sum: {sum(numbers)}\")\nprint(f\"Max: {max(numbers)}\")`,\n category: 'basics',\n tags: ['beginner', 'print', 'variables', 'lists'],\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n difficulty: 'beginner',\n estimatedRunTime: 100,\n isBuiltIn: true,\n usageCount: 0,\n rating: 5,\n requirements: []\n },\n {\n id: 'py-data-analysis',\n name: 'Data Analysis with Pandas',\n description: 'Basic data analysis using pandas and numpy',\n language: 'python',\n code: `import pandas as pd\nimport numpy as np\n\n# Create sample data\ndata = {\n 'Name': ['Alice', 'Bob', 'Charlie', 'Diana', 'Eve'],\n 'Age': [25, 30, 35, 28, 32],\n 'City': ['New York', 'London', 'Paris', 'Tokyo', 'Sydney'],\n 'Salary': [50000, 60000, 70000, 55000, 65000]\n}\n\n# Create DataFrame\ndf = pd.DataFrame(data)\nprint(\"Original DataFrame:\")\nprint(df)\nprint()\n\n# Basic statistics\nprint(\"Basic Statistics:\")\nprint(df.describe())\nprint()\n\n# Data filtering\nprint(\"People over 30:\")\nprint(df[df['Age'] > 30])\nprint()\n\n# Data aggregation\nprint(\"Average salary by city:\")\ncity_avg = df.groupby('City')['Salary'].mean()\nprint(city_avg)\nprint()\n\n# Adding a new column\ndf['Salary_USD'] = df['Salary']\ndf['Salary_EUR'] = df['Salary'] * 0.85 # Example conversion\nprint(\"DataFrame with new columns:\")\nprint(df[['Name', 'Salary_USD', 'Salary_EUR']])`,\n category: 'data-science',\n tags: ['pandas', 'numpy', 'data-analysis', 'intermediate'],\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n difficulty: 'intermediate',\n estimatedRunTime: 500,\n requirements: ['pandas', 'numpy'],\n isBuiltIn: true,\n usageCount: 0,\n rating: 4.6\n },\n {\n id: 'py-matplotlib-plots',\n name: 'Data Visualization with Matplotlib',\n description: 'Create various types of plots using matplotlib',\n language: 'python',\n code: `import matplotlib.pyplot as plt\nimport numpy as np\n\n# Create sample data\nx = np.linspace(0, 10, 100)\ny1 = np.sin(x)\ny2 = np.cos(x)\ny3 = np.sin(x) * np.cos(x)\n\n# Create the plot\nplt.figure(figsize=(10, 6))\n\n# Plot multiple lines\nplt.plot(x, y1, 'b-', label='sin(x)', linewidth=2)\nplt.plot(x, y2, 'r--', label='cos(x)', linewidth=2)\nplt.plot(x, y3, 'g:', label='sin(x)*cos(x)', linewidth=2)\n\n# Customize the plot\nplt.title('Trigonometric Functions', fontsize=16, fontweight='bold')\nplt.xlabel('x', fontsize=12)\nplt.ylabel('y', fontsize=12)\nplt.legend(fontsize=10)\nplt.grid(True, alpha=0.3)\n\n# Add some styling\nplt.tight_layout()\nplt.show()\n\nprint(\"Matplotlib plot created successfully!\")\nprint(\"The plot shows three trigonometric functions:\")\nprint(\"- Blue solid line: sin(x)\")\nprint(\"- Red dashed line: cos(x)\")\nprint(\"- Green dotted line: sin(x)*cos(x)\")`,\n category: 'visualization',\n tags: ['matplotlib', 'visualization', 'plots', 'numpy', 'intermediate'],\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n difficulty: 'intermediate',\n estimatedRunTime: 800,\n requirements: ['matplotlib', 'numpy'],\n isBuiltIn: true,\n usageCount: 0,\n rating: 4.8\n },\n {\n id: 'py-machine-learning',\n name: 'Simple Machine Learning',\n description: 'Basic machine learning example with scikit-learn',\n language: 'python',\n code: `import numpy as np\nfrom sklearn.model_selection import train_test_split\nfrom sklearn.linear_model import LinearRegression\nfrom sklearn.metrics import mean_squared_error, r2_score\nimport matplotlib.pyplot as plt\n\n# Generate sample data\nnp.random.seed(42)\nX = np.random.rand(100, 1) * 10 # Features\ny = 2 * X.ravel() + 1 + np.random.randn(100) * 2 # Target with noise\n\nprint(\"Generated dataset:\")\nprint(f\"Features shape: {X.shape}\")\nprint(f\"Target shape: {y.shape}\")\nprint(f\"Feature range: [{X.min():.2f}, {X.max():.2f}]\")\nprint(f\"Target range: [{y.min():.2f}, {y.max():.2f}]\")\nprint()\n\n# Split the data\nX_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)\n\n# Create and train the model\nmodel = LinearRegression()\nmodel.fit(X_train, y_train)\n\n# Make predictions\ny_pred = model.predict(X_test)\n\n# Calculate metrics\nmse = mean_squared_error(y_test, y_pred)\nr2 = r2_score(y_test, y_pred)\n\nprint(\"Model Performance:\")\nprint(f\"Mean Squared Error: {mse:.2f}\")\nprint(f\"R² Score: {r2:.2f}\")\nprint(f\"Model coefficients: {model.coef_[0]:.2f}\")\nprint(f\"Model intercept: {model.intercept_:.2f}\")\nprint()\n\n# Visualize results\nplt.figure(figsize=(10, 6))\nplt.scatter(X_test, y_test, color='blue', label='Actual', alpha=0.7)\nplt.scatter(X_test, y_pred, color='red', label='Predicted', alpha=0.7)\nplt.plot(X_test, y_pred, color='red', linewidth=2)\nplt.xlabel('Feature')\nplt.ylabel('Target')\nplt.title('Linear Regression: Actual vs Predicted')\nplt.legend()\nplt.grid(True, alpha=0.3)\nplt.show()\n\nprint(\"Machine learning model trained and visualized successfully!\")`,\n category: 'machine-learning',\n tags: ['scikit-learn', 'machine-learning', 'regression', 'matplotlib', 'advanced'],\n createdAt: new Date().toISOString(),\n updatedAt: new Date().toISOString(),\n difficulty: 'advanced',\n estimatedRunTime: 1500,\n requirements: ['scikit-learn', 'matplotlib', 'numpy'],\n isBuiltIn: true,\n usageCount: 0,\n rating: 4.9\n }\n ]\n\n this.templates.push(...builtInTemplates)\n this.initializeCategories()\n }\n\n /**\n * Initialize template categories\n */\n private initializeCategories(): void {\n this.categories.push(\n {\n id: 'basics',\n name: 'Basics',\n description: 'Fundamental programming concepts',\n icon: 'ri:book-line',\n color: '#4ECDC4',\n templates: this.templates.filter(t => t.category === 'basics')\n },\n {\n id: 'async',\n name: 'Async Programming',\n description: 'Asynchronous programming patterns',\n icon: 'ri:time-line',\n color: '#45B7D1',\n templates: this.templates.filter(t => t.category === 'async')\n },\n {\n id: 'graphics',\n name: 'Graphics & Animation',\n description: 'Canvas graphics and animations',\n icon: 'ri:palette-line',\n color: '#FF6B6B',\n templates: this.templates.filter(t => t.category === 'graphics')\n },\n {\n id: 'algorithms',\n name: 'Algorithms',\n description: 'Data structures and algorithms',\n icon: 'ri:mind-map',\n color: '#96CEB4',\n templates: this.templates.filter(t => t.category === 'algorithms')\n },\n {\n id: 'data-science',\n name: 'Data Science',\n description: 'Data analysis and manipulation',\n icon: 'ri:bar-chart-line',\n color: '#FFEAA7',\n templates: this.templates.filter(t => t.category === 'data-science')\n },\n {\n id: 'visualization',\n name: 'Data Visualization',\n description: 'Charts and plots',\n icon: 'ri:line-chart-line',\n color: '#DDA0DD',\n templates: this.templates.filter(t => t.category === 'visualization')\n },\n {\n id: 'machine-learning',\n name: 'Machine Learning',\n description: 'ML algorithms and models',\n icon: 'ri:brain-line',\n color: '#FFA07A',\n templates: this.templates.filter(t => t.category === 'machine-learning')\n }\n )\n }\n\n /**\n * Get all templates\n */\n getTemplates(): CodeTemplate[] {\n return [...this.templates]\n }\n\n /**\n * Get templates by category\n */\n getTemplatesByCategory(categoryId: string): CodeTemplate[] {\n return this.templates.filter(t => t.category === categoryId)\n }\n\n /**\n * Get templates by language\n */\n getTemplatesByLanguage(language: string): CodeTemplate[] {\n return this.templates.filter(t => t.language.toLowerCase() === language.toLowerCase())\n }\n\n /**\n * Get template by ID\n */\n getTemplate(id: string): CodeTemplate | undefined {\n return this.templates.find(t => t.id === id)\n }\n\n /**\n * Search templates\n */\n searchTemplates(query: string, filters?: {\n language?: string\n category?: string\n difficulty?: string\n tags?: string[]\n }): CodeTemplate[] {\n let results = this.templates\n\n // Text search\n if (query.trim()) {\n const searchTerm = query.toLowerCase()\n results = results.filter(template => \n template.name.toLowerCase().includes(searchTerm) ||\n template.description.toLowerCase().includes(searchTerm) ||\n template.tags.some(tag => tag.toLowerCase().includes(searchTerm)) ||\n template.code.toLowerCase().includes(searchTerm)\n )\n }\n\n // Apply filters\n if (filters) {\n if (filters.language) {\n results = results.filter(t => t.language.toLowerCase() === filters.language!.toLowerCase())\n }\n \n if (filters.category) {\n results = results.filter(t => t.category === filters.category)\n }\n \n if (filters.difficulty) {\n results = results.filter(t => t.difficulty === filters.difficulty)\n }\n \n if (filters.tags && filters.tags.length > 0) {\n results = results.filter(t => \n filters.tags!.some(tag => t.tags.includes(tag))\n )\n }\n }\n\n return results\n }\n\n /**\n * Get all categories\n */\n getCategories(): TemplateCategory[] {\n // Update template counts\n this.categories.forEach(category => {\n category.templates = this.templates.filter(t => t.category === category.id)\n })\n return [...this.categories]\n }\n\n /**\n * Get popular templates\n */\n getPopularTemplates(limit = 10): CodeTemplate[] {\n return this.templates\n .sort((a, b) => b.usageCount - a.usageCount)\n .slice(0, limit)\n }\n\n /**\n * Get recent templates\n */\n getRecentTemplates(limit = 10): CodeTemplate[] {\n return this.templates\n .sort((a, b) => new Date(b.updatedAt).getTime() - new Date(a.updatedAt).getTime())\n .slice(0, limit)\n }\n\n /**\n * Add a new template\n */\n addTemplate(template: Omit): string {\n const id = this.generateId()\n const now = new Date().toISOString()\n \n const newTemplate: CodeTemplate = {\n id,\n ...template,\n createdAt: now,\n updatedAt: now,\n isBuiltIn: false,\n usageCount: 0\n }\n \n this.templates.push(newTemplate)\n this.saveUserTemplates()\n \n return id\n }\n\n /**\n * Update a template\n */\n updateTemplate(id: string, updates: Partial): boolean {\n const template = this.templates.find(t => t.id === id)\n if (!template || template.isBuiltIn) {\n return false\n }\n \n Object.assign(template, updates, { updatedAt: new Date().toISOString() })\n this.saveUserTemplates()\n \n return true\n }\n\n /**\n * Delete a template\n */\n deleteTemplate(id: string): boolean {\n const index = this.templates.findIndex(t => t.id === id)\n if (index === -1 || this.templates[index].isBuiltIn) {\n return false\n }\n \n this.templates.splice(index, 1)\n this.saveUserTemplates()\n \n return true\n }\n\n /**\n * Increment template usage count\n */\n incrementUsage(id: string): void {\n const template = this.templates.find(t => t.id === id)\n if (template) {\n template.usageCount++\n this.saveUserTemplates()\n }\n }\n\n /**\n * Rate a template\n */\n rateTemplate(id: string, rating: number): boolean {\n const template = this.templates.find(t => t.id === id)\n if (!template || rating < 1 || rating > 5) {\n return false\n }\n \n template.rating = rating\n this.saveUserTemplates()\n \n return true\n }\n\n /**\n * Export templates\n */\n exportTemplates(): string {\n const userTemplates = this.templates.filter(t => !t.isBuiltIn)\n return JSON.stringify(userTemplates, null, 2)\n }\n\n /**\n * Import templates\n */\n importTemplates(jsonData: string): boolean {\n try {\n const importedTemplates = JSON.parse(jsonData) as CodeTemplate[]\n \n // Validate imported data\n if (!Array.isArray(importedTemplates)) {\n throw new Error('Invalid format: expected array')\n }\n \n // Add imported templates\n importedTemplates.forEach(template => {\n const existingTemplate = this.templates.find(t => t.id === template.id)\n if (!existingTemplate) {\n template.isBuiltIn = false\n template.usageCount = template.usageCount || 0\n this.templates.push(template)\n }\n })\n \n this.saveUserTemplates()\n return true\n } catch (error) {\n console.error('Failed to import templates:', error)\n return false\n }\n }\n\n private generateId(): string {\n return `template_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`\n }\n\n private saveUserTemplates(): void {\n try {\n const userTemplates = this.templates.filter(t => !t.isBuiltIn)\n localStorage.setItem(this.userTemplatesKey, JSON.stringify(userTemplates))\n } catch (error) {\n console.warn('Failed to save user templates:', error)\n }\n }\n\n private loadUserTemplates(): void {\n try {\n const stored = localStorage.getItem(this.userTemplatesKey)\n if (stored) {\n const userTemplates = JSON.parse(stored) as CodeTemplate[]\n this.templates.push(...userTemplates)\n }\n } catch (error) {\n console.warn('Failed to load user templates:', error)\n }\n }\n}\n\n// Export singleton instance\nexport const codeTemplates = CodeTemplatesService.getInstance()\n\n// Export composable for Vue components\nexport function useCodeTemplates() {\n return {\n templates: computed(() => codeTemplates.getTemplates()),\n categories: computed(() => codeTemplates.getCategories()),\n popularTemplates: computed(() => codeTemplates.getPopularTemplates()),\n recentTemplates: computed(() => codeTemplates.getRecentTemplates()),\n getTemplate: codeTemplates.getTemplate.bind(codeTemplates),\n getTemplatesByCategory: codeTemplates.getTemplatesByCategory.bind(codeTemplates),\n getTemplatesByLanguage: codeTemplates.getTemplatesByLanguage.bind(codeTemplates),\n searchTemplates: codeTemplates.searchTemplates.bind(codeTemplates),\n addTemplate: codeTemplates.addTemplate.bind(codeTemplates),\n updateTemplate: codeTemplates.updateTemplate.bind(codeTemplates),\n deleteTemplate: codeTemplates.deleteTemplate.bind(codeTemplates),\n incrementUsage: codeTemplates.incrementUsage.bind(codeTemplates),\n rateTemplate: codeTemplates.rateTemplate.bind(codeTemplates),\n exportTemplates: codeTemplates.exportTemplates.bind(codeTemplates),\n importTemplates: codeTemplates.importTemplates.bind(codeTemplates)\n }\n}" }, { "path": "web/src/store/index.ts", "content": "import type { App } from 'vue'\nimport { createPinia } from 'pinia'\n\nexport const store = createPinia()\n\nexport function setupStore(app: App) {\n app.use(store)\n}\n\nexport * from './modules'\n" }, { "path": "web/src/store/modules/app/helper.ts", "content": "import { ss } from '@/utils/storage'\n\nconst LOCAL_NAME = 'appSetting'\n\nexport type Theme = 'light' | 'dark' | 'auto'\n\nexport type Language = 'zh-CN' | 'zh-TW' | 'en-US'\n\nexport interface AppState {\n siderCollapsed: boolean\n theme: Theme\n language: Language\n}\n\nconst defaultLanguage = navigator.language as Language\n\nexport function defaultSetting(): AppState {\n return { siderCollapsed: false, theme: 'light', language: defaultLanguage }\n}\n\nexport function getLocalSetting(): AppState {\n const localSetting: AppState | undefined = ss.get(LOCAL_NAME)\n return { ...defaultSetting(), ...localSetting }\n}\n\nexport function setLocalSetting(setting: AppState): void {\n ss.set(LOCAL_NAME, setting)\n}\n" }, { "path": "web/src/store/modules/app/index.ts", "content": "import { defineStore } from 'pinia'\nimport type { AppState, Language, Theme } from './helper'\nimport { getLocalSetting, setLocalSetting } from './helper'\nimport { store } from '@/store'\n\nconst langs: Language[] = ['zh-CN', 'en-US', 'zh-CN']\n\nexport const useAppStore = defineStore('app-store', {\n state: (): AppState => getLocalSetting(),\n actions: {\n setSiderCollapsed(collapsed: boolean) {\n this.siderCollapsed = collapsed\n this.recordState()\n },\n\n setTheme(theme: Theme) {\n this.theme = theme\n this.recordState()\n },\n\n setLanguage(language: Language) {\n if (this.language !== language) {\n this.language = language\n this.recordState()\n }\n },\n setNextLanguage() {\n const currentLang = this.language\n const nextLang = langs[(langs.indexOf(currentLang) + 1) % langs.length]\n this.language = nextLang\n this.recordState()\n },\n\n recordState() {\n setLocalSetting(this.$state)\n },\n },\n})\n\nexport function useAppStoreWithOut() {\n return useAppStore(store)\n}\n" }, { "path": "web/src/store/modules/auth/helper.ts", "content": "// Hybrid token approach:\n// - Access tokens: In-memory (short-lived, secure from XSS)\n// - Refresh tokens: httpOnly cookies (long-lived, persistent)\n\nlet accessToken: string | null = null\n\nexport function getToken(): string | null {\n return accessToken\n}\n\nexport function setToken(token: string): void {\n accessToken = token\n}\n\nexport function removeToken(): void {\n accessToken = null\n}\n\n// Expiration tracking can still be useful for UI state\nconst EXPIRE_LOCAL_NAME = 'expiresIn'\n\nexport function getExpiresIn(): number | null {\n const stored = window.localStorage.getItem(EXPIRE_LOCAL_NAME)\n return stored ? parseInt(stored, 10) : null\n}\n\nexport function setExpiresIn(expiresIn: number): void {\n window.localStorage.setItem(EXPIRE_LOCAL_NAME, expiresIn.toString())\n}\n\nexport function removeExpiresIn(): void {\n window.localStorage.removeItem(EXPIRE_LOCAL_NAME)\n}\n" }, { "path": "web/src/store/modules/auth/index.ts", "content": "import { defineStore } from 'pinia'\nimport { watch } from 'vue'\nimport { getExpiresIn, getToken, removeExpiresIn, removeToken, setExpiresIn, setToken } from './helper'\n\nlet activeRefreshPromise: Promise | null = null\n\nexport interface AuthState {\n token: string | null // Access token stored in memory\n expiresIn: number | null\n isRefreshing: boolean // Track if token refresh is in progress\n isInitialized: boolean // Track if auth state has been initialized\n isInitializing: boolean // Track if auth initialization is in progress\n}\n\nexport const useAuthStore = defineStore('auth-store', {\n state: (): AuthState => ({\n token: getToken(), // Load token normally\n expiresIn: getExpiresIn(),\n isRefreshing: false,\n isInitialized: false,\n isInitializing: false,\n }),\n\n getters: {\n isValid(): boolean {\n return !!(this.token && this.expiresIn && this.expiresIn > Date.now() / 1000)\n },\n getToken(): string | null {\n return this.token\n },\n getExpiresIn(): number | null {\n return this.expiresIn\n },\n needsRefresh(): boolean {\n // Check if token expires within next 5 minutes\n const fiveMinutesFromNow = Date.now() / 1000 + 300\n return !!(this.expiresIn && this.expiresIn < fiveMinutesFromNow)\n },\n needPermission(): boolean {\n return this.isInitialized && !this.isInitializing && !this.isValid\n }\n },\n\n actions: {\n async initializeAuth() {\n if (this.isInitialized || this.isInitializing) return\n\n this.isInitializing = true\n\n try {\n const now = Date.now() / 1000\n if (this.expiresIn) {\n const tokenMissing = !this.token\n const expired = this.expiresIn <= now\n if (tokenMissing || expired || this.needsRefresh) {\n console.log('Token expired or about to expire, refreshing...')\n await this.refreshToken()\n }\n } else if (this.token) {\n // Clear expired token\n this.removeToken()\n this.removeExpiresIn()\n }\n } catch (error) {\n // Clear invalid state on error\n this.removeToken()\n this.removeExpiresIn()\n } finally {\n this.isInitializing = false\n this.isInitialized = true\n }\n },\n setToken(token: string) {\n this.token = token\n setToken(token)\n },\n removeToken() {\n this.token = null\n removeToken()\n },\n async refreshToken() {\n if (this.isRefreshing && activeRefreshPromise)\n return activeRefreshPromise\n\n console.log('Starting token refresh...')\n this.isRefreshing = true\n\n const refreshOperation = (async () => {\n try {\n // Call refresh endpoint - refresh token is sent automatically via httpOnly cookie\n const response = await fetch('/api/auth/refresh', {\n method: 'POST',\n credentials: 'include', // Include httpOnly cookies\n })\n\n console.log('Refresh response status:', response.status)\n\n if (response.ok) {\n const data = await response.json()\n console.log('Token refresh successful, setting new token')\n this.setToken(data.accessToken)\n this.setExpiresIn(data.expiresIn)\n return data.accessToken as string\n }\n\n // Refresh failed - user needs to login again\n console.log('Token refresh failed, removing tokens')\n this.removeToken()\n this.removeExpiresIn()\n throw new Error('Token refresh failed')\n } catch (error) {\n console.error('Token refresh error:', error)\n this.removeToken()\n this.removeExpiresIn()\n throw error\n } finally {\n this.isRefreshing = false\n activeRefreshPromise = null\n }\n })()\n\n activeRefreshPromise = refreshOperation\n return refreshOperation\n },\n setExpiresIn(expiresIn: number) {\n this.expiresIn = expiresIn\n setExpiresIn(expiresIn)\n },\n removeExpiresIn() {\n this.expiresIn = null\n removeExpiresIn()\n },\n async waitForInitialization(timeoutMs = 10000) {\n if (!this.isInitializing) {\n return\n }\n\n await new Promise((resolve) => {\n let stopWatcher: (() => void) | null = null\n const timeoutId = setTimeout(() => {\n if (stopWatcher) stopWatcher()\n resolve()\n }, timeoutMs)\n\n stopWatcher = watch(\n () => this.isInitializing,\n (isInit) => {\n if (!isInit) {\n clearTimeout(timeoutId)\n if (stopWatcher) stopWatcher()\n resolve()\n }\n },\n { immediate: false }\n )\n })\n },\n },\n})\n" }, { "path": "web/src/store/modules/index.ts", "content": "export * from './app'\nexport * from './user'\nexport * from './auth'\nexport * from './prompt'\nexport * from './workspace'\nexport * from './session'\nexport * from './message'\n" }, { "path": "web/src/store/modules/message/index.ts", "content": "import { defineStore } from 'pinia'\nimport {\n getChatMessagesBySessionUUID,\n clearSessionChatMessages,\n generateMoreSuggestions,\n} from '@/api'\nimport { deleteChatData } from '@/api'\nimport { createChatPrompt } from '@/api/chat_prompt'\nimport { getDefaultSystemPrompt } from '@/constants/chat'\nimport { nowISO } from '@/utils/date'\nimport { v7 as uuidv7 } from 'uuid'\nimport { useSessionStore } from '../session'\n\nexport interface MessageState {\n chat: Record // sessionUuid -> messages\n isLoading: Record // sessionUuid -> isLoading\n}\n\nexport const useMessageStore = defineStore('message-store', {\n state: (): MessageState => ({\n chat: {},\n isLoading: {},\n }),\n\n getters: {\n getChatSessionDataByUuid(state) {\n return (uuid?: string) => {\n if (uuid) {\n return state.chat[uuid] || []\n }\n return []\n }\n },\n\n getIsLoadingBySession(state) {\n return (sessionUuid: string) => {\n return state.isLoading[sessionUuid] || false\n }\n },\n\n // Get last message for a session\n getLastMessageForSession(state) {\n return (sessionUuid: string) => {\n const messages = state.chat[sessionUuid] || []\n return messages[messages.length - 1] || null\n }\n },\n\n // Get all messages for active session\n activeSessionMessages(state) {\n const sessionStore = useSessionStore()\n if (sessionStore.activeSessionUuid) {\n return state.chat[sessionStore.activeSessionUuid] || []\n }\n return []\n },\n },\n\n actions: {\n async syncChatMessages(sessionUuid: string) {\n if (!sessionUuid) {\n return\n }\n\n this.isLoading[sessionUuid] = true\n\n try {\n const messageData = await getChatMessagesBySessionUUID(sessionUuid)\n const normalizedMessages = Array.isArray(messageData) ? messageData : []\n\n // Initialize batching structure for messages with suggested questions\n const processedMessageData = normalizedMessages.map((message: Chat.Message) => {\n if (message.suggestedQuestions && message.suggestedQuestions.length > 0) {\n // If batches don't exist, create the first batch from existing questions\n if (!message.suggestedQuestionsBatches || message.suggestedQuestionsBatches.length === 0) {\n // Split suggestions into batches of 3 (assuming original suggestions come in groups of 3)\n const batches: string[][] = []\n for (let i = 0; i < message.suggestedQuestions.length; i += 3) {\n batches.push(message.suggestedQuestions.slice(i, i + 3))\n }\n \n return {\n ...message,\n suggestedQuestionsBatches: batches,\n currentSuggestedQuestionsBatch: batches.length - 1, // Show the last batch (most recent)\n suggestedQuestions: batches[batches.length - 1] || message.suggestedQuestions, // Show last batch\n }\n }\n }\n return message\n })\n\n const hasSystemPrompt = processedMessageData.some((message: Chat.Message) => message.isPrompt)\n if (!hasSystemPrompt) {\n try {\n const defaultPrompt = getDefaultSystemPrompt()\n const prompt = await createChatPrompt({\n uuid: uuidv7(),\n chatSessionUuid: sessionUuid,\n role: 'system',\n content: defaultPrompt,\n tokenCount: Math.max(1, Math.ceil(defaultPrompt.length / 4)),\n userId: 0,\n createdBy: 0,\n updatedBy: 0,\n })\n\n const promptMessage: Chat.Message = {\n uuid: prompt.uuid,\n dateTime: prompt.updatedAt || nowISO(),\n text: prompt.content || defaultPrompt,\n inversion: true,\n error: false,\n loading: false,\n isPrompt: true,\n }\n\n const existingPromptIndex = processedMessageData.findIndex(\n (message: Chat.Message) => message.uuid === promptMessage.uuid || message.isPrompt,\n )\n if (existingPromptIndex === -1) {\n processedMessageData.unshift(promptMessage)\n } else {\n processedMessageData[existingPromptIndex] = {\n ...processedMessageData[existingPromptIndex],\n ...promptMessage,\n isPrompt: true,\n }\n }\n } catch (error) {\n console.error(`Failed to create default system prompt for session ${sessionUuid}:`, error)\n }\n }\n\n this.chat[sessionUuid] = processedMessageData\n\n // Update active session if needed\n const sessionStore = useSessionStore()\n if (sessionStore.activeSessionUuid !== sessionUuid) {\n const session = sessionStore.getChatSessionByUuid(sessionUuid)\n if (session?.workspaceUuid) {\n await sessionStore.setActiveSession(session.workspaceUuid, sessionUuid)\n }\n }\n\n return processedMessageData\n } catch (error) {\n console.error(`Failed to sync messages for session ${sessionUuid}:`, error)\n throw error\n } finally {\n this.isLoading[sessionUuid] = false\n }\n },\n\n addMessage(sessionUuid: string, message: Chat.Message) {\n if (!this.chat[sessionUuid]) {\n this.chat[sessionUuid] = []\n }\n this.chat[sessionUuid].push(message)\n },\n\n addMessages(sessionUuid: string, messages: Chat.Message[]) {\n if (!this.chat[sessionUuid]) {\n this.chat[sessionUuid] = []\n }\n this.chat[sessionUuid].push(...messages)\n },\n\n updateMessage(sessionUuid: string, messageUuid: string, updates: Partial) {\n const messages = this.chat[sessionUuid]\n if (messages) {\n const index = messages.findIndex(msg => msg.uuid === messageUuid)\n if (index !== -1) {\n messages[index] = { ...messages[index], ...updates }\n }\n }\n },\n\n async removeMessage(sessionUuid: string, messageUuid: string) {\n try {\n const message = this.chat[sessionUuid]?.find(msg => msg.uuid === messageUuid)\n if (!message) {\n return\n }\n // Call the API to delete the message from the server\n await deleteChatData(message)\n // Remove the message from local state after successful API call\n if (this.chat[sessionUuid]) {\n this.chat[sessionUuid] = this.chat[sessionUuid].filter(\n msg => msg.uuid !== messageUuid\n )\n }\n } catch (error) {\n console.error(`Failed to delete message ${messageUuid}:`, error)\n throw error\n }\n },\n\n clearSessionMessages(sessionUuid: string) {\n try {\n clearSessionChatMessages(sessionUuid)\n // Keep the first message (system prompt) and clear the rest\n const messages = this.chat[sessionUuid] || []\n if (messages.length > 0) {\n this.chat[sessionUuid] = [messages[0]] // Keep only the first message\n } else {\n this.chat[sessionUuid] = []\n }\n } catch (error) {\n console.error(`Failed to clear messages for session ${sessionUuid}:`, error)\n throw error\n }\n },\n\n updateLastMessage(sessionUuid: string, updates: Partial) {\n const messages = this.chat[sessionUuid]\n if (messages && messages.length > 0) {\n const lastIndex = messages.length - 1\n messages[lastIndex] = { ...messages[lastIndex], ...updates }\n }\n },\n\n // Helper method to set loading state\n setLoading(sessionUuid: string, isLoading: boolean) {\n this.isLoading[sessionUuid] = isLoading\n },\n\n // Helper method to get message count for a session\n getMessageCount(sessionUuid: string) {\n return this.chat[sessionUuid]?.length || 0\n },\n\n // Helper method to clear all messages\n clearAllMessages() {\n this.chat = {}\n this.isLoading = {}\n },\n\n // Helper method to remove session data\n removeSessionData(sessionUuid: string) {\n delete this.chat[sessionUuid]\n delete this.isLoading[sessionUuid]\n },\n\n // Helper method to check if session has messages\n hasMessages(sessionUuid: string) {\n return this.chat[sessionUuid]?.length > 0\n },\n\n // Helper method to get messages by type\n getMessagesByType(sessionUuid: string, type: 'user' | 'assistant') {\n const messages = this.chat[sessionUuid] || []\n return messages.filter(msg => {\n if (type === 'user') {\n return msg.inversion\n }\n return !msg.inversion\n })\n },\n\n // Helper method to get pinned messages\n getPinnedMessages(sessionUuid: string) {\n const messages = this.chat[sessionUuid] || []\n return messages.filter(msg => msg.isPin)\n },\n\n // Helper method to get messages with artifacts\n getMessagesWithArtifacts(sessionUuid: string) {\n const messages = this.chat[sessionUuid] || []\n return messages.filter(msg => msg.artifacts && msg.artifacts.length > 0)\n },\n\n // Helper method to get messages by date range\n getMessagesByDateRange(sessionUuid: string, startDate: string, endDate: string) {\n const messages = this.chat[sessionUuid] || []\n return messages.filter(msg => {\n const messageDate = new Date(msg.dateTime)\n return messageDate >= new Date(startDate) && messageDate <= new Date(endDate)\n })\n },\n\n // Helper method to search messages\n searchMessages(sessionUuid: string, query: string) {\n const messages = this.chat[sessionUuid] || []\n const lowercaseQuery = query.toLowerCase()\n return messages.filter(msg =>\n msg.text.toLowerCase().includes(lowercaseQuery)\n )\n },\n\n // Helper method to get loading messages for a session\n getLoadingMessages(sessionUuid: string) {\n const messages = this.chat[sessionUuid] || []\n return messages.filter(msg => msg.loading)\n },\n\n // Helper method to get error messages for a session\n getErrorMessages(sessionUuid: string) {\n const messages = this.chat[sessionUuid] || []\n return messages.filter(msg => msg.error)\n },\n\n // Helper method to get prompt messages for a session\n getPromptMessages(sessionUuid: string) {\n const messages = this.chat[sessionUuid] || []\n return messages.filter(msg => msg.isPrompt)\n },\n\n // Generate more suggested questions for a message\n async generateMoreSuggestedQuestions(sessionUuid: string, messageUuid: string) {\n try {\n // Set generating state for the message\n this.updateMessage(sessionUuid, messageUuid, { suggestedQuestionsGenerating: true })\n\n const response = await generateMoreSuggestions(messageUuid)\n const { newSuggestions, allSuggestions } = response\n\n // Get existing message\n const messages = this.chat[sessionUuid] || []\n const messageIndex = messages.findIndex(msg => msg.uuid === messageUuid)\n \n if (messageIndex !== -1) {\n const message = messages[messageIndex]\n \n // Initialize batches if they don't exist\n let suggestedQuestionsBatches = message.suggestedQuestionsBatches || []\n \n // If this is the first time, create the first batch from existing questions\n if (suggestedQuestionsBatches.length === 0 && message.suggestedQuestions) {\n suggestedQuestionsBatches.push(message.suggestedQuestions)\n }\n \n // Add the new suggestions as a new batch\n suggestedQuestionsBatches.push(newSuggestions)\n \n // Update the message with new data - show the new batch, not all suggestions\n this.updateMessage(sessionUuid, messageUuid, {\n suggestedQuestions: newSuggestions, // Show only the new batch\n suggestedQuestionsBatches,\n currentSuggestedQuestionsBatch: suggestedQuestionsBatches.length - 1, // Set to the new batch\n suggestedQuestionsGenerating: false,\n })\n }\n\n return response\n } catch (error) {\n // Clear generating state on error\n this.updateMessage(sessionUuid, messageUuid, { suggestedQuestionsGenerating: false })\n console.error('Failed to generate more suggestions:', error)\n throw error\n }\n },\n\n // Navigate to previous suggestions batch\n previousSuggestedQuestionsBatch(sessionUuid: string, messageUuid: string) {\n const messages = this.chat[sessionUuid] || []\n const messageIndex = messages.findIndex(msg => msg.uuid === messageUuid)\n \n if (messageIndex !== -1) {\n const message = messages[messageIndex]\n const batches = message.suggestedQuestionsBatches || []\n const currentBatch = message.currentSuggestedQuestionsBatch || 0\n \n if (currentBatch > 0 && batches.length > 0) {\n const newBatchIndex = currentBatch - 1\n this.updateMessage(sessionUuid, messageUuid, {\n suggestedQuestions: batches[newBatchIndex],\n currentSuggestedQuestionsBatch: newBatchIndex,\n })\n }\n }\n },\n\n // Navigate to next suggestions batch\n nextSuggestedQuestionsBatch(sessionUuid: string, messageUuid: string) {\n const messages = this.chat[sessionUuid] || []\n const messageIndex = messages.findIndex(msg => msg.uuid === messageUuid)\n \n if (messageIndex !== -1) {\n const message = messages[messageIndex]\n const batches = message.suggestedQuestionsBatches || []\n const currentBatch = message.currentSuggestedQuestionsBatch || 0\n \n if (currentBatch < batches.length - 1) {\n const newBatchIndex = currentBatch + 1\n this.updateMessage(sessionUuid, messageUuid, {\n suggestedQuestions: batches[newBatchIndex],\n currentSuggestedQuestionsBatch: newBatchIndex,\n })\n }\n }\n },\n },\n})\n" }, { "path": "web/src/store/modules/prompt/helper.ts", "content": "import { ss } from '@/utils/storage'\n\nconst LOCAL_NAME = 'promptStore'\n\nexport type Prompt = {\n key: string\n value: string\n}\nexport type PromptList = [] | Prompt[]\n\nexport interface PromptStore {\n promptList: PromptList\n}\n\nconst defaultChinesePromptList: PromptList = [\n {\n \"key\": \"充当英语翻译和改进者\",\n \"value\": \"我希望你能担任英语翻译、拼写校对和修辞改进的角色。我会用任何语言和你交流，你会识别语言，将其翻译并用更为优美和精炼的英语回答我。请将我简单的词汇和句子替换成更为优美和高雅的表达方式，确保意思不变，但使其更具文学性。请仅回答更正和改进的部分，不要写解释。我的第一句话是“how are you ?”，请翻译它。\"\n },\n {\n \"key\": \"充当英翻中\",\n \"value\": \"下面我让你来充当翻译家，你的目标是把任何语言翻译成中文，请翻译时不要带翻译腔，而是要翻译得自然、流畅和地道，使用优美和高雅的表达方式。请翻译下面这句话：“how are you ?”\"\n },\n {\n \"key\": \"充当英文词典(附中文解释)\",\n \"value\": \"我想让你充当英文词典，对于给出的英文单词，你要给出其中文意思以及英文解释，并且给出一个例句，此外不要有其他反馈，第一个单词是“Hello\\\"\"\n },\n {\n \"key\": \"充当讲故事的人\",\n \"value\": \"我想让你扮演讲故事的角色。您将想出引人入胜、富有想象力和吸引观众的有趣故事。它可以是童话故事、教育故事或任何其他类型的故事，有可能吸引人们的注意力和想象力。根据目标受众，您可以为讲故事环节选择特定的主题或主题，例如，如果是儿童，则可以谈论动物；如果是成年人，那么基于历史的故事可能会更好地吸引他们等等。我的第一个要求是“我需要一个关于毅力的有趣故事。”\"\n },\n {\n \"key\": \"充当 AI 辅助医生\",\n \"value\": \"我想让你扮演一名人工智能辅助医生。我将为您提供患者的详细信息，您的任务是使用最新的人工智能工具，例如医学成像软件和其他机器学习程序，以诊断最可能导致其症状的原因。您还应该将体检、实验室测试等传统方法纳入您的评估过程，以确保准确性。我的第一个请求是“我需要帮助诊断一例严重的腹痛”。\"\n },\n {\n \"key\": \"充当医生\",\n \"value\": \"我想让你扮演医生的角色，想出创造性的治疗方法来治疗疾病。您应该能够推荐常规药物、草药和其他天然替代品。在提供建议时，您还需要考虑患者的年龄、生活方式和病史。我的第一个建议请求是“为患有关节炎的老年患者提出一个侧重于整体治疗方法的治疗计划”。\"\n },\n {\n \"key\": \"担任会计师\",\n \"value\": \"我希望你担任会计师，并想出创造性的方法来管理财务。在为客户制定财务计划时，您需要考虑预算、投资策略和风险管理。在某些情况下，您可能还需要提供有关税收法律法规的建议，以帮助他们实现利润最大化。我的第一个建议请求是“为小型企业制定一个专注于成本节约和长期投资的财务计划”。\"\n },\n {\n \"key\": \"担任厨师\",\n \"value\": \"我需要有人可以推荐美味的食谱，这些食谱包括营养有益但又简单又不费时的食物，因此适合像我们这样忙碌的人以及成本效益等其他因素，因此整体菜肴最终既健康又经济！我的第一个要求——“一些清淡而充实的东西，可以在午休时间快速煮熟”\"\n },\n {\n \"key\": \"充当自助书\",\n \"value\": \"我要你充当一本自助书。您会就如何改善我生活的某些方面（例如人际关系、职业发展或财务规划）向我提供建议和技巧。例如，如果我在与另一半的关系中挣扎，你可以建议有用的沟通技巧，让我们更亲近。我的第一个请求是“我需要帮助在困难时期保持积极性”。\"\n },\n {\n \"key\": \"充当格言书\",\n \"value\": \"我要你充当格言书。您将为我提供明智的建议、鼓舞人心的名言和意味深长的名言，以帮助指导我的日常决策。此外，如有必要，您可以提出将此建议付诸行动或其他相关主题的实用方法。我的第一个请求是“我需要关于如何在逆境中保持积极性的指导”。\"\n },\n {\n \"key\": \"充当心理学家\",\n \"value\": \"我想让你扮演一个心理学家。我会告诉你我的想法。我希望你能给我科学的建议，让我感觉更好。我的第一个想法，{ 在这里输入你的想法，如果你解释得更详细，我想你会得到更准确的答案。}\"\n },\n {\n \"key\": \"充当个人购物员\",\n \"value\": \"我想让你做我的私人采购员。我会告诉你我的预算和喜好，你会建议我购买的物品。您应该只回复您推荐的项目，而不是其他任何内容。不要写解释。我的第一个请求是“我有 100 美元的预算，我正在寻找一件新衣服。”\"\n },\n {\n \"key\": \"充当美食评论家\",\n \"value\": \"我想让你扮演美食评论家。我会告诉你一家餐馆，你会提供对食物和服务的评论。您应该只回复您的评论，而不是其他任何内容。不要写解释。我的第一个请求是“我昨晚去了一家新的意大利餐厅。你能提供评论吗？”\"\n },\n {\n \"key\": \"充当虚拟医生\",\n \"value\": \"我想让你扮演虚拟医生。我会描述我的症状，你会提供诊断和治疗方案。只回复你的诊疗方案，其他不回复。不要写解释。我的第一个请求是“最近几天我一直感到头痛和头晕”。\"\n },\n {\n \"key\": \"担任私人厨师\",\n \"value\": \"我要你做我的私人厨师。我会告诉你我的饮食偏好和过敏，你会建议我尝试的食谱。你应该只回复你推荐的食谱，别无其他。不要写解释。我的第一个请求是“我是一名素食主义者，我正在寻找健康的晚餐点子。”\"\n },\n {\n \"key\": \"担任法律顾问\",\n \"value\": \"我想让你做我的法律顾问。我将描述一种法律情况，您将就如何处理它提供建议。你应该只回复你的建议，而不是其他。不要写解释。我的第一个请求是“我出了车祸，不知道该怎么办”。\"\n },\n {\n \"key\": \"作为个人造型师\",\n \"value\": \"我想让你做我的私人造型师。我会告诉你我的时尚偏好和体型，你会建议我穿的衣服。你应该只回复你推荐的服装，别无其他。不要写解释。我的第一个请求是“我有一个正式的活动要举行，我需要帮助选择一套衣服。”\"\n },\n {\n \"key\": \"担任语言病理学家 (SLP)\",\n \"value\": \"我希望你扮演一名言语语言病理学家 (SLP)，想出新的言语模式、沟通策略，并培养对他们不口吃的沟通能力的信心。您应该能够推荐技术、策略和其他治疗方法。在提供建议时，您还需要考虑患者的年龄、生活方式和顾虑。我的第一个建议要求是“为一位患有口吃和自信地与他人交流有困难的年轻成年男性制定一个治疗计划”\"\n },\n {\n \"key\": \"担任创业技术律师\",\n \"value\": \"我将要求您准备一页纸的设计合作伙伴协议草案，该协议是一家拥有 IP 的技术初创公司与该初创公司技术的潜在客户之间的协议，该客户为该初创公司正在解决的问题空间提供数据和领域专业知识。您将写下大约 1 a4 页的拟议设计合作伙伴协议，涵盖 IP、机密性、商业权利、提供的数据、数据的使用等所有重要方面。\"\n },\n {\n \"key\": \"充当书面作品的标题生成器\",\n \"value\": \"我想让你充当书面作品的标题生成器。我会给你提供一篇文章的主题和关键词，你会生成五个吸引眼球的标题。请保持标题简洁，不超过 20 个字，并确保保持意思。回复将使用主题的语言类型。我的第一个主题是“LearnData，一个建立在 VuePress 上的知识库，里面整合了我所有的笔记和文章，方便我使用和分享。”\"\n },\n {\n \"key\": \"扮演醉汉\",\n \"value\": \"我要你扮演一个喝醉的人。您只会像一个喝醉了的人发短信一样回答，仅此而已。你的醉酒程度会在你的答案中故意和随机地犯很多语法和拼写错误。你也会随机地忽略我说的话，并随机说一些与我提到的相同程度的醉酒。不要在回复上写解释。我的第一句话是“你好吗？”\"\n },\n {\n \"key\": \"担任数学历史老师\",\n \"value\": \"我想让你充当数学历史老师，提供有关数学概念的历史发展和不同数学家的贡献的信息。你应该只提供信息而不是解决数学问题。使用以下格式回答：“{数学家/概念} - {他们的贡献/发展的简要总结}。我的第一个问题是“毕达哥拉斯对数学的贡献是什么？”\"\n },\n]\n\nconst defaultEnglishPromptList: PromptList = [\n {\n \"key\": \"Act as an English Translator and Improver\",\n \"value\": \"I want you to act as an English translator, spelling corrector and improver. I will speak to you in any language and you will detect the language, translate it and answer in the corrected and improved version of my text, in English. I want you to replace my simplified A0-level words and sentences with more beautiful and elegant, upper level English words and sentences. Keep the meaning same, but make them more literary. I want you to only reply the correction, the improvements and nothing else, do not write explanations. My first sentence is \\\"istanbulu cok seviyom burada olmak cok guzel\\\" and I want you to correct it and improve it. I want you to reply me with the corrected and improved version of my sentence, in English.\"\n },\n {\n \"key\": \"Act as a Spoken English Teacher and Improver\",\n \"value\": \"I want you to act as a spoken English teacher and improver. I will speak to you in English and you will reply to me in English to practice my spoken English. I want you to keep your reply neat, limiting the reply to 100 words. I want you to strictly correct my grammar mistakes, typos, and factual errors. I want you to ask me a question in your reply. Now let's start practicing, you could ask me a question first. Remember, I want you to strictly correct my grammar mistakes, typos, and factual errors.\"\n },\n {\n \"key\": \"Act as a Poet\",\n \"value\": \"I want you to act as a poet. You will create poems that evoke emotions and have the power to stir people's soul. Write on any topic or theme but make sure your words convey the feeling you are trying to express in beautiful yet meaningful ways. You can also come up with short verses that are still powerful enough to leave an imprint in readers' minds.\"\n },\n {\n \"key\": \"Act as a Note Taking Assistant\",\n \"value\": \"I want you to act as a note-taking assistant for a lecture. Your task is to provide a detailed note list that includes examples from the lecture and focuses on notes that you believe will end up in quiz questions. Additionally, please make a separate list for notes that have numbers and data in them and another separated list for the examples that included in this lecture. The notes should be concise and easy to read.\"\n },\n {\n \"key\": \"Act as a Career Counselor\",\n \"value\": \"I want you to act as a career counselor. I will provide you with information about my skills, interests, and career goals, and you will suggest career paths that suit me. You should also provide information about required qualifications, potential career progression, and salary ranges. My first request is 'I have a degree in computer science and I enjoy problem-solving and working with data.'\"\n },\n {\n \"key\": \"Act as a Financial Advisor\",\n \"value\": \"I want you to act as a financial advisor. I will provide you with information about my financial situation and goals, and you will suggest ways to manage my money, invest, and save for the future. You should consider factors like risk tolerance, time horizon, and financial obligations. My first request is 'I'm 30 years old, earn $80,000 per year, and want to retire by age 60 with $2 million.'\"\n },\n {\n \"key\": \"Act as a Travel Guide\",\n \"value\": \"I want you to act as a travel guide. I will tell you about a destination I want to visit, and you will suggest places to visit, things to do, local customs to be aware of, and practical travel tips. You should also recommend accommodations and transportation options. My first request is 'I'm planning a trip to Japan for two weeks in spring.'\"\n },\n {\n \"key\": \"Act as a Software Developer\",\n \"value\": \"I want you to act as a software developer. I will provide you with requirements for an application, and you will suggest technologies to use, architecture patterns, and implementation details. You should also provide code snippets when appropriate. My first request is 'I need to build a web application for managing tasks with user authentication.'\"\n },\n {\n \"key\": \"Act as a Personal Trainer\",\n \"value\": \"I want you to act as a personal trainer. I will provide you with information about my fitness level, goals, and available equipment, and you will create a workout plan for me. You should also provide nutrition advice and tips for staying motivated. My first request is 'I'm a beginner who wants to lose weight and tone up. I have access to a gym and basic equipment at home.'\"\n },\n {\n \"key\": \"Act as a Language Learning Partner\",\n \"value\": \"I want you to act as a language learning partner. I will tell you which language I want to learn, and you will engage me in conversation practice, correct my mistakes, and explain grammar concepts. You should also suggest learning resources and techniques. My first request is 'I want to learn Spanish and I'm a complete beginner.'\"\n },\n {\n \"key\": \"Act as a Debate Coach\",\n \"value\": \"I want you to act as a debate coach. I will provide you with a topic, and you will help me develop arguments, counterarguments, and rebuttals. You should also provide tips on delivery and persuasion techniques. My first request is 'The topic is whether social media has had a positive impact on society.'\"\n },\n {\n \"key\": \"Act as a UX/UI Designer\",\n \"value\": \"I want you to act as a UX/UI designer. I will describe a product or feature, and you will suggest user flows, wireframes, and design principles to follow. You should consider usability, accessibility, and visual appeal. My first request is 'I need to design a mobile app for tracking daily water intake.'\"\n }\n]\n\nlet defaultPromptMap: { [key: string]: Prompt[] } = {\n 'zh-CN': defaultChinesePromptList,\n 'zh-TW': defaultChinesePromptList,\n 'en-US': defaultEnglishPromptList,\n}\n\n\nexport function getLocalPromptList(): PromptStore {\n const promptStore: PromptStore | undefined = ss.get(LOCAL_NAME)\n if (promptStore && promptStore?.promptList?.length > 0) {\n return promptStore\n } else {\n let defaultPromptList = defaultPromptMap[navigator.language];\n setLocalPromptList({ promptList: defaultPromptList })\n return { promptList: defaultPromptList }\n }\n}\n\nexport function setLocalPromptList(promptStore: PromptStore): void {\n ss.set(LOCAL_NAME, promptStore)\n}\n" }, { "path": "web/src/store/modules/prompt/index.ts", "content": "import { defineStore } from 'pinia'\nimport type { PromptStore } from './helper'\nimport { getLocalPromptList, setLocalPromptList } from './helper'\n\nexport const usePromptStore = defineStore('prompt-store', {\n state: (): PromptStore => getLocalPromptList(),\n\n actions: {\n updatePromptList(promptList: []) {\n this.$patch({ promptList })\n setLocalPromptList({ promptList })\n },\n getPromptList() {\n return this.$state\n },\n },\n})" }, { "path": "web/src/store/modules/session/index.ts", "content": "import { defineStore } from 'pinia'\nimport { router } from '@/router'\nimport {\n createChatSession,\n deleteChatSession,\n renameChatSession,\n updateChatSession,\n getSessionsByWorkspace,\n createSessionInWorkspace,\n} from '@/api'\nimport { getDefaultSystemPrompt } from '@/constants/chat'\nimport { useWorkspaceStore } from '../workspace'\n\nexport interface SessionState {\n workspaceHistory: Record // workspaceUuid -> sessions\n activeSessionUuid: string | null\n isLoading: boolean\n isCreatingSession: boolean\n isSwitchingSession: boolean\n isNavigating: boolean\n lastRequestedSessionUuid: string | null // Track the most recent session switch request\n}\n\nexport const useSessionStore = defineStore('session-store', {\n state: (): SessionState => ({\n workspaceHistory: {},\n activeSessionUuid: null,\n isLoading: false,\n isCreatingSession: false,\n isSwitchingSession: false,\n isNavigating: false,\n lastRequestedSessionUuid: null,\n }),\n\n getters: {\n getChatSessionByUuid(state) {\n return (uuid?: string) => {\n if (uuid) {\n // Search across all workspace histories\n for (const sessions of Object.values(state.workspaceHistory)) {\n const session = sessions.find(item => item.uuid === uuid)\n if (session) return session\n }\n }\n return null\n }\n },\n\n getSessionsByWorkspace(state) {\n return (workspaceUuid?: string) => {\n if (!workspaceUuid) return []\n return state.workspaceHistory[workspaceUuid] || []\n }\n },\n\n activeSession(state) {\n if (state.activeSessionUuid) {\n // Search across all workspace histories\n for (const sessions of Object.values(state.workspaceHistory)) {\n const session = sessions.find(item => item.uuid === state.activeSessionUuid)\n if (session) return session\n }\n }\n return null\n },\n\n // Get session URL for navigation\n getSessionUrl() {\n return (sessionUuid: string): string => {\n // Search across all workspace histories\n for (const sessions of Object.values(this.workspaceHistory)) {\n const session = sessions.find(item => item.uuid === sessionUuid)\n if (session && session.workspaceUuid) {\n return `/#/workspace/${session.workspaceUuid}/chat/${sessionUuid}`\n }\n }\n return `/#/chat/${sessionUuid}`\n }\n },\n },\n\n actions: {\n async syncWorkspaceSessions(workspaceUuid: string) {\n try {\n this.isLoading = true\n const sessions = await getSessionsByWorkspace(workspaceUuid)\n\n // Map topic to title for frontend compatibility\n const sessionsWithTitle = sessions.map((session: any) => ({\n ...session,\n title: session.topic || session.title || 'Untitled'\n }))\n\n this.workspaceHistory[workspaceUuid] = sessionsWithTitle\n return sessionsWithTitle\n } catch (error) {\n console.error(`Failed to sync sessions for workspace ${workspaceUuid}:`, error)\n throw error\n } finally {\n this.isLoading = false\n }\n },\n\n // Optimized method to load only active workspace sessions\n async syncActiveWorkspaceSessions() {\n try {\n this.isLoading = true\n const workspaceStore = useWorkspaceStore()\n\n if (!workspaceStore.activeWorkspaceUuid) {\n console.log('No active workspace, skipping session sync')\n return\n }\n\n console.log('Loading sessions for active workspace:', workspaceStore.activeWorkspaceUuid)\n const sessions = await getSessionsByWorkspace(workspaceStore.activeWorkspaceUuid)\n\n // Map topic to title for frontend compatibility\n const sessionsWithTitle = sessions.map((session: any) => ({\n ...session,\n title: session.topic || session.title || 'Untitled'\n }))\n\n this.workspaceHistory[workspaceStore.activeWorkspaceUuid] = sessionsWithTitle\n console.log(`✅ Loaded ${sessionsWithTitle.length} sessions for active workspace`)\n\n return sessionsWithTitle\n } catch (error) {\n console.error('Failed to sync active workspace sessions:', error)\n throw error\n } finally {\n this.isLoading = false\n }\n },\n\n async syncAllWorkspaceSessions() {\n try {\n this.isLoading = true\n const workspaceStore = useWorkspaceStore()\n\n // Sync sessions for all workspaces\n for (const workspace of workspaceStore.workspaces) {\n const sessions = await getSessionsByWorkspace(workspace.uuid)\n\n // Map topic to title for frontend compatibility\n const sessionsWithTitle = sessions.map((session: any) => ({\n ...session,\n title: session.topic || session.title || 'Untitled'\n }))\n\n this.workspaceHistory[workspace.uuid] = sessionsWithTitle\n }\n } catch (error) {\n console.error('Failed to sync all workspace sessions:', error)\n throw error\n } finally {\n this.isLoading = false\n }\n },\n\n async createSessionInWorkspace(title: string, workspaceUuid?: string, model?: string) {\n if (this.isCreatingSession) {\n return null\n }\n\n this.isCreatingSession = true\n\n try {\n const workspaceStore = useWorkspaceStore()\n const targetWorkspaceUuid = workspaceUuid || workspaceStore.activeWorkspaceUuid\n\n if (!targetWorkspaceUuid) {\n throw new Error('No workspace available for session creation')\n }\n\n // Get default model if none provided\n let sessionModel = model\n if (!sessionModel) {\n try {\n const { fetchDefaultChatModel } = await import('@/api/chat_model')\n const defaultModel = await fetchDefaultChatModel()\n sessionModel = defaultModel.name\n } catch (error) {\n console.warn('Failed to fetch default model, proceeding without model:', error)\n }\n }\n\n const newSession = await createSessionInWorkspace(targetWorkspaceUuid, {\n topic: title,\n model: sessionModel,\n defaultSystemPrompt: getDefaultSystemPrompt(),\n })\n\n // Map topic to title for frontend compatibility\n const sessionWithTitle = {\n ...newSession,\n title: newSession.topic || title,\n model: newSession.model || sessionModel\n }\n\n // Add to workspace history\n if (!this.workspaceHistory[targetWorkspaceUuid]) {\n this.workspaceHistory[targetWorkspaceUuid] = []\n }\n this.workspaceHistory[targetWorkspaceUuid].unshift(sessionWithTitle)\n\n // Set as active session\n await this.setActiveSession(targetWorkspaceUuid, sessionWithTitle.uuid)\n\n return sessionWithTitle\n } catch (error) {\n console.error('Failed to create session in workspace:', error)\n throw error\n } finally {\n this.isCreatingSession = false\n }\n },\n\n async createLegacySession(session: Chat.Session) {\n try {\n await createChatSession(session.uuid, session.title, session.model, getDefaultSystemPrompt())\n\n // Refresh workspace sessions to get updated list from backend\n const workspaceUuid = session.workspaceUuid\n if (workspaceUuid) {\n await this.syncWorkspaceSessions(workspaceUuid)\n }\n\n await this.setActiveSession(workspaceUuid || null, session.uuid)\n return session\n } catch (error) {\n console.error('Failed to create legacy session:', error)\n throw error\n }\n },\n\n async updateSession(uuid: string, updates: Partial) {\n try {\n console.log('updateSession called with uuid:', uuid, 'updates:', updates)\n console.log('Current workspaceHistory:', this.workspaceHistory)\n\n // Find session across all workspace histories\n for (const workspaceUuid in this.workspaceHistory) {\n const sessions = this.workspaceHistory[workspaceUuid]\n const index = sessions.findIndex(item => item.uuid === uuid)\n if (index !== -1) {\n console.log('Found session in workspace:', workspaceUuid, 'at index:', index)\n // Update local state\n sessions[index] = { ...sessions[index], ...updates }\n\n // Update backend - use the appropriate API method\n if (updates.title !== undefined) {\n // If only title is changing, use the rename endpoint\n await renameChatSession(uuid, sessions[index].title)\n } else {\n // For other updates (like model), use the full update endpoint\n await updateChatSession(uuid, sessions[index])\n }\n\n return sessions[index]\n }\n }\n\n // If session not found locally, try to update it on the backend anyway\n // This handles cases where the session exists on the server but not in local state\n console.log('Session not found locally, attempting backend update')\n try {\n const session = this.getChatSessionByUuid(uuid)\n if (session) {\n console.log('Found session via getter, updating')\n const updatedSession = { ...session, ...updates }\n await updateChatSession(uuid, updatedSession)\n return updatedSession\n }\n } catch (backendError) {\n console.error('Backend update also failed:', backendError)\n }\n\n throw new Error(`Session ${uuid} not found`)\n } catch (error) {\n console.error('Failed to update session:', error)\n throw error\n }\n },\n\n async deleteSession(sessionUuid: string) {\n try {\n // Find session and its workspace\n let workspaceUuid: string | null = null\n for (const [wUuid, sessions] of Object.entries(this.workspaceHistory)) {\n const index = sessions.findIndex(item => item.uuid === sessionUuid)\n if (index !== -1) {\n workspaceUuid = wUuid\n break\n }\n }\n\n if (workspaceUuid) {\n // Remove from workspace history\n this.workspaceHistory[workspaceUuid] = this.workspaceHistory[workspaceUuid].filter(\n session => session.uuid !== sessionUuid\n )\n }\n\n // Delete from backend\n await deleteChatSession(sessionUuid)\n\n // Clear active session if it was deleted\n if (this.activeSessionUuid === sessionUuid) {\n await this.setNextActiveSession(workspaceUuid)\n }\n\n // Clear from workspace active sessions\n if (workspaceUuid) {\n const workspaceStore = useWorkspaceStore()\n workspaceStore.clearActiveSessionForWorkspace(workspaceUuid)\n }\n } catch (error) {\n console.error('Failed to delete session:', error)\n throw error\n }\n },\n\n async setActiveSession(workspaceUuid: string | null, sessionUuid: string) {\n // Early return if this is already the active session\n if (this.activeSessionUuid === sessionUuid) {\n return\n }\n\n // Track this as the most recent requested session\n this.lastRequestedSessionUuid = sessionUuid\n\n // If already switching, wait a bit and check if this request is still the latest\n if (this.isSwitchingSession) {\n console.log('Session switch in progress, deferring request for:', sessionUuid)\n // Wait for current switch to complete\n await new Promise(resolve => setTimeout(resolve, 100))\n // Check if a newer request came in while we were waiting\n if (this.lastRequestedSessionUuid !== sessionUuid) {\n console.log('Ignoring stale session switch request for:', sessionUuid)\n return\n }\n }\n\n this.isSwitchingSession = true\n\n try {\n // Double-check this is still the latest requested session\n if (this.lastRequestedSessionUuid !== sessionUuid) {\n console.log('Aborting session switch, newer request exists:', this.lastRequestedSessionUuid)\n return\n }\n\n this.activeSessionUuid = sessionUuid\n\n // Update workspace active session tracking\n if (workspaceUuid) {\n const workspaceStore = useWorkspaceStore()\n workspaceStore.setActiveSessionForWorkspace(workspaceUuid, sessionUuid)\n }\n\n // Navigate to the session\n await this.navigateToSession(sessionUuid)\n } catch (error) {\n console.error('Failed to set active session:', error)\n throw error\n } finally {\n this.isSwitchingSession = false\n }\n },\n\n // Set active session without navigation (for workspace switching)\n setActiveSessionWithoutNavigation(workspaceUuid: string | null, sessionUuid: string) {\n this.activeSessionUuid = sessionUuid\n\n // Update workspace active session tracking\n if (workspaceUuid) {\n const workspaceStore = useWorkspaceStore()\n workspaceStore.setActiveSessionForWorkspace(workspaceUuid, sessionUuid)\n }\n },\n\n async setNextActiveSession(workspaceUuid: string | null) {\n if (workspaceUuid && this.workspaceHistory[workspaceUuid]?.length > 0) {\n // Set first available session in the same workspace\n const nextSession = this.workspaceHistory[workspaceUuid][0]\n await this.setActiveSession(workspaceUuid, nextSession.uuid)\n } else {\n // Find any available session\n for (const [wUuid, sessions] of Object.entries(this.workspaceHistory)) {\n if (sessions.length > 0) {\n await this.setActiveSession(wUuid, sessions[0].uuid)\n return\n }\n }\n // No sessions available\n this.activeSessionUuid = null\n }\n },\n\n async navigateToSession(sessionUuid: string) {\n // Prevent overlapping navigation calls\n if (this.isNavigating) {\n console.log('Navigation already in progress, skipping')\n return\n }\n\n this.isNavigating = true\n \n try {\n const session = this.getChatSessionByUuid(sessionUuid)\n if (session && session.workspaceUuid) {\n // Check if we're already on the correct route to avoid unnecessary navigation\n const currentRoute = router.currentRoute.value\n const currentWorkspaceUuid = currentRoute.params.workspaceUuid as string\n const currentSessionUuid = currentRoute.params.uuid as string\n \n if (currentWorkspaceUuid === session.workspaceUuid && currentSessionUuid === sessionUuid) {\n console.log('Already on correct route, skipping navigation')\n return\n }\n\n const workspaceStore = useWorkspaceStore()\n await workspaceStore.navigateToWorkspace(session.workspaceUuid, sessionUuid)\n } else {\n // If session doesn't have a workspace, try to assign it to the default workspace\n const workspaceStore = useWorkspaceStore()\n const defaultWorkspace = workspaceStore.workspaces.find(w => w.isDefault) || workspaceStore.workspaces[0]\n\n if (defaultWorkspace) {\n console.log('Session without workspace, navigating to default workspace:', defaultWorkspace.uuid)\n await workspaceStore.navigateToWorkspace(defaultWorkspace.uuid, sessionUuid)\n } else {\n // Last resort: navigate to default route\n console.log('No workspace available, navigating to default route')\n await router.push({ name: 'DefaultWorkspace' })\n }\n }\n } finally {\n this.isNavigating = false\n }\n },\n\n // Helper method to clear all sessions for a workspace\n clearWorkspaceSessions(workspaceUuid: string) {\n this.workspaceHistory[workspaceUuid] = []\n\n // Clear active session if it was in this workspace\n const activeSession = this.activeSession\n if (activeSession && activeSession.workspaceUuid === workspaceUuid) {\n this.activeSessionUuid = null\n }\n },\n\n // Helper method to get all sessions across all workspaces\n getAllSessions() {\n const allSessions: Chat.Session[] = []\n for (const sessions of Object.values(this.workspaceHistory)) {\n allSessions.push(...sessions)\n }\n return allSessions\n },\n\n // Legacy compatibility method - maps to createSessionInWorkspace\n async addSession(session: Chat.Session) {\n return await this.createSessionInWorkspace(session.title, session.workspaceUuid, session.model)\n },\n\n // Centralized session creation method for consistent behavior\n async createNewSession(title?: string, workspaceUuid?: string, model?: string) {\n const workspaceStore = useWorkspaceStore()\n const targetWorkspaceUuid = workspaceUuid || workspaceStore.activeWorkspaceUuid\n\n if (!targetWorkspaceUuid) {\n throw new Error('No workspace available for session creation')\n }\n\n const sessionTitle = title || 'New Chat'\n return await this.createSessionInWorkspace(sessionTitle, targetWorkspaceUuid, model)\n },\n },\n})\n" }, { "path": "web/src/store/modules/user/helper.ts", "content": "import { ss } from '@/utils/storage'\n\nconst LOCAL_NAME = 'userStorage'\n\nexport interface UserInfo {\n name: string\n description: string\n}\n\nexport interface UserState {\n userInfo: UserInfo\n}\n\nexport function defaultSetting(): UserState {\n return {\n userInfo: {\n name: '',\n description: '',\n },\n }\n}\n\nexport function getLocalState(): UserState {\n const localSetting: UserState | undefined = ss.get(LOCAL_NAME)\n return { ...defaultSetting(), ...localSetting }\n}\n\nexport function setLocalState(setting: UserState): void {\n ss.set(LOCAL_NAME, setting)\n}\n" }, { "path": "web/src/store/modules/user/index.ts", "content": "import { defineStore } from 'pinia'\nimport type { UserInfo, UserState } from './helper'\nimport { defaultSetting, getLocalState, setLocalState } from './helper'\n\nexport const useUserStore = defineStore('user-store', {\n state: (): UserState => getLocalState(),\n actions: {\n updateUserInfo(userInfo: Partial) {\n this.userInfo = { ...this.userInfo, ...userInfo }\n this.recordState()\n },\n\n resetUserInfo() {\n this.userInfo = { ...defaultSetting().userInfo }\n this.recordState()\n },\n\n recordState() {\n setLocalState(this.$state)\n },\n },\n})\n" }, { "path": "web/src/store/modules/workspace/index.ts", "content": "import { defineStore } from 'pinia'\nimport { router } from '@/router'\nimport {\n getWorkspaces,\n createWorkspace,\n updateWorkspace,\n deleteWorkspace,\n ensureDefaultWorkspace,\n setDefaultWorkspace,\n updateWorkspaceOrder as updateWorkspaceOrderApi,\n autoMigrateLegacySessions,\n getAllWorkspaceActiveSessions,\n getChatSessionDefault,\n getWorkspace,\n} from '@/api'\n\nimport { useSessionStore } from '@/store/modules/session'\nimport { t } from '@/locales'\n\nexport interface WorkspaceState {\n workspaces: Chat.Workspace[]\n activeWorkspaceUuid: string | null\n workspaceActiveSessions: Record // workspaceUuid -> sessionUuid\n pendingSessionRestore: { workspaceUuid: string; sessionUuid: string } | null\n isLoading: boolean\n}\n\nexport const useWorkspaceStore = defineStore('workspace-store', {\n state: (): WorkspaceState => ({\n workspaces: [],\n activeWorkspaceUuid: null,\n workspaceActiveSessions: {},\n pendingSessionRestore: null,\n isLoading: false,\n }),\n\n getters: {\n getWorkspaceByUuid(state) {\n return (uuid?: string) => {\n if (uuid) {\n return state.workspaces.find(workspace => workspace.uuid === uuid)\n }\n return null\n }\n },\n\n getDefaultWorkspace(state) {\n return state.workspaces.find(workspace => workspace.isDefault) || null\n },\n\n activeWorkspace(state) {\n if (state.activeWorkspaceUuid) {\n return state.workspaces.find(workspace => workspace.uuid === state.activeWorkspaceUuid)\n }\n return null\n },\n\n // Get active session for a specific workspace\n getActiveSessionForWorkspace(state) {\n return (workspaceUuid: string) => {\n return state.workspaceActiveSessions[workspaceUuid] || null\n }\n },\n },\n\n actions: {\n // Optimized initialization that only loads active workspace\n async initializeActiveWorkspace(targetWorkspaceUuid?: string) {\n try {\n console.log('🔄 Starting optimized workspace initialization...')\n\n // Step 1: Handle legacy session migration (if needed)\n try {\n const migrationResult = await autoMigrateLegacySessions()\n if (migrationResult.hasLegacySessions && migrationResult.migratedSessions > 0) {\n console.log(`🔄 Auto-migrated ${migrationResult.migratedSessions} legacy sessions to default workspace`)\n }\n } catch (migrationError) {\n console.warn('⚠️ Legacy session migration failed:', migrationError)\n }\n\n // Step 2: Load only the active/target workspace\n const activeWorkspace = await this.loadActiveWorkspace(targetWorkspaceUuid)\n\n // Step 3: Sync workspace active sessions from backend (for session persistence)\n const routeCurrent = router.currentRoute.value\n const urlWorkspaceUuid = routeCurrent.params.workspaceUuid as string || targetWorkspaceUuid\n const urlSessionUuid = routeCurrent.params.uuid as string\n await this.syncWorkspaceActiveSessions(urlWorkspaceUuid, urlSessionUuid)\n\n // Step 4: Load sessions only for the active workspace\n const sessionStore = useSessionStore()\n await sessionStore.syncActiveWorkspaceSessions()\n\n // Step 5: Ensure user has at least one session in the active workspace\n await this.ensureUserHasSession()\n\n console.log('✅ Optimized workspace initialization completed')\n return activeWorkspace\n } catch (error) {\n console.error('❌ Error in optimized workspace initialization:', error)\n throw error\n }\n },\n\n // Comprehensive initialization method that replaces the old chat store logic\n async initializeApplication() {\n try {\n console.log('🔄 Starting comprehensive application initialization...')\n\n // Step 1: Handle legacy session migration\n try {\n const migrationResult = await autoMigrateLegacySessions()\n if (migrationResult.hasLegacySessions && migrationResult.migratedSessions > 0) {\n console.log(`🔄 Auto-migrated ${migrationResult.migratedSessions} legacy sessions to default workspace`)\n\n // Only force refresh if we're not already on a workspace route\n const currentRoute = router.currentRoute.value\n if (currentRoute.name !== 'WorkspaceChat') {\n console.log('🔄 Refreshing page after migration')\n window.location.reload()\n return // Exit early since we're refreshing\n } else {\n console.log('🔄 Skipping refresh - already on workspace route')\n }\n }\n } catch (migrationError) {\n console.warn('⚠️ Legacy session migration failed:', migrationError)\n // Continue with normal sync - don't block the app\n }\n\n // Step 2: Sync workspaces\n await this.syncWorkspaces()\n\n // Step 3: Determine workspace context from URL\n const routeBeforeSync = router.currentRoute.value\n const urlWorkspaceUuid = routeBeforeSync.name === 'WorkspaceChat' ? routeBeforeSync.params.workspaceUuid as string : null\n const urlSessionUuid = routeBeforeSync.params.uuid as string\n const isOnDefaultRoute = routeBeforeSync.name === 'DefaultWorkspace'\n\n // Step 4: Sync workspace active sessions from backend\n await this.syncWorkspaceActiveSessions(urlWorkspaceUuid || undefined, urlSessionUuid || undefined)\n\n // Step 5: Ensure we have an active workspace\n await this.ensureActiveWorkspace()\n\n // Step 6: Initialize sessions through session store\n const sessionStore = useSessionStore()\n await sessionStore.syncAllWorkspaceSessions()\n\n // Step 7: Handle session creation if needed\n await this.ensureUserHasSession()\n\n // Step 8: Handle navigation\n await this.handleInitialNavigation(urlWorkspaceUuid || undefined, urlSessionUuid || undefined, isOnDefaultRoute)\n\n console.log('✅ Application initialization completed successfully')\n } catch (error) {\n console.error('❌ Error in initializeApplication:', error)\n throw error\n }\n },\n\n // Optimized method to load only the active/default workspace\n async loadActiveWorkspace(targetWorkspaceUuid?: string) {\n try {\n this.isLoading = true\n\n // If specific workspace is requested, try to load it\n if (targetWorkspaceUuid) {\n try {\n const workspace = await getWorkspace(targetWorkspaceUuid)\n this.workspaces = [workspace]\n this.activeWorkspaceUuid = workspace.uuid\n return workspace\n } catch (error) {\n console.warn(`Failed to load specific workspace ${targetWorkspaceUuid}, falling back to default`, error)\n }\n }\n\n // Check if we already have a default workspace loaded\n const existingDefault = this.workspaces.find(w => w.isDefault)\n if (existingDefault) {\n this.activeWorkspaceUuid = existingDefault.uuid\n console.log('✅ Using existing default workspace:', existingDefault.name)\n return existingDefault\n }\n\n // Ensure we have a default workspace (this only loads/creates the default one)\n const defaultWorkspace = await ensureDefaultWorkspace()\n this.workspaces = [defaultWorkspace]\n this.activeWorkspaceUuid = defaultWorkspace.uuid\n\n console.log('✅ Loaded default workspace:', defaultWorkspace.name)\n return defaultWorkspace\n } catch (error) {\n console.error('Failed to load active workspace:', error)\n throw error\n } finally {\n this.isLoading = false\n }\n },\n\n // Load additional workspaces on demand (for workspace selector)\n async loadAllWorkspaces() {\n try {\n const allWorkspaces = await getWorkspaces()\n // Replace workspaces array with all workspaces (this is for workspace selector)\n // Keep the active workspace UUID as is\n this.workspaces = allWorkspaces\n // Ensure activeWorkspaceUuid is still valid\n if (this.activeWorkspaceUuid && !allWorkspaces.find(w => w.uuid === this.activeWorkspaceUuid)) {\n const defaultWs = allWorkspaces.find(w => w.isDefault) || allWorkspaces[0]\n if (defaultWs) {\n this.activeWorkspaceUuid = defaultWs.uuid\n }\n }\n return allWorkspaces\n } catch (error) {\n console.error('Failed to load all workspaces:', error)\n throw error\n }\n },\n\n async syncWorkspaceActiveSessions(urlWorkspaceUuid?: string, urlSessionUuid?: string) {\n try {\n const backendSessions = await getAllWorkspaceActiveSessions()\n\n // Build workspace active sessions mapping\n this.workspaceActiveSessions = {}\n let globalActiveSession = null\n\n for (const session of backendSessions) {\n if (session.workspaceUuid) {\n this.workspaceActiveSessions[session.workspaceUuid] = session.chatSessionUuid\n if (!globalActiveSession) {\n globalActiveSession = session\n }\n }\n }\n\n // Prioritize URL context over backend data\n if (urlWorkspaceUuid && urlSessionUuid) {\n this.activeWorkspaceUuid = urlWorkspaceUuid\n console.log('✅ Used workspace from URL:', { workspaceUuid: urlWorkspaceUuid, sessionUuid: urlSessionUuid })\n\n // Set active session in session store\n const sessionStore = useSessionStore()\n sessionStore.setActiveSessionWithoutNavigation(urlWorkspaceUuid, urlSessionUuid)\n } else if (urlWorkspaceUuid) {\n this.activeWorkspaceUuid = urlWorkspaceUuid\n console.log('✅ Used workspace from URL (no session):', urlWorkspaceUuid)\n\n // Restore active session for this workspace if available\n const activeSessionForWorkspace = this.workspaceActiveSessions[urlWorkspaceUuid]\n if (activeSessionForWorkspace) {\n const sessionStore = useSessionStore()\n sessionStore.setActiveSessionWithoutNavigation(urlWorkspaceUuid, activeSessionForWorkspace)\n console.log('✅ Restored active session for workspace:', activeSessionForWorkspace)\n }\n } else if (globalActiveSession?.workspaceUuid) {\n this.activeWorkspaceUuid = globalActiveSession.workspaceUuid\n console.log('✅ Used workspace from backend:', globalActiveSession.workspaceUuid)\n\n // Set active session from backend\n const sessionStore = useSessionStore()\n sessionStore.setActiveSessionWithoutNavigation(globalActiveSession.workspaceUuid, globalActiveSession.chatSessionUuid)\n console.log('✅ Restored active session from backend:', globalActiveSession.chatSessionUuid)\n }\n } catch (error) {\n console.warn('⚠️ Failed to sync workspace active sessions:', error)\n }\n },\n\n async ensureActiveWorkspace() {\n // If we don't have an active workspace, set to default\n if (!this.activeWorkspaceUuid && this.workspaces.length > 0) {\n const defaultWorkspace = this.workspaces.find(workspace => workspace.isDefault) || this.workspaces[0]\n if (defaultWorkspace) {\n this.activeWorkspaceUuid = defaultWorkspace.uuid\n console.log('✅ Set active workspace to default:', defaultWorkspace.name)\n }\n }\n },\n\n async ensureUserHasSession() {\n const sessionStore = useSessionStore()\n\n // Check if user has any sessions\n const allSessions = sessionStore.getAllSessions()\n if (allSessions.length === 0) {\n console.log('🔄 No sessions found for user, creating default session')\n\n // Ensure we have a default workspace\n const defaultWorkspace = this.workspaces.find(workspace => workspace.isDefault) || null\n if (!defaultWorkspace) {\n console.error('❌ No default workspace found when trying to create default session')\n throw new Error('No default workspace available for session creation')\n }\n\n // Set active workspace\n this.activeWorkspaceUuid = defaultWorkspace.uuid\n\n // Create default session\n const new_chat_text = t('chat.new')\n await sessionStore.createSessionInWorkspace(new_chat_text, defaultWorkspace.uuid)\n console.log('✅ Created default session for new user')\n }\n },\n\n async handleInitialNavigation(urlWorkspaceUuid?: string, urlSessionUuid?: string, isOnDefaultRoute?: boolean) {\n const sessionStore = useSessionStore()\n\n if (urlSessionUuid && sessionStore.getChatSessionByUuid(urlSessionUuid)) {\n // We have a valid session in URL, set it as active\n const session = sessionStore.getChatSessionByUuid(urlSessionUuid)\n if (session) {\n await sessionStore.setActiveSession(session.workspaceUuid || this.activeWorkspaceUuid, urlSessionUuid)\n console.log('✅ Set active session from URL:', urlSessionUuid)\n }\n } else if (this.activeWorkspaceUuid) {\n // Find a session to activate in the active workspace\n const workspaceSessions = sessionStore.getSessionsByWorkspace(this.activeWorkspaceUuid)\n if (workspaceSessions.length > 0) {\n await sessionStore.setActiveSession(this.activeWorkspaceUuid, workspaceSessions[0].uuid)\n console.log('✅ Set first session as active in workspace')\n }\n }\n\n // Handle default route navigation\n if (isOnDefaultRoute && this.activeWorkspaceUuid) {\n console.log('✅ Navigating from default route to active workspace')\n await router.push({\n name: 'WorkspaceChat',\n params: { workspaceUuid: this.activeWorkspaceUuid }\n })\n }\n },\n\n async syncWorkspaces() {\n try {\n this.isLoading = true\n const workspaces = await getWorkspaces()\n this.workspaces = workspaces\n\n // Ensure we have a default workspace\n const defaultWorkspace = this.workspaces.find(workspace => workspace.isDefault) || null\n if (!defaultWorkspace) {\n await this.ensureDefaultWorkspace()\n }\n\n // Set active workspace if not already set\n if (!this.activeWorkspaceUuid && this.workspaces.length > 0) {\n const defaultWs = this.workspaces.find(workspace => workspace.isDefault) || this.workspaces[0]\n this.activeWorkspaceUuid = defaultWs.uuid\n }\n } catch (error) {\n console.error('Failed to sync workspaces:', error)\n throw error\n } finally {\n this.isLoading = false\n }\n },\n\n async ensureDefaultWorkspace() {\n try {\n const defaultWorkspace = await ensureDefaultWorkspace()\n this.workspaces.push(defaultWorkspace)\n this.activeWorkspaceUuid = defaultWorkspace.uuid\n\n // Automatically create a default session for the default workspace\n try {\n const sessionStore = useSessionStore()\n const new_chat_text = t('chat.new')\n await sessionStore.createSessionInWorkspace(new_chat_text, defaultWorkspace.uuid)\n console.log(`✅ Created default session for default workspace: ${defaultWorkspace.name}`)\n } catch (sessionError) {\n console.warn(`⚠️ Failed to create default session for default workspace ${defaultWorkspace.name}:`, sessionError)\n // Don't throw here - workspace creation should succeed even if session creation fails\n }\n\n return defaultWorkspace\n } catch (error) {\n console.error('Failed to ensure default workspace:', error)\n throw error\n }\n },\n\n async setActiveWorkspace(workspaceUuid: string) {\n console.log('🔄 setActiveWorkspace called with:', workspaceUuid)\n console.log('🔍 Current workspaces in store:', this.workspaces.map(w => ({ uuid: w.uuid, name: w.name })))\n\n let workspace = this.workspaces.find(workspace => workspace.uuid === workspaceUuid)\n console.log('🔍 Found workspace in store:', workspace ? workspace.name : 'NOT FOUND')\n\n // If workspace is not loaded, load it on-demand\n if (!workspace) {\n try {\n console.log(`Loading workspace ${workspaceUuid} on-demand...`)\n workspace = await getWorkspace(workspaceUuid)\n this.workspaces.push(workspace)\n console.log(`✅ Loaded workspace on-demand: ${workspace.name}`)\n } catch (error) {\n console.error(`Failed to load workspace ${workspaceUuid}:`, error)\n throw new Error(`Workspace ${workspaceUuid} not found`)\n }\n }\n\n console.log('🔄 Setting activeWorkspaceUuid to:', workspaceUuid)\n this.activeWorkspaceUuid = workspaceUuid\n console.log('✅ activeWorkspaceUuid set, current value:', this.activeWorkspaceUuid)\n\n // Load sessions for this workspace if not already loaded\n const sessionStore = useSessionStore()\n const existingSessions = sessionStore.getSessionsByWorkspace(workspaceUuid)\n console.log('🔍 Existing sessions for workspace:', existingSessions.length)\n\n if (existingSessions.length === 0) {\n console.log(`Loading sessions for workspace ${workspaceUuid}...`)\n await sessionStore.syncWorkspaceSessions(workspaceUuid)\n console.log(`✅ Loaded sessions for workspace: ${workspace.name}`)\n }\n\n // Get the updated sessions list after potential loading\n const sessionsAfterLoad = sessionStore.getSessionsByWorkspace(workspaceUuid)\n\n // Restore the previously active session for this workspace if it is still valid\n const activeSessionForWorkspace = this.workspaceActiveSessions[workspaceUuid]\n const hasSessions = sessionsAfterLoad.length > 0\n const isStoredSessionValid = Boolean(\n activeSessionForWorkspace &&\n sessionsAfterLoad.some(session => session.uuid === activeSessionForWorkspace)\n )\n\n console.log('🔍 Active session for workspace:', activeSessionForWorkspace, 'isValid:', isStoredSessionValid)\n\n if (isStoredSessionValid && activeSessionForWorkspace) {\n sessionStore.setActiveSessionWithoutNavigation(workspaceUuid, activeSessionForWorkspace)\n console.log('✅ Restored previously active session')\n } else if (hasSessions) {\n const firstSession = sessionsAfterLoad[0]\n console.log('🔄 Selecting first available session as active:', firstSession.title)\n sessionStore.setActiveSessionWithoutNavigation(workspaceUuid, firstSession.uuid)\n console.log('✅ Set first session as active')\n } else {\n console.log('⚠️ No sessions available in workspace; clearing active session state')\n this.clearActiveSessionForWorkspace(workspaceUuid)\n sessionStore.activeSessionUuid = null\n }\n\n // Emit an event that the chat view can listen to\n this.$patch((state) => {\n state.pendingSessionRestore = null // Clear any pending restore\n })\n\n console.log('✅ setActiveWorkspace completed successfully')\n },\n\n // Method to handle session restore (called from chat view)\n restoreActiveSession() {\n const pending = this.pendingSessionRestore\n if (pending) {\n const sessionStore = useSessionStore()\n const session = sessionStore.getChatSessionByUuid(pending.sessionUuid)\n if (session) {\n sessionStore.setActiveSession(pending.workspaceUuid, pending.sessionUuid)\n } else {\n // Session no longer exists, clear the tracking\n delete this.workspaceActiveSessions[pending.workspaceUuid]\n }\n // Clear the pending restore\n this.$patch((state) => {\n state.pendingSessionRestore = null\n })\n }\n },\n\n async createWorkspace(name: string, description: string = '', color: string = '#6366f1', icon: string = 'folder') {\n try {\n const newWorkspace = await createWorkspace({\n name,\n description,\n color,\n icon,\n })\n this.workspaces.push(newWorkspace)\n\n // Automatically create a default session for the new workspace\n try {\n const sessionStore = useSessionStore()\n const new_chat_text = t('chat.new')\n await sessionStore.createSessionInWorkspace(new_chat_text, newWorkspace.uuid)\n console.log(`✅ Created default session for new workspace: ${newWorkspace.name}`)\n } catch (sessionError) {\n console.warn(`⚠️ Failed to create default session for workspace ${newWorkspace.name}:`, sessionError)\n // Don't throw here - workspace creation should succeed even if session creation fails\n }\n\n return newWorkspace\n } catch (error) {\n console.error('Failed to create workspace:', error)\n throw error\n }\n },\n\n async updateWorkspace(workspaceUuid: string, updates: any) {\n try {\n const updatedWorkspace = await updateWorkspace(workspaceUuid, updates)\n const index = this.workspaces.findIndex(w => w.uuid === workspaceUuid)\n if (index !== -1) {\n this.workspaces[index] = updatedWorkspace\n }\n return updatedWorkspace\n } catch (error) {\n console.error('Failed to update workspace:', error)\n throw error\n }\n },\n\n async deleteWorkspace(workspaceUuid: string) {\n try {\n await deleteWorkspace(workspaceUuid)\n this.workspaces = this.workspaces.filter(w => w.uuid !== workspaceUuid)\n\n // Remove from active sessions tracking\n delete this.workspaceActiveSessions[workspaceUuid]\n\n // If we deleted the active workspace, switch to default\n if (this.activeWorkspaceUuid === workspaceUuid) {\n const defaultWorkspace = this.workspaces.find(workspace => workspace.isDefault) || null\n if (defaultWorkspace) {\n this.activeWorkspaceUuid = defaultWorkspace.uuid\n } else if (this.workspaces.length > 0) {\n this.activeWorkspaceUuid = this.workspaces[0].uuid\n } else {\n this.activeWorkspaceUuid = null\n }\n }\n } catch (error) {\n console.error('Failed to delete workspace:', error)\n throw error\n }\n },\n\n async setDefaultWorkspace(workspaceUuid: string) {\n try {\n await setDefaultWorkspace(workspaceUuid)\n // Update local state\n this.workspaces.forEach(workspace => {\n workspace.isDefault = workspace.uuid === workspaceUuid\n })\n } catch (error) {\n console.error('Failed to set default workspace:', error)\n throw error\n }\n },\n\n async updateWorkspaceOrder(workspaceUuids: string[]) {\n try {\n if (!Array.isArray(workspaceUuids) || workspaceUuids.length === 0) {\n console.warn('updateWorkspaceOrder expects a non-empty array of UUIDs')\n return\n }\n\n // Persist order positions to backend\n const updatePromises = workspaceUuids.map((uuid, index) => updateWorkspaceOrderApi(uuid, index))\n await Promise.all(updatePromises)\n\n // Reorder locally to reflect saved order\n const reorderedWorkspaces: Chat.Workspace[] = []\n workspaceUuids.forEach((uuid, index) => {\n const workspace = this.workspaces.find(w => w.uuid === uuid)\n if (workspace) {\n reorderedWorkspaces.push({ ...workspace, orderPosition: index })\n }\n })\n\n // If we couldn't build a valid reordered list, avoid wiping the current state\n if (reorderedWorkspaces.length === 0) {\n console.warn('No workspaces matched the provided order list; skipping reorder')\n return\n }\n\n this.workspaces = reorderedWorkspaces\n } catch (error) {\n console.error('Failed to update workspace order:', error)\n throw error\n }\n },\n\n setActiveSessionForWorkspace(workspaceUuid: string, sessionUuid: string) {\n this.workspaceActiveSessions[workspaceUuid] = sessionUuid\n },\n\n clearActiveSessionForWorkspace(workspaceUuid: string) {\n delete this.workspaceActiveSessions[workspaceUuid]\n },\n\n async navigateToWorkspace(workspaceUuid: string, sessionUuid?: string) {\n // Check if we're already on the target route to avoid unnecessary navigation\n const currentRoute = router.currentRoute.value\n const currentWorkspaceUuid = currentRoute.params.workspaceUuid as string\n const currentSessionUuid = currentRoute.params.uuid as string\n\n // If no sessionUuid provided, try to get the active session for this workspace\n let targetSessionUuid = sessionUuid\n if (!targetSessionUuid) {\n const sessionStore = useSessionStore()\n targetSessionUuid = sessionStore.activeSessionUuid || undefined\n }\n\n // More thorough route checking - only skip if route params match exactly\n if (currentRoute.name === 'WorkspaceChat' &&\n currentWorkspaceUuid === workspaceUuid &&\n currentSessionUuid === targetSessionUuid) {\n console.log('Already on exact target route, skipping navigation')\n return\n }\n\n // Additional check: if target matches the last requested session, also skip\n // This prevents navigation loops during rapid switching\n const sessionStore = useSessionStore()\n if (targetSessionUuid &&\n sessionStore.lastRequestedSessionUuid !== targetSessionUuid &&\n sessionStore.lastRequestedSessionUuid !== null) {\n console.log('Navigation target differs from last requested session, skipping to prevent loop')\n return\n }\n\n const route = targetSessionUuid\n ? { name: 'WorkspaceChat', params: { workspaceUuid, uuid: targetSessionUuid } }\n : { name: 'WorkspaceChat', params: { workspaceUuid } }\n\n console.log('Navigating to:', route)\n return router.push(route)\n },\n },\n})\n" }, { "path": "web/src/styles/global.less", "content": "html,\nbody,\n#app {\n\theight: 100%;\n}\n\nbody {\n\tpadding-bottom: constant(safe-area-inset-bottom);\n\tpadding-bottom: env(safe-area-inset-bottom);\n}\n\n.floating-button {\n position: fixed;\n bottom: 10vh;\n right: 10vmin;\n z-index: 99;\n padding: 0.5em;\n border-radius: 50%;\n cursor: pointer;\n background-color: #4ff09a;\n box-shadow: 0 4px 8px 0 rgba(0, 0, 0, 0.2), 0 6px 20px 0 rgba(0, 0, 0, 0.19);\n}\n" }, { "path": "web/src/styles/lib/github-markdown.less", "content": "html.dark {\n .markdown-body {\n color-scheme: dark;\n --color-prettylights-syntax-comment: #8b949e;\n --color-prettylights-syntax-constant: #79c0ff;\n --color-prettylights-syntax-entity: #d2a8ff;\n --color-prettylights-syntax-storage-modifier-import: #c9d1d9;\n --color-prettylights-syntax-entity-tag: #7ee787;\n --color-prettylights-syntax-keyword: #ff7b72;\n --color-prettylights-syntax-string: #a5d6ff;\n --color-prettylights-syntax-variable: #ffa657;\n --color-prettylights-syntax-brackethighlighter-unmatched: #f85149;\n --color-prettylights-syntax-invalid-illegal-text: #f0f6fc;\n --color-prettylights-syntax-invalid-illegal-bg: #8e1519;\n --color-prettylights-syntax-carriage-return-text: #f0f6fc;\n --color-prettylights-syntax-carriage-return-bg: #b62324;\n --color-prettylights-syntax-string-regexp: #7ee787;\n --color-prettylights-syntax-markup-list: #f2cc60;\n --color-prettylights-syntax-markup-heading: #1f6feb;\n --color-prettylights-syntax-markup-italic: #c9d1d9;\n --color-prettylights-syntax-markup-bold: #c9d1d9;\n --color-prettylights-syntax-markup-deleted-text: #ffdcd7;\n --color-prettylights-syntax-markup-deleted-bg: #67060c;\n --color-prettylights-syntax-markup-inserted-text: #aff5b4;\n --color-prettylights-syntax-markup-inserted-bg: #033a16;\n --color-prettylights-syntax-markup-changed-text: #ffdfb6;\n --color-prettylights-syntax-markup-changed-bg: #5a1e02;\n --color-prettylights-syntax-markup-ignored-text: #c9d1d9;\n --color-prettylights-syntax-markup-ignored-bg: #1158c7;\n --color-prettylights-syntax-meta-diff-range: #d2a8ff;\n --color-prettylights-syntax-brackethighlighter-angle: #8b949e;\n --color-prettylights-syntax-sublimelinter-gutter-mark: #484f58;\n --color-prettylights-syntax-constant-other-reference-link: #a5d6ff;\n --color-fg-default: #c9d1d9;\n --color-fg-muted: #8b949e;\n --color-fg-subtle: #6e7681;\n --color-canvas-default: #0d1117;\n --color-canvas-subtle: #161b22;\n --color-border-default: #30363d;\n --color-border-muted: #21262d;\n --color-neutral-muted: rgba(110,118,129,0.4);\n --color-accent-fg: #58a6ff;\n --color-accent-emphasis: #1f6feb;\n --color-attention-subtle: rgba(187,128,9,0.15);\n --color-danger-fg: #f85149;\n }\n}\n\nhtml {\n .markdown-body {\n color-scheme: light;\n --color-prettylights-syntax-comment: #6e7781;\n --color-prettylights-syntax-constant: #0550ae;\n --color-prettylights-syntax-entity: #8250df;\n --color-prettylights-syntax-storage-modifier-import: #24292f;\n --color-prettylights-syntax-entity-tag: #116329;\n --color-prettylights-syntax-keyword: #cf222e;\n --color-prettylights-syntax-string: #0a3069;\n --color-prettylights-syntax-variable: #953800;\n --color-prettylights-syntax-brackethighlighter-unmatched: #82071e;\n --color-prettylights-syntax-invalid-illegal-text: #f6f8fa;\n --color-prettylights-syntax-invalid-illegal-bg: #82071e;\n --color-prettylights-syntax-carriage-return-text: #f6f8fa;\n --color-prettylights-syntax-carriage-return-bg: #cf222e;\n --color-prettylights-syntax-string-regexp: #116329;\n --color-prettylights-syntax-markup-list: #3b2300;\n --color-prettylights-syntax-markup-heading: #0550ae;\n --color-prettylights-syntax-markup-italic: #24292f;\n --color-prettylights-syntax-markup-bold: #24292f;\n --color-prettylights-syntax-markup-deleted-text: #82071e;\n --color-prettylights-syntax-markup-deleted-bg: #ffebe9;\n --color-prettylights-syntax-markup-inserted-text: #116329;\n --color-prettylights-syntax-markup-inserted-bg: #dafbe1;\n --color-prettylights-syntax-markup-changed-text: #953800;\n --color-prettylights-syntax-markup-changed-bg: #ffd8b5;\n --color-prettylights-syntax-markup-ignored-text: #eaeef2;\n --color-prettylights-syntax-markup-ignored-bg: #0550ae;\n --color-prettylights-syntax-meta-diff-range: #8250df;\n --color-prettylights-syntax-brackethighlighter-angle: #57606a;\n --color-prettylights-syntax-sublimelinter-gutter-mark: #8c959f;\n --color-prettylights-syntax-constant-other-reference-link: #0a3069;\n --color-fg-default: #24292f;\n --color-fg-muted: #57606a;\n --color-fg-subtle: #6e7781;\n --color-canvas-default: #ffffff;\n --color-canvas-subtle: #f6f8fa;\n --color-border-default: #d0d7de;\n --color-border-muted: hsla(210,18%,87%,1);\n --color-neutral-muted: rgba(175,184,193,0.2);\n --color-accent-fg: #0969da;\n --color-accent-emphasis: #0969da;\n --color-attention-subtle: #fff8c5;\n --color-danger-fg: #cf222e;\n }\n}\n\n.markdown-body {\n -ms-text-size-adjust: 100%;\n -webkit-text-size-adjust: 100%;\n margin: 0;\n color: var(--color-fg-default);\n background-color: var(--color-canvas-default);\n font-family: -apple-system,BlinkMacSystemFont,\"Segoe UI\",\"Noto Sans\",Helvetica,Arial,sans-serif,\"Apple Color Emoji\",\"Segoe UI Emoji\";\n font-size: 16px;\n line-height: 1.5;\n word-wrap: break-word;\n}\n\n.markdown-body .octicon {\n display: inline-block;\n fill: currentColor;\n vertical-align: text-bottom;\n}\n\n.markdown-body h1:hover .anchor .octicon-link:before,\n.markdown-body h2:hover .anchor .octicon-link:before,\n.markdown-body h3:hover .anchor .octicon-link:before,\n.markdown-body h4:hover .anchor .octicon-link:before,\n.markdown-body h5:hover .anchor .octicon-link:before,\n.markdown-body h6:hover .anchor .octicon-link:before {\n width: 16px;\n height: 16px;\n content: ' ';\n display: inline-block;\n background-color: currentColor;\n -webkit-mask-image: url(\"data:image/svg+xml,\");\n mask-image: url(\"data:image/svg+xml,\");\n}\n\n.markdown-body details,\n.markdown-body figcaption,\n.markdown-body figure {\n display: block;\n}\n\n.markdown-body summary {\n display: list-item;\n}\n\n.markdown-body [hidden] {\n display: none !important;\n}\n\n.markdown-body a {\n background-color: transparent;\n color: var(--color-accent-fg);\n text-decoration: none;\n}\n\n.markdown-body abbr[title] {\n border-bottom: none;\n text-decoration: underline dotted;\n}\n\n.markdown-body b,\n.markdown-body strong {\n font-weight: var(--base-text-weight-semibold, 600);\n}\n\n.markdown-body dfn {\n font-style: italic;\n}\n\n.markdown-body h1 {\n margin: .67em 0;\n font-weight: var(--base-text-weight-semibold, 600);\n padding-bottom: .3em;\n font-size: 2em;\n border-bottom: 1px solid var(--color-border-muted);\n}\n\n.markdown-body mark {\n background-color: var(--color-attention-subtle);\n color: var(--color-fg-default);\n}\n\n.markdown-body small {\n font-size: 90%;\n}\n\n.markdown-body sub,\n.markdown-body sup {\n font-size: 75%;\n line-height: 0;\n position: relative;\n vertical-align: baseline;\n}\n\n.markdown-body sub {\n bottom: -0.25em;\n}\n\n.markdown-body sup {\n top: -0.5em;\n}\n\n.markdown-body img {\n border-style: none;\n max-width: 100%;\n box-sizing: content-box;\n background-color: var(--color-canvas-default);\n}\n\n.markdown-body code,\n.markdown-body kbd,\n.markdown-body pre,\n.markdown-body samp {\n font-family: monospace;\n font-size: 1em;\n}\n\n.markdown-body figure {\n margin: 1em 40px;\n}\n\n.markdown-body hr {\n box-sizing: content-box;\n overflow: hidden;\n background: transparent;\n border-bottom: 1px solid var(--color-border-muted);\n height: .25em;\n padding: 0;\n margin: 24px 0;\n background-color: var(--color-border-default);\n border: 0;\n}\n\n.markdown-body input {\n font: inherit;\n margin: 0;\n overflow: visible;\n font-family: inherit;\n font-size: inherit;\n line-height: inherit;\n}\n\n.markdown-body [type=button],\n.markdown-body [type=reset],\n.markdown-body [type=submit] {\n -webkit-appearance: button;\n}\n\n.markdown-body [type=checkbox],\n.markdown-body [type=radio] {\n box-sizing: border-box;\n padding: 0;\n}\n\n.markdown-body [type=number]::-webkit-inner-spin-button,\n.markdown-body [type=number]::-webkit-outer-spin-button {\n height: auto;\n}\n\n.markdown-body [type=search]::-webkit-search-cancel-button,\n.markdown-body [type=search]::-webkit-search-decoration {\n -webkit-appearance: none;\n}\n\n.markdown-body ::-webkit-input-placeholder {\n color: inherit;\n opacity: .54;\n}\n\n.markdown-body ::-webkit-file-upload-button {\n -webkit-appearance: button;\n font: inherit;\n}\n\n.markdown-body a:hover {\n text-decoration: underline;\n}\n\n.markdown-body ::placeholder {\n color: var(--color-fg-subtle);\n opacity: 1;\n}\n\n.markdown-body hr::before {\n display: table;\n content: \"\";\n}\n\n.markdown-body hr::after {\n display: table;\n clear: both;\n content: \"\";\n}\n\n.markdown-body table {\n border-spacing: 0;\n border-collapse: collapse;\n display: block;\n width: max-content;\n max-width: 100%;\n overflow: auto;\n}\n\n.markdown-body td,\n.markdown-body th {\n padding: 0;\n}\n\n.markdown-body details summary {\n cursor: pointer;\n}\n\n.markdown-body details:not([open])>*:not(summary) {\n display: none !important;\n}\n\n.markdown-body a:focus,\n.markdown-body [role=button]:focus,\n.markdown-body input[type=radio]:focus,\n.markdown-body input[type=checkbox]:focus {\n outline: 2px solid var(--color-accent-fg);\n outline-offset: -2px;\n box-shadow: none;\n}\n\n.markdown-body a:focus:not(:focus-visible),\n.markdown-body [role=button]:focus:not(:focus-visible),\n.markdown-body input[type=radio]:focus:not(:focus-visible),\n.markdown-body input[type=checkbox]:focus:not(:focus-visible) {\n outline: solid 1px transparent;\n}\n\n.markdown-body a:focus-visible,\n.markdown-body [role=button]:focus-visible,\n.markdown-body input[type=radio]:focus-visible,\n.markdown-body input[type=checkbox]:focus-visible {\n outline: 2px solid var(--color-accent-fg);\n outline-offset: -2px;\n box-shadow: none;\n}\n\n.markdown-body a:not([class]):focus,\n.markdown-body a:not([class]):focus-visible,\n.markdown-body input[type=radio]:focus,\n.markdown-body input[type=radio]:focus-visible,\n.markdown-body input[type=checkbox]:focus,\n.markdown-body input[type=checkbox]:focus-visible {\n outline-offset: 0;\n}\n\n.markdown-body kbd {\n display: inline-block;\n padding: 3px 5px;\n font: 11px ui-monospace,SFMono-Regular,SF Mono,Menlo,Consolas,Liberation Mono,monospace;\n line-height: 10px;\n color: var(--color-fg-default);\n vertical-align: middle;\n background-color: var(--color-canvas-subtle);\n border: solid 1px var(--color-neutral-muted);\n border-bottom-color: var(--color-neutral-muted);\n border-radius: 6px;\n box-shadow: inset 0 -1px 0 var(--color-neutral-muted);\n}\n\n.markdown-body h1,\n.markdown-body h2,\n.markdown-body h3,\n.markdown-body h4,\n.markdown-body h5,\n.markdown-body h6 {\n margin-top: 24px;\n margin-bottom: 16px;\n font-weight: var(--base-text-weight-semibold, 600);\n line-height: 1.25;\n}\n\n.markdown-body h2 {\n font-weight: var(--base-text-weight-semibold, 600);\n padding-bottom: .3em;\n font-size: 1.5em;\n border-bottom: 1px solid var(--color-border-muted);\n}\n\n.markdown-body h3 {\n font-weight: var(--base-text-weight-semibold, 600);\n font-size: 1.25em;\n}\n\n.markdown-body h4 {\n font-weight: var(--base-text-weight-semibold, 600);\n font-size: 1em;\n}\n\n.markdown-body h5 {\n font-weight: var(--base-text-weight-semibold, 600);\n font-size: .875em;\n}\n\n.markdown-body h6 {\n font-weight: var(--base-text-weight-semibold, 600);\n font-size: .85em;\n color: var(--color-fg-muted);\n}\n\n.markdown-body p {\n margin-top: 0;\n margin-bottom: 10px;\n}\n\n.markdown-body blockquote {\n margin: 0;\n padding: 0 1em;\n color: var(--color-fg-muted);\n border-left: .25em solid var(--color-border-default);\n}\n\n.markdown-body ul,\n.markdown-body ol {\n margin-top: 0;\n margin-bottom: 0;\n padding-left: 2em;\n}\n\n.markdown-body ol ol,\n.markdown-body ul ol {\n list-style-type: lower-roman;\n}\n\n.markdown-body ul ul ol,\n.markdown-body ul ol ol,\n.markdown-body ol ul ol,\n.markdown-body ol ol ol {\n list-style-type: lower-alpha;\n}\n\n.markdown-body dd {\n margin-left: 0;\n}\n\n.markdown-body tt,\n.markdown-body code,\n.markdown-body samp {\n font-family: ui-monospace,SFMono-Regular,SF Mono,Menlo,Consolas,Liberation Mono,monospace;\n font-size: 12px;\n}\n\n.markdown-body pre {\n margin-top: 0;\n margin-bottom: 0;\n font-family: ui-monospace,SFMono-Regular,SF Mono,Menlo,Consolas,Liberation Mono,monospace;\n font-size: 12px;\n word-wrap: normal;\n}\n\n.markdown-body .octicon {\n display: inline-block;\n overflow: visible !important;\n vertical-align: text-bottom;\n fill: currentColor;\n}\n\n.markdown-body input::-webkit-outer-spin-button,\n.markdown-body input::-webkit-inner-spin-button {\n margin: 0;\n -webkit-appearance: none;\n appearance: none;\n}\n\n.markdown-body::before {\n display: table;\n content: \"\";\n}\n\n.markdown-body::after {\n display: table;\n clear: both;\n content: \"\";\n}\n\n.markdown-body>*:first-child {\n margin-top: 0 !important;\n}\n\n.markdown-body>*:last-child {\n margin-bottom: 0 !important;\n}\n\n.markdown-body a:not([href]) {\n color: inherit;\n text-decoration: none;\n}\n\n.markdown-body .absent {\n color: var(--color-danger-fg);\n}\n\n.markdown-body .anchor {\n float: left;\n padding-right: 4px;\n margin-left: -20px;\n line-height: 1;\n}\n\n.markdown-body .anchor:focus {\n outline: none;\n}\n\n.markdown-body p,\n.markdown-body blockquote,\n.markdown-body ul,\n.markdown-body ol,\n.markdown-body dl,\n.markdown-body table,\n.markdown-body pre,\n.markdown-body details {\n margin-top: 0;\n margin-bottom: 16px;\n}\n\n.markdown-body blockquote>:first-child {\n margin-top: 0;\n}\n\n.markdown-body blockquote>:last-child {\n margin-bottom: 0;\n}\n\n.markdown-body h1 .octicon-link,\n.markdown-body h2 .octicon-link,\n.markdown-body h3 .octicon-link,\n.markdown-body h4 .octicon-link,\n.markdown-body h5 .octicon-link,\n.markdown-body h6 .octicon-link {\n color: var(--color-fg-default);\n vertical-align: middle;\n visibility: hidden;\n}\n\n.markdown-body h1:hover .anchor,\n.markdown-body h2:hover .anchor,\n.markdown-body h3:hover .anchor,\n.markdown-body h4:hover .anchor,\n.markdown-body h5:hover .anchor,\n.markdown-body h6:hover .anchor {\n text-decoration: none;\n}\n\n.markdown-body h1:hover .anchor .octicon-link,\n.markdown-body h2:hover .anchor .octicon-link,\n.markdown-body h3:hover .anchor .octicon-link,\n.markdown-body h4:hover .anchor .octicon-link,\n.markdown-body h5:hover .anchor .octicon-link,\n.markdown-body h6:hover .anchor .octicon-link {\n visibility: visible;\n}\n\n.markdown-body h1 tt,\n.markdown-body h1 code,\n.markdown-body h2 tt,\n.markdown-body h2 code,\n.markdown-body h3 tt,\n.markdown-body h3 code,\n.markdown-body h4 tt,\n.markdown-body h4 code,\n.markdown-body h5 tt,\n.markdown-body h5 code,\n.markdown-body h6 tt,\n.markdown-body h6 code {\n padding: 0 .2em;\n font-size: inherit;\n}\n\n.markdown-body summary h1,\n.markdown-body summary h2,\n.markdown-body summary h3,\n.markdown-body summary h4,\n.markdown-body summary h5,\n.markdown-body summary h6 {\n display: inline-block;\n}\n\n.markdown-body summary h1 .anchor,\n.markdown-body summary h2 .anchor,\n.markdown-body summary h3 .anchor,\n.markdown-body summary h4 .anchor,\n.markdown-body summary h5 .anchor,\n.markdown-body summary h6 .anchor {\n margin-left: -40px;\n}\n\n.markdown-body summary h1,\n.markdown-body summary h2 {\n padding-bottom: 0;\n border-bottom: 0;\n}\n\n.markdown-body ul.no-list,\n.markdown-body ol.no-list {\n padding: 0;\n list-style-type: none;\n}\n\n.markdown-body ol[type=a] {\n list-style-type: lower-alpha;\n}\n\n.markdown-body ol[type=A] {\n list-style-type: upper-alpha;\n}\n\n.markdown-body ol[type=i] {\n list-style-type: lower-roman;\n}\n\n.markdown-body ol[type=I] {\n list-style-type: upper-roman;\n}\n\n.markdown-body ol[type=\"1\"] {\n list-style-type: decimal;\n}\n\n.markdown-body div>ol:not([type]) {\n list-style-type: decimal;\n}\n\n.markdown-body ul ul,\n.markdown-body ul ol,\n.markdown-body ol ol,\n.markdown-body ol ul {\n margin-top: 0;\n margin-bottom: 0;\n}\n\n.markdown-body li>p {\n margin-top: 16px;\n}\n\n.markdown-body li+li {\n margin-top: .25em;\n}\n\n.markdown-body dl {\n padding: 0;\n}\n\n.markdown-body dl dt {\n padding: 0;\n margin-top: 16px;\n font-size: 1em;\n font-style: italic;\n font-weight: var(--base-text-weight-semibold, 600);\n}\n\n.markdown-body dl dd {\n padding: 0 16px;\n margin-bottom: 16px;\n}\n\n.markdown-body table th {\n font-weight: var(--base-text-weight-semibold, 600);\n}\n\n.markdown-body table th,\n.markdown-body table td {\n padding: 6px 13px;\n border: 1px solid var(--color-border-default);\n}\n\n.markdown-body table tr {\n background-color: var(--color-canvas-default);\n border-top: 1px solid var(--color-border-muted);\n}\n\n.markdown-body table tr:nth-child(2n) {\n background-color: var(--color-canvas-subtle);\n}\n\n.markdown-body table img {\n background-color: transparent;\n}\n\n.markdown-body img[align=right] {\n padding-left: 20px;\n}\n\n.markdown-body img[align=left] {\n padding-right: 20px;\n}\n\n.markdown-body .emoji {\n max-width: none;\n vertical-align: text-top;\n background-color: transparent;\n}\n\n.markdown-body span.frame {\n display: block;\n overflow: hidden;\n}\n\n.markdown-body span.frame>span {\n display: block;\n float: left;\n width: auto;\n padding: 7px;\n margin: 13px 0 0;\n overflow: hidden;\n border: 1px solid var(--color-border-default);\n}\n\n.markdown-body span.frame span img {\n display: block;\n float: left;\n}\n\n.markdown-body span.frame span span {\n display: block;\n padding: 5px 0 0;\n clear: both;\n color: var(--color-fg-default);\n}\n\n.markdown-body span.align-center {\n display: block;\n overflow: hidden;\n clear: both;\n}\n\n.markdown-body span.align-center>span {\n display: block;\n margin: 13px auto 0;\n overflow: hidden;\n text-align: center;\n}\n\n.markdown-body span.align-center span img {\n margin: 0 auto;\n text-align: center;\n}\n\n.markdown-body span.align-right {\n display: block;\n overflow: hidden;\n clear: both;\n}\n\n.markdown-body span.align-right>span {\n display: block;\n margin: 13px 0 0;\n overflow: hidden;\n text-align: right;\n}\n\n.markdown-body span.align-right span img {\n margin: 0;\n text-align: right;\n}\n\n.markdown-body span.float-left {\n display: block;\n float: left;\n margin-right: 13px;\n overflow: hidden;\n}\n\n.markdown-body span.float-left span {\n margin: 13px 0 0;\n}\n\n.markdown-body span.float-right {\n display: block;\n float: right;\n margin-left: 13px;\n overflow: hidden;\n}\n\n.markdown-body span.float-right>span {\n display: block;\n margin: 13px auto 0;\n overflow: hidden;\n text-align: right;\n}\n\n.markdown-body code,\n.markdown-body tt {\n padding: .2em .4em;\n margin: 0;\n font-size: 85%;\n white-space: break-spaces;\n background-color: var(--color-neutral-muted);\n border-radius: 6px;\n}\n\n.markdown-body code br,\n.markdown-body tt br {\n display: none;\n}\n\n.markdown-body del code {\n text-decoration: inherit;\n}\n\n.markdown-body samp {\n font-size: 85%;\n}\n\n.markdown-body pre code {\n font-size: 100%;\n}\n\n.markdown-body pre>code {\n padding: 0;\n margin: 0;\n word-break: normal;\n white-space: pre;\n background: transparent;\n border: 0;\n}\n\n.markdown-body .highlight {\n margin-bottom: 16px;\n}\n\n.markdown-body .highlight pre {\n margin-bottom: 0;\n word-break: normal;\n}\n\n.markdown-body .highlight pre,\n.markdown-body pre {\n padding: 16px;\n overflow: auto;\n font-size: 85%;\n line-height: 1.45;\n background-color: var(--color-canvas-subtle);\n border-radius: 6px;\n}\n\n.markdown-body pre code,\n.markdown-body pre tt {\n display: inline;\n max-width: auto;\n padding: 0;\n margin: 0;\n overflow: visible;\n line-height: inherit;\n word-wrap: normal;\n background-color: transparent;\n border: 0;\n}\n\n.markdown-body .csv-data td,\n.markdown-body .csv-data th {\n padding: 5px;\n overflow: hidden;\n font-size: 12px;\n line-height: 1;\n text-align: left;\n white-space: nowrap;\n}\n\n.markdown-body .csv-data .blob-num {\n padding: 10px 8px 9px;\n text-align: right;\n background: var(--color-canvas-default);\n border: 0;\n}\n\n.markdown-body .csv-data tr {\n border-top: 0;\n}\n\n.markdown-body .csv-data th {\n font-weight: var(--base-text-weight-semibold, 600);\n background: var(--color-canvas-subtle);\n border-top: 0;\n}\n\n.markdown-body [data-footnote-ref]::before {\n content: \"[\";\n}\n\n.markdown-body [data-footnote-ref]::after {\n content: \"]\";\n}\n\n.markdown-body .footnotes {\n font-size: 12px;\n color: var(--color-fg-muted);\n border-top: 1px solid var(--color-border-default);\n}\n\n.markdown-body .footnotes ol {\n padding-left: 16px;\n}\n\n.markdown-body .footnotes ol ul {\n display: inline-block;\n padding-left: 16px;\n margin-top: 16px;\n}\n\n.markdown-body .footnotes li {\n position: relative;\n}\n\n.markdown-body .footnotes li:target::before {\n position: absolute;\n top: -8px;\n right: -8px;\n bottom: -8px;\n left: -24px;\n pointer-events: none;\n content: \"\";\n border: 2px solid var(--color-accent-emphasis);\n border-radius: 6px;\n}\n\n.markdown-body .footnotes li:target {\n color: var(--color-fg-default);\n}\n\n.markdown-body .footnotes .data-footnote-backref g-emoji {\n font-family: monospace;\n}\n\n.markdown-body .pl-c {\n color: var(--color-prettylights-syntax-comment);\n}\n\n.markdown-body .pl-c1,\n.markdown-body .pl-s .pl-v {\n color: var(--color-prettylights-syntax-constant);\n}\n\n.markdown-body .pl-e,\n.markdown-body .pl-en {\n color: var(--color-prettylights-syntax-entity);\n}\n\n.markdown-body .pl-smi,\n.markdown-body .pl-s .pl-s1 {\n color: var(--color-prettylights-syntax-storage-modifier-import);\n}\n\n.markdown-body .pl-ent {\n color: var(--color-prettylights-syntax-entity-tag);\n}\n\n.markdown-body .pl-k {\n color: var(--color-prettylights-syntax-keyword);\n}\n\n.markdown-body .pl-s,\n.markdown-body .pl-pds,\n.markdown-body .pl-s .pl-pse .pl-s1,\n.markdown-body .pl-sr,\n.markdown-body .pl-sr .pl-cce,\n.markdown-body .pl-sr .pl-sre,\n.markdown-body .pl-sr .pl-sra {\n color: var(--color-prettylights-syntax-string);\n}\n\n.markdown-body .pl-v,\n.markdown-body .pl-smw {\n color: var(--color-prettylights-syntax-variable);\n}\n\n.markdown-body .pl-bu {\n color: var(--color-prettylights-syntax-brackethighlighter-unmatched);\n}\n\n.markdown-body .pl-ii {\n color: var(--color-prettylights-syntax-invalid-illegal-text);\n background-color: var(--color-prettylights-syntax-invalid-illegal-bg);\n}\n\n.markdown-body .pl-c2 {\n color: var(--color-prettylights-syntax-carriage-return-text);\n background-color: var(--color-prettylights-syntax-carriage-return-bg);\n}\n\n.markdown-body .pl-sr .pl-cce {\n font-weight: bold;\n color: var(--color-prettylights-syntax-string-regexp);\n}\n\n.markdown-body .pl-ml {\n color: var(--color-prettylights-syntax-markup-list);\n}\n\n.markdown-body .pl-mh,\n.markdown-body .pl-mh .pl-en,\n.markdown-body .pl-ms {\n font-weight: bold;\n color: var(--color-prettylights-syntax-markup-heading);\n}\n\n.markdown-body .pl-mi {\n font-style: italic;\n color: var(--color-prettylights-syntax-markup-italic);\n}\n\n.markdown-body .pl-mb {\n font-weight: bold;\n color: var(--color-prettylights-syntax-markup-bold);\n}\n\n.markdown-body .pl-md {\n color: var(--color-prettylights-syntax-markup-deleted-text);\n background-color: var(--color-prettylights-syntax-markup-deleted-bg);\n}\n\n.markdown-body .pl-mi1 {\n color: var(--color-prettylights-syntax-markup-inserted-text);\n background-color: var(--color-prettylights-syntax-markup-inserted-bg);\n}\n\n.markdown-body .pl-mc {\n color: var(--color-prettylights-syntax-markup-changed-text);\n background-color: var(--color-prettylights-syntax-markup-changed-bg);\n}\n\n.markdown-body .pl-mi2 {\n color: var(--color-prettylights-syntax-markup-ignored-text);\n background-color: var(--color-prettylights-syntax-markup-ignored-bg);\n}\n\n.markdown-body .pl-mdr {\n font-weight: bold;\n color: var(--color-prettylights-syntax-meta-diff-range);\n}\n\n.markdown-body .pl-ba {\n color: var(--color-prettylights-syntax-brackethighlighter-angle);\n}\n\n.markdown-body .pl-sg {\n color: var(--color-prettylights-syntax-sublimelinter-gutter-mark);\n}\n\n.markdown-body .pl-corl {\n text-decoration: underline;\n color: var(--color-prettylights-syntax-constant-other-reference-link);\n}\n\n.markdown-body g-emoji {\n display: inline-block;\n min-width: 1ch;\n font-family: \"Apple Color Emoji\",\"Segoe UI Emoji\",\"Segoe UI Symbol\";\n font-size: 1em;\n font-style: normal !important;\n font-weight: var(--base-text-weight-normal, 400);\n line-height: 1;\n vertical-align: -0.075em;\n}\n\n.markdown-body g-emoji img {\n width: 1em;\n height: 1em;\n}\n\n.markdown-body .task-list-item {\n list-style-type: none;\n}\n\n.markdown-body .task-list-item label {\n font-weight: var(--base-text-weight-normal, 400);\n}\n\n.markdown-body .task-list-item.enabled label {\n cursor: pointer;\n}\n\n.markdown-body .task-list-item+.task-list-item {\n margin-top: 4px;\n}\n\n.markdown-body .task-list-item .handle {\n display: none;\n}\n\n.markdown-body .task-list-item-checkbox {\n margin: 0 .2em .25em -1.4em;\n vertical-align: middle;\n}\n\n.markdown-body .contains-task-list:dir(rtl) .task-list-item-checkbox {\n margin: 0 -1.6em .25em .2em;\n}\n\n.markdown-body .contains-task-list {\n position: relative;\n}\n\n.markdown-body .contains-task-list:hover .task-list-item-convert-container,\n.markdown-body .contains-task-list:focus-within .task-list-item-convert-container {\n display: block;\n width: auto;\n height: 24px;\n overflow: visible;\n clip: auto;\n}\n\n.markdown-body ::-webkit-calendar-picker-indicator {\n filter: invert(50%);\n}\n" }, { "path": "web/src/styles/lib/highlight.less", "content": "html.dark {\n\tpre code.hljs {\n\t\tdisplay: block;\n\t\toverflow-x: auto;\n\t\tpadding: 1em\n\t}\n\n\tcode.hljs {\n\t\tpadding: 3px 5px\n\t}\n\n\t.hljs {\n\t\tcolor: #abb2bf;\n\t\tbackground: #282c34\n\t}\n\n\t.hljs-keyword,\n\t.hljs-operator,\n\t.hljs-pattern-match {\n\t\tcolor: #f92672\n\t}\n\n\t.hljs-function,\n\t.hljs-pattern-match .hljs-constructor {\n\t\tcolor: #61aeee\n\t}\n\n\t.hljs-function .hljs-params {\n\t\tcolor: #a6e22e\n\t}\n\n\t.hljs-function .hljs-params .hljs-typing {\n\t\tcolor: #fd971f\n\t}\n\n\t.hljs-module-access .hljs-module {\n\t\tcolor: #7e57c2\n\t}\n\n\t.hljs-constructor {\n\t\tcolor: #e2b93d\n\t}\n\n\t.hljs-constructor .hljs-string {\n\t\tcolor: #9ccc65\n\t}\n\n\t.hljs-comment,\n\t.hljs-quote {\n\t\tcolor: #b18eb1;\n\t\tfont-style: italic\n\t}\n\n\t.hljs-doctag,\n\t.hljs-formula {\n\t\tcolor: #c678dd\n\t}\n\n\t.hljs-deletion,\n\t.hljs-name,\n\t.hljs-section,\n\t.hljs-selector-tag,\n\t.hljs-subst {\n\t\tcolor: #e06c75\n\t}\n\n\t.hljs-literal {\n\t\tcolor: #56b6c2\n\t}\n\n\t.hljs-addition,\n\t.hljs-attribute,\n\t.hljs-meta .hljs-string,\n\t.hljs-regexp,\n\t.hljs-string {\n\t\tcolor: #98c379\n\t}\n\n\t.hljs-built_in,\n\t.hljs-class .hljs-title,\n\t.hljs-title.class_ {\n\t\tcolor: #e6c07b\n\t}\n\n\t.hljs-attr,\n\t.hljs-number,\n\t.hljs-selector-attr,\n\t.hljs-selector-class,\n\t.hljs-selector-pseudo,\n\t.hljs-template-variable,\n\t.hljs-type,\n\t.hljs-variable {\n\t\tcolor: #d19a66\n\t}\n\n\t.hljs-bullet,\n\t.hljs-link,\n\t.hljs-meta,\n\t.hljs-selector-id,\n\t.hljs-symbol,\n\t.hljs-title {\n\t\tcolor: #61aeee\n\t}\n\n\t.hljs-emphasis {\n\t\tfont-style: italic\n\t}\n\n\t.hljs-strong {\n\t\tfont-weight: 700\n\t}\n\n\t.hljs-link {\n\t\ttext-decoration: underline\n\t}\n}\n\nhtml {\n\tpre code.hljs {\n\t\tdisplay: block;\n\t\toverflow-x: auto;\n\t\tpadding: 1em\n\t}\n\n\tcode.hljs {\n\t\tpadding: 3px 5px;\n\t\t&::-webkit-scrollbar {\n\t\t\theight: 4px;\n\t\t}\n\t}\n\n\t.hljs {\n\t\tcolor: #383a42;\n\t\tbackground: #fafafa\n\t}\n\n\t.hljs-comment,\n\t.hljs-quote {\n\t\tcolor: #a0a1a7;\n\t\tfont-style: italic\n\t}\n\n\t.hljs-doctag,\n\t.hljs-formula,\n\t.hljs-keyword {\n\t\tcolor: #a626a4\n\t}\n\n\t.hljs-deletion,\n\t.hljs-name,\n\t.hljs-section,\n\t.hljs-selector-tag,\n\t.hljs-subst {\n\t\tcolor: #e45649\n\t}\n\n\t.hljs-literal {\n\t\tcolor: #0184bb\n\t}\n\n\t.hljs-addition,\n\t.hljs-attribute,\n\t.hljs-meta .hljs-string,\n\t.hljs-regexp,\n\t.hljs-string {\n\t\tcolor: #50a14f\n\t}\n\n\t.hljs-attr,\n\t.hljs-number,\n\t.hljs-selector-attr,\n\t.hljs-selector-class,\n\t.hljs-selector-pseudo,\n\t.hljs-template-variable,\n\t.hljs-type,\n\t.hljs-variable {\n\t\tcolor: #986801\n\t}\n\n\t.hljs-bullet,\n\t.hljs-link,\n\t.hljs-meta,\n\t.hljs-selector-id,\n\t.hljs-symbol,\n\t.hljs-title {\n\t\tcolor: #4078f2\n\t}\n\n\t.hljs-built_in,\n\t.hljs-class .hljs-title,\n\t.hljs-title.class_ {\n\t\tcolor: #c18401\n\t}\n\n\t.hljs-emphasis {\n\t\tfont-style: italic\n\t}\n\n\t.hljs-strong {\n\t\tfont-weight: 700\n\t}\n\n\t.hljs-link {\n\t\ttext-decoration: underline\n\t}\n}\n" }, { "path": "web/src/styles/lib/tailwind.css", "content": "@tailwind base;\n@tailwind components;\n@tailwind utilities;\n" }, { "path": "web/src/types/chat-models.ts", "content": "export interface ChatModel {\n id: number\n name: string\n label: string\n isEnable: boolean\n isDefault: boolean\n orderNumber: number\n lastUsageTime: string\n apiType: string\n maxTokens?: number\n costPer1kTokens?: number\n description?: string\n}\n\nexport interface CreateChatModelRequest {\n name: string\n label: string\n apiType: string\n isEnable?: boolean\n isDefault?: boolean\n orderNumber?: number\n maxTokens?: number\n costPer1kTokens?: number\n description?: string\n}\n\nexport interface UpdateChatModelRequest {\n name?: string\n label?: string\n apiType?: string\n isEnable?: boolean\n isDefault?: boolean\n orderNumber?: number\n maxTokens?: number\n costPer1kTokens?: number\n description?: string\n}\n\nexport interface ChatModelSelectOption {\n label: string | (() => any)\n value: string\n disabled?: boolean\n}\n\nexport interface ChatModelsResponse {\n models: ChatModel[]\n total: number\n}" }, { "path": "web/src/typings/chat.d.ts", "content": "declare namespace Chat {\n\n\tinterface Artifact {\n\t\tuuid: string\n\t\ttype: string // 'code', 'html', 'svg', 'mermaid', 'json', 'markdown'\n\t\ttitle: string\n\t\tcontent: string\n\t\tlanguage?: string // for code artifacts\n\t}\n\n\tinterface Message {\n\t\tuuid: string,\n\t\tdateTime: string\n\t\ttext: string\n\t\tmodel?: string\n\t\tinversion?: boolean\n\t\terror?: boolean\n\t\tloading?: boolean\n\t\tisPrompt?: boolean\n\t\tisPin?: boolean\n\t\tartifacts?: Artifact[]\n\t\tsuggestedQuestions?: string[]\n\t\tsuggestedQuestionsLoading?: boolean\n\t\tsuggestedQuestionsBatches?: string[][]\n\t\tcurrentSuggestedQuestionsBatch?: number\n\t\tsuggestedQuestionsGenerating?: boolean\n\t}\n\n\tinterface Session {\n\t\tuuid: string\n\t\ttitle: string\n\t\tisEdit: boolean\n\t\tmaxLength?: number\n\t\ttemperature?: number\n\t\tmodel?: string\n\t\ttopP?: number\n\t\tn?: number\n\t\tmaxTokens?: number\n\t\tdebug?: boolean\n\t\tsummarizeMode?: boolean\n\t\texploreMode?: boolean\n\t\tartifactEnabled?: boolean\n\t\tworkspaceUuid?: string\n\t}\n\n\tinterface Workspace {\n\t\tuuid: string\n\t\tname: string\n\t\tdescription?: string\n\t\tcolor: string\n\t\ticon: string\n\t\tisDefault: boolean\n\t\torderPosition?: number\n\t\tsessionCount?: number\n\t\tcreatedAt: string\n\t\tupdatedAt: string\n\t}\n\n\tinterface ActiveSession {\n\t\tsessionUuid: string | null\n\t\tworkspaceUuid: string | null\n\t}\n\n\tinterface ChatState {\n\t\tactiveSession: ActiveSession\n\t\tworkspaceActiveSessions: { [workspaceUuid: string]: string } // workspaceUuid -> sessionUuid\n\t\tworkspaces: Workspace[]\n\t\tworkspaceHistory: { [workspaceUuid: string]: Session[] } // workspaceUuid -> Session[]\n\t\tchat: { [uuid: string]: Message[] }\n\t}\n\n\tinterface ConversationRequest {\n\t\tuuid?: string,\n\t\tconversationId?: string\n\t\tparentMessageId?: string\n\t}\n\n\tinterface ConversationResponse {\n\t\tconversationId: string\n\t\tdetail: {\n\t\t\t// rome-ignore lint/suspicious/noExplicitAny: \n\t\t\tchoices: { finish_reason: string; index: number; logprobs: any; text: string }[]\n\t\t\tcreated: number\n\t\t\tid: string\n\t\t\tmodel: string\n\t\t\tobject: string\n\t\t\tusage: { completion_tokens: number; prompt_tokens: number; total_tokens: number }\n\t\t}\n\t\tid: string\n\t\tparentMessageId: string\n\t\trole: string\n\t\ttext: string\n\t}\n\n\tinterface ChatModel {\n\t\tid?: number\n\t\tapiAuthHeader: string\n\t\tapiAuthKey: string\n\t\tapiType: string\n\t\tisDefault: boolean\n\t\tlabel: string\n\t\tname: string\n\t\turl: string\n\t\tenablePerModeRatelimit: boolean,\n\t\tisEnable: boolean,\n\t\tmaxToken?: string,\n\t\tdefaultToken?: string,\n\t\torderNumber?: string,\n\t\thttpTimeOut?: number\n\n\t}\n\n\tinterface ChatModelPrivilege {\n\t\tid: string\n\t\tchatModelName: string\n\t\tfullName: string\n\t\tuserEmail: string\n\t\trateLimit: string\n\t}\n\n\tinterface Comment {\n\t\tuuid: string\n\t\tchatMessageUuid: string\n\t\tcontent: string\n\t\tcreatedAt: string\n\t\tauthorUsername: string\n\t}\n\n\n\n}\n\ndeclare namespace Snapshot {\n\n\tinterface Snapshot {\n\t\tuuid: string;\n\t\ttitle: string;\n\t\tsummary: string;\n\t\ttags: Record;\n\t\tcreatedAt: string;\n\t\ttyp: 'chatbot' | 'snapshot';\n\t}\n\n\tinterface PostLink {\n\t\tuuid: string;\n\t\tdate: string;\n\t\ttitle: string;\n\t}\n}\n\ndeclare namespace Bot {\n\tinterface BotAnswerHistory {\n\t\tid: number\n\t\tbotUuid: string\n\t\tuserId: number\n\t\tprompt: string\n\t\tanswer: string\n\t\tmodel: string\n\t\ttokensUsed: number\n\t\tcreatedAt: string\n\t\tupdatedAt: string\n\t}\n\n}\n" }, { "path": "web/src/typings/global.d.ts", "content": "interface Window {\n $loadingBar?: import('naive-ui').LoadingBarProviderInst;\n $dialog?: import('naive-ui').DialogProviderInst;\n $message?: import('naive-ui').MessageProviderInst;\n $notification?: import('naive-ui').NotificationProviderInst;\n}\n\ninterface SelectOption {\n label: string\n value: string\n}\n\ndeclare module '*.svg' {\n import type { DefineComponent } from 'vue'\n const component: DefineComponent\n export default component\n}" }, { "path": "web/src/utils/__tests__/date.test.ts", "content": "import { expect, describe, it, beforeAll } from 'vitest'\nimport { displayLocaleDate } from '../date'\n\n\ndescribe('displayLocaleDate', () => {\n beforeAll(() => {\n // Set a fixed timezone for all tests\n process.env.TZ = 'Asia/Shanghai'\n })\n\n it('should format ISO date string to local date and time', () => {\n const isoDate = '2025-03-05T12:48:11.990824Z'\n const result = displayLocaleDate(isoDate)\n expect(result).toBe('3/5/2025 8:48 PM')\n })\n\n it('should handle date without milliseconds', () => {\n const isoDate = '2025-02-26T08:58:48Z'\n const result = displayLocaleDate(isoDate)\n expect(result).toBe('2/26/2025 4:58 PM')\n })\n\n it('should handle invalid date string', () => {\n const invalidDate = 'invalid-date'\n const result = displayLocaleDate(invalidDate)\n expect(result).toBe('Invalid DateTime')\n })\n\n})\n" }, { "path": "web/src/utils/artifacts.ts", "content": "import { v7 as uuid } from 'uuid'\n// Use the Chat namespace type\n\nexport type Artifact = Chat.Artifact\n\n// Generate a simple UUID for frontend use\nfunction generateUUID(): string {\n return uuid()\n}\n\n// Extract artifacts from message content (mirrors backend logic)\nexport function extractArtifacts(content: string): Artifact[] {\n const artifacts: Artifact[] = []\n\n // Pattern for HTML artifacts (check specific types first)\n const htmlArtifactRegex = /```html\\s*\\s*\\n(.*?)\\n```/gs\n const htmlMatches = content.matchAll(htmlArtifactRegex)\n\n for (const match of htmlMatches) {\n const title = match[1].trim()\n const artifactContent = match[2].trim()\n\n const artifact: Artifact = {\n uuid: generateUUID(),\n type: 'html',\n title,\n content: artifactContent,\n language: 'html'\n }\n artifacts.push(artifact)\n }\n\n // Pattern for SVG artifacts\n const svgArtifactRegex = /```svg\\s*\\s*\\n(.*?)\\n```/gs\n const svgMatches = content.matchAll(svgArtifactRegex)\n\n for (const match of svgMatches) {\n const title = match[1].trim()\n const artifactContent = match[2].trim()\n\n const artifact: Artifact = {\n uuid: generateUUID(),\n type: 'svg',\n title,\n content: artifactContent,\n language: 'svg'\n }\n artifacts.push(artifact)\n }\n\n // Pattern for Mermaid diagrams\n const mermaidArtifactRegex = /```mermaid\\s*\\s*\\n(.*?)\\n```/gs\n const mermaidMatches = content.matchAll(mermaidArtifactRegex)\n\n for (const match of mermaidMatches) {\n const title = match[1].trim()\n const artifactContent = match[2].trim()\n\n const artifact: Artifact = {\n uuid: generateUUID(),\n type: 'mermaid',\n title,\n content: artifactContent,\n language: 'mermaid'\n }\n artifacts.push(artifact)\n }\n\n // Pattern for JSON artifacts\n const jsonArtifactRegex = /```json\\s*\\s*\\n(.*?)\\n```/gs\n const jsonMatches = content.matchAll(jsonArtifactRegex)\n\n for (const match of jsonMatches) {\n const title = match[1].trim()\n const artifactContent = match[2].trim()\n\n const artifact: Artifact = {\n uuid: generateUUID(),\n type: 'json',\n title,\n content: artifactContent,\n language: 'json'\n }\n artifacts.push(artifact)\n }\n\n // Backward-compatible parsing for legacy executable markers.\n const executableArtifactRegex = /```(\\w+)?\\s*\\s*\\n(.*?)\\n```/gs\n const executableMatches = content.matchAll(executableArtifactRegex)\n\n for (const match of executableMatches) {\n const language = match[1] || 'javascript'\n const title = match[2].trim()\n const artifactContent = match[3].trim()\n\n // Skip if already processed as HTML, SVG, Mermaid, or JSON\n if (language === 'html' || language === 'svg' || language === 'mermaid' || language === 'json') {\n continue\n }\n\n const artifact: Artifact = {\n uuid: generateUUID(),\n type: 'code',\n title,\n content: artifactContent,\n language\n }\n artifacts.push(artifact)\n }\n\n // Pattern for general code artifacts (exclude html, svg, mermaid, json which are handled above)\n const codeArtifactRegex = /```(\\w+)?\\s*\\s*\\n(.*?)\\n```/gs\n const codeMatches = content.matchAll(codeArtifactRegex)\n\n for (const match of codeMatches) {\n const language = match[1] || 'text'\n const title = match[2].trim()\n const artifactContent = match[3].trim()\n\n // Skip if already processed as HTML, SVG, Mermaid, JSON, or executable\n if (language === 'html' || language === 'svg' || language === 'mermaid' || language === 'json') {\n continue\n }\n\n const artifact: Artifact = {\n uuid: generateUUID(),\n type: 'code',\n title,\n content: artifactContent,\n language\n }\n artifacts.push(artifact)\n }\n\n return artifacts\n}\n" }, { "path": "web/src/utils/crypto/index.ts", "content": "import CryptoJS from 'crypto-js'\n\nconst CryptoSecret = '__CRYPTO_SECRET__'\n\n// rome-ignore lint/suspicious/noExplicitAny: \nexport function enCrypto(data: any) {\n const str = JSON.stringify(data)\n return CryptoJS.AES.encrypt(str, CryptoSecret).toString()\n}\n\nexport function deCrypto(data: string) {\n const bytes = CryptoJS.AES.decrypt(data, CryptoSecret)\n const str = bytes.toString(CryptoJS.enc.Utf8)\n\n if (str)\n return JSON.parse(str)\n\n return null\n}\n" }, { "path": "web/src/utils/date.ts", "content": "import { DateTime } from 'luxon';\n\nexport function nowISO(): string {\n return DateTime.now().toISO() || ''\n}\n\nexport function getCurrentDate() {\n const now = DateTime.now()\n const formattedDate = now.toFormat('yyyy-MM-dd-HHmm-ss')\n return formattedDate\n}\n\n// 2025-03-05T12:48:11.990824Z 2025-02-26T08:58:48Z\nexport function displayLocaleDate(ts: string) {\n\n const dateObj = DateTime.fromISO(ts)\n\n const dateString = dateObj.toFormat('D t')\n\n return dateString\n}\n\nexport function formatYearMonth(date: Date): string {\n const year = date.getFullYear().toString()\n const month = (date.getMonth() + 1).toString().padStart(2, '0')\n return `${year}-${month}`\n}\n" }, { "path": "web/src/utils/download.ts", "content": "import { getCurrentDate } from './date'\n\nexport function genTempDownloadLink(imgUrl: string) {\n const tempLink = document.createElement('a')\n tempLink.style.display = 'none'\n tempLink.href = imgUrl\n // generate a file name, chat-shot-2021-08-01.png\n const ts = getCurrentDate()\n tempLink.setAttribute('download', `chat-shot-${ts}.png`)\n if (typeof tempLink.download === 'undefined')\n tempLink.setAttribute('target', '_blank')\n return tempLink\n}\n" }, { "path": "web/src/utils/errorHandler.ts", "content": "import { logger } from './logger'\nimport { useAuthStore } from '@/store'\nimport { \n showNotification, \n showErrorNotification, \n showWarningNotification, \n showSuccessNotification, \n showPersistentNotification,\n showEnhancedErrorNotification,\n showEnhancedWarningNotification,\n showEnhancedInfoNotification\n} from './notificationManager'\n\nexport interface ApiError {\n status: number\n message: string\n code?: string\n details?: any\n}\n\nexport interface ErrorHandlerOptions {\n logError?: boolean\n showToast?: boolean\n redirectOnError?: boolean\n retryCount?: number\n retryDelay?: number\n}\n\nclass ErrorHandler {\n private defaultOptions: ErrorHandlerOptions = {\n logError: true,\n showToast: true,\n redirectOnError: true,\n retryCount: 0,\n retryDelay: 1000,\n }\n\n private isNetworkError(error: any): boolean {\n return (\n !navigator.onLine ||\n error.message === 'Network Error' ||\n error.code === 'ECONNABORTED' ||\n error.code === 'ETIMEDOUT'\n )\n }\n\n private isAuthError(error: any): boolean {\n return error.status === 401 || error.status === 403\n }\n\n private isServerError(error: any): boolean {\n return error.status >= 500 && error.status < 600\n }\n\n private isClientError(error: any): boolean {\n return error.status >= 400 && error.status < 500\n }\n\n private extractErrorMessage(error: any): string {\n if (error.response?.data?.message) {\n return error.response.data.message\n }\n if (error.message) {\n return error.message\n }\n if (typeof error === 'string') {\n return error\n }\n return 'An unknown error occurred'\n }\n\n private logApiError(method: string, url: string, error: any, options: ErrorHandlerOptions): void {\n if (!options.logError) return\n\n const apiError: ApiError = {\n status: error.response?.status || 0,\n message: this.extractErrorMessage(error),\n code: error.code,\n details: error.response?.data,\n }\n\n logger.logApiError(method, url, apiError, apiError.status)\n }\n\n private async handleAuthError(error: any): Promise {\n const authStore = useAuthStore()\n \n try {\n logger.debug('Attempting token refresh for auth error', 'ErrorHandler')\n await authStore.refreshToken()\n } catch (refreshError) {\n logger.error('Token refresh failed, clearing auth state', 'ErrorHandler', refreshError)\n authStore.removeToken()\n authStore.removeExpiresIn()\n \n // Don't redirect immediately - let the auth store handle the UI state change\n // The login modal will appear automatically when authStore.isValid becomes false\n }\n }\n\n private async retryRequest(\n requestFn: () => Promise,\n retryCount: number,\n retryDelay: number\n ): Promise {\n let lastError: any\n\n for (let attempt = 1; attempt <= retryCount; attempt++) {\n try {\n return await requestFn()\n } catch (error) {\n lastError = error\n \n if (attempt === retryCount) {\n throw error\n }\n\n if (this.isAuthError(error)) {\n // Don't retry auth errors\n throw error\n }\n\n logger.debug(`Retrying request (attempt ${attempt + 1}/${retryCount})`, 'ErrorHandler', { error })\n \n // Exponential backoff\n const delay = retryDelay * Math.pow(2, attempt - 1)\n await new Promise(resolve => setTimeout(resolve, delay))\n }\n }\n\n throw lastError\n }\n\n async handleApiRequest(\n requestFn: () => Promise,\n method: string,\n url: string,\n options: ErrorHandlerOptions = {}\n ): Promise {\n const finalOptions = { ...this.defaultOptions, ...options }\n const startTime = Date.now()\n\n try {\n if (finalOptions.retryCount > 0) {\n const result = await this.retryRequest(requestFn, finalOptions.retryCount, finalOptions.retryDelay)\n const duration = Date.now() - startTime\n logger.logApiCall(method, url, 200, duration)\n return result\n } else {\n const result = await requestFn()\n const duration = Date.now() - startTime\n logger.logApiCall(method, url, 200, duration)\n return result\n }\n } catch (error: any) {\n this.logApiError(method, url, error, finalOptions)\n\n // Handle network errors\n if (this.isNetworkError(error)) {\n logger.warn('Network error detected', 'ErrorHandler', { error })\n throw {\n status: 0,\n message: 'Network error. Please check your internet connection.',\n originalError: error,\n }\n }\n\n // Handle authentication errors\n if (this.isAuthError(error)) {\n await this.handleAuthError(error)\n throw {\n status: error.status,\n message: 'Authentication failed. Please login again.',\n originalError: error,\n }\n }\n\n // Handle server errors\n if (this.isServerError(error)) {\n logger.error('Server error occurred', 'ErrorHandler', error)\n throw {\n status: error.status,\n message: 'Server error. Please try again later.',\n originalError: error,\n }\n }\n\n // Handle client errors\n if (this.isClientError(error)) {\n logger.warn('Client error occurred', 'ErrorHandler', error)\n throw {\n status: error.status,\n message: this.extractErrorMessage(error),\n originalError: error,\n }\n }\n\n // Handle unknown errors\n logger.error('Unknown error occurred', 'ErrorHandler', error)\n throw {\n status: 0,\n message: 'An unexpected error occurred.',\n originalError: error,\n }\n }\n }\n\n // Convenience method for GET requests\n async get(url: string, options?: ErrorHandlerOptions): Promise {\n return this.handleApiRequest(\n () => fetch(url, { method: 'GET' }),\n 'GET',\n url,\n options\n )\n }\n\n // Convenience method for POST requests\n async post(url: string, data?: any, options?: ErrorHandlerOptions): Promise {\n return this.handleApiRequest(\n () => fetch(url, {\n method: 'POST',\n headers: { 'Content-Type': 'application/json' },\n body: JSON.stringify(data),\n }),\n 'POST',\n url,\n options\n )\n }\n\n // Convenience method for PUT requests\n async put(url: string, data?: any, options?: ErrorHandlerOptions): Promise {\n return this.handleApiRequest(\n () => fetch(url, {\n method: 'PUT',\n headers: { 'Content-Type': 'application/json' },\n body: JSON.stringify(data),\n }),\n 'PUT',\n url,\n options\n )\n }\n\n // Convenience method for DELETE requests\n async delete(url: string, options?: ErrorHandlerOptions): Promise {\n return this.handleApiRequest(\n () => fetch(url, { method: 'DELETE' }),\n 'DELETE',\n url,\n options\n )\n }\n\n // Global error handler for unhandled promise rejections\n setupGlobalErrorHandler(): void {\n window.addEventListener('unhandledrejection', (event) => {\n logger.error('Unhandled promise rejection', 'GlobalErrorHandler', event.reason)\n \n // Prevent default behavior (logging to console)\n event.preventDefault()\n \n // You could also show a user-friendly error message here\n if (event.reason instanceof Error) {\n console.error('An unexpected error occurred:', event.reason.message)\n }\n })\n\n window.addEventListener('error', (event) => {\n logger.error('Global error occurred', 'GlobalErrorHandler', {\n message: event.message,\n filename: event.filename,\n lineno: event.lineno,\n colno: event.colno,\n error: event.error,\n })\n })\n }\n}\n\n// Export singleton instance\nexport const errorHandler = new ErrorHandler()\n\n// Export convenience functions\nexport const handleApiRequest = (\n requestFn: () => Promise,\n method: string,\n url: string,\n options?: ErrorHandlerOptions\n) => errorHandler.handleApiRequest(requestFn, method, url, options)\n\n// Default export\nexport default errorHandler" }, { "path": "web/src/utils/format/index.ts", "content": "/**\n * 转义 HTML 字符\n * @param source\n */\nexport function encodeHTML(source: string) {\n return source\n .replace(/&/g, '&')\n .replace(//g, '>')\n .replace(/\"/g, '"')\n .replace(/'/g, ''')\n}\n\n/**\n * 判断是否为代码块\n * @param text\n */\nexport function includeCode(text: string | null | undefined) {\n const regexp = /^(?:\\s{4}|\\t).+/gm\n return !!(text?.includes(' = ') || text?.match(regexp))\n}\n\n/**\n * 复制文本\n * @param options\n */\nexport function copyText(options: { text: string; origin?: boolean }) {\n const props = { origin: true, ...options }\n\n let input: HTMLInputElement | HTMLTextAreaElement\n\n if (props.origin)\n input = document.createElement('textarea')\n else\n input = document.createElement('input')\n\n input.setAttribute('readonly', 'readonly')\n input.value = props.text\n document.body.appendChild(input)\n input.select()\n if (document.execCommand('copy'))\n document.execCommand('copy')\n document.body.removeChild(input)\n}\n" }, { "path": "web/src/utils/is/index.ts", "content": "export function isNumber(value: T | unknown): value is number {\n return Object.prototype.toString.call(value) === '[object Number]'\n}\n\nexport function isString(value: T | unknown): value is string {\n return Object.prototype.toString.call(value) === '[object String]'\n}\n\nexport function isBoolean(value: T | unknown): value is boolean {\n return Object.prototype.toString.call(value) === '[object Boolean]'\n}\n\nexport function isNull(value: T | unknown): value is null {\n return Object.prototype.toString.call(value) === '[object Null]'\n}\n\nexport function isUndefined(value: T | unknown): value is undefined {\n return Object.prototype.toString.call(value) === '[object Undefined]'\n}\n\nexport function isObject(value: T | unknown): value is object {\n return Object.prototype.toString.call(value) === '[object Object]'\n}\n\nexport function isArray(value: T | unknown): value is T {\n return Object.prototype.toString.call(value) === '[object Array]'\n}\n\nexport function isFunction any | void | never>(value: T | unknown): value is T {\n return Object.prototype.toString.call(value) === '[object Function]'\n}\n\nexport function isDate(value: T | unknown): value is T {\n return Object.prototype.toString.call(value) === '[object Date]'\n}\n\nexport function isRegExp(value: T | unknown): value is T {\n return Object.prototype.toString.call(value) === '[object RegExp]'\n}\n\nexport function isPromise>(value: T | unknown): value is T {\n return Object.prototype.toString.call(value) === '[object Promise]'\n}\n\nexport function isSet>(value: T | unknown): value is T {\n return Object.prototype.toString.call(value) === '[object Set]'\n}\n\nexport function isMap>(value: T | unknown): value is T {\n return Object.prototype.toString.call(value) === '[object Map]'\n}\n\nexport function isFile(value: T | unknown): value is T {\n return Object.prototype.toString.call(value) === '[object File]'\n}\n\nexport const isASCII = (str: string) => /^[\\x00-\\x7F]*$/.test(str)\n" }, { "path": "web/src/utils/jwt.ts", "content": "import jwt_decode from 'jwt-decode'\n\nexport function isAdmin(token: string): boolean {\n if (token) {\n const decoded: { role: string } = jwt_decode(token)\n if (decoded && decoded.role === 'admin')\n return true\n }\n return false\n}\n" }, { "path": "web/src/utils/logger.ts", "content": "export enum LogLevel {\n DEBUG = 0,\n INFO = 1,\n WARN = 2,\n ERROR = 3,\n}\n\nexport interface LogEntry {\n level: LogLevel\n message: string\n timestamp: string\n context?: string\n data?: any\n}\n\nclass Logger {\n private level: LogLevel\n private isProduction: boolean\n private logs: LogEntry[] = []\n\n constructor() {\n this.level = this.getLogLevelFromEnv()\n this.isProduction = (import.meta as any).env?.VITE_ENV === 'production'\n }\n\n private getLogLevelFromEnv(): LogLevel {\n const envLevel = (import.meta as any).env?.VITE_LOG_LEVEL || 'info'\n switch (envLevel) {\n case 'debug': return LogLevel.DEBUG\n case 'info': return LogLevel.INFO\n case 'warn': return LogLevel.WARN\n case 'error': return LogLevel.ERROR\n default: return this.isProduction ? LogLevel.WARN : LogLevel.DEBUG\n }\n }\n\n private shouldLog(level: LogLevel): boolean {\n return level >= this.level\n }\n\n private createLogEntry(level: LogLevel, message: string, context?: string, data?: any): LogEntry {\n return {\n level,\n message,\n timestamp: new Date().toISOString(),\n context,\n data,\n }\n }\n\n private formatMessage(entry: LogEntry): string {\n const levelName = LogLevel[entry.level]\n const contextStr = entry.context ? `[${entry.context}] ` : ''\n const dataStr = entry.data ? ` ${JSON.stringify(entry.data)}` : ''\n return `${entry.timestamp} [${levelName}] ${contextStr}${entry.message}${dataStr}`\n }\n\n private log(level: LogLevel, message: string, context?: string, data?: any): void {\n if (!this.shouldLog(level)) {\n return\n }\n\n const entry = this.createLogEntry(level, message, context, data)\n this.logs.push(entry)\n\n // Only log to console in development or for errors/warnings\n if (!this.isProduction || level >= LogLevel.ERROR) {\n const formattedMessage = this.formatMessage(entry)\n \n switch (level) {\n case LogLevel.DEBUG:\n console.debug(formattedMessage)\n break\n case LogLevel.INFO:\n console.info(formattedMessage)\n break\n case LogLevel.WARN:\n console.warn(formattedMessage)\n break\n case LogLevel.ERROR:\n console.error(formattedMessage)\n break\n }\n }\n }\n\n // Public logging methods\n debug(message: string, context?: string, data?: any): void {\n this.log(LogLevel.DEBUG, message, context, data)\n }\n\n info(message: string, context?: string, data?: any): void {\n this.log(LogLevel.INFO, message, context, data)\n }\n\n warn(message: string, context?: string, data?: any): void {\n this.log(LogLevel.WARN, message, context, data)\n }\n\n error(message: string, context?: string, data?: any): void {\n this.log(LogLevel.ERROR, message, context, data)\n }\n\n // Specialized logging methods for common scenarios\n logApiCall(method: string, url: string, status?: number, duration?: number): void {\n this.debug(`API ${method} ${url}`, 'API', { method, url, status, duration })\n }\n\n logApiError(method: string, url: string, error: any, status?: number): void {\n this.error(`API Error ${method} ${url}`, 'API', { method, url, error, status })\n }\n\n logStoreAction(action: string, store: string, data?: any): void {\n this.debug(`Store action: ${action}`, store, data)\n }\n\n logPerformance(metric: string, value: number, unit: string = 'ms'): void {\n this.debug(`Performance: ${metric} = ${value}${unit}`, 'Performance', { metric, value, unit })\n }\n\n logUserAction(action: string, details?: any): void {\n this.info(`User action: ${action}`, 'User', details)\n }\n\n // Get logs for debugging\n getLogs(level?: LogLevel): LogEntry[] {\n if (level !== undefined) {\n return this.logs.filter(log => log.level >= level)\n }\n return [...this.logs]\n }\n\n // Clear logs\n clearLogs(): void {\n this.logs = []\n }\n\n // Set log level dynamically\n setLevel(level: LogLevel): void {\n this.level = level\n }\n\n // Export logs for debugging\n exportLogs(): string {\n return JSON.stringify(this.logs, null, 2)\n }\n}\n\n// Export singleton instance\nexport const logger = new Logger()\n\n// Export convenience functions for direct use\nexport const debug = (message: string, context?: string, data?: any) => logger.debug(message, context, data)\nexport const info = (message: string, context?: string, data?: any) => logger.info(message, context, data)\nexport const warn = (message: string, context?: string, data?: any) => logger.warn(message, context, data)\nexport const error = (message: string, context?: string, data?: any) => logger.error(message, context, data)\n\n// Default export\nexport default logger" }, { "path": "web/src/utils/notificationManager.ts", "content": "import { ref, computed, h } from 'vue'\nimport { useMessage } from 'naive-ui'\nimport EnhancedNotification from '@/components/common/EnhancedNotification.vue'\n\ninterface NotificationOptions {\n title?: string\n message: string\n type?: 'success' | 'error' | 'warning' | 'info'\n duration?: number\n action?: {\n text: string\n onClick: () => void\n }\n persistent?: boolean\n closable?: boolean\n enhanced?: boolean // New option to use enhanced notifications\n}\n\ninterface QueuedNotification {\n id: string\n options: NotificationOptions\n timestamp: Date\n}\n\nclass NotificationManager {\n private queue = ref([])\n private activeNotifications = ref>(new Set())\n private messageInstance: any = null\n private maxConcurrent = 3\n private queueEnabled = true\n\n setMessageInstance(instance: any) {\n this.messageInstance = instance\n }\n\n private generateId(): string {\n return `notification_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`\n }\n\n private canShowNotification(): boolean {\n return this.activeNotifications.value.size < this.maxConcurrent\n }\n\n private showNotification(notification: QueuedNotification) {\n if (!this.messageInstance) return\n\n const { id, options } = notification\n this.activeNotifications.value.add(id)\n\n const showFn = this.messageInstance[options.type || 'info']\n const notificationOptions: any = {\n duration: options.persistent ? 0 : (options.duration || 3000),\n closable: options.closable !== false,\n keepAliveOnHover: true,\n onLeave: () => {\n this.activeNotifications.value.delete(id)\n this.processQueue()\n }\n }\n\n if (options.action) {\n notificationOptions.action = options.action\n }\n\n try {\n // Use enhanced notification if requested\n if (options.enhanced) {\n const content = h(EnhancedNotification, {\n type: options.type || 'info',\n title: options.title,\n content: options.message,\n closable: options.closable !== false,\n action: options.action,\n onClose: () => {\n this.activeNotifications.value.delete(id)\n this.processQueue()\n }\n })\n \n showFn(content, {\n ...notificationOptions,\n closable: false // Let the component handle closing\n })\n } else {\n showFn(options.message, notificationOptions)\n }\n } catch (error) {\n console.error('Failed to show notification:', error)\n this.activeNotifications.value.delete(id)\n this.processQueue()\n }\n }\n\n private processQueue() {\n if (!this.queueEnabled || !this.canShowNotification()) return\n\n const nextNotification = this.queue.value.shift()\n if (nextNotification) {\n this.showNotification(nextNotification)\n }\n }\n\n show(options: NotificationOptions): string {\n const id = this.generateId()\n const notification: QueuedNotification = {\n id,\n options,\n timestamp: new Date()\n }\n\n if (this.canShowNotification()) {\n this.showNotification(notification)\n } else {\n this.queue.value.push(notification)\n }\n\n return id\n }\n\n success(message: string, options: Omit = {}): string {\n return this.show({ message, type: 'success', ...options })\n }\n\n error(message: string, options: Omit = {}): string {\n return this.show({ message, type: 'error', ...options })\n }\n\n warning(message: string, options: Omit = {}): string {\n return this.show({ message, type: 'warning', ...options })\n }\n\n info(message: string, options: Omit = {}): string {\n return this.show({ message, type: 'info', ...options })\n }\n\n // Enhanced notification methods with better visual hierarchy\n enhancedSuccess(title: string, message: string, options: Omit = {}): string {\n return this.show({ title, message, type: 'success', enhanced: true, ...options })\n }\n\n enhancedError(title: string, message: string, options: Omit = {}): string {\n return this.show({ title, message, type: 'error', enhanced: true, ...options })\n }\n\n enhancedWarning(title: string, message: string, options: Omit = {}): string {\n return this.show({ title, message, type: 'warning', enhanced: true, ...options })\n }\n\n enhancedInfo(title: string, message: string, options: Omit = {}): string {\n return this.show({ title, message, type: 'info', enhanced: true, ...options })\n }\n\n persistent(message: string, type: 'error' | 'warning' | 'info' = 'error', action?: { text: string; onClick: () => void }): string {\n return this.show({\n message,\n type,\n persistent: true,\n action\n })\n }\n\n remove(id: string): void {\n this.queue.value = this.queue.value.filter(n => n.id !== id)\n this.activeNotifications.value.delete(id)\n }\n\n clear(): void {\n this.queue.value = []\n this.activeNotifications.value.clear()\n if (this.messageInstance) {\n try {\n this.messageInstance.destroyAll()\n } catch (error) {\n console.warn('Failed to clear notifications:', error)\n }\n }\n }\n\n getStats() {\n return {\n queued: this.queue.value.length,\n active: this.activeNotifications.value.size,\n maxConcurrent: this.maxConcurrent\n }\n }\n\n setMaxConcurrent(max: number): void {\n this.maxConcurrent = max\n this.processQueue()\n }\n\n enableQueue(): void {\n this.queueEnabled = true\n this.processQueue()\n }\n\n disableQueue(): void {\n this.queueEnabled = false\n }\n}\n\n// Export singleton instance\nexport const notificationManager = new NotificationManager()\n\n// Vue composable for easy usage in components\nexport function useNotification() {\n const message = useMessage()\n \n // Initialize message instance if not already set\n if (!notificationManager['messageInstance']) {\n notificationManager.setMessageInstance(message)\n }\n\n return {\n show: notificationManager.show.bind(notificationManager),\n success: notificationManager.success.bind(notificationManager),\n error: notificationManager.error.bind(notificationManager),\n warning: notificationManager.warning.bind(notificationManager),\n info: notificationManager.info.bind(notificationManager),\n enhancedSuccess: notificationManager.enhancedSuccess.bind(notificationManager),\n enhancedError: notificationManager.enhancedError.bind(notificationManager),\n enhancedWarning: notificationManager.enhancedWarning.bind(notificationManager),\n enhancedInfo: notificationManager.enhancedInfo.bind(notificationManager),\n persistent: notificationManager.persistent.bind(notificationManager),\n clear: notificationManager.clear.bind(notificationManager),\n stats: computed(() => notificationManager.getStats())\n }\n}\n\n// Global notification functions for non-Vue contexts\nexport function showNotification(options: NotificationOptions): string {\n return notificationManager.show(options)\n}\n\nexport function showSuccessNotification(message: string, options?: Omit): string {\n return notificationManager.success(message, options)\n}\n\nexport function showErrorNotification(message: string, options?: Omit): string {\n return notificationManager.error(message, options)\n}\n\nexport function showWarningNotification(message: string, options?: Omit): string {\n return notificationManager.warning(message, options)\n}\n\nexport function showInfoNotification(message: string, options?: Omit): string {\n return notificationManager.info(message, options)\n}\n\nexport function showPersistentNotification(message: string, type: 'error' | 'warning' | 'info' = 'error', action?: { text: string; onClick: () => void }): string {\n return notificationManager.persistent(message, type, action)\n}\n\n// Enhanced notification functions with better visual hierarchy\nexport function showEnhancedSuccessNotification(title: string, message: string, options?: Omit): string {\n return notificationManager.enhancedSuccess(title, message, options)\n}\n\nexport function showEnhancedErrorNotification(title: string, message: string, options?: Omit): string {\n return notificationManager.enhancedError(title, message, options)\n}\n\nexport function showEnhancedWarningNotification(title: string, message: string, options?: Omit): string {\n return notificationManager.enhancedWarning(title, message, options)\n}\n\nexport function showEnhancedInfoNotification(title: string, message: string, options?: Omit): string {\n return notificationManager.enhancedInfo(title, message, options)\n}\n\nexport function clearAllNotifications(): void {\n notificationManager.clear()\n}" }, { "path": "web/src/utils/prompt.ts", "content": "" }, { "path": "web/src/utils/rand.ts", "content": "/**\n * Generates a random string of length n\n * @param n Length of the random string\n */\nexport function generateRandomString(n: number): string {\n // Array of possible characters\n const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'\n // Random string to be returned\n let randomString = ''\n\n // Loop n times to generate a string of length n\n for (let i = 0; i < n; i++) {\n // Get a random character from the array of characters\n const randomCharacter = characters.charAt(Math.floor(Math.random() * characters.length))\n // Append the random character to the random string\n randomString += randomCharacter\n }\n return randomString\n}\n" }, { "path": "web/src/utils/request/axios.ts", "content": "import axios, { type AxiosResponse } from 'axios'\nimport { useAuthStore } from '@/store'\nimport { logger } from '@/utils/logger'\n\nconst service = axios.create({\n baseURL: \"/api\",\n withCredentials: true, // Include httpOnly cookies (for refresh token)\n})\n\nservice.interceptors.request.use(\n async (config) => {\n const authStore = useAuthStore()\n\n // Skip token validation for authentication endpoints\n const authEndpoints = ['/login', '/signup', '/logout', '/auth/refresh']\n if (authEndpoints.some(endpoint => config.url?.includes(endpoint))) {\n return config\n }\n\n // Wait for auth initialization to complete before making API calls\n if (authStore.isInitializing) {\n logger.debug('Waiting for auth initialization to complete', 'Axios', { url: config.url })\n await authStore.waitForInitialization()\n\n if (authStore.isInitializing) {\n logger.warn('Auth initialization still in progress after timeout', 'Axios', { url: config.url })\n } else {\n logger.debug('Auth initialization completed', 'Axios', { url: config.url })\n }\n }\n\n // Check if token is expired before making request\n if (!authStore.isValid) {\n logger.debug('Token is expired or invalid, attempting refresh', 'Axios', { url: config.url })\n try {\n await authStore.refreshToken()\n // Check again after refresh\n if (!authStore.isValid) {\n logger.warn('Token still invalid after refresh attempt', 'Axios')\n return Promise.reject(new Error('Authentication required'))\n }\n } catch (error) {\n logger.error('Token refresh failed in request interceptor', 'Axios', error)\n return Promise.reject(new Error('Authentication required'))\n }\n } \n // Check if token needs refresh (expires within 5 minutes)\n else if (authStore.needsRefresh && !authStore.isRefreshing) {\n logger.debug('Token needs refresh, refreshing proactively', 'Axios', { url: config.url })\n try {\n await authStore.refreshToken()\n } catch (error) {\n logger.error('Proactive token refresh failed', 'Axios', error)\n // Continue with existing token if proactive refresh fails\n }\n }\n\n // Add access token to Authorization header\n const token = authStore.getToken\n if (token) {\n config.headers.Authorization = `Bearer ${token}`\n }\n\n return config\n },\n (error) => {\n return Promise.reject(error.response)\n },\n)\n\nservice.interceptors.response.use(\n (response: AxiosResponse): AxiosResponse => {\n if (response.status === 200 || response.status === 201 || response.status === 204)\n return response\n throw new Error(response.status.toString())\n },\n async (error) => {\n const authStore = useAuthStore()\n\n logger.logApiError(error.config?.method || 'unknown', error.config?.url || 'unknown', error, error.response?.status)\n\n // Handle 401 errors with automatic token refresh\n if (error.response?.status === 401 && !error.config?.url?.includes('/auth/')) {\n logger.debug('Handling 401 error, attempting token refresh', 'Axios')\n \n // Prevent infinite retry loops\n if (error.config._retryCount >= 1) {\n logger.warn('Already retried once, clearing auth state', 'Axios')\n authStore.removeToken()\n authStore.removeExpiresIn()\n return Promise.reject(new Error('Authentication failed after retry'))\n }\n \n try {\n await authStore.refreshToken()\n // Check if refresh was successful\n if (!authStore.isValid) {\n logger.warn('Token invalid after refresh attempt', 'Axios')\n authStore.removeToken()\n authStore.removeExpiresIn()\n return Promise.reject(new Error('Authentication failed'))\n }\n \n // Retry the original request with new token\n const token = authStore.getToken\n if (token) {\n logger.debug('Retrying request with new token', 'Axios')\n error.config.headers.Authorization = `Bearer ${token}`\n error.config._retryCount = (error.config._retryCount || 0) + 1\n return service.request(error.config)\n }\n } catch (refreshError) {\n // Refresh failed - user needs to login again\n logger.warn('Token refresh failed, clearing auth state', 'Axios', refreshError)\n authStore.removeToken()\n authStore.removeExpiresIn()\n return Promise.reject(new Error('Authentication required'))\n }\n }\n\n return Promise.reject(error)\n },\n)\n\nexport default service\n" }, { "path": "web/src/utils/request/index.ts", "content": "import type { AxiosProgressEvent, AxiosResponse, GenericAbortSignal } from 'axios'\nimport request from './axios'\nimport { useAuthStore } from '@/store'\n\nexport interface HttpOption {\n url: string\n // rome-ignore lint/suspicious/noExplicitAny: \n data?: any\n method?: string\n // rome-ignore lint/suspicious/noExplicitAny: \n headers?: any\n onDownloadProgress?: (progressEvent: AxiosProgressEvent) => void\n signal?: GenericAbortSignal\n beforeRequest?: () => void\n afterRequest?: () => void\n}\n\nexport interface Response {\n data: T\n message: string | null\n status: string\n}\n\nfunction http(\n { url, data, method, headers, onDownloadProgress, signal, beforeRequest, afterRequest }: HttpOption,\n) {\n const successHandler = (res: AxiosResponse>) => {\n const authStore = useAuthStore()\n\n if (res.data.status === 'Success' || typeof res.data === 'string')\n return res.data\n\n if (res.data.status === 'Unauthorized') {\n authStore.removeToken()\n window.location.reload()\n }\n\n return Promise.reject(res.data)\n }\n\n const failHandler = (error: any) => {\n afterRequest?.()\n \n // Enhanced error handling with more detailed error information\n let errorMessage = 'An unexpected error occurred'\n let errorCode = 'UNKNOWN_ERROR'\n \n if (error?.response?.data) {\n errorMessage = error.response.data.message || errorMessage\n errorCode = error.response.data.code || errorCode\n } else if (error?.message) {\n errorMessage = error.message\n } else if (typeof error === 'string') {\n errorMessage = error\n }\n \n // Create enhanced error object with proper typing\n interface EnhancedError extends Error {\n code?: string | number\n status?: number\n originalError?: any\n }\n \n const enhancedError = new Error(errorMessage) as EnhancedError\n enhancedError.name = errorCode\n enhancedError.code = errorCode\n enhancedError.status = error?.response?.status || 0\n enhancedError.originalError = error\n \n throw enhancedError\n }\n\n beforeRequest?.()\n\n method = method || 'GET'\n\n const params = Object.assign(typeof data === 'function' ? data() : data ?? {}, {})\n\n return method === 'GET'\n ? request.get(url, { params, signal, onDownloadProgress }).then(successHandler, failHandler)\n : request.post(url, params, { headers, signal, onDownloadProgress }).then(successHandler, failHandler)\n}\n\nexport function get(\n { url, data, method = 'GET', onDownloadProgress, signal, beforeRequest, afterRequest }: HttpOption,\n): Promise> {\n return http({\n url,\n method,\n data,\n onDownloadProgress,\n signal,\n beforeRequest,\n afterRequest,\n })\n}\n\nexport function post(\n { url, data, method = 'POST', headers, onDownloadProgress, signal, beforeRequest, afterRequest }: HttpOption,\n): Promise> {\n return http({\n url,\n method,\n data,\n headers,\n onDownloadProgress,\n signal,\n beforeRequest,\n afterRequest,\n })\n}\n\nexport default post\n" }, { "path": "web/src/utils/sanitize.ts", "content": "const BLOCKED_TAGS = new Set(['script', 'iframe', 'object', 'embed'])\nconst BLOCKED_SVG_TAGS = new Set(['script', 'foreignobject'])\n\nconst stripUnsafeAttributes = (element: Element) => {\n for (const attr of Array.from(element.attributes)) {\n const name = attr.name.toLowerCase()\n const value = attr.value.trim().toLowerCase()\n\n if (name.startsWith('on')) {\n element.removeAttribute(attr.name)\n continue\n }\n\n if ((name === 'href' || name === 'src' || name === 'xlink:href') && value.startsWith('javascript:')) {\n element.removeAttribute(attr.name)\n }\n }\n}\n\nconst sanitizeElementTree = (root: Element, blockedTags: Set) => {\n const ownerDocument = root.ownerDocument || document\n const walker = ownerDocument.createTreeWalker(root, NodeFilter.SHOW_ELEMENT)\n const toRemove: Element[] = []\n\n let current = walker.currentNode as Element\n while (current) {\n const tagName = current.tagName.toLowerCase()\n if (blockedTags.has(tagName)) {\n toRemove.push(current)\n } else {\n stripUnsafeAttributes(current)\n }\n current = walker.nextNode() as Element\n }\n\n toRemove.forEach(node => node.remove())\n}\n\nexport const sanitizeHtml = (input: string): string => {\n try {\n const parser = new DOMParser()\n const doc = parser.parseFromString(input, 'text/html')\n sanitizeElementTree(doc.body, BLOCKED_TAGS)\n return doc.body.innerHTML\n } catch {\n return ''\n }\n}\n\nexport const sanitizeSvg = (input: string): string => {\n try {\n const parser = new DOMParser()\n const doc = parser.parseFromString(input, 'image/svg+xml')\n const root = doc.documentElement\n if (!root) return ''\n sanitizeElementTree(root, BLOCKED_SVG_TAGS)\n return new XMLSerializer().serializeToString(root)\n } catch {\n return ''\n }\n}\n" }, { "path": "web/src/utils/storage/index.ts", "content": "export * from './local'\n" }, { "path": "web/src/utils/storage/local.ts", "content": "import { deCrypto, enCrypto } from '../crypto'\n\ninterface StorageData {\n data: T\n expire: number | null\n}\n\nexport function createLocalStorage(options?: { expire?: number | null; crypto?: boolean }) {\n const DEFAULT_CACHE_TIME = 60 * 60 * 24 * 7\n\n const { expire, crypto } = Object.assign(\n {\n expire: DEFAULT_CACHE_TIME,\n crypto: true,\n },\n options,\n )\n\n function set(key: string, data: T) {\n const storageData: StorageData = {\n data,\n expire: expire !== null ? new Date().getTime() + expire * 1000 : null,\n }\n\n const json = crypto ? enCrypto(storageData) : JSON.stringify(storageData)\n window.localStorage.setItem(key, json)\n }\n\n function get(key: string) {\n const json = window.localStorage.getItem(key)\n if (json) {\n // rome-ignore lint/suspicious/noExplicitAny: \n let storageData: StorageData | null = null\n\n try {\n storageData = crypto ? deCrypto(json) : JSON.parse(json)\n }\n catch {\n // Prevent failure\n }\n\n if (storageData) {\n const { data, expire } = storageData\n if (expire === null || expire >= Date.now())\n return data\n }\n\n remove(key)\n return null\n }\n }\n\n function remove(key: string) {\n window.localStorage.removeItem(key)\n }\n\n function clear() {\n window.localStorage.clear()\n }\n\n return {\n set,\n get,\n remove,\n clear,\n }\n}\n\nexport const ls = createLocalStorage()\n\nexport const ss = createLocalStorage({ expire: null, crypto: false })\n" }, { "path": "web/src/utils/string.ts", "content": "export function extractStreamingData(streamResponse: string): string {\n const DATA_MARKER = 'data:'\n const SSE_DATA_MARKER = '\\n\\ndata:'\n \n // Handle single data segment at response start (most common after buffer split)\n if (streamResponse.startsWith(DATA_MARKER)) {\n return streamResponse.slice(DATA_MARKER.length).trim()\n }\n\n // Handle Server-Sent Events with multiple data segments - extract the last one\n const lastSSEDataPosition = streamResponse.lastIndexOf(SSE_DATA_MARKER)\n if (lastSSEDataPosition === -1) {\n return streamResponse.trim() // No SSE format detected, return original\n }\n\n // Extract data after the last SSE marker\n const dataStartPosition = lastSSEDataPosition + SSE_DATA_MARKER.length\n return streamResponse.slice(dataStartPosition).trim()\n}\n\nexport function escapeDollarNumber(text: string) {\n let escapedText = ''\n for (let i = 0; i < text.length; i += 1) {\n let char = text[i]\n const nextChar = text[i + 1] || ' '\n if (char === '$' && nextChar >= '0' && nextChar <= '9')\n char = '\\\\$'\n escapedText += char\n }\n return escapedText\n }\nexport function escapeBrackets(text: string) {\n const pattern = /(```[\\s\\S]*?```|`.*?`)|\\\\\\[([\\s\\S]*?[^\\\\])\\\\\\]|\\\\\\((.*?)\\\\\\)/g\n return text.replace(pattern, (match, codeBlock, squareBracket, roundBracket) => {\n if (codeBlock)\n return codeBlock\n else if (squareBracket)\n return `$$${squareBracket}$$`\n else if (roundBracket)\n return `$${roundBracket}$`\n return match\n })\n }" }, { "path": "web/src/utils/tooling.ts", "content": "export type ToolCall = {\n name: string\n arguments: Record\n}\n\nconst toolCallRegex = /```tool_call\\s*([\\s\\S]*?)```/gi\nconst toolResultRegex = /```tool_result\\s*([\\s\\S]*?)```/gi\n\nexport const extractToolCalls = (text: string) => {\n const calls: ToolCall[] = []\n let cleanedText = text\n\n cleanedText = cleanedText.replace(toolCallRegex, (_, jsonPayload) => {\n try {\n const parsed = JSON.parse(jsonPayload.trim())\n if (parsed && typeof parsed === 'object' && parsed.name) {\n calls.push(parsed as ToolCall)\n }\n } catch {\n // Ignore malformed tool calls.\n }\n return ''\n })\n\n return {\n calls,\n cleanedText: cleanedText.trim(),\n }\n}\n\nexport const stripToolBlocks = (text: string) => {\n return text.replace(toolCallRegex, '').replace(toolResultRegex, '').trim()\n}\n\nexport const isToolResultMessage = (text: string) => {\n const trimmed = text.trim()\n return trimmed.startsWith('[[TOOL_RESULT]]') || toolResultRegex.test(trimmed)\n}\n" }, { "path": "web/src/utils/workspaceUrls.ts", "content": "/**\n * Utility functions for generating and handling workspace-aware URLs\n */\n\n// Get base URL for the application\nfunction getBaseUrl(): string {\n return `${window.location.protocol}//${window.location.host}`\n}\n\n// Generate shareable URL for a session within a workspace\nexport function generateSessionUrl(sessionUuid: string, workspaceUuid?: string): string {\n const baseUrl = getBaseUrl()\n \n if (workspaceUuid) {\n return `${baseUrl}/#/workspace/${workspaceUuid}/chat/${sessionUuid}`\n }\n \n return `${baseUrl}/#/chat/${sessionUuid}`\n}\n\n// Generate shareable URL for a workspace\nexport function generateWorkspaceUrl(workspaceUuid: string): string {\n const baseUrl = getBaseUrl()\n return `${baseUrl}/#/workspace/${workspaceUuid}/chat`\n}\n\n// Extract workspace and session UUIDs from a URL\nexport function parseWorkspaceUrl(url: string): { workspaceUuid?: string; sessionUuid?: string } {\n try {\n const urlObj = new URL(url)\n const hash = urlObj.hash.substring(1) // Remove the # character\n \n // Match patterns: /workspace/:workspaceUuid/chat/:sessionUuid? or /chat/:sessionUuid?\n const workspaceMatch = hash.match(/^\\/workspace\\/([^\\/]+)\\/chat\\/?([^\\/]+)?/)\n const chatMatch = hash.match(/^\\/chat\\/?([^\\/]+)?/)\n \n if (workspaceMatch) {\n return {\n workspaceUuid: workspaceMatch[1],\n sessionUuid: workspaceMatch[2]\n }\n }\n \n if (chatMatch) {\n return {\n sessionUuid: chatMatch[1]\n }\n }\n \n return {}\n } catch (error) {\n console.error('Error parsing workspace URL:', error)\n return {}\n }\n}\n\n// Check if a URL is a valid workspace URL\nexport function isValidWorkspaceUrl(url: string): boolean {\n const parsed = parseWorkspaceUrl(url)\n return parsed.workspaceUuid !== undefined || parsed.sessionUuid !== undefined\n}\n\n// Copy URL to clipboard with error handling\nexport async function copyUrlToClipboard(url: string): Promise {\n try {\n if (navigator.clipboard && window.isSecureContext) {\n await navigator.clipboard.writeText(url)\n return true\n } else {\n // Fallback for older browsers or non-HTTPS\n const textArea = document.createElement('textarea')\n textArea.value = url\n textArea.style.position = 'fixed'\n textArea.style.left = '-999999px'\n textArea.style.top = '-999999px'\n document.body.appendChild(textArea)\n textArea.focus()\n textArea.select()\n \n const success = document.execCommand('copy')\n document.body.removeChild(textArea)\n return success\n }\n } catch (error) {\n console.error('Failed to copy URL to clipboard:', error)\n return false\n }\n}\n\n// Generate QR code data URL for sharing (requires qr-code library)\nexport function generateQRCodeUrl(url: string): string {\n // This would require a QR code library like 'qrcode'\n // For now, return a placeholder\n return `https://api.qrserver.com/v1/create-qr-code/?size=200x200&data=${encodeURIComponent(url)}`\n}\n\n// Validate workspace UUID format\nexport function isValidWorkspaceUuid(uuid: string): boolean {\n const uuidRegex = /^[0-9a-f]{8}-[0-9a-f]{4}-[1-5][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i\n return uuidRegex.test(uuid)\n}\n\n// Validate session UUID format\nexport function isValidSessionUuid(uuid: string): boolean {\n return isValidWorkspaceUuid(uuid) // Same format\n}\n\n// Create a URL-safe workspace name for potential future slug-based URLs\nexport function createWorkspaceSlug(name: string): string {\n return name\n .toLowerCase()\n .replace(/[^\\w\\s-]/g, '') // Remove special characters\n .replace(/[\\s_-]+/g, '-') // Replace spaces and underscores with hyphens\n .replace(/^-+|-+$/g, '') // Remove leading/trailing hyphens\n}\n\n// Social sharing URLs\nexport const socialShareUrls = {\n twitter: (url: string, text: string = 'Check out this chat workspace') => \n `https://twitter.com/intent/tweet?url=${encodeURIComponent(url)}&text=${encodeURIComponent(text)}`,\n \n facebook: (url: string) => \n `https://www.facebook.com/sharer/sharer.php?u=${encodeURIComponent(url)}`,\n \n linkedin: (url: string, title: string = 'Chat Workspace') => \n `https://www.linkedin.com/sharing/share-offsite/?url=${encodeURIComponent(url)}&title=${encodeURIComponent(title)}`,\n \n email: (url: string, subject: string = 'Chat Workspace', body: string = 'Check out this workspace:') => \n `mailto:?subject=${encodeURIComponent(subject)}&body=${encodeURIComponent(body)}%20${encodeURIComponent(url)}`\n}" }, { "path": "web/src/views/admin/index.vue", "content": "\n\n\n" }, { "path": "web/src/views/admin/model/AddModelForm.vue", "content": "\n\n\n" }, { "path": "web/src/views/admin/model/index.vue", "content": "\n\n\n" }, { "path": "web/src/views/admin/modelRateLimit/addChatModelForm.vue", "content": "\n\n\n" }, { "path": "web/src/views/admin/modelRateLimit/index.vue", "content": "\n\n\n" }, { "path": "web/src/views/admin/user/index.vue", "content": "\n\n\n" }, { "path": "web/src/views/bot/all.vue", "content": "\n\n\n" }, { "path": "web/src/views/bot/components/AnswerHistory.vue", "content": "\n\n\n" }, { "path": "web/src/views/bot/components/Message/index.vue", "content": "\n\n" }, { "path": "web/src/views/bot/components/Message/style.less", "content": ".markdown-body {\n\tbackground-color: transparent;\n\tfont-size: 14px;\n\n\tp {\n\t\twhite-space: pre-wrap;\n\t}\n\n\tol {\n\t\tlist-style-type: decimal;\n\t}\n\n\tul {\n\t\tlist-style-type: disc;\n\t}\n\n\tpre code,\n\tpre tt {\n\t\tline-height: 1.65;\n\t}\n\n\t.highlight pre,\n\tpre {\n\t\tbackground-color: #fff;\n\t}\n\n\tcode.hljs {\n\t\tpadding: 0;\n\t}\n\n\t.code-block {\n\t\t&-wrapper {\n\t\t\tposition: relative;\n\t\t\tpadding-top: 24px;\n\t\t}\n\n\t\t&-header {\n\t\t\tposition: absolute;\n\t\t\ttop: 5px;\n\t\t\tright: 0;\n\t\t\twidth: 100%;\n\t\t\tpadding: 0 1rem;\n\t\t\tdisplay: flex;\n\t\t\tjustify-content: flex-end;\n\t\t\talign-items: center;\n\t\t\tcolor: #b3b3b3;\n\n\t\t\t&__copy{\n\t\t\t\tcursor: pointer;\n\t\t\t\tmargin-left: 0.5rem;\n\t\t\t\tuser-select: none;\n\t\t\t\t&:hover {\n\t\t\t\t\tcolor: #65a665;\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n}\n\nhtml.dark {\n\n\t.highlight pre,\n\tpre {\n\t\tbackground-color: #282c34;\n\t}\n}\n" }, { "path": "web/src/views/bot/page.vue", "content": "\n\n\n" }, { "path": "web/src/views/chat/components/ArtifactGallery.vue", "content": "\n\n\n\n\n" }, { "path": "web/src/views/chat/components/AudioPlayer/index.vue", "content": "\n\n\n\n \n" }, { "path": "web/src/views/chat/components/Conversation.vue", "content": "\n\n\n\n\n" }, { "path": "web/src/views/chat/components/HeaderMobile/index.vue", "content": "\n\n\n" }, { "path": "web/src/views/chat/components/JumpToBottom.vue", "content": "\n\n\n\n\n" }, { "path": "web/src/views/chat/components/Message/ArtifactContent.vue", "content": "