jawadsaghir12 commited on
Commit
d45f514
Β·
1 Parent(s): 623f57d
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)