Adding some bug fixes for MCP agent (#85)

* Adding script execution in the testcase

* Enhance MCP client configuration by adding support for custom headers… (#84)

Enhance MCP client configuration by adding support for custom headers in SSE and streamable HTTP transports.

---------

Co-authored-by: Shahal Thayyil <148101954+Shahalt1@users.noreply.github.com>

Author: shriyanshagnihotri

testzeus_hercules/utils/mcp_helper.py

@@ -254,8 +254,8 @@ async def create_toolkit_and_store(session: ClientSession) -> bool:
                     raise ValueError(
                         f"SSE transport requires 'url' field in server config for '{server_name}'"
                     )
-
-                client_cm = sse_client(url=url)
+                headers = server_config.get("headers", None)
+                client_cm = sse_client(url=url, headers=headers)
                 read_stream, write_stream = await getattr(client_cm, "__aenter__")()
                 session = ClientSession(
                     read_stream, write_stream, read_timeout_seconds=timedelta(seconds=timeout_seconds)
@@ -270,8 +270,8 @@ async def create_toolkit_and_store(session: ClientSession) -> bool:
                     raise ValueError(
                         f"Streamable HTTP transport requires 'url' field in server config for '{server_name}'"
                     )
-
-                client_cm = streamablehttp_client(url)
+                headers = server_config.get("headers", None)
+                client_cm = streamablehttp_client(url, headers)
                 read_stream, write_stream, _ = await getattr(client_cm, "__aenter__")()
                 session = ClientSession(
                     read_stream, write_stream, read_timeout_seconds=timedelta(seconds=timeout_seconds)