From 91c3adb208d8135cf0842ef1ce0af1433cd43bb3 Mon Sep 17 00:00:00 2001
From: Vignesh Aigal <vignesh.aigal@gmail.com>
Date: Wed, 28 Feb 2024 20:35:19 -0800
Subject: [PATCH] If redirectURL in session use that as redirect_url

---
 llmstack/server/adapters.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/llmstack/server/adapters.py b/llmstack/server/adapters.py
index 6f3312a117e..b1750fcd8d1 100644
--- a/llmstack/server/adapters.py
+++ b/llmstack/server/adapters.py
@@ -7,6 +7,21 @@
 
 
 class CustomAccountAdapter(DefaultAccountAdapter):
+    def post_login(self, request, user, *, email_verification, signal_kwargs, email, signup, redirect_url):
+        if "redirectUrl" in request.session:
+            redirect_url = request.session["redirectUrl"]
+            del request.session["redirectUrl"]
+
+        return super().post_login(
+            request,
+            user,
+            email_verification=email_verification,
+            signal_kwargs=signal_kwargs,
+            email=email,
+            signup=signup,
+            redirect_url=redirect_url,
+        )
+
     def save_user(self, request, user, form, commit=True):
         user = super().save_user(request, user, form, False)
         if not user.username: