Merge pull request #362 from MODSetter/dev

chore: updated litellm and langchain-litellm

surfsense_backend/app/services/llm_service.py

@@ -1,6 +1,6 @@
 import logging
 
-from langchain_community.chat_models import ChatLiteLLM
+from langchain_litellm import ChatLiteLLM
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.future import select
 
@@ -81,7 +81,7 @@ async def get_user_llm_instance(
                 "MISTRAL": "mistral",
                 "AZURE_OPENAI": "azure",
                 "OPENROUTER": "openrouter",
-                "COMETAPI": "openai",
+                "COMETAPI": "cometapi",
                 # Add more mappings as needed
             }
             provider_prefix = provider_map.get(

surfsense_backend/pyproject.toml

@@ -21,7 +21,6 @@ dependencies = [
     "langchain-unstructured>=0.1.6",
     "langgraph>=0.3.29",
     "linkup-sdk>=0.2.4",
-    "litellm>=1.61.4,<1.70.0",
     "llama-cloud-services>=0.6.25",
     "markdownify>=0.14.1",
     "notion-client>=2.3.0",
@@ -42,6 +41,8 @@ dependencies = [
     "uvicorn[standard]>=0.34.0",
     "validators>=0.34.0",
     "youtube-transcript-api>=1.0.3",
+    "litellm>=1.77.5",
+    "langchain-litellm>=0.2.3",
 ]
 
 [dependency-groups]

surfsense_backend/uv.lock

@@ -1232,6 +1232,36 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/fe/84/9c2917a70ed570ddbfd1d32ac23200c1d011e36c332e59950d2f6d204941/fastavro-1.11.1-cp313-cp313t-musllinux_1_2_x86_64.whl", hash = "sha256:1bc2824e9969c04ab6263d269a1e0e5d40b9bd16ade6b70c29d6ffbc4f3cc102", size = 3387171 },
 ]
 
+[[package]]
+name = "fastuuid"
+version = "0.13.5"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/15/80/3c16a1edad2e6cd82fbd15ac998cc1b881f478bf1f80ca717d941c441874/fastuuid-0.13.5.tar.gz", hash = "sha256:d4976821ab424d41542e1ea39bc828a9d454c3f8a04067c06fca123c5b95a1a1", size = 18255 }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/21/36/434f137c5970cac19e57834e1f7680e85301619d49891618c00666700c61/fastuuid-0.13.5-cp312-cp312-macosx_10_12_x86_64.macosx_11_0_arm64.macosx_10_12_universal2.whl", hash = "sha256:35fe8045e866bc6846f8de6fa05acb1de0c32478048484a995e96d31e21dff2a", size = 494638 },
+    { url = "https://files.pythonhosted.org/packages/ca/3c/083de2ac007b2b305523b9c006dba5051e5afd87a626ef1a39f76e2c6b82/fastuuid-0.13.5-cp312-cp312-macosx_10_12_x86_64.whl", hash = "sha256:02a460333f52d731a006d18a52ef6fcb2d295a1f5b1a5938d30744191b2f77b7", size = 253138 },
+    { url = "https://files.pythonhosted.org/packages/73/5e/630cffa1c8775db526e39e9e4c5c7db0c27be0786bb21ba82c912ae19f63/fastuuid-0.13.5-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:74b0e4f8c307b9f477a5d7284db4431ce53a3c1e3f4173db7a97db18564a6202", size = 244521 },
+    { url = "https://files.pythonhosted.org/packages/4d/51/55d78705f4fbdadf88fb40f382f508d6c7a4941ceddd7825fafebb4cc778/fastuuid-0.13.5-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:6955a99ef455c2986f3851f4e0ccc35dec56ac1a7720f2b92e88a75d6684512e", size = 271557 },
+    { url = "https://files.pythonhosted.org/packages/6a/2b/1b89e90a8635e5587ccdbbeb169c590672ce7637880f2c047482a0359950/fastuuid-0.13.5-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:f10c77b826738c1a27dcdaa92ea4dc1ec9d869748a99e1fde54f1379553d4854", size = 272334 },
+    { url = "https://files.pythonhosted.org/packages/0c/06/4c8207894eeb30414999e5c3f66ac039bc4003437eb4060d8a1bceb4cc6f/fastuuid-0.13.5-cp312-cp312-manylinux_2_5_i686.manylinux1_i686.whl", hash = "sha256:bb25dccbeb249d16d5e664f65f17ebec05136821d5ef462c4110e3f76b86fb86", size = 290594 },
+    { url = "https://files.pythonhosted.org/packages/50/69/96d221931a31d77a47cc2487bdfacfb3091edfc2e7a04b1795df1aec05df/fastuuid-0.13.5-cp312-cp312-musllinux_1_1_aarch64.whl", hash = "sha256:a5becc646a3eeafb76ce0a6783ba190cd182e3790a8b2c78ca9db2b5e87af952", size = 452835 },
+    { url = "https://files.pythonhosted.org/packages/25/ef/bf045f0a47dcec96247497ef3f7a31d86ebc074330e2dccc34b8dbc0468a/fastuuid-0.13.5-cp312-cp312-musllinux_1_1_i686.whl", hash = "sha256:69b34363752d06e9bb0dbdf02ae391ec56ac948c6f2eb00be90dad68e80774b9", size = 468225 },
+    { url = "https://files.pythonhosted.org/packages/30/46/4817ab5a3778927155a4bde92540d4c4fa996161ec8b8e080c8928b0984e/fastuuid-0.13.5-cp312-cp312-musllinux_1_1_x86_64.whl", hash = "sha256:57d0768afcad0eab8770c9b8cf904716bd3c547e8b9a4e755ee8a673b060a3a3", size = 444907 },
+    { url = "https://files.pythonhosted.org/packages/80/27/ab284117ce4dc9b356a7196bdbf220510285f201d27f1f078592cdc8187b/fastuuid-0.13.5-cp312-cp312-win32.whl", hash = "sha256:8ac6c6f5129d52eaa6ef9ea4b6e2f7c69468a053f3ab8e439661186b9c06bb85", size = 145415 },
+    { url = "https://files.pythonhosted.org/packages/f4/0c/f970a4222773b248931819f8940800b760283216ca3dda173ed027e94bdd/fastuuid-0.13.5-cp312-cp312-win_amd64.whl", hash = "sha256:ad630e97715beefef07ec37c9c162336e500400774e2c1cbe1a0df6f80d15b9a", size = 150840 },
+    { url = "https://files.pythonhosted.org/packages/4f/62/74fc53f6e04a4dc5b36c34e4e679f85a4c14eec800dcdb0f2c14b5442217/fastuuid-0.13.5-cp313-cp313-macosx_10_12_x86_64.macosx_11_0_arm64.macosx_10_12_universal2.whl", hash = "sha256:ea17dfd35e0e91920a35d91e65e5f9c9d1985db55ac4ff2f1667a0f61189cefa", size = 494678 },
+    { url = "https://files.pythonhosted.org/packages/09/ba/f28b9b7045738a8bfccfb9cd6aff4b91fce2669e6b383a48b0694ee9b3ff/fastuuid-0.13.5-cp313-cp313-macosx_10_12_x86_64.whl", hash = "sha256:be6ad91e5fefbcc2a4b478858a2715e386d405834ea3ae337c3b6b95cc0e47d6", size = 253162 },
+    { url = "https://files.pythonhosted.org/packages/b1/18/13fac89cb4c9f0cd7e81a9154a77ecebcc95d2b03477aa91d4d50f7227ee/fastuuid-0.13.5-cp313-cp313-macosx_11_0_arm64.whl", hash = "sha256:ea6df13a306aab3e0439d58c312ff1e6f4f07f09f667579679239b4a6121f64a", size = 244546 },
+    { url = "https://files.pythonhosted.org/packages/04/bf/9691167804d59411cc4269841df949f6dd5e76452ab10dcfcd1dbe04c5bc/fastuuid-0.13.5-cp313-cp313-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:2354c1996d3cf12dc2ba3752e2c4d6edc46e1a38c63893146777b1939f3062d4", size = 271528 },
+    { url = "https://files.pythonhosted.org/packages/a9/b5/7a75a03d1c7aa0b6d573032fcca39391f0aef7f2caabeeb45a672bc0bd3c/fastuuid-0.13.5-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:a6cf9b7469fc26d1f9b1c43ac4b192e219e85b88fdf81d71aa755a6c08c8a817", size = 272292 },
+    { url = "https://files.pythonhosted.org/packages/c0/db/fa0f16cbf76e6880599533af4ef01bb586949c5320612e9d884eff13e603/fastuuid-0.13.5-cp313-cp313-manylinux_2_5_i686.manylinux1_i686.whl", hash = "sha256:92ba539170097b9047551375f1ca09d8d2b4aefcc79eeae3e1c43fe49b42072e", size = 290466 },
+    { url = "https://files.pythonhosted.org/packages/1e/02/6b8c45bfbc8500994dd94edba7f59555f9683c4d8c9a164ae1d25d03c7c7/fastuuid-0.13.5-cp313-cp313-musllinux_1_1_aarch64.whl", hash = "sha256:dbb81d05617bc2970765c1ad82db7e8716f6a2b7a361a14b83de5b9240ade448", size = 452838 },
+    { url = "https://files.pythonhosted.org/packages/27/12/85d95a84f265b888e8eb9f9e2b5aaf331e8be60c0a7060146364b3544b6a/fastuuid-0.13.5-cp313-cp313-musllinux_1_1_i686.whl", hash = "sha256:d973bd6bf9d754d3cca874714ac0a6b22a47f239fb3d3c8687569db05aac3471", size = 468149 },
+    { url = "https://files.pythonhosted.org/packages/ad/da/dd9a137e9ea707e883c92470113a432233482ec9ad3e9b99c4defc4904e6/fastuuid-0.13.5-cp313-cp313-musllinux_1_1_x86_64.whl", hash = "sha256:e725ceef79486423f05ee657634d4b4c1ca5fb2c8a94e0708f5d6356a83f2a83", size = 444933 },
+    { url = "https://files.pythonhosted.org/packages/12/f4/ab363d7f4ac3989691e2dc5ae2d8391cfb0b4169e52ef7fa0ac363e936f0/fastuuid-0.13.5-cp313-cp313-win32.whl", hash = "sha256:a1c430a332ead0b2674f1ef71b17f43b8139ec5a4201182766a21f131a31e021", size = 145462 },
+    { url = "https://files.pythonhosted.org/packages/aa/8a/52eb77d9c294a54caa0d2d8cc9f906207aa6d916a22de963687ab6db8b86/fastuuid-0.13.5-cp313-cp313-win_amd64.whl", hash = "sha256:241fdd362fd96e6b337db62a65dd7cb3dfac20adf854573247a47510e192db6f", size = 150923 },
+]
+
 [[package]]
 name = "filelock"
 version = "3.18.0"
@@ -2208,6 +2238,19 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/54/f0/31db18b7b8213266aed926ce89b5bdd84ccde7ee2edf4cab14e3dd2bfcf1/langchain_core-0.3.65-py3-none-any.whl", hash = "sha256:80e8faf6e9f331f8ef728f3fe793549f1d3fb244fcf9e1bdcecab6a6f4669394", size = 438052 },
 ]
 
+[[package]]
+name = "langchain-litellm"
+version = "0.2.3"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "langchain-core" },
+    { name = "litellm" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/40/8f/08032033cd4bdff1d177d6a9e3a1021e47c4c63fd1d8c564af6f3c7e9f8d/langchain_litellm-0.2.3.tar.gz", hash = "sha256:0e11687373ae6a99efee5a04d3a76de4fab0e1459edc0e84adb6f60ca76ebf79", size = 10829 }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/76/26/271b1dad80b39a0e9df7ab13f63fa3fad52ce8288ddf73dec32a2212219f/langchain_litellm-0.2.3-py3-none-any.whl", hash = "sha256:422254b8742893aed6380f5ee73e6ae77869b218758edd0888d14ebd2c439352", size = 11571 },
+]
+
 [[package]]
 name = "langchain-text-splitters"
 version = "0.3.8"
@@ -2377,11 +2420,12 @@ wheels = [
 
 [[package]]
 name = "litellm"
-version = "1.69.3"
+version = "1.77.5"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "aiohttp" },
     { name = "click" },
+    { name = "fastuuid" },
     { name = "httpx" },
     { name = "importlib-metadata" },
     { name = "jinja2" },
@@ -2392,9 +2436,9 @@ dependencies = [
     { name = "tiktoken" },
     { name = "tokenizers" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/7d/25/2ce407b0ef7e3a2b05300336b7264ec781872c5e60e6ead7a10c9410adcd/litellm-1.69.3.tar.gz", hash = "sha256:748fe9dffea743bf683c9e28e4632c14894863e62c3fbf057560ea7324d89390", size = 7493451 }
+sdist = { url = "https://files.pythonhosted.org/packages/e1/a3/85fc92d998ec9645c9fac108618681ef411ca4b338cc7544d6b3aad57699/litellm-1.77.5.tar.gz", hash = "sha256:8e8a83b49c4a6ae044b1a1c01adfbdef72b0031b86f1463dd743e267fa1d7b99", size = 10351819 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/aa/c0/d1538ca18a32adb2669312342e11066e3ddb7dffdf328c1fad3027f78b3e/litellm-1.69.3-py3-none-any.whl", hash = "sha256:31f17024d06824aa8c1798e2c6ac44b69d4f721cefbc3bcd0d53a4568831075e", size = 7761394 },
+    { url = "https://files.pythonhosted.org/packages/94/4c/89553f7e375ef39497d86f2266a0cdb37371a07e9e0aa8949f33c15a4198/litellm-1.77.5-py3-none-any.whl", hash = "sha256:07f53964c08d555621d4376cc42330458301ae889bfb6303155dcabc51095fbf", size = 9165458 },
 ]
 
 [[package]]
@@ -3324,7 +3368,7 @@ wheels = [
 
 [[package]]
 name = "openai"
-version = "1.75.0"
+version = "2.1.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "anyio" },
@@ -3336,9 +3380,9 @@ dependencies = [
     { name = "tqdm" },
     { name = "typing-extensions" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/99/b1/318f5d4c482f19c5fcbcde190801bfaaaec23413cda0b88a29f6897448ff/openai-1.75.0.tar.gz", hash = "sha256:fb3ea907efbdb1bcfd0c44507ad9c961afd7dce3147292b54505ecfd17be8fd1", size = 429492 }
+sdist = { url = "https://files.pythonhosted.org/packages/1a/dd/4d4d46a06943e37c95b6e388237e1e38d1e9aab264ff070f86345d60b7a4/openai-2.1.0.tar.gz", hash = "sha256:47f3463a5047340a989b4c0cd5378054acfca966ff61a96553b22f098e3270a2", size = 572998 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/80/9a/f34f163294345f123673ed03e77c33dee2534f3ac1f9d18120384457304d/openai-1.75.0-py3-none-any.whl", hash = "sha256:fe6f932d2ded3b429ff67cc9ad118c71327db32eb9d32dd723de3acfca337125", size = 646972 },
+    { url = "https://files.pythonhosted.org/packages/68/83/88f64fc8f037885efa8a629d1215f5bc1f037453bab4d4f823b5533319eb/openai-2.1.0-py3-none-any.whl", hash = "sha256:33172e8c06a4576144ba4137a493807a9ca427421dcabc54ad3aa656daf757d3", size = 964939 },
 ]
 
 [[package]]
@@ -5292,6 +5336,7 @@ dependencies = [
     { name = "google-auth-oauthlib" },
     { name = "kokoro" },
     { name = "langchain-community" },
+    { name = "langchain-litellm" },
     { name = "langchain-unstructured" },
     { name = "langgraph" },
     { name = "linkup-sdk" },
@@ -5338,10 +5383,11 @@ requires-dist = [
     { name = "google-auth-oauthlib", specifier = ">=1.2.1" },
     { name = "kokoro", specifier = ">=0.9.4" },
     { name = "langchain-community", specifier = ">=0.3.17" },
+    { name = "langchain-litellm", specifier = ">=0.2.3" },
     { name = "langchain-unstructured", specifier = ">=0.1.6" },
     { name = "langgraph", specifier = ">=0.3.29" },
     { name = "linkup-sdk", specifier = ">=0.2.4" },
-    { name = "litellm", specifier = ">=1.61.4,<1.70.0" },
+    { name = "litellm", specifier = ">=1.77.5" },
     { name = "llama-cloud-services", specifier = ">=0.6.25" },
     { name = "markdownify", specifier = ">=0.14.1" },
     { name = "notion-client", specifier = ">=2.3.0" },

surfsense_web/app/dashboard/[search_space_id]/chats/chats-client.tsx

@@ -1,7 +1,6 @@
 "use client";
 
 import { format } from "date-fns";
-import { AnimatePresence, motion, type Variants } from "motion/react";
 import {
 	Calendar,
 	CheckCircle,
@@ -14,6 +13,7 @@ import {
 	Tag,
 	Trash2,
 } from "lucide-react";
+import { AnimatePresence, motion, type Variants } from "motion/react";
 import { useRouter, useSearchParams } from "next/navigation";
 import { useEffect, useState } from "react";
 import { toast } from "sonner";

surfsense_web/app/dashboard/[search_space_id]/connectors/(manage)/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { format } from "date-fns";
-import { motion } from "motion/react";
 import { Calendar as CalendarIcon, Edit, Plus, RefreshCw, Trash2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useEffect, useState } from "react";
 import { toast } from "sonner";

surfsense_web/app/dashboard/[search_space_id]/connectors/[connector_id]/edit/page.tsx

@@ -1,7 +1,7 @@
 "use client";
 
-import { motion } from "motion/react";
 import { ArrowLeft, Check, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useEffect } from "react";
 import { toast } from "sonner";

surfsense_web/app/dashboard/[search_space_id]/connectors/[connector_id]/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, Info, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useEffect, useState } from "react";
 import { useForm } from "react-hook-form";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/airtable-connector/page.tsx

@@ -1,7 +1,7 @@
 "use client";
 
-import { motion } from "motion/react";
 import { ArrowLeft, Check, ExternalLink, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import Link from "next/link";
 import { useParams, useRouter } from "next/navigation";
 import { useEffect, useState } from "react";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/confluence-connector/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, Info, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useState } from "react";
 import { useForm } from "react-hook-form";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/discord-connector/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, Info, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useState } from "react";
 import { useForm } from "react-hook-form";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/github-connector/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, CircleAlert, Github, Info, ListChecks, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useState } from "react";
 import { useForm } from "react-hook-form";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/google-calendar-connector/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, ExternalLink, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import Link from "next/link";
 import { useParams, useRouter, useSearchParams } from "next/navigation";
 import { useEffect, useState } from "react";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/google-gmail-connector/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, ExternalLink, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import Link from "next/link";
 import { useParams, useRouter, useSearchParams } from "next/navigation";
 import { useEffect, useState } from "react";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/jira-connector/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, Info, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useState } from "react";
 import { useForm } from "react-hook-form";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/linear-connector/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, Info, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useState } from "react";
 import { useForm } from "react-hook-form";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/linkup-api/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, Info, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useState } from "react";
 import { useForm } from "react-hook-form";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/luma-connector/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, Key, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import Link from "next/link";
 import { useParams, useRouter } from "next/navigation";
 import { useEffect, useState } from "react";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/notion-connector/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, Info, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useState } from "react";
 import { useForm } from "react-hook-form";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/serper-api/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, Info, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useState } from "react";
 import { useForm } from "react-hook-form";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/slack-connector/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, Info, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useState } from "react";
 import { useForm } from "react-hook-form";

surfsense_web/app/dashboard/[search_space_id]/connectors/add/tavily-api/page.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { zodResolver } from "@hookform/resolvers/zod";
-import { motion } from "motion/react";
 import { ArrowLeft, Check, Info, Loader2 } from "lucide-react";
+import { motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useState } from "react";
 import { useForm } from "react-hook-form";

surfsense_web/app/dashboard/[search_space_id]/documents/(manage)/components/DocumentsFilters.tsx

@@ -1,7 +1,7 @@
 "use client";
 
-import { AnimatePresence, motion, type Variants } from "motion/react";
 import { CircleAlert, CircleX, Columns3, Filter, ListFilter, Trash } from "lucide-react";
+import { AnimatePresence, motion, type Variants } from "motion/react";
 import React, { useMemo, useRef } from "react";
 import {
 	AlertDialog,

surfsense_web/app/dashboard/[search_space_id]/documents/(manage)/components/DocumentsTableShell.tsx

@@ -1,7 +1,7 @@
 "use client";
 
-import { motion } from "motion/react";
 import { ChevronDown, ChevronUp, FileX } from "lucide-react";
+import { motion } from "motion/react";
 import React from "react";
 import { DocumentViewer } from "@/components/document-viewer";
 import { Button } from "@/components/ui/button";

surfsense_web/app/dashboard/[search_space_id]/documents/(manage)/components/PaginationControls.tsx

@@ -1,7 +1,7 @@
 "use client";
 
-import { motion } from "motion/react";
 import { ChevronFirst, ChevronLast, ChevronLeft, ChevronRight } from "lucide-react";
+import { motion } from "motion/react";
 import { Button } from "@/components/ui/button";
 import { Label } from "@/components/ui/label";
 import { Pagination, PaginationContent, PaginationItem } from "@/components/ui/pagination";

surfsense_web/app/dashboard/[search_space_id]/documents/(manage)/page.tsx

@@ -42,18 +42,11 @@ export default function DocumentsTable() {
 	const [selectedIds, setSelectedIds] = useState<Set<number>>(new Set());
 
 	// Use server-side pagination and search
-	const {
+	const { documents, total, loading, error, fetchDocuments, searchDocuments, deleteDocument } =
-		documents,
+		useDocuments(searchSpaceId, {
-		total,
+			page: pageIndex,
-		loading,
+			pageSize: pageSize,
-		error,
+		});
-		fetchDocuments,
-		searchDocuments,
-		deleteDocument,
-	} = useDocuments(searchSpaceId, {
-		page: pageIndex,
-		pageSize: pageSize,
-	});
 
 	// Refetch when pagination changes or when search/filters change
 	useEffect(() => {
@@ -172,7 +165,6 @@ export default function DocumentsTable() {
 				}}
 			/>
 
-
 			<PaginationControls
 				pageIndex={pageIndex}
 				pageSize={pageSize}

surfsense_web/app/dashboard/[search_space_id]/documents/upload/page.tsx

@@ -1,7 +1,7 @@
 "use client";
 
-import { AnimatePresence, motion } from "motion/react";
 import { CheckCircle2, FileType, Info, Tag, Upload, X } from "lucide-react";
+import { AnimatePresence, motion } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useCallback, useState } from "react";
 import { useDropzone } from "react-dropzone";

surfsense_web/app/dashboard/[search_space_id]/documents/youtube/page.tsx

@@ -2,8 +2,8 @@
 
 import { IconBrandYoutube } from "@tabler/icons-react";
 import { type Tag, TagInput } from "emblor";
-import { motion, type Variants } from "motion/react";
 import { Loader2 } from "lucide-react";
+import { motion, type Variants } from "motion/react";
 import { useParams, useRouter } from "next/navigation";
 import { useState } from "react";
 import { toast } from "sonner";

surfsense_web/app/dashboard/[search_space_id]/logs/(manage)/page.tsx

@@ -15,7 +15,6 @@ import {
 	useReactTable,
 	type VisibilityState,
 } from "@tanstack/react-table";
-import { AnimatePresence, motion, type Variants } from "motion/react";
 import {
 	Activity,
 	AlertCircle,
@@ -42,6 +41,7 @@ import {
 	X,
 	Zap,
 } from "lucide-react";
+import { AnimatePresence, motion, type Variants } from "motion/react";
 import { useParams } from "next/navigation";
 import React, { useContext, useId, useMemo, useRef, useState } from "react";
 import { toast } from "sonner";

surfsense_web/app/dashboard/api-key/api-key-client.tsx

@@ -1,8 +1,8 @@
 "use client";
 
 import { IconCheck, IconCopy, IconKey } from "@tabler/icons-react";
-import { AnimatePresence, motion } from "motion/react";
 import { ArrowLeft } from "lucide-react";
+import { AnimatePresence, motion } from "motion/react";
 import { useRouter } from "next/navigation";
 import { Alert, AlertDescription, AlertTitle } from "@/components/ui/alert";
 import { Button } from "@/components/ui/button";

surfsense_web/app/dashboard/page.tsx

@@ -1,7 +1,7 @@
 "use client";
 
-import { motion, type Variants } from "motion/react";
 import { AlertCircle, Loader2, Plus, Search, Trash2 } from "lucide-react";
+import { motion, type Variants } from "motion/react";
 import Image from "next/image";
 import Link from "next/link";
 import { useRouter } from "next/navigation";

surfsense_web/app/onboard/page.tsx

@@ -1,7 +1,7 @@
 "use client";
 
-import { AnimatePresence, motion } from "motion/react";
 import { ArrowLeft, ArrowRight, Bot, CheckCircle, Sparkles } from "lucide-react";
+import { AnimatePresence, motion } from "motion/react";
 import { useRouter } from "next/navigation";
 import { useEffect, useState } from "react";
 import { Logo } from "@/components/Logo";

surfsense_web/components/pricing.tsx

@@ -2,8 +2,8 @@
 
 import NumberFlow from "@number-flow/react";
 import confetti from "canvas-confetti";
-import { motion } from "motion/react";
 import { Check, Star } from "lucide-react";
+import { motion } from "motion/react";
 import Link from "next/link";
 import { useRef, useState } from "react";
 import { buttonVariants } from "@/components/ui/button";
@@ -91,53 +91,55 @@ export function Pricing({
 				</span>
 			</div>
 
-		<div className={cn(
+			<div
-			"grid grid-cols-1 gap-4",
+				className={cn(
-			plans.length === 2 ? "md:grid-cols-2 max-w-5xl mx-auto" : "md:grid-cols-3"
+					"grid grid-cols-1 gap-4",
-		)}>
+					plans.length === 2 ? "md:grid-cols-2 max-w-5xl mx-auto" : "md:grid-cols-3"
-			{plans.map((plan, index) => (
+				)}
-				<motion.div
+			>
-					key={index}
+				{plans.map((plan, index) => (
-					initial={{ y: 50, opacity: 1 }}
+					<motion.div
-					whileInView={
+						key={index}
-						isDesktop
+						initial={{ y: 50, opacity: 1 }}
-							? plans.length === 2
+						whileInView={
-								? {
+							isDesktop
-										y: plan.isPopular ? -20 : 0,
+								? plans.length === 2
-										opacity: 1,
+									? {
-										scale: plan.isPopular ? 1.0 : 0.96,
+											y: plan.isPopular ? -20 : 0,
-									}
+											opacity: 1,
-								: {
+											scale: plan.isPopular ? 1.0 : 0.96,
-										y: plan.isPopular ? -20 : 0,
+										}
-										opacity: 1,
+									: {
-										x: index === 2 ? -30 : index === 0 ? 30 : 0,
+											y: plan.isPopular ? -20 : 0,
-										scale: index === 0 || index === 2 ? 0.94 : 1.0,
+											opacity: 1,
-									}
+											x: index === 2 ? -30 : index === 0 ? 30 : 0,
-							: {}
+											scale: index === 0 || index === 2 ? 0.94 : 1.0,
-					}
+										}
-					viewport={{ once: true }}
+								: {}
-					transition={{
+						}
-						duration: 1.6,
+						viewport={{ once: true }}
-						type: "spring",
+						transition={{
-						stiffness: 100,
+							duration: 1.6,
-						damping: 30,
+							type: "spring",
-						delay: 0.4,
+							stiffness: 100,
-						opacity: { duration: 0.5 },
+							damping: 30,
-					}}
+							delay: 0.4,
-					className={cn(
+							opacity: { duration: 0.5 },
-						`rounded-2xl border-[1px] p-6 bg-background text-center lg:flex lg:flex-col lg:justify-center relative`,
+						}}
-						plan.isPopular ? "border-primary border-2" : "border-border",
+						className={cn(
-						"flex flex-col",
+							`rounded-2xl border-[1px] p-6 bg-background text-center lg:flex lg:flex-col lg:justify-center relative`,
-						!plan.isPopular && "mt-5",
+							plan.isPopular ? "border-primary border-2" : "border-border",
-						plans.length === 3 && (index === 0 || index === 2)
+							"flex flex-col",
-							? "z-0 transform translate-x-0 translate-y-0 -translate-z-[50px] rotate-y-[10deg]"
+							!plan.isPopular && "mt-5",
-							: plans.length === 2 && !plan.isPopular
+							plans.length === 3 && (index === 0 || index === 2)
-								? "z-0"
+								? "z-0 transform translate-x-0 translate-y-0 -translate-z-[50px] rotate-y-[10deg]"
-								: "z-10",
+								: plans.length === 2 && !plan.isPopular
-						plans.length === 3 && index === 0 && "origin-right",
+									? "z-0"
-						plans.length === 3 && index === 2 && "origin-left"
+									: "z-10",
-					)}
+							plans.length === 3 && index === 0 && "origin-right",
-				>
+							plans.length === 3 && index === 2 && "origin-left"
+						)}
+					>
 						{plan.isPopular && (
 							<div className="absolute top-0 right-0 bg-primary py-0.5 px-2 rounded-bl-xl rounded-tr-xl flex items-center">
 								<Star className="text-primary-foreground h-4 w-4 fill-current" />
@@ -147,39 +149,39 @@ export function Pricing({
 							</div>
 						)}
 						<div className="flex-1 flex flex-col">
-						<p className="text-base font-semibold text-muted-foreground">{plan.name}</p>
+							<p className="text-base font-semibold text-muted-foreground">{plan.name}</p>
-						<div className="mt-6 flex items-center justify-center gap-x-2">
+							<div className="mt-6 flex items-center justify-center gap-x-2">
-							<span className="text-5xl font-bold tracking-tight text-foreground">
+								<span className="text-5xl font-bold tracking-tight text-foreground">
-								{isNaN(Number(plan.price)) ? (
+									{isNaN(Number(plan.price)) ? (
-									<span>{isMonthly ? plan.price : plan.yearlyPrice}</span>
+										<span>{isMonthly ? plan.price : plan.yearlyPrice}</span>
-								) : (
+									) : (
-									<NumberFlow
+										<NumberFlow
-										value={isMonthly ? Number(plan.price) : Number(plan.yearlyPrice)}
+											value={isMonthly ? Number(plan.price) : Number(plan.yearlyPrice)}
-										format={{
+											format={{
-											style: "currency",
+												style: "currency",
-											currency: "USD",
+												currency: "USD",
-											minimumFractionDigits: 0,
+												minimumFractionDigits: 0,
-											maximumFractionDigits: 0,
+												maximumFractionDigits: 0,
-										}}
+											}}
-										transformTiming={{
+											transformTiming={{
-											duration: 500,
+												duration: 500,
-											easing: "ease-out",
+												easing: "ease-out",
-										}}
+											}}
-										willChange
+											willChange
-										className="font-variant-numeric: tabular-nums"
+											className="font-variant-numeric: tabular-nums"
-									/>
+										/>
-								)}
+									)}
-							</span>
-							{plan.period && plan.period !== "Next 3 months" && (
-								<span className="text-sm font-semibold leading-6 tracking-wide text-muted-foreground">
-									/ {plan.period}
 								</span>
-							)}
+								{plan.period && plan.period !== "Next 3 months" && (
-						</div>
+									<span className="text-sm font-semibold leading-6 tracking-wide text-muted-foreground">
+										/ {plan.period}
+									</span>
+								)}
+							</div>
 
-						<p className="text-xs leading-5 text-muted-foreground">
+							<p className="text-xs leading-5 text-muted-foreground">
-							{isNaN(Number(plan.price)) ? "" : isMonthly ? "billed monthly" : "billed annually"}
+								{isNaN(Number(plan.price)) ? "" : isMonthly ? "billed monthly" : "billed annually"}
-						</p>
+							</p>
 
 							<ul className="mt-5 gap-2 flex flex-col">
 								{plan.features.map((feature, idx) => (

surfsense_web/hooks/use-media-query.ts

@@ -6,34 +6,32 @@ import { useEffect, useState } from "react";
  * @returns boolean - True if the media query matches, false otherwise
  */
 export function useMediaQuery(query: string): boolean {
-  const [matches, setMatches] = useState(false);
+	const [matches, setMatches] = useState(false);
 
-  useEffect(() => {
+	useEffect(() => {
-    // Check if we're in the browser (handle SSR)
+		// Check if we're in the browser (handle SSR)
-    if (typeof window === "undefined") {
+		if (typeof window === "undefined") {
-      return;
+			return;
-    }
+		}
 
-    const mediaQuery = window.matchMedia(query);
+		const mediaQuery = window.matchMedia(query);
-    
-    // Set initial value
-    setMatches(mediaQuery.matches);
 
-    // Create event listener
+		// Set initial value
-    const handler = (event: MediaQueryListEvent) => {
+		setMatches(mediaQuery.matches);
-      setMatches(event.matches);
-    };
 
-    // Add event listener
+		// Create event listener
-    mediaQuery.addEventListener("change", handler);
+		const handler = (event: MediaQueryListEvent) => {
+			setMatches(event.matches);
+		};
 
-    // Cleanup
+		// Add event listener
-    return () => {
+		mediaQuery.addEventListener("change", handler);
-      mediaQuery.removeEventListener("change", handler);
-    };
-  }, [query]);
 
-  return matches;
+		// Cleanup
+		return () => {
+			mediaQuery.removeEventListener("change", handler);
+		};
+	}, [query]);
+
+	return matches;
 }
-
-