| 91 | assigned_user_ids: list[str], |
| 92 | ): |
| 93 | async def fake_transform_request( |
| 94 | self, |
| 95 | request, |
| 96 | excluded_tokens=None, |
| 97 | excluded_guest_user_ids=None, |
| 98 | ): |
| 99 | session = await pool.acquire(exclude_user_ids=excluded_guest_user_ids) |
| 100 | assigned_user_ids.append(session.user_id) |
| 101 | return { |
| 102 | "url": f"https://upstream.test/{session.user_id}", |
| 103 | "headers": {"x-guest-user-id": session.user_id}, |
| 104 | "body": {"model": request.model}, |
| 105 | "token": session.token, |
| 106 | "chat_id": f"chat-{session.user_id}", |
| 107 | "model": request.model, |
| 108 | "user_id": session.user_id, |
| 109 | "auth_mode": "guest", |
| 110 | "token_source": "guest_pool", |
| 111 | "guest_user_id": session.user_id, |
| 112 | } |
| 113 | |
| 114 | async def fake_transform_response(self, response, request, transformed): |
| 115 | return { |