diff --git a/src/app/api/chat/route.ts b/src/app/api/chat/route.ts
index a783a3f..bb13fd6 100644
--- a/src/app/api/chat/route.ts
+++ b/src/app/api/chat/route.ts
@@ -45,6 +45,7 @@ export async function POST(req: Request) {
const session = await verifyUser();
const userId = session?.data?.data?.id;
const publicKey = session?.data?.data?.publicKey;
+ const degenMode = session?.data?.data?.degenMode;
if (!userId) {
return new Response('Unauthorized', { status: 401 });
@@ -124,6 +125,7 @@ export async function POST(req: Request) {
`User Solana wallet public key: ${publicKey}`,
`User ID: ${userId}`,
`Conversation ID: ${conversationId}`,
+ `Degen Mode: ${degenMode}`,
].join('\n\n');
// Filter out empty messages and ensure sorting by createdAt ascending
@@ -168,7 +170,12 @@ export async function POST(req: Request) {
// Exclude the confirmation tool if we are handling a confirmation
const { toolsRequired, usage: orchestratorUsage } =
- await getToolsFromOrchestrator(relevant, confirmationHandled);
+ await getToolsFromOrchestrator(
+ relevant,
+ degenMode || confirmationHandled,
+ );
+
+ console.log('toolsRequired', toolsRequired);
logWithTiming(
startTime,
diff --git a/src/components/ui/switch.tsx b/src/components/ui/switch.tsx
new file mode 100644
index 0000000..5f4117f
--- /dev/null
+++ b/src/components/ui/switch.tsx
@@ -0,0 +1,29 @@
+"use client"
+
+import * as React from "react"
+import * as SwitchPrimitives from "@radix-ui/react-switch"
+
+import { cn } from "@/lib/utils"
+
+const Switch = React.forwardRef<
+ React.ElementRef,
+ React.ComponentPropsWithoutRef
+>(({ className, ...props }, ref) => (
+
+
+
+))
+Switch.displayName = SwitchPrimitives.Root.displayName
+
+export { Switch }
diff --git a/src/server/actions/user.ts b/src/server/actions/user.ts
index d150b25..ba15c3b 100644
--- a/src/server/actions/user.ts
+++ b/src/server/actions/user.ts
@@ -1,9 +1,8 @@
'use server';
+import { revalidateTag } from 'next/cache';
import { cookies } from 'next/headers';
-import { Wallet } from '@prisma/client';
-import { Email } from '@privy-io/react-auth';
import { PrivyClient } from '@privy-io/server-auth';
import { WalletWithMetadata } from '@privy-io/server-auth';
import { z } from 'zod';
@@ -92,7 +91,12 @@ const getOrCreateUser = actionClient
});
export const verifyUser = actionClient.action<
- ActionResponse<{ id: string; publicKey?: string }>
+ ActionResponse<{
+ id: string;
+ degenMode: boolean;
+ publicKey?: string;
+ privyId: string;
+ }>
>(async () => {
const token = (await cookies()).get('privy-token')?.value;
if (!token) {
@@ -108,6 +112,8 @@ export const verifyUser = actionClient.action<
where: { privyId: claims.userId },
select: {
id: true,
+ degenMode: true,
+ privyId: true,
wallets: {
select: {
publicKey: true,
@@ -130,7 +136,9 @@ export const verifyUser = actionClient.action<
success: true,
data: {
id: user.id,
+ privyId: user.privyId,
publicKey: user.wallets[0]?.publicKey,
+ degenMode: user.degenMode,
},
};
} catch {
@@ -240,3 +248,26 @@ export const syncEmbeddedWallets = actionClient.action<
export const getPrivyClient = actionClient.action(
async () => PRIVY_SERVER_CLIENT,
);
+
+export type UserUpdateData = {
+ degenMode?: boolean;
+};
+export async function updateUser(data: UserUpdateData) {
+ try {
+ const authResult = await verifyUser();
+ const userId = authResult?.data?.data?.id;
+ const privyId = authResult?.data?.data?.privyId;
+
+ if (!userId) {
+ return { success: false, error: 'UNAUTHORIZED' };
+ }
+ await prisma.user.update({
+ where: { id: userId },
+ data,
+ });
+ revalidateTag(`user-${privyId}`);
+ return { success: true };
+ } catch (error) {
+ return { success: false, error: 'Failed to update user' };
+ }
+}
diff --git a/src/types/db.ts b/src/types/db.ts
index d9b22af..e24496a 100644
--- a/src/types/db.ts
+++ b/src/types/db.ts
@@ -32,7 +32,13 @@ export type PrismaUser = _PrismaUser & {
export type NeurUser = Pick<
PrismaUser,
- 'id' | 'privyId' | 'createdAt' | 'updatedAt' | 'earlyAccess' | 'wallets'
+ | 'id'
+ | 'privyId'
+ | 'createdAt'
+ | 'updatedAt'
+ | 'earlyAccess'
+ | 'wallets'
+ | 'degenMode'
> & {
privyUser: PrivyUser;
hasEAP: boolean;