va/agent
0
0
Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

Switch to shttp transport

Browse Source
This commit is contained in:
Anibal Angulo
2026-02-25 19:30:58 +00:00
parent 73fb20553d
commit 58d777754f
4 changed files with 29 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/va_agent/agent.py
View File

@@ -4,7 +4,7 @@ from google import genai
from google.adk.agents.llm_agent import Agent
from google.adk.runners import Runner
from google.adk.tools.mcp_tool import McpToolset
from google.adk.tools.mcp_tool.mcp_session_manager import SseConnectionParams
from google.adk.tools.mcp_tool.mcp_session_manager import StreamableHTTPConnectionParams
from google.cloud.firestore_v1.async_client import AsyncClient
from va_agent.auth import auth_headers_provider
@@ -12,7 +12,7 @@ from va_agent.config import settings
from va_agent.session import FirestoreSessionService
toolset = McpToolset(
connection_params=SseConnectionParams(url=settings.mcp_remote_url),
connection_params=StreamableHTTPConnectionParams(url=settings.mcp_remote_url),
header_provider=auth_headers_provider,
)