fix: dashboard url, memories crud

Author: itsskofficial

src/client/app/chat/page.js

@@ -103,7 +103,7 @@ const UpgradeToProModal = ({ isOpen, onClose }) => {
 	const handleUpgrade = () => {
 		const dashboardUrl = process.env.NEXT_PUBLIC_LANDING_PAGE_URL
 		if (dashboardUrl) {
-			window.open(`${dashboardUrl}/dashboard`, "_blank")
+			window.location.href = `${dashboardUrl}/dashboard`
 		}
 		onClose()
 	}

src/client/app/integrations/page.js

@@ -112,7 +112,7 @@ const UpgradeToProModal = ({ isOpen, onClose }) => {
 	const handleUpgrade = () => {
 		const dashboardUrl = process.env.NEXT_PUBLIC_LANDING_PAGE_URL
 		if (dashboardUrl) {
-			window.open(`${dashboardUrl}/dashboard`, "_blank")
+			window.location.href = `${dashboardUrl}/dashboard`
 		}
 		onClose()
 	}
@@ -714,7 +714,9 @@ const IntegrationsPage = () => {
 	const fetchIntegrations = useCallback(async () => {
 		setLoading(true)
 		try {
-			const response = await fetch("/api/settings/integrations")
+			const response = await fetch("/api/settings/integrations", {
+				cache: "no-store"
+			})
 			const data = await response.json()
 			if (!response.ok)
 				throw new Error(data.error || "Failed to fetch integrations")
@@ -756,7 +758,29 @@ const IntegrationsPage = () => {
 		setUpgradeModalOpen(true)
 	}
 
-	const handleConnect = (integration) => {
+	const handleConnect = async (integration) => {
+		const isProFeature = PRO_ONLY_INTEGRATIONS.includes(integration.name)
+		if (isProFeature && !isPro) {
+			handleUpgradeClick()
+			return
+		}
+
+		// If it's a pro feature, refresh the session cookie before redirecting
+		// to ensure the backend gets a fresh token with the correct roles.
+		if (isProFeature) {
+			const toastId = toast.loading("Preparing secure connection...")
+			try {
+				const res = await fetch("/api/auth/refresh-session")
+				if (!res.ok) throw new Error("Session refresh failed")
+				toast.dismiss(toastId)
+			} catch (e) {
+				toast.error("Could not prepare connection. Please try again.", {
+					id: toastId
+				})
+				return // Stop if refresh fails
+			}
+		}
+
 		if (integration.name === "whatsapp") {
 			setWhatsAppToConnect(integration)
 			return
@@ -1257,17 +1281,17 @@ const IntegrationsPage = () => {
 									</button>
 								) : (
 									<button
-										onClick={(e) => {
+										onClick={async (e) => {
 											e.stopPropagation()
 											if (
 												integration.auth_type ===
 												"composio"
 											) {
-												handleComposioConnect(
+												await handleComposioConnect(
 													integration
 												)
 											} else {
-												handleConnect(integration)
+												await handleConnect(integration)
 											}
 										}}
 										className="flex items-center justify-center gap-2 w-full py-2 px-3 rounded-md bg-brand-orange hover:bg-brand-orange/90 text-brand-black font-semibold text-sm transition-colors"
@@ -1452,4 +1476,4 @@ const IntegrationsPage = () => {
 	)
 }
 
-export default IntegrationsPage
+export default IntegrationsPage

src/client/app/memories/page.js

@@ -484,12 +484,16 @@ export default function MemoriesPage() {
 		setIsLoading(true)
 		try {
 			if (view === "list") {
-				const response = await fetch("/api/memories")
+				const response = await fetch("/api/memories", {
+					cache: "no-store"
+				})
 				if (!response.ok) throw new Error("Failed to fetch memories.")
 				const data = await response.json()
 				setMemories(data.memories || [])
 			} else {
-				const response = await fetch("/api/memories/graph")
+				const response = await fetch("/api/memories/graph", {
+					cache: "no-store"
+				})
 				if (!response.ok)
 					throw new Error("Failed to fetch memory graph data.")
 				const data = await response.json()

src/client/app/tasks/page.js

@@ -59,7 +59,7 @@ const UpgradeToProModal = ({ isOpen, onClose }) => {
 	const handleUpgrade = () => {
 		const dashboardUrl = process.env.NEXT_PUBLIC_LANDING_PAGE_URL
 		if (dashboardUrl) {
-			window.open(`${dashboardUrl}/dashboard`, "_blank")
+			window.location.href = `${dashboardUrl}/dashboard`
 		}
 		onClose()
 	}

src/server/main/memories/utils.py

@@ -135,11 +135,14 @@ async def update_memory(user_id: str, memory_id: int, new_content: str) -> str:
             raise ValueError("Failed to analyze updated memory content.")
 
         new_embedding = _get_normalized_embedding(new_content, task_type="RETRIEVAL_DOCUMENT")
+        # Also re-evaluate the expiration based on the new content analysis
+        expires_at = parse_duration(analysis.get("duration")) if analysis.get("memory_type") == "short-term" else None
 
         async with conn.transaction():
             await conn.execute(
-                "UPDATE facts SET content = $1, embedding = $2, updated_at = NOW() WHERE id = $3",
-                new_content, new_embedding, memory_id
+                # Update content, embedding, timestamp, and expiration
+                "UPDATE facts SET content = $1, embedding = $2, updated_at = NOW(), expires_at = $4 WHERE id = $3",
+                new_content, new_embedding, memory_id, expires_at
             )
             await conn.execute("DELETE FROM fact_topics WHERE fact_id = $1", memory_id)
             topic_names = analysis.get("topics", ["Miscellaneous"])