librechat-data-provider 0.1.0

package/jest.config.js

@@ -0,0 +1,18 @@
+module.exports = {
+  collectCoverageFrom: ['src/**/*.{js,jsx,ts,tsx}', '!<rootDir>/node_modules/'],
+  coveragePathIgnorePatterns: ['/node_modules/', '/dist/'],
+  coverageReporters: ['text', 'cobertura'],
+  testResultsProcessor: 'jest-junit',
+  moduleNameMapper: {
+    '^@src/(.*)$': '<rootDir>/src/$1',
+  },
+  // coverageThreshold: {
+  //   global: {
+  //     statements: 58,
+  //     branches: 49,
+  //     functions: 50,
+  //     lines: 57,
+  //   },
+  // },
+  restoreMocks: true,
+};

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "librechat-data-provider",
+  "version": "0.1.0",
+  "description": "data services for librechat apps",
+  "main": "dist/index.js",
+  "module": "dist/index.es.js",
+  "types": "types/",
+  "scripts": {
+    "clean": "rimraf dist",
+    "build": "npm run clean && rollup -c --silent --bundleConfigAsCjs",
+    "build:watch": "rollup -c -w",
+    "test": "jest --coverage --watch",
+    "test:ci": "jest --coverage --ci",
+    "verify": "npm run test:ci"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/danny-avila/LibreChat.git"
+  },
+  "author": "",
+  "license": "ISC",
+  "bugs": {
+    "url": "https://github.com/danny-avila/LibreChat/issues"
+  },
+  "homepage": "https://github.com/danny-avila/LibreChat#readme",
+  "dependencies": {
+    "@tanstack/react-query": "^4.28.0",
+    "axios": "^1.3.4"
+  },
+  "devDependencies": {
+    "@babel/preset-env": "^7.21.5",
+    "@babel/preset-react": "^7.18.6",
+    "@babel/preset-typescript": "^7.21.0",
+    "@rollup/plugin-commonjs": "^25.0.2",
+    "@rollup/plugin-node-resolve": "^15.1.0",
+    "@tanstack/query-core": "^4.29.19",
+    "@types/jest": "^29.5.2",
+    "@types/node": "^20.3.0",
+    "jest": "^29.5.0",
+    "jest-junit": "^16.0.0",
+    "rimraf": "^5.0.1",
+    "rollup": "^3.26.0",
+    "rollup-plugin-typescript2": "^0.35.0",
+    "typescript": "^5.0.4"
+  },
+  "publishConfig": {
+    "registry": "https://registry.npmjs.org/"
+  }  
+}

package/rollup.config.js

@@ -0,0 +1,31 @@
+import typescript from 'rollup-plugin-typescript2';
+import resolve from '@rollup/plugin-node-resolve';
+import pkg from './package.json';
+
+export default [
+  {
+    input: 'src/index.ts',
+    output: [
+      {
+        file: pkg.main,
+        format: 'cjs',
+      },
+      {
+        file: pkg.module,
+        format: 'esm',
+      },
+    ],
+    ...{
+      external: [
+        ...Object.keys(pkg.dependencies || {}),
+        ...Object.keys(pkg.devDependencies || {}),
+        ...Object.keys(pkg.peerDependencies || {}),
+      ],
+      preserveSymlinks: true,
+      plugins: [
+        resolve(),
+        typescript({ useTsconfigDeclarationDir: true, tsconfig: './tsconfig.json' }),
+      ],
+    },
+  },
+];

package/src/api-endpoints.ts

@@ -0,0 +1,95 @@
+export const user = () => {
+  return '/api/user';
+};
+
+export const userPlugins = () => {
+  return '/api/user/plugins';
+};
+
+export const messages = (id: string) => {
+  return `/api/messages/${id}`;
+};
+
+export const abortRequest = (endpoint: string) => {
+  return `/api/ask/${endpoint}/abort`;
+};
+
+export const conversations = (pageNumber: string) => {
+  return `/api/convos?pageNumber=${pageNumber}`;
+};
+
+export const conversationById = (id: string) => {
+  return `/api/convos/${id}`;
+};
+
+export const updateConversation = () => {
+  return '/api/convos/update';
+};
+
+export const deleteConversation = () => {
+  return '/api/convos/clear';
+};
+
+export const search = (q: string, pageNumber: string) => {
+  return `/api/search?q=${q}&pageNumber=${pageNumber}`;
+};
+
+export const searchEnabled = () => {
+  return '/api/search/enable';
+};
+
+export const presets = () => {
+  return '/api/presets';
+};
+
+export const deletePreset = () => {
+  return '/api/presets/delete';
+};
+
+export const aiEndpoints = () => {
+  return '/api/endpoints';
+};
+
+export const tokenizer = () => {
+  return '/api/tokenizer';
+};
+
+export const login = () => {
+  return '/api/auth/login';
+};
+
+export const logout = () => {
+  return '/api/auth/logout';
+};
+
+export const register = () => {
+  return '/api/auth/register';
+};
+
+export const loginFacebook = () => {
+  return '/api/auth/facebook';
+};
+
+export const loginGoogle = () => {
+  return '/api/auth/google';
+};
+
+export const refreshToken = () => {
+  return '/api/auth/refresh';
+};
+
+export const requestPasswordReset = () => {
+  return '/api/auth/requestPasswordReset';
+};
+
+export const resetPassword = () => {
+  return '/api/auth/resetPassword';
+};
+
+export const plugins = () => {
+  return '/api/plugins';
+};
+
+export const config = () => {
+  return '/api/config';
+};

package/src/createPayload.ts

@@ -0,0 +1,28 @@
+import type { TConversation, TSubmission, EModelEndpoint } from './types';
+
+export default function createPayload(submission: TSubmission) {
+  const { conversation, message, endpointOption } = submission;
+  const { conversationId } = conversation as TConversation;
+  const { endpoint } = endpointOption as { endpoint: EModelEndpoint };
+
+  const endpointUrlMap = {
+    azureOpenAI: '/api/ask/azureOpenAI',
+    openAI: '/api/ask/openAI',
+    google: '/api/ask/google',
+    bingAI: '/api/ask/bingAI',
+    chatGPT: '/api/ask/chatGPT',
+    chatGPTBrowser: '/api/ask/chatGPTBrowser',
+    gptPlugins: '/api/ask/gptPlugins',
+    anthropic: '/api/ask/anthropic',
+  };
+
+  const server = endpointUrlMap[endpoint];
+
+  const payload = {
+    ...message,
+    ...endpointOption,
+    conversationId,
+  };
+
+  return { server, payload };
+}

package/src/data-service.ts

@@ -0,0 +1,119 @@
+import * as t from './types';
+import request from './request';
+import * as endpoints from './api-endpoints';
+
+export function getConversations(pageNumber: string): Promise<t.TGetConversationsResponse> {
+  return request.get(endpoints.conversations(pageNumber));
+}
+
+export function abortRequestWithMessage(
+  endpoint: string,
+  abortKey: string,
+  message: string,
+): Promise<void> {
+  return request.post(endpoints.abortRequest(endpoint), { arg: { abortKey, message } });
+}
+
+export function deleteConversation(payload: t.TDeleteConversationRequest) {
+  //todo: this should be a DELETE request
+  return request.post(endpoints.deleteConversation(), { arg: payload });
+}
+
+export function clearAllConversations(): Promise<unknown> {
+  return request.post(endpoints.deleteConversation(), { arg: {} });
+}
+
+export function getMessagesByConvoId(id: string): Promise<t.TMessage[]> {
+  return request.get(endpoints.messages(id));
+}
+
+export function getConversationById(id: string): Promise<t.TConversation> {
+  return request.get(endpoints.conversationById(id));
+}
+
+export function updateConversation(
+  payload: t.TUpdateConversationRequest,
+): Promise<t.TUpdateConversationResponse> {
+  return request.post(endpoints.updateConversation(), { arg: payload });
+}
+
+export function getPresets(): Promise<t.TPreset[]> {
+  return request.get(endpoints.presets());
+}
+
+export function createPreset(payload: t.TPreset): Promise<t.TPreset[]> {
+  return request.post(endpoints.presets(), payload);
+}
+
+export function updatePreset(payload: t.TPreset): Promise<t.TPreset[]> {
+  return request.post(endpoints.presets(), payload);
+}
+
+export function deletePreset(arg: t.TPreset | object): Promise<t.TPreset[]> {
+  return request.post(endpoints.deletePreset(), arg);
+}
+
+export function getSearchEnabled(): Promise<boolean> {
+  return request.get(endpoints.searchEnabled());
+}
+
+export function getUser(): Promise<t.TUser> {
+  return request.get(endpoints.user());
+}
+
+export const searchConversations = async (
+  q: string,
+  pageNumber: string,
+): Promise<t.TSearchResults> => {
+  return request.get(endpoints.search(q, pageNumber));
+};
+
+export const getAIEndpoints = () => {
+  return request.get(endpoints.aiEndpoints());
+};
+
+export const updateTokenCount = (text: string) => {
+  return request.post(endpoints.tokenizer(), { arg: text });
+};
+
+export const login = (payload: t.TLoginUser) => {
+  return request.post(endpoints.login(), payload);
+};
+
+export const logout = () => {
+  return request.post(endpoints.logout());
+};
+
+export const register = (payload: t.TRegisterUser) => {
+  return request.post(endpoints.register(), payload);
+};
+
+export const refreshToken = () => {
+  return request.post(endpoints.refreshToken());
+};
+
+export const getLoginGoogle = () => {
+  return request.get(endpoints.loginGoogle());
+};
+
+export const requestPasswordReset = (
+  payload: t.TRequestPasswordReset,
+): Promise<t.TRequestPasswordResetResponse> => {
+  return request.post(endpoints.requestPasswordReset(), payload);
+};
+
+export const resetPassword = (payload: t.TResetPassword) => {
+  return request.post(endpoints.resetPassword(), payload);
+};
+
+export const getAvailablePlugins = (): Promise<t.TPlugin[]> => {
+  return request.get(endpoints.plugins());
+};
+
+export const updateUserPlugins = (payload: t.TUpdateUserPlugins) => {
+  return request.post(endpoints.userPlugins(), payload);
+};
+
+export const getStartupConfig = (): Promise<t.TStartupConfig> => {
+  return request.get(endpoints.config());
+};

package/src/headers-helpers.ts

@@ -0,0 +1,9 @@
+import axios from 'axios';
+
+export function setAcceptLanguageHeader(value: string): void {
+  axios.defaults.headers.common['Accept-Language'] = value;
+}
+
+export function setTokenHeader(token: string) {
+  axios.defaults.headers.common['Authorization'] = 'Bearer ' + token;
+}

package/src/index.ts

@@ -0,0 +1,7 @@
+export * from './data-service';
+export * from './request';
+export * from './types';
+export * from './react-query-service';
+export * from './headers-helpers';
+export * from './sse.mjs';
+export { default as createPayload } from './createPayload';