Multi-user authentication not working when setting up access_tokens and cloud_ids dynamically

[YasiruOps]

I'm currently using MCPServerStreamableHTTP and have overridden it to dynamically set headers (access token and cloud ID) at runtime. However, I'm encountering an issue where the first request consistently fails, while subsequent requests succeed without problems.

For context:
I'm running everything locally as well as Docker.
When I hardcode the access token and cloud ID, the first request works as expected (as per the setup in the original repository).
The issue seems related to the dynamic header setup or initial connection state.
I'll share the code structure below for reference.

class MCPServerHttpWithDynamicHeaders(MCPServerStreamableHTTP):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)

    async def _get_headers(self) -> dict:
        client = AsyncIOMotorClient("")
        db = client[""]
        collection = db[""]

        user_details = await collection.find_one({"email": ""})

        if not user_details:
            raise ValueError("User details not found for email")

        access_token = user_details.get("access_token")
        cloud_id = user_details.get("cloud_id")

        if not access_token:
            raise ValueError("access_token missing for user")
        if not cloud_id:
            raise ValueError("cloud_id missing for user")

        headers = {
            "Authorization": f"Bearer {access_token}",
            "X-Atlassian-Cloud-Id": cloud_id,
        }
        logger.info(f"Generated headers: {headers}")
        return headers

    async def call_tool(self, tool_name: str, params: dict):
        self.headers = await self._get_headers()
        return await super().call_tool(tool_name, params)

jira_server = MCPServerHttpWithDynamicHeaders(url="http://localhost:9000/mcp/")

--tried with trailing slash and without
--this is how i setup the agent

JiraAgent = Agent(
    model=model,
    name="jira_agent",
    system_prompt=system_prompt,
    deps_type=JiraAgentDeps,
    mcp_servers=[jira_server],
    instrument=True,
)

i run this with this JiraAgent.run_mcp_servers(): JiraAgent.run_stream(deps)
let me know if u need more context.. thank you

[LazySeaSeal]

@YasiruOps Hey , any updates pls , Im trying to do the same thing !

[YasiruOps]

@YasiruOps Hey , any updates pls , Im trying to do the same thing !

Hi no luck from my end. Were u able to figure it out by any chance..

[LazySeaSeal]

Same here ... But Im still trying to ... if you want we can jump into a call and maybe get things figured out ?