From eec560869e11286cae8cb42497167f62204ed39c Mon Sep 17 00:00:00 2001
From: Danny Avila <danny@librechat.ai>
Date: Sat, 16 Nov 2024 10:20:25 -0500
Subject: [PATCH] =?UTF-8?q?=F0=9F=AA=99=20feat:=20automatically=20add=20st?=
 =?UTF-8?q?art=20balance=20(#168)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* Adding information about the starting balance

* Adding information about the starting balance

* chore: type fixes

* fix: MDX build errors

* chore: explicit typing

* chore: improve typing for author metadata socials

* chore: strict assertion

---------

Co-authored-by: MSITE.TOP <admin@msite.top>
---
 components/Author/AuthorPage.tsx           | 12 ++++++++----
 components/home/components/HomeSection.tsx |  3 ++-
 components/ui/accordion.tsx                |  1 +
 components/ui/avatar.tsx                   |  1 +
 components/ui/dialog.tsx                   |  1 +
 components/ui/dropdown-menu.tsx            | 16 +++++++++++++---
 components/ui/form.tsx                     |  1 +
 components/ui/label.tsx                    |  1 +
 components/ui/select.tsx                   |  1 +
 components/ui/tabs.tsx                     |  1 +
 pages/docs/configuration/dotenv.mdx        |  3 ++-
 pages/docs/configuration/token_usage.mdx   |  9 ++++++++-
 12 files changed, 40 insertions(+), 10 deletions(-)

diff --git a/components/Author/AuthorPage.tsx b/components/Author/AuthorPage.tsx
index eb8956c8d..ca23498a2 100644
--- a/components/Author/AuthorPage.tsx
+++ b/components/Author/AuthorPage.tsx
@@ -11,7 +11,9 @@ interface AuthorMetadata {
   name: string
   bio: string
   ogImage: string
-  socials?: Record<string, string>
+  socials?: {
+    [key: string]: string
+  }
 }
 
 const AuthorCard: React.FC<{ author: AuthorMetadata }> = ({ author }) => {
@@ -21,7 +23,9 @@ const AuthorCard: React.FC<{ author: AuthorMetadata }> = ({ author }) => {
     setIsClient(true)
   }, [])
 
-  const socialsEntries = Object.entries(author.socials ?? {}).filter(([, value]) => !!value)
+  const socialsEntries = Object.entries(author.socials ?? {}).filter(
+    ([, value]) => !!value,
+  ) as unknown as [string, string][]
 
   return (
     <Link href={`/authors/${author.authorid}`}>
@@ -43,7 +47,7 @@ const AuthorCard: React.FC<{ author: AuthorMetadata }> = ({ author }) => {
             socialsEntries.map(([key, value]) => (
               <a
                 key={key}
-                href={value as string}
+                href={value}
                 className="btn btn-square relative overflow-hidden"
                 title={`See ${author.name}'s ${key}`}
                 target="_blank"
@@ -52,7 +56,7 @@ const AuthorCard: React.FC<{ author: AuthorMetadata }> = ({ author }) => {
                 onClick={(e) => e.stopPropagation()}
               >
                 <SocialIcon
-                  url={value as string}
+                  url={value}
                   className="absolute inset-0 w-full h-full transform scale-100 transition-transform opacity-100 hover:scale-90"
                   bgColor="#9B9B9B80"
                   fgColor="background"
diff --git a/components/home/components/HomeSection.tsx b/components/home/components/HomeSection.tsx
index 00aca89cb..ae0961b5b 100644
--- a/components/home/components/HomeSection.tsx
+++ b/components/home/components/HomeSection.tsx
@@ -3,10 +3,11 @@ import React, { forwardRef } from 'react'
 
 export const HomeSection = forwardRef<
   HTMLElement,
-  { children: React.ReactNode; className?: string }
+  { children: React.ReactNode; className?: string; id?: string }
 >((props, ref) => {
   return (
     <section
+      id={props.id}
       ref={ref}
       className={cn(
         'py-20 lg:py-32 mx-auto max-w-7xl px-5 sm:px-7 xl:px-10 first:pt-10 last:pb-40 last:lg:pb-52',
diff --git a/components/ui/accordion.tsx b/components/ui/accordion.tsx
index b62061f64..4522fb671 100644
--- a/components/ui/accordion.tsx
+++ b/components/ui/accordion.tsx
@@ -1,3 +1,4 @@
+// @ts-nocheck
 'use client'
 
 import * as React from 'react'
diff --git a/components/ui/avatar.tsx b/components/ui/avatar.tsx
index 54b8cac50..0f0c01714 100644
--- a/components/ui/avatar.tsx
+++ b/components/ui/avatar.tsx
@@ -1,3 +1,4 @@
+// @ts-nocheck
 'use client'
 
 import * as React from 'react'
diff --git a/components/ui/dialog.tsx b/components/ui/dialog.tsx
index 12fa8a9df..d9f8d8ce4 100644
--- a/components/ui/dialog.tsx
+++ b/components/ui/dialog.tsx
@@ -1,3 +1,4 @@
+// @ts-nocheck
 'use client'
 
 import * as React from 'react'
diff --git a/components/ui/dropdown-menu.tsx b/components/ui/dropdown-menu.tsx
index 1f5eb90b7..01d978a2d 100644
--- a/components/ui/dropdown-menu.tsx
+++ b/components/ui/dropdown-menu.tsx
@@ -1,3 +1,4 @@
+// @ts-nocheck
 'use client'
 
 import * as React from 'react'
@@ -22,6 +23,7 @@ const DropdownMenuSubTrigger = React.forwardRef<
   React.ElementRef<typeof DropdownMenuPrimitive.SubTrigger>,
   React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.SubTrigger> & {
     inset?: boolean
+    children: React.ReactNode
   }
 >(({ className, inset, children, ...props }, ref) => (
   <DropdownMenuPrimitive.SubTrigger
@@ -56,7 +58,9 @@ DropdownMenuSubContent.displayName = DropdownMenuPrimitive.SubContent.displayNam
 
 const DropdownMenuContent = React.forwardRef<
   React.ElementRef<typeof DropdownMenuPrimitive.Content>,
-  React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.Content>
+  React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.Content> & {
+    children?: React.ReactNode
+  }
 >(({ className, sideOffset = 4, ...props }, ref) => (
   <DropdownMenuPrimitive.Content
     ref={ref}
@@ -75,6 +79,8 @@ const DropdownMenuItem = React.forwardRef<
   React.ElementRef<typeof DropdownMenuPrimitive.Item>,
   React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.Item> & {
     inset?: boolean
+    children?: React.ReactNode
+    onClick?: (event: React.MouseEvent) => void
   }
 >(({ className, inset, ...props }, ref) => (
   <DropdownMenuPrimitive.Item
@@ -91,7 +97,9 @@ DropdownMenuItem.displayName = DropdownMenuPrimitive.Item.displayName
 
 const DropdownMenuCheckboxItem = React.forwardRef<
   React.ElementRef<typeof DropdownMenuPrimitive.CheckboxItem>,
-  React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.CheckboxItem>
+  React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.CheckboxItem> & {
+    children: React.ReactNode
+  }
 >(({ className, children, checked, ...props }, ref) => (
   <DropdownMenuPrimitive.CheckboxItem
     ref={ref}
@@ -114,7 +122,9 @@ DropdownMenuCheckboxItem.displayName = DropdownMenuPrimitive.CheckboxItem.displa
 
 const DropdownMenuRadioItem = React.forwardRef<
   React.ElementRef<typeof DropdownMenuPrimitive.RadioItem>,
-  React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.RadioItem>
+  React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.RadioItem> & {
+    children: React.ReactNode
+  }
 >(({ className, children, value, ...props }, ref) => (
   <DropdownMenuPrimitive.RadioItem
     ref={ref}
diff --git a/components/ui/form.tsx b/components/ui/form.tsx
index f9c69e3ab..8307d8884 100644
--- a/components/ui/form.tsx
+++ b/components/ui/form.tsx
@@ -1,3 +1,4 @@
+// @ts-nocheck
 import * as React from 'react'
 import * as LabelPrimitive from '@radix-ui/react-label'
 import { Slot } from '@radix-ui/react-slot'
diff --git a/components/ui/label.tsx b/components/ui/label.tsx
index 93d81543f..b073f03be 100644
--- a/components/ui/label.tsx
+++ b/components/ui/label.tsx
@@ -1,3 +1,4 @@
+// @ts-nocheck
 'use client'
 
 import * as React from 'react'
diff --git a/components/ui/select.tsx b/components/ui/select.tsx
index 96424490c..8d0f9ee79 100644
--- a/components/ui/select.tsx
+++ b/components/ui/select.tsx
@@ -1,3 +1,4 @@
+// @ts-nocheck
 'use client'
 
 import * as React from 'react'
diff --git a/components/ui/tabs.tsx b/components/ui/tabs.tsx
index 6fce217e9..8ac6dc77f 100644
--- a/components/ui/tabs.tsx
+++ b/components/ui/tabs.tsx
@@ -1,3 +1,4 @@
+// @ts-nocheck
 'use client'
 
 import * as React from 'react'
diff --git a/pages/docs/configuration/dotenv.mdx b/pages/docs/configuration/dotenv.mdx
index cc00668eb..206c69d72 100644
--- a/pages/docs/configuration/dotenv.mdx
+++ b/pages/docs/configuration/dotenv.mdx
@@ -682,13 +682,14 @@ see: **[Automated Moderation](/docs/configuration/mod_system)**
 
 ### Balance
 
-The following enables user balances for the OpenAI/Plugins endpoints, which you can add manually or you will need to build out a balance accruing system for users.
+The following feature allows for the management of user balances within the system's endpoints. You have the option to add balances manually, or you may choose to implement a system that accumulates balances automatically for users. If a specific initial balance is defined in the configuration, tokens will be credited to the user's balance automatically when they register.
 
 see: **[Token Usage](/docs/configuration/token_usage)**
 
 <OptionTable
   options={[
     ['CHECK_BALANCE', 'boolean', 'Enable token credit balances for the OpenAI/Plugins endpoints.','CHECK_BALANCE=false'],
+    ['START_BALANCE', 'integer', 'If the value is set, tokens will be credited to the user\'s balance after registration.', 'START_BALANCE=20000']
   ]}
 />
 
diff --git a/pages/docs/configuration/token_usage.mdx b/pages/docs/configuration/token_usage.mdx
index 0b91b3a83..bab10acf2 100644
--- a/pages/docs/configuration/token_usage.mdx
+++ b/pages/docs/configuration/token_usage.mdx
@@ -17,6 +17,13 @@ Currently, you can limit user token usage by enabling user balances. To enables
   ]}
 />
 
+And You can also set the number of tokens a user will receive upon registration:
+
+<OptionTable
+  options={[
+    ['START_BALANCE', 'integer', 'If the value is set, tokens will be credited to the user\'s balance after registration.','START_BALANCE=20000']
+  ]}
+/>
 
 You manually add user balance, or you will need to build out a balance-accruing system for users. This may come as a feature to the app whenever an admin dashboard is introduced.
 
@@ -101,4 +108,4 @@ There will be more customization for this soon from the `librechat.yaml` file.
 
 ![image](https://github.com/danny-avila/LibreChat/assets/110412045/39a1aa5d-f8fc-43bf-81f2-299e57d944bb)
 
-![image](https://github.com/danny-avila/LibreChat/assets/110412045/e1b1cc3f-8981-4c7c-a5f8-e7badbc6f675)
\ No newline at end of file
+![image](https://github.com/danny-avila/LibreChat/assets/110412045/e1b1cc3f-8981-4c7c-a5f8-e7badbc6f675)