bugfix/fix model name

frontend/src/auth.ts

@@ -5,7 +5,7 @@ import { decodeJwt } from 'jose';
 export async function parseAuthToken(header: string) {
   if (!header) {
     console.error('No auth token provided to parse');
-    return { email: null, roles: [] };
+    return { email: null };
   }
 
   // Decode without verification since we're using auth-at-the-edge and can trust all traffic
@@ -14,27 +14,16 @@ export async function parseAuthToken(header: string) {
     tokenContent = decodeJwt(header);
   } catch(error) {
     console.error('Malformed JWT during decoding: ' + header, error);
-    return { email: null, roles: [] };
+    return { email: null };
   }
 
   const email = tokenContent.email as string | undefined;
   if (!email) {
     console.error('No email found in token');
-    return null;
+    return { email: null };
   }
 
-  const realmAccess = tokenContent.realm_access as { roles?: string[] } | undefined;
-  if (!realmAccess) {
-    console.error('No realm access information found in token');
-    return { email: null, roles: [] };
-  }
-
-  const roles = realmAccess.roles || [];
-  // console.debug(`Roles found for user ${email}: ${roles}`);
-  return {
-    email,
-    roles,
-  };
+  return { email };
 }
 
 

frontend/src/layouts/Chat.astro

@@ -92,7 +92,7 @@ const singleServer = mcpServers.find((server) => server.name.toLowerCase() === s
           <model-selector class="[tool-selector+&]:left-21 top-0 md:mt-0 md:absolute">
             <label class="sr-only" for="model-selector">Mode</label>
             <select class="border-pink border-solid border-1 cursor-pointer inline-block min-h-[36px] mr-5 outline-pink px-2 py-[5px] rounded-sm text-pink w-40 focus-visible:outline-2" id="model-selector" name="model">
-              <option value="locai-l1" class="p-[4px]">
+              <option value="locailabs/locai-l1-large-2011" class="p-[4px]">
                 <svg focusable="false" aria-hidden="true" fill="none" viewBox="0 0 124 124" width="20px" height="20px">
                   <rect x="0" y="0" width="110" height="110" rx="60" fill="#CCFF00"/>
                   <rect x="30" y="25" width="10" height="60"  fill="#000000"/>
@@ -106,13 +106,13 @@ const singleServer = mcpServers.find((server) => server.name.toLowerCase() === s
                 <svg focusable="false" aria-hidden="true" width="20px" height="20px" viewBox="0 0 1200 1200"><path fill="#d97757" stroke="none" d="M 233.959793 800.214905 L 468.644287 668.536987 L 472.590637 657.100647 L 468.644287 650.738403 L 457.208069 650.738403 L 417.986633 648.322144 L 283.892639 644.69812 L 167.597321 639.865845 L 54.926208 633.825623 L 26.577238 627.785339 L 3.3e-05 592.751709 L 2.73832 575.27533 L 26.577238 559.248352 L 60.724873 562.228149 L 136.187973 567.382629 L 249.422867 575.194763 L 331.570496 580.026978 L 453.261841 592.671082 L 472.590637 592.671082 L 475.328857 584.859009 L 468.724915 580.026978 L 463.570557 575.194763 L 346.389313 495.785217 L 219.543671 411.865906 L 153.100723 363.543762 L 117.181267 339.060425 L 99.060455 316.107361 L 91.248367 266.01355 L 123.865784 230.093994 L 167.677887 233.073853 L 178.872513 236.053772 L 223.248367 270.201477 L 318.040283 343.570496 L 441.825592 434.738342 L 459.946411 449.798706 L 467.194672 444.64447 L 468.080597 441.020203 L 459.946411 427.409485 L 392.617493 305.718323 L 320.778564 181.932983 L 288.80542 130.630859 L 280.348999 99.865845 C 277.369171 87.221436 275.194641 76.590698 275.194641 63.624268 L 312.322174 13.20813 L 332.8591 6.604126 L 382.389313 13.20813 L 403.248352 31.328979 L 434.013519 101.71814 L 483.865753 212.537048 L 561.181274 363.221497 L 583.812134 407.919434 L 595.892639 449.315491 L 600.40271 461.959839 L 608.214783 461.959839 L 608.214783 454.711609 L 614.577271 369.825623 L 626.335632 265.61084 L 637.771851 131.516846 L 641.718201 93.745117 L 660.402832 48.483276 L 697.530334 24.000122 L 726.52356 37.852417 L 750.362549 72 L 747.060486 94.067139 L 732.886047 186.201416 L 705.100708 330.52356 L 686.979919 427.167847 L 697.530334 427.167847 L 709.61084 415.087341 L 758.496704 350.174561 L 840.644348 247.490051 L 876.885925 206.738342 L 919.167847 161.71814 L 946.308838 140.29541 L 997.61084 140.29541 L 1035.38269 196.429626 L 1018.469849 254.416199 L 965.637634 321.422852 L 921.825562 378.201538 L 859.006714 462.765259 L 819.785278 530.41626 L 823.409424 535.812073 L 832.75177 534.92627 L 974.657776 504.724915 L 1051.328979 490.872559 L 1142.818848 475.167786 L 1184.214844 494.496582 L 1188.724854 514.147644 L 1172.456421 554.335693 L 1074.604126 578.496765 L 959.838989 601.449829 L 788.939636 641.879272 L 786.845764 643.409485 L 789.261841 646.389343 L 866.255127 653.637634 L 899.194702 655.409424 L 979.812134 655.409424 L 1129.932861 666.604187 L 1169.154419 692.537109 L 1192.671265 724.268677 L 1188.724854 748.429688 L 1128.322144 779.194641 L 1046.818848 759.865845 L 856.590759 714.604126 L 791.355774 698.335754 L 782.335693 698.335754 L 782.335693 703.731567 L 836.69812 756.885986 L 936.322205 846.845581 L 1061.073975 962.81897 L 1067.436279 991.490112 L 1051.409424 1014.120911 L 1034.496704 1011.704712 L 924.885986 929.234924 L 882.604126 892.107544 L 786.845764 811.48999 L 780.483276 811.48999 L 780.483276 819.946289 L 802.550415 852.241699 L 919.087341 1027.409424 L 925.127625 1081.127686 L 916.671204 1098.604126 L 886.469849 1109.154419 L 853.288696 1103.114136 L 785.073914 1007.355835 L 714.684631 899.516785 L 657.906067 802.872498 L 650.979858 806.81897 L 617.476624 1167.704834 L 601.771851 1186.147705 L 565.530212 1200 L 535.328857 1177.046997 L 519.302124 1139.919556 L 535.328857 1066.550537 L 554.657776 970.792053 L 570.362488 894.68457 L 584.536926 800.134277 L 592.993347 768.724976 L 592.429626 766.630859 L 585.503479 767.516968 L 514.22821 865.369263 L 405.825531 1011.865906 L 320.053711 1103.677979 L 299.516815 1111.812256 L 263.919525 1093.369263 L 267.221497 1060.429688 L 287.114136 1031.114136 L 405.825531 880.107361 L 477.422913 786.52356 L 523.651062 732.483276 L 523.328918 724.671265 L 520.590698 724.671265 L 205.288605 929.395935 L 149.154434 936.644409 L 124.993355 914.01355 L 127.973183 876.885986 L 139.409409 864.80542 L 234.201385 799.570435 L 233.879227 799.8927 Z"/></svg>
                 Claude Sonnet 3.7
               </option>
-              <option value="gemini-2.5-flash" class="p-[4px]">
+              <option value="gemini-2.5-flash-uk" class="p-[4px]">
                 <svg focusable="false" aria-hidden="true" fill="none" viewBox="0 0 16 16" width="20px" height="20px"><path d="M16 8.016A8.522 8.522 0 008.016 16h-.032A8.521 8.521 0 000 8.016v-.032A8.521 8.521 0 007.984 0h.032A8.522 8.522 0 0016 7.984v.032z" fill="url(#prefix__paint0_radial_980_20147)"/><defs><radialGradient id="prefix__paint0_radial_980_20147" cx="0" cy="0" r="1" gradientUnits="userSpaceOnUse" gradientTransform="matrix(16.1326 5.4553 -43.70045 129.2322 1.588 6.503)"><stop offset=".067" stop-color="#9168C0"/><stop offset=".343" stop-color="#5684D1"/><stop offset=".672" stop-color="#1BA1E3"/></radialGradient></defs></svg>
                 Gemini Flash 2.5
               </option>
-              <option value="o4-mini" class="p-[4px]">
+              <option value="gpt-4.1-nano" class="p-[4px]">
                 <svg focusable="false" aria-hidden="true" fill="#000000" width="20px" height="20px" viewBox="0 0 24 24" role="img"><path d="M22.2819 9.8211a5.9847 5.9847 0 0 0-.5157-4.9108 6.0462 6.0462 0 0 0-6.5098-2.9A6.0651 6.0651 0 0 0 4.9807 4.1818a5.9847 5.9847 0 0 0-3.9977 2.9 6.0462 6.0462 0 0 0 .7427 7.0966 5.98 5.98 0 0 0 .511 4.9107 6.051 6.051 0 0 0 6.5146 2.9001A5.9847 5.9847 0 0 0 13.2599 24a6.0557 6.0557 0 0 0 5.7718-4.2058 5.9894 5.9894 0 0 0 3.9977-2.9001 6.0557 6.0557 0 0 0-.7475-7.0729zm-9.022 12.6081a4.4755 4.4755 0 0 1-2.8764-1.0408l.1419-.0804 4.7783-2.7582a.7948.7948 0 0 0 .3927-.6813v-6.7369l2.02 1.1686a.071.071 0 0 1 .038.052v5.5826a4.504 4.504 0 0 1-4.4945 4.4944zm-9.6607-4.1254a4.4708 4.4708 0 0 1-.5346-3.0137l.142.0852 4.783 2.7582a.7712.7712 0 0 0 .7806 0l5.8428-3.3685v2.3324a.0804.0804 0 0 1-.0332.0615L9.74 19.9502a4.4992 4.4992 0 0 1-6.1408-1.6464zM2.3408 7.8956a4.485 4.485 0 0 1 2.3655-1.9728V11.6a.7664.7664 0 0 0 .3879.6765l5.8144 3.3543-2.0201 1.1685a.0757.0757 0 0 1-.071 0l-4.8303-2.7865A4.504 4.504 0 0 1 2.3408 7.872zm16.5963 3.8558L13.1038 8.364 15.1192 7.2a.0757.0757 0 0 1 .071 0l4.8303 2.7913a4.4944 4.4944 0 0 1-.6765 8.1042v-5.6772a.79.79 0 0 0-.407-.667zm2.0107-3.0231l-.142-.0852-4.7735-2.7818a.7759.7759 0 0 0-.7854 0L9.409 9.2297V6.8974a.0662.0662 0 0 1 .0284-.0615l4.8303-2.7866a4.4992 4.4992 0 0 1 6.6802 4.66zM8.3065 12.863l-2.02-1.1638a.0804.0804 0 0 1-.038-.0567V6.0742a4.4992 4.4992 0 0 1 7.3757-3.4537l-.142.0805L8.704 5.459a.7948.7948 0 0 0-.3927.6813zm1.0976-2.3654l2.602-1.4998 2.6069 1.4998v2.9994l-2.5974 1.4997-2.6067-1.4997Z"/></svg>
-                o4 mini
+                GPT 4.1 nano
               </option>
             </select>
           </model-selector>

frontend/src/logic/ai3.ts

@@ -36,13 +36,19 @@ export const getLlmResponse = async(messages: Message[], selectedServers: FormDa
 
   let agentModel;
   if (process.env['USE_LITE_LLM'] === 'true') {
-    console.log('Using Lite LLM');
+    const apiKey = process.env['LITELLM_GOVAI_CLIENT_OPENAI_API_KEY'];
+    const baseURL = process.env['LLM_GATEWAY_URL'];
+    console.log('LiteLLM configuration:', {
+      baseURL,
+      apiKey: apiKey ? `${apiKey.substring(0, 8)}...${apiKey.substring(apiKey.length - 4)}` : 'missing',
+      model: selectedModel,
+    });
     agentModel = new ChatOpenAI({
-      openAIApiKey: process.env['LITELLM_GOVAI_CLIENT_OPENAI_API_KEY'],
+      apiKey,
       configuration: {
-        baseURL: process.env['LLM_GATEWAY_URL'],
+        baseURL,
       },
-      modelName: `${selectedModel}`,
+      model: selectedModel,
       callbackManager,
     });
   } else {

frontend/src/logic/get-servers.ts

@@ -109,18 +109,17 @@ const getTools = async(mcpServer: MCP_SERVER, authToken?: string) => {
 
   try {
     console.log(`${mcpServer.name}: Connected using Streamable HTTP transport`);
+    const serverMcpTools = await loadMcpTools(mcpServer.url, client) as unknown as Tool[];
+    const toolList = await client.listTools();
+    serverMcpTools.forEach((tool, toolIndex) => {
+      tool.serverName = mcpServer.name;
+      tool.annotations = toolList.tools[toolIndex].annotations;
+    });
+    tools.push(...serverMcpTools);
   } catch(error) { /* eslint @typescript-eslint/no-unused-vars: "off" */
-    console.log(`${mcpServer.name}: Error connecting via StreamableHTTTP`, error);
+    console.log(`${mcpServer.name}: Error loading tools`, error);
   }
 
-  const serverMcpTools = await loadMcpTools(mcpServer.url, client) as unknown as Tool[];
-  const toolList = await client.listTools();
-  serverMcpTools.forEach((tool, toolIndex) => {
-    tool.serverName = mcpServer.name;
-    tool.annotations = toolList.tools[toolIndex].annotations;
-  });
-
-  tools.push(...serverMcpTools);
   return tools;
 };
 
@@ -141,13 +140,15 @@ const getPrompt = async(promptName: string, mcpServer: MCP_SERVER, authToken?: s
 
 const cachedServers: MCP_SERVER[] = [];
 let caddyServer: MCP_SERVER | undefined;
+let testServer: MCP_SERVER | undefined;
 
 
-// Cache all servers except Caddy
+// Cache all servers except Caddy and test-mcp-server
 (async() => {
   const servers = getServerList();
   caddyServer = servers.find((server) => server.name === 'Caddy');
-  for (const server of servers.filter((server) => server.name !== 'Caddy')) {
+  testServer = servers.find((server) => server.name === 'test-mcp-server');
+  for (const server of servers.filter((server) => server.name !== 'Caddy' && server.name !== 'test-mcp-server')) {
     server.tools = await getTools(server);
     cachedServers.push(server);
   }
@@ -158,25 +159,30 @@ export const getMcpServers = async(authToken: string) => {
 
   // Get Caddy collections
   const caddyServers: MCP_SERVER[] = [];
-  if (!caddyServer) {
-    return cachedServers;
+  if (caddyServer) {
+    const caddyTools = await getTools(caddyServer, authToken);
+
+    for (const tool of caddyTools) {
+      const prompt = await getPrompt(tool.name, caddyServer, authToken);
+      caddyServers.push({
+        name: tool.annotations?.title?.replace('Search ', '') || tool.name,
+        description: tool.description,
+        url: caddyServer?.url || '',
+        accessToken: caddyServer?.accessToken,
+        tools: [tool],
+        customPrompt: prompt.trim(),
+        isCaddy: true,
+      });
+    }
   }
 
-  const caddyTools = await getTools(caddyServer, authToken);
-
-  for (const tool of caddyTools) {
-    const prompt = await getPrompt(tool.name, caddyServer, authToken);
-    caddyServers.push({
-      name: tool.annotations?.title?.replace('Search ', '') || tool.name,
-      description: tool.description,
-      url: caddyServer?.url || '',
-      accessToken: caddyServer?.accessToken,
-      tools: [tool],
-      customPrompt: prompt.trim(),
-      isCaddy: true,
-    });
+  // Load test server tools dynamically (not cached)
+  const testServers: MCP_SERVER[] = [];
+  if (testServer) {
+    testServer.tools = await getTools(testServer, authToken);
+    testServers.push(testServer);
   }
 
-  return [...caddyServers, ...cachedServers];
+  return [...caddyServers, ...testServers, ...cachedServers];
 
 };

frontend/src/middleware.ts

@@ -1,4 +1,5 @@
 import 'dotenv/config';
+import type { MiddlewareNext } from 'astro';
 import { parseAuthToken } from './auth.ts';
 
 // Define paths that should be public (no authorisation required)
@@ -11,7 +12,7 @@ const PUBLIC_PATHS = [
 ];
 const TEST_AUTHORISATION_JWT = 'eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOiIxNzM5ODk2MTk5IiwiaWF0IjoiMTczOTg5NTg5OSIsImF1dGhfdGltZSI6IjE3Mzk4OTM1MjkiLCJqdGkiOiIyYmVmOGI1ZS0yOGY0LTQ2OWQtYWQ2My1lZjJlNDgxNzliODYiLCJpc3MiOiJodHRwczovL2xvY2FsLXRlc3Rpbmcub2JmdXNjYXRlZC50ZXN0LmRvbWFpbi5nb3YudWsvcmVhbG1zL29iZnVzY2F0ZWQiLCJhdWQiOiJhY2NvdW50Iiwic3ViIjoiYmMzNzNkZTQtNDAyMi00NmIyLTgxNTEtZjA0NjEzNzhlOWNiIiwidHlwIjoiQmVhcmVyIiwiYXpwIjoibWludXRlIiwic2lkIjoiYzM2NmE5ZmUtMDNiNC00MjIxLWI0ZWItOTE0MzMzNWFhNjUyIiwiYWNyIjoiMSIsImFsbG93ZWQtb3JpZ2lucyI6WyJodHRwczovL2xvY2FsLXRlc3Rpbmcub2JmdXNjYXRlZC50ZXN0LmRvbWFpbi5nb3YudWsiXSwicmVhbG1fYWNjZXNzIjp7InJvbGVzIjpbImxvY2FsLXRlc3RpbmciXX0sInJlc291cmNlX2FjY2VzcyI6eyJhY2NvdW50Ijp7InJvbGVzIjpbIm1hbmFnZS1hY2NvdW50IiwibWFuYWdlLWFjY291bnQtbGlua3MiLCJ2aWV3LXByb2ZpbGUiXX19LCJzY29wZSI6Im9wZW5pZCBwcm9maWxlIGVtYWlsIiwiZW1haWxfdmVyaWZpZWQiOiJ0cnVlIiwicHJlZmVycmVkX3VzZXJuYW1lIjoidGVzdEB0ZXN0LmNvLnVrIiwiZW1haWwiOiJ0ZXN0QHRlc3QuY28udWsifQ.Pmlltl1M0Q9EAkU96J_zkPJUjjh2TGhQGzfi0v2J-IrxUt1KTnGEcnEk09TUJjdCuyIgO9YEH-uGj5MihnGj6PqCQjq17lWP5YUjYyjgrULfgM6jZ_659RK31wZdRg_72yiy-BeVd-c-v7UzRtdTXIMkwn_aWEIp7own__jfZV_E_32KfelgtwzljVGHjGXdz_Irg6_2B4lbRn8ipWAn3SDlM9Cj8aJw7q5qq7XPk9KkXclivi4bMQJ9RNgMxtgitFtdINRF1A9_pkbERM1LliAgvW-FTLwmVECAGDQyoE8xDQuti8JgixvM22WfpdznSLd2gWAWMiyYZJwRxzFSVw'; // pragma: allowlist secret
 
-export async function onRequest(context, next) {
+export async function onRequest(context: any, next: MiddlewareNext) { // eslint-disable-line @typescript-eslint/no-explicit-any
   const pathname = new URL(context.request.url).pathname;
 
   // Check if the requested path is public
@@ -33,7 +34,7 @@ export async function onRequest(context, next) {
       return redirectToUnauthorised(context);
     }
 
-    const { email, roles } = await parseAuthToken(token);
+    const { email } = await parseAuthToken(token);
 
     // If the current user doesn't match the user for the session, destroy existing session data - it may be a shared device
     const storedUserEmail = await context.session.get('user-email');
@@ -42,18 +43,13 @@ export async function onRequest(context, next) {
       context.session.set('user-email', email);
     }
 
-    // allow any role (rather than the specific role for gov-ai-client)
-    if (!roles) {
-      return redirectToUnauthorised(context);
-    }
-
     return next();
   } catch(error) {
     console.error('Error authorising token:', error);
     return redirectToUnauthorised(context);
   }
 }
 
-function redirectToUnauthorised(context) {
+function redirectToUnauthorised(context: any) { // eslint-disable-line @typescript-eslint/no-explicit-any
   return context.redirect('/unauthorised');
 }

frontend/src/pages/post-message.ts

@@ -3,6 +3,7 @@ import { getLlmResponse } from '../logic/ai3.ts';
 import type { Message } from '../logic/ai3.ts';
 import { getChat, saveChat } from '../logic/database.ts';
 import { sendMessage } from './api/sse.ts';
+import { parseAuthToken } from '../auth.ts';
 
 export async function POST(context: APIContext) {
 
@@ -27,8 +28,23 @@ export async function POST(context: APIContext) {
     }
   }
 
+  // get user email from JWT
+  const TEST_AUTHORISATION_JWT = 'eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOiIxNzM5ODk2MTk5IiwiaWF0IjoiMTczOTg5NTg5OSIsImF1dGhfdGltZSI6IjE3Mzk4OTM1MjkiLCJqdGkiOiIyYmVmOGI1ZS0yOGY0LTQ2OWQtYWQ2My1lZjJlNDgxNzliODYiLCJpc3MiOiJodHRwczovL2xvY2FsLXRlc3Rpbmcub2JmdXNjYXRlZC50ZXN0LmRvbWFpbi5nb3YudWsvcmVhbG1zL29iZnVzY2F0ZWQiLCJhdWQiOiJhY2NvdW50Iiwic3ViIjoiYmMzNzNkZTQtNDAyMi00NmIyLTgxNTEtZjA0NjEzNzhlOWNiIiwidHlwIjoiQmVhcmVyIiwiYXpwIjoibWludXRlIiwic2lkIjoiYzM2NmE5ZmUtMDNiNC00MjIxLWI0ZWItOTE0MzMzNWFhNjUyIiwiYWNyIjoiMSIsImFsbG93ZWQtb3JpZ2lucyI6WyJodHRwczovL2xvY2FsLXRlc3Rpbmcub2JmdXNjYXRlZC50ZXN0LmRvbWFpbi5nb3YudWsiXSwicmVhbG1fYWNjZXNzIjp7InJvbGVzIjpbImxvY2FsLXRlc3RpbmciXX0sInJlc291cmNlX2FjY2VzcyI6eyJhY2NvdW50Ijp7InJvbGVzIjpbIm1hbmFnZS1hY2NvdW50IiwibWFuYWdlLWFjY291bnQtbGlua3MiLCJ2aWV3LXByb2ZpbGUiXX19LCJzY29wZSI6Im9wZW5pZCBwcm9maWxlIGVtYWlsIiwiZW1haWxfdmVyaWZpZWQiOiJ0cnVlIiwicHJlZmVycmVkX3VzZXJuYW1lIjoidGVzdEB0ZXN0LmNvLnVrIiwiZW1haWwiOiJ0ZXN0QHRlc3QuY28udWsifQ.Pmlltl1M0Q9EAkU96J_zkPJUjjh2TGhQGzfi0v2J-IrxUt1KTnGEcnEk09TUJjdCuyIgO9YEH-uGj5MihnGj6PqCQjq17lWP5YUjYyjgrULfgM6jZ_659RK31wZdRg_72yiy-BeVd-c-v7UzRtdTXIMkwn_aWEIp7own__jfZV_E_32KfelgtwzljVGHjGXdz_Irg6_2B4lbRn8ipWAn3SDlM9Cj8aJw7q5qq7XPk9KkXclivi4bMQJ9RNgMxtgitFtdINRF1A9_pkbERM1LliAgvW-FTLwmVECAGDQyoE8xDQuti8JgixvM22WfpdznSLd2gWAWMiyYZJwRxzFSVw'; // pragma: allowlist secret
+  let oidcDataToken;
+  if (process.env.ENVIRONMENT === 'local') {
+    oidcDataToken = TEST_AUTHORISATION_JWT;
+  } else {
+    oidcDataToken = context.request.headers.get('x-amzn-oidc-data') || '';
+  }
+  const authResult = await parseAuthToken(oidcDataToken);
+  const userEmail = authResult.email;
+
+  if (!userEmail) {
+    console.error('No user email found in token');
+    return new Response('Unauthorized: No user email in token', { status: 401 });
+  }
+
   // add user prompt to session data
-  const userEmail = await context.session?.get('user-email');
   let messages: Message[] | undefined = (await getChat(userEmail, chatId))?.messages;
   if (!messages) {
     messages = [];

terraform/ecs.tf

@@ -1,7 +1,7 @@
 locals {
   frontend_port          = 8081
   additional_policy_arns = { for idx, arn in [aws_iam_policy.ecs_exec_custom_policy.arn] : idx => arn }
-  llm_gateway_name       = var.env == "dev" || var.env == "preprod" ? "llm-gateway.${var.env}" : "llm-gateway"
+  llm_gateway_name       = "llm-gateway"
   llm_gateway_url        = "https://${local.llm_gateway_name}.i.ai.gov.uk"
 }
 

tests/playwright.config.ts

@@ -4,9 +4,9 @@ import { defineConfig, devices } from '@playwright/test';
  * Read environment variables from file.
  * https://github.com/motdotla/dotenv
  */
-// import dotenv from 'dotenv';
-// import path from 'path';
-// dotenv.config({ path: path.resolve(__dirname, '.env') });
+import dotenv from 'dotenv';
+import path from 'path';
+dotenv.config({ path: path.resolve(__dirname, '../.env') });
 
 /**
  * See https://playwright.dev/docs/test-configuration.
@@ -78,17 +78,23 @@ export default defineConfig({
 
   /* Run local dev servers before starting the tests */
   webServer: [
-    {
-      cwd: '../',
-      command: 'make run_frontend',
-      url: 'http://localhost:4321',
-      reuseExistingServer: !process.env.CI,
-    },
     {
       cwd: '../mcp-server-demo',
       command: 'npm run start',
       url: 'http://localhost:3210/sse',
       reuseExistingServer: !process.env.CI,
     },
+    {
+      cwd: '../',
+      command: 'make run_frontend',
+      url: 'http://localhost:4321',
+      reuseExistingServer: !process.env.CI,
+      env: {
+        ...process.env,
+        USE_LITE_LLM: process.env.USE_LITE_LLM || '',
+        LITELLM_GOVAI_CLIENT_OPENAI_API_KEY: process.env.LITELLM_GOVAI_CLIENT_OPENAI_API_KEY || '',
+        LLM_GATEWAY_URL: process.env.LLM_GATEWAY_URL || '',
+      },
+    },
   ]
 });