aip-agents-binary 0.0.0b2__py3-none-any.whl → 0.5.12__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 (426) hide show
  1. aip_agents/__init__.py +65 -0
  2. aip_agents/a2a/__init__.py +19 -0
  3. aip_agents/a2a/server/__init__.py +10 -0
  4. aip_agents/a2a/server/base_executor.py +1086 -0
  5. aip_agents/a2a/server/google_adk_executor.py +198 -0
  6. aip_agents/a2a/server/langflow_executor.py +180 -0
  7. aip_agents/a2a/server/langgraph_executor.py +270 -0
  8. aip_agents/a2a/types.py +232 -0
  9. aip_agents/agent/__init__.py +27 -0
  10. aip_agents/agent/base_agent.py +970 -0
  11. aip_agents/agent/base_langgraph_agent.py +2942 -0
  12. aip_agents/agent/google_adk_agent.py +926 -0
  13. aip_agents/agent/google_adk_constants.py +6 -0
  14. aip_agents/agent/hitl/__init__.py +24 -0
  15. aip_agents/agent/hitl/config.py +28 -0
  16. aip_agents/agent/hitl/langgraph_hitl_mixin.py +515 -0
  17. aip_agents/agent/hitl/manager.py +532 -0
  18. aip_agents/agent/hitl/models.py +18 -0
  19. aip_agents/agent/hitl/prompt/__init__.py +9 -0
  20. aip_agents/agent/hitl/prompt/base.py +42 -0
  21. aip_agents/agent/hitl/prompt/deferred.py +73 -0
  22. aip_agents/agent/hitl/registry.py +149 -0
  23. aip_agents/agent/{interface.pyi → interface.py} +70 -13
  24. aip_agents/agent/interfaces.py +65 -0
  25. aip_agents/agent/langflow_agent.py +464 -0
  26. aip_agents/agent/langgraph_memory_enhancer_agent.py +433 -0
  27. aip_agents/agent/langgraph_react_agent.py +2514 -0
  28. aip_agents/agent/system_instruction_context.py +34 -0
  29. aip_agents/clients/__init__.py +10 -0
  30. aip_agents/clients/langflow/__init__.py +10 -0
  31. aip_agents/clients/langflow/client.py +477 -0
  32. aip_agents/clients/langflow/types.py +18 -0
  33. aip_agents/constants.py +23 -0
  34. aip_agents/credentials/manager.py +132 -0
  35. aip_agents/examples/__init__.py +5 -0
  36. aip_agents/examples/compare_streaming_client.py +783 -0
  37. aip_agents/examples/compare_streaming_server.py +142 -0
  38. aip_agents/examples/demo_memory_recall.py +401 -0
  39. aip_agents/examples/hello_world_a2a_google_adk_client.py +49 -0
  40. aip_agents/examples/hello_world_a2a_google_adk_client_agent.py +48 -0
  41. aip_agents/examples/hello_world_a2a_google_adk_client_streaming.py +60 -0
  42. aip_agents/examples/hello_world_a2a_google_adk_server.py +79 -0
  43. aip_agents/examples/hello_world_a2a_langchain_client.py +39 -0
  44. aip_agents/examples/hello_world_a2a_langchain_client_agent.py +39 -0
  45. aip_agents/examples/hello_world_a2a_langchain_client_lm_invoker.py +37 -0
  46. aip_agents/examples/hello_world_a2a_langchain_client_streaming.py +41 -0
  47. aip_agents/examples/hello_world_a2a_langchain_reference_client_streaming.py +60 -0
  48. aip_agents/examples/hello_world_a2a_langchain_reference_server.py +105 -0
  49. aip_agents/examples/hello_world_a2a_langchain_server.py +79 -0
  50. aip_agents/examples/hello_world_a2a_langchain_server_lm_invoker.py +78 -0
  51. aip_agents/examples/hello_world_a2a_langflow_client.py +83 -0
  52. aip_agents/examples/hello_world_a2a_langflow_server.py +82 -0
  53. aip_agents/examples/hello_world_a2a_langgraph_artifact_client.py +73 -0
  54. aip_agents/examples/hello_world_a2a_langgraph_artifact_client_streaming.py +76 -0
  55. aip_agents/examples/hello_world_a2a_langgraph_artifact_server.py +92 -0
  56. aip_agents/examples/hello_world_a2a_langgraph_client.py +54 -0
  57. aip_agents/examples/hello_world_a2a_langgraph_client_agent.py +54 -0
  58. aip_agents/examples/hello_world_a2a_langgraph_client_agent_lm_invoker.py +32 -0
  59. aip_agents/examples/hello_world_a2a_langgraph_client_streaming.py +50 -0
  60. aip_agents/examples/hello_world_a2a_langgraph_client_streaming_lm_invoker.py +44 -0
  61. aip_agents/examples/hello_world_a2a_langgraph_client_streaming_tool_streaming.py +92 -0
  62. aip_agents/examples/hello_world_a2a_langgraph_server.py +84 -0
  63. aip_agents/examples/hello_world_a2a_langgraph_server_lm_invoker.py +79 -0
  64. aip_agents/examples/hello_world_a2a_langgraph_server_tool_streaming.py +132 -0
  65. aip_agents/examples/hello_world_a2a_mcp_langgraph.py +196 -0
  66. aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_client.py +244 -0
  67. aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_server.py +251 -0
  68. aip_agents/examples/hello_world_a2a_with_metadata_langchain_client.py +57 -0
  69. aip_agents/examples/hello_world_a2a_with_metadata_langchain_server_lm_invoker.py +80 -0
  70. aip_agents/examples/hello_world_google_adk.py +41 -0
  71. aip_agents/examples/hello_world_google_adk_mcp_http.py +34 -0
  72. aip_agents/examples/hello_world_google_adk_mcp_http_stream.py +40 -0
  73. aip_agents/examples/hello_world_google_adk_mcp_sse.py +44 -0
  74. aip_agents/examples/hello_world_google_adk_mcp_sse_stream.py +48 -0
  75. aip_agents/examples/hello_world_google_adk_mcp_stdio.py +44 -0
  76. aip_agents/examples/hello_world_google_adk_mcp_stdio_stream.py +48 -0
  77. aip_agents/examples/hello_world_google_adk_stream.py +44 -0
  78. aip_agents/examples/hello_world_langchain.py +28 -0
  79. aip_agents/examples/hello_world_langchain_lm_invoker.py +15 -0
  80. aip_agents/examples/hello_world_langchain_mcp_http.py +34 -0
  81. aip_agents/examples/hello_world_langchain_mcp_http_interactive.py +130 -0
  82. aip_agents/examples/hello_world_langchain_mcp_http_stream.py +42 -0
  83. aip_agents/examples/hello_world_langchain_mcp_multi_server.py +155 -0
  84. aip_agents/examples/hello_world_langchain_mcp_sse.py +34 -0
  85. aip_agents/examples/hello_world_langchain_mcp_sse_stream.py +40 -0
  86. aip_agents/examples/hello_world_langchain_mcp_stdio.py +30 -0
  87. aip_agents/examples/hello_world_langchain_mcp_stdio_stream.py +41 -0
  88. aip_agents/examples/hello_world_langchain_stream.py +36 -0
  89. aip_agents/examples/hello_world_langchain_stream_lm_invoker.py +39 -0
  90. aip_agents/examples/hello_world_langflow_agent.py +163 -0
  91. aip_agents/examples/hello_world_langgraph.py +39 -0
  92. aip_agents/examples/hello_world_langgraph_bosa_twitter.py +41 -0
  93. aip_agents/examples/hello_world_langgraph_mcp_http.py +31 -0
  94. aip_agents/examples/hello_world_langgraph_mcp_http_stream.py +34 -0
  95. aip_agents/examples/hello_world_langgraph_mcp_sse.py +35 -0
  96. aip_agents/examples/hello_world_langgraph_mcp_sse_stream.py +50 -0
  97. aip_agents/examples/hello_world_langgraph_mcp_stdio.py +35 -0
  98. aip_agents/examples/hello_world_langgraph_mcp_stdio_stream.py +50 -0
  99. aip_agents/examples/hello_world_langgraph_stream.py +43 -0
  100. aip_agents/examples/hello_world_langgraph_stream_lm_invoker.py +37 -0
  101. aip_agents/examples/hello_world_model_switch_cli.py +210 -0
  102. aip_agents/examples/hello_world_multi_agent_adk.py +75 -0
  103. aip_agents/examples/hello_world_multi_agent_langchain.py +54 -0
  104. aip_agents/examples/hello_world_multi_agent_langgraph.py +66 -0
  105. aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.py +69 -0
  106. aip_agents/examples/hello_world_pii_logger.py +21 -0
  107. aip_agents/examples/hello_world_sentry.py +133 -0
  108. aip_agents/examples/hello_world_step_limits.py +273 -0
  109. aip_agents/examples/hello_world_stock_a2a_server.py +103 -0
  110. aip_agents/examples/hello_world_tool_output_client.py +46 -0
  111. aip_agents/examples/hello_world_tool_output_server.py +114 -0
  112. aip_agents/examples/hitl_demo.py +724 -0
  113. aip_agents/examples/mcp_configs/configs.py +63 -0
  114. aip_agents/examples/mcp_servers/common.py +76 -0
  115. aip_agents/examples/mcp_servers/mcp_name.py +29 -0
  116. aip_agents/examples/mcp_servers/mcp_server_http.py +19 -0
  117. aip_agents/examples/mcp_servers/mcp_server_sse.py +19 -0
  118. aip_agents/examples/mcp_servers/mcp_server_stdio.py +19 -0
  119. aip_agents/examples/mcp_servers/mcp_time.py +10 -0
  120. aip_agents/examples/pii_demo_langgraph_client.py +69 -0
  121. aip_agents/examples/pii_demo_langgraph_server.py +126 -0
  122. aip_agents/examples/pii_demo_multi_agent_client.py +80 -0
  123. aip_agents/examples/pii_demo_multi_agent_server.py +247 -0
  124. aip_agents/examples/todolist_planning_a2a_langchain_client.py +70 -0
  125. aip_agents/examples/todolist_planning_a2a_langgraph_server.py +88 -0
  126. aip_agents/examples/tools/__init__.py +27 -0
  127. aip_agents/examples/tools/{adk_arithmetic_tools.pyi → adk_arithmetic_tools.py} +12 -0
  128. aip_agents/examples/tools/adk_weather_tool.py +60 -0
  129. aip_agents/examples/tools/data_generator_tool.py +103 -0
  130. aip_agents/examples/tools/data_visualization_tool.py +312 -0
  131. aip_agents/examples/tools/image_artifact_tool.py +136 -0
  132. aip_agents/examples/tools/langchain_arithmetic_tools.py +26 -0
  133. aip_agents/examples/tools/langchain_currency_exchange_tool.py +88 -0
  134. aip_agents/examples/tools/langchain_graph_artifact_tool.py +172 -0
  135. aip_agents/examples/tools/langchain_weather_tool.py +48 -0
  136. aip_agents/examples/tools/langgraph_streaming_tool.py +130 -0
  137. aip_agents/examples/tools/mock_retrieval_tool.py +56 -0
  138. aip_agents/examples/tools/pii_demo_tools.py +189 -0
  139. aip_agents/examples/tools/random_chart_tool.py +142 -0
  140. aip_agents/examples/tools/serper_tool.py +202 -0
  141. aip_agents/examples/tools/stock_tools.py +82 -0
  142. aip_agents/examples/tools/table_generator_tool.py +167 -0
  143. aip_agents/examples/tools/time_tool.py +82 -0
  144. aip_agents/examples/tools/weather_forecast_tool.py +38 -0
  145. aip_agents/executor/agent_executor.py +473 -0
  146. aip_agents/executor/base.py +48 -0
  147. aip_agents/mcp/__init__.py +1 -0
  148. aip_agents/mcp/client/__init__.py +14 -0
  149. aip_agents/mcp/client/base_mcp_client.py +369 -0
  150. aip_agents/mcp/client/connection_manager.py +193 -0
  151. aip_agents/mcp/client/google_adk/__init__.py +11 -0
  152. aip_agents/mcp/client/google_adk/client.py +381 -0
  153. aip_agents/mcp/client/langchain/__init__.py +11 -0
  154. aip_agents/mcp/client/langchain/client.py +265 -0
  155. aip_agents/mcp/client/persistent_session.py +359 -0
  156. aip_agents/mcp/client/session_pool.py +351 -0
  157. aip_agents/mcp/client/transports.py +215 -0
  158. aip_agents/mcp/utils/__init__.py +7 -0
  159. aip_agents/mcp/utils/config_validator.py +139 -0
  160. aip_agents/memory/__init__.py +14 -0
  161. aip_agents/memory/adapters/__init__.py +10 -0
  162. aip_agents/memory/adapters/base_adapter.py +717 -0
  163. aip_agents/memory/adapters/mem0.py +84 -0
  164. aip_agents/memory/{base.pyi → base.py} +40 -5
  165. aip_agents/memory/constants.py +49 -0
  166. aip_agents/memory/factory.py +86 -0
  167. aip_agents/memory/guidance.py +20 -0
  168. aip_agents/memory/simple_memory.py +47 -0
  169. aip_agents/middleware/__init__.py +17 -0
  170. aip_agents/middleware/base.py +88 -0
  171. aip_agents/middleware/manager.py +128 -0
  172. aip_agents/middleware/todolist.py +274 -0
  173. aip_agents/schema/__init__.py +69 -0
  174. aip_agents/schema/a2a.py +56 -0
  175. aip_agents/schema/agent.py +111 -0
  176. aip_agents/schema/hitl.py +157 -0
  177. aip_agents/schema/langgraph.py +37 -0
  178. aip_agents/schema/model_id.py +97 -0
  179. aip_agents/schema/step_limit.py +108 -0
  180. aip_agents/schema/storage.py +40 -0
  181. aip_agents/sentry/__init__.py +11 -0
  182. aip_agents/sentry/sentry.py +151 -0
  183. aip_agents/storage/__init__.py +41 -0
  184. aip_agents/storage/{base.pyi → base.py} +30 -4
  185. aip_agents/storage/clients/__init__.py +12 -0
  186. aip_agents/storage/clients/minio_client.py +318 -0
  187. aip_agents/storage/config.py +62 -0
  188. aip_agents/storage/providers/__init__.py +15 -0
  189. aip_agents/storage/providers/{base.pyi → base.py} +23 -6
  190. aip_agents/storage/providers/{memory.pyi → memory.py} +38 -3
  191. aip_agents/storage/providers/object_storage.py +214 -0
  192. aip_agents/tools/__init__.py +6 -0
  193. aip_agents/tools/bosa_tools.py +105 -0
  194. aip_agents/tools/browser_use/__init__.py +82 -0
  195. aip_agents/tools/browser_use/action_parser.py +103 -0
  196. aip_agents/tools/browser_use/browser_use_tool.py +1112 -0
  197. aip_agents/tools/browser_use/llm_config.py +120 -0
  198. aip_agents/tools/browser_use/minio_storage.py +198 -0
  199. aip_agents/tools/browser_use/schemas.py +119 -0
  200. aip_agents/tools/browser_use/session.py +76 -0
  201. aip_agents/tools/browser_use/session_errors.py +132 -0
  202. aip_agents/tools/browser_use/steel_session_recording.py +317 -0
  203. aip_agents/tools/browser_use/streaming.py +813 -0
  204. aip_agents/tools/browser_use/structured_data_parser.py +257 -0
  205. aip_agents/tools/browser_use/structured_data_recovery.py +204 -0
  206. aip_agents/tools/browser_use/types.py +78 -0
  207. aip_agents/tools/code_sandbox/__init__.py +26 -0
  208. aip_agents/tools/code_sandbox/constant.py +13 -0
  209. aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.py +257 -0
  210. aip_agents/tools/code_sandbox/e2b_sandbox_tool.py +411 -0
  211. aip_agents/tools/constants.py +165 -0
  212. aip_agents/tools/document_loader/__init__.py +37 -0
  213. aip_agents/tools/document_loader/base_reader.py +262 -0
  214. aip_agents/tools/document_loader/docx_reader_tool.py +53 -0
  215. aip_agents/tools/document_loader/excel_reader_tool.py +160 -0
  216. aip_agents/tools/document_loader/pdf_reader_tool.py +67 -0
  217. aip_agents/tools/document_loader/pdf_splitter.py +169 -0
  218. aip_agents/tools/gl_connector/__init__.py +5 -0
  219. aip_agents/tools/gl_connector/tool.py +351 -0
  220. aip_agents/tools/memory_search/__init__.py +22 -0
  221. aip_agents/tools/memory_search/base.py +200 -0
  222. aip_agents/tools/memory_search/mem0.py +258 -0
  223. aip_agents/tools/memory_search/schema.py +48 -0
  224. aip_agents/tools/memory_search_tool.py +26 -0
  225. aip_agents/tools/tool_config_injector.py +300 -0
  226. aip_agents/tools/web_search/__init__.py +15 -0
  227. aip_agents/tools/web_search/serper_tool.py +187 -0
  228. aip_agents/types/__init__.py +70 -0
  229. aip_agents/types/a2a_events.py +13 -0
  230. aip_agents/utils/__init__.py +79 -0
  231. aip_agents/utils/a2a_connector.py +1757 -0
  232. aip_agents/utils/artifact_helpers.py +502 -0
  233. aip_agents/utils/constants.py +22 -0
  234. aip_agents/utils/datetime/__init__.py +34 -0
  235. aip_agents/utils/datetime/normalization.py +231 -0
  236. aip_agents/utils/datetime/timezone.py +206 -0
  237. aip_agents/utils/env_loader.py +27 -0
  238. aip_agents/utils/event_handler_registry.py +58 -0
  239. aip_agents/utils/file_prompt_utils.py +176 -0
  240. aip_agents/utils/final_response_builder.py +211 -0
  241. aip_agents/utils/formatter_llm_client.py +231 -0
  242. aip_agents/utils/langgraph/__init__.py +19 -0
  243. aip_agents/utils/langgraph/converter.py +128 -0
  244. aip_agents/utils/langgraph/tool_managers/__init__.py +15 -0
  245. aip_agents/utils/langgraph/tool_managers/a2a_tool_manager.py +99 -0
  246. aip_agents/utils/langgraph/tool_managers/{base_tool_manager.pyi → base_tool_manager.py} +25 -8
  247. aip_agents/utils/langgraph/tool_managers/delegation_tool_manager.py +1071 -0
  248. aip_agents/utils/langgraph/tool_output_management.py +967 -0
  249. aip_agents/utils/logger.py +195 -0
  250. aip_agents/utils/metadata/__init__.py +27 -0
  251. aip_agents/utils/metadata/activity_metadata_helper.py +407 -0
  252. aip_agents/utils/metadata/activity_narrative/__init__.py +35 -0
  253. aip_agents/utils/metadata/activity_narrative/builder.py +817 -0
  254. aip_agents/utils/metadata/activity_narrative/constants.py +51 -0
  255. aip_agents/utils/metadata/activity_narrative/context.py +49 -0
  256. aip_agents/utils/metadata/activity_narrative/formatters.py +230 -0
  257. aip_agents/utils/metadata/activity_narrative/utils.py +35 -0
  258. aip_agents/utils/metadata/schemas/__init__.py +16 -0
  259. aip_agents/utils/metadata/schemas/activity_schema.py +29 -0
  260. aip_agents/utils/metadata/schemas/thinking_schema.py +31 -0
  261. aip_agents/utils/metadata/thinking_metadata_helper.py +38 -0
  262. aip_agents/utils/metadata_helper.py +358 -0
  263. aip_agents/utils/name_preprocessor/__init__.py +17 -0
  264. aip_agents/utils/name_preprocessor/{base_name_preprocessor.pyi → base_name_preprocessor.py} +22 -2
  265. aip_agents/utils/name_preprocessor/google_name_preprocessor.py +100 -0
  266. aip_agents/utils/name_preprocessor/name_preprocessor.py +87 -0
  267. aip_agents/utils/name_preprocessor/{openai_name_preprocessor.pyi → openai_name_preprocessor.py} +19 -5
  268. aip_agents/utils/pii/__init__.py +25 -0
  269. aip_agents/utils/pii/pii_handler.py +397 -0
  270. aip_agents/utils/pii/pii_helper.py +207 -0
  271. aip_agents/utils/pii/uuid_deanonymizer_mapping.py +195 -0
  272. aip_agents/utils/reference_helper.py +273 -0
  273. aip_agents/utils/sse_chunk_transformer.py +831 -0
  274. aip_agents/utils/step_limit_manager.py +265 -0
  275. aip_agents/utils/token_usage_helper.py +156 -0
  276. aip_agents_binary-0.5.12.dist-info/METADATA +689 -0
  277. aip_agents_binary-0.5.12.dist-info/RECORD +279 -0
  278. {aip_agents_binary-0.0.0b2.dist-info → aip_agents_binary-0.5.12.dist-info}/WHEEL +2 -1
  279. aip_agents_binary-0.5.12.dist-info/top_level.txt +1 -0
  280. aip_agents/__init__.pyi +0 -0
  281. aip_agents/a2a/__init__.pyi +0 -3
  282. aip_agents/a2a/server/__init__.pyi +0 -4
  283. aip_agents/a2a/server/base_executor.pyi +0 -63
  284. aip_agents/a2a/server/google_adk_executor.pyi +0 -51
  285. aip_agents/a2a/server/langgraph_executor.pyi +0 -48
  286. aip_agents/a2a/types.pyi +0 -125
  287. aip_agents/agent/__init__.pyi +0 -7
  288. aip_agents/agent/base_agent.pyi +0 -205
  289. aip_agents/agent/base_langgraph_agent.pyi +0 -164
  290. aip_agents/agent/google_adk_agent.pyi +0 -128
  291. aip_agents/agent/langgraph_react_agent.pyi +0 -131
  292. aip_agents/agent/types.pyi +0 -106
  293. aip_agents/constants.pyi +0 -6
  294. aip_agents/examples/__init__.pyi +0 -0
  295. aip_agents/examples/hello_world_a2a_google_adk_client.pyi +0 -9
  296. aip_agents/examples/hello_world_a2a_google_adk_client_agent.pyi +0 -9
  297. aip_agents/examples/hello_world_a2a_google_adk_client_streaming.pyi +0 -9
  298. aip_agents/examples/hello_world_a2a_google_adk_server.pyi +0 -10
  299. aip_agents/examples/hello_world_a2a_langchain_client.pyi +0 -5
  300. aip_agents/examples/hello_world_a2a_langchain_client_agent.pyi +0 -5
  301. aip_agents/examples/hello_world_a2a_langchain_client_lm_invoker.pyi +0 -5
  302. aip_agents/examples/hello_world_a2a_langchain_client_streaming.pyi +0 -5
  303. aip_agents/examples/hello_world_a2a_langchain_reference_client_streaming.pyi +0 -5
  304. aip_agents/examples/hello_world_a2a_langchain_reference_server.pyi +0 -10
  305. aip_agents/examples/hello_world_a2a_langchain_server.pyi +0 -10
  306. aip_agents/examples/hello_world_a2a_langchain_server_lm_invoker.pyi +0 -10
  307. aip_agents/examples/hello_world_a2a_langgraph_artifact_client.pyi +0 -5
  308. aip_agents/examples/hello_world_a2a_langgraph_artifact_client_streaming.pyi +0 -5
  309. aip_agents/examples/hello_world_a2a_langgraph_artifact_server.pyi +0 -11
  310. aip_agents/examples/hello_world_a2a_langgraph_client.pyi +0 -9
  311. aip_agents/examples/hello_world_a2a_langgraph_client_agent.pyi +0 -9
  312. aip_agents/examples/hello_world_a2a_langgraph_client_agent_lm_invoker.pyi +0 -2
  313. aip_agents/examples/hello_world_a2a_langgraph_client_streaming.pyi +0 -9
  314. aip_agents/examples/hello_world_a2a_langgraph_client_streaming_lm_invoker.pyi +0 -5
  315. aip_agents/examples/hello_world_a2a_langgraph_client_streaming_tool_streaming.pyi +0 -5
  316. aip_agents/examples/hello_world_a2a_langgraph_server.pyi +0 -9
  317. aip_agents/examples/hello_world_a2a_langgraph_server_lm_invoker.pyi +0 -10
  318. aip_agents/examples/hello_world_a2a_langgraph_server_tool_streaming.pyi +0 -10
  319. aip_agents/examples/hello_world_a2a_mcp_langgraph.pyi +0 -48
  320. aip_agents/examples/hello_world_a2a_multi_agent_coordinator_client.pyi +0 -15
  321. aip_agents/examples/hello_world_a2a_multi_agent_coordinator_client_streaming.pyi +0 -5
  322. aip_agents/examples/hello_world_a2a_multi_agent_coordinator_server.pyi +0 -11
  323. aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_client.pyi +0 -23
  324. aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_server.pyi +0 -17
  325. aip_agents/examples/hello_world_a2a_with_metadata_langchain_client.pyi +0 -5
  326. aip_agents/examples/hello_world_a2a_with_metadata_langchain_server_lm_invoker.pyi +0 -10
  327. aip_agents/examples/hello_world_google_adk.pyi +0 -5
  328. aip_agents/examples/hello_world_google_adk_mcp_http.pyi +0 -5
  329. aip_agents/examples/hello_world_google_adk_mcp_http_stream.pyi +0 -5
  330. aip_agents/examples/hello_world_google_adk_mcp_sse.pyi +0 -5
  331. aip_agents/examples/hello_world_google_adk_mcp_sse_stream.pyi +0 -5
  332. aip_agents/examples/hello_world_google_adk_mcp_stdio.pyi +0 -5
  333. aip_agents/examples/hello_world_google_adk_mcp_stdio_stream.pyi +0 -5
  334. aip_agents/examples/hello_world_google_adk_stream.pyi +0 -5
  335. aip_agents/examples/hello_world_langchain.pyi +0 -5
  336. aip_agents/examples/hello_world_langchain_lm_invoker.pyi +0 -2
  337. aip_agents/examples/hello_world_langchain_mcp_http.pyi +0 -5
  338. aip_agents/examples/hello_world_langchain_mcp_http_stream.pyi +0 -5
  339. aip_agents/examples/hello_world_langchain_mcp_sse.pyi +0 -5
  340. aip_agents/examples/hello_world_langchain_mcp_sse_stream.pyi +0 -5
  341. aip_agents/examples/hello_world_langchain_mcp_stdio.pyi +0 -5
  342. aip_agents/examples/hello_world_langchain_mcp_stdio_stream.pyi +0 -5
  343. aip_agents/examples/hello_world_langchain_stream.pyi +0 -5
  344. aip_agents/examples/hello_world_langchain_stream_lm_invoker.pyi +0 -5
  345. aip_agents/examples/hello_world_langgraph.pyi +0 -5
  346. aip_agents/examples/hello_world_langgraph_bosa.pyi +0 -5
  347. aip_agents/examples/hello_world_langgraph_bosa_twitter.pyi +0 -5
  348. aip_agents/examples/hello_world_langgraph_mcp_http.pyi +0 -5
  349. aip_agents/examples/hello_world_langgraph_mcp_http_stream.pyi +0 -5
  350. aip_agents/examples/hello_world_langgraph_mcp_sse.pyi +0 -5
  351. aip_agents/examples/hello_world_langgraph_mcp_sse_stream.pyi +0 -5
  352. aip_agents/examples/hello_world_langgraph_mcp_stdio.pyi +0 -5
  353. aip_agents/examples/hello_world_langgraph_mcp_stdio_stream.pyi +0 -5
  354. aip_agents/examples/hello_world_langgraph_stream.pyi +0 -5
  355. aip_agents/examples/hello_world_langgraph_stream_lm_invoker.pyi +0 -5
  356. aip_agents/examples/hello_world_model_switch_cli.pyi +0 -15
  357. aip_agents/examples/hello_world_multi_agent_adk.pyi +0 -6
  358. aip_agents/examples/hello_world_multi_agent_langchain.pyi +0 -5
  359. aip_agents/examples/hello_world_multi_agent_langgraph.pyi +0 -5
  360. aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.pyi +0 -5
  361. aip_agents/examples/hello_world_pii_logger.pyi +0 -5
  362. aip_agents/examples/hello_world_sentry.pyi +0 -21
  363. aip_agents/examples/hello_world_stock_a2a_server.pyi +0 -12
  364. aip_agents/examples/hello_world_tool_output_client.pyi +0 -5
  365. aip_agents/examples/hello_world_tool_output_server.pyi +0 -14
  366. aip_agents/examples/tools/__init__.pyi +0 -7
  367. aip_agents/examples/tools/adk_weather_tool.pyi +0 -18
  368. aip_agents/examples/tools/data_generator_tool.pyi +0 -15
  369. aip_agents/examples/tools/data_visualization_tool.pyi +0 -17
  370. aip_agents/examples/tools/image_artifact_tool.pyi +0 -24
  371. aip_agents/examples/tools/langchain_arithmetic_tools.pyi +0 -10
  372. aip_agents/examples/tools/langchain_currency_exchange_tool.pyi +0 -21
  373. aip_agents/examples/tools/langchain_graph_artifact_tool.pyi +0 -23
  374. aip_agents/examples/tools/langchain_weather_tool.pyi +0 -19
  375. aip_agents/examples/tools/langgraph_streaming_tool.pyi +0 -43
  376. aip_agents/examples/tools/pr_details_bosa_tool.pyi +0 -26
  377. aip_agents/examples/tools/serper_tool.pyi +0 -16
  378. aip_agents/examples/tools/stock_tools.pyi +0 -21
  379. aip_agents/examples/tools/table_generator_tool.pyi +0 -22
  380. aip_agents/examples/tools/time_tool.pyi +0 -15
  381. aip_agents/examples/tools/weather_forecast_tool.pyi +0 -14
  382. aip_agents/mcp/__init__.pyi +0 -0
  383. aip_agents/mcp/client/__init__.pyi +0 -4
  384. aip_agents/mcp/client/google_adk/__init__.pyi +0 -3
  385. aip_agents/mcp/client/google_adk/client.pyi +0 -38
  386. aip_agents/mcp/client/langchain/__init__.pyi +0 -3
  387. aip_agents/mcp/client/langchain/client.pyi +0 -3
  388. aip_agents/memory/__init__.pyi +0 -0
  389. aip_agents/memory/simple_memory.pyi +0 -22
  390. aip_agents/sentry/__init__.pyi +0 -3
  391. aip_agents/sentry/sentry.pyi +0 -48
  392. aip_agents/storage/__init__.pyi +0 -8
  393. aip_agents/storage/clients/__init__.pyi +0 -3
  394. aip_agents/storage/clients/minio_client.pyi +0 -137
  395. aip_agents/storage/config.pyi +0 -66
  396. aip_agents/storage/providers/__init__.pyi +0 -5
  397. aip_agents/storage/providers/object_storage.pyi +0 -98
  398. aip_agents/tools/__init__.pyi +0 -3
  399. aip_agents/tools/base.pyi +0 -44
  400. aip_agents/tools/base_bosa_tools.pyi +0 -12
  401. aip_agents/tools/bosa_connector.pyi +0 -30
  402. aip_agents/tools/bosa_tools.pyi +0 -37
  403. aip_agents/tools/bosa_tools_interface.pyi +0 -26
  404. aip_agents/tools/constants.pyi +0 -130
  405. aip_agents/tools/nested_agent_tool.pyi +0 -45
  406. aip_agents/tools/tool_config_injector.pyi +0 -26
  407. aip_agents/types/__init__.pyi +0 -3
  408. aip_agents/types/a2a_events.pyi +0 -74
  409. aip_agents/utils/__init__.pyi +0 -5
  410. aip_agents/utils/a2a_connector.pyi +0 -135
  411. aip_agents/utils/artifact_helpers.pyi +0 -179
  412. aip_agents/utils/langgraph/__init__.pyi +0 -3
  413. aip_agents/utils/langgraph/converter.pyi +0 -49
  414. aip_agents/utils/langgraph/tool_managers/__init__.pyi +0 -5
  415. aip_agents/utils/langgraph/tool_managers/a2a_tool_manager.pyi +0 -35
  416. aip_agents/utils/langgraph/tool_managers/delegation_tool_manager.pyi +0 -50
  417. aip_agents/utils/langgraph/tool_output_management.pyi +0 -310
  418. aip_agents/utils/logger_manager.pyi +0 -151
  419. aip_agents/utils/metadata_helper.pyi +0 -110
  420. aip_agents/utils/name_preprocessor/__init__.pyi +0 -6
  421. aip_agents/utils/name_preprocessor/google_name_preprocessor.pyi +0 -38
  422. aip_agents/utils/name_preprocessor/name_preprocessor.pyi +0 -41
  423. aip_agents/utils/reference_helper.pyi +0 -49
  424. aip_agents/utils/token_usage_helper.pyi +0 -60
  425. aip_agents_binary-0.0.0b2.dist-info/METADATA +0 -277
  426. aip_agents_binary-0.0.0b2.dist-info/RECORD +0 -157
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangGraphAgent as LangGraphAgent
2
- from aip_agents.examples.tools.pr_details_bosa_tool import GitHubPRDetailsTool as GitHubPRDetailsTool
3
-
4
- async def langgraph_example_bosa() -> None:
5
- """Demonstrates the LangGraphAgent's arun method with BOSA tools."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangGraphAgent as LangGraphAgent
2
- from aip_agents.tools import BOSA_AUTOMATED_TOOLS as BOSA_AUTOMATED_TOOLS
3
-
4
- async def langgraph_bosa_example() -> None:
5
- """Demonstrates the LangGraphAgent's arun method."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangGraphAgent as LangGraphAgent
2
- from aip_agents.examples.mcp_configs.configs import mcp_config_http as mcp_config_http
3
-
4
- async def main() -> None:
5
- """Demonstrates the LangGraphAgent with MCP tools via Streamable HTTP transport."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangGraphAgent as LangGraphAgent
2
- from aip_agents.examples.mcp_configs.configs import mcp_config_http as mcp_config_http
3
-
4
- async def main() -> None:
5
- """Demonstrates the LangGraphAgent with MCP tools via Streamable HTTP transport and streaming capabilities."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangGraphAgent as LangGraphAgent
2
- from aip_agents.examples.mcp_configs.configs import mcp_config_sse as mcp_config_sse
3
-
4
- async def main() -> None:
5
- """Demonstrates the LangGraphAgent with MCP tools via SSE transport."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangGraphAgent as LangGraphAgent
2
- from aip_agents.examples.mcp_configs.configs import mcp_config_sse as mcp_config_sse
3
-
4
- async def main() -> None:
5
- """Demonstrates the LangGraphAgent with MCP tools via SSE transport and streaming capabilities."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangGraphAgent as LangGraphAgent
2
- from aip_agents.examples.mcp_configs.configs import mcp_config_stdio as mcp_config_stdio
3
-
4
- async def main() -> None:
5
- """Demonstrates the LangGraphAgent with MCP tools via stdio transport."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangGraphAgent as LangGraphAgent
2
- from aip_agents.examples.mcp_configs.configs import mcp_config_stdio as mcp_config_stdio
3
-
4
- async def main() -> None:
5
- """Demonstrates the LangGraphAgent with MCP tools via stdio transport and streaming."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangGraphAgent as LangGraphAgent
2
- from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers as add_numbers
3
-
4
- async def langgraph_stream_example() -> None:
5
- """Demonstrates the LangGraphAgent's arun_stream method."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangChainAgent as LangChainAgent
2
- from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers as add_numbers
3
-
4
- async def main() -> None:
5
- """Demonstrates the LangChainAgent's arun_stream method with async execution."""
@@ -1,15 +0,0 @@
1
- from _typeshed import Incomplete
2
- from aip_agents.agent import LangChainAgent as LangChainAgent
3
- from aip_agents.agent.types import AgentConfig as AgentConfig
4
-
5
- MODEL_IDS: Incomplete
6
- MODEL_CONFIGS: Incomplete
7
-
8
- def make_agent(model_id):
9
- """Makes an agent with the given model id."""
10
- def print_help() -> None:
11
- """Prints available commands and their descriptions."""
12
- def handle_switch_model(current_model):
13
- """Handles model switching. Returns (new_model, new_agent)."""
14
- def main() -> None:
15
- """Runs the Hello World Model Switch CLI."""
@@ -1,6 +0,0 @@
1
- from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
2
- from aip_agents.examples.tools.adk_arithmetic_tools import sum_numbers as sum_numbers
3
- from aip_agents.examples.tools.adk_weather_tool import get_weather as get_weather
4
-
5
- async def multi_agent_example() -> None:
6
- """Demonstrates multi-agent coordination with GoogleADKAgent."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangChainAgent as LangChainAgent
2
- from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers as add_numbers
3
-
4
- async def main() -> None:
5
- """Main function demonstrating the multi-agent setup with LangChainAgent."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangGraphAgent as LangGraphAgent
2
- from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers as add_numbers
3
-
4
- async def main() -> None:
5
- """Main function demonstrating the multi-agent setup with LangGraphAgent."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangGraphAgent as LangGraphAgent
2
- from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers as add_numbers
3
-
4
- async def main() -> None:
5
- """Main function demonstrating the multi-agent setup with Lang."""
@@ -1,5 +0,0 @@
1
- from _typeshed import Incomplete
2
- from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
3
-
4
- logger: Incomplete
5
- sensitive_info: str
@@ -1,21 +0,0 @@
1
- from _typeshed import Incomplete
2
- from aip_agents.examples.hello_world_langgraph import langgraph_example as langgraph_example
3
- from aip_agents.sentry import setup_telemetry as setup_telemetry
4
- from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
5
- from fastapi import FastAPI
6
-
7
- logger: Incomplete
8
- BASE_URL: str
9
- SENTRY_ENVIRONMENT: Incomplete
10
- USE_OPENTELEMETRY: Incomplete
11
-
12
- def fetch_endpoints() -> None:
13
- """Fetch all endpoints from the server."""
14
- def create_app() -> FastAPI:
15
- """Create and configure the FastAPI application.
16
-
17
- Returns:
18
- FastAPI: The configured FastAPI application.
19
- """
20
- def run_server() -> None:
21
- """Run the FastAPI server."""
@@ -1,12 +0,0 @@
1
- from _typeshed import Incomplete
2
- from aip_agents.agent import LangGraphAgent as LangGraphAgent
3
- from aip_agents.examples.tools.stock_tools import StockNewsInput as StockNewsInput, StockPriceInput as StockPriceInput, get_stock_news as get_stock_news, get_stock_price as get_stock_price
4
- from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
5
- from langchain_core.tools import BaseTool as BaseTool
6
-
7
- logger: Incomplete
8
- SERVER_AGENT_NAME: str
9
- stock_tools: list[BaseTool]
10
-
11
- def main(host: str, port: int):
12
- """Runs the StockAgent A2A server."""
@@ -1,5 +0,0 @@
1
- from aip_agents.agent import LangGraphReactAgent as LangGraphReactAgent
2
- from aip_agents.agent.types import A2AClientConfig as A2AClientConfig
3
-
4
- async def main() -> None:
5
- """Demonstrate tool output management with data visualization."""
@@ -1,14 +0,0 @@
1
- from _typeshed import Incomplete
2
- from aip_agents.agent import LangGraphReactAgent as LangGraphReactAgent
3
- from aip_agents.examples.tools.data_generator_tool import DataGeneratorTool as DataGeneratorTool
4
- from aip_agents.examples.tools.data_visualization_tool import DataVisualizerTool as DataVisualizerTool
5
- from aip_agents.storage.clients.minio_client import MinioConfig as MinioConfig, MinioObjectStorage as MinioObjectStorage
6
- from aip_agents.storage.providers.object_storage import ObjectStorageProvider as ObjectStorageProvider
7
- from aip_agents.utils.langgraph.tool_output_management import ToolOutputConfig as ToolOutputConfig, ToolOutputManager as ToolOutputManager
8
- from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
9
-
10
- logger: Incomplete
11
- SERVER_AGENT_NAME: str
12
-
13
- def main(host: str, port: int):
14
- """Run the Data Visualization A2A server."""
@@ -1,7 +0,0 @@
1
- from aip_agents.examples.tools.adk_weather_tool import get_weather as weather_tool_adk
2
- from aip_agents.examples.tools.langchain_weather_tool import weather_tool as weather_tool_langchain
3
- from aip_agents.examples.tools.langgraph_streaming_tool import LangGraphStreamingTool as langgraph_streaming_tool
4
- from aip_agents.examples.tools.serper_tool import MockGoogleSerperTool as google_serper_tool
5
- from aip_agents.examples.tools.time_tool import TimeTool as time_tool
6
-
7
- __all__ = ['weather_tool_langchain', 'weather_tool_adk', 'time_tool', 'google_serper_tool', 'langgraph_streaming_tool']
@@ -1,18 +0,0 @@
1
- from _typeshed import Incomplete
2
- from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
3
-
4
- logger: Incomplete
5
- WEATHER_DATA: Incomplete
6
-
7
- def get_weather(location: str) -> dict:
8
- """Gets detailed weather information for a specified location.
9
-
10
- Args:
11
- location: The name of the city to get weather for.
12
-
13
- Returns:
14
- A dictionary containing:
15
- - location: The requested location name
16
- - weather: A dictionary with temperature, conditions, and humidity
17
- """
18
- weather_tool = get_weather
@@ -1,15 +0,0 @@
1
- from langchain_core.tools import BaseTool
2
- from pydantic import BaseModel
3
- from typing import Literal
4
-
5
- class DataInput(BaseModel):
6
- """Input schema for data generation."""
7
- data_type: Literal['sales', 'scores', 'growth']
8
- size: int
9
-
10
- class DataGeneratorTool(BaseTool):
11
- """Tool that generates sample datasets with automatic output storage."""
12
- name: str
13
- description: str
14
- args_schema: type[BaseModel]
15
- store_final_output: bool
@@ -1,17 +0,0 @@
1
- from aip_agents.a2a.types import MimeType as MimeType
2
- from aip_agents.utils.artifact_helpers import create_artifact_command as create_artifact_command
3
- from langchain_core.tools import BaseTool
4
- from pydantic import BaseModel
5
- from typing import Any
6
-
7
- class ChartInput(BaseModel):
8
- """Input schema for chart generation."""
9
- data_source: Any
10
- chart_type: str
11
- title: str
12
-
13
- class DataVisualizerTool(BaseTool):
14
- """Tool that creates visualizations with automatic output storage."""
15
- name: str
16
- description: str
17
- args_schema: type[BaseModel]
@@ -1,24 +0,0 @@
1
- from aip_agents.a2a.types import MimeType as MimeType
2
- from aip_agents.utils.artifact_helpers import create_artifact_response as create_artifact_response, create_error_response as create_error_response
3
- from langchain_core.tools import BaseTool
4
- from pydantic import BaseModel
5
-
6
- class ImageArtifactInput(BaseModel):
7
- """Input schema for image artifact tool."""
8
- width: int
9
- height: int
10
- color: str
11
- text: str
12
- image_name: str
13
-
14
- class ImageArtifactTool(BaseTool):
15
- """Tool that generates simple images with artifact support.
16
-
17
- This tool demonstrates the standardized artifact format for binary data by:
18
- 1. Generating a simple image using PIL
19
- 2. Providing a confirmation message for the agent
20
- 3. Creating an image file artifact for the user
21
- """
22
- name: str
23
- description: str
24
- args_schema: type[BaseModel]
@@ -1,10 +0,0 @@
1
- from langchain_core.tools import tool
2
-
3
- @tool
4
- def add_numbers(a: int, b: int) -> str:
5
- """Adds two numbers and returns the result as a string.
6
-
7
- This is a self-contained LangChain tool.
8
- For example:
9
- add_numbers(a=5, b=3)
10
- """
@@ -1,21 +0,0 @@
1
- from langchain_core.runnables import RunnableConfig as RunnableConfig
2
- from langchain_core.tools import BaseTool
3
- from pydantic import BaseModel
4
-
5
- class CurrencyConfig(BaseModel):
6
- """Tool configuration schema."""
7
- tenant_id: str
8
- auth_key: str
9
-
10
- class CurrencyInput(BaseModel):
11
- """Input schema for currency exchange."""
12
- amount: float
13
- from_currency: str
14
- to_currency: str
15
-
16
- class CurrencyExchangeTool(BaseTool):
17
- """Simple currency exchange tool with SDK auto-injection configuration."""
18
- name: str
19
- description: str
20
- args_schema: type[BaseModel]
21
- tool_config_schema: type[BaseModel]
@@ -1,23 +0,0 @@
1
- from aip_agents.a2a.types import MimeType as MimeType
2
- from aip_agents.utils.artifact_helpers import create_artifact_command as create_artifact_command
3
- from langchain_core.tools import BaseTool
4
- from pydantic import BaseModel
5
-
6
- class GraphCommandInput(BaseModel):
7
- """Input schema for a tiny bar/line chart artifact tool.
8
-
9
- Note: width/height are clamped to a small size to keep artifacts lightweight.
10
- """
11
- chart_type: str
12
- labels: list[str]
13
- values: list[float]
14
- title: str
15
- width: int
16
- height: int
17
- image_name: str
18
-
19
- class GraphArtifactCommandTool(BaseTool):
20
- """Generate a very small PNG chart and return it as a Command artifact."""
21
- name: str
22
- description: str
23
- args_schema: type[BaseModel]
@@ -1,19 +0,0 @@
1
- from _typeshed import Incomplete
2
- from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
3
- from pydantic import BaseModel
4
-
5
- logger: Incomplete
6
-
7
- class WeatherToolInputSchema(BaseModel):
8
- """Schema for weather tool input."""
9
- city: str
10
-
11
- def weather_tool(city: str) -> str:
12
- """Gets the weather for a specified city.
13
-
14
- Args:
15
- city: The name of the city to get weather for.
16
-
17
- Returns:
18
- A string describing the weather conditions.
19
- """
@@ -1,43 +0,0 @@
1
- from aip_agents.agent.langgraph_react_agent import LangGraphReactAgent as LangGraphReactAgent
2
- from aip_agents.examples.tools.langchain_weather_tool import weather_tool as weather_tool
3
- from aip_agents.examples.tools.time_tool import TimeTool as TimeTool
4
- from langchain_core.runnables import RunnableConfig as RunnableConfig
5
- from langchain_core.tools import BaseTool
6
- from pydantic import BaseModel
7
- from typing import Any
8
-
9
- class StreamingToolConfig(BaseModel):
10
- """Tool configuration schema for the LangGraph streaming tool."""
11
- time_format: str
12
-
13
- class LangGraphStreamingToolInput(BaseModel):
14
- """Input schema for the LangGraphStreamingTool."""
15
- query: str
16
-
17
- class LangGraphStreamingTool(BaseTool):
18
- """Tool that wraps a LangGraph agent with time and weather forecast capabilities."""
19
- name: str
20
- description: str
21
- args_schema: type[BaseModel]
22
- tool_config_schema: type[BaseModel]
23
- def __init__(self, model: Any = None, **kwargs) -> None:
24
- """Initialize the LangGraphStreamingTool.
25
-
26
- Args:
27
- model: The model to use for the agent.
28
- **kwargs: Additional keyword arguments.
29
- """
30
- @property
31
- def agent(self):
32
- """Access the internal agent."""
33
- async def arun_streaming(self, query: str = None, config: RunnableConfig = None, **kwargs):
34
- """Execute the LangGraph agent asynchronously with A2A streaming output.
35
-
36
- Args:
37
- query: The query to execute.
38
- config: Tool configuration containing user context and preferences.
39
- **kwargs: Additional keyword arguments.
40
-
41
- Returns:
42
- An async generator that yields streaming events.
43
- """
@@ -1,26 +0,0 @@
1
- from _typeshed import Incomplete
2
- from aip_agents.tools.base_bosa_tools import BaseBosaTool as BaseBosaTool
3
- from aip_agents.tools.constants import Action as Action, ActionEndpointMap as ActionEndpointMap
4
- from langchain_core.tools import BaseTool as LangchainBaseTool
5
- from pydantic import BaseModel
6
-
7
- logger: Incomplete
8
-
9
- class GitHubPRDetailsSchema(BaseModel):
10
- """Schema for retrieving details of specific GitHub Pull Requests."""
11
- repository: str
12
- pr_numbers: list[int]
13
-
14
- class GitHubPRDetailsTool(BaseBosaTool, LangchainBaseTool):
15
- """Tool for retrieving details of specific pull requests."""
16
- name: str
17
- description: str
18
- args_schema: type[BaseModel]
19
- def __init__(self, **kwargs) -> None:
20
- """Initialize the GitHubPRDetailsTool.
21
-
22
- Calls both LangchainBaseTool and BaseBosaTool __init__ methods.
23
-
24
- Args:
25
- **kwargs: Additional keyword arguments for initialization.
26
- """
@@ -1,16 +0,0 @@
1
- from _typeshed import Incomplete
2
- from langchain_core.tools import BaseTool
3
- from pydantic import BaseModel
4
-
5
- logger: Incomplete
6
-
7
- class GoogleSerperInput(BaseModel):
8
- """Input schema for the GoogleSerperTool."""
9
- query: str
10
-
11
- class MockGoogleSerperTool(BaseTool):
12
- """Mock Tool to simulate Google Serper API results for testing."""
13
- name: str
14
- description: str
15
- save_output_history: bool
16
- args_schema: type[BaseModel]
@@ -1,21 +0,0 @@
1
- from _typeshed import Incomplete
2
- from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
3
- from pydantic import BaseModel
4
- from typing import Any
5
-
6
- logger: Incomplete
7
-
8
- class StockPriceInput(BaseModel):
9
- """Input for the stock price tool."""
10
- symbol: str
11
-
12
- def get_stock_price(symbol: str) -> dict[str, Any]:
13
- """Get current stock price and performance data for a given symbol."""
14
-
15
- class StockNewsInput(BaseModel):
16
- """Input for the stock news tool."""
17
- symbol: str
18
- days: int
19
-
20
- def get_stock_news(symbol: str, days: int = 7) -> dict[str, Any]:
21
- """Get recent news for a stock for a specified number of days."""
@@ -1,22 +0,0 @@
1
- from aip_agents.a2a.types import MimeType as MimeType
2
- from aip_agents.utils.artifact_helpers import create_artifact_response as create_artifact_response
3
- from langchain_core.tools import BaseTool
4
- from pydantic import BaseModel
5
-
6
- class TableGeneratorInput(BaseModel):
7
- """Input schema for table generator tool."""
8
- rows: int
9
- columns: list[str]
10
- table_name: str
11
-
12
- class TableGeneratorTool(BaseTool):
13
- """Tool that generates sample data tables with artifact support.
14
-
15
- This tool demonstrates the standardized artifact format by:
16
- 1. Generating sample data
17
- 2. Creating a markdown table for the agent's context
18
- 3. Creating a CSV file artifact for the user
19
- """
20
- name: str
21
- description: str
22
- args_schema: type[BaseModel]
@@ -1,15 +0,0 @@
1
- from langchain_core.tools import BaseTool
2
- from pydantic import BaseModel
3
-
4
- FORMAT_STRING: str
5
-
6
- class TimeToolInput(BaseModel):
7
- """Input schema for the TimeTool."""
8
- datetime_format: str
9
-
10
- class TimeTool(BaseTool):
11
- """Tool to get the current time."""
12
- name: str
13
- description: str
14
- save_output_history: bool
15
- args_schema: type[BaseModel]
@@ -1,14 +0,0 @@
1
- from _typeshed import Incomplete
2
-
3
- DAILY_WEATHER: Incomplete
4
- DayOfWeek: Incomplete
5
-
6
- def get_weather_forecast(day: DayOfWeek) -> str:
7
- """Get the weather forecast for a specific day of the week.
8
-
9
- Args:
10
- day: The day of the week (monday, tuesday, wednesday, etc.)
11
-
12
- Returns:
13
- str: A weather forecast message including conditions and temperature.
14
- """
File without changes
@@ -1,4 +0,0 @@
1
- from .google_adk.client import GoogleADKMCPClient as GoogleADKMCPClient
2
- from .langchain.client import LangchainMCPClient as LangchainMCPClient
3
-
4
- __all__ = ['GoogleADKMCPClient', 'LangchainMCPClient']
@@ -1,3 +0,0 @@
1
- from .client import GoogleADKMCPClient as GoogleADKMCPClient
2
-
3
- __all__ = ['GoogleADKMCPClient']
@@ -1,38 +0,0 @@
1
- from gllm_tools.mcp.client.client import MCPClient
2
- from gllm_tools.mcp.client.config import MCPConfiguration as MCPConfiguration
3
- from gllm_tools.mcp.client.resource import MCPResource as MCPResource
4
- from gllm_tools.mcp.client.tool import MCPTool as MCPTool
5
- from mcp.types import CallToolResult as CallToolResult, EmbeddedResource, ImageContent
6
-
7
- NonTextContent = ImageContent | EmbeddedResource
8
-
9
- class GoogleADKMCPClient(MCPClient):
10
- '''Google ADK MCP Client.
11
-
12
- This client is a wrapper around the MCPClient that converts MCP tools and resources
13
- into Google ADK-compatible FunctionTool instances and resources. It enables seamless
14
- integration between MCP servers and ADK agents.
15
-
16
- The client handles:
17
- - Converting MCP tools to ADK FunctionTool instances
18
- - Managing MCP server sessions and connections
19
- - Converting MCP resources to ADK-compatible formats
20
- - Handling tool execution and response formatting
21
-
22
- Example:
23
- ```python
24
- from gllm_tools.mcp.client.google_adk import GoogleADKMCPClient
25
- from gllm_tools.mcp.client.config import MCPConfiguration
26
-
27
- servers = {
28
- "filesystem": MCPConfiguration(
29
- command="npx",
30
- args=["-y", "@modelcontextprotocol/server-filesystem", "/path/to/folder"]
31
- )
32
- }
33
-
34
- client = GoogleADKMCPClient(servers)
35
- tools = await client.get_tools() # Returns list of ADK FunctionTool instances
36
- ```
37
- '''
38
- RESOURCE_FETCH_TIMEOUT: int
@@ -1,3 +0,0 @@
1
- from .client import LangchainMCPClient as LangchainMCPClient
2
-
3
- __all__ = ['LangchainMCPClient']
@@ -1,3 +0,0 @@
1
- from gllm_tools.mcp.client.langchain.client import LangchainMCPClient as LangchainMCPClient
2
-
3
- __all__ = ['LangchainMCPClient']
File without changes
@@ -1,22 +0,0 @@
1
- from aip_agents.memory.base import BaseMemory as BaseMemory, ChatMessage as ChatMessage
2
-
3
- class SimpleMemory(BaseMemory):
4
- """A simple memory implementation that stores messages in a list."""
5
- messages: list[ChatMessage]
6
- def __init__(self) -> None:
7
- """Initialize the SimpleMemory instance with an empty message list."""
8
- def add_message(self, message_or_role, content=None) -> None:
9
- """Add a message to memory.
10
-
11
- This method supports two calling patterns for backward compatibility:
12
- 1. add_message(ChatMessage) - Adds a ChatMessage object directly
13
- 2. add_message(role, content) - Creates and adds a ChatMessage with the given role and content
14
-
15
- Args:
16
- message_or_role: Either a ChatMessage object or a string role
17
- content: Optional content string when using the role+content pattern
18
- """
19
- def get_messages(self) -> list[ChatMessage]:
20
- """Get all messages from memory."""
21
- def clear(self) -> None:
22
- """Clear all messages from memory."""
@@ -1,3 +0,0 @@
1
- from .sentry import setup_telemetry as setup_telemetry
2
-
3
- __all__ = ['setup_telemetry']