SDK Integration

1
const client = new ClientCredentials({
2
  auth: {
3
    tokenHost: process.env.RB_OAUTH2_URL,
4
    tokenPath: process.env.RB_OAUTH2_TOKEN_PATH,
5
  },
6
  client: {
7
    id: process.env.RB_CLIENT_ID,
8
    secret: process.env.RB_CLIENT_SECRET,
9
  },
10
});
11
const accessToken = await client.getToken({});

1
// app/tasks/product-listing/route.ts
2
export async function POST(request: Request) {
3
  const token = await getAccessToken();
4
  const body = await request.json();
5
6
  const rightbrain = RightBrainClient.getInstance({
7
    accessToken: token.token.access_token,
8
    baseUrl: process.env.RB_API_URL!,
9
    organizationId: process.env.RB_ORG_ID!,
10
    projectId: process.env.RB_PROJECT_ID!,
11
  });
12
13
  const response = await rightbrain.runGenerateImageBasedProductListing({
14
    inputs: { product_name: body.product_name },
15
  });
16
17
  return Response.json(response);
18
}

1
// hooks/useGenerateImageBasedProductListing.ts
2
export function useGenerateImageBasedProductListing(taskId: string) {
3
  const [loading, setLoading] = useState(false);
4
  const [data, setData] = useState(null);
5
6
  const execute = async (inputs) => {
7
    setLoading(true);
8
    const res = await fetch(`/api/tasks/${taskId}`, {
9
      method: "POST",
10
      headers: { "Content-Type": "application/json" },
11
      body: JSON.stringify(inputs),
12
    });
13
    const json = await res.json();
14
    setData(json.response);
15
    setLoading(false);
16
  };
17
18
  return { execute, data, loading };
19
}