feat: initial implementation 🚀

Author: rpidanny

.gitignore

@@ -32,4 +32,6 @@ lerna-debug.log*
 !.vscode/settings.json
 !.vscode/tasks.json
 !.vscode/launch.json
-!.vscode/extensions.json
+!.vscode/extensions.json
+
+.env

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+  "typescript.tsdk": "node_modules/typescript/lib"
+}

Dockerfile

@@ -0,0 +1,40 @@
+#
+# Base image
+#
+
+FROM mcr.microsoft.com/playwright:v1.32.0-jammy as base
+
+RUN apt update -y && apt upgrade -y && \
+  apt install -y bash && \
+  mkdir /opt/app
+
+WORKDIR /opt/app
+
+#
+# Build image
+#
+
+FROM base as build
+
+RUN apt update -y && apt upgrade -y && \
+  apt install -y make gcc g++
+
+COPY package.json yarn.lock ./
+
+RUN yarn 
+
+COPY . .
+
+RUN yarn build
+
+#
+# Runtime image
+#
+
+FROM base
+
+ENV NODE_ENV production
+
+COPY --from=build /opt/app .
+
+CMD bash -c 'node dist/main'

README.md

@@ -1,73 +1,3 @@
-<p align="center">
-  <a href="http://nestjs.com/" target="blank"><img src="https://nestjs.com/img/logo-small.svg" width="200" alt="Nest Logo" /></a>
-</p>
+# ChatGPT Browser API Proxy
 
-[circleci-image]: https://img.shields.io/circleci/build/github/nestjs/nest/master?token=abc123def456
-[circleci-url]: https://circleci.com/gh/nestjs/nest
-
-  <p align="center">A progressive <a href="http://nodejs.org" target="_blank">Node.js</a> framework for building efficient and scalable server-side applications.</p>
-    <p align="center">
-<a href="https://www.npmjs.com/~nestjscore" target="_blank"><img src="https://img.shields.io/npm/v/@nestjs/core.svg" alt="NPM Version" /></a>
-<a href="https://www.npmjs.com/~nestjscore" target="_blank"><img src="https://img.shields.io/npm/l/@nestjs/core.svg" alt="Package License" /></a>
-<a href="https://www.npmjs.com/~nestjscore" target="_blank"><img src="https://img.shields.io/npm/dm/@nestjs/common.svg" alt="NPM Downloads" /></a>
-<a href="https://circleci.com/gh/nestjs/nest" target="_blank"><img src="https://img.shields.io/circleci/build/github/nestjs/nest/master" alt="CircleCI" /></a>
-<a href="https://coveralls.io/github/nestjs/nest?branch=master" target="_blank"><img src="https://coveralls.io/repos/github/nestjs/nest/badge.svg?branch=master#9" alt="Coverage" /></a>
-<a href="https://discord.gg/G7Qnnhy" target="_blank"><img src="https://img.shields.io/badge/discord-online-brightgreen.svg" alt="Discord"/></a>
-<a href="https://opencollective.com/nest#backer" target="_blank"><img src="https://opencollective.com/nest/backers/badge.svg" alt="Backers on Open Collective" /></a>
-<a href="https://opencollective.com/nest#sponsor" target="_blank"><img src="https://opencollective.com/nest/sponsors/badge.svg" alt="Sponsors on Open Collective" /></a>
-  <a href="https://paypal.me/kamilmysliwiec" target="_blank"><img src="https://img.shields.io/badge/Donate-PayPal-ff3f59.svg"/></a>
-    <a href="https://opencollective.com/nest#sponsor"  target="_blank"><img src="https://img.shields.io/badge/Support%20us-Open%20Collective-41B883.svg" alt="Support us"></a>
-  <a href="https://twitter.com/nestframework" target="_blank"><img src="https://img.shields.io/twitter/follow/nestframework.svg?style=social&label=Follow"></a>
-</p>
-  <!--[![Backers on Open Collective](https://opencollective.com/nest/backers/badge.svg)](https://opencollective.com/nest#backer)
-  [![Sponsors on Open Collective](https://opencollective.com/nest/sponsors/badge.svg)](https://opencollective.com/nest#sponsor)-->
-
-## Description
-
-[Nest](https://github.com/nestjs/nest) framework TypeScript starter repository.
-
-## Installation
-
-```bash
-$ yarn install
-```
-
-## Running the app
-
-```bash
-# development
-$ yarn run start
-
-# watch mode
-$ yarn run start:dev
-
-# production mode
-$ yarn run start:prod
-```
-
-## Test
-
-```bash
-# unit tests
-$ yarn run test
-
-# e2e tests
-$ yarn run test:e2e
-
-# test coverage
-$ yarn run test:cov
-```
-
-## Support
-
-Nest is an MIT-licensed open source project. It can grow thanks to the sponsors and support by the amazing backers. If you'd like to join them, please [read more here](https://docs.nestjs.com/support).
-
-## Stay in touch
-
-- Author - [Kamil Myśliwiec](https://kamilmysliwiec.com)
-- Website - [https://nestjs.com](https://nestjs.com/)
-- Twitter - [@nestframework](https://twitter.com/nestframework)
-
-## License
-
-Nest is [MIT licensed](LICENSE).
+Exposes OpenAI APIs using ChatGPT Browser API.

example.env

@@ -0,0 +1 @@
+OPENAI_API_TOKEN=<Your OpenAI API Token>

package.json

@@ -22,9 +22,12 @@
   },
   "dependencies": {
     "@nestjs/common": "^9.0.0",
+    "@nestjs/config": "^2.3.2",
     "@nestjs/core": "^9.0.0",
     "@nestjs/mapped-types": "*",
     "@nestjs/platform-express": "^9.0.0",
+    "class-transformer": "^0.5.1",
+    "class-validator": "^0.14.0",
     "playwright": "^1.33.0",
     "reflect-metadata": "^0.1.13",
     "rxjs": "^7.2.0"

src/app.module.ts

@@ -1,11 +1,31 @@
-import { Module } from '@nestjs/common';
+import {
+  Logger,
+  MiddlewareConsumer,
+  Module,
+  RequestMethod,
+} from '@nestjs/common';
+import { ConfigModule } from '@nestjs/config';
 
 import { CompletionModule } from './completion/completion.module.js';
+import { ChatModule } from './chat/chat.module.js';
 import { GlobalModule } from './global/global.module.js';
+import { RequestLoggerMiddleware } from './global/middlewares/request-logger.js';
 
 @Module({
-  imports: [GlobalModule.forRootAsync(), CompletionModule],
+  imports: [
+    ConfigModule.forRoot(),
+    GlobalModule.forRootAsync(),
+    CompletionModule,
+    ChatModule,
+  ],
   controllers: [],
-  providers: [],
+  providers: [Logger],
 })
-export class AppModule {}
+export class AppModule {
+  configure(consumer: MiddlewareConsumer): any {
+    consumer.apply(RequestLoggerMiddleware).forRoutes({
+      path: '*',
+      method: RequestMethod.ALL,
+    });
+  }
+}

src/chat/chat.controller.spec.ts

@@ -0,0 +1,20 @@
+import { Test, TestingModule } from '@nestjs/testing';
+import { ChatController } from './chat.controller.js';
+import { ChatService } from './chat.service.js';
+
+describe('ChatController', () => {
+  let controller: ChatController;
+
+  beforeEach(async () => {
+    const module: TestingModule = await Test.createTestingModule({
+      controllers: [ChatController],
+      providers: [ChatService],
+    }).compile();
+
+    controller = module.get<ChatController>(ChatController);
+  });
+
+  it('should be defined', () => {
+    expect(controller).toBeDefined();
+  });
+});

src/chat/chat.controller.ts

@@ -0,0 +1,19 @@
+import { Body, Controller, Post } from '@nestjs/common';
+
+import { ChatService } from './chat.service.js';
+import {
+  ChatCompletionRequestV1DTO,
+  ChatCompletionResponseV1DTO,
+} from './chat.dto.js';
+
+@Controller('v1/chat')
+export class ChatController {
+  constructor(private readonly chatService: ChatService) {}
+
+  @Post('completions')
+  async complete(
+    @Body() ChatDto: ChatCompletionRequestV1DTO
+  ): Promise<ChatCompletionResponseV1DTO> {
+    return await this.chatService.complete(ChatDto.messages);
+  }
+}

src/chat/chat.dto.ts

@@ -0,0 +1,103 @@
+import { Expose } from 'class-transformer';
+import {
+  IsArray,
+  IsBoolean,
+  IsNumber,
+  IsObject,
+  IsOptional,
+  IsString,
+} from 'class-validator';
+
+@Expose()
+export class LogprobsV1DTO {
+  tokens: string[];
+  token_logprobs: number[];
+  top_logprobs: Record<string, number[]>;
+  text_offset: number[];
+  text: string;
+  finish_reason: string;
+}
+
+export class UsageV1DTO {
+  prompt_tokens: number;
+  completion_tokens: number;
+  total_tokens: number;
+}
+
+export class MessageV1DTO {
+  role: string;
+  content: string;
+  name?: string
+}
+
+export class ChatCompletionRequestV1DTO {
+  @IsString()
+  model!: string;
+
+  @IsArray()
+  messages!: MessageV1DTO[];
+
+
+  @IsNumber()
+  @IsOptional()
+  temperature?: number;
+
+  @IsNumber()
+  @IsOptional()
+  top_p?: number;
+
+  @IsNumber()
+  @IsOptional()
+  n?: number;
+
+  @IsBoolean()
+  @IsOptional()
+  stream?: number;
+
+  @IsOptional()
+  stop?: string | string[];
+
+  @IsNumber()
+  @IsOptional()
+  max_tokens?: number;
+
+  @IsNumber()
+  @IsOptional()
+  presence_penalty?: number;
+
+  @IsNumber()
+  @IsOptional()
+  frequency_penalty?: number;
+
+  @IsObject()
+  @IsOptional()
+  logit_bias?: Record<string, number>;
+
+  @IsString()
+  @IsOptional()
+  user?: string;
+}
+
+export class ChatCompletionResponseV1DTO {
+  @Expose()
+  id: string;
+
+  @Expose()
+  object: string;
+
+  @Expose()
+  created: number;
+
+  @Expose()
+  choices: ChoiceV1DTO[];
+
+  @Expose()
+  usage: UsageV1DTO;
+}
+
+@Expose()
+export class ChoiceV1DTO {
+  index: number;
+  message: MessageV1DTO;
+  finish_reason: string;
+}

src/chat/chat.module.ts

@@ -0,0 +1,10 @@
+import { Module } from '@nestjs/common';
+
+import { ChatController } from './chat.controller.js';
+import { ChatService } from './chat.service.js';
+
+@Module({
+  controllers: [ChatController],
+  providers: [ChatService],
+})
+export class ChatModule {}

src/chat/chat.service.spec.ts

@@ -0,0 +1,19 @@
+import { Test, TestingModule } from '@nestjs/testing';
+
+import { ChatService } from './chat.service.js';
+
+describe('ChatService', () => {
+  let service: ChatService;
+
+  beforeEach(async () => {
+    const module: TestingModule = await Test.createTestingModule({
+      providers: [ChatService],
+    }).compile();
+
+    service = module.get<ChatService>(ChatService);
+  });
+
+  it('should be defined', () => {
+    expect(service).toBeDefined();
+  });
+});