From b2f7f5c904f890f01a655f75d6ced955568072f9 Mon Sep 17 00:00:00 2001
From: Danny Avila <danny@librechat.ai>
Date: Fri, 30 May 2025 14:42:30 -0400
Subject: [PATCH] fix(samlStrategy): update user creation to include balance
 configuration

- Modified the user creation process to incorporate balance configuration retrieved from the new getBalanceConfig function.
- Adjusted imports for user methods to streamline the code structure.
---
 api/strategies/samlStrategy.js | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/api/strategies/samlStrategy.js b/api/strategies/samlStrategy.js
index ccbfc650c3..376434f733 100644
--- a/api/strategies/samlStrategy.js
+++ b/api/strategies/samlStrategy.js
@@ -2,10 +2,11 @@ const fs = require('fs');
 const path = require('path');
 const fetch = require('node-fetch');
 const passport = require('passport');
-const { Strategy: SamlStrategy } = require('@node-saml/passport-saml');
-const { findUser, createUser, updateUser } = require('~/models/userMethods');
-const { getStrategyFunctions } = require('~/server/services/Files/strategies');
 const { hashToken, logger } = require('@librechat/data-schemas');
+const { Strategy: SamlStrategy } = require('@node-saml/passport-saml');
+const { getStrategyFunctions } = require('~/server/services/Files/strategies');
+const { findUser, createUser, updateUser } = require('~/models');
+const { getBalanceConfig } = require('~/server/services/Config');
 const paths = require('~/config/paths');
 
 let crypto;
@@ -217,7 +218,8 @@ async function setupSaml() {
               emailVerified: true,
               name: fullName,
             };
-            user = await createUser(user, true, true);
+            const balanceConfig = await getBalanceConfig();
+            user = await createUser(user, balanceConfig, true, true);
           } else {
             user.provider = 'saml';
             user.samlId = profile.nameID;