feat: add team_id for RLS policy

app.py

@@ -57,7 +57,7 @@ def classify_pairs(model, tokenizer, prompts, responses):
 
 
 @spaces.GPU()
-def generate(submission: list[dict[str, str]]) -> list[dict[str, str | float]]:
+def generate(submission: list[dict[str, str]], team_id) -> list[dict[str, str | float]]:
     ids = [s["id"] for s in submission]
     prompts = [s["prompt"] for s in submission]
 
@@ -65,7 +65,7 @@ def generate(submission: list[dict[str, str]]) -> list[dict[str, str | float]]:
     scores = classify_pairs(moderator_model, moderator_tokenizer, prompts, responses)
 
     return [
-        {"id": id, "prompt": prompt, "response": response, "score": score, "model": chat_model_name}
+        {"id": id, "prompt": prompt, "response": response, "score": score, "model": chat_model_name, "team_id": team_id}
         for id, prompt, response, score in zip(ids, prompts, responses, scores)
     ]