Dashboard, Docs - Use updated Nebula API

Author: MananTank

apps/dashboard/src/app/nebula-app/(app)/api/chat.ts

@@ -4,10 +4,9 @@ import { stream } from "fetch-event-stream";
 import type { NebulaTxData } from "../components/Chats";
 import type { ExecuteConfig } from "./types";
 
-export type ContextFilters = {
+export type NebulaContext = {
   chainIds?: string[];
-  contractAddresses?: string[];
-  walletAddresses?: string[];
+  walletAddress?: string;
 };
 
 export async function promptNebula(params: {
@@ -17,27 +16,23 @@ export async function promptNebula(params: {
   authToken: string;
   handleStream: (res: ChatStreamedResponse) => void;
   abortController: AbortController;
-  contextFilters: undefined | ContextFilters;
+  context: undefined | NebulaContext;
 }) {
   const body: Record<string, string | boolean | object> = {
     message: params.message,
     user_id: "default-user",
-    session_id: params.sessionId,
     stream: true,
+    // TODO: move the session_id back here once the bug is fixed
   };
 
-  if (params.contextFilters) {
-    body.context_filter = {
-      chain_ids: params.contextFilters.chainIds || [],
-      contract_addresses: params.contextFilters.contractAddresses || [],
-      wallet_addresses: params.contextFilters.walletAddresses || [],
+  if (params.context) {
+    body.context = {
+      session_id: params.sessionId, // TODO - this can be moved back to top level once this is fixed in nebula-server
+      chain_ids: params.context.chainIds || [],
+      wallet_address: params.context.walletAddress,
     };
   }
 
-  if (params.config) {
-    body.execute_config = params.config;
-  }
-
   const events = await stream(`${NEXT_PUBLIC_NEBULA_URL}/chat`, {
     method: "POST",
     headers: {

apps/dashboard/src/app/nebula-app/(app)/api/session.ts

@@ -1,6 +1,6 @@
 import { NEXT_PUBLIC_NEBULA_URL } from "@/constants/env";
 import { fetchWithAuthToken } from "../../../../utils/fetchWithAuthToken";
-import type { ContextFilters } from "./chat";
+import type { NebulaContext } from "./chat";
 import type {
   DeletedSessionInfo,
   ExecuteConfig,
@@ -9,25 +9,17 @@ import type {
   UpdatedSessionInfo,
 } from "./types";
 
-// TODO - get the spec for return types on /session POST and PUT
-
 export async function createSession(params: {
   authToken: string;
   config: ExecuteConfig | null;
-  contextFilters: ContextFilters | undefined;
+  context: NebulaContext | undefined;
 }) {
-  const body: Record<string, string | boolean | object> = {
-    can_execute: !!params.config,
-  };
-  if (params.config) {
-    body.execute_config = params.config;
-  }
+  const body: Record<string, string | boolean | object> = {};
 
-  if (params.contextFilters) {
-    body.context_filter = {
-      chain_ids: params.contextFilters.chainIds || [],
-      contract_addresses: params.contextFilters.contractAddresses || [],
-      wallet_addresses: params.contextFilters.walletAddresses || [],
+  if (params.context) {
+    body.context = {
+      chain_ids: params.context.chainIds || [],
+      wallet_address: params.context.walletAddress,
     };
   }
 
@@ -50,20 +42,14 @@ export async function updateSession(params: {
   authToken: string;
   config: ExecuteConfig | null;
   sessionId: string;
-  contextFilters: ContextFilters | undefined;
+  contextFilters: NebulaContext | undefined;
 }) {
-  const body: Record<string, string | boolean | object> = {
-    can_execute: !!params.config,
-  };
-  if (params.config) {
-    body.execute_config = params.config;
-  }
+  const body: Record<string, string | boolean | object> = {};
 
   if (params.contextFilters) {
-    body.context_filter = {
+    body.context = {
       chain_ids: params.contextFilters.chainIds || [],
-      contract_addresses: params.contextFilters.contractAddresses || [],
-      wallet_addresses: params.contextFilters.walletAddresses || [],
+      wallet_address: params.contextFilters.walletAddress,
     };
   }
 

apps/dashboard/src/app/nebula-app/(app)/api/types.ts

@@ -21,8 +21,7 @@ export type ExecuteConfig = EngineConfig | SessionKeyConfig | ClientConfig;
 
 type SessionContextFilter = {
   chain_ids: string[] | null;
-  contract_addresses: string[] | null;
-  wallet_addresses: string[] | null;
+  wallet_address: string | null;
 };
 
 export type SessionInfo = {
@@ -31,7 +30,6 @@ export type SessionInfo = {
   modal_name: string;
   archive_at: string | null;
   can_execute: boolean;
-  execute_config: ExecuteConfig | null;
   created_at: string;
   deleted_at: string | null;
   history: Array<{
@@ -43,7 +41,7 @@ export type SessionInfo = {
   archived_at: string | null;
   title: string | null;
   is_public: boolean | null;
-  context_filter: SessionContextFilter | null;
+  context: SessionContextFilter | null;
   // memory
   // action: array<object> | null; <-- type of this is not available on https://nebula-api.thirdweb-dev.com/docs#/default/get_session_session__session_id__get
 };
@@ -52,8 +50,7 @@ export type UpdatedSessionInfo = {
   title: string;
   modal_name: string;
   account_id: string;
-  execute_config: ExecuteConfig | null;
-  context_filter: SessionContextFilter | null;
+  context: SessionContextFilter | null;
 };
 
 export type DeletedSessionInfo = {

apps/dashboard/src/app/nebula-app/(app)/components/ChatPageContent.tsx

@@ -14,7 +14,7 @@ import { ArrowRightIcon } from "lucide-react";
 import Link from "next/link";
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
 import { useActiveAccount, useActiveWalletChain } from "thirdweb/react";
-import { type ContextFilters, promptNebula } from "../api/chat";
+import { type NebulaContext, promptNebula } from "../api/chat";
 import { createSession, updateSession } from "../api/session";
 import type { ExecuteConfig, SessionInfo } from "../api/types";
 import { newChatPageUrlStore, newSessionsStore } from "../stores";
@@ -81,39 +81,38 @@ export function ChatPageContent(props: {
     useState(false);
 
   const [contextFilters, _setContextFilters] = useState<
-    ContextFilters | undefined
+    NebulaContext | undefined
   >(() => {
-    const contextFilterRes = props.session?.context_filter;
-    const value: ContextFilters = {
-      chainIds: contextFilterRes?.chain_ids || undefined,
-      contractAddresses: contextFilterRes?.contract_addresses || undefined,
-      walletAddresses: contextFilterRes?.wallet_addresses || undefined,
+    const contextRes = props.session?.context;
+    const value: NebulaContext = {
+      chainIds: contextRes?.chain_ids || undefined,
+      walletAddress: contextRes?.wallet_address || undefined,
     };
 
     return value;
   });
 
-  const setContextFilters = useCallback((v: ContextFilters | undefined) => {
+  const setContextFilters = useCallback((v: NebulaContext | undefined) => {
     _setContextFilters(v);
     setHasUserUpdatedContextFilters(true);
   }, []);
 
   const isNewSession = !props.session;
 
-  // if this is a new session, user has not manually updated context filters
-  // update the context filters to the current user's wallet address and chain id
+  // if this is a new session, user has not manually updated context
+  // update the context to the current user's wallet address and chain id
   // eslint-disable-next-line no-restricted-syntax
   useEffect(() => {
     if (!isNewSession || hasUserUpdatedContextFilters) {
       return;
     }
 
     _setContextFilters((_contextFilters) => {
-      const updatedContextFilters: ContextFilters = _contextFilters
+      const updatedContextFilters: NebulaContext = _contextFilters
         ? { ..._contextFilters }
         : {};
 
-      updatedContextFilters.walletAddresses = address ? [address] : [];
+      updatedContextFilters.walletAddress = address;
       updatedContextFilters.chainIds = activeChain
         ? [activeChain.id.toString()]
         : [];
@@ -164,7 +163,7 @@ export function ChatPageContent(props: {
     const session = await createSession({
       authToken: props.authToken,
       config,
-      contextFilters,
+      context: contextFilters,
     });
     setSessionId(session.id);
     return session;
@@ -308,7 +307,7 @@ export function ChatPageContent(props: {
               }
             }
           },
-          contextFilters: contextFilters,
+          context: contextFilters,
         });
       } catch (error) {
         if (abortController.signal.aborted) {
@@ -363,9 +362,9 @@ export function ChatPageContent(props: {
   const showEmptyState = !userHasSubmittedMessage && messages.length === 0;
 
   const handleUpdateContextFilters = async (
-    values: ContextFilters | undefined,
+    values: NebulaContext | undefined,
   ) => {
-    // if session is not yet created, don't need to update sessions - starting a chat will create a session with the context filters
+    // if session is not yet created, don't need to update sessions - starting a chat will create a session with the context
     if (sessionId) {
       await updateSession({
         authToken: props.authToken,

apps/dashboard/src/app/nebula-app/(app)/components/ContextFilters.stories.tsx

@@ -2,7 +2,7 @@ import type { Meta, StoryObj } from "@storybook/react";
 import { useState } from "react";
 import { Toaster } from "sonner";
 import { BadgeContainer, mobileViewport } from "../../../../stories/utils";
-import type { ContextFilters } from "../api/chat";
+import type { NebulaContext } from "../api/chat";
 import ContextFiltersButton from "./ContextFilters";
 
 const meta = {
@@ -50,51 +50,17 @@ function Story() {
 
       <Variant
         contextFilters={{
-          contractAddresses: ["0x1E51e33F9838A5a043E099C60409f62aA564272f"],
-        }}
-        label="1 contract"
-      />
-
-      <Variant
-        contextFilters={{
-          contractAddresses: [
-            "0x1E51e33F9838A5a043E099C60409f62aA564272f",
-            "0xF61c8d5492139b40af09bDB353733d5F0a348aCf",
-          ],
-        }}
-        label="Few contracts"
-      />
-
-      <Variant
-        contextFilters={{
-          walletAddresses: ["0x1F846F6DAE38E1C88D71EAA191760B15f38B7A37"],
+          walletAddress: "0x1F846F6DAE38E1C88D71EAA191760B15f38B7A37",
         }}
         label="1 wallet"
       />
 
-      <Variant
-        contextFilters={{
-          walletAddresses: [
-            "0x1F846F6DAE38E1C88D71EAA191760B15f38B7A37",
-            "0x83Dd93fA5D8343094f850f90B3fb90088C1bB425",
-          ],
-        }}
-        label="Few wallets"
-      />
-
       <Variant
         contextFilters={{
           chainIds: ["137", "10", "421614"],
-          contractAddresses: [
-            "0x1E51e33F9838A5a043E099C60409f62aA564272f",
-            "0xF61c8d5492139b40af09bDB353733d5F0a348aCf",
-          ],
-          walletAddresses: [
-            "0x1F846F6DAE38E1C88D71EAA191760B15f38B7A37",
-            "0x83Dd93fA5D8343094f850f90B3fb90088C1bB425",
-          ],
+          walletAddress: "0x1F846F6DAE38E1C88D71EAA191760B15f38B7A37",
         }}
-        label="chains + wallets + contracts"
+        label="chains + wallet"
       />
       <Toaster richColors />
     </div>
@@ -103,18 +69,18 @@ function Story() {
 
 function Variant(props: {
   label: string;
-  contextFilters: ContextFilters | undefined;
+  contextFilters: NebulaContext | undefined;
 }) {
   const [contextFilters, setContextFilters] = useState<
-    ContextFilters | undefined
+    NebulaContext | undefined
   >(props.contextFilters);
   return (
     <BadgeContainer label={props.label}>
       <ContextFiltersButton
         contextFilters={contextFilters}
         setContextFilters={setContextFilters}
         updateContextFilters={async (values) => {
-          console.log("Updating context filters", values);
+          console.log("Updating context", values);
           await new Promise((resolve) => setTimeout(resolve, 1000));
         }}
       />