Spaces:
Sleeping
Sleeping
Commit Β·
d45f514
1
Parent(s): 623f57d
fd
Browse files
src/Agentic_System/__pycache__/google_mcp_config.cpython-311.pyc
ADDED
|
Binary file (12.1 kB). View file
|
|
|
src/Agentic_System/__pycache__/google_mcp_config.cpython-314.pyc
CHANGED
|
Binary files a/src/Agentic_System/__pycache__/google_mcp_config.cpython-314.pyc and b/src/Agentic_System/__pycache__/google_mcp_config.cpython-314.pyc differ
|
|
|
src/Agentic_System/__pycache__/mcp_runtime.cpython-314.pyc
CHANGED
|
Binary files a/src/Agentic_System/__pycache__/mcp_runtime.cpython-314.pyc and b/src/Agentic_System/__pycache__/mcp_runtime.cpython-314.pyc differ
|
|
|
src/Agentic_System/google_mcp_config.py
CHANGED
|
@@ -82,6 +82,14 @@ def get_mcp_headers() -> dict[str, str] | None:
|
|
| 82 |
return headers
|
| 83 |
|
| 84 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 85 |
# βββ Per-service short names (used with --tools flag) ββββββββββββββββββββββββ
|
| 86 |
SERVICE_MAP = {
|
| 87 |
"sheets": "sheets",
|
|
@@ -226,6 +234,10 @@ def create_google_mcp_server(service: str, tool_names: list[str]):
|
|
| 226 |
"""
|
| 227 |
from agents.mcp import MCPServerStreamableHttp
|
| 228 |
from agents.mcp import create_static_tool_filter
|
|
|
|
|
|
|
|
|
|
|
|
|
| 229 |
|
| 230 |
allowed_tool_names = list(dict.fromkeys([*tool_names, AUTH_BOOTSTRAP_TOOL]))
|
| 231 |
tool_filter = create_static_tool_filter(allowed_tool_names=allowed_tool_names)
|
|
@@ -234,6 +246,8 @@ def create_google_mcp_server(service: str, tool_names: list[str]):
|
|
| 234 |
logger.info("[MCP-FACTORY] Creating MCPServerStreamableHttp for service: %s", service)
|
| 235 |
logger.info("[MCP-FACTORY] URL : %s", MCP_SERVER_URL)
|
| 236 |
logger.info("[MCP-FACTORY] Headers set : %s", bool(get_mcp_headers()))
|
|
|
|
|
|
|
| 237 |
logger.info("[MCP-FACTORY] Tool filter : %s", allowed_tool_names)
|
| 238 |
logger.info("=" * 60)
|
| 239 |
|
|
|
|
| 82 |
return headers
|
| 83 |
|
| 84 |
|
| 85 |
+
def _describe_mcp_auth_mode() -> str:
|
| 86 |
+
if MCP_SERVER_BEARER_TOKEN:
|
| 87 |
+
return "bearer-token"
|
| 88 |
+
if MCP_SERVER_API_KEY:
|
| 89 |
+
return "api-key-bearer"
|
| 90 |
+
return "none"
|
| 91 |
+
|
| 92 |
+
|
| 93 |
# βββ Per-service short names (used with --tools flag) ββββββββββββββββββββββββ
|
| 94 |
SERVICE_MAP = {
|
| 95 |
"sheets": "sheets",
|
|
|
|
| 234 |
"""
|
| 235 |
from agents.mcp import MCPServerStreamableHttp
|
| 236 |
from agents.mcp import create_static_tool_filter
|
| 237 |
+
try:
|
| 238 |
+
from .mcp_runtime import get_mcp_start_timeout
|
| 239 |
+
except Exception:
|
| 240 |
+
from mcp_runtime import get_mcp_start_timeout
|
| 241 |
|
| 242 |
allowed_tool_names = list(dict.fromkeys([*tool_names, AUTH_BOOTSTRAP_TOOL]))
|
| 243 |
tool_filter = create_static_tool_filter(allowed_tool_names=allowed_tool_names)
|
|
|
|
| 246 |
logger.info("[MCP-FACTORY] Creating MCPServerStreamableHttp for service: %s", service)
|
| 247 |
logger.info("[MCP-FACTORY] URL : %s", MCP_SERVER_URL)
|
| 248 |
logger.info("[MCP-FACTORY] Headers set : %s", bool(get_mcp_headers()))
|
| 249 |
+
logger.info("[MCP-FACTORY] Auth mode : %s", _describe_mcp_auth_mode())
|
| 250 |
+
logger.info("[MCP-FACTORY] Start timeout (s): %s", get_mcp_start_timeout())
|
| 251 |
logger.info("[MCP-FACTORY] Tool filter : %s", allowed_tool_names)
|
| 252 |
logger.info("=" * 60)
|
| 253 |
|
src/Agentic_System/test/_do_oauth.py
CHANGED
|
@@ -26,8 +26,10 @@ from urllib.parse import parse_qs, urlparse
|
|
| 26 |
|
| 27 |
# Add google-mcp-server to path so we can use its auth modules
|
| 28 |
MCP_SERVER_DIR = os.path.abspath(
|
| 29 |
-
os.path.join(os.path.dirname(__file__), "..", "..", "google-mcp-server")
|
| 30 |
)
|
|
|
|
|
|
|
| 31 |
sys.path.insert(0, MCP_SERVER_DIR)
|
| 32 |
|
| 33 |
from dotenv import load_dotenv
|
|
@@ -194,6 +196,9 @@ async def main():
|
|
| 194 |
flow.fetch_token(code=code)
|
| 195 |
credentials = flow.credentials
|
| 196 |
print(f" Token exchange successful!")
|
|
|
|
|
|
|
|
|
|
| 197 |
|
| 198 |
# Get user info
|
| 199 |
user_info = get_user_info(credentials)
|
|
|
|
| 26 |
|
| 27 |
# Add google-mcp-server to path so we can use its auth modules
|
| 28 |
MCP_SERVER_DIR = os.path.abspath(
|
| 29 |
+
os.path.join(os.path.dirname(__file__), "..", "..", "..", "google-mcp-server")
|
| 30 |
)
|
| 31 |
+
if not os.path.isdir(MCP_SERVER_DIR):
|
| 32 |
+
raise RuntimeError(f"google-mcp-server not found at {MCP_SERVER_DIR}")
|
| 33 |
sys.path.insert(0, MCP_SERVER_DIR)
|
| 34 |
|
| 35 |
from dotenv import load_dotenv
|
|
|
|
| 196 |
flow.fetch_token(code=code)
|
| 197 |
credentials = flow.credentials
|
| 198 |
print(f" Token exchange successful!")
|
| 199 |
+
if os.getenv("PRINT_ACCESS_TOKEN", "false").lower() in {"1", "true", "yes", "on"}:
|
| 200 |
+
print(f" Access token: {credentials.token}")
|
| 201 |
+
print(" NOTE: Treat this as a secret.")
|
| 202 |
|
| 203 |
# Get user info
|
| 204 |
user_info = get_user_info(credentials)
|