fix: use cjs extension for commonjs files

* chore(.gitignore): add auth.json to gitignore
test(landing.spec.js): remove commented out code and add check for landing page title
test(login.spec.js): add test for login page title
feat(package.json): add e2e:auth script to generate auth.json storage file for e2e tests

* test(landing.spec.js): add beforeEach hook to create a new browser context with auth.json storage state
test(landing.spec.js): change test name from 'landing page' to 'Landing title'
fix(package.json): change e2e:auth script to save auth.json in e2e directory

.dockerignore

@@ -1,2 +1,3 @@
 **/node_modules
-**/.env
+api/.env
+client/dist/images

.eslintrc.cjs

@@ -0,0 +1,114 @@
+module.exports = {
+  env: {
+    browser: true,
+    es2021: true,
+    node: true,
+    commonjs: true,
+    es6: true
+  },
+  extends: [
+    'eslint:recommended',
+    'plugin:react/recommended',
+    'plugin:react-hooks/recommended',
+    'prettier'
+  ],
+  parser: '@typescript-eslint/parser',
+  parserOptions: {
+    ecmaVersion: 'latest',
+    sourceType: 'module',
+    ecmaFeatures: {
+      jsx: true
+    }
+  },
+  plugins: ['react', 'react-hooks', '@typescript-eslint'],
+  rules: {
+    'react/react-in-jsx-scope': 'off',
+    '@typescript-eslint/ban-ts-comment': ['error', { 'ts-ignore': 'allow-with-description' }],
+    indent: ['error', 2, { SwitchCase: 1 }],
+    'max-len': [
+      'error',
+      {
+        code: 150,
+        ignoreStrings: true,
+        ignoreTemplateLiterals: true,
+        ignoreComments: true
+      }
+    ],
+    'linebreak-style': 0,
+    // "arrow-parens": [2, "as-needed", { requireForBlockBody: true }],
+    // 'no-plusplus': ['error', { allowForLoopAfterthoughts: true }],
+    'no-console': 'off',
+    'import/extensions': 'off',
+    'no-use-before-define': [
+      'error',
+      {
+        functions: false
+      }
+    ],
+    'no-promise-executor-return': 'off',
+    'no-param-reassign': 'off',
+    'no-continue': 'off',
+    'no-restricted-syntax': 'off',
+    'react/prop-types': ['off'],
+    'react/display-name': ['off']
+  },
+  overrides: [
+    {
+      files: ['**/*.ts', '**/*.tsx'],
+      rules: {
+        'no-unused-vars': 'off', // off because it conflicts with '@typescript-eslint/no-unused-vars'
+        'react/display-name': 'off',
+        '@typescript-eslint/no-unused-vars': 'warn'
+      }
+    },
+    {
+      files: ['rollup.config.js', '.eslintrc.js', 'jest.config.js'],
+      env: {
+        node: true
+      }
+    },
+    {
+      files: [
+        '**/*.test.js',
+        '**/*.test.jsx',
+        '**/*.test.ts',
+        '**/*.test.tsx',
+        '**/*.spec.js',
+        '**/*.spec.jsx',
+        '**/*.spec.ts',
+        '**/*.spec.tsx',
+        'setupTests.js'
+      ],
+      env: {
+        jest: true,
+        node: true
+      },
+      rules: {
+        'react/display-name': 'off',
+        'react/prop-types': 'off',
+        'react/no-unescaped-entities': 'off'
+      }
+    },
+    {
+      files: '**/*.+(ts)',
+      parser: '@typescript-eslint/parser',
+      parserOptions: {
+        project: './client/tsconfig.json'
+      },
+      plugins: ['@typescript-eslint/eslint-plugin'],
+      extends: [
+        'plugin:@typescript-eslint/eslint-recommended',
+        'plugin:@typescript-eslint/recommended'
+      ]
+    }
+  ],
+  settings: {
+    react: {
+      createClass: 'createReactClass', // Regex for Component Factory to use,
+      // default to "createReactClass"
+      pragma: 'React', // Pragma to use, default to "React"
+      fragment: 'Fragment', // Fragment to use (may be a property of <pragma>), default to "Fragment"
+      version: 'detect' // React version. "detect" automatically picks the version you have installed.
+    }
+  }
+};

.github/dependabot.yml

@@ -0,0 +1,41 @@
+# To get started with Dependabot version updates, you'll need to specify which
+# package ecosystems to update and where the package manifests are located.
+# Please see the documentation for all configuration options:
+# https://docs.github.com/github/administering-a-repository/configuration-options-for-dependency-updates
+
+version: 2
+updates:
+  - package-ecosystem: "npm" # See documentation for possible values
+    directory: "/api" # Location of package manifests
+    schedule:
+      interval: "weekly"
+    allow:
+      # Allow both direct and indirect updates for all packages
+      - dependency-type: "all"
+    commit-message:
+      prefix: "npm api prod"
+      prefix-development: "npm api dev"
+      include: "scope"
+  - package-ecosystem: "npm" # See documentation for possible values
+    directory: "/client" # Location of package manifests
+    schedule:
+      interval: "weekly"
+    allow:
+      # Allow both direct and indirect updates for all packages
+      - dependency-type: "all"
+    commit-message:
+      prefix: "npm client prod"
+      prefix-development: "npm client dev"
+      include: "scope"
+  - package-ecosystem: "npm" # See documentation for possible values
+    directory: "/" # Location of package manifests
+    schedule:
+      interval: "weekly"
+    allow:
+      # Allow both direct and indirect updates for all packages
+      - dependency-type: "all"
+    commit-message:
+      prefix: "npm all prod"
+      prefix-development: "npm all dev"
+      include: "scope" 
+     

.github/playwright.yml

@@ -0,0 +1,72 @@
+name: Playwright Tests
+on:
+  push:
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
+jobs:
+  tests_e2e:
+    name: Run end-to-end tests
+    timeout-minutes: 60
+    runs-on: ubuntu-latest
+    env:
+      BINGAI_TOKEN: ${{ secrets.BINGAI_TOKEN }}
+      CHATGPT_TOKEN: ${{ secrets.CHATGPT_TOKEN }}
+      MONGO_URI: ${{ secrets.MONGO_URI }}
+      OPENAI_KEY: ${{ secrets.OPENAI_KEY }}
+    steps:
+    - uses: actions/checkout@v3
+    - uses: actions/setup-node@v3
+      with:
+        node-version: 18
+        cache: 'npm'
+
+    - name: Cache API dependencies
+      uses: actions/cache@v2
+      with:
+        path: ./api/node_modules
+        key: api-${{ runner.os }}-node-${{ hashFiles('./api/package-lock.json') }}
+        restore-keys: |
+          api-${{ runner.os }}-node-
+
+    - name: Install API dependencies
+      working-directory: ./api
+      run: npm ci
+
+    - name: Cache Client dependencies
+      uses: actions/cache@v2
+      with:
+        path: ./client/node_modules
+        key: client-${{ runner.os }}-node-${{ hashFiles('./client/package-lock.json') }}
+        restore-keys: |
+          client-${{ runner.os }}-node-
+
+    - name: Install Client dependencies
+      working-directory: ./client
+      run: npm ci
+
+    - name: Build Client
+      working-directory: ./client
+      run: npm run build
+
+    - name: Install global dependencies
+      run: npm ci
+
+    - name: Install Playwright Browsers
+      run: npx playwright install --with-deps
+
+    - name: Start API server
+      working-directory: ./api
+      run: |
+        npm run start &
+        sleep 10 # Wait for the server to start
+
+    - name: Run Playwright tests
+      run: npx playwright test
+
+    - uses: actions/upload-artifact@v3
+      if: always()
+      with:
+        name: playwright-report
+        path: e2e/playwright-report/
+        retention-days: 30

.gitignore

@@ -57,3 +57,11 @@ src/style - official.css
 /e2e/specs/.test-results/
 /e2e/playwright-report/
 /playwright/.cache/
+.DS_Store
+*.code-workspace
+.idea
+
+# meilisearch
+meilisearch
+data.ms/*
+auth.json

.husky/pre-commit

@@ -0,0 +1,5 @@
+#!/usr/bin/env sh
+. "$(dirname -- "$0")/_/husky.sh"
+
+npx lint-staged
+

.prettierrc.cjs

@@ -0,0 +1,19 @@
+module.exports = {
+  printWidth: 100,
+  useTabs: false,
+  tabWidth: 2,
+  semi: true,
+  singleQuote: true,
+  // bracketSpacing: false,
+  trailingComma: 'none',
+  arrowParens: 'always',
+  embeddedLanguageFormatting: 'auto',
+  insertPragma: false,
+  proseWrap: 'preserve',
+  quoteProps: 'as-needed',
+  requirePragma: false,
+  rangeStart: 0,
+  endOfLine: 'auto',
+  jsxBracketSameLine: false,
+  jsxSingleQuote: false,
+};

CHANGELOG.md

@@ -0,0 +1,136 @@
+# # Changelog
+<details open>
+<summary><strong>2023-05-14</strong></summary>
+ 
+**Released [v0.4.4](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.4.4):**
+
+1. The Msg Clipboard was changed to a checkmark for improved user experience by @techwithanirudh in PR [#247](https://github.com/danny-avila/chatgpt-clone/pull/247).
+2. A typo in the auth.json path for accessing Google Palm was corrected by @antonme in PR [#266](https://github.com/danny-avila/chatgpt-clone/pull/266).
+3. @techwithanirudh added a Popup Menu to save sidebar space in PR [#260](https://github.com/danny-avila/chatgpt-clone/pull/260). 
+4. The default pageSize in Conversation.js was increased from 12 to 14 by @danny-avila in PR [#267](https://github.com/danny-avila/chatgpt-clone/pull/267).
+5. Fonts were updated by @techwithanirudh in PR [#261](https://github.com/danny-avila/chatgpt-clone/pull/261).
+6. Font file paths in style.css were changed by @danny-avila in PR [#268](https://github.com/danny-avila/chatgpt-clone/pull/268).
+7. Code was fixed to adjust max_tokens according to model selection by @p4w4n in PR [#263](https://github.com/danny-avila/chatgpt-clone/pull/263).
+8. Various improvements were made, such as fixing react errors and adjusting the mobile view, by @danny-avila in PR [#269](https://github.com/danny-avila/chatgpt-clone/pull/269).
+
+New contributors to the project include:
+
+- @techwithanirudh, who made their first contribution in PR [#247](https://github.com/danny-avila/chatgpt-clone/pull/247).
+- @antonme, who made their first contribution in PR [#266](https://github.com/danny-avila/chatgpt-clone/pull/266).
+- @p4w4n, who made their first contribution in PR [#263](https://github.com/danny-avila/chatgpt-clone/pull/263).
+
+The [full changelog can be found here](https://github.com/danny-avila/chatgpt-clone/compare/v0.4.3...v0.4.4)
+</details>
+ <details>
+<summary><strong>2023-05-13</strong></summary>
+ 
+**Released [v0.4.3](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.4.3) which now supports Google's PaLM 2!**
+  
+  ![image](https://github.com/danny-avila/chatgpt-clone/assets/110412045/ec5e8ff3-6c3a-4f25-9687-d8558435d094)
+ 
+**How to Setup PaLM 2 (via Google Cloud Vertex AI API)**
+
+- Enable the Vertex AI API on Google Cloud:
+- - https://console.cloud.google.com/vertex-ai
+- Create a Service Account:
+- - https://console.cloud.google.com/projectselector/iam-admin/serviceaccounts/create?walkthrough_id=iam--create-service-account#step_index=1
+- Make sure to click 'Create and Continue' to give at least the 'Vertex AI User' role.
+- Create a JSON key, rename as 'auth.json' and save it in /api/data/.
+
+**Alternatively**
+
+- In your ./api/.env file, set PALM_KEY as "user_provided" to allow the user to provide a Service Account key JSON from the UI.
+- They will follow the steps above except for renaming the file, simply importing the JSON when prompted.
+- The key is sent to the server but never saved except in your local storage
+
+**Note:**
+
+- Vertex AI does not (yet) support response streaming for text generations, so response may seem to take long when generating a lot of text.
+- Text streaming is simulated
+
+
+You can check the full changelog in between v0.4.2 and v0.4.3 [here](https://github.com/danny-avila/chatgpt-clone/compare/v0.4.2...v0.4.3).
+</details>
+
+<details>
+<summary><strong>2023-05-11</strong></summary>
+ 
+**Released [v0.4.2](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.4.2)**
+ 
+ChatGPT-Clone received some important upgrades and improvements. A new contributor, [@qcgm1978](https://github.com/qcgm1978), makes their first contribution by adding a null check for adaptiveCards variable. Additionally, support for titling conversations with the Azure endpoint is added by [@danny-avila](https://github.com/danny-avila) in PR [#234](https://github.com/danny-avila/chatgpt-clone/pull/234). In PR [#235](https://github.com/danny-avila/chatgpt-clone/pull/235), [@danny-avila](https://github.com/danny-avila) also makes some necessary fixes to titling, quotation marks, and endpoints being unavailable with only the Azure key provided. The logging system is now powered by Pino and sanitization, thanks to [@danorlando](https://github.com/danorlando) in PR [#227](https://github.com/danny-avila/chatgpt-clone/pull/227). To bulletproof the Docker container, the .dockerignore file is updated to include the client/.env file by [@danny-avila](https://github.com/danny-avila) in PR [#241](https://github.com/danny-avila/chatgpt-clone/pull/241). This issue was brought to our attention on discord.
+
+There is active work on the new Plugins feature, converting the frontend to Typescript, and looking to integrate Palm2, google's new generative AI accessible via API, to the project as a new endpoint.
+
+You can check the full changelog in between [v0.4.1](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.4.1) and [v0.4.2](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.4.2) [here](https://github.com/danny-avila/chatgpt-clone/compare/v0.4.1...v0.4.2)."
+
+For discussion and suggestion you can join us: **[community discord server](https://discord.gg/NGaa9RPCft)**
+</details>
+
+<details>
+<summary><strong>2023-05-09</strong></summary>
+ 
+**Released [v0.4.1](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.4.1)**
+ 
+ * update user system section of readme by @danorlando in #207
+ * remove github-passport and update package.lock files by @danorlando in #208
+ * Update README.md by @fuegovic in #209
+ * fix: fix browser refresh redirecting to /chat/new by @danorlando in #210
+ * fix: fix issue with validation when google account has multiple spaces in username by @danorlando in #211
+ * chore: update docker image version to use latest by @danny-avila in #218
+ * update documentation structure by @fuegovic in #220
+ * Feat: Add Azure support by @danny-avila in #219
+ * Update Message.js by @DavidDev1334 in #191
+ 
+⚠️ **IMPORTANT :** Since V0.4.0 You should register and login with a local account (email and password) for the first time sign-up. if you use login for the first time with a social login account (eg. Google, facebook, etc.), the conversations and presets that you created before the user system was implemented will NOT be migrated to that account.
+
+⚠️ **Breaking - new Env Variables :** Since V0.4.0 You will need to add the new env variables from .env.example for the app to work, even if you're not using multiple users for your purposes.
+
+For discussion and suggestion you can join us: **[community discord server](https://discord.gg/NGaa9RPCft)**
+</details>
+ <details>
+<summary><strong>2023-05-07</strong></summary>
+
+**Released [v0.4.0](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.4.0)**, Introducing User/Auth System and OAuth2/Social Login! You can now register and login with an email account or use Google login. Your your previous conversations and presets will migrate to your new profile upon creation. Check out the details in the [User/Auth System](#userauth-system) section of the README.md.
+ 
+⚠️ **IMPORTANT :** You should register and login with a local account (email and password) for the first time sign-up. if you use login for the first time with a social login account (eg. Google, facebook, etc.), the conversations and presets that you created before the user system was implemented will NOT be migrated to that account.
+
+⚠️ **Breaking - new Env Variables :** You will need to add the new env variables from .env.example for the app to work, even if you're not using multiple users for your purposes.
+
+For discussion and suggestion you can join us: **[community discord server](https://discord.gg/NGaa9RPCft)**
+</details>
+ 
+
+<details>
+<summary><strong>2023-04-05</strong></summary>
+
+**Released [v0.3.0](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.3.0)**, Introducing more customization for both OpenAI & BingAI conversations! This is one of the biggest updates yet and will make integrating future LLM's a lot easier, providing a lot of customization features as well, including sharing presets! Please feel free to share them in the **[community discord server](https://discord.gg/NGaa9RPCft)**
+</details>
+
+  
+<details>
+<summary><strong>2023-03-23</strong></summary>
+
+**Released [v0.1.0](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.1.0)**, **searching messages/conversations is live!** Up next is more custom parameters for customGpt's. Join the discord server for more immediate assistance and update: **[community discord server](https://discord.gg/NGaa9RPCft)**
+</details>
+
+  
+<details>
+<summary><strong>2023-03-22</strong></summary>
+
+**Released [v0.0.6](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.0.6)**, the latest stable release before **Searching messages** goes live tomorrow. See exact updates to date in the tag link. By request, there is now also a **[community discord server](https://s
+</details>
+
+<details>
+<summary><strong>2023-03-20</strong></summary>
+
+**Searching messages** is almost here as I test more of its functionality. There've been a lot of great features requested and great contributions and I will work on some soon, namely, further customizing the custom gpt params with sliders similar to the OpenAI playground, and including the custom params and system messages available to Bing.
+
+The above features are next and then I will have to focus on building the **test environment.** I would **greatly appreciate** help in this area with any test environment you're familiar with (mocha, chai, jest, playwright, puppeteer). This is to aid in the velocity of contributing and to save time I spend debugging.
+
+On that note, I had to switch the default branch due to some breaking changes that haven't been straight forward to debug, mainly related to node-chat-gpt the main dependency of the project. Thankfully, my working branch, now switched to default as main, is working as expected.
+</details>
+
+
+##
+
+## [Go Back to ReadMe](README.md)

CONTRIBUTORS.md

@@ -0,0 +1,26 @@
+# Contributors List
+
+We appreciate all the contributors who helped make this project possible:
+
+-   danny-avila (Admin)
+-   wtlyu (Contributor)
+-   danorlando (Contributor)
+-   alfredo-f (Contributor)
+-   HyunggyuJang (Contributor)
+-   fuegovic (Contributor)
+-   DavidDev1334
+-	toordog (Contributor)
+-   heathriel (External Contributor)
+-   hackreactor-bot (Contributor)
+-   git-bruh (Contributor)
+-   zhangsean (Contributor)
+-   llk89 (Contributor)
+-   adamrb (Contributor)
+
+
+
+If you have contributed to this project and would like to be added to the list of contributors, please submit a pull request updating this file with your name and GitHub username.
+
+##
+
+## [Go Back to ReadMe](README.md)

Dockerfile

@@ -1,35 +1,32 @@
-FROM node:19-alpine AS react-client
+# Base node image
+FROM node:19-alpine AS base
+WORKDIR /api
+COPY /api/package*.json /api/
 WORKDIR /client
-# copy package.json into the container at /client
 COPY /client/package*.json /client/
-# install dependencies
+WORKDIR /
+COPY /package*.json /
 RUN npm ci
-# Copy the current directory contents into the container at /client
+
+# React client build
+FROM base AS react-client
+WORKDIR /client
 COPY /client/ /client/
-# Build webpack artifacts
+ENV NODE_OPTIONS="--max-old-space-size=2048"
 RUN npm run build
 
-FROM node:19-alpine AS node-api
+# Node API setup
+FROM base AS node-api
 WORKDIR /api
-# copy package.json into the container at /api
-COPY /api/package*.json /api/
-# install dependencies
-RUN npm ci
-# Copy the current directory contents into the container at /api
 COPY /api/ /api/
-# Copy the client side code
 COPY --from=react-client /client/dist /client/dist
-# Make port 3080 available to the world outside this container
 EXPOSE 3080
-# Expose the server to 0.0.0.0
 ENV HOST=0.0.0.0
-# Run the app when the container launches
 CMD ["npm", "start"]
 
 # Optional: for client with nginx routing
 FROM nginx:stable-alpine AS nginx-client
 WORKDIR /usr/share/nginx/html
-COPY --from=react-client /client/public /usr/share/nginx/html
-# Add your nginx.conf
-COPY /client/nginx.conf /etc/nginx/conf.d/default.conf
+COPY --from=react-client /client/dist /usr/share/nginx/html
+COPY client/nginx.conf /etc/nginx/conf.d/default.conf
 ENTRYPOINT ["nginx", "-g", "daemon off;"]

Dockerfile-app

@@ -0,0 +1,35 @@
+# ./Dockerfile
+
+FROM node:19-alpine
+WORKDIR /app
+
+# Copy package.json files for client and api
+COPY /client/package*.json /app/client/
+COPY /api/package*.json /app/api/
+COPY /package*.json /app/
+
+# Install dependencies for both client and api
+RUN npm ci
+
+# Copy the current directory contents into the container
+COPY /client/ /app/client/
+COPY /api/ /app/api/
+
+# Set the memory limit for Node.js
+ENV NODE_OPTIONS="--max-old-space-size=2048"
+
+# Build artifacts for the client
+RUN cd /app/client && npm run build
+
+# Create the necessary directory and copy the client side code to the api directory
+RUN mkdir -p /app/api/client && cp -R /app/client/dist /app/api/client/dist
+
+# Make port 3080 available to the world outside this container
+EXPOSE 3080
+
+# Expose the server to 0.0.0.0
+ENV HOST=0.0.0.0
+
+# Run the app when the container launches
+WORKDIR /app/api
+CMD ["npm", "start"]

LICENSE.md

@@ -1,7 +1,7 @@
-MIT License
+# MIT License
 
 Copyright (c) 2023 Danny Avila
-
+##
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal
 in the Software without restriction, including without limitation the rights
@@ -12,6 +12,8 @@ furnished to do so, subject to the following conditions:
 The above copyright notice and this permission notice shall be included in all
 copies or substantial portions of the Software.
 
+##
+
 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
@@ -19,3 +21,7 @@ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
+
+##
+
+## [Go Back to ReadMe](README.md)

README.md

@@ -1,5 +1,6 @@
+
 <p align="center">
-  <a href="https://discord.gg/sDfH4MwDWJ">
+  <a href="https://discord.gg/NGaa9RPCft">
     <picture>
       <source media="(prefers-color-scheme: dark)" srcset="https://user-images.githubusercontent.com/110412045/228325485-9d3e618f-a980-44fe-89e9-d6d39164680e.png">
       <img src="https://user-images.githubusercontent.com/110412045/228325485-9d3e618f-a980-44fe-89e9-d6d39164680e.png" height="128">
@@ -9,7 +10,7 @@
 </p>
 
 <p align="center">
-  <a aria-label="Join the community on Discord" href="https://discord.gg/sDfH4MwDWJ">
+  <a aria-label="Join the community on Discord" href="https://discord.gg/NGaa9RPCft">
     <img alt="" src="https://img.shields.io/badge/Join%20the%20community-blueviolet.svg?style=for-the-badge&logo=DISCORD&labelColor=000000&logoWidth=20">
   </a>
   <a aria-label="Sponsors" href="#sponsors">
@@ -20,376 +21,129 @@
 ## All AI Conversations under One Roof. ##
   Assistant AIs are the future and OpenAI revolutionized this movement with ChatGPT. While numerous UIs exist, this app commemorates the original styling of ChatGPT, with the ability to integrate any current/future AI models, while integrating and improving upon original client features, such as conversation/message search and prompt templates (currently WIP). Through this clone, you can avoid ChatGPT Plus in favor of free or pay-per-call APIs. I will soon deploy a demo of this app. Feel free to contribute, clone, or fork. Currently dockerized.
   
-  <div align="center">
-  <video src="https://user-images.githubusercontent.com/110412045/223754183-8b7f45ce-6517-4bd5-9b39-c624745bf399.mp4" width=400/>
-</div>
+  ![clone3](https://user-images.githubusercontent.com/110412045/230538752-9b99dc6e-cd02-483a-bff0-6c6e780fa7ae.gif)
 
-## Sponsors
-
-  Sponsored by <a href="https://github.com/DavidDev1334"><b>@DavidDev1334</b></a> & <a href="https://github.com/mjtechguy"><b>@mjtechguy</b></a>
-
-
-## Updates
-<details open>
-<summary><strong>2023-03-23</strong></summary>
-
-
-
-**Released [v0.1.0](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.1.0)**, **searching messages/conversations is live!** Up next is more custom parameters for customGpt's. Join the discord server for more immediate assistance and update: **[community discord server](https://discord.gg/NGaa9RPCft)**
-
-</details>
-
-<details>
-<summary><strong>Previous Updates</strong></summary>
-  
-<details>
-<summary><strong>2023-03-22</strong></summary>
-
-
-
-**Released [v0.0.6](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.0.6)**, the latest stable release before **Searching messages** goes live tomorrow. See exact updates to date in the tag link. By request, there is now also a **[community discord server](https://discord.gg/NGaa9RPCft)**
-
-</details>
-<details>
-<summary><strong>2023-03-20</strong></summary>
-
-
-
-**Searching messages** is almost here as I test more of its functionality. There've been a lot of great features requested and great contributions and I will work on some soon, namely, further customizing the custom gpt params with sliders similar to the OpenAI playground, and including the custom params and system messages available to Bing.
-
-The above features are next and then I will have to focus on building the **test environment.** I would **greatly appreciate** help in this area with any test environment you're familiar with (mocha, chai, jest, playwright, puppeteer). This is to aid in the velocity of contributing and to save time I spend debugging.
-
-On that note, I had to switch the default branch due to some breaking changes that haven't been straight forward to debug, mainly related to node-chat-gpt the main dependency of the project. Thankfully, my working branch, now switched to default as main, is working as expected.
-
-</details>
-<details>
-<summary><strong>2023-03-16</strong></summary>
-
-
-
-[Latest release (v0.0.4)](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.0.4) includes Resubmitting messages & Branching messages, which mirrors official ChatGPT feature of editing a sent message, that then branches the conversation into separate message paths (works only with ChatGPT)
-
-Full details and [example here](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.0.4). Message search is on the docket
-
-</details>
-<details>
-<summary><strong>2023-03-12</strong></summary>
-
-
-
-
-Really thankful for all the issues reported and contributions made, the project's features and improvements have accelerated as result. Honorable mention is [wtlyu](https://github.com/wtlyu) for contributing a lot of mindful code, namely hostname configuration and mobile styling. I will upload images on next release for faster docker setup, and starting updating them simultaneously with this repo.
-
-
-
-Many improvements across the board, the biggest is being able to start conversations simultaneously (again thanks to [wtlyu](https://github.com/wtlyu) for bringing it to my attention), as you can switch conversations or start a new chat without any response streaming from a prior one, as the backend will still process/save client responses. Just watch out for any rate limiting from OpenAI/Microsoft if this is done excessively.
-
-
-Adding support for conversation search is next! Thank you [mysticaltech](https://github.com/mysticaltech) for bringing up a method I can use for this.
-</details>
-<details>
-<summary><strong>2023-03-09</strong></summary>
-Released v.0.0.2
-
-Adds Sydney (jailbroken Bing AI) to the model menu. Thank you [DavesDevFails](https://github.com/DavesDevFails) for bringing it to my attention in this [issue](https://github.com/danny-avila/chatgpt-clone/issues/13). Bing/Sydney now correctly cite links, more styling to come. Fix some overlooked bugs, and model menu doesn't close upon deleting a customGpt.
-
-
-I've re-enabled the ChatGPT browser client (free version) since it might be working for most people, it no longer works for me. Sydney is the best free route anyway.
-</details>
-<details>
-<summary><strong>2023-03-07</strong></summary>
-Due to increased interest in the repo, I've dockerized the app as of this update for quick setup! See setup instructions below. I realize this still takes some time with installing docker dependencies, so it's on the roadmap to have a deployed demo. Besides this, I've made major improvements for a lot of the existing features across the board, mainly UI/UX.
-
-
-Also worth noting, the method to access the Free Version is no longer working, so I've removed it from model selection until further notice.
-</details>
-<details>
-<summary><strong>2023-03-04</strong></summary>
-Custom prompt prefixing and labeling is now supported through the official API. This nets some interesting results when you need ChatGPT for specific uses or entertainment. Select 'CustomGPT' in the model menu to configure this, and you can choose to save the configuration or reference it by conversation. Model selection will change by conversation.
-</details>
-<details>
-<summary><strong>2023-03-01</strong></summary>
-Official ChatGPT API is out! Removed davinci since the official API is extremely fast and 10x less expensive. Since user labeling and prompt prefixing is officially supported, I will add a View feature so you can set this within chat, which gives the UI an added use case. I've kept the BrowserClient, since it's free to use like the official site.
-
-The Messages UI correctly mirrors code syntax highlighting. The exact replication of the cursor is not 1-to-1 yet, but pretty close. Later on in the project, I'll implement tests for code edge cases and explore the possibility of running code in-browser. Right now, unknown code defaults to javascript, but will detect language as close as possible.
-</details>
-<details>
-<summary><strong>2023-02-21</strong></summary>
-BingAI is integrated (although sadly limited by Microsoft with the 5 msg/convo limit, 50 msgs/day). I will need to handle the case when Bing refuses to give more answers on top of the other styling features I have in mind. Official ChatGPT use is back with the new BrowserClient. Brainstorming how to handle the UI when the Ai model changes, since conversations can't be persisted between them (or perhaps build a way to achieve this at some level).
-</details>
-<details >
-<summary><strong>2023-02-15</strong></summary>
-Just got access to Bing AI so I'll be focusing on integrating that through waylaidwanderer's 'experimental' BingAIClient.
-</details>
-<details>
-<summary><strong>2023-02-14</strong></summary>
-
-Official ChatGPT use is no longer possible though I recently used it with waylaidwanderer's [reverse proxy method](https://github.com/waylaidwanderer/node-chatgpt-api/blob/main/README.md#using-a-reverse-proxy), and before that, through leaked models he also discovered.
-
-Currently, this project is only functional with the `text-davinci-003` model.
-</details>
-</details>
-
-# Table of Contents
-- [ChatGPT Clone](#chatgpt-clone)
-  - [All AI Conversations under One Roof.](#all-ai-conversations-under-one-roof)
-  - [Updates](#updates)
-- [Table of Contents](#table-of-contents)
-  - [Roadmap](#roadmap)
-    - [Features](#features)
-    - [Tech Stack](#tech-stack)
-  - [Getting Started](#getting-started)
-    - [Prerequisites](#prerequisites)
-  - [Usage](#usage)
-    - [Local](#local)
-    - [Docker](#docker)
-    - [Access Tokens](#access-tokens)
-    - [Proxy](#proxy)
-    - [User System](#user-system)
-    - [Updating](#updating)
-  - [Use Cases](#use-cases)
-  - [Origin](#origin)
-  - [Caveats](#caveats)
-    - [Regarding use of Official ChatGPT API](#regarding-use-of-official-chatgpt-api)
-  - [Contributing](#contributing)
-  - [License](#license)
-
-## Roadmap
-
-> **Warning**
-
->  This is a work in progress. I'm building this in public. FYI there is still a lot of tech debt to cleanup. You can follow the progress here or on my [Linkedin](https://www.linkedin.com/in/danny-avila).
-
-
-
-<details>
-<summary><strong>Here are my recently completed and planned features:</strong></summary>
-
-- [x] Persistent conversation
-- [x] Rename, delete conversations
-- [x] UI Error handling
-- [x] Bing AI integration
-- [x] AI model change handling (start new convos within existing, remembers last selected)
-- [x] Code block handling (highlighting, markdown, clipboard, language detection)
-- [x] Markdown handling
-- [x] Customize prompt prefix/label (custom ChatGPT using official API)
-- [x] Server convo pagination (limit fetch and load more with 'show more' button)
-- [x] Config file for easy startup (docker compose)
-- [x] Mobile styling (thanks to [wtlyu](https://github.com/wtlyu))
-- [x] Resubmit/edit sent messages (thanks to [wtlyu](https://github.com/wtlyu))
-- [ ] Message Search
-- [ ] Custom params for ChatGPT API (temp, top_p, presence_penalty)
-- [ ] Bing AI Styling (params, suggested responses, convo end, etc.) - **In progress**
-- [ ] Add warning before clearing convos
-- [ ] Build test suite for CI/CD
-- [ ] Prompt Templates/Search
-- [ ] Refactor/clean up code (tech debt)
-- [ ] Optional use of local storage for credentials
-- [ ] Deploy demo
-
-</details>
-
-### Features
+# Features
 
 - Response streaming identical to ChatGPT through server-sent events
 - UI from original ChatGPT, including Dark mode
-- AI model selection (official ChatGPT API, BingAI, ChatGPT Free)
-- Create and Save custom ChatGPTs*
+- AI model selection (through 3 endpoints: OpenAI API, BingAI, and ChatGPT Browser)
+- Create, Save, & Share custom presets for OpenAI and BingAI endpoints - [More info on customization here](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.3.0)
 - Edit and Resubmit messages just like the official site (with conversation branching)
-- Search all messages/conversations - [see details here](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.1.0)
+- Search all messages/conversations - [More info here](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.1.0)
+- Integrating plugins soon
 
-^* ChatGPT can be 'customized' by setting a system message or prompt prefix and alternate 'role' to the API request^
+##
+# Sponsors
 
-[More info here](https://platform.openai.com/docs/guides/chat/instructing-chat-models). Here's an [example from this app.]()
-
-### Tech Stack
+  Sponsored by <a href="https://github.com/DavidDev1334"><b>@DavidDev1334</b></a>, <a href="https://github.com/mjtechguy"><b>@mjtechguy</b></a>, <a href="https://github.com/Pharrcyde"><b>@Pharrcyde</b></a>, & <a href="https://github.com/fuegovic"><b>@fuegovic</b></a>
 
+##
 
+## **Google's PaLM 2 is now supported as of [v0.4.3](https://github.com/danny-avila/chatgpt-clone/releases/tag/v0.4.3)**
+  
+  ![image](https://github.com/danny-avila/chatgpt-clone/assets/110412045/ec5e8ff3-6c3a-4f25-9687-d8558435d094)
+ 
 <details>
-<summary><strong>This project uses:</strong></summary>
+<summary><strong>How to Setup PaLM 2 (via Google Cloud Vertex AI API)</strong></summary>
+- Enable the Vertex AI API on Google Cloud:
+- - https://console.cloud.google.com/vertex-ai
+- Create a Service Account:
+- - https://console.cloud.google.com/projectselector/iam-admin/serviceaccounts/create?walkthrough_id=iam--create-service-account#step_index=1
+- Make sure to click 'Create and Continue' to give at least the 'Vertex AI User' role.
+- Create a JSON key, rename as 'auth.json' and save it in /api/data/.
 
+**Alternatively**
 
+- In your ./api/.env file, set PALM_KEY as "user_provided" to allow the user to provide a Service Account key JSON from the UI.
+- They will follow the steps above except for renaming the file, simply importing the JSON when prompted.
+- The key is sent to the server but never saved except in your local storage
 
-- [node-chatgpt-api](https://github.com/waylaidwanderer/node-chatgpt-api)
-- No React boilerplate/toolchain/clone tutorials, created from scratch with react@latest
-- Use of Tailwind CSS and [shadcn/ui](https://github.com/shadcn/ui) components
-- Docker, useSWR, Redux, Express, MongoDB, [Keyv](https://www.npmjs.com/package/keyv)
+**Note:**
+
+- Vertex AI does not (yet) support response streaming for text generations, so response may seem to take long when generating a lot of text.
+- Text streaming is simulated
 </details>
 
+---
 
+## [Read all Latest Updates here](CHANGELOG.md)
 
-## Getting Started
+<h1>Table of Contents</h1>
 
-### Prerequisites
-- npm
-- Node.js >= 19.0.0
-- MongoDB installed or [MongoDB Atlas](https://account.mongodb.com/account/login) (required if not using Docker)
-    - MongoDB does not support older ARM CPUs like those found in Raspberry Pis. However, you can make it work by setting MongoDB's version to mongo:4.4.18 in docker-compose.yml, the most recent version compatible with
-- [Docker (optional)](https://www.docker.com/get-started/)
-- [OpenAI API key](https://platform.openai.com/account/api-keys)
-- BingAI, ChatGPT access tokens (optional, free AIs)
-
-## Usage
-
-- **Clone/download** the repo down where desired
-```bash
-  git clone https://github.com/danny-avila/chatgpt-clone.git
-```
-- If using MongoDB Atlas, remove `&w=majority` from default connection string.
-
-### Local
-### **[In-depth instructions here!](https://github.com/danny-avila/chatgpt-clone/blob/0d4f0f74c04337aaf51b9a3eef898165a7009156/LOCAL_INSTALL.md)** 
-- thank you [@fuegovic](https://github.com/fuegovic)!
-
-### Docker
-
-- **Provide** all credentials, (API keys, access tokens, and Mongo Connection String) in [docker-compose.yml](docker-compose.yml) under api service
-- **Run** `docker-compose up` to start the app
-- Note: MongoDB does not support older ARM CPUs like those found in Raspberry Pis. However, you can make it work by setting MongoDB's version to mongo:4.4.18 in docker-compose.yml, the most recent version compatible with
-
-### Access Tokens
-
-<details>
-<summary><strong>ChatGPT Free Instructions</strong></summary>
-
-To get your Access token For ChatGPT 'Free Version', login to chat.openai.com, then visit https://chat.openai.com/api/auth/session.
-
-
-**Warning:** There may be a high chance of your account being banned with this method. Continue doing so at your own risk.
+<details open>
+  <summary><strong>Getting Started</strong></summary>
 
+  * [Docker Install](/documents/install/docker_install.md)
+  * [Linux Install](documents/install/linux_install.md)
+  * [Mac Install](documents/install/mac_install.md)
+  * [Windows Install](documents/install/windows_install.md)
 </details>
 
 <details>
-<summary><strong>BingAI Instructions</strong></summary>
-The Bing Access Token is the "_U" cookie from bing.com. Use dev tools or an extension while logged into the site to view it.
+  <summary><strong>General Information</strong></summary>
 
-**Note:** Specific error handling and styling for this model is still in progress.
-</details>
-
-### Proxy
-
-If your server cannot connect to the chatGPT API server by some reason, (eg in China). You can set a environment variable `PROXY`. This will be transmitted to `node-chatgpt-api` interface.
-
-**Warning:** `PROXY` is not `reverseProxyUrl` in `node-chatgpt-api`
-
-<details>
-<summary><strong>Set up proxy in local environment </strong></summary>
-
-Here is two ways to set proxy.
-- Option 1: system level environment
-`export PROXY="http://127.0.0.1:7890"`
-- Option 2: set in .env file
-`PROXY="http://127.0.0.1:7890"`
-
-**Change `http://127.0.0.1:7890` to your proxy server**
+  * [Project Origin](documents/general_info/project_origin.md)
+  * [Multilingual Information](documents/general_info/multilingual_information.md)
+  * [Roadmap](documents/general_info/roadmap.md)
+  * [Tech Stack](documents/general_info/tech_stack.md)
+  * [Changelog](CHANGELOG.md)
+  * [Bing Jailbreak Info](documents/general_info/bing_jailbreak_info.md)
 </details>
 
 <details>
-<summary><strong>Set up proxy in docker environment </strong></summary>
-
-set in docker-compose.yml file, under services - api - environment
-
-```
-    api:
-        ...
-        environment:
-                ...
-                - "PROXY=http://127.0.0.1:7890"
-                # add this line ↑
-```
-
-**Change `http://127.0.0.1:7890` to your proxy server**
+  <summary><strong>Features</strong></summary>
 
+  * [User Auth System](documents/features/user_auth_system.md)
+  * [Proxy](documents/features/proxy.md)
 </details>
 
-### User System
-
-By default, there is no user system enabled, so anyone can access your server.
-
-**This project is not designed to provide a complete and full-featured user system.** It's not high priority task and might never be provided.
-
-[wtlyu](https://github.com/wtlyu) provide a sample user system structure, that you can implement your own user system. It's simple and not a ready-for-use edition. 
-
-(If you want to implement your user system, open this ↓)
-
 <details>
-<summary><strong>Implement your own user system </strong></summary>
-
-To enable the user system, set `ENABLE_USER_SYSTEM=1` in your `.env` file.
-
-The sample structure is simple. It provide three basic endpoint:
-
-1. `/auth/login` will redirect to your own login url. In the sample code, it's `/auth/your_login_page`.
-2. `/auth/logout` will redirect to your own logout url. In the sample code, it's `/auth/your_login_page/logout`.
-3. `/api/me` will return the userinfo: `{ username, display }`.
-   1. `username` will be used in db, used to distinguish between users.
-   2. `display` will be displayed in UI.
-
-The only one thing that drive user system work is `req.session.user`. Once it's set, the client will be trusted. Set to `null` if logout.
-
-Please refer to `/api/server/routes/authYourLogin.js` file. It's very clear and simple to tell you how to implement your user system.
-
-Or you can ask chatGPT to write the code for you, here is one example to connect LDAP:
-
-```
-Please write me an express module, that serve the login and logout endpoint as a router. The login and logout uri is '/' and '/logout'. Once loginned, save display name and username in session.user, as {display, username}. Then redirect to '/'. Please write the code using express and other lib, and storage any server configuration in a config variable. I want the user to be connected to my LDAP server.
-```
+  <summary><strong>Cloud Deployment</strong></summary>
 
+  * [Heroku](documents/deployment/heroku.md)
 </details>
 
+<details>
+  <summary><strong>Contributions</strong></summary>
 
-### Updating
-
-- As the project is still a work-in-progress, you should pull the latest and run the steps over. Reset your browser cache/clear site data.
-
-## Use Cases ##
-
-
+  * [Code of Conduct](documents/contributions/code_of_conduct.md)
+  * [Contributor Guidelines](documents/contributions/contributor_guidelines.md)
+  * [Documentation Guidelines](documents/contributions/documentation_guidelines.md)
+  * [Code Standards and Conventions](documents/contributions/coding_conventions.md)
+  * [Testing](documents/contributions/testing.md)
+  * [Pull Request Template](documents/contributions/pull_request_template.md)
+  * [Contributors](CONTRIBUTORS.md)
+  * [Trello Board](https://trello.com/b/17z094kq/chatgpt-clone)
+</details>
 
 <details>
-<summary><strong> Why use this project? </strong></summary>
+  <summary><strong>Report Templates</strong></summary>
 
-  - One stop shop for all conversational AIs, with the added bonus of searching past conversations.
-  - Using the official API, you'd have to generate 7.5 million words to expense the same cost as ChatGPT Plus ($20).
-  - ChatGPT/Google Bard/Bing AI conversations are lost in space or
-  cannot be searched past a certain timeframe.
-  - **Customize ChatGPT**
+  * [Bug Report Template](documents/report_templates/bug_report_template.md)
+  * [Custom Issue Template](documents/report_templates/custom_issue_template.md)
+  * [Feature Request Template](documents/report_templates/feature_request_template.md)
+</details>
 
-    ![use case example](./images/use_case3.png "Make a Custom GPT")
+##
+### [Alternative Documentation](https://chatgpt-clone.gitbook.io/chatgpt-clone-docs/get-started/docker)
 
-  - **API is not as limited as ChatGPT Free (at [chat.openai.com](https://chat.openai.com/chat))**
+##
 
-    ![use case example](./images/use_case2.png "chat.openai.com is getting more limited by the day!")
+## Star History
 
-  - **ChatGPT Free is down.**
+[![Star History Chart](https://api.star-history.com/svg?repos=danny-avila/chatgpt-clone&type=Date)](https://star-history.com/#danny-avila/chatgpt-clone&Date)
 
-    ![use case example](./images/use_case.png "GPT is down! Plus is too expensive!")
-    
-    </details>
-
-
-## Origin ##
-    This project was started early in Feb '23, anticipating the release of the official ChatGPT API from OpenAI, which is now used. It was originally created as a Minimum Viable Product (or MVP) for the [@HackReactor](https://github.com/hackreactor/) Bootcamp. It was built with OpenAI response streaming and most of the UI completed in under 20 hours. During the end of that time, I had most of the UI and basic functionality done. This was created without using any boilerplates or templates, including create-react-app and other toolchains. I didn't follow any 'un-official chatgpt' video tutorials, and simply referenced the official site for the UI. The purpose of the exercise was to learn setting up a full stack project from scratch. Please feel free to give feedback, suggestions, or fork the project for your own use.
-
-
-## Caveats
-### Regarding use of Official ChatGPT API
-From [@waylaidwanderer](https://github.com/waylaidwanderer/node-chatgpt-api/blob/main/README.md#caveats):
-
-Since `gpt-3.5-turbo` is ChatGPT's underlying model, I had to do my best to replicate the way the official ChatGPT website uses it.
-This means my implementation or the underlying model may not behave exactly the same in some ways:
-- Conversations are not tied to any user IDs, so if that's important to you, you should implement your own user ID system.
-- ChatGPT's model parameters (temperature, frequency penalty, etc.) are unknown, so I set some defaults that I thought would be reasonable.
-- Conversations are limited to roughly the last 3000 tokens, so earlier messages may be forgotten during longer conversations.
-  - This works in a similar way to ChatGPT, except I'm pretty sure they have some additional way of retrieving context from earlier messages when needed (which can probably be achieved with embeddings, but I consider that out-of-scope for now).
-
-## Contributing
-
-Contributions and suggestions welcome! Bug reports and fixes are welcome!
+## Contributors
+Contributions and suggestions bug reports and fixes are welcome!
+Please read the documentation before you do!
 
 For new features, components, or extensions, please open an issue and discuss before sending a PR. 
 
 - Join the [Discord community](https://discord.gg/NGaa9RPCft)
 
-## License
-This project is licensed under the MIT License.
+This project exists in its current state thanks to all the people who contribute
+---
+<a href="https://github.com/danny-avila/chatgpt-clone/graphs/contributors">
+  <img src="https://contrib.rocks/image?repo=danny-avila/chatgpt-clone" />
+</a>

SECURITY.md

@@ -0,0 +1,47 @@
+# Security Policy
+
+## Reporting a Vulnerability
+We take security seriously and appreciate the efforts of security researchers to improve the security of our codebase. 
+If you discover a security vulnerability within our project, please follow these guidelines to report it to us:
+
+**Note: Only report sensible vulnerability report details via Github Security Advisory System. Every other communication channel are public and should be used only to initiate first contact and to initiate a private communication channel.**
+
+### Communication channels
+  - **Option 1: GitHub Security Advisory System**: We encourage you to use GitHub's Security Advisory system to report any security vulnerabilities you find. This allows us to receive vulnerability reports directly through GitHub. You can find more information on how to submit a security advisory report in the [GitHub Security Advisories documentation](https://docs.github.com/en/code-security/getting-started-with-security-vulnerability-alerts/about-github-security-advisories).
+  - **Option 2: Github issues**: You can initiate first contact via Github Issues. **Please note that initial contact through Discord should not include any sensitive details.** 
+  - **Option 3: Discord Server**: You can join our [Discord community](https://discord.gg/NGaa9RPCft) and initiate first contact in the `#issues` channel. **Please note that initial contact through Discord should not include any sensitive details.** 
+
+_After initial contact, we will use this initial contact to establish a private communication channel for further discussion._
+
+
+### When submitting a vulnerability report, please provide us with the following information:
+  - A clear description of the vulnerability, including steps to reproduce it
+  - The version(s) of the project affected by the vulnerability
+  - Any additional information that may be useful for understanding and addressing the issue
+We will make every effort to acknowledge your report within 72 hours and keep you informed of its progress towards resolution.
+
+
+## Security Updates and Patching
+We are committed to maintaining the security of our open-source project named ChatGPT-Clone and promptly addressing any identified vulnerabilities. To ensure the security of our project, we follow these practices:
+  - We prioritize security updates for the current major release of our software.
+  - We actively monitor the GitHub Security Advisory system and the `#security-reports` channel on Discord for any vulnerability reports.
+  - We promptly review and validate reported vulnerabilities and take appropriate actions to address them.
+  - We release security patches and updates in a timely manner to mitigate any identified vulnerabilities.
+
+Please note that as a security-conscious community, we may not always disclose detailed information about security issues until we have determined that doing so would not put our users or the project at risk. We appreciate your understanding and cooperation in these matters.
+
+## Scope
+This security policy applies to the following GitHub repository:
+- Repository: [ChatGPT-Clone](https://github.com/danny-avila/chatgpt-clone)
+
+## Contact
+If you have any questions or concerns regarding the security of our project, please join our [Discord community](https://discord.gg/NGaa9RPCft) and report them in the appropriate channel. 
+You can also reach out to us by [opening an issue](https://github.com/danny-avila/chatgpt-clone/issues/new) on GitHub.
+Please note that the response time may vary depending on the nature and severity of the inquiry.
+
+## Acknowledgments
+We would like to express our gratitude to the security researchers and community members who help us improve the security of our project. Your contributions are invaluable, and we sincerely appreciate your efforts.
+
+## Bug Bounty Program
+We do not currently have a bug bounty program in place. However, we welcome and appreciate any security-related contributions through pull requests (PRs) that address vulnerabilities in our codebase. 
+We believe in the power of collaboration to improve the security of our project and invite you to join us in making it more robust.

api/.env.example

@@ -1,107 +1,177 @@
-# Server configuration. 
-# The server will listen to localhost:3080 request by default. You can set the target ip as you want.
-# If you want this server can be used outside your local machine, for example to share with other 
-# machine or expose this from a docker container, set HOST=0.0.0.0 or your external ip interface.
-#
-# Tips: HOST=0.0.0.0 means listening on all interface. It's not a real ip. Use localhost:port rather
-# than 0.0.0.0:port to open it.
-HOST=localhost 
+##########################
+# Server configuration: 
+##########################
+
+# The server will listen to localhost:3080 by default. You can change the target IP as you want. 
+# If you want to make this server available externally, for example to share the server with others 
+# or expose this from a Docker container, set host to 0.0.0.0 or your external IP interface. 
+# Tips: Setting host to 0.0.0.0 means listening on all interfaces. It's not a real IP.  
+# Use localhost:port rather than 0.0.0.0:port to access the server. 
+# Set Node env to development if running in dev mode.
+HOST=localhost
 PORT=3080
-NODE_ENV=development
+NODE_ENV=production 
 
-# Change this to proxy any API request. It's useful if your machine have difficulty calling the original API server.
-# PROXY="http://YOUR_PROXY_SERVER"
+# Change this to proxy any API request. 
+# It's useful if your machine has difficulty calling the original API server. 
+# PROXY=
 
-# Change this to your MongoDB URI if different and I recommend appending chatgpt-clone
-MONGO_URI="mongodb://127.0.0.1:27017/chatgpt-clone"
+# Change this to your MongoDB URI if different. I recommend appending chatgpt-clone. 
+MONGO_URI=mongodb://127.0.0.1:27017/chatgpt-clone
 
+##########################
+# OpenAI Endpoint: 
+##########################
 
-#############################
-# Endpoint OpenAI:
-#############################
-
-# Access key from OpenAI platform
-# Leave it blank to disable this endpoint
+# Access key from OpenAI platform. 
+# Leave it blank to disable this feature.
+# Set to "user_provided" to allow the user to provide their API key from the UI.
 OPENAI_KEY=
 
-# Identify the available models, sperate by comma, and not space in it
-# Leave it blank to use internal settings.
-# OPENAI_MODELS=gpt-4,text-davinci-003,gpt-3.5-turbo,gpt-3.5-turbo-0301
+# Identify the available models, separated by commas *without spaces*. 
+# The first will be default. 
+# Leave it blank to use internal settings. 
+OPENAI_MODELS=gpt-3.5-turbo,gpt-3.5-turbo-0301,text-davinci-003,gpt-4
 
-# Reverse proxy setting for OpenAI
-# https://github.com/waylaidwanderer/node-chatgpt-api#using-a-reverse-proxy
-# OPENAI_REVERSE_PROXY=<YOUR REVERSE PROXY>
+# Reverse proxy settings for OpenAI: 
+# https://github.com/waylaidwanderer/node-chatgpt-api#using-a-reverse-proxy 
+# OPENAI_REVERSE_PROXY=
 
+##########################
+# AZURE Endpoint:   
+##########################
 
-#############################
-# Endpoint BingAI (Also jailbreak Sydney):
-#############################
+# To use Azure with this project, set the following variables. These will be used to build the API URL.
+# Chat completion: 
+#     `https://{AZURE_OPENAI_API_INSTANCE_NAME}.openai.azure.com/openai/deployments/{AZURE_OPENAI_API_DEPLOYMENT_NAME}/chat/completions?api-version={AZURE_OPENAI_API_VERSION}`;
+# You should also consider changing the `OPENAI_MODELS` variable above to the models available in your instance/deployment.
+# Note: I've noticed that the Azure API is much faster than the OpenAI API, so the streaming looks almost instantaneous.
 
-# BingAI Tokens: the "_U" cookies value from bing.com
-# Leave it and BINGAI_USER_TOKEN blank to disable this endpoint.
-BINGAI_TOKEN=
+# AZURE_OPENAI_API_KEY=
+# AZURE_OPENAI_API_INSTANCE_NAME=
+# AZURE_OPENAI_API_DEPLOYMENT_NAME=
+# AZURE_OPENAI_API_VERSION=
+# AZURE_OPENAI_API_COMPLETIONS_DEPLOYMENT_NAME= # Optional, but may be used in future updates
+# AZURE_OPENAI_API_EMBEDDINGS_DEPLOYMENT_NAME= # Optional, but may be used in future updates
 
-# BingAI User defined Token
-# Allow user to set their own token by client
-# Uncomment this to enable this feature.
-# (Not implemented yet.)
-# BINGAI_USER_TOKEN=1
+##########################
+# BingAI Endpoint:   
+##########################
 
+# Also used for Sydney and jailbreak
 
-#############################
-# Endpoint chatGPT:
-#############################
+# BingAI Tokens: the "_U" cookies value from bing.com 
+# Set to "user_provided" to allow the user to provide its token from the UI.
+# Leave it blank to disable this endpoint. 
+BINGAI_TOKEN="user_provided"
 
-# ChatGPT Browser Client (free but use at your own risk)
-# Access token from https://chat.openai.com/api/auth/session
-# Exposes your access token to CHATGPT_REVERSE_PROXY
-# Leave it blank to disable this endpoint
-CHATGPT_TOKEN=
+# BingAI Host: 
+# Necessary for some people in different countries, e.g. China (https://cn.bing.com) 
+# Leave it blank to use default server. 
+# BINGAI_HOST=https://cn.bing.com 
 
-# Identify the available models, sperate by comma, and not space in it
-# Leave it blank to use internal settings.
-# CHATGPT_MODELS=text-davinci-002-render-sha,text-davinci-002-render-paid,gpt-4
+##########################
+# ChatGPT Endpoint: 
+##########################
 
-# Reverse proxy setting for OpenAI
-# https://github.com/waylaidwanderer/node-chatgpt-api#using-a-reverse-proxy
-# By default it will use the node-chatgpt-api recommended proxy, (it's a third party server)
-# CHATGPT_REVERSE_PROXY=<YOUR REVERSE PROXY>
+# ChatGPT Browser Client (free but use at your own risk) 
+# Access token from https://chat.openai.com/api/auth/session 
+# Exposes your access token to `CHATGPT_REVERSE_PROXY`
+# Set to "user_provided" to allow the user to provide its token from the UI.
+# Leave it blank to disable this endpoint 
+CHATGPT_TOKEN="user_provided"
 
+# Identify the available models, separated by commas. The first will be default. 
+# Leave it blank to use internal settings. 
+CHATGPT_MODELS=text-davinci-002-render-sha,gpt-4
+# NOTE: you can add gpt-4-plugins, gpt-4-code-interpreter, and gpt-4-browsing to the list above and use the models for these features;
+# however, the view/display portion of these features are not supported, but you can use the underlying models, which have higher token context
+# Also: text-davinci-002-render-paid is deprecated as of May 2023
 
-#############################
-# Search:
-#############################
+# Reverse proxy settings for ChatGPT 
+# https://github.com/waylaidwanderer/node-chatgpt-api#using-a-reverse-proxy 
+# By default, the server will use the node-chatgpt-api recommended proxy (a third party server). 
+# CHATGPT_REVERSE_PROXY= 
 
-# ENABLING SEARCH MESSAGES/CONVOS
-# Requires installation of free self-hosted Meilisearch or Paid Remote Plan (Remote not tested)
-# The easiest setup for this is through docker-compose, which takes care of it for you.
-# SEARCH=1
-SEARCH=1
+##########################
+# PaLM (Google) Endpoint: 
+##########################
 
-# REQUIRED FOR SEARCH: MeiliSearch Host, mainly for api server to connect to the search server.
-# must replace '0.0.0.0' with 'meilisearch' if serving meilisearch with docker-compose
-# MEILI_HOST='http://meilisearch:7700' # <-- docker-compose (should already be setup on docker-compose.yml)
-MEILI_HOST='http://0.0.0.0:7700' # <-- local/remote
+# PaLM 2 Client (via Google Cloud Vertex AI API)
+# Steps:
+    # Enable the Vertex AI API on Google Cloud:
+        # https://console.cloud.google.com/vertex-ai
+    # Create a Service Account:
+        # https://console.cloud.google.com/projectselector/iam-admin/serviceaccounts/create?walkthrough_id=iam--create-service-account#step_index=1
+    # Make sure to click 'Create and Continue' to give at least the 'Vertex AI User' role.
+    # Create a JSON key, rename as 'auth.json' and save it in /api/data/.
+# Alternatively
+    # Uncomment below PALM_KEY and set as "user_provided" to allow the user to provide a Service Account key JSON from the UI.
+    # They will follow the steps above except for renaming the file.
+# Leave blank or omit to disable this endpoint 
 
-# REQUIRED FOR SEARCH: MeiliSearch HTTP Address, mainly for docker-compose to expose the search server.
-# must replace '0.0.0.0' with 'meilisearch' if serving meilisearch with docker-compose
-# MEILI_HTTP_ADDR='meilisearch:7700'  # <-- docker-compose (should already be setup on docker-compose.yml)
-MEILI_HTTP_ADDR='0.0.0.0:7700'  # <-- local/remote
+# PALM_KEY="user_provided"
 
-# REQUIRED FOR SEARCH: In production env., needs a secure key, feel free to generate your own.
-# This master key must be at least 16 bytes, composed of valid UTF-8 characters.
-# Meilisearch will throw an error and refuse to launch if no master key is provided or if it is under 16 bytes, 
-# Meilisearch will suggest a secure autogenerated master key.
+# In case you need a reverse proxy for this endpoint:
+# GOOGLE_REVERSE_PROXY= 
+
+##########################
+# Proxy: To be Used by all endpoints
+##########################
+PROXY=
+
+##########################
+# Search: 
+##########################
+
+# ENABLING SEARCH MESSAGES/CONVOS 
+# Requires the installation of the free self-hosted Meilisearch or a paid Remote Plan (Remote not tested) 
+# The easiest setup for this is through docker-compose, which takes care of it for you. 
+SEARCH=false
+
+# REQUIRED FOR SEARCH: MeiliSearch Host, mainly for the API server to connect to the search server. 
+# Replace '0.0.0.0' with 'meilisearch' if serving MeiliSearch with docker-compose. 
+MEILI_HOST=http://0.0.0.0:7700
+
+# REQUIRED FOR SEARCH: MeiliSearch HTTP Address, mainly for docker-compose to expose the search server. 
+# Replace '0.0.0.0' with 'meilisearch' if serving MeiliSearch with docker-compose. 
+MEILI_HTTP_ADDR=0.0.0.0:7700
+
+# REQUIRED FOR SEARCH: In production env., a secure key is needed. You can generate your own. 
+# This master key must be at least 16 bytes, composed of valid UTF-8 characters. 
+# MeiliSearch will throw an error and refuse to launch if no master key is provided, 
+# or if it is under 16 bytes. MeiliSearch will suggest a secure autogenerated master key.
 # Using docker, it seems recognized as production so use a secure key.
-# MEILI_MASTER_KEY=                                           # <-- empty/insecure key works for local/remote
-MEILI_MASTER_KEY=JKMW-hGc7v_D1FkJVdbRSDNFLZcUv3S75yrxXP0SmcU # <-- ready made secure key for docker-compose
+# This is a ready made secure key for docker-compose, you can replace it with your own.
+MEILI_MASTER_KEY=DrhYf7zENyR6AlUCKmnz0eYASOQdl6zxH7s7MKFSfFCt
 
+##########################
+# User System:
+##########################
 
-#############################
-# User System
-#############################
+# Google:
+# Add your Google Client ID and Secret here, you must register an app with Google Cloud to get these values
+# https://cloud.google.com/
+GOOGLE_CLIENT_ID=
+GOOGLE_CLIENT_SECRET=
+GOOGLE_CALLBACK_URL=/oauth/google/callback
 
-# Enable the user system.
-# this is not a ready to use user system.
-# dont't use it, unless you can write your own code.
-# ENABLE_USER_SYSTEM=  # <-- make sure you don't comment this back in if you're not using your own user system
+#JWT:
+JWT_SECRET_DEV=secret
+
+# Add a secure secret for production if deploying to live domain.
+JWT_SECRET_PROD=secret
+
+# Set the expiration delay for the secure cookie with the JWT token
+# Delay is in millisecond e.g. 7 days is 1000*60*60*24*7
+SESSION_EXPIRY=1000 * 60 * 60 * 24 * 7
+
+# Site URLs:
+# Don't forget to set Node env to development in the Server configuration section above
+# if you want to run in dev mode
+CLIENT_URL_DEV=http://localhost:3090
+SERVER_URL_DEV=http://localhost:3080
+
+# Change these values to domain if deploying:
+CLIENT_URL_PROD=http://localhost:3080
+SERVER_URL_PROD=http://localhost:3080

api/.eslintrc.js

@@ -1,39 +0,0 @@
-module.exports = {
-  env: {
-    es2021: true,
-    node: true
-  },
-  extends: ['eslint:recommended'],
-  overrides: [],
-  parserOptions: {
-    ecmaVersion: 'latest',
-    sourceType: 'module'
-  },
-  rules: {
-    indent: ['error', 2, { SwitchCase: 1 }],
-    'max-len': [
-      'error',
-      {
-        code: 150,
-        ignoreStrings: true,
-        ignoreTemplateLiterals: true,
-        ignoreComments: true
-      }
-    ],
-    'linebreak-style': 0,
-    'arrow-parens': [2, 'as-needed', { requireForBlockBody: true }],
-    // 'no-plusplus': ['error', { allowForLoopAfterthoughts: true }],
-    'no-console': 'off',
-    'import/extensions': 'off',
-    'no-use-before-define': [
-      'error',
-      {
-        functions: false
-      }
-    ],
-    'no-promise-executor-return': 'off',
-    'no-param-reassign': 'off',
-    'no-continue': 'off',
-    'no-restricted-syntax': 'off'
-  }
-};

api/.prettierrc

@@ -1,22 +0,0 @@
-{
-  "arrowParens": "avoid",
-  "bracketSpacing": true,
-  "endOfLine": "lf",
-  "htmlWhitespaceSensitivity": "css",
-  "insertPragma": false,
-  "singleAttributePerLine": true,
-  "bracketSameLine": false,
-  "jsxBracketSameLine": false,
-  "jsxSingleQuote": false,
-  "printWidth": 110,
-  "proseWrap": "preserve",
-  "quoteProps": "as-needed",
-  "requirePragma": false,
-  "semi": true,
-  "singleQuote": true,
-  "tabWidth": 2,
-  "trailingComma": "none",
-  "useTabs": false,
-  "vueIndentScriptAndStyle": false,
-  "parser": "babel"
-}

api/app/clients/bingai.js

@@ -13,6 +13,7 @@ const askBing = async ({
   clientId,
   invocationId,
   toneStyle,
+  token,
   onProgress
 }) => {
   const { BingAIClient } = await import('@waylaidwanderer/chatgpt-api');
@@ -22,11 +23,13 @@ const askBing = async ({
 
   const bingAIClient = new BingAIClient({
     // "_U" cookie from bing.com
-    userToken: process.env.BINGAI_TOKEN,
+    userToken:
+      process.env.BINGAI_TOKEN == 'user_provided' ? token : process.env.BINGAI_TOKEN ?? null,
     // If the above doesn't work, provide all your cookies as a string instead
     // cookies: '',
     debug: false,
     cache: store,
+    host: process.env.BINGAI_HOST || null,
     proxy: process.env.PROXY || null
   });
 

api/app/clients/chatgpt-browser.js

@@ -6,8 +6,10 @@ const browserClient = async ({
   parentMessageId,
   conversationId,
   model,
+  token,
   onProgress,
-  abortController
+  abortController,
+  userId
 }) => {
   const { ChatGPTBrowserClient } = await import('@waylaidwanderer/chatgpt-api');
   const store = {
@@ -16,12 +18,15 @@ const browserClient = async ({
 
   const clientOptions = {
     // Warning: This will expose your access token to a third party. Consider the risks before using this.
-    reverseProxyUrl: process.env.CHATGPT_REVERSE_PROXY || 'https://bypass.churchless.tech/api/conversation',
+    reverseProxyUrl:
+      process.env.CHATGPT_REVERSE_PROXY || 'https://ai.fakeopen.com/api/conversation',
     // Access token from https://chat.openai.com/api/auth/session
-    accessToken: process.env.CHATGPT_TOKEN,
+    accessToken:
+      process.env.CHATGPT_TOKEN == 'user_provided' ? token : process.env.CHATGPT_TOKEN ?? null,
     model: model,
-    // debug: true
-    proxy: process.env.PROXY || null
+    debug: false,
+    proxy: process.env.PROXY || null,
+    user: userId
   };
 
   const client = new ChatGPTBrowserClient(clientOptions, store);

api/app/clients/chatgpt-client.js

@@ -1,12 +1,15 @@
 require('dotenv').config();
 const { KeyvFile } = require('keyv-file');
-// const set = new Set(['gpt-4', 'text-davinci-003', 'gpt-3.5-turbo', 'gpt-3.5-turbo-0301']);
+const { genAzureEndpoint } = require('../../utils/genAzureEndpoints');
+const tiktoken = require('@dqbd/tiktoken');
+const encoding_for_model = tiktoken.encoding_for_model;
 
 const askClient = async ({
   text,
   parentMessageId,
   conversationId,
   model,
+  oaiApiKey,
   chatGptLabel,
   promptPrefix,
   temperature,
@@ -14,40 +17,76 @@ const askClient = async ({
   presence_penalty,
   frequency_penalty,
   onProgress,
-  abortController
+  abortController,
+  userId
 }) => {
-  const ChatGPTClient = (await import('@waylaidwanderer/chatgpt-api')).default;
+  const { ChatGPTClient } = await import('@waylaidwanderer/chatgpt-api');
   const store = {
     store: new KeyvFile({ filename: './data/cache.json' })
   };
 
+  const azure = process.env.AZURE_OPENAI_API_KEY ? true : false;
+  if (promptPrefix == null) {
+    promptText = 'You are ChatGPT, a large language model trained by OpenAI.';
+  } else {
+    promptText = promptPrefix;
+  }
+  const maxContextTokens = model === 'gpt-4' ? 8191 : model === 'gpt-4-32k' ? 32767 : 4095; // 1 less than maximum
   const clientOptions = {
-    // Warning: This will expose your access token to a third party. Consider the risks before using this.
     reverseProxyUrl: process.env.OPENAI_REVERSE_PROXY || null,
-
+    azure,
+    maxContextTokens,
     modelOptions: {
-      model: model,
+      model,
       temperature,
       top_p,
       presence_penalty,
       frequency_penalty
     },
-
     chatGptLabel,
     promptPrefix,
-    proxy: process.env.PROXY || null,
-    debug: false
+    proxy: process.env.PROXY || null
+    // debug: true
   };
 
-  const client = new ChatGPTClient(process.env.OPENAI_KEY, clientOptions, store);
-  let options = { onProgress, abortController };
+  let apiKey = oaiApiKey ? oaiApiKey : process.env.OPENAI_KEY || null;
 
-  if (!!parentMessageId && !!conversationId) {
-    options = { ...options, parentMessageId, conversationId };
+  if (azure) {
+    apiKey = oaiApiKey ? oaiApiKey : process.env.AZURE_OPENAI_API_KEY || null;
+    clientOptions.reverseProxyUrl = genAzureEndpoint({
+      azureOpenAIApiInstanceName: process.env.AZURE_OPENAI_API_INSTANCE_NAME,
+      azureOpenAIApiDeploymentName: process.env.AZURE_OPENAI_API_DEPLOYMENT_NAME,
+      azureOpenAIApiVersion: process.env.AZURE_OPENAI_API_VERSION
+    });
   }
 
-  const res = await client.sendMessage(text, options);
-  return res;
+  const client = new ChatGPTClient(apiKey, clientOptions, store);
+
+  const options = {
+    onProgress,
+    abortController,
+    ...(parentMessageId && conversationId ? { parentMessageId, conversationId } : {})
+  };
+
+  const enc = encoding_for_model(model);
+  const text_tokens = enc.encode(text);
+  const prompt_tokens = enc.encode(promptText);
+  // console.log("Prompt tokens = ", prompt_tokens.length);
+  // console.log("Message Tokens = ", text_tokens.length);
+
+  const res = await client.sendMessage(text, { ...options, userId });
+  // return res;
+  // create a new response object that includes the token counts
+  const newRes = {
+    ...res,
+    usage: {
+      prompt_tokens: prompt_tokens.length,
+      completion_tokens: text_tokens.length,
+      total_tokens: prompt_tokens.length + text_tokens.length
+    }
+  };
+
+  return newRes;
 };
 
 module.exports = { askClient };

api/app/google/GoogleClient.js

@@ -0,0 +1,397 @@
+const crypto = require('crypto');
+const TextStream = require('../stream');
+const { google } = require('googleapis');
+const { Agent, ProxyAgent } = require('undici');
+const { getMessages, saveMessage, saveConvo } = require('../../models');
+const {
+  encoding_for_model: encodingForModel,
+  get_encoding: getEncoding
+} = require('@dqbd/tiktoken');
+
+const tokenizersCache = {};
+
+class GoogleAgent {
+  constructor(credentials, options = {}) {
+    this.client_email = credentials.client_email;
+    this.project_id = credentials.project_id;
+    this.private_key = credentials.private_key;
+    this.setOptions(options);
+    this.currentDateString = new Date().toLocaleDateString('en-us', {
+      year: 'numeric',
+      month: 'long',
+      day: 'numeric'
+    });
+  }
+
+  constructUrl() {
+    return `https://us-central1-aiplatform.googleapis.com/v1/projects/${this.project_id}/locations/us-central1/publishers/google/models/${this.modelOptions.model}:predict`;
+  }
+
+  setOptions(options) {
+    if (this.options && !this.options.replaceOptions) {
+      // nested options aren't spread properly, so we need to do this manually
+      this.options.modelOptions = {
+        ...this.options.modelOptions,
+        ...options.modelOptions
+      };
+      delete options.modelOptions;
+      // now we can merge options
+      this.options = {
+        ...this.options,
+        ...options
+      };
+    } else {
+      this.options = options;
+    }
+
+    this.options.examples = this.options.examples.filter(
+      (obj) => obj.input.content !== '' && obj.output.content !== ''
+    );
+
+    const modelOptions = this.options.modelOptions || {};
+    this.modelOptions = {
+      ...modelOptions,
+      // set some good defaults (check for undefined in some cases because they may be 0)
+      model: modelOptions.model || 'chat-bison',
+      temperature: typeof modelOptions.temperature === 'undefined' ? 0.2 : modelOptions.temperature, // 0 - 1, 0.2 is recommended
+      topP: typeof modelOptions.topP === 'undefined' ? 0.95 : modelOptions.topP, // 0 - 1, default: 0.95
+      topK: typeof modelOptions.topK === 'undefined' ? 40 : modelOptions.topK // 1-40, default: 40
+      // stop: modelOptions.stop // no stop method for now
+    };
+
+    this.isChatModel = this.modelOptions.model.startsWith('chat-');
+    const { isChatModel } = this;
+    this.isTextModel = this.modelOptions.model.startsWith('text-');
+    const { isTextModel } = this;
+
+    this.maxContextTokens = this.options.maxContextTokens || (isTextModel ? 8000 : 4096);
+    // The max prompt tokens is determined by the max context tokens minus the max response tokens.
+    // Earlier messages will be dropped until the prompt is within the limit.
+    this.maxResponseTokens = this.modelOptions.maxOutputTokens || 1024;
+    this.maxPromptTokens =
+      this.options.maxPromptTokens || this.maxContextTokens - this.maxResponseTokens;
+
+    if (this.maxPromptTokens + this.maxResponseTokens > this.maxContextTokens) {
+      throw new Error(
+        `maxPromptTokens + maxOutputTokens (${this.maxPromptTokens} + ${this.maxResponseTokens} = ${
+          this.maxPromptTokens + this.maxResponseTokens
+        }) must be less than or equal to maxContextTokens (${this.maxContextTokens})`
+      );
+    }
+
+    this.userLabel = this.options.userLabel || 'User';
+    this.modelLabel = this.options.modelLabel || 'Assistant';
+
+    if (isChatModel) {
+      // Use these faux tokens to help the AI understand the context since we are building the chat log ourselves.
+      // Trying to use "<|im_start|>" causes the AI to still generate "<" or "<|" at the end sometimes for some reason,
+      // without tripping the stop sequences, so I'm using "||>" instead.
+      this.startToken = '||>';
+      this.endToken = '';
+      this.gptEncoder = this.constructor.getTokenizer('cl100k_base');
+    } else if (isTextModel) {
+      this.startToken = '<|im_start|>';
+      this.endToken = '<|im_end|>';
+      this.gptEncoder = this.constructor.getTokenizer('text-davinci-003', true, {
+        '<|im_start|>': 100264,
+        '<|im_end|>': 100265
+      });
+    } else {
+      // Previously I was trying to use "<|endoftext|>" but there seems to be some bug with OpenAI's token counting
+      // system that causes only the first "<|endoftext|>" to be counted as 1 token, and the rest are not treated
+      // as a single token. So we're using this instead.
+      this.startToken = '||>';
+      this.endToken = '';
+      try {
+        this.gptEncoder = this.constructor.getTokenizer(this.modelOptions.model, true);
+      } catch {
+        this.gptEncoder = this.constructor.getTokenizer('text-davinci-003', true);
+      }
+    }
+
+    if (!this.modelOptions.stop) {
+      const stopTokens = [this.startToken];
+      if (this.endToken && this.endToken !== this.startToken) {
+        stopTokens.push(this.endToken);
+      }
+      stopTokens.push(`\n${this.userLabel}:`);
+      stopTokens.push('<|diff_marker|>');
+      // I chose not to do one for `modelLabel` because I've never seen it happen
+      this.modelOptions.stop = stopTokens;
+    }
+
+    if (this.options.reverseProxyUrl) {
+      this.completionsUrl = this.options.reverseProxyUrl;
+    } else {
+      this.completionsUrl = this.constructUrl();
+    }
+
+    return this;
+  }
+
+  static getTokenizer(encoding, isModelName = false, extendSpecialTokens = {}) {
+    if (tokenizersCache[encoding]) {
+      return tokenizersCache[encoding];
+    }
+    let tokenizer;
+    if (isModelName) {
+      tokenizer = encodingForModel(encoding, extendSpecialTokens);
+    } else {
+      tokenizer = getEncoding(encoding, extendSpecialTokens);
+    }
+    tokenizersCache[encoding] = tokenizer;
+    return tokenizer;
+  }
+
+  async getClient() {
+    const scopes = ['https://www.googleapis.com/auth/cloud-platform'];
+    const jwtClient = new google.auth.JWT(this.client_email, null, this.private_key, scopes);
+
+    jwtClient.authorize((err) => {
+      if (err) {
+        console.log(err);
+        throw err;
+      }
+    });
+
+    return jwtClient;
+  }
+
+  buildPayload(input, { messages = [] }) {
+    let payload = {
+      instances: [
+        {
+          messages: [...messages, { author: this.userLabel, content: input }]
+        }
+      ],
+      parameters: this.options.modelOptions
+    };
+
+    if (this.options.promptPrefix) {
+      payload.instances[0].context = this.options.promptPrefix;
+    }
+
+    if (this.options.examples.length > 0) {
+      payload.instances[0].examples = this.options.examples;
+    }
+
+    if (this.isTextModel) {
+      payload.instances = [
+        {
+          prompt: input
+        }
+      ];
+    }
+
+    if (this.options.debug) {
+      console.debug('buildPayload');
+      console.dir(payload, { depth: null });
+    }
+
+    return payload;
+  }
+
+  async getCompletion(input, messages = [], abortController = null) {
+    if (!abortController) {
+      abortController = new AbortController();
+    }
+    const { debug } = this.options;
+    const url = this.completionsUrl;
+    if (debug) {
+      console.debug();
+      console.debug(url);
+      console.debug(this.modelOptions);
+      console.debug();
+    }
+    const opts = {
+      method: 'POST',
+      agent: new Agent({
+        bodyTimeout: 0,
+        headersTimeout: 0
+      }),
+      signal: abortController.signal
+    };
+
+    if (this.options.proxy) {
+      opts.agent = new ProxyAgent(this.options.proxy);
+    }
+
+    const client = await this.getClient();
+    const payload = this.buildPayload(input, { messages });
+    const res = await client.request({ url, method: 'POST', data: payload });
+    console.dir(res.data, { depth: null });
+    return res.data;
+  }
+
+  async loadHistory(conversationId, parentMessageId = null) {
+    if (this.options.debug) {
+      console.debug('Loading history for conversation', conversationId, parentMessageId);
+    }
+
+    if (!parentMessageId) {
+      return [];
+    }
+
+    const messages = (await getMessages({ conversationId })) || [];
+
+    if (messages.length === 0) {
+      this.currentMessages = [];
+      return [];
+    }
+
+    const orderedMessages = this.constructor.getMessagesForConversation(messages, parentMessageId);
+    return orderedMessages.map((message) => {
+      return {
+        author: message.isCreatedByUser ? this.userLabel : this.modelLabel,
+        content: message.content
+      };
+    });
+  }
+
+  async saveMessageToDatabase(message, user = null) {
+    await saveMessage({ ...message, unfinished: false });
+    await saveConvo(user, {
+      conversationId: message.conversationId,
+      endpoint: 'google',
+      ...this.modelOptions
+    });
+  }
+
+  async sendMessage(message, opts = {}) {
+    if (opts && typeof opts === 'object') {
+      this.setOptions(opts);
+    }
+    console.log('sendMessage', message, opts);
+
+    const user = opts.user || null;
+    const conversationId = opts.conversationId || crypto.randomUUID();
+    const parentMessageId = opts.parentMessageId || '00000000-0000-0000-0000-000000000000';
+    const userMessageId = crypto.randomUUID();
+    const responseMessageId = crypto.randomUUID();
+    const messages = await this.loadHistory(conversationId, this.options?.parentMessageId);
+
+    const userMessage = {
+      messageId: userMessageId,
+      parentMessageId,
+      conversationId,
+      sender: 'User',
+      text: message,
+      isCreatedByUser: true
+    };
+
+    if (typeof opts?.getIds === 'function') {
+      opts.getIds({
+        userMessage,
+        conversationId,
+        responseMessageId
+      });
+    }
+
+    console.log('userMessage', userMessage);
+
+    await this.saveMessageToDatabase(userMessage, user);
+    let reply = '';
+    let blocked = false;
+    try {
+      const result = await this.getCompletion(message, messages, opts.abortController);
+      blocked = result?.predictions?.[0]?.safetyAttributes?.blocked;
+      reply =
+        result?.predictions?.[0]?.candidates?.[0]?.content ||
+        result?.predictions?.[0]?.content ||
+        '';
+      if (blocked === true) {
+        reply = `Google blocked a proper response to your message:\n${JSON.stringify(
+          result.predictions[0].safetyAttributes
+        )}${reply.length > 0 ? `\nAI Response:\n${reply}` : ''}`;
+      }
+      if (this.options.debug) {
+        console.debug('result');
+        console.debug(result);
+      }
+    } catch (err) {
+      console.error(err);
+    }
+
+    if (this.options.debug) {
+      console.debug('options');
+      console.debug(this.options);
+    }
+
+    if (!blocked) {
+      const textStream = new TextStream(reply, { delay: 0.5 });
+      await textStream.processTextStream(opts.onProgress);
+    }
+
+    const responseMessage = {
+      messageId: responseMessageId,
+      conversationId,
+      parentMessageId: userMessage.messageId,
+      sender: 'PaLM2',
+      text: reply,
+      error: blocked,
+      isCreatedByUser: false
+    };
+
+    await this.saveMessageToDatabase(responseMessage, user);
+    return responseMessage;
+  }
+
+  getTokenCount(text) {
+    return this.gptEncoder.encode(text, 'all').length;
+  }
+
+  /**
+   * Algorithm adapted from "6. Counting tokens for chat API calls" of
+   * https://github.com/openai/openai-cookbook/blob/main/examples/How_to_count_tokens_with_tiktoken.ipynb
+   *
+   * An additional 2 tokens need to be added for metadata after all messages have been counted.
+   *
+   * @param {*} message
+   */
+  getTokenCountForMessage(message) {
+    // Map each property of the message to the number of tokens it contains
+    const propertyTokenCounts = Object.entries(message).map(([key, value]) => {
+      // Count the number of tokens in the property value
+      const numTokens = this.getTokenCount(value);
+
+      // Subtract 1 token if the property key is 'name'
+      const adjustment = key === 'name' ? 1 : 0;
+      return numTokens - adjustment;
+    });
+
+    // Sum the number of tokens in all properties and add 4 for metadata
+    return propertyTokenCounts.reduce((a, b) => a + b, 4);
+  }
+
+  /**
+   * Iterate through messages, building an array based on the parentMessageId.
+   * Each message has an id and a parentMessageId. The parentMessageId is the id of the message that this message is a reply to.
+   * @param messages
+   * @param parentMessageId
+   * @returns {*[]} An array containing the messages in the order they should be displayed, starting with the root message.
+   */
+  static getMessagesForConversation(messages, parentMessageId) {
+    const orderedMessages = [];
+    let currentMessageId = parentMessageId;
+    while (currentMessageId) {
+      // eslint-disable-next-line no-loop-func
+      const message = messages.find((m) => m.messageId === currentMessageId);
+      if (!message) {
+        break;
+      }
+      orderedMessages.unshift(message);
+      currentMessageId = message.parentMessageId;
+    }
+
+    if (orderedMessages.length === 0) {
+      return [];
+    }
+
+    return orderedMessages.map((msg) => ({
+      isCreatedByUser: msg.isCreatedByUser,
+      content: msg.text
+    }));
+  }
+}
+
+module.exports = GoogleAgent;

api/app/stream.js

@@ -0,0 +1,59 @@
+const { Readable } = require('stream');
+
+class TextStream extends Readable {
+  constructor(text, options = {}) {
+    super(options);
+    this.text = text;
+    this.currentIndex = 0;
+    this.delay = options.delay || 20; // Time in milliseconds
+  }
+
+  _read() {
+    const minChunkSize = 2;
+    const maxChunkSize = 4;
+    const { delay } = this;
+
+    if (this.currentIndex < this.text.length) {
+      setTimeout(() => {
+        const remainingChars = this.text.length - this.currentIndex;
+        const chunkSize = Math.min(this.randomInt(minChunkSize, maxChunkSize + 1), remainingChars);
+
+        const chunk = this.text.slice(this.currentIndex, this.currentIndex + chunkSize);
+        this.push(chunk);
+        this.currentIndex += chunkSize;
+      }, delay);
+    } else {
+      this.push(null); // signal end of data
+    }
+  }
+
+  randomInt(min, max) {
+    return Math.floor(Math.random() * (max - min)) + min;
+  }
+
+  async processTextStream(onProgressCallback) {
+    const streamPromise = new Promise((resolve, reject) => {
+      this.on('data', (chunk) => {
+        onProgressCallback(chunk.toString());
+      });
+
+      this.on('end', () => {
+        console.log('Stream ended');
+        resolve();
+      });
+
+      this.on('error', (err) => {
+        reject(err);
+      });
+    });
+
+    try {
+      await streamPromise;
+    } catch (err) {
+      console.error('Error processing text stream:', err);
+      // Handle the error appropriately, e.g., return an error message or throw an error
+    }
+  }
+}
+
+module.exports = TextStream;

api/app/titleConvo.js

@@ -1,7 +1,8 @@
 const { Configuration, OpenAIApi } = require('openai');
 const _ = require('lodash');
+const { genAzureEndpoint } = require('../utils/genAzureEndpoints');
 
-const proxyEnvToAxiosProxy = proxyString => {
+const proxyEnvToAxiosProxy = (proxyString) => {
   if (!proxyString) return null;
 
   const regex = /^([^:]+):\/\/(?:([^:@]*):?([^:@]*)@)?([^:]+)(?::(\d+))?/;
@@ -16,51 +17,53 @@ const proxyEnvToAxiosProxy = proxyString => {
   return proxyConfig;
 };
 
-const titleConvo = async ({ endpoint, text, response }) => {
+const titleConvo = async ({ endpoint, text, response, oaiApiKey }) => {
   let title = 'New Chat';
-  const messages = [
-    {
-      role: 'system',
-      content:
-        // `You are a title-generator with one job: giving a conversation, detect the language and titling the conversation provided by a user, using the same language. The requirement are: 1. If possible, generate in 5 words or less, 2. Using title case, 3. must give the title using the language as the user said. 4. Don't refer to the participants of the conversation. 5. Do not include punctuation or quotation marks. 6. Your response should be in title case, exclusively containing the title. 7. don't say anything except the title.
-        `Detect user language and write in the same language an extremely concise title for this conversation, which you must accurately detect. Write in the detected language. Title in 5 Words or Less. No Punctuation/Quotation. All first letters of every word should be capitalized and complete only the title in User Language only.
-
-||>User:
-"${text}"
-||>Response:
-"${JSON.stringify(response?.text)}"
-
-||>Title:`
-    }
-    // {
-    //   role: 'user',
-    //   content: `User:\n "${text}"\n\n${model}: \n"${JSON.stringify(response?.text)}"\n\n`
-    // }
-  ];
-
-  // console.log('Title Prompt', messages[0]);
-
-  const request = {
-    model: 'gpt-3.5-turbo',
-    messages,
-    temperature: 0,
-    presence_penalty: 0,
-    frequency_penalty: 0
-  };
-
-  // console.log('REQUEST', request);
+  const ChatGPTClient = (await import('@waylaidwanderer/chatgpt-api')).default;
 
   try {
-    const configuration = new Configuration({
-      apiKey: process.env.OPENAI_KEY
-    });
-    const openai = new OpenAIApi(configuration);
-    const completion = await openai.createChatCompletion(request, {
-      proxy: proxyEnvToAxiosProxy(process.env.PROXY || null)
-    });
+    const instructionsPayload = {
+      role: 'system',
+      content: `Detect user language and write in the same language an extremely concise title for this conversation, which you must accurately detect. Write in the detected language. Title in 5 Words or Less. No Punctuation or Quotation. All first letters of every word should be capitalized and complete only the title in User Language only.
 
-    //eslint-disable-next-line
-    title = completion.data.choices[0].message.content.replace(/["\.]/g, '');
+    ||>User:
+    "${text}"
+    ||>Response:
+    "${JSON.stringify(response?.text)}"
+    
+    ||>Title:`
+    };
+
+    const azure = process.env.AZURE_OPENAI_API_KEY ? true : false;
+    const options = {
+      azure,
+      reverseProxyUrl: process.env.OPENAI_REVERSE_PROXY || null,
+      proxy: process.env.PROXY || null
+    };
+
+    const titleGenClientOptions = JSON.parse(JSON.stringify(options));
+
+    titleGenClientOptions.modelOptions = {
+      model: 'gpt-3.5-turbo',
+      temperature: 0,
+      presence_penalty: 0,
+      frequency_penalty: 0
+    };
+
+    let apiKey = oaiApiKey || process.env.OPENAI_KEY;
+
+    if (azure) {
+      apiKey = process.env.AZURE_OPENAI_API_KEY;
+      titleGenClientOptions.reverseProxyUrl = genAzureEndpoint({
+        azureOpenAIApiInstanceName: process.env.AZURE_OPENAI_API_INSTANCE_NAME,
+        azureOpenAIApiDeploymentName: process.env.AZURE_OPENAI_API_DEPLOYMENT_NAME,
+        azureOpenAIApiVersion: process.env.AZURE_OPENAI_API_VERSION
+      });
+    }
+
+    const titleGenClient = new ChatGPTClient(apiKey, titleGenClientOptions);
+    const result = await titleGenClient.getCompletion([instructionsPayload], null);
+    title = result.choices[0].message.content.replace(/\s+/g, ' ').replaceAll('"', '').trim();
   } catch (e) {
     console.error(e);
     console.log('There was an issue generating title, see error above');

api/lib/db/migrateDb.js

@@ -110,7 +110,7 @@ async function migrateDb() {
   ret[0] = await migrateToStrictFollowParentMessageIdChain();
   ret[1] = await migrateToSupportBetterCustomization();
 
-  const isMigrated = !!ret.find(element => !element?.noNeed);
+  const isMigrated = !!ret.find((element) => !element?.noNeed);
 
   if (!isMigrated) console.log('[Migrate] Nothing to migrate');
 }

api/lib/parse/getCitations.js

@@ -2,7 +2,8 @@
 const regex = / \[.*?]\(.*?\)/g;
 
 const getCitations = (res) => {
-  const textBlocks = res.details.adaptiveCards[0].body;
+  const adaptiveCards = res.details.adaptiveCards;
+  const textBlocks = adaptiveCards && adaptiveCards[0].body;
   if (!textBlocks) return '';
   let links = textBlocks[textBlocks.length - 1]?.text.match(regex);
   if (links?.length === 0 || !links) return '';
@@ -10,4 +11,4 @@ const getCitations = (res) => {
   return links.join('\n');
 };
 
-module.exports = getCitations;
+module.exports = getCitations;

api/lib/utils/mergeSort.js

@@ -2,11 +2,11 @@ function mergeSort(arr, compareFn) {
   if (arr.length <= 1) {
     return arr;
   }
-  
+
   const mid = Math.floor(arr.length / 2);
   const leftArr = arr.slice(0, mid);
   const rightArr = arr.slice(mid);
-  
+
   return merge(mergeSort(leftArr, compareFn), mergeSort(rightArr, compareFn), compareFn);
 }
 
@@ -14,7 +14,7 @@ function merge(leftArr, rightArr, compareFn) {
   const result = [];
   let leftIndex = 0;
   let rightIndex = 0;
-  
+
   while (leftIndex < leftArr.length && rightIndex < rightArr.length) {
     if (compareFn(leftArr[leftIndex], rightArr[rightIndex]) < 0) {
       result.push(leftArr[leftIndex++]);
@@ -22,8 +22,8 @@ function merge(leftArr, rightArr, compareFn) {
       result.push(rightArr[rightIndex++]);
     }
   }
-  
+
   return result.concat(leftArr.slice(leftIndex)).concat(rightArr.slice(rightIndex));
 }
 
-module.exports = mergeSort;
+module.exports = mergeSort;

api/middleware/requireJwtAuth.js

@@ -0,0 +1,5 @@
+const passport = require('passport');
+
+const requireJwtAuth = passport.authenticate('jwt', { session: false });
+
+module.exports = requireJwtAuth;

api/middleware/requireLocalAuth.js

@@ -0,0 +1,31 @@
+const passport = require('passport');
+const DebugControl = require('../utils/debug.js');
+
+function log({ title, parameters }) {
+  DebugControl.log.functionName(title);
+  if (parameters) {
+    DebugControl.log.parameters(parameters);
+  }
+}
+
+const requireLocalAuth = (req, res, next) => {
+  passport.authenticate('local', (err, user, info) => {
+    if (err) {
+      log({
+        title: '(requireLocalAuth) Error at passport.authenticate',
+        parameters: [{ name: 'error', value: err }]
+      });
+      return next(err);
+    }
+    if (!user) {
+      log({
+        title: '(requireLocalAuth) Error: No user'
+      });
+      return res.status(422).send(info);
+    }
+    req.user = user;
+    next();
+  })(req, res, next);
+};
+
+module.exports = requireLocalAuth;

api/models/Conversation.js

@@ -13,48 +13,24 @@ const getConvo = async (user, conversationId) => {
 
 module.exports = {
   Conversation,
-  saveConvo: async (user, { conversationId, newConversationId, title, ...convo }) => {
+  saveConvo: async (user, { conversationId, newConversationId, ...convo }) => {
     try {
       const messages = await getMessages({ conversationId });
-      const update = { ...convo, messages };
-      if (title) {
-        update.title = title;
-        update.user = user;
-      }
+      const update = { ...convo, messages, user };
       if (newConversationId) {
         update.conversationId = newConversationId;
       }
-      if (!update.jailbreakConversationId) {
-        update.jailbreakConversationId = null;
-      }
 
-      return await Conversation.findOneAndUpdate(
-        { conversationId: conversationId, user },
-        { $set: update },
-        { new: true, upsert: true }
-      ).exec();
+      return await Conversation.findOneAndUpdate({ conversationId: conversationId, user }, update, {
+        new: true,
+        upsert: true
+      }).exec();
     } catch (error) {
       console.log(error);
       return { message: 'Error saving conversation' };
     }
   },
-  updateConvo: async (user, { conversationId, oldConvoId, ...update }) => {
-    try {
-      let convoId = conversationId;
-      if (oldConvoId) {
-        convoId = oldConvoId;
-        update.conversationId = conversationId;
-      }
-
-      return await Conversation.findOneAndUpdate({ conversationId: convoId, user }, update, {
-        new: true
-      }).exec();
-    } catch (error) {
-      console.log(error);
-      return { message: 'Error updating conversation' };
-    }
-  },
-  getConvosByPage: async (user, pageNumber = 1, pageSize = 12) => {
+  getConvosByPage: async (user, pageNumber = 1, pageSize = 14) => {
     try {
       const totalConvos = (await Conversation.countDocuments({ user })) || 1;
       const totalPages = Math.ceil(totalConvos / pageSize);
@@ -63,14 +39,13 @@ module.exports = {
         .skip((pageNumber - 1) * pageSize)
         .limit(pageSize)
         .exec();
-
       return { conversations: convos, pages: totalPages, pageNumber, pageSize };
     } catch (error) {
       console.log(error);
       return { message: 'Error getting conversations' };
     }
   },
-  getConvosQueried: async (user, convoIds, pageNumber = 1, pageSize = 12) => {
+  getConvosQueried: async (user, convoIds, pageNumber = 1, pageSize = 14) => {
     try {
       if (!convoIds || convoIds.length === 0) {
         return { conversations: [], pages: 1, pageNumber, pageSize };
@@ -82,7 +57,7 @@ module.exports = {
       // will handle a syncing solution soon
       const deletedConvoIds = [];
 
-      convoIds.forEach(convo =>
+      convoIds.forEach((convo) =>
         promises.push(
           Conversation.findOne({
             user,
@@ -145,7 +120,7 @@ module.exports = {
   },
   deleteConvos: async (user, filter) => {
     let toRemove = await Conversation.find({ ...filter, user }).select('conversationId');
-    const ids = toRemove.map(instance => instance.conversationId);
+    const ids = toRemove.map((instance) => instance.conversationId);
     let deleteCount = await Conversation.deleteMany({ ...filter, user }).exec();
     deleteCount.messages = await deleteMessages({ conversationId: { $in: ids } });
     return deleteCount;

api/models/Message.js

@@ -1,7 +1,9 @@
 const Message = require('./schema/messageSchema');
+
 module.exports = {
   Message,
-  saveMessage: async ({
+
+  async saveMessage({
     messageId,
     newMessageId,
     conversationId,
@@ -9,9 +11,12 @@ module.exports = {
     sender,
     text,
     isCreatedByUser = false,
-    error
-  }) => {
+    error,
+    unfinished,
+    cancelled
+  }) {
     try {
+      // may also need to update the conversation here
       await Message.findOneAndUpdate(
         { messageId },
         {
@@ -21,43 +26,57 @@ module.exports = {
           sender,
           text,
           isCreatedByUser,
-          error
+          error,
+          unfinished,
+          cancelled
         },
         { upsert: true, new: true }
       );
-      return { messageId, conversationId, parentMessageId, sender, text, isCreatedByUser };
-    } catch (error) {
-      console.error(error);
-      return { message: 'Error saving message' };
+
+      return {
+        messageId,
+        conversationId,
+        parentMessageId,
+        sender,
+        text,
+        isCreatedByUser
+      };
+    } catch (err) {
+      console.error(`Error saving message: ${err}`);
+      throw new Error('Failed to save message.');
     }
   },
-  deleteMessagesSince: async ({ messageId, conversationId }) => {
+
+  async deleteMessagesSince({ messageId, conversationId }) {
     try {
       const message = await Message.findOne({ messageId }).exec();
 
-      if (message)
+      if (message) {
         return await Message.find({ conversationId })
           .deleteMany({ createdAt: { $gt: message.createdAt } })
           .exec();
-    } catch (error) {
-      console.error(error);
-      return { message: 'Error deleting messages' };
+      }
+    } catch (err) {
+      console.error(`Error deleting messages: ${err}`);
+      throw new Error('Failed to delete messages.');
     }
   },
-  getMessages: async filter => {
+
+  async getMessages(filter) {
     try {
       return await Message.find(filter).sort({ createdAt: 1 }).exec();
-    } catch (error) {
-      console.error(error);
-      return { message: 'Error getting messages' };
+    } catch (err) {
+      console.error(`Error getting messages: ${err}`);
+      throw new Error('Failed to get messages.');
     }
   },
-  deleteMessages: async filter => {
+
+  async deleteMessages(filter) {
     try {
       return await Message.deleteMany(filter).exec();
-    } catch (error) {
-      console.error(error);
-      return { message: 'Error deleting messages' };
+    } catch (err) {
+      console.error(`Error deleting messages: ${err}`);
+      throw new Error('Failed to delete messages.');
     }
   }
 };

api/models/Preset.js

@@ -39,7 +39,7 @@ module.exports = {
   },
   deletePresets: async (user, filter) => {
     let toRemove = await Preset.find({ ...filter, user }).select('presetId');
-    const ids = toRemove.map(instance => instance.presetId);
+    const ids = toRemove.map((instance) => instance.presetId);
     let deleteCount = await Preset.deleteMany({ ...filter, user }).exec();
     return deleteCount;
   }

api/models/Prompt.js

@@ -1,18 +1,21 @@
 const mongoose = require('mongoose');
 
-const promptSchema = mongoose.Schema({
-  title: {
-    type: String,
-    required: true
+const promptSchema = mongoose.Schema(
+  {
+    title: {
+      type: String,
+      required: true
+    },
+    prompt: {
+      type: String,
+      required: true
+    },
+    category: {
+      type: String
+    }
   },
-  prompt: {
-    type: String,
-    required: true
-  },
-  category: {
-    type: String,
-  },
-}, { timestamps: true });
+  { timestamps: true }
+);
 
 const Prompt = mongoose.models.Prompt || mongoose.model('Prompt', promptSchema);
 
@@ -31,7 +34,7 @@ module.exports = {
   },
   getPrompts: async (filter) => {
     try {
-      return await Prompt.find(filter).exec()
+      return await Prompt.find(filter).exec();
     } catch (error) {
       console.error(error);
       return { prompt: 'Error getting prompts' };
@@ -39,10 +42,10 @@ module.exports = {
   },
   deletePrompts: async (filter) => {
     try {
-      return await Prompt.deleteMany(filter).exec()
+      return await Prompt.deleteMany(filter).exec();
     } catch (error) {
       console.error(error);
       return { prompt: 'Error deleting prompts' };
     }
   }
-}
+};

api/models/User.js

@@ -0,0 +1,176 @@
+const mongoose = require('mongoose');
+const bcrypt = require('bcryptjs');
+const jwt = require('jsonwebtoken');
+const Joi = require('joi');
+const DebugControl = require('../utils/debug.js');
+
+function log({ title, parameters }) {
+  DebugControl.log.functionName(title);
+  DebugControl.log.parameters(parameters);
+}
+
+const Session = mongoose.Schema({
+  refreshToken: {
+    type: String,
+    default: ''
+  }
+});
+
+const userSchema = mongoose.Schema(
+  {
+    name: {
+      type: String
+    },
+    username: {
+      type: String,
+      lowercase: true,
+      required: [true, "can't be blank"],
+      match: [/^[a-zA-Z0-9_]+$/, 'is invalid'],
+      index: true
+    },
+    email: {
+      type: String,
+      required: [true, "can't be blank"],
+      lowercase: true,
+      unique: true,
+      match: [/\S+@\S+\.\S+/, 'is invalid'],
+      index: true
+    },
+    emailVerified: {
+      type: Boolean,
+      required: true,
+      default: false
+    },
+    password: {
+      type: String,
+      trim: true,
+      minlength: 8,
+      maxlength: 60
+    },
+    avatar: {
+      type: String,
+      required: false
+    },
+    provider: {
+      type: String,
+      required: true,
+      default: 'local'
+    },
+    role: {
+      type: String,
+      default: 'USER'
+    },
+    googleId: {
+      type: String,
+      unique: true,
+      sparse: true
+    },
+    facebookId: {
+      type: String,
+      unique: true,
+      sparse: true
+    },
+    refreshToken: {
+      type: [Session]
+    }
+  },
+  { timestamps: true }
+);
+
+//Remove refreshToken from the response
+userSchema.set('toJSON', {
+  transform: function (_doc, ret,) {
+    delete ret.refreshToken;
+    return ret;
+  }
+});
+
+userSchema.methods.toJSON = function () {
+  return {
+    id: this._id,
+    provider: this.provider,
+    email: this.email,
+    name: this.name,
+    username: this.username,
+    avatar: this.avatar,
+    role: this.role,
+    emailVerified: this.emailVerified,
+    createdAt: this.createdAt,
+    updatedAt: this.updatedAt
+  };
+};
+
+const isProduction = process.env.NODE_ENV === 'production';
+const secretOrKey = isProduction ? process.env.JWT_SECRET_PROD : process.env.JWT_SECRET_DEV;
+const refreshSecret = isProduction
+  ? process.env.REFRESH_TOKEN_SECRET_PROD
+  : process.env.REFRESH_TOKEN_SECRET_DEV;
+
+userSchema.methods.generateToken = function () {
+  const token = jwt.sign(
+    {
+      id: this._id,
+      username: this.username,
+      provider: this.provider,
+      email: this.email
+    },
+    secretOrKey,
+    { expiresIn: eval(process.env.SESSION_EXPIRY) }
+  );
+  return token;
+};
+
+userSchema.methods.generateRefreshToken = function () {
+  const refreshToken = jwt.sign(
+    {
+      id: this._id,
+      username: this.username,
+      provider: this.provider,
+      email: this.email
+    },
+    refreshSecret,
+    { expiresIn: eval(process.env.REFRESH_TOKEN_EXPIRY) }
+  );
+  return refreshToken;
+};
+
+userSchema.methods.comparePassword = function (candidatePassword, callback) {
+  bcrypt.compare(candidatePassword, this.password, (err, isMatch) => {
+    if (err) return callback(err);
+    callback(null, isMatch);
+  });
+};
+
+module.exports.hashPassword = async (password) => {
+  const hashedPassword = await new Promise((resolve, reject) => {
+    bcrypt.hash(password, 10, function (err, hash) {
+      if (err) reject(err);
+      else resolve(hash);
+    });
+  });
+
+  return hashedPassword;
+};
+
+module.exports.validateUser = (user) => {
+  log({
+    title: 'Validate User',
+    parameters: [{ name: 'Validate User', value: user }]
+  });
+  const schema = {
+    avatar: Joi.any(),
+    name: Joi.string().min(2).max(80).required(),
+    username: Joi.string()
+      .min(2)
+      .max(80)
+      .regex(/^[a-zA-Z0-9_]+$/)
+      .required(),
+    password: Joi.string().min(8).max(60).allow('').allow(null)
+  };
+
+  return schema.validate(user);
+};
+
+const User = mongoose.model('User', userSchema);
+
+module.exports = User;

api/models/index.js

@@ -1,5 +1,5 @@
 const { getMessages, saveMessage, deleteMessagesSince, deleteMessages } = require('./Message');
-const { getConvoTitle, getConvo, saveConvo, updateConvo } = require('./Conversation');
+const { getConvoTitle, getConvo, saveConvo } = require('./Conversation');
 const { getPreset, getPresets, savePreset, deletePresets } = require('./Preset');
 
 module.exports = {
@@ -11,7 +11,6 @@ module.exports = {
   getConvoTitle,
   getConvo,
   saveConvo,
-  updateConvo,
 
   getPreset,
   getPresets,

api/models/plugins/mongoMeili.js

@@ -19,10 +19,7 @@ const createMeiliMongooseModel = function ({ index, indexName, client, attribute
     static async clearMeiliIndex() {
       await index.delete();
       // await index.deleteAllDocuments();
-      await this.collection.updateMany(
-        { _meiliIndex: true },
-        { $set: { _meiliIndex: false } }
-      );
+      await this.collection.updateMany({ _meiliIndex: true }, { $set: { _meiliIndex: false } });
     }
 
     static async resetIndex() {
@@ -57,7 +54,7 @@ const createMeiliMongooseModel = function ({ index, indexName, client, attribute
         // Find objects into mongodb matching `objectID` from Meili search
         const query = {};
         // query[primaryKey] = { $in: _.map(data.hits, primaryKey) };
-        query[primaryKey] = _.map(data.hits, hit => cleanUpPrimaryKeyValue(hit[primaryKey]));
+        query[primaryKey] = _.map(data.hits, (hit) => cleanUpPrimaryKeyValue(hit[primaryKey]));
         // console.log('query', query);
         const hitsFromMongoose = await this.find(
           query,
@@ -67,7 +64,7 @@ const createMeiliMongooseModel = function ({ index, indexName, client, attribute
               return { ...results, [key]: 1 };
             },
             { _id: 1 }
-          ),
+          )
         );
 
         // Add additional data from mongodb into Meili search hits
@@ -198,8 +195,8 @@ module.exports = function mongoMeili(schema, options) {
     if (Object.prototype.hasOwnProperty.call(schema.obj, 'messages')) {
       console.log('Syncing convos...');
       mongoose.model('Conversation').syncWithMeili();
-    } 
-    
+    }
+
     if (Object.prototype.hasOwnProperty.call(schema.obj, 'messageId')) {
       console.log('Syncing messages...');
       mongoose.model('Message').syncWithMeili();

api/models/schema/conversationPreset.js

@@ -17,6 +17,12 @@ module.exports = {
     default: null,
     required: false
   },
+  // for google only
+  modelLabel: {
+    type: String,
+    default: null,
+    required: false
+  },
   promptPrefix: {
     type: String,
     default: null,
@@ -32,6 +38,22 @@ module.exports = {
     default: 1,
     required: false
   },
+  // for google only
+  topP: {
+    type: Number,
+    default: 0.95,
+    required: false
+  },
+  topK: {
+    type: Number,
+    default: 40,
+    required: false
+  },
+  maxOutputTokens: {
+    type: Number,
+    default: 1024,
+    required: false
+  },
   presence_penalty: {
     type: Number,
     default: 0,

api/models/schema/convoSchema.js

@@ -20,6 +20,8 @@ const convoSchema = mongoose.Schema(
       default: null
     },
     messages: [{ type: mongoose.Schema.Types.ObjectId, ref: 'Message' }],
+    // google only
+    examples: [{ type: mongoose.Schema.Types.Mixed }],
     ...conversationPreset,
     // for bingAI only
     jailbreakConversationId: {

api/models/schema/messageSchema.js

@@ -43,6 +43,14 @@ const messageSchema = mongoose.Schema(
       required: true,
       default: false
     },
+    unfinished: {
+      type: Boolean,
+      default: false
+    },
+    cancelled: {
+      type: Boolean,
+      default: false
+    },
     error: {
       type: Boolean,
       default: false

api/models/schema/presetSchema.js

@@ -17,6 +17,8 @@ const presetSchema = mongoose.Schema(
       type: String,
       default: null
     },
+    // google only
+    examples: [{ type: mongoose.Schema.Types.Mixed }],
     ...conversationPreset
   },
   { timestamps: true }

api/models/schema/tokenSchema.js

@@ -0,0 +1,22 @@
+const mongoose = require('mongoose');
+const Schema = mongoose.Schema;
+
+const tokenSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    required: true,
+    ref: 'user'
+  },
+  token: {
+    type: String,
+    required: true
+  },
+  createdAt: {
+    type: Date,
+    required: true,
+    default: Date.now,
+    expires: 900
+  }
+});
+
+module.exports = mongoose.model('Token', tokenSchema);

api/package.json

@@ -1,6 +1,6 @@
 {
-  "name": "chatgpt-clone",
-  "version": "0.3.0",
+  "name": "chat-backend",
+  "version": "0.4.6",
   "description": "",
   "main": "server/index.js",
   "scripts": {
@@ -21,21 +21,35 @@
   "dependencies": {
     "@dqbd/tiktoken": "^1.0.2",
     "@keyv/mongo": "^2.1.8",
-    "@waylaidwanderer/chatgpt-api": "^1.33.2",
+    "@waylaidwanderer/chatgpt-api": "^1.36.0",
     "axios": "^1.3.4",
+    "bcrypt": "^5.1.0",
+    "bcryptjs": "^2.4.3",
+    "cookie": "^0.5.0",
+    "cookie-parser": "^1.4.6",
     "cors": "^2.8.5",
     "crypto": "^1.0.1",
     "dotenv": "^16.0.3",
     "eslint": "^8.36.0",
     "express": "^4.18.2",
-    "express-session": "^1.17.3",
+    "googleapis": "^118.0.0",
+    "handlebars": "^4.7.7",
     "html": "^1.0.0",
+    "joi": "^17.9.2",
+    "jsonwebtoken": "^9.0.0",
     "keyv": "^4.5.2",
     "keyv-file": "^0.2.0",
     "lodash": "^4.17.21",
-    "meilisearch": "^0.31.1",
-    "mongoose": "^6.9.0",
+    "meilisearch": "^0.32.3",
+    "mongoose": "^7.1.1",
+    "nodemailer": "^6.9.1",
     "openai": "^3.1.0",
+    "passport": "^0.6.0",
+    "passport-facebook": "^3.0.0",
+    "passport-google-oauth20": "^2.0.0",
+    "passport-jwt": "^4.0.1",
+    "passport-local": "^1.0.0",
+    "pino": "^8.12.1",
     "sanitize": "^2.1.2"
   },
   "devDependencies": {

api/server/controllers/auth.controller.js

@@ -0,0 +1,168 @@
+const {
+  loginUser,
+  logoutUser,
+  registerUser,
+  requestPasswordReset,
+  resetPassword
+} = require('../services/auth.service');
+
+const isProduction = process.env.NODE_ENV === 'production';
+
+const loginController = async (req, res) => {
+  try {
+    const token = req.user.generateToken();
+    const user = await loginUser(req.user);
+    if (user) {
+      res.cookie('token', token, {
+        expires: new Date(Date.now() + eval(process.env.SESSION_EXPIRY)),
+        httpOnly: false,
+        secure: isProduction
+      });
+      res.status(200).send({ token, user });
+    } else {
+      return res.status(400).json({ message: 'Invalid credentials' });
+    }
+  } catch (err) {
+    console.log(err);
+    return res.status(500).json({ message: err.message });
+  }
+};
+
+const logoutController = async (req, res) => {
+  const { signedCookies = {} } = req;
+  const { refreshToken } = signedCookies;
+  try {
+    const logout = await logoutUser(req.user, refreshToken);
+    console.log(logout);
+    const { status, message } = logout;
+    if (status === 200) {
+      res.clearCookie('token');
+      res.clearCookie('refreshToken');
+      res.status(status).send({ message });
+    } else {
+      res.status(status).send({ message });
+    }
+  } catch (err) {
+    console.log(err);
+    return res.status(500).json({ message: err.message });
+  }
+};
+
+const registrationController = async (req, res) => {
+  try {
+    const response = await registerUser(req.body);
+    if (response.status === 200) {
+      const { status, user } = response;
+      const token = user.generateToken();
+      //send token for automatic login
+      res.cookie('token', token, {
+        expires: new Date(Date.now() + eval(process.env.SESSION_EXPIRY)),
+        httpOnly: false,
+        secure: isProduction
+      });
+      res.status(status).send({ user });
+    } else {
+      const { status, message } = response;
+      res.status(status).send({ message });
+    }
+  } catch (err) {
+    console.log(err);
+    return res.status(500).json({ message: err.message });
+  }
+};
+
+const getUserController = async (req, res) => {
+  return res.status(200).send(req.user);
+};
+
+const resetPasswordRequestController = async (req, res) => {
+  try {
+    const resetService = await requestPasswordReset(req.body.email);
+    if (resetService.link) {
+      return res.status(200).json(resetService);
+    } else {
+      return res.status(400).json(resetService);
+    }
+  } catch (e) {
+    console.log(e);
+    return res.status(400).json({ message: e.message });
+  }
+};
+
+const resetPasswordController = async (req, res) => {
+  try {
+    const resetPasswordService = await resetPassword(
+      req.body.userId,
+      req.body.token,
+      req.body.password
+    );
+    if (resetPasswordService instanceof Error) {
+      return res.status(400).json(resetPasswordService);
+    } else {
+      return res.status(200).json(resetPasswordService);
+    }
+  } catch (e) {
+    console.log(e);
+    return res.status(400).json({ message: e.message });
+  }
+};
+
+const refreshController = async (req, res, next) => {
+  const { signedCookies = {} } = req;
+  const { refreshToken } = signedCookies;
+  //TODO
+  // if (refreshToken) {
+  //   try {
+  //     const payload = jwt.verify(refreshToken, process.env.REFRESH_TOKEN_SECRET);
+  //     const userId = payload._id;
+  //     User.findOne({ _id: userId }).then(
+  //       (user) => {
+  //         if (user) {
+  //           // Find the refresh token against the user record in database
+  //           const tokenIndex = user.refreshToken.findIndex(item => item.refreshToken === refreshToken);
+
+  //           if (tokenIndex === -1) {
+  //             res.statusCode = 401;
+  //             res.send('Unauthorized');
+  //           } else {
+  //             const token = req.user.generateToken();
+  //             // If the refresh token exists, then create new one and replace it.
+  //             const newRefreshToken = req.user.generateRefreshToken();
+  //             user.refreshToken[tokenIndex] = { refreshToken: newRefreshToken };
+  //             user.save((err) => {
+  //               if (err) {
+  //                 res.statusCode = 500;
+  //                 res.send(err);
+  //               } else {
+  //               //  setTokenCookie(res, newRefreshToken);
+  //                 const user = req.user.toJSON();
+  //                 res.status(200).send({ token, user });
+  //               }
+  //             });
+  //           }
+  //         } else {
+  //           res.statusCode = 401;
+  //           res.send('Unauthorized');
+  //         }
+  //       },
+  //       err => next(err)
+  //     );
+  //   } catch (err) {
+  //     res.statusCode = 401;
+  //     res.send('Unauthorized');
+  //   }
+  // } else {
+  //   res.statusCode = 401;
+  //   res.send('Unauthorized');
+  // }
+};
+
+module.exports = {
+  getUserController,
+  loginController,
+  logoutController,
+  refreshController,
+  registrationController,
+  resetPasswordRequestController,
+  resetPasswordController
+};

api/server/controllers/error.controller.js

@@ -10,8 +10,8 @@ const handleDuplicateKeyError = (err, res) => {
 //handle validation errors
 const handleValidationError = (err, res) => {
   console.log('congrats you hit the validation middleware');
-  let errors = Object.values(err.errors).map(el => el.message);
-  let fields = Object.values(err.errors).map(el => el.path);
+  let errors = Object.values(err.errors).map((el) => el.message);
+  let fields = Object.values(err.errors).map((el) => el.path);
   let code = 400;
   if (errors.length > 1) {
     const formattedErrors = errors.join(' ');

api/server/index.js

@@ -1,12 +1,12 @@
 const express = require('express');
-const session = require('express-session');
 const connectDb = require('../lib/db/connectDb');
 const migrateDb = require('../lib/db/migrateDb');
 const indexSync = require('../lib/db/indexSync');
 const path = require('path');
 const cors = require('cors');
 const routes = require('./routes');
-const errorController = require('./controllers/errorController');
+const errorController = require('./controllers/error.controller');
+const passport = require('passport');
 
 const port = process.env.PORT || 3080;
 const host = process.env.HOST || 'localhost';
@@ -20,44 +20,36 @@ const projectPath = path.join(__dirname, '..', '..', 'client');
 
   const app = express();
   app.use(errorController);
-  app.use(cors());
   app.use(express.json());
+  app.use(express.urlencoded({ extended: true }));
   app.use(express.static(path.join(projectPath, 'dist')));
   app.set('trust proxy', 1); // trust first proxy
-  app.use(
-    session({
-      secret: 'chatgpt-clone-random-secrect',
-      resave: false,
-      saveUninitialized: true,
-      cookie: { maxAge: 7 * 24 * 60 * 60 * 1000 } // 7 days
-    })
-  );
-
-  // ROUTES
-
-  /* chore: potential redirect error here, can only comment out this block; 
-    comment back in if using auth routes i guess */
-  // app.get('/', routes.authenticatedOrRedirect, function (req, res) {
-  //   console.log(path.join(projectPath, 'public', 'index.html'));
-  //   res.sendFile(path.join(projectPath, 'public', 'index.html'));
-  // });
+  app.use(cors());
 
+  // OAUTH
+  app.use(passport.initialize());
+  require('../strategies/jwtStrategy');
+  require('../strategies/localStrategy');
+  if (process.env.GOOGLE_CLIENT_ID && process.env.GOOGLE_CLIENT_SECRET) {
+    require('../strategies/googleStrategy');
+  }
+  if (process.env.FACEBOOK_CLIENT_ID && process.env.FACEBOOK_CLIENT_SECRET) {
+    require('../strategies/facebookStrategy');
+  }
+  app.use('/oauth', routes.oauth);
   // api endpoint
-  app.use('/api/search', routes.authenticatedOr401, routes.search);
-  app.use('/api/ask', routes.authenticatedOr401, routes.ask);
-  app.use('/api/messages', routes.authenticatedOr401, routes.messages);
-  app.use('/api/convos', routes.authenticatedOr401, routes.convos);
-  app.use('/api/presets', routes.authenticatedOr401, routes.presets);
-  app.use('/api/prompts', routes.authenticatedOr401, routes.prompts);
-  app.use('/api/tokenizer', routes.authenticatedOr401, routes.tokenizer);
-  app.use('/api/endpoints', routes.authenticatedOr401, routes.endpoints);
-
-  // user system
-  app.use('/auth', routes.auth);
-  app.use('/api/me', routes.me);
+  app.use('/api/auth', routes.auth);
+  app.use('/api/search', routes.search);
+  app.use('/api/ask', routes.ask);
+  app.use('/api/messages', routes.messages);
+  app.use('/api/convos', routes.convos);
+  app.use('/api/presets', routes.presets);
+  app.use('/api/prompts', routes.prompts);
+  app.use('/api/tokenizer', routes.tokenizer);
+  app.use('/api/endpoints', routes.endpoints);
 
   // static files
-  app.get('/*', routes.authenticatedOrRedirect, function (req, res) {
+  app.get('/*', function (req, res) {
     res.sendFile(path.join(projectPath, 'dist', 'index.html'));
   });
 
@@ -66,12 +58,13 @@ const projectPath = path.join(__dirname, '..', '..', 'client');
       console.log(
         `Server listening on all interface at port ${port}. Use http://localhost:${port} to access it`
       );
-    else console.log(`Server listening at http://${host == '0.0.0.0' ? 'localhost' : host}:${port}`);
+    else
+      console.log(`Server listening at http://${host == '0.0.0.0' ? 'localhost' : host}:${port}`);
   });
 })();
 
 let messageCount = 0;
-process.on('uncaughtException', err => {
+process.on('uncaughtException', (err) => {
   if (!err.message.includes('fetch failed')) {
     console.error('There was an uncaught error:', err.message);
   }

api/server/routes/ask/addToCache.js

@@ -0,0 +1,64 @@
+const Keyv = require('keyv');
+const { KeyvFile } = require('keyv-file');
+
+const addToCache = async ({ endpoint, endpointOption, userMessage, responseMessage }) => {
+  try {
+    const conversationsCache = new Keyv({
+      store: new KeyvFile({ filename: './data/cache.json' }),
+      namespace: 'chatgpt' // should be 'bing' for bing/sydney
+    });
+
+    const {
+      conversationId,
+      messageId: userMessageId,
+      parentMessageId: userParentMessageId,
+      text: userText
+    } = userMessage;
+    const {
+      messageId: responseMessageId,
+      parentMessageId: responseParentMessageId,
+      text: responseText
+    } = responseMessage;
+
+    let conversation = await conversationsCache.get(conversationId);
+    // used to generate a title for the conversation if none exists
+    // let isNewConversation = false;
+    if (!conversation) {
+      conversation = {
+        messages: [],
+        createdAt: Date.now()
+      };
+      // isNewConversation = true;
+    }
+
+    const roles = (options) => {
+      if (endpoint === 'openAI') {
+        return options?.chatGptLabel || 'ChatGPT';
+      } else if (endpoint === 'bingAI') {
+        return options?.jailbreak ? 'Sydney' : 'BingAI';
+      }
+    };
+
+    let _userMessage = {
+      id: userMessageId,
+      parentMessageId: userParentMessageId,
+      role: 'User',
+      message: userText
+    };
+
+    let _responseMessage = {
+      id: responseMessageId,
+      parentMessageId: responseParentMessageId,
+      role: roles(endpointOption),
+      message: responseText
+    };
+
+    conversation.messages.push(_userMessage, _responseMessage);
+
+    await conversationsCache.set(conversationId, conversation);
+  } catch (error) {
+    console.error('Trouble adding to cache', error);
+  }
+};
+
+module.exports = addToCache;

api/server/routes/ask/askBingAI.js

@@ -4,8 +4,9 @@ const router = express.Router();
 const { titleConvo, askBing } = require('../../../app');
 const { saveMessage, getConvoTitle, saveConvo, getConvo } = require('../../../models');
 const { handleError, sendMessage, createOnProgress, handleText } = require('./handlers');
+const requireJwtAuth = require('../../../middleware/requireJwtAuth');
 
-router.post('/', async (req, res) => {
+router.post('/', requireJwtAuth, async (req, res) => {
   const {
     endpoint,
     text,
@@ -35,21 +36,23 @@ router.post('/', async (req, res) => {
   let endpointOption = {};
   if (req.body?.jailbreak)
     endpointOption = {
-      jailbreak: req.body?.jailbreak || false,
-      jailbreakConversationId: req.body?.jailbreakConversationId || null,
-      systemMessage: req.body?.systemMessage || null,
-      context: req.body?.context || null,
-      toneStyle: req.body?.toneStyle || 'fast'
+      jailbreak: req.body?.jailbreak ?? false,
+      jailbreakConversationId: req.body?.jailbreakConversationId ?? null,
+      systemMessage: req.body?.systemMessage ?? null,
+      context: req.body?.context ?? null,
+      toneStyle: req.body?.toneStyle ?? 'fast',
+      token: req.body?.token ?? null
     };
   else
     endpointOption = {
-      jailbreak: req.body?.jailbreak || false,
-      systemMessage: req.body?.systemMessage || null,
-      context: req.body?.context || null,
-      conversationSignature: req.body?.conversationSignature || null,
-      clientId: req.body?.clientId || null,
-      invocationId: req.body?.invocationId || null,
-      toneStyle: req.body?.toneStyle || 'fast'
+      jailbreak: req.body?.jailbreak ?? false,
+      systemMessage: req.body?.systemMessage ?? null,
+      context: req.body?.context ?? null,
+      conversationSignature: req.body?.conversationSignature ?? null,
+      clientId: req.body?.clientId ?? null,
+      invocationId: req.body?.invocationId ?? null,
+      toneStyle: req.body?.toneStyle ?? 'fast',
+      token: req.body?.token ?? null
     };
 
   console.log('ask log', {
@@ -60,7 +63,7 @@ router.post('/', async (req, res) => {
 
   if (!overrideParentMessageId) {
     await saveMessage(userMessage);
-    await saveConvo(req?.session?.user?.username, {
+    await saveConvo(req.user.id, {
       ...userMessage,
       ...endpointOption,
       conversationId,
@@ -93,6 +96,8 @@ const ask = async ({
 }) => {
   let { text, parentMessageId: userParentMessageId, messageId: userMessageId } = userMessage;
 
+  let responseMessageId = crypto.randomUUID();
+
   res.writeHead(200, {
     Connection: 'keep-alive',
     'Content-Type': 'text/event-stream',
@@ -104,9 +109,26 @@ const ask = async ({
   if (preSendRequest) sendMessage(res, { message: userMessage, created: true });
 
   try {
-    const progressCallback = createOnProgress();
+    let lastSavedTimestamp = 0;
+    const { onProgress: progressCallback, getPartialText } = createOnProgress({
+      onProgress: ({ text }) => {
+        const currentTimestamp = Date.now();
+        if (currentTimestamp - lastSavedTimestamp > 500) {
+          lastSavedTimestamp = currentTimestamp;
+          saveMessage({
+            messageId: responseMessageId,
+            sender: endpointOption?.jailbreak ? 'Sydney' : 'BingAI',
+            conversationId,
+            parentMessageId: overrideParentMessageId || userMessageId,
+            text: text,
+            unfinished: true,
+            cancelled: false,
+            error: false
+          });
+        }
+      }
+    });
     const abortController = new AbortController();
-    res.on('close', () => abortController.abort());
     let response = await askBing({
       text,
       parentMessageId: userParentMessageId,
@@ -122,33 +144,34 @@ const ask = async ({
 
     console.log('BING RESPONSE', response);
 
+    const newConversationId = endpointOption?.jailbreak
+      ? response.jailbreakConversationId
+      : response.conversationId || conversationId;
+    const newUserMassageId =
+      response.parentMessageId || response.details.requestId || userMessageId;
+    const newResponseMessageId = response.messageId || response.details.messageId;
+
     // STEP1 generate response message
-    response.text = response.response || response.details.spokenText || '**Bing refused to answer.**';
+    response.text =
+      response.response || response.details.spokenText || '**Bing refused to answer.**';
 
     let responseMessage = {
+      conversationId: newConversationId,
+      messageId: responseMessageId,
+      newMessageId: newResponseMessageId,
+      parentMessageId: overrideParentMessageId || newUserMassageId,
+      sender: endpointOption?.jailbreak ? 'Sydney' : 'BingAI',
       text: await handleText(response, true),
       suggestions:
-        response.details.suggestedResponses && response.details.suggestedResponses.map(s => s.text),
-      jailbreak: endpointOption?.jailbreak
+        response.details.suggestedResponses &&
+        response.details.suggestedResponses.map((s) => s.text),
+      unfinished: false,
+      cancelled: false,
+      error: false
     };
-    // // response.text = await handleText(response, true);
-    // response.suggestions =
-    //   response.details.suggestedResponses && response.details.suggestedResponses.map(s => s.text);
-
-    if (endpointOption?.jailbreak) {
-      responseMessage.conversationId = response.jailbreakConversationId;
-      responseMessage.messageId = response.messageId || response.details.messageId;
-      responseMessage.parentMessageId = overrideParentMessageId || response.parentMessageId || userMessageId;
-      responseMessage.sender = 'Sydney';
-    } else {
-      responseMessage.conversationId = response.conversationId;
-      responseMessage.messageId = response.messageId || response.details.messageId;
-      responseMessage.parentMessageId =
-        overrideParentMessageId || response.parentMessageId || response.details.requestId || userMessageId;
-      responseMessage.sender = 'BingAI';
-    }
 
     await saveMessage(responseMessage);
+    responseMessage.messageId = newResponseMessageId;
 
     // STEP2 update the convosation.
 
@@ -159,14 +182,22 @@ const ask = async ({
     // Attition: the api will also create new conversationId while using invalid userMessage.parentMessageId,
     // but in this situation, don't change the conversationId, but create new convo.
 
-    let conversationUpdate = { conversationId, endpoint: 'bingAI' };
-    if (conversationId != responseMessage.conversationId && isNewConversation)
-      conversationUpdate = {
-        ...conversationUpdate,
-        conversationId: conversationId,
-        newConversationId: responseMessage.conversationId || conversationId
-      };
-    conversationId = responseMessage.conversationId || conversationId;
+    let conversationUpdate = { conversationId: newConversationId, endpoint: 'bingAI' };
+    if (conversationId != newConversationId)
+      if (isNewConversation) {
+        // change the conversationId to new one
+        conversationUpdate = {
+          ...conversationUpdate,
+          conversationId: conversationId,
+          newConversationId: newConversationId
+        };
+      } else {
+        // create new conversation
+        conversationUpdate = {
+          ...conversationUpdate,
+          ...endpointOption
+        };
+      }
 
     if (endpointOption?.jailbreak) {
       conversationUpdate.jailbreak = true;
@@ -178,32 +209,39 @@ const ask = async ({
       conversationUpdate.invocationId = response.invocationId;
     }
 
-    await saveConvo(req?.session?.user?.username, conversationUpdate);
+    await saveConvo(req.user.id, conversationUpdate);
+    conversationId = newConversationId;
 
     // STEP3 update the user message
-    userMessage.conversationId = conversationId;
-    userMessage.messageId = responseMessage.parentMessageId;
+    userMessage.conversationId = newConversationId;
+    userMessage.messageId = newUserMassageId;
 
     // If response has parentMessageId, the fake userMessage.messageId should be updated to the real one.
-    if (!overrideParentMessageId) {
-      const oldUserMessageId = userMessageId;
-      await saveMessage({ ...userMessage, messageId: oldUserMessageId, newMessageId: userMessage.messageId });
-    }
-    userMessageId = userMessage.messageId;
+    if (!overrideParentMessageId)
+      await saveMessage({
+        ...userMessage,
+        messageId: userMessageId,
+        newMessageId: newUserMassageId
+      });
+    userMessageId = newUserMassageId;
 
     sendMessage(res, {
-      title: await getConvoTitle(req?.session?.user?.username, conversationId),
+      title: await getConvoTitle(req.user.id, conversationId),
       final: true,
-      conversation: await getConvo(req?.session?.user?.username, conversationId),
+      conversation: await getConvo(req.user.id, conversationId),
       requestMessage: userMessage,
       responseMessage: responseMessage
     });
     res.end();
 
     if (userParentMessageId == '00000000-0000-0000-0000-000000000000') {
-      const title = await titleConvo({ endpoint: endpointOption?.endpoint, text, response: responseMessage });
+      const title = await titleConvo({
+        endpoint: endpointOption?.endpoint,
+        text,
+        response: responseMessage
+      });
 
-      await saveConvo(req?.session?.user?.username, {
+      await saveConvo(req.user.id, {
         conversationId: conversationId,
         title
       });
@@ -211,10 +249,12 @@ const ask = async ({
   } catch (error) {
     console.log(error);
     const errorMessage = {
-      messageId: crypto.randomUUID(),
+      messageId: responseMessageId,
       sender: endpointOption?.jailbreak ? 'Sydney' : 'BingAI',
       conversationId,
       parentMessageId: overrideParentMessageId || userMessageId,
+      unfinished: false,
+      cancelled: false,
       error: true,
       text: error.message
     };

api/server/routes/ask/askChatGPTBrowser.js

@@ -2,11 +2,12 @@ const express = require('express');
 const crypto = require('crypto');
 const router = express.Router();
 const { getChatGPTBrowserModels } = require('../endpoints');
-const { titleConvo, browserClient } = require('../../../app/');
-const { saveMessage, getConvoTitle, saveConvo, updateConvo, getConvo } = require('../../../models');
+const { browserClient } = require('../../../app/');
+const { saveMessage, getConvoTitle, saveConvo, getConvo } = require('../../../models');
 const { handleError, sendMessage, createOnProgress, handleText } = require('./handlers');
+const requireJwtAuth = require('../../../middleware/requireJwtAuth');
 
-router.post('/', async (req, res) => {
+router.post('/', requireJwtAuth, async (req, res) => {
   const {
     endpoint,
     text,
@@ -33,11 +34,12 @@ router.post('/', async (req, res) => {
 
   // build endpoint option
   const endpointOption = {
-    model: req.body?.model || 'text-davinci-002-render-sha'
+    model: req.body?.model ?? 'text-davinci-002-render-sha',
+    token: req.body?.token ?? null
   };
 
   const availableModels = getChatGPTBrowserModels();
-  if (availableModels.find(model => model === endpointOption.model) === undefined)
+  if (availableModels.find((model) => model === endpointOption.model) === undefined)
     return handleError(res, { text: 'Illegal request: model' });
 
   console.log('ask log', {
@@ -48,7 +50,7 @@ router.post('/', async (req, res) => {
 
   if (!overrideParentMessageId) {
     await saveMessage(userMessage);
-    await saveConvo(req?.session?.user?.username, {
+    await saveConvo(req.user.id, {
       ...userMessage,
       ...endpointOption,
       conversationId,
@@ -80,6 +82,7 @@ const ask = async ({
   res
 }) => {
   let { text, parentMessageId: userParentMessageId, messageId: userMessageId } = userMessage;
+  const userId = req.user.id;
 
   res.writeHead(200, {
     Connection: 'keep-alive',
@@ -91,81 +94,124 @@ const ask = async ({
 
   if (preSendRequest) sendMessage(res, { message: userMessage, created: true });
 
+  let responseMessageId = crypto.randomUUID();
+
   try {
-    const progressCallback = createOnProgress();
+    let lastSavedTimestamp = 0;
+    const { onProgress: progressCallback, getPartialText } = createOnProgress({
+      onProgress: ({ text }) => {
+        const currentTimestamp = Date.now();
+        if (currentTimestamp - lastSavedTimestamp > 500) {
+          lastSavedTimestamp = currentTimestamp;
+          saveMessage({
+            messageId: responseMessageId,
+            sender: endpointOption?.jailbreak ? 'Sydney' : 'BingAI',
+            conversationId,
+            parentMessageId: overrideParentMessageId || userMessageId,
+            text: text,
+            unfinished: true,
+            cancelled: false,
+            error: false
+          });
+        }
+      }
+    });
     const abortController = new AbortController();
-    res.on('close', () => abortController.abort());
     let response = await browserClient({
       text,
       parentMessageId: userParentMessageId,
       conversationId,
       ...endpointOption,
       onProgress: progressCallback.call(null, { res, text }),
-      abortController
+      abortController,
+      userId
     });
 
     console.log('CLIENT RESPONSE', response);
 
+    const newConversationId = response.conversationId || conversationId;
+    const newUserMassageId = response.parentMessageId || userMessageId;
+    const newResponseMessageId = response.messageId;
+
     // STEP1 generate response message
     response.text = response.response || '**ChatGPT refused to answer.**';
 
     let responseMessage = {
-      conversationId: response.conversationId,
-      messageId: response.messageId,
-      parentMessageId: overrideParentMessageId || response.parentMessageId || userMessageId,
+      conversationId: newConversationId,
+      messageId: responseMessageId,
+      newMessageId: newResponseMessageId,
+      parentMessageId: overrideParentMessageId || newUserMassageId,
       text: await handleText(response),
-      sender: endpointOption?.chatGptLabel || 'ChatGPT'
+      sender: endpointOption?.chatGptLabel || 'ChatGPT',
+      unfinished: false,
+      cancelled: false,
+      error: false
     };
 
     await saveMessage(responseMessage);
+    responseMessage.messageId = newResponseMessageId;
 
     // STEP2 update the conversation
 
     // First update conversationId if needed
-    let conversationUpdate = { conversationId, endpoint: 'chatGPTBrowser' };
-    if (conversationId != responseMessage.conversationId && isNewConversation)
-      conversationUpdate = {
-        ...conversationUpdate,
-        conversationId: conversationId,
-        newConversationId: responseMessage.conversationId || conversationId
-      };
-    conversationId = responseMessage.conversationId || conversationId;
+    let conversationUpdate = { conversationId: newConversationId, endpoint: 'chatGPTBrowser' };
+    if (conversationId != newConversationId)
+      if (isNewConversation) {
+        // change the conversationId to new one
+        conversationUpdate = {
+          ...conversationUpdate,
+          conversationId: conversationId,
+          newConversationId: newConversationId
+        };
+      } else {
+        // create new conversation
+        conversationUpdate = {
+          ...conversationUpdate,
+          ...endpointOption
+        };
+      }
 
-    await saveConvo(req?.session?.user?.username, conversationUpdate);
+    await saveConvo(req.user.id, conversationUpdate);
+    conversationId = newConversationId;
 
     // STEP3 update the user message
-    userMessage.conversationId = conversationId;
-    userMessage.messageId = responseMessage.parentMessageId;
+    userMessage.conversationId = newConversationId;
+    userMessage.messageId = newUserMassageId;
 
     // If response has parentMessageId, the fake userMessage.messageId should be updated to the real one.
-    if (!overrideParentMessageId) {
-      const oldUserMessageId = userMessageId;
-      await saveMessage({ ...userMessage, messageId: oldUserMessageId, newMessageId: userMessage.messageId });
-    }
-    userMessageId = userMessage.messageId;
+    if (!overrideParentMessageId)
+      await saveMessage({
+        ...userMessage,
+        messageId: userMessageId,
+        newMessageId: newUserMassageId
+      });
+    userMessageId = newUserMassageId;
 
     sendMessage(res, {
-      title: await getConvoTitle(req?.session?.user?.username, conversationId),
+      title: await getConvoTitle(req.user.id, conversationId),
       final: true,
-      conversation: await getConvo(req?.session?.user?.username, conversationId),
+      conversation: await getConvo(req.user.id, conversationId),
       requestMessage: userMessage,
       responseMessage: responseMessage
     });
     res.end();
 
     if (userParentMessageId == '00000000-0000-0000-0000-000000000000') {
-      const title = await titleConvo({ endpoint: endpointOption?.endpoint, text, response: responseMessage });
-      await updateConvo(req?.session?.user?.username, {
+      // const title = await titleConvo({ endpoint: endpointOption?.endpoint, text, response: responseMessage });
+      const title = await response.details.title;
+      await saveConvo(req.user.id, {
         conversationId: conversationId,
         title
       });
     }
   } catch (error) {
     const errorMessage = {
-      messageId: crypto.randomUUID(),
+      messageId: responseMessageId,
       sender: 'ChatGPT',
       conversationId,
       parentMessageId: overrideParentMessageId || userMessageId,
+      unfinished: false,
+      cancelled: false,
       error: true,
       text: error.message
     };

api/server/routes/ask/askGoogle.js

@@ -0,0 +1,156 @@
+const express = require('express');
+const router = express.Router();
+const { titleConvo } = require('../../../app/');
+const GoogleClient = require('../../../app/google/GoogleClient');
+const { saveMessage, getConvoTitle, saveConvo, getConvo } = require('../../../models');
+const { handleError, sendMessage, createOnProgress } = require('./handlers');
+const requireJwtAuth = require('../../../middleware/requireJwtAuth');
+
+router.post('/', requireJwtAuth, async (req, res) => {
+  const { endpoint, text, parentMessageId, conversationId } = req.body;
+  if (text.length === 0) return handleError(res, { text: 'Prompt empty or too short' });
+  if (endpoint !== 'google') return handleError(res, { text: 'Illegal request' });
+
+  // build endpoint option
+  const endpointOption = {
+    examples: req.body?.examples ?? [{ input: { content: '' }, output: { content: '' } }],
+    promptPrefix: req.body?.promptPrefix ?? null,
+    token: req.body?.token ?? null,
+    modelOptions: {
+      model: req.body?.model ?? 'chat-bison',
+      modelLabel: req.body?.modelLabel ?? null,
+      temperature: req.body?.temperature ?? 0.2,
+      maxOutputTokens: req.body?.maxOutputTokens ?? 1024,
+      topP: req.body?.topP ?? 0.95,
+      topK: req.body?.topK ?? 40
+    }
+  };
+
+  const availableModels = ['chat-bison', 'text-bison'];
+  if (availableModels.find((model) => model === endpointOption.modelOptions.model) === undefined) {
+    return handleError(res, { text: `Illegal request: model` });
+  }
+
+  // eslint-disable-next-line no-use-before-define
+  return await ask({
+    text,
+    endpointOption,
+    conversationId,
+    parentMessageId,
+    req,
+    res
+  });
+});
+
+const ask = async ({ text, endpointOption, parentMessageId = null, conversationId, req, res }) => {
+  res.writeHead(200, {
+    Connection: 'keep-alive',
+    'Content-Type': 'text/event-stream',
+    'Cache-Control': 'no-cache, no-transform',
+    'Access-Control-Allow-Origin': '*',
+    'X-Accel-Buffering': 'no'
+  });
+  let userMessage;
+  let userMessageId;
+  let responseMessageId;
+  let lastSavedTimestamp = 0;
+
+  try {
+    const getIds = (data) => {
+      userMessage = data.userMessage;
+      userMessageId = userMessage.messageId;
+      responseMessageId = data.responseMessageId;
+      if (!conversationId) {
+        conversationId = data.conversationId;
+      }
+    };
+
+    const { onProgress: progressCallback } = createOnProgress({
+      onProgress: ({ text: partialText }) => {
+        const currentTimestamp = Date.now();
+        if (currentTimestamp - lastSavedTimestamp > 500) {
+          lastSavedTimestamp = currentTimestamp;
+          saveMessage({
+            messageId: responseMessageId,
+            sender: 'PaLM2',
+            conversationId,
+            parentMessageId: userMessageId,
+            text: partialText,
+            unfinished: true,
+            cancelled: false,
+            error: false
+          });
+        }
+      }
+    });
+
+    const abortController = new AbortController();
+
+    let key;
+    if (endpointOption.token) {
+      key = JSON.parse(endpointOption.token);
+      delete endpointOption.token;
+      console.log('Using service account key provided by User for PaLM models');
+    }
+
+    try {
+      if (!key) {
+        key = require('../../../data/auth.json');
+      }
+    } catch (e) {
+      console.log("No 'auth.json' file (service account key) found in /api/data/ for PaLM models");
+    }
+
+    const clientOptions = {
+      // debug: true, // for testing
+      reverseProxyUrl: process.env.GOOGLE_REVERSE_PROXY || null,
+      proxy: process.env.PROXY || null,
+      ...endpointOption
+    };
+
+    const client = new GoogleClient(key, clientOptions);
+
+    let response = await client.sendMessage(text, {
+      getIds,
+      user: req.user.id,
+      parentMessageId,
+      conversationId,
+      onProgress: progressCallback.call(null, { res, text, parentMessageId: userMessageId }),
+      abortController
+    });
+
+    await saveMessage(response);
+    sendMessage(res, {
+      title: await getConvoTitle(req.user.id, conversationId),
+      final: true,
+      conversation: await getConvo(req.user.id, conversationId),
+      requestMessage: userMessage,
+      responseMessage: response
+    });
+    res.end();
+
+    if (parentMessageId == '00000000-0000-0000-0000-000000000000') {
+      const title = await titleConvo({ text, response });
+      await saveConvo(req.user.id, {
+        conversationId: conversationId,
+        title
+      });
+    }
+  } catch (error) {
+    console.error(error);
+    const errorMessage = {
+      messageId: responseMessageId,
+      sender: 'PaLM2',
+      conversationId,
+      parentMessageId,
+      unfinished: false,
+      cancelled: false,
+      error: true,
+      text: error.message
+    };
+    await saveMessage(errorMessage);
+    handleError(res, errorMessage);
+  }
+};
+
+module.exports = router;

api/server/routes/ask/askOpenAI.js

@@ -1,12 +1,33 @@
 const express = require('express');
 const crypto = require('crypto');
 const router = express.Router();
+const addToCache = require('./addToCache');
 const { getOpenAIModels } = require('../endpoints');
 const { titleConvo, askClient } = require('../../../app/');
-const { saveMessage, getConvoTitle, saveConvo, updateConvo, getConvo } = require('../../../models');
+const { saveMessage, getConvoTitle, saveConvo, getConvo } = require('../../../models');
 const { handleError, sendMessage, createOnProgress, handleText } = require('./handlers');
+const requireJwtAuth = require('../../../middleware/requireJwtAuth');
 
-router.post('/', async (req, res) => {
+const abortControllers = new Map();
+
+router.post('/abort', requireJwtAuth, async (req, res) => {
+  const { abortKey } = req.body;
+  console.log(`req.body`, req.body);
+  if (!abortControllers.has(abortKey)) {
+    return res.status(404).send('Request not found');
+  }
+
+  const { abortController } = abortControllers.get(abortKey);
+
+  abortControllers.delete(abortKey);
+  const ret = await abortController.abortAsk();
+  console.log('Aborted request', abortKey);
+  console.log('Aborted message:', ret);
+
+  res.send(JSON.stringify(ret));
+});
+
+router.post('/', requireJwtAuth, async (req, res) => {
   const {
     endpoint,
     text,
@@ -19,6 +40,7 @@ router.post('/', async (req, res) => {
 
   // build user message
   const conversationId = oldConversationId || crypto.randomUUID();
+  const isNewConversation = !oldConversationId;
   const userMessageId = crypto.randomUUID();
   const userParentMessageId = parentMessageId || '00000000-0000-0000-0000-000000000000';
   const userMessage = {
@@ -32,17 +54,17 @@ router.post('/', async (req, res) => {
 
   // build endpoint option
   const endpointOption = {
-    model: req.body?.model || 'gpt-3.5-turbo',
-    chatGptLabel: req.body?.chatGptLabel || null,
-    promptPrefix: req.body?.promptPrefix || null,
-    temperature: req.body?.temperature || 1,
-    top_p: req.body?.top_p || 1,
-    presence_penalty: req.body?.presence_penalty || 0,
-    frequency_penalty: req.body?.frequency_penalty || 0
+    model: req.body?.model ?? 'gpt-3.5-turbo',
+    chatGptLabel: req.body?.chatGptLabel ?? null,
+    promptPrefix: req.body?.promptPrefix ?? null,
+    temperature: req.body?.temperature ?? 1,
+    top_p: req.body?.top_p ?? 1,
+    presence_penalty: req.body?.presence_penalty ?? 0,
+    frequency_penalty: req.body?.frequency_penalty ?? 0
   };
 
   const availableModels = getOpenAIModels();
-  if (availableModels.find(model => model === endpointOption.model) === undefined)
+  if (availableModels.find((model) => model === endpointOption.model) === undefined)
     return handleError(res, { text: 'Illegal request: model' });
 
   console.log('ask log', {
@@ -53,7 +75,7 @@ router.post('/', async (req, res) => {
 
   if (!overrideParentMessageId) {
     await saveMessage(userMessage);
-    await saveConvo(req?.session?.user?.username, {
+    await saveConvo(req.user.id, {
       ...userMessage,
       ...endpointOption,
       conversationId,
@@ -63,6 +85,7 @@ router.post('/', async (req, res) => {
 
   // eslint-disable-next-line no-use-before-define
   return await ask({
+    isNewConversation,
     userMessage,
     endpointOption,
     conversationId,
@@ -74,6 +97,7 @@ router.post('/', async (req, res) => {
 });
 
 const ask = async ({
+  isNewConversation,
   userMessage,
   endpointOption,
   conversationId,
@@ -83,8 +107,8 @@ const ask = async ({
   res
 }) => {
   let { text, parentMessageId: userParentMessageId, messageId: userMessageId } = userMessage;
-
-  const client = askClient;
+  const userId = req.user.id;
+  let responseMessageId = crypto.randomUUID();
 
   res.writeHead(200, {
     Connection: 'keep-alive',
@@ -97,66 +121,147 @@ const ask = async ({
   if (preSendRequest) sendMessage(res, { message: userMessage, created: true });
 
   try {
-    const progressCallback = createOnProgress();
-    const abortController = new AbortController();
-    res.on('close', () => abortController.abort());
-    let response = await client({
+    let lastSavedTimestamp = 0;
+    const { onProgress: progressCallback, getPartialText } = createOnProgress({
+      onProgress: ({ text }) => {
+        const currentTimestamp = Date.now();
+        if (currentTimestamp - lastSavedTimestamp > 500) {
+          lastSavedTimestamp = currentTimestamp;
+          saveMessage({
+            messageId: responseMessageId,
+            sender: endpointOption?.chatGptLabel || 'ChatGPT',
+            conversationId,
+            parentMessageId: overrideParentMessageId || userMessageId,
+            text: text,
+            unfinished: true,
+            cancelled: false,
+            error: false
+          });
+        }
+      }
+    });
+
+    let abortController = new AbortController();
+    abortController.abortAsk = async function () {
+      this.abort();
+
+      const responseMessage = {
+        messageId: responseMessageId,
+        sender: endpointOption?.chatGptLabel || 'ChatGPT',
+        conversationId,
+        parentMessageId: overrideParentMessageId || userMessageId,
+        text: getPartialText(),
+        unfinished: false,
+        cancelled: true,
+        error: false
+      };
+
+      saveMessage(responseMessage);
+      await addToCache({ endpoint: 'openAI', endpointOption, userMessage, responseMessage });
+
+      return {
+        title: await getConvoTitle(req.user.id, conversationId),
+        final: true,
+        conversation: await getConvo(req.user.id, conversationId),
+        requestMessage: userMessage,
+        responseMessage: responseMessage
+      };
+    };
+    const abortKey = conversationId;
+    abortControllers.set(abortKey, { abortController, ...endpointOption });
+    const oaiApiKey = req.body?.token ?? null;
+
+    let response = await askClient({
       text,
       parentMessageId: userParentMessageId,
       conversationId,
+      oaiApiKey,
       ...endpointOption,
       onProgress: progressCallback.call(null, {
         res,
         text,
         parentMessageId: overrideParentMessageId || userMessageId
       }),
-      abortController
+      abortController,
+      userId
     });
 
+    abortControllers.delete(abortKey);
     console.log('CLIENT RESPONSE', response);
 
+    const newConversationId = response.conversationId || conversationId;
+    const newUserMassageId = response.parentMessageId || userMessageId;
+    const newResponseMessageId = response.messageId;
+
     // STEP1 generate response message
     response.text = response.response || '**ChatGPT refused to answer.**';
 
     let responseMessage = {
-      conversationId: response.conversationId,
-      messageId: response.messageId,
-      parentMessageId: overrideParentMessageId || userMessageId,
+      conversationId: newConversationId,
+      messageId: responseMessageId,
+      newMessageId: newResponseMessageId,
+      parentMessageId: overrideParentMessageId || newUserMassageId,
       text: await handleText(response),
-      sender: endpointOption?.chatGptLabel || 'ChatGPT'
+      sender: endpointOption?.chatGptLabel || 'ChatGPT',
+      unfinished: false,
+      cancelled: false,
+      error: false
     };
 
     await saveMessage(responseMessage);
+    responseMessage.messageId = newResponseMessageId;
 
     // STEP2 update the conversation
-    conversationId = responseMessage.conversationId || conversationId;
-    // it seems openAI will not change the conversationId.
-    // let conversationUpdate = { conversationId, endpoint: 'openAI' };
-    // await saveConvo(req?.session?.user?.username, conversationUpdate);
+    let conversationUpdate = { conversationId: newConversationId, endpoint: 'openAI' };
+    if (conversationId != newConversationId)
+      if (isNewConversation) {
+        // change the conversationId to new one
+        conversationUpdate = {
+          ...conversationUpdate,
+          conversationId: conversationId,
+          newConversationId: newConversationId
+        };
+      } else {
+        // create new conversation
+        conversationUpdate = {
+          ...conversationUpdate,
+          ...endpointOption
+        };
+      }
+
+    await saveConvo(req.user.id, conversationUpdate);
+    conversationId = newConversationId;
 
     // STEP3 update the user message
-    userMessage.conversationId = conversationId;
-    userMessage.messageId = responseMessage.parentMessageId;
+    userMessage.conversationId = newConversationId;
+    userMessage.messageId = newUserMassageId;
 
     // If response has parentMessageId, the fake userMessage.messageId should be updated to the real one.
-    if (!overrideParentMessageId) {
-      const oldUserMessageId = userMessageId;
-      await saveMessage({ ...userMessage, messageId: oldUserMessageId, newMessageId: userMessage.messageId });
-    }
-    userMessageId = userMessage.messageId;
+    if (!overrideParentMessageId)
+      await saveMessage({
+        ...userMessage,
+        messageId: userMessageId,
+        newMessageId: newUserMassageId
+      });
+    userMessageId = newUserMassageId;
 
     sendMessage(res, {
-      title: await getConvoTitle(req?.session?.user?.username, conversationId),
+      title: await getConvoTitle(req.user.id, conversationId),
       final: true,
-      conversation: await getConvo(req?.session?.user?.username, conversationId),
+      conversation: await getConvo(req.user.id, conversationId),
       requestMessage: userMessage,
       responseMessage: responseMessage
     });
     res.end();
 
     if (userParentMessageId == '00000000-0000-0000-0000-000000000000') {
-      const title = await titleConvo({ endpoint: endpointOption?.endpoint, text, response: responseMessage });
-      await updateConvo(req?.session?.user?.username, {
+      const title = await titleConvo({
+        endpoint: endpointOption?.endpoint,
+        text,
+        response: responseMessage,
+        oaiApiKey
+      });
+      await saveConvo(req.user.id, {
         conversationId: conversationId,
         title
       });
@@ -164,10 +269,12 @@ const ask = async ({
   } catch (error) {
     console.error(error);
     const errorMessage = {
-      messageId: crypto.randomUUID(),
+      messageId: responseMessageId,
       sender: endpointOption?.chatGptLabel || 'ChatGPT',
       conversationId,
       parentMessageId: overrideParentMessageId || userMessageId,
+      unfinished: false,
+      cancelled: false,
       error: true,
       text: error.message
     };

api/server/routes/ask/handlers.js

@@ -17,7 +17,7 @@ const sendMessage = (res, message) => {
   res.write(`event: message\ndata: ${JSON.stringify(message)}\n\n`);
 };
 
-const createOnProgress = () => {
+const createOnProgress = ({ onProgress: _onProgress }) => {
   let i = 0;
   let code = '';
   let tokens = '';
@@ -65,14 +65,21 @@ const createOnProgress = () => {
     }
 
     sendMessage(res, { text: tokens + cursor, message: true, initial: i === 0, ...rest });
+
+    _onProgress && _onProgress({ text: tokens, message: true, initial: i === 0, ...rest });
+
     i++;
   };
 
-  const onProgress = opts => {
+  const onProgress = (opts) => {
     return _.partialRight(progressCallback, opts);
   };
 
-  return onProgress;
+  const getPartialText = () => {
+    return tokens;
+  };
+
+  return { onProgress, getPartialText };
 };
 
 const handleText = async (response, bing = false) => {

api/server/routes/ask/index.js

@@ -2,11 +2,13 @@ const express = require('express');
 const router = express.Router();
 // const askAzureOpenAI = require('./askAzureOpenAI';)
 const askOpenAI = require('./askOpenAI');
+const askGoogle = require('./askGoogle');
 const askBingAI = require('./askBingAI');
 const askChatGPTBrowser = require('./askChatGPTBrowser');
 
 // router.use('/azureOpenAI', askAzureOpenAI);
 router.use('/openAI', askOpenAI);
+router.use('/google', askGoogle);
 router.use('/bingAI', askBingAI);
 router.use('/chatGPTBrowser', askChatGPTBrowser);
 

api/server/routes/auth.js

@@ -1,57 +1,25 @@
 const express = require('express');
+const {
+  resetPasswordRequestController,
+  resetPasswordController,
+  getUserController,
+  loginController,
+  logoutController,
+  refreshController,
+  registrationController
+} = require('../controllers/auth.controller');
+const requireJwtAuth = require('../../middleware/requireJwtAuth');
+const requireLocalAuth = require('../../middleware/requireLocalAuth');
+
 const router = express.Router();
-const authYourLogin = require('./authYourLogin');
-const userSystemEnabled = !!process.env.ENABLE_USER_SYSTEM || false;
 
-router.get('/login', function (req, res) {
-  if (userSystemEnabled) {
-    res.redirect('/auth/your_login_page');
-  } else {
-    res.redirect('/');
-  }
-});
+//Local
+router.get('/user', requireJwtAuth, getUserController);
+router.post('/logout', requireJwtAuth, logoutController);
+router.post('/login', requireLocalAuth, loginController);
+router.post('/refresh', requireJwtAuth, refreshController);
+router.post('/register', registrationController);
+router.post('/requestPasswordReset', resetPasswordRequestController);
+router.post('/resetPassword', resetPasswordController);
 
-router.get('/logout', function (req, res) {
-  // clear the session
-  req.session.user = null;
-
-  req.session.save(function () {
-    if (userSystemEnabled) {
-      res.redirect('/auth/your_login_page/logout');
-    } else {
-      res.redirect('/');
-    }
-  });
-});
-
-const authenticatedOr401 = (req, res, next) => {
-  if (userSystemEnabled) {
-    const user = req?.session?.user;
-
-    if (user) {
-      next();
-    } else {
-      res.status(401).end();
-    }
-  } else {
-    next();
-  }
-};
-
-const authenticatedOrRedirect = (req, res, next) => {
-  if (userSystemEnabled) {
-    const user = req?.session?.user;
-
-    if (user) {
-      next();
-    } else {
-      res.redirect('/auth/login');
-    }
-  } else next();
-};
-
-if (userSystemEnabled) {
-  router.use('/your_login_page', authYourLogin);
-}
-
-module.exports = { router, authenticatedOr401, authenticatedOrRedirect };
+module.exports = router;

api/server/routes/authYourLogin.js

@@ -1,44 +0,0 @@
-const express = require('express');
-const router = express.Router();
-
-// WARNING!
-// THIS IS NOT A READY TO USE USER SYSTEM
-// PLEASE IMPLEMENT YOUR OWN USER SYSTEM
-
-const userSystemEnabled = process.env.ENABLE_USER_SYSTEM || false;
-
-// Logout
-router.get('/logout', (req, res) => {
-  // Do anything you want
-  console.warn('logout not implemented!');
-
-  // finish
-  res.redirect('/');
-});
-
-// Login
-router.get('/', async (req, res) => {
-  // Do anything you want
-  console.warn('login not implemented! Automatic passed as sample user');
-
-  // save the user info into session
-  // username will be used in db
-  // display will be used in UI
-  if (userSystemEnabled) {
-    req.session.user = {
-      username: null, // was 'sample_user', but would break previous relationship with previous conversations before v0.1.0
-      display: 'Sample User'
-    };
-  }
-
-  req.session.save(function (error) {
-    if (error) {
-      console.log(error);
-      res.send(`<h1>Login Failed. An error occurred. Please see the server logs for details.</h1>`);
-    } else {
-      res.redirect('/');
-    }
-  });
-});
-
-module.exports = router;

api/server/routes/convos.js

@@ -1,24 +1,23 @@
 const express = require('express');
 const router = express.Router();
-const { titleConvo } = require('../../app/');
-const { getConvo, saveConvo, getConvoTitle } = require('../../models');
-const { getConvosByPage, deleteConvos, updateConvo } = require('../../models/Conversation');
-const { getMessages } = require('../../models/Message');
+const { getConvo, saveConvo } = require('../../models');
+const { getConvosByPage, deleteConvos } = require('../../models/Conversation');
+const requireJwtAuth = require('../../middleware/requireJwtAuth');
 
-router.get('/', async (req, res) => {
+router.get('/', requireJwtAuth, async (req, res) => {
   const pageNumber = req.query.pageNumber || 1;
-  res.status(200).send(await getConvosByPage(req?.session?.user?.username, pageNumber));
+  res.status(200).send(await getConvosByPage(req.user.id, pageNumber));
 });
 
-router.get('/:conversationId', async (req, res) => {
+router.get('/:conversationId', requireJwtAuth, async (req, res) => {
   const { conversationId } = req.params;
-  const convo = await getConvo(req?.session?.user?.username, conversationId);
+  const convo = await getConvo(req.user.id, conversationId);
 
   if (convo) res.status(200).send(convo.toObject());
   else res.status(404).end();
 });
 
-router.post('/clear', async (req, res) => {
+router.post('/clear', requireJwtAuth, async (req, res) => {
   let filter = {};
   const { conversationId, source } = req.body.arg;
   if (conversationId) {
@@ -32,7 +31,7 @@ router.post('/clear', async (req, res) => {
   }
 
   try {
-    const dbResponse = await deleteConvos(req?.session?.user?.username, filter);
+    const dbResponse = await deleteConvos(req.user.id, filter);
     res.status(201).send(dbResponse);
   } catch (error) {
     console.error(error);
@@ -40,11 +39,11 @@ router.post('/clear', async (req, res) => {
   }
 });
 
-router.post('/update', async (req, res) => {
+router.post('/update', requireJwtAuth, async (req, res) => {
   const update = req.body.arg;
 
   try {
-    const dbResponse = await updateConvo(req?.session?.user?.username, update);
+    const dbResponse = await saveConvo(req.user.id, update);
     res.status(201).send(dbResponse);
   } catch (error) {
     console.error(error);

api/server/routes/endpoints.js

@@ -9,19 +9,52 @@ const getOpenAIModels = () => {
 };
 
 const getChatGPTBrowserModels = () => {
-  let models = ['text-davinci-002-render-sha', 'text-davinci-002-render-paid', 'gpt-4'];
+  let models = ['text-davinci-002-render-sha', 'gpt-4'];
   if (process.env.CHATGPT_MODELS) models = String(process.env.CHATGPT_MODELS).split(',');
 
   return models;
 };
 
-router.get('/', function (req, res) {
-  const azureOpenAI = !!process.env.AZURE_OPENAI_KEY;
-  const openAI = process.env.OPENAI_KEY ? { availableModels: getOpenAIModels() } : false;
-  const bingAI = !!process.env.BINGAI_TOKEN;
-  const chatGPTBrowser = process.env.CHATGPT_TOKEN ? { availableModels: getChatGPTBrowserModels() } : false;
+let i = 0;
+router.get('/', async function (req, res) {
+  let key, palmUser;
+  try {
+    key = require('../../data/auth.json');
+  } catch (e) {
+    if (i === 0) {
+      console.log("No 'auth.json' file (service account key) found in /api/data/ for PaLM models");
+      i++;
+    }
+  }
 
-  res.send(JSON.stringify({ azureOpenAI, openAI, bingAI, chatGPTBrowser }));
+  if (process.env.PALM_KEY === 'user_provided') {
+    palmUser = true;
+    if (i <= 1) {
+      console.log('User will provide key for PaLM models');
+      i++;
+    }
+  }
+
+  const google =
+    key || palmUser
+      ? { userProvide: palmUser, availableModels: ['chat-bison', 'text-bison'] }
+      : false;
+  const azureOpenAI = !!process.env.AZURE_OPENAI_KEY;
+  const apiKey = process.env.OPENAI_KEY || process.env.AZURE_OPENAI_API_KEY;
+  const openAI = apiKey
+    ? { availableModels: getOpenAIModels(), userProvide: apiKey === 'user_provided' }
+    : false;
+  const bingAI = process.env.BINGAI_TOKEN
+    ? { userProvide: process.env.BINGAI_TOKEN == 'user_provided' }
+    : false;
+  const chatGPTBrowser = process.env.CHATGPT_TOKEN
+    ? {
+      userProvide: process.env.CHATGPT_TOKEN == 'user_provided',
+      availableModels: getChatGPTBrowserModels()
+    }
+    : false;
+
+  res.send(JSON.stringify({ azureOpenAI, openAI, google, bingAI, chatGPTBrowser }));
 });
 
 module.exports = { router, getOpenAIModels, getChatGPTBrowserModels };

api/server/routes/index.js

@@ -5,9 +5,9 @@ const presets = require('./presets');
 const prompts = require('./prompts');
 const search = require('./search');
 const tokenizer = require('./tokenizer');
-const me = require('./me');
+const auth = require('./auth');
+const oauth = require('./oauth');
 const { router: endpoints } = require('./endpoints');
-const { router: auth, authenticatedOr401, authenticatedOrRedirect } = require('./auth');
 
 module.exports = {
   search,
@@ -17,9 +17,7 @@ module.exports = {
   presets,
   prompts,
   auth,
+  oauth,
   tokenizer,
-  me,
-  endpoints,
-  authenticatedOr401,
-  authenticatedOrRedirect
+  endpoints
 };

api/server/routes/me.js

@@ -1,16 +0,0 @@
-const express = require('express');
-const router = express.Router();
-const userSystemEnabled = !!process.env.ENABLE_USER_SYSTEM || false;
-
-router.get('/', function (req, res) {
-  if (userSystemEnabled) {
-    const user = req?.session?.user;
-
-    if (user) res.send(JSON.stringify({ username: user?.username, display: user?.display }));
-    else res.send(JSON.stringify(null));
-  } else {
-    res.send(JSON.stringify({ username: 'anonymous_user', display: 'Anonymous User' }));
-  }
-});
-
-module.exports = router;

api/server/routes/messages.js

@@ -1,8 +1,9 @@
 const express = require('express');
 const router = express.Router();
 const { getMessages } = require('../../models/Message');
+const requireJwtAuth = require('../../middleware/requireJwtAuth');
 
-router.get('/:conversationId', async (req, res) => {
+router.get('/:conversationId', requireJwtAuth, async (req, res) => {
   const { conversationId } = req.params;
   res.status(200).send(await getMessages({ conversationId }));
 });

api/server/routes/oauth.js

@@ -0,0 +1,64 @@
+const passport = require('passport');
+const express = require('express');
+
+const router = express.Router();
+
+const isProduction = process.env.NODE_ENV === 'production';
+const clientUrl = isProduction ? process.env.CLIENT_URL_PROD : process.env.CLIENT_URL_DEV;
+
+// Social
+router.get(
+  '/google',
+  passport.authenticate('google', {
+    scope: ['openid', 'profile', 'email'],
+    session: false
+  })
+);
+
+router.get(
+  '/google/callback',
+  passport.authenticate('google', {
+    failureRedirect: `${clientUrl}/login`,
+    failureMessage: true,
+    session: false,
+    scope: ['openid', 'profile', 'email']
+  }),
+  (req, res) => {
+    const token = req.user.generateToken();
+    res.cookie('token', token, {
+      expires: new Date(Date.now() + eval(process.env.SESSION_EXPIRY)),
+      httpOnly: false,
+      secure: isProduction
+    });
+    res.redirect(clientUrl);
+  }
+);
+
+router.get(
+  '/facebook',
+  passport.authenticate('facebook', {
+    scope: ['public_profile', 'email'],
+    session: false
+  })
+);
+
+router.get(
+  '/facebook/callback',
+  passport.authenticate('facebook', {
+    failureRedirect: `${clientUrl}/login`,
+    failureMessage: true,
+    session: false,
+    scope: ['public_profile', 'email']
+  }),
+  (req, res) => {
+    const token = req.user.generateToken();
+    res.cookie('token', token, {
+      expires: new Date(Date.now() + eval(process.env.SESSION_EXPIRY)),
+      httpOnly: false,
+      secure: isProduction
+    });
+    res.redirect(clientUrl);
+  }
+);
+
+module.exports = router;

api/server/routes/presets.js

@@ -2,23 +2,24 @@ const express = require('express');
 const router = express.Router();
 const { getPresets, savePreset, deletePresets } = require('../../models');
 const crypto = require('crypto');
+const requireJwtAuth = require('../../middleware/requireJwtAuth');
 
-router.get('/', async (req, res) => {
-  const presets = (await getPresets(req?.session?.user?.username)).map((preset) => {
+router.get('/', requireJwtAuth, async (req, res) => {
+  const presets = (await getPresets(req.user.id)).map((preset) => {
     return preset.toObject();
   });
   res.status(200).send(presets);
 });
 
-router.post('/', async (req, res) => {
+router.post('/', requireJwtAuth, async (req, res) => {
   const update = req.body || {};
 
   update.presetId = update?.presetId || crypto.randomUUID();
 
   try {
-    await savePreset(req?.session?.user?.username, update);
+    await savePreset(req.user.id, update);
 
-    const presets = (await getPresets(req?.session?.user?.username)).map((preset) => {
+    const presets = (await getPresets(req.user.id)).map((preset) => {
       return preset.toObject();
     });
     res.status(201).send(presets);
@@ -28,7 +29,7 @@ router.post('/', async (req, res) => {
   }
 });
 
-router.post('/delete', async (req, res) => {
+router.post('/delete', requireJwtAuth, async (req, res) => {
   let filter = {};
   const { presetId } = req.body.arg || {};
 
@@ -37,9 +38,9 @@ router.post('/delete', async (req, res) => {
   console.log('delete preset filter', filter);
 
   try {
-    await deletePresets(req?.session?.user?.username, filter);
+    await deletePresets(req.user.id, filter);
 
-    const presets = (await getPresets(req?.session?.user?.username)).map(preset => preset.toObject());
+    const presets = (await getPresets(req.user.id)).map((preset) => preset.toObject());
 
     // console.log('delete preset response', presets);
     res.status(201).send(presets);

api/server/routes/search.js

@@ -5,6 +5,8 @@ const { Message } = require('../../models/Message');
 const { Conversation, getConvosQueried } = require('../../models/Conversation');
 const { reduceHits } = require('../../lib/utils/reduceHits');
 const { cleanUpPrimaryKeyValue } = require('../../lib/utils/misc');
+const requireJwtAuth = require('../../middleware/requireJwtAuth');
+
 const cache = new Map();
 
 router.get('/sync', async function (req, res) {
@@ -13,9 +15,9 @@ router.get('/sync', async function (req, res) {
   res.send('synced');
 });
 
-router.get('/', async function (req, res) {
+router.get('/', requireJwtAuth, async function (req, res) {
   try {
-    let user = req?.session?.user?.username;
+    let user = req.user.id;
     user = user ?? null;
     const { q } = req.query;
     const pageNumber = req.query.pageNumber || 1;
@@ -25,7 +27,9 @@ router.get('/', async function (req, res) {
       console.log('cache hit', key);
       const cached = cache.get(key);
       const { pages, pageSize, messages } = cached;
-      res.status(200).send({ conversations: cached[pageNumber], pages, pageNumber, pageSize, messages });
+      res
+        .status(200)
+        .send({ conversations: cached[pageNumber], pages, pageNumber, pageSize, messages });
       return;
     } else {
       cache.clear();
@@ -42,7 +46,7 @@ router.get('/', async function (req, res) {
         },
         true
       )
-    ).hits.map(message => {
+    ).hits.map((message) => {
       const { _formatted, ...rest } = message;
       return {
         ...rest,
@@ -93,12 +97,12 @@ router.get('/clear', async function (req, res) {
 
 router.get('/test', async function (req, res) {
   const { q } = req.query;
-  const messages = (await Message.meiliSearch(q, { attributesToHighlight: ['text'] }, true)).hits.map(
-    message => {
-      const { _formatted, ...rest } = message;
-      return { ...rest, searchResult: true, text: _formatted.text };
-    }
-  );
+  const messages = (
+    await Message.meiliSearch(q, { attributesToHighlight: ['text'] }, true)
+  ).hits.map((message) => {
+    const { _formatted, ...rest } = message;
+    return { ...rest, searchResult: true, text: _formatted.text };
+  });
   res.send(messages);
 });
 

api/server/routes/tokenizer.js

@@ -4,15 +4,23 @@ const { Tiktoken } = require('@dqbd/tiktoken/lite');
 const { load } = require('@dqbd/tiktoken/load');
 const registry = require('@dqbd/tiktoken/registry.json');
 const models = require('@dqbd/tiktoken/model_to_encoding.json');
+const requireJwtAuth = require('../../middleware/requireJwtAuth');
 
-router.post('/', async (req, res) => {
-  const { arg } = req.body;
-  // console.log(typeof req.body === 'object' ? { ...req.body, ...req.query } : req.query);
-  const model = await load(registry[models['gpt-3.5-turbo']]);
-  const encoder = new Tiktoken(model.bpe_ranks, model.special_tokens, model.pat_str);
-  const tokens = encoder.encode(arg.text);
-  encoder.free();
-  res.send({ count: tokens.length });
+router.post('/', requireJwtAuth, async (req, res) => {
+  try {
+    const { arg } = req.body;
+
+    // console.log('context:', arg, req.body);
+    // console.log(typeof req.body === 'object' ? { ...req.body, ...req.query } : req.query);
+    const model = await load(registry[models['gpt-3.5-turbo']]);
+    const encoder = new Tiktoken(model.bpe_ranks, model.special_tokens, model.pat_str);
+    const tokens = encoder.encode(arg.text);
+    encoder.free();
+    res.send({ count: tokens.length });
+  } catch (e) {
+    console.error(e);
+    res.status(500).send(e.message);
+  }
 });
 
 module.exports = router;

api/server/services/auth.service.js

@@ -0,0 +1,192 @@
+const User = require('../../models/User');
+const Token = require('../../models/schema/tokenSchema');
+const sendEmail = require('../../utils/sendEmail');
+const crypto = require('crypto');
+const bcrypt = require('bcrypt');
+const DebugControl = require('../../utils/debug.js');
+const { registerSchema } = require('../../strategies/validators');
+const migrateDataToFirstUser = require('../../utils/migrateDataToFirstUser');
+
+function log({ title, parameters }) {
+  DebugControl.log.functionName(title);
+  DebugControl.log.parameters(parameters);
+}
+
+const isProduction = process.env.NODE_ENV === 'production';
+const clientUrl = isProduction ? process.env.CLIENT_URL_PROD : process.env.CLIENT_URL_DEV;
+
+const loginUser = async (user) => {
+  // const refreshToken = req.user.generateRefreshToken();
+  const dbUser = await User.findById(user._id);
+  //todo: save refresh token
+
+  return dbUser;
+};
+
+const logoutUser = async (user, refreshToken) => {
+  try {
+    const userFound = await User.findById(user._id);
+    const tokenIndex = userFound.refreshToken.findIndex((item) => item.refreshToken === refreshToken);
+
+    if (tokenIndex !== -1) {
+      userFound.refreshToken.id(userFound.refreshToken[tokenIndex]._id).remove();
+    }
+
+    await userFound.save();
+    //res.clearCookie('refreshToken', COOKIE_OPTIONS);
+    // removeTokenCookie(res);
+    return { status: 200, message: 'Logout successful' };
+  } catch (err) {
+    return { status: 500, message: err.message };
+  }
+}
+
+const registerUser = async (user) => {
+  let response = {};
+  const { error } = registerSchema.validate(user);
+  if (error) {
+    log({
+      title: 'Route: register - Joi Validation Error',
+      parameters: [
+        { name: 'Request params:', value: user },
+        { name: 'Validation error:', value: error.details }
+      ]
+    });
+    response = { status: 422, message: error.details[0].message };
+    return response;
+  }
+
+  const { email, password, name, username } = user;
+
+  try {
+    const existingUser = await User.findOne({ email });
+
+    if (existingUser) {
+      log({
+        title: 'Register User - Email in use',
+        parameters: [
+          { name: 'Request params:', value: user },
+          { name: 'Existing user:', value: existingUser }
+        ]
+      });
+      response = { status: 422, message: 'Email is in use' };
+      return response;
+    }
+
+    //determine if this is the first registered user (not counting anonymous_user)
+    const isFirstRegisteredUser = (await User.countDocuments({})) === 0;
+
+    try {
+      const newUser = await new User({
+        provider: 'local',
+        email,
+        password,
+        username,
+        name,
+        avatar: null,
+        role: isFirstRegisteredUser ? 'ADMIN' : 'USER'
+      });
+
+      // todo: implement refresh token
+      // const refreshToken = newUser.generateRefreshToken();
+      // newUser.refreshToken.push({ refreshToken });
+      bcrypt.genSalt(10, (err, salt) => {
+        bcrypt.hash(newUser.password, salt, (errh, hash) => {
+          if (err) {
+            console.log(err);
+          }
+          // set pasword to hash
+          newUser.password = hash;
+          newUser.save();
+        });
+      });
+      console.log('newUser', newUser);
+      if (isFirstRegisteredUser) {
+        migrateDataToFirstUser(newUser);
+        // console.log(migrate);
+      }
+      response = { status: 200, user: newUser };
+      return response;
+    } catch (err) {
+      response = { status: 500, message: err.message };
+      return response;
+    }
+  } catch (err) {
+    response = { status: 500, message: err.message };
+    return response;
+  }
+};
+
+const requestPasswordReset = async (email) => {
+  const user = await User.findOne({ email });
+  if (!user) {
+    return new Error('Email does not exist');
+  }
+
+  let token = await Token.findOne({ userId: user._id });
+  if (token) await token.deleteOne();
+
+  let resetToken = crypto.randomBytes(32).toString('hex');
+  const hash = await bcrypt.hash(resetToken, 10);
+
+  await new Token({
+    userId: user._id,
+    token: hash,
+    createdAt: Date.now()
+  }).save();
+
+  const link = `${clientUrl}/reset-password?token=${resetToken}&userId=${user._id}`;
+
+  sendEmail(
+    user.email,
+    'Password Reset Request',
+    {
+      name: user.name,
+      link: link
+    },
+    './template/requestResetPassword.handlebars'
+  );
+  return { link };
+};
+
+const resetPassword = async (userId, token, password) => {
+  let passwordResetToken = await Token.findOne({ userId });
+
+  if (!passwordResetToken) {
+    return new Error('Invalid or expired password reset token');
+  }
+
+  const isValid = await bcrypt.compare(token, passwordResetToken.token);
+
+  if (!isValid) {
+    return new Error('Invalid or expired password reset token');
+  }
+
+  const hash = await bcrypt.hash(password, 10);
+
+  await User.updateOne({ _id: userId }, { $set: { password: hash } }, { new: true });
+
+  const user = await User.findById({ _id: userId });
+
+  sendEmail(
+    user.email,
+    'Password Reset Successfnodeully',
+    {
+      name: user.name
+    },
+    './template/resetPassword.handlebars'
+  );
+
+  await passwordResetToken.deleteOne();
+
+  return { message: 'Password reset was successful' };
+};
+
+module.exports = {
+  // signup,
+  registerUser,
+  loginUser,
+  logoutUser,
+  requestPasswordReset,
+  resetPassword
+};

api/strategies/facebookStrategy.js

@@ -0,0 +1,60 @@
+const passport = require('passport');
+const FacebookStrategy = require('passport-facebook').Strategy;
+const User = require('../models/User');
+
+const serverUrl =
+  process.env.NODE_ENV === 'production' ? process.env.SERVER_URL_PROD : process.env.SERVER_URL_DEV;
+
+// facebook strategy
+const facebookLogin = new FacebookStrategy(
+  {
+    clientID: process.env.FACEBOOK_APP_ID,
+    clientSecret: process.env.FACEBOOK_SECRET,
+    callbackURL: `${serverUrl}${process.env.FACEBOOK_CALLBACK_URL}`,
+    proxy: true
+    // profileFields: [
+    //   'id',
+    //   'email',
+    //   'gender',
+    //   'profileUrl',
+    //   'displayName',
+    //   'locale',
+    //   'name',
+    //   'timezone',
+    //   'updated_time',
+    //   'verified',
+    //   'picture.type(large)'
+    // ]
+  },
+  async (accessToken, refreshToken, profile, done) => {
+    console.log('facebookLogin => profile', profile);
+    try {
+      const oldUser = await User.findOne({ email: profile.emails[0].value });
+
+      if (oldUser) {
+        console.log('FACEBOOK LOGIN => found user', oldUser);
+        return done(null, oldUser);
+      }
+    } catch (err) {
+      console.log(err);
+    }
+
+    // register user
+    try {
+      const newUser = await new User({
+        provider: 'facebook',
+        facebookId: profile.id,
+        username: profile.name.givenName + profile.name.familyName,
+        email: profile.emails[0].value,
+        name: profile.displayName,
+        avatar: profile.photos[0].value
+      }).save();
+
+      done(null, newUser);
+    } catch (err) {
+      console.log(err);
+    }
+  }
+);
+
+passport.use(facebookLogin);

api/strategies/googleStrategy.js

@@ -0,0 +1,44 @@
+const passport = require('passport');
+const { Strategy: GoogleStrategy } = require('passport-google-oauth20');
+
+const User = require('../models/User');
+
+const serverUrl =
+  process.env.NODE_ENV === 'production' ? process.env.SERVER_URL_PROD : process.env.SERVER_URL_DEV;
+
+// google strategy
+const googleLogin = new GoogleStrategy(
+  {
+    clientID: process.env.GOOGLE_CLIENT_ID,
+    clientSecret: process.env.GOOGLE_CLIENT_SECRET,
+    callbackURL: `${serverUrl}${process.env.GOOGLE_CALLBACK_URL}`,
+    proxy: true
+  },
+  async (accessToken, refreshToken, profile, cb) => {
+    try {
+      const oldUser = await User.findOne({ email: profile.emails[0].value });
+      if (oldUser) {
+        return cb(null, oldUser);
+      }
+    } catch (err) {
+      console.log(err);
+    }
+
+    try {
+      const newUser = await new User({
+        provider: 'google',
+        googleId: profile.id,
+        username: profile.name.givenName,
+        email: profile.emails[0].value,
+        emailVerified: profile.emails[0].verified,
+        name: `${profile.name.givenName} ${profile.name.familyName}`,
+        avatar: profile.photos[0].value
+      }).save();
+      cb(null, newUser);
+    } catch (err) {
+      console.log(err);
+    }
+  }
+);
+
+passport.use(googleLogin);

api/strategies/jwtStrategy.js

@@ -0,0 +1,29 @@
+const passport = require('passport');
+const { Strategy: JwtStrategy, ExtractJwt } = require('passport-jwt');
+const User = require('../models/User');
+
+const isProduction = process.env.NODE_ENV === 'production';
+const secretOrKey = isProduction ? process.env.JWT_SECRET_PROD : process.env.JWT_SECRET_DEV;
+
+// JWT strategy
+const jwtLogin = new JwtStrategy(
+  {
+    jwtFromRequest: ExtractJwt.fromAuthHeaderAsBearerToken(),
+    secretOrKey
+  },
+  async (payload, done) => {
+    try {
+      const user = await User.findById(payload.id);
+      if (user) {
+        done(null, user);
+      } else {
+        console.log('JwtStrategy => no user found');
+        done(null, false);
+      }
+    } catch (err) {
+      done(err, false);
+    }
+  }
+);
+
+passport.use(jwtLogin);

api/strategies/localStrategy.js

@@ -0,0 +1,66 @@
+const passport = require('passport');
+const PassportLocalStrategy = require('passport-local').Strategy;
+
+const User = require('../models/User');
+const { loginSchema } = require('./validators');
+const DebugControl = require('../utils/debug.js');
+
+const passportLogin = new PassportLocalStrategy(
+  {
+    usernameField: 'email',
+    passwordField: 'password',
+    session: false,
+    passReqToCallback: true
+  },
+  async (req, email, password, done) => {
+    const { error } = loginSchema.validate(req.body);
+    if (error) {
+      log({
+        title: 'Passport Local Strategy - Validation Error',
+        parameters: [{ name: 'req.body', value: req.body }]
+      });
+      return done(null, false, { message: error.details[0].message });
+    }
+
+    try {
+      const user = await User.findOne({ email: email.trim() });
+      if (!user) {
+        log({
+          title: 'Passport Local Strategy - User Not Found',
+          parameters: [{ name: 'email', value: email }]
+        });
+        return done(null, false, { message: 'Email does not exists.' });
+      }
+
+      user.comparePassword(password, function (err, isMatch) {
+        if (err) {
+          log({
+            title: 'Passport Local Strategy - Compare password error',
+            parameters: [{ name: 'error', value: err }]
+          });
+          return done(err);
+        }
+        if (!isMatch) {
+          log({
+            title: 'Passport Local Strategy - Password does not match',
+            parameters: [{ name: 'isMatch', value: isMatch }]
+          });
+          return done(null, false, { message: 'Incorrect password.' });
+        }
+
+        return done(null, user);
+      });
+    } catch (err) {
+      return done(err);
+    }
+  }
+);
+
+passport.use(passportLogin);
+
+function log({ title, parameters }) {
+  DebugControl.log.functionName(title);
+  if (parameters) {
+    DebugControl.log.parameters(parameters);
+  }
+}

api/strategies/validators.js

@@ -0,0 +1,24 @@
+const Joi = require('joi');
+
+const loginSchema = Joi.object().keys({
+  email: Joi.string().trim().email().required(),
+  password: Joi.string().trim().min(6).max(20).required()
+});
+
+const registerSchema = Joi.object().keys({
+  name: Joi.string().trim().min(2).max(30).required(),
+  username: Joi.string()
+    .trim()
+    .min(2)
+    .max(20)
+    .regex(/^[a-zA-Z0-9_]+$/)
+    .required(),
+  email: Joi.string().trim().email().required(),
+  password: Joi.string().trim().min(6).max(20).required(),
+  confirm_password: Joi.string().trim().min(6).max(20).required()
+});
+
+module.exports = {
+  loginSchema,
+  registerSchema
+};

api/utils/LoggingSystem.js

@@ -0,0 +1,125 @@
+const pino = require('pino');
+
+const logger = pino({
+  level: 'info',
+  redact: {
+    paths: [
+      // List of Paths to redact from the logs (https://getpino.io/#/docs/redaction)
+      'env.OPENAI_KEY',
+      'env.BINGAI_TOKEN',
+      'env.CHATGPT_TOKEN',
+      'env.MEILI_MASTER_KEY',
+      'env.GOOGLE_CLIENT_SECRET',
+      'env.JWT_SECRET_DEV',
+      'env.JWT_SECRET_PROD',
+      'newUser.password'
+    ], // See example to filter object class instances
+    censor: '***' // Redaction character
+  }
+});
+
+// Sanitize outside the logger paths. This is useful for sanitizing variables directly with Regex and patterns.
+const redactPatterns = [
+  // Array of regular expressions for redacting patterns
+  /api[-_]?key/i,
+  /password/i,
+  /token/i,
+  /secret/i,
+  /key/i,
+  /certificate/i,
+  /client[-_]?id/i,
+  /authorization[-_]?code/i,
+  /authorization[-_]?login[-_]?hint/i,
+  /authorization[-_]?acr[-_]?values/i,
+  /authorization[-_]?response[-_]?mode/i,
+  /authorization[-_]?nonce/i
+];
+
+/*
+  // Example of redacting sensitive data from object class instances
+  function redactSensitiveData(obj) {
+    if (obj instanceof User) {
+      return {
+        ...obj.toObject(),
+        password: '***', // Redact the password field
+      };
+    }
+    return obj;
+  }
+
+  // Example of redacting sensitive data from object class instances
+  logger.info({ newUser: redactSensitiveData(newUser) }, 'newUser');
+*/
+
+const levels = {
+  TRACE: 10,
+  DEBUG: 20,
+  INFO: 30,
+  WARN: 40,
+  ERROR: 50,
+  FATAL: 60
+};
+
+let level = levels.INFO;
+
+module.exports = {
+  levels,
+  setLevel: (l) => (level = l),
+  log: {
+    trace: (msg) => {
+      if (level <= levels.TRACE) return;
+      logger.trace(msg);
+    },
+    debug: (msg) => {
+      if (level <= levels.DEBUG) return;
+      logger.debug(msg);
+    },
+    info: (msg) => {
+      if (level <= levels.INFO) return;
+      logger.info(msg);
+    },
+    warn: (msg) => {
+      if (level <= levels.WARN) return;
+      logger.warn(msg);
+    },
+    error: (msg) => {
+      if (level <= levels.ERROR) return;
+      logger.error(msg);
+    },
+    fatal: (msg) => {
+      if (level <= levels.FATAL) return;
+      logger.fatal(msg);
+    },
+
+    // Custom loggers
+    parameters: (parameters) => {
+      if (level <= levels.TRACE) return;
+      logger.debug({ parameters }, 'Function Parameters');
+    },
+    functionName: (name) => {
+      if (level <= levels.TRACE) return;
+      logger.debug(`EXECUTING: ${name}`);
+    },
+    flow: (flow) => {
+      if (level <= levels.INFO) return;
+      logger.debug(`BEGIN FLOW: ${flow}`);
+    },
+    variable: ({ name, value }) => {
+      if (level <= levels.DEBUG) return;
+      // Check if the variable name matches any of the redact patterns and redact the value
+      let sanitizedValue = value;
+      for (const pattern of redactPatterns) {
+        if (pattern.test(name)) {
+          sanitizedValue = '***';
+          break;
+        }
+      }
+      logger.debug({ variable: { name, value: sanitizedValue } }, `VARIABLE ${name}`);
+    },
+    request: () => (req, res, next) => {
+      if (level < levels.DEBUG) return next();
+      logger.debug({ query: req.query, body: req.body }, `Hit URL ${req.url} with following`);
+      return next();
+    }
+  }
+};

api/utils/debug.js

@@ -0,0 +1,46 @@
+const levels = {
+  NONE: 0,
+  LOW: 1,
+  MEDIUM: 2,
+  HIGH: 3
+};
+
+let level = levels.HIGH;
+
+module.exports = {
+  levels,
+  setLevel: (l) => (level = l),
+  log: {
+    parameters: (parameters) => {
+      if (levels.HIGH > level) return;
+      console.group();
+      parameters.forEach((p) => console.log(`${p.name}:`, p.value));
+      console.groupEnd();
+    },
+    functionName: (name) => {
+      if (levels.MEDIUM > level) return;
+      console.log(`\nEXECUTING: ${name}\n`);
+    },
+    flow: (flow) => {
+      if (levels.LOW > level) return;
+      console.log(`\n\n\nBEGIN FLOW: ${flow}\n\n\n`);
+    },
+    variable: ({ name, value }) => {
+      if (levels.HIGH > level) return;
+      console.group();
+      console.group();
+      console.log(`VARIABLE ${name}:`, value);
+      console.groupEnd();
+      console.groupEnd();
+    },
+    request: () => (req, res, next) => {
+      if (levels.HIGH > level) return next();
+      console.log('Hit URL', req.url, 'with following:');
+      console.group();
+      console.log('Query:', req.query);
+      console.log('Body:', req.body);
+      console.groupEnd();
+      return next();
+    }
+  }
+};

api/utils/emails/passwordReset.handlebars

@@ -0,0 +1,11 @@
+<html>
+    <head>
+        <style>
+
+        </style>
+    </head>
+    <body>
+        <p>Hi {{name}},</p>
+        <p>Your password has been changed successfully.</p>
+    </body>
+</html>

api/utils/emails/requestPasswordReset.handlebars

@@ -0,0 +1,13 @@
+<html>
+    <head>
+        <style>
+
+        </style>
+    </head>
+    <body>
+        <p>Hi {{name}},</p>
+        <h1>You have requested to reset your password.</h1>
+        <p> Please click the link below to reset your password.</p>
+        <a href="{{link}}">Reset Password</a>
+    </body>
+</html>

api/utils/genAzureEndpoints.js

@@ -0,0 +1,9 @@
+function genAzureEndpoint({
+  azureOpenAIApiInstanceName,
+  azureOpenAIApiDeploymentName,
+  azureOpenAIApiVersion
+}) {
+  return `https://${azureOpenAIApiInstanceName}.openai.azure.com/openai/deployments/${azureOpenAIApiDeploymentName}/chat/completions?api-version=${azureOpenAIApiVersion}`;
+}
+
+module.exports = { genAzureEndpoint };

api/utils/migrateDataToFirstUser.js

@@ -0,0 +1,29 @@
+const Conversation = require('../models/schema/convoSchema');
+const Preset = require('../models/schema/presetSchema');
+
+const migrateConversations = async (userId) => {
+  try {
+    return await Conversation.updateMany({ user: null }, { $set: { user: userId } }).exec();
+  } catch (error) {
+    console.log(error);
+    return { message: 'Error saving conversation' };
+  }
+};
+
+const migratePresets = async (userId) => {
+  try {
+    return await Preset.updateMany({ user: null }, { $set: { user: userId } }).exec();
+  } catch (error) {
+    console.log(error);
+    return { message: 'Error saving conversation' };
+  }
+};
+
+const migrateDataToFirstUser = async (user) => {
+  const conversations = await migrateConversations(user.id);
+  console.log(conversations);
+  const presets = await migratePresets(user.id);
+  console.log(presets);
+};
+
+module.exports = migrateDataToFirstUser;

api/utils/sendEmail.js

@@ -0,0 +1,54 @@
+const nodemailer = require('nodemailer');
+const handlebars = require('handlebars');
+const fs = require('fs');
+const path = require('path');
+
+const sendEmail = async (email, subject, payload, template) => {
+  try {
+    // create reusable transporter object using the default SMTP transport
+    const transporter = nodemailer.createTransport({
+      host: process.env.EMAIL_HOST,
+      port: 465,
+      auth: {
+        user: process.env.EMAIL_USERNAME,
+        pass: process.env.EMAIL_PASSWORD
+      }
+    });
+
+    const source = fs.readFileSync(path.join(__dirname, template), 'utf8');
+    const compiledTemplate = handlebars.compile(source);
+    const options = () => {
+      return {
+        from: process.env.FROM_EMAIL,
+        to: email,
+        subject: subject,
+        html: compiledTemplate(payload)
+      };
+    };
+
+    // Send email
+    transporter.sendMail(options(), (error, info) => {
+      if (error) {
+        return error;
+      } else {
+        return res.status(200).json({
+          success: true
+        });
+      }
+    });
+  } catch (error) {
+    return error;
+  }
+};
+
+/*
+Example:
+sendEmail(
+  "youremail@gmail.com,
+  "Email subject",
+  { name: "Eze" },
+  "./templates/layouts/main.handlebars"
+);
+*/
+
+module.exports = sendEmail;

client/.env.example

@@ -0,0 +1,23 @@
+###########################
+# App configuration:
+###########################
+
+# Custom app name, this text will be displayed in the landing page and the footer.
+VITE_APP_TITLE="ChatGPT Clone"
+
+###########################
+# Server URL configuration: 
+###########################
+
+# The social login domain uses this to redirect to localhost:3080 when you run the app in dev mode with Vite.
+# Use your domain name as the Prod URL when you deploy the app to a live domain.
+# Please note that:  
+# Social login features will not work if you run the build version on port 3080 locally after modifying the Prod URL 
+VITE_SERVER_URL_DEV=http://localhost:3080
+VITE_SERVER_URL_PROD=http://localhost:3080
+
+# Enable Social Login
+# This enables/disables the Login with Google button on the login page.
+# Set to true if you have registered the app with google cloud services 
+# and have set the GOOGLE_CLIENT_ID and GOOGLE_CLIENT_SECRET in the /api/.env file
+VITE_SHOW_GOOGLE_LOGIN_OPTION=false

client/.eslintrc.js

@@ -1,30 +0,0 @@
-module.exports = {
-  "env": {
-      "browser": true,
-      "es2021": true,
-      "node": true,
-      "commonjs": true,
-      "es6": true,
-  },
-  "extends": [
-      "eslint:recommended",
-      "plugin:react/recommended",
-      "plugin:react-hooks/recommended"
-  ],
-  "overrides": [
-  ],
-  "parserOptions": {
-      "ecmaFeatures": {
-          "jsx": true
-      },
-      "ecmaVersion": "latest",
-      "sourceType": "module"
-  },
-  "plugins": [
-      "react"
-  ],
-  "rules": {
-      'react/prop-types': ['off'],
-      'react/display-name': ['off'],
-  }
-}

client/.prettierrc

@@ -1,22 +0,0 @@
-{
-  "arrowParens": "avoid",
-  "bracketSpacing": true,
-  "endOfLine": "lf",
-  "htmlWhitespaceSensitivity": "css",
-  "insertPragma": false,
-  "singleAttributePerLine": true,
-  "bracketSameLine": false,
-  "jsxBracketSameLine": false,
-  "jsxSingleQuote": false,
-  "printWidth": 110,
-  "proseWrap": "preserve",
-  "quoteProps": "as-needed",
-  "requirePragma": false,
-  "semi": true,
-  "singleQuote": true,
-  "tabWidth": 2,
-  "trailingComma": "none",
-  "useTabs": false,
-  "vueIndentScriptAndStyle": false,
-  "parser": "babel"
-}

client/package.json

@@ -1,13 +1,12 @@
 {
-  "name": "chatgpt-clone",
-  "version": "0.3.0",
+  "name": "chat-frontend",
+  "version": "0.4.6",
   "description": "",
   "type": "module",
   "scripts": {
     "build": "vite build",
     "dev": "vite",
-    "preview-prod": "vite preview",
-    "build-dev": "Webpack . --watch"
+    "preview-prod": "vite preview"
   },
   "repository": {
     "type": "git",
@@ -21,35 +20,48 @@
   },
   "homepage": "https://github.com/danny-avila/chatgpt-clone#readme",
   "dependencies": {
+    "@fortawesome/fontawesome-svg-core": "^6.4.0",
+    "@fortawesome/free-brands-svg-icons": "^6.4.0",
+    "@fortawesome/free-regular-svg-icons": "^6.4.0",
+    "@fortawesome/free-solid-svg-icons": "^6.4.0",
+    "@fortawesome/react-fontawesome": "^0.2.0",
     "@headlessui/react": "^1.7.13",
     "@radix-ui/react-alert-dialog": "^1.0.2",
     "@radix-ui/react-checkbox": "^1.0.3",
     "@radix-ui/react-dialog": "^1.0.2",
     "@radix-ui/react-dropdown-menu": "^2.0.2",
     "@radix-ui/react-hover-card": "^1.0.5",
+    "@radix-ui/react-icons": "^1.3.0",
     "@radix-ui/react-label": "^2.0.0",
     "@radix-ui/react-slider": "^1.1.1",
     "@radix-ui/react-tabs": "^1.0.3",
+    "@tailwindcss/forms": "^0.5.3",
+    "@tanstack/react-query": "^4.28.0",
     "@types/jest": "^29.5.0",
     "@types/node": "^18.15.10",
     "@types/react": "^18.0.30",
     "@types/react-dom": "^18.0.11",
     "@zattoo/use-double-click": "1.2.0",
     "axios": "^1.3.4",
-    "class-variance-authority": "^0.4.0",
+    "class-variance-authority": "^0.6.0",
     "clsx": "^1.2.1",
     "copy-to-clipboard": "^3.3.3",
     "crypto-browserify": "^3.12.0",
+    "downloadjs": "^1.4.7",
     "esbuild": "0.17.15",
     "export-from-json": "^1.7.2",
+    "filenamify": "^6.0.0",
+    "html2canvas": "^1.4.1",
     "lodash": "^4.17.21",
-    "lucide-react": "^0.113.0",
+    "lucide-react": "^0.220.0",
+    "pino": "^8.12.1",
     "rc-input-number": "^7.4.2",
     "react": "^18.2.0",
     "react-dom": "^18.2.0",
+    "react-hook-form": "^7.43.9",
     "react-lazy-load": "^4.0.1",
     "react-markdown": "^8.0.6",
-    "react-router-dom": "^6.9.0",
+    "react-router-dom": "^6.11.2",
     "react-string-replace": "^1.1.0",
     "react-textarea-autosize": "^8.4.0",
     "react-transition-group": "^4.4.5",
@@ -60,7 +72,6 @@
     "remark-gfm": "^3.0.1",
     "remark-math": "^5.1.1",
     "remark-supersub": "^1.0.0",
-    "swr": "^2.0.3",
     "tailwind-merge": "^1.9.1",
     "tailwindcss-animate": "^1.0.5",
     "tailwindcss-radix": "^2.8.0",
@@ -74,34 +85,29 @@
     "@babel/plugin-transform-runtime": "^7.19.6",
     "@babel/preset-env": "^7.20.2",
     "@babel/preset-react": "^7.18.6",
+    "@babel/preset-typescript": "^7.21.0",
     "@babel/runtime": "^7.20.13",
+    "@tanstack/react-query-devtools": "^4.29.0",
+    "@types/jest": "^29.5.0",
+    "@types/node": "^18.15.10",
+    "@types/react": "^18.0.30",
+    "@types/react-dom": "^18.0.11",
     "@vitejs/plugin-react": "^3.1.0",
     "autoprefixer": "^10.4.13",
     "babel-loader": "^9.1.2",
     "babel-plugin-root-import": "^6.6.0",
     "babel-preset-react": "^6.24.1",
     "css-loader": "^6.7.3",
-    "eslint": "^8.33.0",
-    "eslint-config-airbnb-base": "^15.0.0",
-    "eslint-config-prettier": "^8.6.0",
-    "eslint-plugin-jest": "^27.2.1",
-    "eslint-plugin-react": "^7.32.2",
-    "eslint-plugin-react-hooks": "^4.6.0",
     "path": "^0.12.7",
     "postcss": "^8.4.21",
     "postcss-loader": "^7.1.0",
     "postcss-preset-env": "^8.2.0",
-    "prettier": "^2.8.3",
-    "prettier-plugin-tailwindcss": "^0.2.2",
-    "source-map-loader": "^1.1.3",
+    "source-map-loader": "^4.0.1",
     "style-loader": "^3.3.1",
     "tailwindcss": "^3.2.6",
     "ts-loader": "^9.4.2",
-    "typescript": "^4.9.5",
+    "typescript": "^5.0.4",
     "vite": "^4.2.1",
-    "vite-plugin-html": "^3.2.0",
-    "webpack": "^5.77.0",
-    "webpack-cli": "^5.0.1",
-    "webpack-dev-server": "^4.11.1"
+    "vite-plugin-html": "^3.2.0"
   }
 }

client/src/App.jsx

@@ -1,100 +1,92 @@
-import React, { useEffect, useState } from 'react';
-import { createBrowserRouter, RouterProvider, Navigate } from 'react-router-dom';
+import { createBrowserRouter, RouterProvider, Navigate, Outlet } from 'react-router-dom';
 import Root from './routes/Root';
 import Chat from './routes/Chat';
 import Search from './routes/Search';
-import store from './store';
-import userAuth from './utils/userAuth';
-import { useRecoilState, useSetRecoilState } from 'recoil';
-
-import axios from 'axios';
+import { ScreenshotProvider } from './utils/screenshotContext.jsx';
+import { ReactQueryDevtools } from '@tanstack/react-query-devtools';
+import { Login, Registration, RequestPasswordReset, ResetPassword } from './components/Auth';
+import { AuthContextProvider } from './hooks/AuthContext';
+import { RecoilRoot } from 'recoil';
+import { QueryClient, QueryClientProvider, QueryCache } from '@tanstack/react-query';
+import { ThemeProvider } from './hooks/ThemeContext';
+import { useApiErrorBoundary } from './hooks/ApiErrorBoundaryContext';
+import ApiErrorWatcher from './components/Auth/ApiErrorWatcher';
 
+const AuthLayout = () => (
+  <AuthContextProvider>
+    <Outlet />
+    <ApiErrorWatcher />
+  </AuthContextProvider>
+);
 const router = createBrowserRouter([
   {
-    path: '/',
-    element: <Root />,
+    path: 'register',
+    element: <Registration />
+  },
+  {
+    path: 'forgot-password',
+    element: <RequestPasswordReset />
+  },
+  {
+    path: 'reset-password',
+    element: <ResetPassword />
+  },
+  {
+    element: <AuthLayout />,
     children: [
       {
-        index: true,
-        element: (
-          <Navigate
-            to="/chat/new"
-            replace={true}
-          />
-        )
+        path: 'login',
+        element: <Login />
       },
       {
-        path: 'chat/:conversationId?',
-        element: <Chat />
-      },
-      {
-        path: 'search/:query?',
-        element: <Search />
+        path: '/',
+        element: <Root />,
+        children: [
+          {
+            index: true,
+            element: <Navigate to="/chat/new" replace={true} />
+          },
+          {
+            path: 'chat/:conversationId?',
+            element: <Chat />
+          },
+          {
+            path: 'search/:query?',
+            element: <Search />
+          }
+        ]
       }
     ]
   }
 ]);
 
 const App = () => {
-  const [user, setUser] = useRecoilState(store.user);
-  const setIsSearchEnabled = useSetRecoilState(store.isSearchEnabled);
-  const setEndpointsConfig = useSetRecoilState(store.endpointsConfig);
-  const setPresets = useSetRecoilState(store.presets);
+  const { setError } = useApiErrorBoundary();
 
-  useEffect(() => {
-    // fetch if seatch enabled
-    axios
-      .get('/api/search/enable', {
-        timeout: 1000,
-        withCredentials: true
-      })
-      .then(res => {
-        setIsSearchEnabled(res.data);
-      });
+  const queryClient = new QueryClient({
+    queryCache: new QueryCache({
+      onError: (error) => {
+        if (error?.response?.status === 401) {
+          setError(error);
+        }
+      }
+    })
+  });
 
-    // fetch user
-    userAuth()
-      .then(user => setUser(user))
-      .catch(err => console.log(err));
-
-    // fetch models
-    axios
-      .get('/api/endpoints', {
-        timeout: 1000,
-        withCredentials: true
-      })
-      .then(({ data }) => {
-        setEndpointsConfig(data);
-      })
-      .catch(error => {
-        console.error(error);
-        console.log('Not login!');
-        window.location.href = '/auth/login';
-      });
-
-    // fetch presets
-    axios
-      .get('/api/presets', {
-        timeout: 1000,
-        withCredentials: true
-      })
-      .then(({ data }) => {
-        setPresets(data);
-      })
-      .catch(error => {
-        console.error(error);
-        console.log('Not login!');
-        window.location.href = '/auth/login';
-      });
-  }, []);
-
-  if (user)
-    return (
-      <div>
-        <RouterProvider router={router} />
-      </div>
-    );
-  else return <div className="flex h-screen"></div>;
+  return (
+    <QueryClientProvider client={queryClient}>
+      <RecoilRoot>
+        <ThemeProvider>
+          <RouterProvider router={router} />
+          <ReactQueryDevtools initialIsOpen={false} />
+        </ThemeProvider>
+      </RecoilRoot>
+    </QueryClientProvider>
+  );
 };
 
-export default App;
+export default () => (
+  <ScreenshotProvider>
+    <App />
+  </ScreenshotProvider>
+);

client/src/components/Auth/ApiErrorWatcher.tsx

@@ -0,0 +1,18 @@
+import React from 'react';
+import { useApiErrorBoundary } from '~/hooks/ApiErrorBoundaryContext';
+import { useNavigate } from 'react-router-dom';
+
+const ApiErrorWatcher = () => {
+  const { error } = useApiErrorBoundary();
+  const navigate = useNavigate();
+  React.useEffect(() => {
+    if (error?.response?.status === 500) {
+      // do something with error
+      // navigate('/login');
+    }
+  }, [error, navigate]);
+
+  return null;
+};
+
+export default ApiErrorWatcher;

client/src/components/Auth/Login.tsx

@@ -0,0 +1,198 @@
+import { useEffect } from 'react';
+import { useForm } from 'react-hook-form';
+import { TLoginUser } from '~/data-provider';
+import { useAuthContext } from '~/hooks/AuthContext';
+import { useNavigate } from 'react-router-dom';
+
+function Login() {
+  const { login, error, isAuthenticated } = useAuthContext();
+  const {
+    register,
+    handleSubmit,
+    formState: { errors }
+  } = useForm<TLoginUser>();
+
+  const navigate = useNavigate();
+
+  useEffect(() => {
+    if (isAuthenticated) {
+      navigate('/chat/new');
+    }
+  }, [isAuthenticated, navigate]);
+
+  const SERVER_URL = import.meta.env.DEV
+    ? import.meta.env.VITE_SERVER_URL_DEV
+    : import.meta.env.VITE_SERVER_URL_PROD;
+  const showGoogleLogin = import.meta.env.VITE_SHOW_GOOGLE_LOGIN_OPTION === 'true';
+
+  return (
+    <div className="flex min-h-screen flex-col items-center justify-center bg-white pt-6 sm:pt-0">
+      <div className="mt-6 w-96 overflow-hidden bg-white px-6 py-4 sm:max-w-md sm:rounded-lg">
+        <h1 className="mb-4 text-center text-3xl font-semibold">Welcome back</h1>
+        {error && (
+          <div
+            className="relative mt-4 rounded border border-red-400 bg-red-100 px-4 py-3 text-red-700"
+            role="alert"
+          >
+            Unable to login with the information provided. Please check your credentials and try
+            again.
+          </div>
+        )}
+        <form
+          className="mt-6"
+          aria-label="Login form"
+          method="POST"
+          onSubmit={handleSubmit((data) => login(data))}
+        >
+          <div className="mb-2">
+            <div className="relative">
+              <input
+                type="email"
+                id="email"
+                autoComplete="email"
+                aria-label="Email"
+                {...register('email', {
+                  required: 'Email is required',
+                  minLength: {
+                    value: 3,
+                    message: 'Email must be at least 6 characters'
+                  },
+                  maxLength: {
+                    value: 120,
+                    message: 'Email should not be longer than 120 characters'
+                  },
+                  pattern: {
+                    value: /\S+@\S+\.\S+/,
+                    message: 'You must enter a valid email address'
+                  }
+                })}
+                aria-invalid={!!errors.email}
+                className="peer block w-full appearance-none rounded-t-md border-0 border-b-2 border-gray-300 bg-gray-50 px-2.5 pb-2.5 pt-5 text-sm text-gray-900 focus:border-green-500 focus:outline-none focus:ring-0"
+                placeholder=" "
+              ></input>
+              <label
+                htmlFor="email"
+                className="absolute left-2.5 top-4 z-10 origin-[0] -translate-y-4 scale-75 transform text-gray-500 duration-300 peer-placeholder-shown:translate-y-0 peer-placeholder-shown:scale-100 peer-focus:-translate-y-4 peer-focus:scale-75 peer-focus:text-green-500"
+              >
+                Email address
+              </label>
+            </div>
+            {errors.email && (
+              <span role="alert" className="mt-1 text-sm text-red-600">
+                {/* @ts-ignore */}
+                {errors.email.message}
+              </span>
+            )}
+          </div>
+          <div className="mb-2">
+            <div className="relative">
+              <input
+                type="password"
+                id="password"
+                autoComplete="current-password"
+                aria-label="Password"
+                {...register('password', {
+                  required: 'Password is required',
+                  minLength: {
+                    value: 8,
+                    message: 'Password must be at least 8 characters'
+                  },
+                  maxLength: {
+                    value: 40,
+                    message: 'Password must be less than 40 characters'
+                  }
+                })}
+                aria-invalid={!!errors.password}
+                className="peer block w-full appearance-none rounded-t-md border-0 border-b-2 border-gray-300 bg-gray-50 px-2.5 pb-2.5 pt-5 text-sm text-gray-900 focus:border-green-500 focus:outline-none focus:ring-0"
+                placeholder=" "
+              ></input>
+              <label
+                htmlFor="password"
+                className="absolute left-2.5 top-4 z-10 origin-[0] -translate-y-4 scale-75 transform text-gray-500 duration-300 peer-placeholder-shown:translate-y-0 peer-placeholder-shown:scale-100 peer-focus:-translate-y-4 peer-focus:scale-75 peer-focus:text-green-500"
+              >
+                Password
+              </label>
+            </div>
+
+            {errors.password && (
+              <span role="alert" className="mt-1 text-sm text-red-600">
+                {/* @ts-ignore */}
+                {errors.password.message}
+              </span>
+            )}
+          </div>
+          <a href="/forgot-password" className="text-sm text-green-500 hover:underline">
+            Forgot Password?
+          </a>
+          <div className="mt-6">
+            <button
+              aria-label="Sign in"
+              type="submit"
+              className="w-full transform rounded-sm bg-green-500 px-4 py-3 tracking-wide text-white transition-colors duration-200 hover:bg-green-600 focus:bg-green-600 focus:outline-none"
+            >
+              Continue
+            </button>
+          </div>
+        </form>
+        <p className="my-4 text-center text-sm font-light text-gray-700">
+          {' '}
+          Don't have an account?{' '}
+          <a href="/register" className="p-1 text-green-500 hover:underline">
+            Sign up
+          </a>
+        </p>
+        {showGoogleLogin && (
+          <>
+            <div className="relative mt-6 flex w-full items-center justify-center border border-t uppercase">
+              <div className="absolute bg-white px-3 text-xs">Or</div>
+            </div>
+            <div className="mt-4 flex gap-x-2">
+              <a
+                aria-label="Login with Google"
+                className="justify-left flex w-full items-center space-x-3 rounded-md border border-gray-300 px-5 py-3 hover:bg-gray-50 focus:ring-2 focus:ring-violet-600 focus:ring-offset-1"
+                href={`${SERVER_URL}/oauth/google`}
+              >
+                <svg
+                  xmlns="http://www.w3.org/2000/svg"
+                  viewBox="0 0 512 512"
+                  id="google"
+                  className="h-5 w-5"
+                >
+                  <path
+                    fill="#fbbb00"
+                    d="M113.47 309.408 95.648 375.94l-65.139 1.378C11.042 341.211 0 299.9 0 256c0-42.451 10.324-82.483 28.624-117.732h.014L86.63 148.9l25.404 57.644c-5.317 15.501-8.215 32.141-8.215 49.456.002 18.792 3.406 36.797 9.651 53.408z"
+                  ></path>
+                  <path
+                    fill="#518ef8"
+                    d="M507.527 208.176C510.467 223.662 512 239.655 512 256c0 18.328-1.927 36.206-5.598 53.451-12.462 58.683-45.025 109.925-90.134 146.187l-.014-.014-73.044-3.727-10.338-64.535c29.932-17.554 53.324-45.025 65.646-77.911h-136.89V208.176h245.899z"
+                  ></path>
+                  <path
+                    fill="#28b446"
+                    d="m416.253 455.624.014.014C372.396 490.901 316.666 512 256 512c-97.491 0-182.252-54.491-225.491-134.681l82.961-67.91c21.619 57.698 77.278 98.771 142.53 98.771 28.047 0 54.323-7.582 76.87-20.818l83.383 68.262z"
+                  ></path>
+                  <path
+                    fill="#f14336"
+                    d="m419.404 58.936-82.933 67.896C313.136 112.246 285.552 103.82 256 103.82c-66.729 0-123.429 42.957-143.965 102.724l-83.397-68.276h-.014C71.23 56.123 157.06 0 256 0c62.115 0 119.068 22.126 163.404 58.936z"
+                  ></path>
+                </svg>
+                <p>Login with Google</p>
+              </a>
+
+              {/* <a 
+                aria-label="Login with Facebook"
+                className="flex w-full items-center justify-center rounded-md border border-gray-600 p-2 focus:ring-2 focus:ring-violet-600 focus:ring-offset-1"
+                href="http://localhost:3080/auth/facebook">
+                <FontAwesomeIcon
+                  icon={faFacebook} 
+                  size={'lg'}
+                />
+              </a> */}
+            </div>
+          </>
+        )}
+      </div>
+    </div>
+  );
+}
+
+export default Login;

client/src/components/Auth/Registration.tsx

@@ -0,0 +1,329 @@
+import { useState } from 'react';
+import { useNavigate } from 'react-router-dom';
+import { useForm } from 'react-hook-form';
+import { useRegisterUserMutation, TRegisterUser } from '~/data-provider';
+import { FontAwesomeIcon } from '@fortawesome/react-fontawesome';
+import { faFacebook } from '@fortawesome/free-brands-svg-icons';
+import { faGoogle } from '@fortawesome/free-brands-svg-icons';
+
+function Registration() {
+  const SERVER_URL = import.meta.env.DEV
+    ? import.meta.env.VITE_SERVER_URL_DEV
+    : import.meta.env.VITE_SERVER_URL_PROD;
+  const showGoogleLogin = import.meta.env.VITE_SHOW_GOOGLE_LOGIN_OPTION === 'true';
+
+  const navigate = useNavigate();
+  const {
+    register,
+    watch,
+    handleSubmit,
+    formState: { errors }
+  } = useForm<TRegisterUser>({ mode: 'onChange' });
+  const [error, setError] = useState<boolean>(false);
+  const [errorMessage, setErrorMessage] = useState<string>('');
+  const registerUser = useRegisterUserMutation();
+
+  const password = watch('password');
+
+  const onRegisterUserFormSubmit = (data: TRegisterUser) => {
+    registerUser.mutate(data, {
+      onSuccess: () => {
+        navigate('/chat/new');
+      },
+      onError: (error) => {
+        setError(true);
+        if (error.response?.data?.message) {
+          setErrorMessage(error.response?.data?.message);
+        }
+      }
+    });
+  };
+
+  return (
+    <div className="flex min-h-screen flex-col items-center justify-center bg-white pt-6 sm:pt-0">
+      <div className="mt-6 w-96 overflow-hidden bg-white px-6 py-4 sm:max-w-md sm:rounded-lg">
+        <h1 className="mb-4 text-center text-3xl font-semibold">Create your account</h1>
+        {error && (
+          <div
+            className="relative mt-4 rounded border border-red-400 bg-red-100 px-4 py-3 text-red-700"
+            role="alert"
+          >
+            There was an error attempting to register your account. Please try again. {errorMessage}
+          </div>
+        )}
+        <form
+          className="mt-6"
+          aria-label="Registration form"
+          method="POST"
+          onSubmit={handleSubmit((data) => onRegisterUserFormSubmit(data))}
+        >
+          <div className="mb-2">
+            <div className="relative">
+              <input
+                id="name"
+                type="text"
+                autoComplete="name"
+                aria-label="Name"
+                // uncomment to prevent pasting in confirm field
+                onPaste={(e) => {
+                  e.preventDefault();
+                  return false;
+                }}
+                {...register('name', {
+                  required: 'Name is required',
+                  minLength: {
+                    value: 3,
+                    message: 'Name must be at least 3 characters'
+                  },
+                  maxLength: {
+                    value: 80,
+                    message: 'Name must be less than 80 characters'
+                  }
+                })}
+                aria-invalid={!!errors.name}
+                className="peer block w-full appearance-none rounded-t-md border-0 border-b-2 border-gray-300 bg-gray-50 px-2.5 pb-2.5 pt-5 text-sm text-gray-900 focus:border-green-500 focus:outline-none focus:ring-0"
+                placeholder=" "
+              ></input>
+              <label
+                htmlFor="name"
+                className="absolute left-2.5 top-4 z-10 origin-[0] -translate-y-4 scale-75 transform text-sm text-gray-500 duration-300 peer-placeholder-shown:translate-y-0 peer-placeholder-shown:scale-100 peer-focus:-translate-y-4 peer-focus:scale-75 peer-focus:text-green-500"
+              >
+                Full Name
+              </label>
+            </div>
+
+            {errors.name && (
+              <span role="alert" className="mt-1 text-sm text-red-600">
+                {/* @ts-ignore */}
+                {errors.name.message}
+              </span>
+            )}
+          </div>
+          <div className="mb-2">
+            <div className="relative">
+              <input
+                type="text"
+                id="username"
+                aria-label="Username"
+                {...register('username', {
+                  required: 'Username is required',
+                  minLength: {
+                    value: 3,
+                    message: 'Username must be at least 3 characters'
+                  },
+                  maxLength: {
+                    value: 20,
+                    message: 'Username must be less than 20 characters'
+                  }
+                })}
+                aria-invalid={!!errors.username}
+                className="peer block w-full appearance-none rounded-t-md border-0 border-b-2 border-gray-300 bg-gray-50 px-2.5 pb-2.5 pt-5 text-sm text-gray-900 focus:border-green-500 focus:outline-none focus:ring-0"
+                placeholder=" "
+                autoComplete="off"
+              ></input>
+              <label
+                htmlFor="username"
+                className="absolute left-2.5 top-4 z-10 origin-[0] -translate-y-4 scale-75 transform text-sm text-gray-500 duration-300 peer-placeholder-shown:translate-y-0 peer-placeholder-shown:scale-100 peer-focus:-translate-y-4 peer-focus:scale-75 peer-focus:text-green-500"
+              >
+                Username
+              </label>
+            </div>
+
+            {errors.username && (
+              <span role="alert" className="mt-1 text-sm text-red-600">
+                {/* @ts-ignore */}
+                {errors.username.message}
+              </span>
+            )}
+          </div>
+          <div className="mb-2">
+            <div className="relative">
+              <input
+                type="email"
+                id="email"
+                autoComplete="email"
+                aria-label="Email"
+                {...register('email', {
+                  required: 'Email is required',
+                  minLength: {
+                    value: 3,
+                    message: 'Email must be at least 6 characters'
+                  },
+                  maxLength: {
+                    value: 120,
+                    message: 'Email should not be longer than 120 characters'
+                  },
+                  pattern: {
+                    value: /\S+@\S+\.\S+/,
+                    message: 'You must enter a valid email address'
+                  }
+                })}
+                aria-invalid={!!errors.email}
+                className="peer block w-full appearance-none rounded-t-md border-0 border-b-2 border-gray-300 bg-gray-50 px-2.5 pb-2.5 pt-5 text-sm text-gray-900 focus:border-green-500 focus:outline-none focus:ring-0"
+                placeholder=" "
+              ></input>
+              <label
+                htmlFor="email"
+                className="absolute left-2.5 top-4 z-10 origin-[0] -translate-y-4 scale-75 transform text-sm text-gray-500 duration-300 peer-placeholder-shown:translate-y-0 peer-placeholder-shown:scale-100 peer-focus:-translate-y-4 peer-focus:scale-75 peer-focus:text-green-500"
+              >
+                Email
+              </label>
+            </div>
+            {errors.email && (
+              <span role="alert" className="mt-1 text-sm text-red-600">
+                {/* @ts-ignore */}
+                {errors.email.message}
+              </span>
+            )}
+          </div>
+          <div className="mb-2">
+            <div className="relative">
+              <input
+                type="password"
+                id="password"
+                autoComplete="current-password"
+                aria-label="Password"
+                {...register('password', {
+                  required: 'Password is required',
+                  minLength: {
+                    value: 8,
+                    message: 'Password must be at least 8 characters'
+                  },
+                  maxLength: {
+                    value: 40,
+                    message: 'Password must be less than 40 characters'
+                  }
+                })}
+                aria-invalid={!!errors.password}
+                className="peer block w-full appearance-none rounded-t-md border-0 border-b-2 border-gray-300 bg-gray-50 px-2.5 pb-2.5 pt-5 text-sm text-gray-900 focus:border-green-500 focus:outline-none focus:ring-0"
+                placeholder=" "
+              ></input>
+              <label
+                htmlFor="password"
+                className="absolute left-2.5 top-4 z-10 origin-[0] -translate-y-4 scale-75 transform text-sm text-gray-500 duration-300 peer-placeholder-shown:translate-y-0 peer-placeholder-shown:scale-100 peer-focus:-translate-y-4 peer-focus:scale-75 peer-focus:text-green-500"
+              >
+                Password
+              </label>
+            </div>
+
+            {errors.password && (
+              <span role="alert" className="mt-1 text-sm text-red-600">
+                {/* @ts-ignore */}
+                {errors.password.message}
+              </span>
+            )}
+          </div>
+          <div className="mb-2">
+            <div className="relative">
+              <input
+                type="password"
+                id="confirm_password"
+                aria-label="Confirm Password"
+                // uncomment to prevent pasting in confirm field
+                onPaste={(e) => {
+                  e.preventDefault();
+                  return false;
+                }}
+                {...register('confirm_password', {
+                  validate: (value) => value === password || 'Passwords do not match'
+                })}
+                aria-invalid={!!errors.confirm_password}
+                className="peer block w-full appearance-none rounded-t-md border-0 border-b-2 border-gray-300 bg-gray-50 px-2.5 pb-2.5 pt-5 text-sm text-gray-900 focus:border-green-500 focus:outline-none focus:ring-0"
+                placeholder=" "
+              ></input>
+              <label
+                htmlFor="confirm_password"
+                className="absolute left-2.5 top-4 z-10 origin-[0] -translate-y-4 scale-75 transform text-sm text-gray-500 duration-300 peer-placeholder-shown:translate-y-0 peer-placeholder-shown:scale-100 peer-focus:-translate-y-4 peer-focus:scale-75 peer-focus:text-green-500"
+              >
+                Confirm Password
+              </label>
+            </div>
+
+            {errors.confirm_password && (
+              <span role="alert" className="mt-1 text-sm text-red-600">
+                {/* @ts-ignore */}
+                {errors.confirm_password.message}
+              </span>
+            )}
+          </div>
+          <div className="mt-6">
+            <button
+              disabled={
+                !!errors.email ||
+                !!errors.name ||
+                !!errors.password ||
+                !!errors.username ||
+                !!errors.confirm_password
+              }
+              type="submit"
+              aria-label="Submit registration"
+              className="w-full transform rounded-sm bg-green-500 px-4 py-3 tracking-wide text-white transition-colors duration-200 hover:bg-green-600 focus:bg-green-600 focus:outline-none"
+            >
+              Continue
+            </button>
+          </div>
+        </form>
+        <p className="my-4 text-center text-sm font-light text-gray-700">
+          {' '}
+          Already have an account?{' '}
+          <a href="/login" className="p-1 font-medium text-green-500 hover:underline">
+            Login
+          </a>
+        </p>
+        {showGoogleLogin && (
+          <>
+            <div className="relative mt-6 flex w-full items-center justify-center border border-t uppercase">
+              <div className="absolute bg-white px-3 text-xs">Or</div>
+            </div>
+
+            <div className="mt-4 flex gap-x-2">
+              <a
+                aria-label="Login with Google"
+                href={`${SERVER_URL}/oauth/google`}
+                className="justify-left flex w-full items-center space-x-3 rounded-md border border-gray-300 px-5 py-3 hover:bg-gray-50 focus:ring-2 focus:ring-violet-600 focus:ring-offset-1"
+              >
+                <svg
+                  xmlns="http://www.w3.org/2000/svg"
+                  viewBox="0 0 512 512"
+                  id="google"
+                  className="h-5 w-5"
+                >
+                  <path
+                    fill="#fbbb00"
+                    d="M113.47 309.408 95.648 375.94l-65.139 1.378C11.042 341.211 0 299.9 0 256c0-42.451 10.324-82.483 28.624-117.732h.014L86.63 148.9l25.404 57.644c-5.317 15.501-8.215 32.141-8.215 49.456.002 18.792 3.406 36.797 9.651 53.408z"
+                  ></path>
+                  <path
+                    fill="#518ef8"
+                    d="M507.527 208.176C510.467 223.662 512 239.655 512 256c0 18.328-1.927 36.206-5.598 53.451-12.462 58.683-45.025 109.925-90.134 146.187l-.014-.014-73.044-3.727-10.338-64.535c29.932-17.554 53.324-45.025 65.646-77.911h-136.89V208.176h245.899z"
+                  ></path>
+                  <path
+                    fill="#28b446"
+                    d="m416.253 455.624.014.014C372.396 490.901 316.666 512 256 512c-97.491 0-182.252-54.491-225.491-134.681l82.961-67.91c21.619 57.698 77.278 98.771 142.53 98.771 28.047 0 54.323-7.582 76.87-20.818l83.383 68.262z"
+                  ></path>
+                  <path
+                    fill="#f14336"
+                    d="m419.404 58.936-82.933 67.896C313.136 112.246 285.552 103.82 256 103.82c-66.729 0-123.429 42.957-143.965 102.724l-83.397-68.276h-.014C71.23 56.123 157.06 0 256 0c62.115 0 119.068 22.126 163.404 58.936z"
+                  ></path>
+                </svg>
+                <p>Login with Google</p>
+              </a>
+              {/* <button
+                  aria-label="Login with Facebook"
+                  role="button"
+                  className="flex w-full items-center justify-center space-x-3 rounded-md border p-4 focus:ring-2 focus:ring-violet-400 focus:ring-offset-1 dark:border-gray-400"
+                >
+                  <FontAwesomeIcon
+                    icon={faFacebook} 
+                    size={'lg'}
+                  />
+                  <p>Login with Facebook</p>
+                </button> */}
+            </div>
+          </>
+        )}
+      </div>
+    </div>
+  );
+}
+
+export default Registration;

client/src/components/Auth/RequestPasswordReset.tsx

@@ -0,0 +1,118 @@
+import { useState } from 'react';
+import { useForm } from 'react-hook-form';
+import { useRequestPasswordResetMutation, TRequestPasswordReset } from '~/data-provider';
+
+function RequestPasswordReset() {
+  const {
+    register,
+    handleSubmit,
+    formState: { errors }
+  } = useForm<TRequestPasswordReset>();
+  const requestPasswordReset = useRequestPasswordResetMutation();
+  const [success, setSuccess] = useState<boolean>(false);
+  const [requestError, setRequestError] = useState<boolean>(false);
+  const [resetLink, setResetLink] = useState<string>('');
+
+  const onSubmit = (data: TRequestPasswordReset) => {
+    requestPasswordReset.mutate(data, {
+      onSuccess: (data) => {
+        setSuccess(true);
+        setResetLink(data.link);
+      },
+      onError: () => {
+        setRequestError(true);
+        setTimeout(() => {
+          setRequestError(false);
+        }, 5000);
+      }
+    });
+  };
+
+  return (
+    <div className="flex min-h-screen flex-col items-center justify-center bg-white pt-6 sm:pt-0">
+      <div className="mt-6 w-96 overflow-hidden bg-white px-6 py-4 sm:max-w-md sm:rounded-lg">
+        <h1 className="mb-4 text-center text-3xl font-semibold">Reset your password</h1>
+        {success && (
+          <div
+            className="relative mt-4 rounded border border-green-400 bg-green-100 px-4 py-3 text-green-700"
+            role="alert"
+          >
+            Click{' '}
+            <a className="text-green-600 hover:underline" href={resetLink}>
+              HERE
+            </a>{' '}
+            to reset your password.
+            {/* An email has been sent with instructions on how to reset your password. */}
+          </div>
+        )}
+        {requestError && (
+          <div
+            className="relative mt-4 rounded border border-red-400 bg-red-100 px-4 py-3 text-red-700"
+            role="alert"
+          >
+            There was a problem resetting your password. There was no user found with the email
+            address provided. Please try again.
+          </div>
+        )}
+        <form
+          className="mt-6"
+          aria-label="Password reset form"
+          method="POST"
+          onSubmit={handleSubmit(onSubmit)}
+        >
+          <div className="mb-2">
+            <div className="relative">
+              <input
+                type="email"
+                id="email"
+                autoComplete="off"
+                aria-label="Email"
+                {...register('email', {
+                  required: 'Email is required',
+                  minLength: {
+                    value: 3,
+                    message: 'Email must be at least 6 characters'
+                  },
+                  maxLength: {
+                    value: 120,
+                    message: 'Email should not be longer than 120 characters'
+                  },
+                  pattern: {
+                    value: /\S+@\S+\.\S+/,
+                    message: 'You must enter a valid email address'
+                  }
+                })}
+                aria-invalid={!!errors.email}
+                className="peer block w-full appearance-none rounded-t-md border-0 border-b-2 border-gray-300 bg-gray-50 px-2.5 pb-2.5 pt-5 text-sm text-gray-900 focus:border-green-500 focus:outline-none focus:ring-0"
+                placeholder=" "
+              ></input>
+              <label
+                htmlFor="email"
+                className="absolute left-2.5 top-4 z-10 origin-[0] -translate-y-4 scale-75 transform text-gray-500 duration-300 peer-placeholder-shown:translate-y-0 peer-placeholder-shown:scale-100 peer-focus:-translate-y-4 peer-focus:scale-75 peer-focus:text-green-500"
+              >
+                Email address
+              </label>
+            </div>
+            {errors.email && (
+              <span role="alert" className="mt-1 text-sm text-red-600">
+                {/* @ts-ignore */}
+                {errors.email.message}
+              </span>
+            )}
+          </div>
+          <div className="mt-6">
+            <button
+              type="submit"
+              disabled={!!errors.email}
+              className="w-full rounded-sm border border-transparent bg-green-500 px-4 py-2 text-sm font-medium text-white shadow-sm hover:bg-green-600 focus:outline-none active:bg-green-500"
+            >
+              Continue
+            </button>
+          </div>
+        </form>
+      </div>
+    </div>
+  );
+}
+
+export default RequestPasswordReset;