aip-agents-binary 0.0.0b2__py3-none-any.whl → 0.5.11__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 (506) hide show
  1. aip_agents/__init__.py +65 -0
  2. aip_agents/__init__.pyi +19 -0
  3. aip_agents/a2a/__init__.py +19 -0
  4. aip_agents/a2a/__init__.pyi +1 -1
  5. aip_agents/a2a/server/__init__.py +10 -0
  6. aip_agents/a2a/server/base_executor.py +1086 -0
  7. aip_agents/a2a/server/base_executor.pyi +16 -6
  8. aip_agents/a2a/server/google_adk_executor.py +198 -0
  9. aip_agents/a2a/server/google_adk_executor.pyi +11 -11
  10. aip_agents/a2a/server/langflow_executor.py +180 -0
  11. aip_agents/a2a/server/langflow_executor.pyi +43 -0
  12. aip_agents/a2a/server/langgraph_executor.py +267 -0
  13. aip_agents/a2a/server/langgraph_executor.pyi +11 -12
  14. aip_agents/a2a/types.py +232 -0
  15. aip_agents/a2a/types.pyi +54 -47
  16. aip_agents/agent/__init__.py +27 -0
  17. aip_agents/agent/__init__.pyi +3 -1
  18. aip_agents/agent/base_agent.py +970 -0
  19. aip_agents/agent/base_agent.pyi +24 -8
  20. aip_agents/agent/base_langgraph_agent.py +2942 -0
  21. aip_agents/agent/base_langgraph_agent.pyi +93 -25
  22. aip_agents/agent/google_adk_agent.py +926 -0
  23. aip_agents/agent/google_adk_agent.pyi +34 -21
  24. aip_agents/agent/google_adk_constants.py +6 -0
  25. aip_agents/agent/google_adk_constants.pyi +3 -0
  26. aip_agents/agent/hitl/__init__.py +24 -0
  27. aip_agents/agent/hitl/__init__.pyi +6 -0
  28. aip_agents/agent/hitl/config.py +28 -0
  29. aip_agents/agent/hitl/config.pyi +15 -0
  30. aip_agents/agent/hitl/langgraph_hitl_mixin.py +515 -0
  31. aip_agents/agent/hitl/langgraph_hitl_mixin.pyi +42 -0
  32. aip_agents/agent/hitl/manager.py +522 -0
  33. aip_agents/agent/hitl/manager.pyi +199 -0
  34. aip_agents/agent/hitl/models.py +18 -0
  35. aip_agents/agent/hitl/models.pyi +3 -0
  36. aip_agents/agent/hitl/prompt/__init__.py +9 -0
  37. aip_agents/agent/hitl/prompt/__init__.pyi +4 -0
  38. aip_agents/agent/hitl/prompt/base.py +42 -0
  39. aip_agents/agent/hitl/prompt/base.pyi +24 -0
  40. aip_agents/agent/hitl/prompt/deferred.py +73 -0
  41. aip_agents/agent/hitl/prompt/deferred.pyi +30 -0
  42. aip_agents/agent/interface.py +138 -0
  43. aip_agents/agent/interface.pyi +5 -5
  44. aip_agents/agent/interfaces.py +65 -0
  45. aip_agents/agent/interfaces.pyi +44 -0
  46. aip_agents/agent/langflow_agent.py +464 -0
  47. aip_agents/agent/langflow_agent.pyi +133 -0
  48. aip_agents/agent/langgraph_memory_enhancer_agent.py +433 -0
  49. aip_agents/agent/langgraph_memory_enhancer_agent.pyi +49 -0
  50. aip_agents/agent/langgraph_react_agent.py +2514 -0
  51. aip_agents/agent/langgraph_react_agent.pyi +51 -56
  52. aip_agents/agent/system_instruction_context.py +34 -0
  53. aip_agents/agent/system_instruction_context.pyi +13 -0
  54. aip_agents/clients/__init__.py +10 -0
  55. aip_agents/clients/__init__.pyi +4 -0
  56. aip_agents/clients/langflow/__init__.py +10 -0
  57. aip_agents/clients/langflow/__init__.pyi +4 -0
  58. aip_agents/clients/langflow/client.py +477 -0
  59. aip_agents/clients/langflow/client.pyi +140 -0
  60. aip_agents/clients/langflow/types.py +18 -0
  61. aip_agents/clients/langflow/types.pyi +7 -0
  62. aip_agents/constants.py +23 -0
  63. aip_agents/constants.pyi +1 -0
  64. aip_agents/credentials/manager.py +132 -0
  65. aip_agents/examples/__init__.py +5 -0
  66. aip_agents/examples/compare_streaming_client.py +783 -0
  67. aip_agents/examples/compare_streaming_client.pyi +48 -0
  68. aip_agents/examples/compare_streaming_server.py +142 -0
  69. aip_agents/examples/compare_streaming_server.pyi +18 -0
  70. aip_agents/examples/demo_memory_recall.py +401 -0
  71. aip_agents/examples/demo_memory_recall.pyi +58 -0
  72. aip_agents/examples/hello_world_a2a_google_adk_client.py +49 -0
  73. aip_agents/examples/hello_world_a2a_google_adk_client.pyi +2 -2
  74. aip_agents/examples/hello_world_a2a_google_adk_client_agent.py +48 -0
  75. aip_agents/examples/hello_world_a2a_google_adk_client_agent.pyi +2 -2
  76. aip_agents/examples/hello_world_a2a_google_adk_client_streaming.py +60 -0
  77. aip_agents/examples/hello_world_a2a_google_adk_client_streaming.pyi +2 -2
  78. aip_agents/examples/hello_world_a2a_google_adk_server.py +79 -0
  79. aip_agents/examples/hello_world_a2a_google_adk_server.pyi +7 -2
  80. aip_agents/examples/hello_world_a2a_langchain_client.py +39 -0
  81. aip_agents/examples/hello_world_a2a_langchain_client.pyi +1 -1
  82. aip_agents/examples/hello_world_a2a_langchain_client_agent.py +39 -0
  83. aip_agents/examples/hello_world_a2a_langchain_client_agent.pyi +1 -1
  84. aip_agents/examples/hello_world_a2a_langchain_client_lm_invoker.py +37 -0
  85. aip_agents/examples/hello_world_a2a_langchain_client_lm_invoker.pyi +1 -1
  86. aip_agents/examples/hello_world_a2a_langchain_client_streaming.py +41 -0
  87. aip_agents/examples/hello_world_a2a_langchain_client_streaming.pyi +1 -1
  88. aip_agents/examples/hello_world_a2a_langchain_reference_client_streaming.py +60 -0
  89. aip_agents/examples/hello_world_a2a_langchain_reference_client_streaming.pyi +1 -1
  90. aip_agents/examples/hello_world_a2a_langchain_reference_server.py +105 -0
  91. aip_agents/examples/hello_world_a2a_langchain_reference_server.pyi +8 -3
  92. aip_agents/examples/hello_world_a2a_langchain_server.py +79 -0
  93. aip_agents/examples/hello_world_a2a_langchain_server.pyi +7 -2
  94. aip_agents/examples/hello_world_a2a_langchain_server_lm_invoker.py +78 -0
  95. aip_agents/examples/hello_world_a2a_langchain_server_lm_invoker.pyi +7 -2
  96. aip_agents/examples/hello_world_a2a_langflow_client.py +83 -0
  97. aip_agents/examples/hello_world_a2a_langflow_client.pyi +9 -0
  98. aip_agents/examples/hello_world_a2a_langflow_server.py +82 -0
  99. aip_agents/examples/hello_world_a2a_langflow_server.pyi +14 -0
  100. aip_agents/examples/hello_world_a2a_langgraph_artifact_client.py +73 -0
  101. aip_agents/examples/hello_world_a2a_langgraph_artifact_client.pyi +1 -1
  102. aip_agents/examples/hello_world_a2a_langgraph_artifact_client_streaming.py +76 -0
  103. aip_agents/examples/hello_world_a2a_langgraph_artifact_client_streaming.pyi +1 -1
  104. aip_agents/examples/hello_world_a2a_langgraph_artifact_server.py +92 -0
  105. aip_agents/examples/hello_world_a2a_langgraph_artifact_server.pyi +7 -2
  106. aip_agents/examples/hello_world_a2a_langgraph_client.py +54 -0
  107. aip_agents/examples/hello_world_a2a_langgraph_client.pyi +2 -2
  108. aip_agents/examples/hello_world_a2a_langgraph_client_agent.py +54 -0
  109. aip_agents/examples/hello_world_a2a_langgraph_client_agent.pyi +2 -2
  110. aip_agents/examples/hello_world_a2a_langgraph_client_agent_lm_invoker.py +32 -0
  111. aip_agents/examples/hello_world_a2a_langgraph_client_agent_lm_invoker.pyi +1 -1
  112. aip_agents/examples/hello_world_a2a_langgraph_client_streaming.py +50 -0
  113. aip_agents/examples/hello_world_a2a_langgraph_client_streaming.pyi +3 -3
  114. aip_agents/examples/hello_world_a2a_langgraph_client_streaming_lm_invoker.py +44 -0
  115. aip_agents/examples/hello_world_a2a_langgraph_client_streaming_lm_invoker.pyi +1 -1
  116. aip_agents/examples/hello_world_a2a_langgraph_client_streaming_tool_streaming.py +92 -0
  117. aip_agents/examples/hello_world_a2a_langgraph_client_streaming_tool_streaming.pyi +1 -1
  118. aip_agents/examples/hello_world_a2a_langgraph_server.py +84 -0
  119. aip_agents/examples/hello_world_a2a_langgraph_server.pyi +7 -2
  120. aip_agents/examples/hello_world_a2a_langgraph_server_lm_invoker.py +79 -0
  121. aip_agents/examples/hello_world_a2a_langgraph_server_lm_invoker.pyi +7 -2
  122. aip_agents/examples/hello_world_a2a_langgraph_server_tool_streaming.py +132 -0
  123. aip_agents/examples/hello_world_a2a_langgraph_server_tool_streaming.pyi +7 -2
  124. aip_agents/examples/hello_world_a2a_mcp_langgraph.py +196 -0
  125. aip_agents/examples/hello_world_a2a_mcp_langgraph.pyi +1 -1
  126. aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_client.py +244 -0
  127. aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_client.pyi +31 -6
  128. aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_server.py +251 -0
  129. aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_server.pyi +33 -5
  130. aip_agents/examples/hello_world_a2a_with_metadata_langchain_client.py +57 -0
  131. aip_agents/examples/hello_world_a2a_with_metadata_langchain_client.pyi +1 -1
  132. aip_agents/examples/hello_world_a2a_with_metadata_langchain_server_lm_invoker.py +80 -0
  133. aip_agents/examples/hello_world_a2a_with_metadata_langchain_server_lm_invoker.pyi +7 -2
  134. aip_agents/examples/hello_world_google_adk.py +41 -0
  135. aip_agents/examples/hello_world_google_adk_mcp_http.py +34 -0
  136. aip_agents/examples/hello_world_google_adk_mcp_http_stream.py +40 -0
  137. aip_agents/examples/hello_world_google_adk_mcp_sse.py +44 -0
  138. aip_agents/examples/hello_world_google_adk_mcp_sse_stream.py +48 -0
  139. aip_agents/examples/hello_world_google_adk_mcp_stdio.py +44 -0
  140. aip_agents/examples/hello_world_google_adk_mcp_stdio_stream.py +48 -0
  141. aip_agents/examples/hello_world_google_adk_stream.py +44 -0
  142. aip_agents/examples/hello_world_langchain.py +28 -0
  143. aip_agents/examples/hello_world_langchain_lm_invoker.py +15 -0
  144. aip_agents/examples/hello_world_langchain_mcp_http.py +34 -0
  145. aip_agents/examples/hello_world_langchain_mcp_http_interactive.py +130 -0
  146. aip_agents/examples/hello_world_langchain_mcp_http_interactive.pyi +16 -0
  147. aip_agents/examples/hello_world_langchain_mcp_http_stream.py +42 -0
  148. aip_agents/examples/hello_world_langchain_mcp_multi_server.py +155 -0
  149. aip_agents/examples/hello_world_langchain_mcp_multi_server.pyi +18 -0
  150. aip_agents/examples/hello_world_langchain_mcp_sse.py +34 -0
  151. aip_agents/examples/hello_world_langchain_mcp_sse_stream.py +40 -0
  152. aip_agents/examples/hello_world_langchain_mcp_stdio.py +30 -0
  153. aip_agents/examples/hello_world_langchain_mcp_stdio_stream.py +41 -0
  154. aip_agents/examples/hello_world_langchain_stream.py +36 -0
  155. aip_agents/examples/hello_world_langchain_stream_lm_invoker.py +39 -0
  156. aip_agents/examples/hello_world_langflow_agent.py +163 -0
  157. aip_agents/examples/hello_world_langflow_agent.pyi +35 -0
  158. aip_agents/examples/hello_world_langgraph.py +39 -0
  159. aip_agents/examples/hello_world_langgraph_bosa_twitter.py +41 -0
  160. aip_agents/examples/hello_world_langgraph_mcp_http.py +31 -0
  161. aip_agents/examples/hello_world_langgraph_mcp_http_stream.py +34 -0
  162. aip_agents/examples/hello_world_langgraph_mcp_sse.py +35 -0
  163. aip_agents/examples/hello_world_langgraph_mcp_sse_stream.py +50 -0
  164. aip_agents/examples/hello_world_langgraph_mcp_stdio.py +35 -0
  165. aip_agents/examples/hello_world_langgraph_mcp_stdio_stream.py +50 -0
  166. aip_agents/examples/hello_world_langgraph_stream.py +43 -0
  167. aip_agents/examples/hello_world_langgraph_stream_lm_invoker.py +37 -0
  168. aip_agents/examples/hello_world_model_switch_cli.py +210 -0
  169. aip_agents/examples/hello_world_model_switch_cli.pyi +18 -3
  170. aip_agents/examples/hello_world_multi_agent_adk.py +75 -0
  171. aip_agents/examples/hello_world_multi_agent_langchain.py +54 -0
  172. aip_agents/examples/hello_world_multi_agent_langgraph.py +66 -0
  173. aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.py +69 -0
  174. aip_agents/examples/hello_world_pii_logger.py +21 -0
  175. aip_agents/examples/hello_world_pii_logger.pyi +1 -1
  176. aip_agents/examples/hello_world_sentry.py +133 -0
  177. aip_agents/examples/hello_world_sentry.pyi +1 -1
  178. aip_agents/examples/hello_world_step_limits.py +273 -0
  179. aip_agents/examples/hello_world_step_limits.pyi +17 -0
  180. aip_agents/examples/hello_world_stock_a2a_server.py +103 -0
  181. aip_agents/examples/hello_world_stock_a2a_server.pyi +7 -2
  182. aip_agents/examples/hello_world_tool_output_client.py +46 -0
  183. aip_agents/examples/hello_world_tool_output_client.pyi +1 -1
  184. aip_agents/examples/hello_world_tool_output_server.py +114 -0
  185. aip_agents/examples/hello_world_tool_output_server.pyi +7 -2
  186. aip_agents/examples/hitl_demo.py +724 -0
  187. aip_agents/examples/hitl_demo.pyi +67 -0
  188. aip_agents/examples/mcp_configs/configs.py +63 -0
  189. aip_agents/examples/mcp_servers/common.py +76 -0
  190. aip_agents/examples/mcp_servers/mcp_name.py +29 -0
  191. aip_agents/examples/mcp_servers/mcp_server_http.py +19 -0
  192. aip_agents/examples/mcp_servers/mcp_server_sse.py +19 -0
  193. aip_agents/examples/mcp_servers/mcp_server_stdio.py +19 -0
  194. aip_agents/examples/mcp_servers/mcp_time.py +10 -0
  195. aip_agents/examples/pii_demo_langgraph_client.py +69 -0
  196. aip_agents/examples/pii_demo_langgraph_client.pyi +5 -0
  197. aip_agents/examples/pii_demo_langgraph_server.py +126 -0
  198. aip_agents/examples/pii_demo_langgraph_server.pyi +20 -0
  199. aip_agents/examples/pii_demo_multi_agent_client.py +80 -0
  200. aip_agents/examples/pii_demo_multi_agent_client.pyi +5 -0
  201. aip_agents/examples/pii_demo_multi_agent_server.py +247 -0
  202. aip_agents/examples/pii_demo_multi_agent_server.pyi +40 -0
  203. aip_agents/examples/todolist_planning_a2a_langchain_client.py +70 -0
  204. aip_agents/examples/todolist_planning_a2a_langchain_client.pyi +5 -0
  205. aip_agents/examples/todolist_planning_a2a_langgraph_server.py +88 -0
  206. aip_agents/examples/todolist_planning_a2a_langgraph_server.pyi +19 -0
  207. aip_agents/examples/tools/__init__.py +27 -0
  208. aip_agents/examples/tools/__init__.pyi +3 -1
  209. aip_agents/examples/tools/adk_arithmetic_tools.py +36 -0
  210. aip_agents/examples/tools/adk_weather_tool.py +60 -0
  211. aip_agents/examples/tools/adk_weather_tool.pyi +1 -1
  212. aip_agents/examples/tools/data_generator_tool.py +103 -0
  213. aip_agents/examples/tools/data_visualization_tool.py +312 -0
  214. aip_agents/examples/tools/data_visualization_tool.pyi +2 -0
  215. aip_agents/examples/tools/image_artifact_tool.py +136 -0
  216. aip_agents/examples/tools/image_artifact_tool.pyi +2 -0
  217. aip_agents/examples/tools/langchain_arithmetic_tools.py +26 -0
  218. aip_agents/examples/tools/langchain_arithmetic_tools.pyi +7 -0
  219. aip_agents/examples/tools/langchain_currency_exchange_tool.py +88 -0
  220. aip_agents/examples/tools/langchain_currency_exchange_tool.pyi +0 -1
  221. aip_agents/examples/tools/langchain_graph_artifact_tool.py +172 -0
  222. aip_agents/examples/tools/langchain_graph_artifact_tool.pyi +2 -0
  223. aip_agents/examples/tools/langchain_weather_tool.py +48 -0
  224. aip_agents/examples/tools/langchain_weather_tool.pyi +1 -1
  225. aip_agents/examples/tools/langgraph_streaming_tool.py +130 -0
  226. aip_agents/examples/tools/langgraph_streaming_tool.pyi +1 -1
  227. aip_agents/examples/tools/mock_retrieval_tool.py +56 -0
  228. aip_agents/examples/tools/mock_retrieval_tool.pyi +13 -0
  229. aip_agents/examples/tools/pii_demo_tools.py +189 -0
  230. aip_agents/examples/tools/pii_demo_tools.pyi +54 -0
  231. aip_agents/examples/tools/random_chart_tool.py +142 -0
  232. aip_agents/examples/tools/random_chart_tool.pyi +20 -0
  233. aip_agents/examples/tools/serper_tool.py +202 -0
  234. aip_agents/examples/tools/stock_tools.py +82 -0
  235. aip_agents/examples/tools/stock_tools.pyi +18 -3
  236. aip_agents/examples/tools/table_generator_tool.py +167 -0
  237. aip_agents/examples/tools/time_tool.py +82 -0
  238. aip_agents/examples/tools/weather_forecast_tool.py +38 -0
  239. aip_agents/executor/agent_executor.py +473 -0
  240. aip_agents/executor/base.py +48 -0
  241. aip_agents/mcp/__init__.py +1 -0
  242. aip_agents/mcp/client/__init__.py +14 -0
  243. aip_agents/mcp/client/__init__.pyi +4 -3
  244. aip_agents/mcp/client/base_mcp_client.py +369 -0
  245. aip_agents/mcp/client/base_mcp_client.pyi +148 -0
  246. aip_agents/mcp/client/connection_manager.py +193 -0
  247. aip_agents/mcp/client/connection_manager.pyi +48 -0
  248. aip_agents/mcp/client/google_adk/__init__.py +11 -0
  249. aip_agents/mcp/client/google_adk/__init__.pyi +1 -1
  250. aip_agents/mcp/client/google_adk/client.py +381 -0
  251. aip_agents/mcp/client/google_adk/client.pyi +50 -13
  252. aip_agents/mcp/client/langchain/__init__.py +11 -0
  253. aip_agents/mcp/client/langchain/__init__.pyi +1 -1
  254. aip_agents/mcp/client/langchain/client.py +265 -0
  255. aip_agents/mcp/client/langchain/client.pyi +47 -2
  256. aip_agents/mcp/client/persistent_session.py +359 -0
  257. aip_agents/mcp/client/persistent_session.pyi +113 -0
  258. aip_agents/mcp/client/session_pool.py +351 -0
  259. aip_agents/mcp/client/session_pool.pyi +101 -0
  260. aip_agents/mcp/client/transports.py +215 -0
  261. aip_agents/mcp/client/transports.pyi +123 -0
  262. aip_agents/mcp/utils/__init__.py +7 -0
  263. aip_agents/mcp/utils/__init__.pyi +0 -0
  264. aip_agents/mcp/utils/config_validator.py +139 -0
  265. aip_agents/mcp/utils/config_validator.pyi +82 -0
  266. aip_agents/memory/__init__.py +14 -0
  267. aip_agents/memory/__init__.pyi +5 -0
  268. aip_agents/memory/adapters/__init__.py +10 -0
  269. aip_agents/memory/adapters/__init__.pyi +4 -0
  270. aip_agents/memory/adapters/base_adapter.py +717 -0
  271. aip_agents/memory/adapters/base_adapter.pyi +150 -0
  272. aip_agents/memory/adapters/mem0.py +84 -0
  273. aip_agents/memory/adapters/mem0.pyi +22 -0
  274. aip_agents/memory/base.py +84 -0
  275. aip_agents/memory/base.pyi +15 -4
  276. aip_agents/memory/constants.py +49 -0
  277. aip_agents/memory/constants.pyi +25 -0
  278. aip_agents/memory/factory.py +86 -0
  279. aip_agents/memory/factory.pyi +24 -0
  280. aip_agents/memory/guidance.py +20 -0
  281. aip_agents/memory/guidance.pyi +3 -0
  282. aip_agents/memory/simple_memory.py +47 -0
  283. aip_agents/memory/simple_memory.pyi +6 -5
  284. aip_agents/middleware/__init__.py +17 -0
  285. aip_agents/middleware/__init__.pyi +5 -0
  286. aip_agents/middleware/base.py +88 -0
  287. aip_agents/middleware/base.pyi +71 -0
  288. aip_agents/middleware/manager.py +128 -0
  289. aip_agents/middleware/manager.pyi +80 -0
  290. aip_agents/middleware/todolist.py +274 -0
  291. aip_agents/middleware/todolist.pyi +125 -0
  292. aip_agents/schema/__init__.py +69 -0
  293. aip_agents/schema/__init__.pyi +9 -0
  294. aip_agents/schema/a2a.py +56 -0
  295. aip_agents/schema/a2a.pyi +40 -0
  296. aip_agents/schema/agent.py +111 -0
  297. aip_agents/schema/agent.pyi +65 -0
  298. aip_agents/schema/hitl.py +157 -0
  299. aip_agents/schema/hitl.pyi +89 -0
  300. aip_agents/schema/langgraph.py +37 -0
  301. aip_agents/schema/langgraph.pyi +28 -0
  302. aip_agents/schema/model_id.py +97 -0
  303. aip_agents/schema/model_id.pyi +54 -0
  304. aip_agents/schema/step_limit.py +108 -0
  305. aip_agents/schema/step_limit.pyi +63 -0
  306. aip_agents/schema/storage.py +40 -0
  307. aip_agents/schema/storage.pyi +21 -0
  308. aip_agents/sentry/__init__.py +11 -0
  309. aip_agents/sentry/__init__.pyi +1 -1
  310. aip_agents/sentry/sentry.py +151 -0
  311. aip_agents/sentry/sentry.pyi +2 -2
  312. aip_agents/storage/__init__.py +41 -0
  313. aip_agents/storage/base.py +85 -0
  314. aip_agents/storage/base.pyi +1 -2
  315. aip_agents/storage/clients/__init__.py +12 -0
  316. aip_agents/storage/clients/minio_client.py +318 -0
  317. aip_agents/storage/config.py +62 -0
  318. aip_agents/storage/config.pyi +9 -46
  319. aip_agents/storage/providers/__init__.py +15 -0
  320. aip_agents/storage/providers/base.py +106 -0
  321. aip_agents/storage/providers/base.pyi +2 -3
  322. aip_agents/storage/providers/memory.py +114 -0
  323. aip_agents/storage/providers/object_storage.py +214 -0
  324. aip_agents/storage/providers/object_storage.pyi +1 -1
  325. aip_agents/tools/__init__.py +6 -0
  326. aip_agents/tools/__init__.pyi +3 -2
  327. aip_agents/tools/bosa_tools.py +105 -0
  328. aip_agents/tools/bosa_tools.pyi +2 -2
  329. aip_agents/tools/browser_use/__init__.py +82 -0
  330. aip_agents/tools/browser_use/__init__.pyi +14 -0
  331. aip_agents/tools/browser_use/action_parser.py +103 -0
  332. aip_agents/tools/browser_use/action_parser.pyi +18 -0
  333. aip_agents/tools/browser_use/browser_use_tool.py +1112 -0
  334. aip_agents/tools/browser_use/browser_use_tool.pyi +50 -0
  335. aip_agents/tools/browser_use/llm_config.py +120 -0
  336. aip_agents/tools/browser_use/llm_config.pyi +52 -0
  337. aip_agents/tools/browser_use/minio_storage.py +198 -0
  338. aip_agents/tools/browser_use/minio_storage.pyi +109 -0
  339. aip_agents/tools/browser_use/schemas.py +119 -0
  340. aip_agents/tools/browser_use/schemas.pyi +32 -0
  341. aip_agents/tools/browser_use/session.py +76 -0
  342. aip_agents/tools/browser_use/session.pyi +4 -0
  343. aip_agents/tools/browser_use/session_errors.py +132 -0
  344. aip_agents/tools/browser_use/session_errors.pyi +53 -0
  345. aip_agents/tools/browser_use/steel_session_recording.py +317 -0
  346. aip_agents/tools/browser_use/steel_session_recording.pyi +63 -0
  347. aip_agents/tools/browser_use/streaming.py +813 -0
  348. aip_agents/tools/browser_use/streaming.pyi +81 -0
  349. aip_agents/tools/browser_use/structured_data_parser.py +257 -0
  350. aip_agents/tools/browser_use/structured_data_parser.pyi +86 -0
  351. aip_agents/tools/browser_use/structured_data_recovery.py +204 -0
  352. aip_agents/tools/browser_use/structured_data_recovery.pyi +43 -0
  353. aip_agents/tools/browser_use/types.py +78 -0
  354. aip_agents/tools/browser_use/types.pyi +45 -0
  355. aip_agents/tools/code_sandbox/__init__.py +26 -0
  356. aip_agents/tools/code_sandbox/__init__.pyi +3 -0
  357. aip_agents/tools/code_sandbox/constant.py +13 -0
  358. aip_agents/tools/code_sandbox/constant.pyi +4 -0
  359. aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.py +257 -0
  360. aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.pyi +86 -0
  361. aip_agents/tools/code_sandbox/e2b_sandbox_tool.py +411 -0
  362. aip_agents/tools/code_sandbox/e2b_sandbox_tool.pyi +29 -0
  363. aip_agents/tools/constants.py +165 -0
  364. aip_agents/tools/constants.pyi +105 -100
  365. aip_agents/tools/document_loader/__init__.py +37 -0
  366. aip_agents/tools/document_loader/__init__.pyi +7 -0
  367. aip_agents/tools/document_loader/base_reader.py +262 -0
  368. aip_agents/tools/document_loader/base_reader.pyi +62 -0
  369. aip_agents/tools/document_loader/docx_reader_tool.py +53 -0
  370. aip_agents/tools/document_loader/docx_reader_tool.pyi +6 -0
  371. aip_agents/tools/document_loader/excel_reader_tool.py +160 -0
  372. aip_agents/tools/document_loader/excel_reader_tool.pyi +26 -0
  373. aip_agents/tools/document_loader/pdf_reader_tool.py +67 -0
  374. aip_agents/tools/document_loader/pdf_reader_tool.pyi +11 -0
  375. aip_agents/tools/document_loader/pdf_splitter.py +169 -0
  376. aip_agents/tools/document_loader/pdf_splitter.pyi +18 -0
  377. aip_agents/tools/gl_connector/__init__.py +5 -0
  378. aip_agents/tools/gl_connector/__init__.pyi +3 -0
  379. aip_agents/tools/gl_connector/tool.py +351 -0
  380. aip_agents/tools/gl_connector/tool.pyi +74 -0
  381. aip_agents/tools/memory_search/__init__.py +22 -0
  382. aip_agents/tools/memory_search/__init__.pyi +5 -0
  383. aip_agents/tools/memory_search/base.py +200 -0
  384. aip_agents/tools/memory_search/base.pyi +69 -0
  385. aip_agents/tools/memory_search/mem0.py +258 -0
  386. aip_agents/tools/memory_search/mem0.pyi +19 -0
  387. aip_agents/tools/memory_search/schema.py +48 -0
  388. aip_agents/tools/memory_search/schema.pyi +15 -0
  389. aip_agents/tools/memory_search_tool.py +26 -0
  390. aip_agents/tools/memory_search_tool.pyi +3 -0
  391. aip_agents/tools/tool_config_injector.py +300 -0
  392. aip_agents/tools/web_search/__init__.py +15 -0
  393. aip_agents/tools/web_search/__init__.pyi +3 -0
  394. aip_agents/tools/web_search/serper_tool.py +187 -0
  395. aip_agents/tools/web_search/serper_tool.pyi +19 -0
  396. aip_agents/types/__init__.py +70 -0
  397. aip_agents/types/__init__.pyi +35 -2
  398. aip_agents/types/a2a_events.py +13 -0
  399. aip_agents/types/a2a_events.pyi +2 -73
  400. aip_agents/utils/__init__.py +79 -0
  401. aip_agents/utils/__init__.pyi +8 -2
  402. aip_agents/utils/a2a_connector.py +1757 -0
  403. aip_agents/utils/a2a_connector.pyi +15 -4
  404. aip_agents/utils/artifact_helpers.py +502 -0
  405. aip_agents/utils/artifact_helpers.pyi +26 -2
  406. aip_agents/utils/constants.py +22 -0
  407. aip_agents/utils/constants.pyi +10 -0
  408. aip_agents/utils/datetime/__init__.py +34 -0
  409. aip_agents/utils/datetime/__init__.pyi +4 -0
  410. aip_agents/utils/datetime/normalization.py +231 -0
  411. aip_agents/utils/datetime/normalization.pyi +95 -0
  412. aip_agents/utils/datetime/timezone.py +206 -0
  413. aip_agents/utils/datetime/timezone.pyi +48 -0
  414. aip_agents/utils/env_loader.py +27 -0
  415. aip_agents/utils/env_loader.pyi +10 -0
  416. aip_agents/utils/event_handler_registry.py +58 -0
  417. aip_agents/utils/event_handler_registry.pyi +23 -0
  418. aip_agents/utils/file_prompt_utils.py +176 -0
  419. aip_agents/utils/file_prompt_utils.pyi +21 -0
  420. aip_agents/utils/final_response_builder.py +211 -0
  421. aip_agents/utils/final_response_builder.pyi +34 -0
  422. aip_agents/utils/formatter_llm_client.py +231 -0
  423. aip_agents/utils/formatter_llm_client.pyi +71 -0
  424. aip_agents/utils/langgraph/__init__.py +19 -0
  425. aip_agents/utils/langgraph/converter.py +128 -0
  426. aip_agents/utils/langgraph/converter.pyi +1 -1
  427. aip_agents/utils/langgraph/tool_managers/__init__.py +15 -0
  428. aip_agents/utils/langgraph/tool_managers/a2a_tool_manager.py +99 -0
  429. aip_agents/utils/langgraph/tool_managers/a2a_tool_manager.pyi +2 -2
  430. aip_agents/utils/langgraph/tool_managers/base_tool_manager.py +66 -0
  431. aip_agents/utils/langgraph/tool_managers/base_tool_manager.pyi +2 -3
  432. aip_agents/utils/langgraph/tool_managers/delegation_tool_manager.py +1071 -0
  433. aip_agents/utils/langgraph/tool_managers/delegation_tool_manager.pyi +17 -11
  434. aip_agents/utils/langgraph/tool_output_management.py +967 -0
  435. aip_agents/utils/langgraph/tool_output_management.pyi +15 -33
  436. aip_agents/utils/logger.py +195 -0
  437. aip_agents/utils/logger.pyi +60 -0
  438. aip_agents/utils/metadata/__init__.py +27 -0
  439. aip_agents/utils/metadata/__init__.pyi +5 -0
  440. aip_agents/utils/metadata/activity_metadata_helper.py +407 -0
  441. aip_agents/utils/metadata/activity_metadata_helper.pyi +25 -0
  442. aip_agents/utils/metadata/activity_narrative/__init__.py +35 -0
  443. aip_agents/utils/metadata/activity_narrative/__init__.pyi +7 -0
  444. aip_agents/utils/metadata/activity_narrative/builder.py +817 -0
  445. aip_agents/utils/metadata/activity_narrative/builder.pyi +35 -0
  446. aip_agents/utils/metadata/activity_narrative/constants.py +51 -0
  447. aip_agents/utils/metadata/activity_narrative/constants.pyi +10 -0
  448. aip_agents/utils/metadata/activity_narrative/context.py +49 -0
  449. aip_agents/utils/metadata/activity_narrative/context.pyi +32 -0
  450. aip_agents/utils/metadata/activity_narrative/formatters.py +230 -0
  451. aip_agents/utils/metadata/activity_narrative/formatters.pyi +48 -0
  452. aip_agents/utils/metadata/activity_narrative/utils.py +35 -0
  453. aip_agents/utils/metadata/activity_narrative/utils.pyi +12 -0
  454. aip_agents/utils/metadata/schemas/__init__.py +16 -0
  455. aip_agents/utils/metadata/schemas/__init__.pyi +4 -0
  456. aip_agents/utils/metadata/schemas/activity_schema.py +29 -0
  457. aip_agents/utils/metadata/schemas/activity_schema.pyi +18 -0
  458. aip_agents/utils/metadata/schemas/thinking_schema.py +31 -0
  459. aip_agents/utils/metadata/schemas/thinking_schema.pyi +20 -0
  460. aip_agents/utils/metadata/thinking_metadata_helper.py +38 -0
  461. aip_agents/utils/metadata/thinking_metadata_helper.pyi +4 -0
  462. aip_agents/utils/metadata_helper.py +358 -0
  463. aip_agents/utils/metadata_helper.pyi +55 -48
  464. aip_agents/utils/name_preprocessor/__init__.py +17 -0
  465. aip_agents/utils/name_preprocessor/base_name_preprocessor.py +73 -0
  466. aip_agents/utils/name_preprocessor/base_name_preprocessor.pyi +1 -2
  467. aip_agents/utils/name_preprocessor/google_name_preprocessor.py +100 -0
  468. aip_agents/utils/name_preprocessor/google_name_preprocessor.pyi +1 -1
  469. aip_agents/utils/name_preprocessor/name_preprocessor.py +87 -0
  470. aip_agents/utils/name_preprocessor/name_preprocessor.pyi +4 -4
  471. aip_agents/utils/name_preprocessor/openai_name_preprocessor.py +48 -0
  472. aip_agents/utils/name_preprocessor/openai_name_preprocessor.pyi +1 -1
  473. aip_agents/utils/pii/__init__.py +25 -0
  474. aip_agents/utils/pii/__init__.pyi +5 -0
  475. aip_agents/utils/pii/pii_handler.py +397 -0
  476. aip_agents/utils/pii/pii_handler.pyi +96 -0
  477. aip_agents/utils/pii/pii_helper.py +207 -0
  478. aip_agents/utils/pii/pii_helper.pyi +78 -0
  479. aip_agents/utils/pii/uuid_deanonymizer_mapping.py +195 -0
  480. aip_agents/utils/pii/uuid_deanonymizer_mapping.pyi +73 -0
  481. aip_agents/utils/reference_helper.py +273 -0
  482. aip_agents/utils/reference_helper.pyi +38 -6
  483. aip_agents/utils/sse_chunk_transformer.py +831 -0
  484. aip_agents/utils/sse_chunk_transformer.pyi +166 -0
  485. aip_agents/utils/step_limit_manager.py +265 -0
  486. aip_agents/utils/step_limit_manager.pyi +112 -0
  487. aip_agents/utils/token_usage_helper.py +156 -0
  488. aip_agents/utils/token_usage_helper.pyi +1 -1
  489. aip_agents_binary-0.5.11.dist-info/METADATA +689 -0
  490. aip_agents_binary-0.5.11.dist-info/RECORD +542 -0
  491. {aip_agents_binary-0.0.0b2.dist-info → aip_agents_binary-0.5.11.dist-info}/WHEEL +2 -1
  492. aip_agents_binary-0.5.11.dist-info/top_level.txt +1 -0
  493. aip_agents/agent/types.pyi +0 -106
  494. aip_agents/examples/hello_world_a2a_multi_agent_coordinator_client.pyi +0 -15
  495. aip_agents/examples/hello_world_a2a_multi_agent_coordinator_client_streaming.pyi +0 -5
  496. aip_agents/examples/hello_world_a2a_multi_agent_coordinator_server.pyi +0 -11
  497. aip_agents/examples/hello_world_langgraph_bosa.pyi +0 -5
  498. aip_agents/examples/tools/pr_details_bosa_tool.pyi +0 -26
  499. aip_agents/tools/base.pyi +0 -44
  500. aip_agents/tools/base_bosa_tools.pyi +0 -12
  501. aip_agents/tools/bosa_connector.pyi +0 -30
  502. aip_agents/tools/bosa_tools_interface.pyi +0 -26
  503. aip_agents/tools/nested_agent_tool.pyi +0 -45
  504. aip_agents/utils/logger_manager.pyi +0 -151
  505. aip_agents_binary-0.0.0b2.dist-info/METADATA +0 -277
  506. aip_agents_binary-0.0.0b2.dist-info/RECORD +0 -157
@@ -1,4 +1,4 @@
1
- from langchain_core.messages.ai import AIMessage as AIMessage, UsageMetadata as UsageMetadata
1
+ from langchain_core.messages.ai import AIMessage, UsageMetadata
2
2
  from langchain_core.messages.tool import ToolMessage as ToolMessage
3
3
  from langgraph.types import Command
4
4
  from typing import Any
@@ -0,0 +1,689 @@
1
+ Metadata-Version: 2.4
2
+ Name: aip-agents-binary
3
+ Version: 0.5.11
4
+ Summary: A library for managing agents in Gen AI applications.
5
+ Author-email: Raymond Christopher <raymond.christopher@gdplabs.id>
6
+ Requires-Python: <3.13,>=3.11
7
+ Description-Content-Type: text/markdown
8
+ Requires-Dist: a2a-sdk<0.3.0,>=0.2.4
9
+ Requires-Dist: aiostream<0.7.0,>=0.6.0
10
+ Requires-Dist: authlib<2.0.0,>=1.6.4
11
+ Requires-Dist: bosa-connectors-binary<0.4.0,>=0.3.1
12
+ Requires-Dist: bosa-core-binary[logger,telemetry]<0.10.0,>=0.9.6
13
+ Requires-Dist: colorama<0.5.0,>=0.4.6
14
+ Requires-Dist: deprecated<2.0.0,>=1.2.18
15
+ Requires-Dist: fastapi<0.121.0,>=0.120.0
16
+ Requires-Dist: gllm-core-binary<0.4.0,>=0.3.18
17
+ Requires-Dist: gllm-inference-binary[anthropic,bedrock,google-genai,google-vertexai,openai]<0.6.0,>=0.5.90
18
+ Requires-Dist: gllm-tools-binary<0.2.0,>=0.1.3
19
+ Requires-Dist: google-adk<0.6.0,>=0.5.0
20
+ Requires-Dist: langchain<0.4.0,>=0.3.0
21
+ Requires-Dist: langchain-openai<0.4.0,>=0.3.17
22
+ Requires-Dist: langchain-mcp-adapters<0.1.0,>=0.0.10
23
+ Requires-Dist: langchain-experimental<0.4.0,>=0.3.4
24
+ Requires-Dist: langgraph<0.3.0,>=0.2.16
25
+ Requires-Dist: minio<8.0.0,>=7.2.20
26
+ Requires-Dist: pydantic<3.0.0,>=2.11.7
27
+ Requires-Dist: python-dotenv<2.0.0,>=1.1.0
28
+ Requires-Dist: requests<3.0.0,>=2.32.4
29
+ Requires-Dist: uvicorn<0.35.0,>=0.34.3
30
+ Provides-Extra: memory
31
+ Requires-Dist: gllm-memory-binary[mem0ai]<0.2.0,>=0.1.1; extra == "memory"
32
+ Provides-Extra: privacy
33
+ Requires-Dist: gllm-privacy-binary<0.5.0,>=0.4.7; extra == "privacy"
34
+ Provides-Extra: code-sandbox
35
+ Requires-Dist: e2b<2.0.0,>=1.5.4; extra == "code-sandbox"
36
+ Provides-Extra: browser-use-tool
37
+ Requires-Dist: browser-use==0.5.9; extra == "browser-use-tool"
38
+ Requires-Dist: steel-sdk>=0.7.0; extra == "browser-use-tool"
39
+ Requires-Dist: json-repair>=0.52.3; extra == "browser-use-tool"
40
+ Provides-Extra: document-loader
41
+ Requires-Dist: gllm-docproc-binary[docx,pdf,xlsx]<0.8.0,>=0.7.0; extra == "document-loader"
42
+ Requires-Dist: PyPDF2<4.0.0,>=3.0.0; extra == "document-loader"
43
+ Requires-Dist: unidecode<2.0.0,>=1.3.0; extra == "document-loader"
44
+ Provides-Extra: gl-connector
45
+ Requires-Dist: bosa-connectors-binary<0.4.0,>=0.3.1; extra == "gl-connector"
46
+ Provides-Extra: local
47
+ Requires-Dist: e2b<2.0.0,>=1.5.4; extra == "local"
48
+ Requires-Dist: browser-use==0.5.9; extra == "local"
49
+ Requires-Dist: steel-sdk>=0.7.0; extra == "local"
50
+ Requires-Dist: json-repair>=0.52.3; extra == "local"
51
+ Requires-Dist: PyPDF2<4.0.0,>=3.0.0; extra == "local"
52
+ Requires-Dist: unidecode<2.0.0,>=1.3.0; extra == "local"
53
+ Requires-Dist: bosa-connectors-binary<0.4.0,>=0.3.1; extra == "local"
54
+ Provides-Extra: dev
55
+ Requires-Dist: coverage<8.0.0,>=7.4.4; extra == "dev"
56
+ Requires-Dist: mypy<2.0.0,>=1.15.0; extra == "dev"
57
+ Requires-Dist: pre-commit<4.0.0,>=3.7.0; extra == "dev"
58
+ Requires-Dist: pytest<9.0.0,>=8.1.1; extra == "dev"
59
+ Requires-Dist: pytest-asyncio<0.24.0,>=0.23.6; extra == "dev"
60
+ Requires-Dist: pytest-cov<6.0.0,>=5.0.0; extra == "dev"
61
+ Requires-Dist: pytest-xdist>=3.8.0; extra == "dev"
62
+ Requires-Dist: ruff<0.7.0,>=0.6.7; extra == "dev"
63
+ Requires-Dist: pillow<12.0.0,>=11.3.0; extra == "dev"
64
+
65
+ # AIP Agents
66
+
67
+ ## Description
68
+
69
+ The core agent library for **local execution** in GL AIP (GDP Labs AI Agents Package). `aip-agents` is part of the GL AIP ecosystem:
70
+
71
+ - **`aip-agents`** (this library) — The underlying agent library for local execution
72
+ - **[`ai-agent-platform`](https://github.com/GDP-ADMIN/ai-agent-platform)** — The platform that provides remote server/run capabilities (uses `aip-agents` internally)
73
+ - **[`glaip-sdk`](https://github.com/GDP-ADMIN/glaip-sdk/tree/main/python/glaip-sdk)** — The SDK that end users use to run agents either locally (directly via `aip-agents`) or on the remote server (via `ai-agent-platform`)
74
+
75
+ You can use `aip-agents` directly for local execution, or let `glaip-sdk` manage local vs remote mode for you. See the [GL AIP overview](../glaip-sdk/docs/overview.md) for the full architecture.
76
+
77
+ ## Installation
78
+
79
+ ### Prerequisites
80
+ - Python 3.11 - 3.12 - [Install here](https://www.python.org/downloads/)
81
+ - Pip (if using Pip) - [Install here](https://pip.pypa.io/en/stable/installation/)
82
+ - Poetry 1.8.1+ (if using Poetry) - [Install here](https://python-poetry.org/docs/#installation)
83
+ - Git (if using Git) - [Install here](https://git-scm.com/downloads)
84
+ - For git installation:
85
+ - Access to the [GDP Labs SDK repository](https://github.com/GDP-ADMIN/glaip-sdk)
86
+
87
+ ### 1. Installation from the GDP Labs registry
88
+ This package is published to the internal GDP Labs registry. Ensure your pip/Poetry config includes the registry:
89
+ `https://glsdk.gdplabs.id/gen-ai-internal/simple/`.
90
+
91
+ #### Using pip
92
+ ```bash
93
+ pip install aip-agents
94
+ ```
95
+
96
+ #### Using Poetry
97
+ ```bash
98
+ poetry add aip-agents
99
+ ```
100
+
101
+ ### 2. Development Installation (Git)
102
+ For development purposes, you can install directly from the Git repository:
103
+ ```bash
104
+ poetry add "git+ssh://git@github.com/GDP-ADMIN/glaip-sdk.git#subdirectory=python/aip-agents"
105
+ ```
106
+
107
+ ### 3. Recommended: install via glaip-sdk for local mode
108
+ If you want local + remote compatibility, install the SDK's local extra instead:
109
+ ```bash
110
+ pip install "glaip-sdk[local]"
111
+ ```
112
+
113
+ ### Optional Extras
114
+ ```bash
115
+ pip install "aip-agents[memory]"
116
+ pip install "aip-agents[privacy]"
117
+ ```
118
+
119
+ ## Managing Dependencies
120
+ 1. Go to the `aip-agents` module root, e.g. `cd python/aip-agents`.
121
+ 2. Run `poetry shell` to create a virtual environment.
122
+ 3. Run `poetry install` to install the `aip-agents` requirements (Poetry will generate a local lock file for you if needed; the repository ignores it).
123
+ 4. Run `poetry update` if you change any dependency versions in `pyproject.toml`.
124
+
125
+ ## Contributing
126
+ Please refer to this [Python Style Guide](https://docs.google.com/document/d/1uRggCrHnVfDPBnG641FyQBwUwLoFw0kTzNqRm92vUwM/edit?usp=sharing)
127
+ to get information about code style, documentation standard, and SCA that you need to use when contributing to this project
128
+
129
+ 1. Activate `pre-commit` hooks using `pre-commit install`
130
+ 2. Run `poetry shell` to create a virtual environment.
131
+ 3. Run `poetry install` to install the `aip-agents` requirements (this will also create a local lock file that stays local).
132
+ 4. Run `which python` to get the path to be referenced at Visual Studio Code interpreter path (`Ctrl`+`Shift`+`P` or `Cmd`+`Shift`+`P`)
133
+ 5. Try running the unit test to see if it's working:
134
+ ```bash
135
+ poetry run pytest -s tests/unit_tests/
136
+ ```
137
+
138
+ ## Hello World Examples
139
+
140
+ ### Prerequisites
141
+ - Python 3.11 - 3.12
142
+ - Install the package:
143
+
144
+ ```bash
145
+ pip install aip-agents
146
+ ```
147
+
148
+ - For OpenAI: Set your API key in the environment:
149
+ ```bash
150
+ export OPENAI_API_KEY=your-openai-key
151
+ ```
152
+ - For Google ADK: Set your API key in the environment:
153
+ ```bash
154
+ export GOOGLE_API_KEY=your-google-api-key
155
+ ```
156
+
157
+ ### Run the Hello World Examples
158
+
159
+ The example scripts are located in `aip_agents/examples` in the source repo. You can run them individually or use the `run_all_examples.py` script.
160
+
161
+ **1. Running Individual Examples:**
162
+
163
+ Navigate to the library's root directory (e.g., `python/aip-agents` if you cloned the repository).
164
+
165
+ **LangGraph (OpenAI):**
166
+ ```bash
167
+ python aip_agents/examples/hello_world_langgraph.py
168
+ ```
169
+
170
+ **LangGraph with BOSA Connector (OpenAI):**
171
+ ```bash
172
+ python aip_agents/examples/hello_world_langgraph_bosa_twitter.py
173
+ ```
174
+
175
+ **LangGraph Streaming (OpenAI):**
176
+ ```bash
177
+ python aip_agents/examples/hello_world_langgraph_stream.py
178
+ ```
179
+
180
+ **LangGraph Multi-Agent Coordinator (OpenAI):**
181
+ ```bash
182
+ python aip_agents/examples/hello_world_a2a_multi_agent_coordinator_server.py
183
+ ```
184
+
185
+ **Google ADK:**
186
+ ```bash
187
+ python aip_agents/examples/hello_world_google_adk.py
188
+ ```
189
+
190
+ **Google ADK Streaming:**
191
+ ```bash
192
+ python aip_agents/examples/hello_world_google_adk_stream.py
193
+ ```
194
+
195
+ **LangChain (OpenAI):**
196
+ ```bash
197
+ python aip_agents/examples/hello_world_langchain.py
198
+ ```
199
+
200
+ **LangChain Streaming (OpenAI):**
201
+ ```bash
202
+ python aip_agents/examples/hello_world_langchain_stream.py
203
+ ```
204
+
205
+ **HITL (Human-in-the-Loop) Approval Demo:**
206
+ ```bash
207
+ python aip_agents/examples/hitl_demo.py
208
+ ```
209
+
210
+ **2. Running MCP Examples**
211
+
212
+ ### Prerequisites
213
+
214
+ Ensure you have set the environment variables for API keys:
215
+
216
+ ```bash
217
+ export OPENAI_API_KEY="your-openai-key"
218
+ export GOOGLE_API_KEY="your-google-api-key"
219
+ ```
220
+
221
+ For examples that use stateful MCP tools like browser automation, start the Playwright MCP server in a separate terminal:
222
+
223
+ ```bash
224
+ npx @playwright/mcp@latest --headless --port 8931
225
+ ```
226
+
227
+ **Note:** Use the `--headless` flag to run the server without a visible browser window, which is recommended if the browser is not installed yet to avoid failures. For using an actual (non-headless) browser, refer to the [Playwright MCP documentation](https://github.com/microsoft/playwright-mcp).
228
+
229
+ ### Local MCP Servers
230
+
231
+ For STDIO, SSE, and HTTP transports using local servers, open a terminal in the library root (`python/aip-agents`) and run:
232
+
233
+ - For STDIO:
234
+
235
+ ```bash
236
+ poetry run python aip_agents/examples/mcp_servers/mcp_server_stdio.py
237
+ ```
238
+
239
+ - For SSE:
240
+
241
+ ```bash
242
+ poetry run python aip_agents/examples/mcp_servers/mcp_server_sse.py
243
+ ```
244
+
245
+ - For HTTP:
246
+
247
+ ```bash
248
+ poetry run python aip_agents/examples/mcp_servers/mcp_server_http.py
249
+ ```
250
+
251
+ Note: Start the appropriate server before running the client examples for that transport.
252
+
253
+ ### Running Examples
254
+
255
+ All examples are run from the library root using `poetry run python aip_agents/examples/<file>.py`. Examples support OpenAI for LangGraph/LangChain and Google ADK where specified.
256
+
257
+ #### LangChain Examples
258
+
259
+ ##### STDIO Transport
260
+ - Non-Streaming:
261
+ ```bash
262
+ poetry run python aip_agents/examples/hello_world_langchain_mcp_stdio.py
263
+ ```
264
+ - Streaming:
265
+ ```bash
266
+ poetry run python aip_agents/examples/hello_world_langchain_mcp_stdio_stream.py
267
+ ```
268
+
269
+ ##### SSE Transport
270
+ - Non-Streaming:
271
+ ```bash
272
+ poetry run python aip_agents/examples/hello_world_langchain_mcp_sse.py
273
+ ```
274
+ - Streaming:
275
+ ```bash
276
+ poetry run python aip_agents/examples/hello_world_langchain_mcp_sse_stream.py
277
+ ```
278
+
279
+ ##### HTTP Transport
280
+ - Non-Streaming:
281
+ ```bash
282
+ poetry run python aip_agents/examples/hello_world_langchain_mcp_http.py
283
+ ```
284
+ - Streaming:
285
+ ```bash
286
+ poetry run python aip_agents/examples/hello_world_langchain_mcp_http_stream.py
287
+ ```
288
+
289
+ #### Google ADK Examples
290
+
291
+ ##### STDIO Transport
292
+ - Non-Streaming:
293
+ ```bash
294
+ poetry run python aip_agents/examples/hello_world_google_adk_mcp_stdio.py
295
+ ```
296
+ - Streaming:
297
+ ```bash
298
+ poetry run python aip_agents/examples/hello_world_google_adk_mcp_stdio_stream.py
299
+ ```
300
+
301
+ ##### SSE Transport
302
+ - Non-Streaming:
303
+ ```bash
304
+ poetry run python aip_agents/examples/hello_world_google_adk_mcp_sse.py
305
+ ```
306
+ - Streaming:
307
+ ```bash
308
+ poetry run python aip_agents/examples/hello_world_google_adk_mcp_sse_stream.py
309
+ ```
310
+
311
+ ##### HTTP Transport
312
+ - Non-Streaming:
313
+ ```bash
314
+ poetry run python aip_agents/examples/hello_world_google_adk_mcp_http.py
315
+ ```
316
+ - Streaming:
317
+ ```bash
318
+ poetry run python aip_agents/examples/hello_world_google_adk_mcp_http_stream.py
319
+ ```
320
+
321
+ #### LangGraph Examples (OpenAI)
322
+
323
+ ##### STDIO Transport
324
+ - Non-Streaming:
325
+ ```bash
326
+ poetry run python aip_agents/examples/hello_world_langgraph_mcp_stdio.py
327
+ ```
328
+ - Streaming:
329
+ ```bash
330
+ poetry run python aip_agents/examples/hello_world_langgraph_mcp_stdio_stream.py
331
+ ```
332
+
333
+ ##### SSE Transport
334
+ - Non-Streaming:
335
+ ```bash
336
+ poetry run python aip_agents/examples/hello_world_langgraph_mcp_sse.py
337
+ ```
338
+ - Streaming:
339
+ ```bash
340
+ poetry run python aip_agents/examples/hello_world_langgraph_mcp_sse_stream.py
341
+ ```
342
+
343
+ ##### HTTP Transport
344
+ - Non-Streaming:
345
+ ```bash
346
+ poetry run python aip_agents/examples/hello_world_langgraph_mcp_http.py
347
+ ```
348
+ - Streaming:
349
+ ```bash
350
+ poetry run python aip_agents/examples/hello_world_langgraph_mcp_http_stream.py
351
+ ```
352
+
353
+ ### Multi-Server Example
354
+
355
+ This LangChain example uses multiple MCP servers: Playwright (for browser actions) and a random name generator (SSE transport) with persistent sessions across multiple `arun` calls.
356
+
357
+ 1. Start the Playwright server:
358
+
359
+ ```bash
360
+ npx @playwright/mcp@latest --headless --port 8931
361
+ ```
362
+
363
+ 2. In another terminal, start the Name Generator SSE server:
364
+
365
+ ```bash
366
+ poetry run python aip_agents/examples/mcp_servers/mcp_name.py
367
+ ```
368
+
369
+ 3. Run the multi-server client example:
370
+
371
+ ```bash
372
+ poetry run python aip_agents/examples/hello_world_langchain_mcp_multi_server.py
373
+ ```
374
+ **3. Running Individual A2A Examples:**
375
+
376
+ * Navigate to the library's root directory (e.g., `libs/aip-agents` if you cloned the repository).
377
+ * Open a new terminal and navigate to the `aip_agents/examples` directory to run the A2A server.
378
+
379
+ **LangChain Server:**
380
+ ```bash
381
+ python hello_world_a2a_langchain_server.py
382
+ ```
383
+
384
+ * Open a new terminal and navigate to the `aip_agents/examples` directory to run the A2A client.
385
+
386
+ **LangChain Client:**
387
+ ```bash
388
+ python hello_world_a2a_langchain_client.py
389
+ ```
390
+
391
+ **LangChain Client Integrated with Agent Workflow:**
392
+ ```bash
393
+ python hello_world_a2a_langchain_client_agent.py
394
+ ```
395
+
396
+ **LangChain Client Streaming:**
397
+ ```bash
398
+ python hello_world_a2a_langchain_client_stream.py
399
+ ```
400
+
401
+
402
+ ## Human-in-the-Loop (HITL) Approval
403
+
404
+ AIP Agents supports Human-in-the-Loop approval for tool execution, allowing human operators to review and approve high-risk tool calls before they execute.
405
+
406
+ ### Features
407
+
408
+ - **Configurable Approval Policies**: Set approval requirements per tool with customizable timeouts and behaviors
409
+ - **Interactive CLI Prompts**: Clear, structured prompts showing tool details and context
410
+ - **Structured Logging**: All approval decisions are logged with full metadata
411
+ - **Timeout Handling**: Configurable behavior when approval requests time out
412
+ - **Non-blocking**: Tools without HITL configuration execute normally
413
+
414
+ ### Quick Start
415
+
416
+ Configure HITL for specific tools in your agent:
417
+
418
+ ```python
419
+ from aip_agents.agent import LangGraphReactAgent
420
+ from aip_agents.agent.hitl.config import ToolApprovalConfig
421
+
422
+ # Create agent with tools
423
+ agent = LangGraphReactAgent(
424
+ name="My Agent",
425
+ tools=[send_email_tool, search_tool],
426
+ )
427
+
428
+ # Configure HITL via tool_configs
429
+ agent.tool_configs = {
430
+ "tool_configs": {
431
+ "send_email": {"hitl": {"timeout_seconds": 300}}
432
+ }
433
+ }
434
+ ```
435
+
436
+ When the agent attempts to use the `send_email` tool, it will:
437
+
438
+ 1. Emit a pending approval event via `DeferredPromptHandler`
439
+ 2. Wait for `ApprovalManager.resolve_pending_request()` to be called
440
+ 3. Execute the tool only if approved
441
+ 4. Log the decision for audit purposes
442
+
443
+ ### Configuration Options
444
+
445
+ | Option | Type | Default | Description |
446
+ |--------|------|---------|-------------|
447
+ | `timeout_seconds` | int | 300 | Seconds to wait for operator input |
448
+
449
+ ### Logging
450
+
451
+ All HITL decisions are logged with structured data:
452
+
453
+ ```json
454
+ {
455
+ "event": "hitl_decision",
456
+ "tool": "send_email",
457
+ "decision": "approved",
458
+ "operator_input": "A",
459
+ "latency_ms": 2500,
460
+ "timestamp": "2025-09-25T10:15:00Z"
461
+ }
462
+ ```
463
+
464
+ ### Demo
465
+
466
+ Run the interactive demo to see HITL in action:
467
+
468
+ ```bash
469
+ python aip_agents/examples/hitl_demo.py
470
+ ```
471
+
472
+ ## Architectural Notes
473
+
474
+ ### Memory Features
475
+
476
+ The library supports Mem0 as a memory backend for long-term conversation recall. Key features:
477
+ - Automatic persistence of user-agent interactions via `memory_backend="mem0"`.
478
+ - Semantic search for relevant past conversations.
479
+ - New `built_in_mem0_search` tool for explicit recall by time period (e.g., "yesterday", "last week", "July 2025").
480
+ - Date range parsing for natural language time filters using `dateparser`.
481
+ - Conditional auto-augmentation (disabled by default to reduce noise; enable with `memory_auto_augment=True`).
482
+
483
+ #### Mem0 Date Recall Example
484
+
485
+ Use the coordinator example with memory enabled:
486
+
487
+ ```bash
488
+ poetry run python aip_agents/examples/hello_world_a2a_mem0_coordinator_server.py
489
+ ```
490
+
491
+ In client:
492
+ ```python
493
+ agent = LangGraphAgent(
494
+ name="client",
495
+ instruction="...",
496
+ model="gpt-4o-mini",
497
+ memory_backend="mem0",
498
+ )
499
+ ```
500
+
501
+ Test recall: After some interactions, query "What did we discuss yesterday?" – agent uses tool to filter by created_at.
502
+
503
+ ## Deep Agents Middleware
504
+
505
+ The Deep Agents Middleware system provides composable components for enhancing agent capabilities with planning, context management, and custom lifecycle hooks.
506
+
507
+ ### Quick Start
508
+
509
+ Enable deep agent capabilities with a single parameter:
510
+
511
+ ```python
512
+ from aip_agents.agent.langgraph_react_agent import LangGraphReactAgent
513
+
514
+ # Enable planning + filesystem for complex multi-step tasks
515
+ agent = LangGraphReactAgent(
516
+ name="research_agent",
517
+ model="gpt-4",
518
+ planning=True, # Enables TodoListMiddleware for task decomposition
519
+ tools=[search_tool, calculator_tool],
520
+ )
521
+ ```
522
+
523
+ ### Understanding Planning vs Filesystem
524
+
525
+ **Important**: `planning` and `filesystem` are **completely independent** features:
526
+
527
+ - **`planning=True`**
528
+ - Adds `write_todos` tool for task decomposition
529
+ - Stores todos in **in-memory dictionary** (per thread_id)
530
+ - Does NOT use or require filesystem
531
+ - Perfect for breaking down complex tasks into steps
532
+ - Example: "Research quantum computing" → agent creates 5 subtasks
533
+
534
+ - **`filesystem=True`**
535
+ - Adds file operation tools: `ls`, `read_file`, `write_file`, `edit_file`, `grep`
536
+ - Stores data in **pluggable backend** (default: InMemoryBackend)
537
+ - Does NOT interact with planning/todos
538
+ - Perfect for offloading large tool results to prevent context overflow
539
+ - Example: Web search returns 50KB → agent writes to `/research/results.txt`
540
+
541
+ - **Both together** (`planning=True, filesystem=True`)
542
+ - Agent can plan tasks AND manage large data
543
+ - Todos stored separately in memory, files in backend
544
+ - Most powerful combination for complex research/analysis tasks
545
+
546
+ ### Planning Only
547
+
548
+ For task decomposition without filesystem:
549
+
550
+ ```python
551
+ agent = LangGraphReactAgent(
552
+ name="planner_agent",
553
+ model="gpt-4",
554
+ planning=True,
555
+ tools=[...],
556
+ )
557
+ ```
558
+
559
+ ### Filesystem Only
560
+
561
+ For context offloading without planning:
562
+
563
+ ```python
564
+ agent = LangGraphReactAgent(
565
+ name="data_processor",
566
+ model="gpt-4",
567
+ filesystem=True, # Enables FilesystemMiddleware
568
+ tools=[...],
569
+ )
570
+ ```
571
+
572
+ ### Custom Middleware
573
+
574
+ Create domain-specific middleware by implementing the `AgentMiddleware` protocol:
575
+
576
+ ```python
577
+ from aip_agents.middleware.base import AgentMiddleware, ModelRequest
578
+
579
+ class CustomMiddleware:
580
+ def __init__(self):
581
+ self.tools = [] # Add custom tools here
582
+ self.system_prompt_additions = "Custom instructions..."
583
+
584
+ def before_model(self, state: dict) -> dict:
585
+ # Hook executed before model invocation
586
+ return {}
587
+
588
+ def modify_model_request(self, request: ModelRequest, state: dict) -> ModelRequest:
589
+ # Modify the model request (add tools, adjust params, etc.)
590
+ return request
591
+
592
+ def after_model(self, state: dict) -> dict:
593
+ # Hook executed after model invocation
594
+ return {}
595
+
596
+ # COMPOSITION (not override): Custom middlewares EXTEND built-in middleware
597
+ agent = LangGraphReactAgent(
598
+ name="custom_agent",
599
+ model="gpt-4",
600
+ planning=True, # Adds TodoListMiddleware
601
+ filesystem=True, # Adds FilesystemMiddleware
602
+ middlewares=[CustomMiddleware()], # EXTENDS (doesn't replace) the above
603
+ tools=[...],
604
+ )
605
+ # Result: Agent has ALL THREE middleware active:
606
+ # 1. TodoListMiddleware (from planning=True)
607
+ # 2. FilesystemMiddleware (from filesystem=True)
608
+ # 3. CustomMiddleware (from middlewares parameter)
609
+ ```
610
+
611
+ **Key Points:**
612
+ - ✅ `middlewares` parameter **extends** (never replaces) auto-configured middleware
613
+ - ✅ `planning` and `filesystem` are **independent** - use either, both, or neither
614
+ - ✅ `planning=True` stores todos in **memory** (does NOT require filesystem)
615
+ - ✅ Execution order: built-in middleware (planning, filesystem) → custom middlewares
616
+ - ✅ All hooks from all middleware execute in sequence
617
+
618
+ **Common Combinations:**
619
+ ```python
620
+ # Planning only (no filesystem)
621
+ # → Todos stored in memory, no file operations available
622
+ agent = LangGraphReactAgent(planning=True)
623
+ # → [TodoListMiddleware]
624
+
625
+ # Filesystem only (no planning)
626
+ # → File operations available, no todo planning
627
+ agent = LangGraphReactAgent(filesystem=True)
628
+ # → [FilesystemMiddleware]
629
+
630
+ # Both planning and filesystem
631
+ # → Todos in memory + file operations (most powerful combination)
632
+ agent = LangGraphReactAgent(planning=True, filesystem=True)
633
+ # → [TodoListMiddleware, FilesystemMiddleware]
634
+
635
+ # Custom only (no auto-configuration)
636
+ agent = LangGraphReactAgent(middlewares=[CustomMiddleware()])
637
+ # → [CustomMiddleware]
638
+
639
+ # All together (composition)
640
+ agent = LangGraphReactAgent(
641
+ planning=True,
642
+ filesystem=True,
643
+ middlewares=[CustomMiddleware()]
644
+ )
645
+ # → [TodoListMiddleware, FilesystemMiddleware, CustomMiddleware]
646
+ ```
647
+
648
+ ### Advanced: Custom Storage Backend
649
+
650
+ Provide your own storage backend for filesystem operations:
651
+
652
+ ```python
653
+ from aip_agents.middleware.backends.protocol import BackendProtocol
654
+ from aip_agents.middleware.backends.memory import InMemoryBackend
655
+
656
+ # Use custom backend (e.g., PostgreSQL, S3, Redis)
657
+ custom_backend = MyCustomBackend()
658
+
659
+ agent = LangGraphReactAgent(
660
+ name="agent",
661
+ model="gpt-4",
662
+ filesystem=custom_backend, # Pass BackendProtocol instance
663
+ tools=[...],
664
+ )
665
+ ```
666
+
667
+ ### Benefits
668
+
669
+ - **Context Window Management**: Automatically offload large tool results to files
670
+ - **Task Decomposition**: Break down complex multi-step tasks into trackable todos
671
+ - **Incremental Development**: Add capabilities gradually (filesystem first, then planning)
672
+ - **Zero Breaking Changes**: Existing agents work unchanged (backward compatible)
673
+ - **Extensible**: Compose custom middleware with built-in components
674
+
675
+ For detailed documentation, see `docs/deep_agents_guide.md` (coming soon).
676
+
677
+ ### Agent Interface (`AgentInterface`)
678
+
679
+ The `aip_agents.agent.interface.AgentInterface` class defines a standardized contract for all agent implementations within the AIP Agents ecosystem. It ensures that different agent types (e.g., LangGraph-based, Google ADK-based) expose a consistent set of methods for core operations.
680
+
681
+ Key methods defined by `AgentInterface` typically include:
682
+ - `arun()`: For asynchronous execution of the agent that returns a final consolidated response.
683
+ - `arun_stream()`: For asynchronous execution that streams back partial responses or events from the agent.
684
+
685
+ By adhering to this interface, users can interact with various agents in a uniform way, making it easier to switch between or combine different agent technologies.
686
+
687
+ ### Inversion of Control (IoC) / Dependency Injection (DI)
688
+
689
+ The agent implementations (e.g., `LangGraphAgent`, `GoogleADKAgent`) utilize Dependency Injection. For instance, `LangGraphAgent` accepts an `agent_executor` (like one created by LangGraph's `create_react_agent`) in its constructor. Similarly, `GoogleADKAgent` accepts a native `adk_native_agent`. This allows the core execution logic to be provided externally, promoting flexibility and decoupling the agent wrapper from the specific instantiation details of its underlying engine.