From 7b4c23ac180cf2078ca3f82e52a25a5784e765f4 Mon Sep 17 00:00:00 2001
From: linearcombination <4829djaskdfj@gmail.com>
Date: Fri, 5 Jun 2026 10:05:29 -0700
Subject: [PATCH] Limit STET production languages

Limit STET production languages to the intersection of en, es-419,
pt-br and STET languages whose input document provides a 4th column.
---
 backend/stet/entrypoints/routes.py  | 13 ++++++++++---
 frontend/tests/e2e/passages_test.ts |  2 +-
 frontend/tests/e2e/stet_test.ts     |  2 +-
 3 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/backend/stet/entrypoints/routes.py b/backend/stet/entrypoints/routes.py
index 5c0f155c..ce0501d9 100644
--- a/backend/stet/entrypoints/routes.py
+++ b/backend/stet/entrypoints/routes.py
@@ -10,7 +10,6 @@
 from fastapi.responses import JSONResponse
 from stet.domain import document_generator, model
 
-
 router = APIRouter()
 
 logger = settings.logger(__name__)
@@ -20,10 +19,15 @@
 async def source_lang_codes_and_names(
     request: Request,
     stet_dir: str = settings.STET_DIR,
+    production_lang_codes: Sequence[str] = ["en", "es-419", "pt-br"],
 ) -> Sequence[tuple[str, str, bool]]:
     """
-    Return list of all available language code, name tuples for which Translation Services has provided a source document.
+    Return list of all available language code, name tuples for which
+    Translation Services has provided a source document unless the
+    request comes from production server in which case limit the
+    languages according to special rules.
     """
+    logger.debug("headers: %s", dict(request.headers))
     is_production = request.headers.get("x-is-production") == "true"
     logger.debug("is_production: %s", is_production)
     # Scan what source docs are available and make sure to filter
@@ -48,7 +52,10 @@ async def source_lang_codes_and_names(
     languages = []
     for lang_code_and_name in resource_lookup.lang_codes_and_names_having_usfm():
         if is_production:
-            if lang_code_and_name[0] in ietf_codes_for_docs_with_fourth_column:
+            if (
+                lang_code_and_name[0] in production_lang_codes
+                and lang_code_and_name[0] in ietf_codes_for_docs_with_fourth_column
+            ):
                 languages.append(lang_code_and_name)
         else:
             if lang_code_and_name[0] in ietf_codes:
diff --git a/frontend/tests/e2e/passages_test.ts b/frontend/tests/e2e/passages_test.ts
index 4025df8e..e9a6b3db 100644
--- a/frontend/tests/e2e/passages_test.ts
+++ b/frontend/tests/e2e/passages_test.ts
@@ -1,6 +1,6 @@
 import { test, expect } from '@playwright/test'
 
-test.only('add passages', async ({ page }) => {
+test.skip('add passages', async ({ page }) => {
   await page.goto('http://localhost:8001/passages')
   await page.getByText('Español Latin America (Latin').click()
   await page.getByRole('button', { name: 'Next' }).click()
diff --git a/frontend/tests/e2e/stet_test.ts b/frontend/tests/e2e/stet_test.ts
index b63c8b1e..ea8c0818 100644
--- a/frontend/tests/e2e/stet_test.ts
+++ b/frontend/tests/e2e/stet_test.ts
@@ -97,7 +97,7 @@ test.describe('Desktop Tests', () => {
     await expect(page.getByText('Tok Pisin')).toBeVisible({ timeout: 64_000 })
   })
 
-  test('stet input docs available in production should be limited to those with 4th column', async ({
+  test('stet input docs available in production should be limited to those with 4th column and in [en, es-419, pt-br]', async ({
     page
   }) => {
     // log every request to the backend endpoint