feat: support scoping by project id (#11)

Author: adrianlyjak

test-proj/ui/src/lib/client.ts

@@ -11,12 +11,17 @@ import { EXTRACTED_DATA_COLLECTION } from "./config";
 
 const platformToken = import.meta.env.VITE_LLAMA_CLOUD_API_KEY;
 const apiBaseUrl = import.meta.env.VITE_LLAMA_CLOUD_BASE_URL;
+const projectId = import.meta.env.VITE_LLAMA_CLOUD_PROJECT_ID;
 
 // Configure the platform client
 cloudApiClient.setConfig({
   baseUrl: apiBaseUrl,
   headers: {
+    // optionally use a backend API token scoped to a project. For local development,
     ...(platformToken && { authorization: `Bearer ${platformToken}` }),
+    // This header is required for requests to correctly scope to the agent's project
+    // when authenticating with a user cookie
+    ...(projectId && { "Project-Id": projectId }),
   },
 });
 

test-proj/ui/vite.config.ts

@@ -6,6 +6,7 @@ import path from "path";
 export default defineConfig(({}) => {
   const deploymentId = process.env.LLAMA_DEPLOY_NEXTJS_DEPLOYMENT_NAME;
   const basePath = `/deployments/${deploymentId}/ui`;
+  const projectId = process.env.LLAMA_DEPLOY_PROJECT_ID;
 
   return {
     plugins: [react()],
@@ -30,6 +31,10 @@ export default defineConfig(({}) => {
       "import.meta.env.VITE_LLAMA_DEPLOY_DEPLOYMENT_NAME":
         JSON.stringify(deploymentId),
       "import.meta.env.VITE_LLAMA_DEPLOY_BASE_PATH": JSON.stringify(basePath),
+      ...(projectId && {
+        "import.meta.env.VITE_LLAMA_CLOUD_PROJECT_ID":
+          JSON.stringify(projectId),
+      }),
     },
   };
 });

ui/src/lib/client.ts

@@ -11,12 +11,17 @@ import { EXTRACTED_DATA_COLLECTION } from "./config";
 
 const platformToken = import.meta.env.VITE_LLAMA_CLOUD_API_KEY;
 const apiBaseUrl = import.meta.env.VITE_LLAMA_CLOUD_BASE_URL;
+const projectId = import.meta.env.VITE_LLAMA_CLOUD_PROJECT_ID;
 
 // Configure the platform client
 cloudApiClient.setConfig({
   baseUrl: apiBaseUrl,
   headers: {
+    // optionally use a backend API token scoped to a project. For local development,
     ...(platformToken && { authorization: `Bearer ${platformToken}` }),
+    // This header is required for requests to correctly scope to the agent's project
+    // when authenticating with a user cookie
+    ...(projectId && { "Project-Id": projectId }),
   },
 });
 

ui/vite.config.ts

@@ -6,6 +6,7 @@ import path from "path";
 export default defineConfig(({}) => {
   const deploymentId = process.env.LLAMA_DEPLOY_NEXTJS_DEPLOYMENT_NAME;
   const basePath = `/deployments/${deploymentId}/ui`;
+  const projectId = process.env.LLAMA_DEPLOY_PROJECT_ID;
 
   return {
     plugins: [react()],
@@ -30,6 +31,10 @@ export default defineConfig(({}) => {
       "import.meta.env.VITE_LLAMA_DEPLOY_DEPLOYMENT_NAME":
         JSON.stringify(deploymentId),
       "import.meta.env.VITE_LLAMA_DEPLOY_BASE_PATH": JSON.stringify(basePath),
+      ...(projectId && {
+        "import.meta.env.VITE_LLAMA_CLOUD_PROJECT_ID":
+          JSON.stringify(projectId),
+      }),
     },
   };
 });