glaip-sdk 0.6.15b2__py3-none-any.whl → 0.6.16__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (154) hide show
  1. glaip_sdk/agents/__init__.py +27 -0
  2. glaip_sdk/agents/base.py +1196 -0
  3. glaip_sdk/cli/__init__.py +9 -0
  4. glaip_sdk/cli/account_store.py +540 -0
  5. glaip_sdk/cli/agent_config.py +78 -0
  6. glaip_sdk/cli/auth.py +699 -0
  7. glaip_sdk/cli/commands/__init__.py +5 -0
  8. glaip_sdk/cli/commands/accounts.py +746 -0
  9. glaip_sdk/cli/commands/agents.py +1509 -0
  10. glaip_sdk/cli/commands/common_config.py +104 -0
  11. glaip_sdk/cli/commands/configure.py +896 -0
  12. glaip_sdk/cli/commands/mcps.py +1356 -0
  13. glaip_sdk/cli/commands/models.py +69 -0
  14. glaip_sdk/cli/commands/tools.py +576 -0
  15. glaip_sdk/cli/commands/transcripts.py +755 -0
  16. glaip_sdk/cli/commands/update.py +61 -0
  17. glaip_sdk/cli/config.py +95 -0
  18. glaip_sdk/cli/constants.py +38 -0
  19. glaip_sdk/cli/context.py +150 -0
  20. glaip_sdk/cli/core/__init__.py +79 -0
  21. glaip_sdk/cli/core/context.py +124 -0
  22. glaip_sdk/cli/core/output.py +851 -0
  23. glaip_sdk/cli/core/prompting.py +649 -0
  24. glaip_sdk/cli/core/rendering.py +187 -0
  25. glaip_sdk/cli/display.py +355 -0
  26. glaip_sdk/cli/hints.py +57 -0
  27. glaip_sdk/cli/io.py +112 -0
  28. glaip_sdk/cli/main.py +615 -0
  29. glaip_sdk/cli/masking.py +136 -0
  30. glaip_sdk/cli/mcp_validators.py +287 -0
  31. glaip_sdk/cli/pager.py +266 -0
  32. glaip_sdk/cli/parsers/__init__.py +7 -0
  33. glaip_sdk/cli/parsers/json_input.py +177 -0
  34. glaip_sdk/cli/resolution.py +67 -0
  35. glaip_sdk/cli/rich_helpers.py +27 -0
  36. glaip_sdk/cli/slash/__init__.py +15 -0
  37. glaip_sdk/cli/slash/accounts_controller.py +578 -0
  38. glaip_sdk/cli/slash/accounts_shared.py +75 -0
  39. glaip_sdk/cli/slash/agent_session.py +285 -0
  40. glaip_sdk/cli/slash/prompt.py +256 -0
  41. glaip_sdk/cli/slash/remote_runs_controller.py +566 -0
  42. glaip_sdk/cli/slash/session.py +1708 -0
  43. glaip_sdk/cli/slash/tui/__init__.py +9 -0
  44. glaip_sdk/cli/slash/tui/accounts_app.py +876 -0
  45. glaip_sdk/cli/slash/tui/background_tasks.py +72 -0
  46. glaip_sdk/cli/slash/tui/loading.py +58 -0
  47. glaip_sdk/cli/slash/tui/remote_runs_app.py +628 -0
  48. glaip_sdk/cli/transcript/__init__.py +31 -0
  49. glaip_sdk/cli/transcript/cache.py +536 -0
  50. glaip_sdk/cli/transcript/capture.py +329 -0
  51. glaip_sdk/cli/transcript/export.py +38 -0
  52. glaip_sdk/cli/transcript/history.py +815 -0
  53. glaip_sdk/cli/transcript/launcher.py +77 -0
  54. glaip_sdk/cli/transcript/viewer.py +374 -0
  55. glaip_sdk/cli/update_notifier.py +290 -0
  56. glaip_sdk/cli/utils.py +263 -0
  57. glaip_sdk/cli/validators.py +238 -0
  58. glaip_sdk/client/__init__.py +11 -0
  59. glaip_sdk/client/_agent_payloads.py +520 -0
  60. glaip_sdk/client/agent_runs.py +147 -0
  61. glaip_sdk/client/agents.py +1335 -0
  62. glaip_sdk/client/base.py +502 -0
  63. glaip_sdk/client/main.py +249 -0
  64. glaip_sdk/client/mcps.py +370 -0
  65. glaip_sdk/client/run_rendering.py +700 -0
  66. glaip_sdk/client/shared.py +21 -0
  67. glaip_sdk/client/tools.py +661 -0
  68. glaip_sdk/client/validators.py +198 -0
  69. glaip_sdk/config/constants.py +52 -0
  70. glaip_sdk/mcps/__init__.py +21 -0
  71. glaip_sdk/mcps/base.py +345 -0
  72. glaip_sdk/models/__init__.py +90 -0
  73. glaip_sdk/models/agent.py +47 -0
  74. glaip_sdk/models/agent_runs.py +116 -0
  75. glaip_sdk/models/common.py +42 -0
  76. glaip_sdk/models/mcp.py +33 -0
  77. glaip_sdk/models/tool.py +33 -0
  78. glaip_sdk/payload_schemas/__init__.py +7 -0
  79. glaip_sdk/payload_schemas/agent.py +85 -0
  80. glaip_sdk/registry/__init__.py +55 -0
  81. glaip_sdk/registry/agent.py +164 -0
  82. glaip_sdk/registry/base.py +139 -0
  83. glaip_sdk/registry/mcp.py +253 -0
  84. glaip_sdk/registry/tool.py +232 -0
  85. glaip_sdk/runner/__init__.py +59 -0
  86. glaip_sdk/runner/base.py +84 -0
  87. glaip_sdk/runner/deps.py +112 -0
  88. glaip_sdk/runner/langgraph.py +782 -0
  89. glaip_sdk/runner/mcp_adapter/__init__.py +13 -0
  90. glaip_sdk/runner/mcp_adapter/base_mcp_adapter.py +43 -0
  91. glaip_sdk/runner/mcp_adapter/langchain_mcp_adapter.py +257 -0
  92. glaip_sdk/runner/mcp_adapter/mcp_config_builder.py +95 -0
  93. glaip_sdk/runner/tool_adapter/__init__.py +18 -0
  94. glaip_sdk/runner/tool_adapter/base_tool_adapter.py +44 -0
  95. glaip_sdk/runner/tool_adapter/langchain_tool_adapter.py +219 -0
  96. glaip_sdk/tools/__init__.py +22 -0
  97. glaip_sdk/tools/base.py +435 -0
  98. glaip_sdk/utils/__init__.py +86 -0
  99. glaip_sdk/utils/a2a/__init__.py +34 -0
  100. glaip_sdk/utils/a2a/event_processor.py +188 -0
  101. glaip_sdk/utils/agent_config.py +194 -0
  102. glaip_sdk/utils/bundler.py +267 -0
  103. glaip_sdk/utils/client.py +111 -0
  104. glaip_sdk/utils/client_utils.py +486 -0
  105. glaip_sdk/utils/datetime_helpers.py +58 -0
  106. glaip_sdk/utils/discovery.py +78 -0
  107. glaip_sdk/utils/display.py +135 -0
  108. glaip_sdk/utils/export.py +143 -0
  109. glaip_sdk/utils/general.py +61 -0
  110. glaip_sdk/utils/import_export.py +168 -0
  111. glaip_sdk/utils/import_resolver.py +492 -0
  112. glaip_sdk/utils/instructions.py +101 -0
  113. glaip_sdk/utils/rendering/__init__.py +115 -0
  114. glaip_sdk/utils/rendering/formatting.py +264 -0
  115. glaip_sdk/utils/rendering/layout/__init__.py +64 -0
  116. glaip_sdk/utils/rendering/layout/panels.py +156 -0
  117. glaip_sdk/utils/rendering/layout/progress.py +202 -0
  118. glaip_sdk/utils/rendering/layout/summary.py +74 -0
  119. glaip_sdk/utils/rendering/layout/transcript.py +606 -0
  120. glaip_sdk/utils/rendering/models.py +85 -0
  121. glaip_sdk/utils/rendering/renderer/__init__.py +55 -0
  122. glaip_sdk/utils/rendering/renderer/base.py +1024 -0
  123. glaip_sdk/utils/rendering/renderer/config.py +27 -0
  124. glaip_sdk/utils/rendering/renderer/console.py +55 -0
  125. glaip_sdk/utils/rendering/renderer/debug.py +178 -0
  126. glaip_sdk/utils/rendering/renderer/factory.py +138 -0
  127. glaip_sdk/utils/rendering/renderer/stream.py +202 -0
  128. glaip_sdk/utils/rendering/renderer/summary_window.py +79 -0
  129. glaip_sdk/utils/rendering/renderer/thinking.py +273 -0
  130. glaip_sdk/utils/rendering/renderer/toggle.py +182 -0
  131. glaip_sdk/utils/rendering/renderer/tool_panels.py +442 -0
  132. glaip_sdk/utils/rendering/renderer/transcript_mode.py +162 -0
  133. glaip_sdk/utils/rendering/state.py +204 -0
  134. glaip_sdk/utils/rendering/step_tree_state.py +100 -0
  135. glaip_sdk/utils/rendering/steps/__init__.py +34 -0
  136. glaip_sdk/utils/rendering/steps/event_processor.py +778 -0
  137. glaip_sdk/utils/rendering/steps/format.py +176 -0
  138. glaip_sdk/utils/rendering/steps/manager.py +387 -0
  139. glaip_sdk/utils/rendering/timing.py +36 -0
  140. glaip_sdk/utils/rendering/viewer/__init__.py +21 -0
  141. glaip_sdk/utils/rendering/viewer/presenter.py +184 -0
  142. glaip_sdk/utils/resource_refs.py +195 -0
  143. glaip_sdk/utils/run_renderer.py +41 -0
  144. glaip_sdk/utils/runtime_config.py +425 -0
  145. glaip_sdk/utils/serialization.py +424 -0
  146. glaip_sdk/utils/sync.py +142 -0
  147. glaip_sdk/utils/tool_detection.py +33 -0
  148. glaip_sdk/utils/validation.py +264 -0
  149. {glaip_sdk-0.6.15b2.dist-info → glaip_sdk-0.6.16.dist-info}/METADATA +4 -5
  150. glaip_sdk-0.6.16.dist-info/RECORD +160 -0
  151. glaip_sdk-0.6.15b2.dist-info/RECORD +0 -12
  152. {glaip_sdk-0.6.15b2.dist-info → glaip_sdk-0.6.16.dist-info}/WHEEL +0 -0
  153. {glaip_sdk-0.6.15b2.dist-info → glaip_sdk-0.6.16.dist-info}/entry_points.txt +0 -0
  154. {glaip_sdk-0.6.15b2.dist-info → glaip_sdk-0.6.16.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,13 @@
1
+ """MCP adapter module for local agent runtime.
2
+
3
+ This module provides MCP adapters for converting glaip-sdk MCP references
4
+ to backend-specific MCP configuration formats.
5
+
6
+ Authors:
7
+ Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
8
+ """
9
+
10
+ from glaip_sdk.runner.mcp_adapter.base_mcp_adapter import BaseMCPAdapter
11
+ from glaip_sdk.runner.mcp_adapter.langchain_mcp_adapter import LangChainMCPAdapter
12
+
13
+ __all__ = ["BaseMCPAdapter", "LangChainMCPAdapter"]
@@ -0,0 +1,43 @@
1
+ """Base MCP adapter for local agent runtime.
2
+
3
+ This module defines the abstract base class for MCP adapters.
4
+ Different backends (LangGraph, Google ADK) implement their own adapters.
5
+
6
+ Authors:
7
+ Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
8
+ """
9
+
10
+ from __future__ import annotations
11
+
12
+ from abc import ABC, abstractmethod
13
+ from typing import Any
14
+
15
+
16
+ class BaseMCPAdapter(ABC):
17
+ """Abstract base class for MCP adapters.
18
+
19
+ One Interface, Multiple Implementations:
20
+ - LangChainMCPAdapter: Adapts to aip-agents mcp_config format
21
+ - GoogleADKMCPAdapter: Adapts to Google ADK format (future)
22
+
23
+ Each backend implements this interface to adapt glaip-sdk MCPs
24
+ to their specific MCP configuration format.
25
+ """
26
+
27
+ @abstractmethod
28
+ def adapt_mcps(self, mcp_refs: list[Any]) -> dict[str, Any]:
29
+ """Adapt glaip-sdk MCP references to backend-specific format.
30
+
31
+ Args:
32
+ mcp_refs: List of MCP references from Agent definition.
33
+ Can be: MCP class instances, MCP.from_native() refs, etc.
34
+
35
+ Returns:
36
+ Backend-specific MCP configuration.
37
+ For LangChain/aip-agents: dict[str, dict[str, Any]] mapping server names to config.
38
+ For Google ADK: dict in Google ADK MCP format.
39
+
40
+ Raises:
41
+ ValueError: If MCP is not supported by this backend.
42
+ """
43
+ ...
@@ -0,0 +1,257 @@
1
+ """LangChain MCP adapter for local agent runtime.
2
+
3
+ This module handles adaptation of glaip-sdk MCP references to aip-agents
4
+ MCP configuration format for local execution.
5
+
6
+ Authors:
7
+ Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
8
+ """
9
+
10
+ from typing import Any
11
+
12
+ from glaip_sdk.runner.mcp_adapter.base_mcp_adapter import BaseMCPAdapter
13
+ from glaip_sdk.runner.mcp_adapter.mcp_config_builder import MCPConfigBuilder
14
+ from gllm_core.utils import LoggerManager
15
+
16
+ logger = LoggerManager().get_logger(__name__)
17
+
18
+
19
+ class LangChainMCPAdapter(BaseMCPAdapter):
20
+ """Adapts glaip-sdk MCPs to aip-agents mcp_config dict format.
21
+
22
+ Handles:
23
+ - MCP class with http transport → convert to mcp_config
24
+ - MCP class with sse transport → convert to mcp_config
25
+ - MCP class with stdio transport → convert to mcp_config
26
+
27
+ Rejects:
28
+ - MCP.from_native() → platform-specific
29
+ - String MCP references → platform-specific
30
+ """
31
+
32
+ def adapt_mcps(self, mcp_refs: list[Any]) -> dict[str, Any]:
33
+ """Adapt MCP references to aip-agents mcp_config format.
34
+
35
+ Args:
36
+ mcp_refs: List of MCP references from Agent definition.
37
+
38
+ Returns:
39
+ Dictionary mapping server names to configuration dicts.
40
+ Format: {server_name: {transport: ..., url: ..., ...}}
41
+
42
+ Raises:
43
+ ValueError: If MCP is not supported in local mode.
44
+ """
45
+ mcp_configs = {}
46
+
47
+ for mcp_ref in mcp_refs:
48
+ server_name, config = self._adapt_single_mcp(mcp_ref)
49
+ mcp_configs[server_name] = config
50
+
51
+ logger.debug("Adapted %d MCPs to aip-agents format", len(mcp_configs))
52
+ return mcp_configs
53
+
54
+ def _adapt_single_mcp(self, mcp_ref: Any) -> tuple[str, dict[str, Any]]:
55
+ """Adapt a single MCP reference.
56
+
57
+ Args:
58
+ mcp_ref: Single MCP reference to adapt.
59
+
60
+ Returns:
61
+ Tuple of (server_name, config_dict).
62
+
63
+ Raises:
64
+ ValueError: If MCP is not supported.
65
+ """
66
+ # 1. String references (not supported)
67
+ if isinstance(mcp_ref, str):
68
+ raise ValueError(self._get_platform_mcp_error(mcp_ref))
69
+
70
+ # 2. MCP instance - check if it's local or platform
71
+ if self._is_local_mcp(mcp_ref):
72
+ config = self._convert_mcp_config(mcp_ref)
73
+ return mcp_ref.name, config
74
+
75
+ # 3. Platform MCP (from_native, from_id)
76
+ if self._is_platform_mcp(mcp_ref):
77
+ raise ValueError(self._get_platform_mcp_error(mcp_ref))
78
+
79
+ # 4. Unknown type
80
+ raise ValueError(
81
+ f"Unsupported MCP type for local mode: {type(mcp_ref)}. "
82
+ "Local mode only supports MCP class instances with http/sse/stdio transport."
83
+ )
84
+
85
+ def _is_local_mcp(self, ref: Any) -> bool:
86
+ """Check if ref is a local MCP (has transport config)."""
87
+ return (
88
+ hasattr(ref, "transport")
89
+ and hasattr(ref, "name")
90
+ and getattr(ref, "transport", None) in ("http", "sse", "stdio")
91
+ and not self._is_lookup_only(ref)
92
+ )
93
+
94
+ def _is_lookup_only(self, ref: Any) -> bool:
95
+ """Check if MCP is lookup-only (platform reference)."""
96
+ return hasattr(ref, "_lookup_only") and getattr(ref, "_lookup_only", False)
97
+
98
+ def _convert_mcp_config(self, mcp: Any) -> dict[str, Any]:
99
+ """Convert glaip-sdk MCP to aip-agents mcp_config format.
100
+
101
+ Args:
102
+ mcp: glaip-sdk MCP instance.
103
+
104
+ Returns:
105
+ aip-agents compatible MCP config dict.
106
+ """
107
+ # Start with user-provided config
108
+ config = mcp.config.copy() if mcp.config else {}
109
+
110
+ # Ensure transport is set
111
+ config["transport"] = mcp.transport
112
+
113
+ # Map server_url to url if needed (aip-agents uses 'url')
114
+ if "server_url" in config and "url" not in config:
115
+ config["url"] = config.pop("server_url")
116
+
117
+ self._validate_converted_config(
118
+ mcp_name=mcp.name,
119
+ transport=mcp.transport,
120
+ config=config,
121
+ )
122
+
123
+ # Convert authentication to headers using MCPConfigBuilder
124
+ # Merge with existing headers (auth headers take precedence for conflicts)
125
+ if hasattr(mcp, "authentication") and mcp.authentication:
126
+ auth_headers = MCPConfigBuilder.build_headers_from_auth(mcp.authentication)
127
+ if auth_headers:
128
+ existing_headers = config.get("headers", {})
129
+ config["headers"] = {**existing_headers, **auth_headers}
130
+ else:
131
+ logger.warning("Failed to build headers from authentication for MCP '%s'", mcp.name)
132
+
133
+ logger.debug("Converted MCP '%s' with transport '%s'", mcp.name, mcp.transport)
134
+ return config
135
+
136
+ def _validate_converted_config(self, mcp_name: str, transport: str, config: dict[str, Any]) -> None:
137
+ """Validate converted MCP config matches aip-agents schema expectations.
138
+
139
+ This method performs transport-specific validation after the glaip-sdk MCP
140
+ has been converted into the `aip-agents` `mcp_config` dictionary.
141
+
142
+ Args:
143
+ mcp_name: The MCP server name.
144
+ transport: The MCP transport type.
145
+ config: The converted MCP configuration dictionary.
146
+
147
+ Raises:
148
+ ValueError: If the configuration is invalid for the chosen transport.
149
+ """
150
+ self._validate_transport_config(mcp_name, transport)
151
+ if transport in ("http", "sse"):
152
+ self._validate_http_sse_config(
153
+ mcp_name=mcp_name,
154
+ transport=transport,
155
+ config=config,
156
+ )
157
+ return
158
+ if transport == "stdio":
159
+ self._validate_stdio_config(
160
+ mcp_name=mcp_name,
161
+ config=config,
162
+ )
163
+
164
+ def _validate_transport_config(self, mcp_name: str, transport: str) -> None:
165
+ """Validate that the MCP transport is supported by local mode.
166
+
167
+ Args:
168
+ mcp_name: The MCP server name.
169
+ transport: The MCP transport type.
170
+
171
+ Raises:
172
+ ValueError: If the transport is not one of 'http', 'sse', or 'stdio'.
173
+ """
174
+ if transport not in ("http", "sse", "stdio"):
175
+ raise ValueError(
176
+ f"Invalid MCP config for '{mcp_name}': transport must be one of "
177
+ f"'http', 'sse', or 'stdio'. Got: {transport!r}"
178
+ )
179
+
180
+ def _validate_http_sse_config(self, mcp_name: str, transport: str, config: dict[str, Any]) -> None:
181
+ """Validate http/sse config has a usable URL.
182
+
183
+ Args:
184
+ mcp_name: The MCP server name.
185
+ transport: The MCP transport type ('http' or 'sse').
186
+ config: The converted MCP configuration dictionary.
187
+
188
+ Raises:
189
+ ValueError: If url is missing/empty or does not use http(s) scheme.
190
+ """
191
+ url = config.get("url")
192
+ if not isinstance(url, str) or not url:
193
+ raise ValueError(
194
+ f"Invalid MCP config for '{mcp_name}': transport='{transport}' "
195
+ "requires config['url'] as a non-empty string."
196
+ )
197
+
198
+ if not (url.startswith("http://") or url.startswith("https://")):
199
+ raise ValueError(
200
+ f"Invalid MCP config for '{mcp_name}': config['url'] must start with "
201
+ f"'http://' or 'https://'. Got: {url!r}"
202
+ )
203
+
204
+ def _validate_stdio_config(self, mcp_name: str, config: dict[str, Any]) -> None:
205
+ """Validate stdio config has a usable command and optional args list.
206
+
207
+ Args:
208
+ mcp_name: The MCP server name.
209
+ config: The converted MCP configuration dictionary.
210
+
211
+ Raises:
212
+ ValueError: If command is missing/empty or args is not a list of strings.
213
+ """
214
+ command = config.get("command")
215
+ if not isinstance(command, str) or not command:
216
+ raise ValueError(
217
+ f"Invalid MCP config for '{mcp_name}': transport='stdio' "
218
+ "requires config['command'] as a non-empty string."
219
+ )
220
+
221
+ args = config.get("args")
222
+ if args is not None and (not isinstance(args, list) or any(not isinstance(x, str) for x in args)):
223
+ raise ValueError(
224
+ f"Invalid MCP config for '{mcp_name}': transport='stdio' expects "
225
+ "config['args'] to be a list[str] if provided."
226
+ )
227
+
228
+ def _is_platform_mcp(self, ref: Any) -> bool:
229
+ """Check if ref is platform-specific (not supported locally)."""
230
+ # MCP.from_native() or MCP.from_id() instances
231
+ if self._is_lookup_only(ref):
232
+ return True
233
+
234
+ # MCP with ID but no local transport
235
+ if hasattr(ref, "id") and getattr(ref, "id") and not self._is_local_mcp(ref):
236
+ return True
237
+
238
+ return False
239
+
240
+ def _get_platform_mcp_error(self, ref: Any) -> str:
241
+ """Get error message for platform MCPs."""
242
+ if isinstance(ref, str):
243
+ mcp_name = ref
244
+ else:
245
+ mcp_name = getattr(ref, "name", "<unknown>")
246
+
247
+ return (
248
+ f"MCP '{mcp_name}' is not supported in local mode.\n\n"
249
+ "Local mode only supports MCPs with local transport configurations:\n"
250
+ " - MCP(name='...', transport='http', config={{'url': '...'}})\n"
251
+ " - MCP(name='...', transport='sse', config={{'url': '...'}})\n"
252
+ " - MCP(name='...', transport='stdio', config={{'command': '...'}})\n\n"
253
+ "Alternatives:\n"
254
+ " 1. Configure MCP with a local server URL\n"
255
+ " 2. Deploy the agent to use platform MCPs: agent.deploy()\n"
256
+ " 3. Remove MCP for local testing"
257
+ )
@@ -0,0 +1,95 @@
1
+ """MCP configuration builder helper.
2
+
3
+ This module provides utilities for building MCP configurations,
4
+ particularly for handling authentication conversion to headers.
5
+
6
+ Authors:
7
+ Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
8
+ """
9
+
10
+ from typing import Any
11
+
12
+ from gllm_core.utils import LoggerManager
13
+
14
+ logger = LoggerManager().get_logger(__name__)
15
+
16
+
17
+ class MCPConfigBuilder:
18
+ """Helper class for building MCP configurations.
19
+
20
+ Handles authentication-to-headers conversion and configuration validation.
21
+ Simplified version focused on local MCP support needs.
22
+ """
23
+
24
+ @staticmethod
25
+ def build_headers_from_auth(authentication: dict[str, Any]) -> dict[str, str] | None:
26
+ """Build HTTP headers from authentication configuration.
27
+
28
+ Args:
29
+ authentication: Authentication configuration dict with 'type' and auth-specific fields.
30
+
31
+ Returns:
32
+ dict[str, str] | None: HTTP headers or None if invalid/no-auth.
33
+ """
34
+ if not authentication or "type" not in authentication:
35
+ return None
36
+
37
+ auth_type = str(authentication["type"]).lower()
38
+
39
+ # Dispatch to type-specific handlers
40
+ handlers = {
41
+ "no-auth": MCPConfigBuilder._handle_no_auth,
42
+ "custom-header": MCPConfigBuilder._handle_custom_header,
43
+ "bearer-token": MCPConfigBuilder._handle_bearer_token,
44
+ "api-key": MCPConfigBuilder._handle_api_key,
45
+ }
46
+
47
+ handler = handlers.get(auth_type)
48
+ if handler:
49
+ return handler(authentication)
50
+
51
+ logger.warning("Unsupported authentication type: %s", auth_type)
52
+ return None
53
+
54
+ @staticmethod
55
+ def _handle_no_auth(authentication: dict[str, Any]) -> None: # noqa: ARG004
56
+ """Handle no-auth type."""
57
+ return None
58
+
59
+ @staticmethod
60
+ def _handle_custom_header(authentication: dict[str, Any]) -> dict[str, str] | None:
61
+ """Handle custom-header auth type."""
62
+ headers = authentication.get("headers")
63
+ if isinstance(headers, dict) and all(isinstance(k, str) and isinstance(v, str) for k, v in headers.items()):
64
+ return headers
65
+ logger.warning("custom-header auth requires 'headers' dict with string keys/values")
66
+ return None
67
+
68
+ @staticmethod
69
+ def _handle_bearer_token(authentication: dict[str, Any]) -> dict[str, str] | None:
70
+ """Handle bearer-token auth type."""
71
+ # Check if headers provided directly
72
+ headers = authentication.get("headers")
73
+ if isinstance(headers, dict):
74
+ return headers
75
+ # Otherwise build from token
76
+ token = authentication.get("token")
77
+ if token:
78
+ return {"Authorization": f"Bearer {token}"}
79
+ logger.warning("bearer-token auth requires 'token' field or 'headers' dict")
80
+ return None
81
+
82
+ @staticmethod
83
+ def _handle_api_key(authentication: dict[str, Any]) -> dict[str, str] | None:
84
+ """Handle api-key auth type."""
85
+ # Check if headers provided directly
86
+ headers = authentication.get("headers")
87
+ if isinstance(headers, dict):
88
+ return headers
89
+ # Otherwise build from key/value
90
+ key = authentication.get("key")
91
+ value = authentication.get("value")
92
+ if key and value:
93
+ return {str(key): str(value)}
94
+ logger.warning("api-key auth requires 'key' and 'value' fields or 'headers' dict")
95
+ return None
@@ -0,0 +1,18 @@
1
+ """Tool adapter for local agent runtime.
2
+
3
+ This package provides adapters to convert glaip-sdk tool references
4
+ to backend-specific formats for local execution.
5
+
6
+ Authors:
7
+ Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
8
+ """
9
+
10
+ from glaip_sdk.runner.tool_adapter.base_tool_adapter import BaseToolAdapter
11
+ from glaip_sdk.runner.tool_adapter.langchain_tool_adapter import (
12
+ LangChainToolAdapter,
13
+ )
14
+
15
+ __all__ = [
16
+ "BaseToolAdapter",
17
+ "LangChainToolAdapter",
18
+ ]
@@ -0,0 +1,44 @@
1
+ """Base tool adapter for local agent runtime.
2
+
3
+ This module defines the abstract base class for tool adapters.
4
+ Different backends (LangGraph, Google ADK) implement their own adapters.
5
+
6
+ Authors:
7
+ Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
8
+ """
9
+
10
+ from __future__ import annotations
11
+
12
+ from abc import ABC, abstractmethod
13
+ from typing import Any
14
+
15
+
16
+ class BaseToolAdapter(ABC):
17
+ """Abstract base class for tool adapters.
18
+
19
+ One Interface, Multiple Implementations:
20
+ - LangChainToolAdapter: Adapts to LangChain BaseTool (for aip-agents)
21
+ - GoogleADKToolAdapter: Adapts to Google ADK format (future)
22
+
23
+ Each backend implements this interface to adapt glaip-sdk tools
24
+ to their specific tool format.
25
+ """
26
+
27
+ @abstractmethod
28
+ def adapt_tools(self, tool_refs: list[Any]) -> list[Any]:
29
+ """Adapt glaip-sdk tool references to backend-specific format.
30
+
31
+ Args:
32
+ tool_refs: List of tool references from Agent definition.
33
+ Can be: LangChain classes/instances, Tool.from_langchain(),
34
+ Tool.from_native(), string names, etc.
35
+
36
+ Returns:
37
+ List of tools in backend-specific format.
38
+ For LangChain: list of BaseTool instances.
39
+ For Google ADK: list of Google ADK tool objects.
40
+
41
+ Raises:
42
+ ValueError: If tool is not supported by this backend.
43
+ """
44
+ ...
@@ -0,0 +1,219 @@
1
+ """LangChain tool adapter for local agent runtime.
2
+
3
+ This module handles adaptation of glaip-sdk tool references to LangChain
4
+ BaseTool instances for local execution with aip-agents (LangGraph backend).
5
+
6
+ Authors:
7
+ Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
8
+ """
9
+
10
+ from typing import Any
11
+
12
+ from glaip_sdk.runner.tool_adapter.base_tool_adapter import BaseToolAdapter
13
+ from gllm_core.utils import LoggerManager
14
+
15
+ logger = LoggerManager().get_logger(__name__)
16
+
17
+ # Constant for unknown tool name placeholder
18
+ _UNKNOWN_TOOL_NAME = "<unknown>"
19
+
20
+
21
+ class LangChainToolAdapter(BaseToolAdapter):
22
+ """Adapts glaip-sdk tools to LangChain BaseTool format for aip-agents.
23
+
24
+ Handles:
25
+ - LangChain BaseTool classes → instantiate
26
+ - LangChain BaseTool instances → return as-is
27
+ - Tool.from_langchain() → extract underlying tool
28
+ - @tool_plugin decorator → ignore (just metadata)
29
+
30
+ Rejects:
31
+ - Tool.from_native() → platform-specific
32
+ - String tool names → platform-specific
33
+ """
34
+
35
+ def adapt_tools(self, tool_refs: list[Any]) -> list[Any]:
36
+ """Adapt tool references to LangChain BaseTool instances.
37
+
38
+ Args:
39
+ tool_refs: List of tool references from Agent definition.
40
+
41
+ Returns:
42
+ List of LangChain BaseTool instances.
43
+
44
+ Raises:
45
+ ValueError: If tool is not supported in local mode.
46
+ """
47
+ langchain_tools = []
48
+
49
+ for tool_ref in tool_refs:
50
+ langchain_tool = self._adapt_single_tool(tool_ref)
51
+ langchain_tools.append(langchain_tool)
52
+
53
+ logger.debug("Adapted %d tools to LangChain format", len(langchain_tools))
54
+ return langchain_tools
55
+
56
+ def _adapt_single_tool(self, tool_ref: Any) -> Any:
57
+ """Adapt a single tool reference.
58
+
59
+ Args:
60
+ tool_ref: Single tool reference to adapt.
61
+
62
+ Returns:
63
+ LangChain BaseTool instance.
64
+
65
+ Raises:
66
+ ValueError: If tool is not supported.
67
+ """
68
+ # 1. Tool.from_langchain() wrapper
69
+ if self._is_tool_wrapper(tool_ref):
70
+ return self._extract_from_wrapper(tool_ref)
71
+
72
+ # 2. Direct LangChain BaseTool
73
+ if self._is_langchain_tool(tool_ref):
74
+ return self._instantiate_langchain_tool(tool_ref)
75
+
76
+ # 3. Platform tools (not supported)
77
+ if self._is_platform_tool(tool_ref):
78
+ raise ValueError(self._get_platform_tool_error(tool_ref))
79
+
80
+ # 4. Unknown type
81
+ raise ValueError(
82
+ f"Unsupported tool type for local mode: {type(tool_ref)}. "
83
+ "Local mode only supports LangChain BaseTool classes/instances."
84
+ )
85
+
86
+ def _has_explicit_attr(self, ref: Any, attr: str) -> bool:
87
+ """Check if attribute is explicitly set on the object.
88
+
89
+ This avoids false positives from objects like MagicMock, where hasattr()
90
+ can return True even if the attribute was never set.
91
+ """
92
+ ref_dict = getattr(ref, "__dict__", None)
93
+ return isinstance(ref_dict, dict) and attr in ref_dict
94
+
95
+ def _is_tool_wrapper(self, ref: Any) -> bool:
96
+ """Check if ref is a Tool.from_langchain() wrapper.
97
+
98
+ Args:
99
+ ref: Object to check.
100
+
101
+ Returns:
102
+ True if ref is a Tool.from_langchain() wrapper.
103
+ """
104
+ if self._has_explicit_attr(ref, "langchain_tool") and hasattr(ref, "id") and hasattr(ref, "name"):
105
+ return True
106
+
107
+ if self._has_explicit_attr(ref, "tool_class"):
108
+ return getattr(ref, "tool_class", None) is not None
109
+
110
+ return False
111
+
112
+ def _extract_from_wrapper(self, wrapper: Any) -> Any:
113
+ """Extract underlying LangChain tool from Tool.from_langchain().
114
+
115
+ Args:
116
+ wrapper: Tool.from_langchain() wrapper object.
117
+
118
+ Returns:
119
+ LangChain BaseTool instance.
120
+
121
+ Raises:
122
+ ValueError: If the wrapper's underlying tool is not a valid LangChain tool.
123
+ """
124
+ langchain_tool = getattr(wrapper, "langchain_tool", None)
125
+ if langchain_tool is None:
126
+ langchain_tool = getattr(wrapper, "tool_class", None)
127
+
128
+ # Validate the extracted object is a valid LangChain tool
129
+ if langchain_tool is None:
130
+ wrapper_name = getattr(wrapper, "name", _UNKNOWN_TOOL_NAME)
131
+ raise ValueError(
132
+ f"Tool wrapper '{wrapper_name}' does not contain a valid LangChain tool. "
133
+ "Ensure Tool.from_langchain() was called with a LangChain BaseTool class or instance."
134
+ )
135
+
136
+ # Validate it's actually a LangChain tool (class or instance)
137
+ if not self._is_langchain_tool(langchain_tool):
138
+ wrapper_name = getattr(wrapper, "name", _UNKNOWN_TOOL_NAME)
139
+ raise ValueError(
140
+ f"Tool wrapper '{wrapper_name}' contains an invalid tool type: {type(langchain_tool)}. "
141
+ "Expected a LangChain BaseTool class or instance."
142
+ )
143
+
144
+ # If it's a class, instantiate it
145
+ if isinstance(langchain_tool, type):
146
+ langchain_tool = langchain_tool()
147
+
148
+ logger.debug(
149
+ "Extracted LangChain tool from wrapper: %s",
150
+ getattr(langchain_tool, "name", _UNKNOWN_TOOL_NAME),
151
+ )
152
+ return langchain_tool
153
+
154
+ def _is_langchain_tool(self, ref: Any) -> bool:
155
+ """Check if ref is a LangChain BaseTool class or instance.
156
+
157
+ Args:
158
+ ref: Object to check.
159
+
160
+ Returns:
161
+ True if ref is a LangChain BaseTool.
162
+ """
163
+ from glaip_sdk.utils.tool_detection import is_langchain_tool # noqa: PLC0415
164
+
165
+ return is_langchain_tool(ref)
166
+
167
+ def _instantiate_langchain_tool(self, ref: Any) -> Any:
168
+ """Instantiate LangChain tool if class, return as-is if instance.
169
+
170
+ Args:
171
+ ref: LangChain BaseTool class or instance.
172
+
173
+ Returns:
174
+ LangChain BaseTool instance.
175
+ """
176
+ if isinstance(ref, type):
177
+ # It's a class, instantiate it
178
+ # Note: @tool_plugin decorator doesn't affect instantiation
179
+ return ref()
180
+ return ref
181
+
182
+ def _is_platform_tool(self, ref: Any) -> bool:
183
+ """Check if ref is platform-specific (not supported locally).
184
+
185
+ Args:
186
+ ref: Object to check.
187
+
188
+ Returns:
189
+ True if ref is a platform-specific tool.
190
+ """
191
+ # String tool names
192
+ if isinstance(ref, str):
193
+ return True
194
+
195
+ # Tool.from_native() instances
196
+ if hasattr(ref, "id") and hasattr(ref, "name") and not self._has_explicit_attr(ref, "langchain_tool"):
197
+ tool_class = getattr(ref, "tool_class", None) if self._has_explicit_attr(ref, "tool_class") else None
198
+ if tool_class is None:
199
+ return True
200
+
201
+ return False
202
+
203
+ def _get_platform_tool_error(self, ref: Any) -> str:
204
+ """Get error message for platform tools.
205
+
206
+ Args:
207
+ ref: Platform tool reference.
208
+
209
+ Returns:
210
+ Error message string.
211
+ """
212
+ from glaip_sdk.runner.deps import ( # noqa: PLC0415
213
+ get_local_mode_not_supported_for_tool_message,
214
+ )
215
+
216
+ tool_name = ref if isinstance(ref, str) else getattr(ref, "name", None)
217
+ if tool_name is None:
218
+ tool_name = getattr(getattr(ref, "tool_class", None), "__name__", _UNKNOWN_TOOL_NAME)
219
+ return get_local_mode_not_supported_for_tool_message(tool_name)