@aws/agentcore 1.0.0-preview.9 → 1.0.0-rc.3

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 (386) hide show
  1. package/README.md +1218 -145
  2. package/dist/assets/agent-inspector/index.css.asset +1 -0
  3. package/dist/assets/agent-inspector/index.js.asset +289 -0
  4. package/dist/assets/cdk/README.md +43 -17
  5. package/dist/assets/cdk/bin/cdk.ts +22 -123
  6. package/dist/assets/cdk/lib/cdk-stack.ts +18 -69
  7. package/dist/assets/cdk/package.json +6 -13
  8. package/dist/assets/cdk/tsconfig.json +1 -1
  9. package/dist/assets/evaluators/python-lambda/README.md +32 -0
  10. package/dist/assets/evaluators/python-lambda/pyproject.toml +1 -1
  11. package/dist/assets/templates/a2a-python-strands/README.md +31 -0
  12. package/dist/assets/templates/a2a-python-strands/main.py +100 -0
  13. package/dist/assets/templates/a2a-python-strands/memory/session.py +35 -0
  14. package/dist/assets/templates/a2a-python-strands/model/load.py +6 -0
  15. package/dist/assets/templates/a2a-python-strands/pyproject.toml +20 -0
  16. package/dist/assets/templates/agent-python-langchain/README.md +56 -0
  17. package/dist/assets/{python/a2a/googleadk/base → templates/agent-python-langchain}/gitignore.template +1 -1
  18. package/dist/assets/templates/agent-python-langchain/main.py +50 -0
  19. package/dist/assets/templates/agent-python-langchain/model/load.py +9 -0
  20. package/dist/assets/templates/agent-python-langchain/pyproject.toml +21 -0
  21. package/dist/assets/templates/agent-python-minimal/README.md +31 -0
  22. package/dist/assets/templates/agent-python-minimal/main.py +13 -0
  23. package/dist/assets/templates/agent-python-minimal/pyproject.toml +18 -0
  24. package/dist/assets/{container/python/Dockerfile → templates/agent-python-strands/Dockerfile.template} +3 -1
  25. package/dist/assets/templates/agent-python-strands/README.md +60 -0
  26. package/dist/assets/{python/a2a/langchain_langgraph/base → templates/agent-python-strands}/gitignore.template +1 -1
  27. package/dist/assets/templates/agent-python-strands/main.py +65 -0
  28. package/dist/assets/templates/agent-python-strands/memory/__init__.py +0 -0
  29. package/dist/assets/templates/agent-python-strands/memory/session.py +31 -0
  30. package/dist/assets/{python/http/strands/base → templates/agent-python-strands}/model/load.py +49 -3
  31. package/dist/assets/templates/agent-python-strands/parse.py +41 -0
  32. package/dist/assets/templates/agent-python-strands/pyproject.toml +24 -0
  33. package/dist/assets/templates/agent-typescript-strands/README.md +50 -0
  34. package/dist/assets/templates/agent-typescript-strands/gitignore.template +22 -0
  35. package/dist/assets/templates/agent-typescript-strands/main.ts +77 -0
  36. package/dist/assets/templates/agent-typescript-strands/memory/memory.ts +33 -0
  37. package/dist/assets/templates/agent-typescript-strands/model/load.ts +5 -0
  38. package/dist/assets/templates/agent-typescript-strands/package.json.template +28 -0
  39. package/dist/assets/templates/agent-typescript-strands/tsconfig.json +19 -0
  40. package/dist/assets/templates/agent-typescript-vercel/README.md +22 -0
  41. package/dist/assets/templates/agent-typescript-vercel/gitignore.template +22 -0
  42. package/dist/assets/templates/agent-typescript-vercel/main.ts +35 -0
  43. package/dist/assets/templates/agent-typescript-vercel/package.json.template +24 -0
  44. package/dist/assets/templates/agent-typescript-vercel/tsconfig.json +19 -0
  45. package/dist/assets/templates/agui-python-strands/README.md +39 -0
  46. package/dist/assets/{python/a2a/strands/base → templates/agui-python-strands}/gitignore.template +1 -1
  47. package/dist/assets/{python/agui/strands/base → templates/agui-python-strands}/main.py +11 -14
  48. package/dist/assets/templates/agui-python-strands/memory/__init__.py +0 -0
  49. package/dist/assets/templates/agui-python-strands/memory/session.py +35 -0
  50. package/dist/assets/templates/agui-python-strands/model/load.py +6 -0
  51. package/dist/assets/templates/agui-python-strands/pyproject.toml +22 -0
  52. package/dist/assets/{python/http/strands/base → templates/export-harness-python}/README.md +15 -8
  53. package/dist/assets/templates/export-harness-python/gitignore.template +41 -0
  54. package/dist/assets/templates/export-harness-python/main.py +584 -0
  55. package/dist/assets/templates/export-harness-python/mcp_client/client.py +63 -0
  56. package/dist/assets/templates/export-harness-python/memory/__init__.py +0 -0
  57. package/dist/assets/templates/export-harness-python/memory/session.py +47 -0
  58. package/dist/assets/templates/export-harness-python/model/load.py +249 -0
  59. package/dist/assets/templates/export-harness-python/model/mantle_compat.py +21 -0
  60. package/dist/assets/templates/export-harness-python/pyproject.toml +22 -0
  61. package/dist/assets/templates/export-harness-python/skills/fetcher.py +279 -0
  62. package/dist/assets/templates/harness/harness.yaml +157 -0
  63. package/dist/assets/templates/mcp-python-fastmcp/README.md +32 -0
  64. package/dist/assets/templates/mcp-python-fastmcp/gitignore.template +41 -0
  65. package/dist/assets/templates/mcp-python-fastmcp/main.py +83 -0
  66. package/dist/assets/{python/mcp/standalone/base → templates/mcp-python-fastmcp}/pyproject.toml +3 -2
  67. package/dist/assets/templates/shared/env.local.template +11 -0
  68. package/dist/assets/templates/shared/gitignore.template +19 -0
  69. package/dist/index.js +4 -27
  70. package/dist/main.js +1582 -0
  71. package/package.json +76 -148
  72. package/LICENSE +0 -175
  73. package/dist/agent-inspector/index.css +0 -1
  74. package/dist/agent-inspector/index.js +0 -279
  75. package/dist/assets/README.md +0 -104
  76. package/dist/assets/__tests__/__snapshots__/assets.snapshot.test.ts.snap +0 -5673
  77. package/dist/assets/__tests__/__snapshots__/dockerfile-render.test.ts.snap +0 -77
  78. package/dist/assets/__tests__/assets.snapshot.test.ts +0 -125
  79. package/dist/assets/__tests__/dockerfile-render.test.ts +0 -24
  80. package/dist/assets/agents/AGENTS.md +0 -141
  81. package/dist/assets/cdk/.prettierrc +0 -8
  82. package/dist/assets/cdk/jest.config.js +0 -9
  83. package/dist/assets/cdk/npmignore.template +0 -6
  84. package/dist/assets/cdk/test/cdk.test.ts +0 -28
  85. package/dist/assets/harness/invoke.py.template +0 -74
  86. package/dist/assets/mcp/python/README.md +0 -27
  87. package/dist/assets/mcp/python/pyproject.toml +0 -22
  88. package/dist/assets/mcp/python/server.py +0 -117
  89. package/dist/assets/mcp/python-lambda/README.md +0 -22
  90. package/dist/assets/mcp/python-lambda/handler.py +0 -144
  91. package/dist/assets/mcp/python-lambda/pyproject.toml +0 -15
  92. package/dist/assets/python/a2a/googleadk/base/README.md +0 -22
  93. package/dist/assets/python/a2a/googleadk/base/main.py +0 -110
  94. package/dist/assets/python/a2a/googleadk/base/model/load.py +0 -41
  95. package/dist/assets/python/a2a/googleadk/base/pyproject.toml +0 -20
  96. package/dist/assets/python/a2a/langchain_langgraph/base/README.md +0 -22
  97. package/dist/assets/python/a2a/langchain_langgraph/base/main.py +0 -130
  98. package/dist/assets/python/a2a/langchain_langgraph/base/model/load.py +0 -123
  99. package/dist/assets/python/a2a/langchain_langgraph/base/pyproject.toml +0 -25
  100. package/dist/assets/python/a2a/strands/base/README.md +0 -22
  101. package/dist/assets/python/a2a/strands/base/main.py +0 -107
  102. package/dist/assets/python/a2a/strands/base/model/load.py +0 -123
  103. package/dist/assets/python/a2a/strands/base/pyproject.toml +0 -23
  104. package/dist/assets/python/a2a/strands/capabilities/memory/session.py +0 -43
  105. package/dist/assets/python/agui/googleadk/base/README.md +0 -30
  106. package/dist/assets/python/agui/googleadk/base/gitignore.template +0 -41
  107. package/dist/assets/python/agui/googleadk/base/main.py +0 -31
  108. package/dist/assets/python/agui/googleadk/base/model/load.py +0 -41
  109. package/dist/assets/python/agui/googleadk/base/pyproject.toml +0 -24
  110. package/dist/assets/python/agui/langchain_langgraph/base/README.md +0 -22
  111. package/dist/assets/python/agui/langchain_langgraph/base/gitignore.template +0 -41
  112. package/dist/assets/python/agui/langchain_langgraph/base/main.py +0 -74
  113. package/dist/assets/python/agui/langchain_langgraph/base/model/load.py +0 -123
  114. package/dist/assets/python/agui/langchain_langgraph/base/pyproject.toml +0 -30
  115. package/dist/assets/python/agui/strands/base/README.md +0 -22
  116. package/dist/assets/python/agui/strands/base/gitignore.template +0 -41
  117. package/dist/assets/python/agui/strands/base/model/__init__.py +0 -1
  118. package/dist/assets/python/agui/strands/base/model/load.py +0 -123
  119. package/dist/assets/python/agui/strands/base/pyproject.toml +0 -27
  120. package/dist/assets/python/agui/strands/capabilities/memory/__init__.py +0 -1
  121. package/dist/assets/python/agui/strands/capabilities/memory/session.py +0 -43
  122. package/dist/assets/python/http/autogen/base/README.md +0 -39
  123. package/dist/assets/python/http/autogen/base/gitignore.template +0 -41
  124. package/dist/assets/python/http/autogen/base/main.py +0 -112
  125. package/dist/assets/python/http/autogen/base/mcp_client/__init__.py +0 -1
  126. package/dist/assets/python/http/autogen/base/mcp_client/client.py +0 -28
  127. package/dist/assets/python/http/autogen/base/model/__init__.py +0 -1
  128. package/dist/assets/python/http/autogen/base/model/load.py +0 -136
  129. package/dist/assets/python/http/autogen/base/pyproject.toml +0 -34
  130. package/dist/assets/python/http/googleadk/base/README.md +0 -39
  131. package/dist/assets/python/http/googleadk/base/gitignore.template +0 -41
  132. package/dist/assets/python/http/googleadk/base/main.py +0 -160
  133. package/dist/assets/python/http/googleadk/base/mcp_client/__init__.py +0 -1
  134. package/dist/assets/python/http/googleadk/base/mcp_client/client.py +0 -75
  135. package/dist/assets/python/http/googleadk/base/model/__init__.py +0 -1
  136. package/dist/assets/python/http/googleadk/base/model/load.py +0 -41
  137. package/dist/assets/python/http/googleadk/base/pyproject.toml +0 -22
  138. package/dist/assets/python/http/langchain_langgraph/base/README.md +0 -39
  139. package/dist/assets/python/http/langchain_langgraph/base/gitignore.template +0 -41
  140. package/dist/assets/python/http/langchain_langgraph/base/main.py +0 -176
  141. package/dist/assets/python/http/langchain_langgraph/base/mcp_client/__init__.py +0 -1
  142. package/dist/assets/python/http/langchain_langgraph/base/mcp_client/client.py +0 -77
  143. package/dist/assets/python/http/langchain_langgraph/base/model/__init__.py +0 -1
  144. package/dist/assets/python/http/langchain_langgraph/base/model/load.py +0 -123
  145. package/dist/assets/python/http/langchain_langgraph/base/pyproject.toml +0 -37
  146. package/dist/assets/python/http/openaiagents/base/README.md +0 -39
  147. package/dist/assets/python/http/openaiagents/base/gitignore.template +0 -41
  148. package/dist/assets/python/http/openaiagents/base/main.py +0 -169
  149. package/dist/assets/python/http/openaiagents/base/mcp_client/__init__.py +0 -1
  150. package/dist/assets/python/http/openaiagents/base/mcp_client/client.py +0 -74
  151. package/dist/assets/python/http/openaiagents/base/model/__init__.py +0 -1
  152. package/dist/assets/python/http/openaiagents/base/model/load.py +0 -37
  153. package/dist/assets/python/http/openaiagents/base/pyproject.toml +0 -21
  154. package/dist/assets/python/http/strands/base/main.py +0 -217
  155. package/dist/assets/python/http/strands/base/mcp_client/__init__.py +0 -1
  156. package/dist/assets/python/http/strands/base/mcp_client/client.py +0 -73
  157. package/dist/assets/python/http/strands/base/model/__init__.py +0 -1
  158. package/dist/assets/python/http/strands/base/pyproject.toml +0 -25
  159. package/dist/assets/python/http/strands/capabilities/memory/__init__.py +0 -1
  160. package/dist/assets/python/http/strands/capabilities/memory/session.py +0 -44
  161. package/dist/assets/python/mcp/standalone/base/README.md +0 -36
  162. package/dist/assets/python/mcp/standalone/base/gitignore.template +0 -41
  163. package/dist/assets/python/mcp/standalone/base/main.py +0 -25
  164. package/dist/cli/index.mjs +0 -1718
  165. package/dist/index.d.ts +0 -9
  166. package/dist/index.d.ts.map +0 -1
  167. package/dist/index.js.map +0 -1
  168. package/dist/lib/constants.d.ts +0 -37
  169. package/dist/lib/constants.d.ts.map +0 -1
  170. package/dist/lib/constants.js +0 -69
  171. package/dist/lib/constants.js.map +0 -1
  172. package/dist/lib/errors/config.d.ts +0 -49
  173. package/dist/lib/errors/config.d.ts.map +0 -1
  174. package/dist/lib/errors/config.js +0 -170
  175. package/dist/lib/errors/config.js.map +0 -1
  176. package/dist/lib/errors/index.d.ts +0 -2
  177. package/dist/lib/errors/index.d.ts.map +0 -1
  178. package/dist/lib/errors/index.js +0 -18
  179. package/dist/lib/errors/index.js.map +0 -1
  180. package/dist/lib/index.d.ts +0 -7
  181. package/dist/lib/index.d.ts.map +0 -1
  182. package/dist/lib/index.js +0 -44
  183. package/dist/lib/index.js.map +0 -1
  184. package/dist/lib/packaging/build-args.d.ts +0 -6
  185. package/dist/lib/packaging/build-args.d.ts.map +0 -1
  186. package/dist/lib/packaging/build-args.js +0 -18
  187. package/dist/lib/packaging/build-args.js.map +0 -1
  188. package/dist/lib/packaging/container.d.ts +0 -10
  189. package/dist/lib/packaging/container.d.ts.map +0 -1
  190. package/dist/lib/packaging/container.js +0 -78
  191. package/dist/lib/packaging/container.js.map +0 -1
  192. package/dist/lib/packaging/errors.d.ts +0 -16
  193. package/dist/lib/packaging/errors.d.ts.map +0 -1
  194. package/dist/lib/packaging/errors.js +0 -36
  195. package/dist/lib/packaging/errors.js.map +0 -1
  196. package/dist/lib/packaging/helpers.d.ts +0 -54
  197. package/dist/lib/packaging/helpers.d.ts.map +0 -1
  198. package/dist/lib/packaging/helpers.js +0 -463
  199. package/dist/lib/packaging/helpers.js.map +0 -1
  200. package/dist/lib/packaging/index.d.ts +0 -40
  201. package/dist/lib/packaging/index.d.ts.map +0 -1
  202. package/dist/lib/packaging/index.js +0 -98
  203. package/dist/lib/packaging/index.js.map +0 -1
  204. package/dist/lib/packaging/node.d.ts +0 -22
  205. package/dist/lib/packaging/node.d.ts.map +0 -1
  206. package/dist/lib/packaging/node.js +0 -109
  207. package/dist/lib/packaging/node.js.map +0 -1
  208. package/dist/lib/packaging/python.d.ts +0 -23
  209. package/dist/lib/packaging/python.d.ts.map +0 -1
  210. package/dist/lib/packaging/python.js +0 -165
  211. package/dist/lib/packaging/python.js.map +0 -1
  212. package/dist/lib/packaging/types/index.d.ts +0 -2
  213. package/dist/lib/packaging/types/index.d.ts.map +0 -1
  214. package/dist/lib/packaging/types/index.js +0 -3
  215. package/dist/lib/packaging/types/index.js.map +0 -1
  216. package/dist/lib/packaging/types/packaging.d.ts +0 -57
  217. package/dist/lib/packaging/types/packaging.d.ts.map +0 -1
  218. package/dist/lib/packaging/types/packaging.js +0 -3
  219. package/dist/lib/packaging/types/packaging.js.map +0 -1
  220. package/dist/lib/packaging/uv.d.ts +0 -7
  221. package/dist/lib/packaging/uv.d.ts.map +0 -1
  222. package/dist/lib/packaging/uv.js +0 -40
  223. package/dist/lib/packaging/uv.js.map +0 -1
  224. package/dist/lib/schemas/io/config-io.d.ts +0 -119
  225. package/dist/lib/schemas/io/config-io.d.ts.map +0 -1
  226. package/dist/lib/schemas/io/config-io.js +0 -323
  227. package/dist/lib/schemas/io/config-io.js.map +0 -1
  228. package/dist/lib/schemas/io/global-config.d.ts +0 -33
  229. package/dist/lib/schemas/io/global-config.d.ts.map +0 -1
  230. package/dist/lib/schemas/io/global-config.js +0 -89
  231. package/dist/lib/schemas/io/global-config.js.map +0 -1
  232. package/dist/lib/schemas/io/index.d.ts +0 -3
  233. package/dist/lib/schemas/io/index.d.ts.map +0 -1
  234. package/dist/lib/schemas/io/index.js +0 -18
  235. package/dist/lib/schemas/io/index.js.map +0 -1
  236. package/dist/lib/schemas/io/path-resolver.d.ts +0 -119
  237. package/dist/lib/schemas/io/path-resolver.d.ts.map +0 -1
  238. package/dist/lib/schemas/io/path-resolver.js +0 -207
  239. package/dist/lib/schemas/io/path-resolver.js.map +0 -1
  240. package/dist/lib/utils/aws-account.d.ts +0 -7
  241. package/dist/lib/utils/aws-account.d.ts.map +0 -1
  242. package/dist/lib/utils/aws-account.js +0 -24
  243. package/dist/lib/utils/aws-account.js.map +0 -1
  244. package/dist/lib/utils/credentials.d.ts +0 -86
  245. package/dist/lib/utils/credentials.d.ts.map +0 -1
  246. package/dist/lib/utils/credentials.js +0 -153
  247. package/dist/lib/utils/credentials.js.map +0 -1
  248. package/dist/lib/utils/env.d.ts +0 -22
  249. package/dist/lib/utils/env.d.ts.map +0 -1
  250. package/dist/lib/utils/env.js +0 -65
  251. package/dist/lib/utils/env.js.map +0 -1
  252. package/dist/lib/utils/index.d.ts +0 -9
  253. package/dist/lib/utils/index.d.ts.map +0 -1
  254. package/dist/lib/utils/index.js +0 -27
  255. package/dist/lib/utils/index.js.map +0 -1
  256. package/dist/lib/utils/json-rpc.d.ts +0 -3
  257. package/dist/lib/utils/json-rpc.d.ts.map +0 -1
  258. package/dist/lib/utils/json-rpc.js +0 -27
  259. package/dist/lib/utils/json-rpc.js.map +0 -1
  260. package/dist/lib/utils/platform.d.ts +0 -63
  261. package/dist/lib/utils/platform.d.ts.map +0 -1
  262. package/dist/lib/utils/platform.js +0 -88
  263. package/dist/lib/utils/platform.js.map +0 -1
  264. package/dist/lib/utils/subprocess.d.ts +0 -29
  265. package/dist/lib/utils/subprocess.d.ts.map +0 -1
  266. package/dist/lib/utils/subprocess.js +0 -116
  267. package/dist/lib/utils/subprocess.js.map +0 -1
  268. package/dist/lib/utils/time-parser.d.ts +0 -11
  269. package/dist/lib/utils/time-parser.d.ts.map +0 -1
  270. package/dist/lib/utils/time-parser.js +0 -47
  271. package/dist/lib/utils/time-parser.js.map +0 -1
  272. package/dist/lib/utils/zod.d.ts +0 -14
  273. package/dist/lib/utils/zod.d.ts.map +0 -1
  274. package/dist/lib/utils/zod.js +0 -32
  275. package/dist/lib/utils/zod.js.map +0 -1
  276. package/dist/schema/constants.d.ts +0 -119
  277. package/dist/schema/constants.d.ts.map +0 -1
  278. package/dist/schema/constants.js +0 -170
  279. package/dist/schema/constants.js.map +0 -1
  280. package/dist/schema/index.d.ts +0 -4
  281. package/dist/schema/index.d.ts.map +0 -1
  282. package/dist/schema/index.js +0 -21
  283. package/dist/schema/index.js.map +0 -1
  284. package/dist/schema/schemas/agent-env.d.ts +0 -186
  285. package/dist/schema/schemas/agent-env.d.ts.map +0 -1
  286. package/dist/schema/schemas/agent-env.js +0 -240
  287. package/dist/schema/schemas/agent-env.js.map +0 -1
  288. package/dist/schema/schemas/agentcore-project.d.ts +0 -838
  289. package/dist/schema/schemas/agentcore-project.d.ts.map +0 -1
  290. package/dist/schema/schemas/agentcore-project.js +0 -335
  291. package/dist/schema/schemas/agentcore-project.js.map +0 -1
  292. package/dist/schema/schemas/auth.d.ts +0 -140
  293. package/dist/schema/schemas/auth.d.ts.map +0 -1
  294. package/dist/schema/schemas/auth.js +0 -114
  295. package/dist/schema/schemas/auth.js.map +0 -1
  296. package/dist/schema/schemas/aws-targets.d.ts +0 -71
  297. package/dist/schema/schemas/aws-targets.d.ts.map +0 -1
  298. package/dist/schema/schemas/aws-targets.js +0 -57
  299. package/dist/schema/schemas/aws-targets.js.map +0 -1
  300. package/dist/schema/schemas/deployed-state.d.ts +0 -725
  301. package/dist/schema/schemas/deployed-state.d.ts.map +0 -1
  302. package/dist/schema/schemas/deployed-state.js +0 -218
  303. package/dist/schema/schemas/deployed-state.js.map +0 -1
  304. package/dist/schema/schemas/index.d.ts +0 -9
  305. package/dist/schema/schemas/index.d.ts.map +0 -1
  306. package/dist/schema/schemas/index.js +0 -26
  307. package/dist/schema/schemas/index.js.map +0 -1
  308. package/dist/schema/schemas/mcp-defs.d.ts +0 -52
  309. package/dist/schema/schemas/mcp-defs.d.ts.map +0 -1
  310. package/dist/schema/schemas/mcp-defs.js +0 -50
  311. package/dist/schema/schemas/mcp-defs.js.map +0 -1
  312. package/dist/schema/schemas/mcp.d.ts +0 -820
  313. package/dist/schema/schemas/mcp.d.ts.map +0 -1
  314. package/dist/schema/schemas/mcp.js +0 -564
  315. package/dist/schema/schemas/mcp.js.map +0 -1
  316. package/dist/schema/schemas/primitives/ab-test.d.ts +0 -141
  317. package/dist/schema/schemas/primitives/ab-test.d.ts.map +0 -1
  318. package/dist/schema/schemas/primitives/ab-test.js +0 -97
  319. package/dist/schema/schemas/primitives/ab-test.js.map +0 -1
  320. package/dist/schema/schemas/primitives/config-bundle.d.ts +0 -31
  321. package/dist/schema/schemas/primitives/config-bundle.d.ts.map +0 -1
  322. package/dist/schema/schemas/primitives/config-bundle.js +0 -38
  323. package/dist/schema/schemas/primitives/config-bundle.js.map +0 -1
  324. package/dist/schema/schemas/primitives/evaluator.d.ts +0 -102
  325. package/dist/schema/schemas/primitives/evaluator.d.ts.map +0 -1
  326. package/dist/schema/schemas/primitives/evaluator.js +0 -84
  327. package/dist/schema/schemas/primitives/evaluator.js.map +0 -1
  328. package/dist/schema/schemas/primitives/harness.d.ts +0 -363
  329. package/dist/schema/schemas/primitives/harness.d.ts.map +0 -1
  330. package/dist/schema/schemas/primitives/harness.js +0 -261
  331. package/dist/schema/schemas/primitives/harness.js.map +0 -1
  332. package/dist/schema/schemas/primitives/http-gateway.d.ts +0 -21
  333. package/dist/schema/schemas/primitives/http-gateway.d.ts.map +0 -1
  334. package/dist/schema/schemas/primitives/http-gateway.js +0 -34
  335. package/dist/schema/schemas/primitives/http-gateway.js.map +0 -1
  336. package/dist/schema/schemas/primitives/index.d.ts +0 -15
  337. package/dist/schema/schemas/primitives/index.d.ts.map +0 -1
  338. package/dist/schema/schemas/primitives/index.js +0 -61
  339. package/dist/schema/schemas/primitives/index.js.map +0 -1
  340. package/dist/schema/schemas/primitives/memory.d.ts +0 -51
  341. package/dist/schema/schemas/primitives/memory.d.ts.map +0 -1
  342. package/dist/schema/schemas/primitives/memory.js +0 -73
  343. package/dist/schema/schemas/primitives/memory.js.map +0 -1
  344. package/dist/schema/schemas/primitives/online-eval-config.d.ts +0 -14
  345. package/dist/schema/schemas/primitives/online-eval-config.d.ts.map +0 -1
  346. package/dist/schema/schemas/primitives/online-eval-config.js +0 -30
  347. package/dist/schema/schemas/primitives/online-eval-config.js.map +0 -1
  348. package/dist/schema/schemas/primitives/policy.d.ts +0 -49
  349. package/dist/schema/schemas/primitives/policy.d.ts.map +0 -1
  350. package/dist/schema/schemas/primitives/policy.js +0 -62
  351. package/dist/schema/schemas/primitives/policy.js.map +0 -1
  352. package/dist/schema/schemas/primitives/tags.d.ts +0 -6
  353. package/dist/schema/schemas/primitives/tags.d.ts.map +0 -1
  354. package/dist/schema/schemas/primitives/tags.js +0 -27
  355. package/dist/schema/schemas/primitives/tags.js.map +0 -1
  356. package/dist/schema/schemas/zod-util.d.ts +0 -10
  357. package/dist/schema/schemas/zod-util.d.ts.map +0 -1
  358. package/dist/schema/schemas/zod-util.js +0 -23
  359. package/dist/schema/schemas/zod-util.js.map +0 -1
  360. package/dist/schema/types/index.d.ts +0 -2
  361. package/dist/schema/types/index.d.ts.map +0 -1
  362. package/dist/schema/types/index.js +0 -18
  363. package/dist/schema/types/index.js.map +0 -1
  364. package/dist/schema/types/path.d.ts +0 -27
  365. package/dist/schema/types/path.d.ts.map +0 -1
  366. package/dist/schema/types/path.js +0 -13
  367. package/dist/schema/types/path.js.map +0 -1
  368. package/scripts/bump-version.ts +0 -393
  369. package/scripts/bundle.mjs +0 -219
  370. package/scripts/check-old-cli.lib.mjs +0 -102
  371. package/scripts/check-old-cli.mjs +0 -10
  372. package/scripts/copy-assets.mjs +0 -64
  373. package/scripts/generate-schema.mjs +0 -40
  374. package/scripts/run-e2e-local.sh +0 -112
  375. package/scripts/start-tui-harness.sh +0 -90
  376. /package/dist/{agent-inspector/favicon.svg → assets/agent-inspector/favicon.svg.asset} +0 -0
  377. /package/dist/{agent-inspector/index.html → assets/agent-inspector/index.html.asset} +0 -0
  378. /package/dist/assets/{python/http/strands/base → templates/a2a-python-strands}/gitignore.template +0 -0
  379. /package/dist/assets/{typescript/.gitkeep → templates/a2a-python-strands/memory/__init__.py} +0 -0
  380. /package/dist/assets/{python/a2a/googleadk/base → templates/a2a-python-strands}/model/__init__.py +0 -0
  381. /package/dist/assets/{python/a2a/langchain_langgraph/base → templates/agent-python-langchain}/model/__init__.py +0 -0
  382. /package/dist/assets/{container/python → templates/agent-python-strands}/dockerignore.template +0 -0
  383. /package/dist/assets/{python/a2a/strands/base → templates/agent-python-strands}/model/__init__.py +0 -0
  384. /package/dist/assets/{python/a2a/strands/capabilities/memory → templates/agui-python-strands/model}/__init__.py +0 -0
  385. /package/dist/assets/{python/agui/googleadk/base/model → templates/export-harness-python/mcp_client}/__init__.py +0 -0
  386. /package/dist/assets/{python/agui/langchain_langgraph/base → templates/export-harness-python}/model/__init__.py +0 -0
@@ -1,22 +0,0 @@
1
- # {{ name }}
2
-
3
- An A2A (Agent-to-Agent) agent deployed on Amazon Bedrock AgentCore using Strands SDK.
4
-
5
- ## Overview
6
-
7
- This agent implements the A2A protocol, enabling agent-to-agent communication. Other agents can discover and interact with this agent via the `/.well-known/agent-card.json` endpoint.
8
-
9
- ## Local Development
10
-
11
- ```bash
12
- uv sync
13
- uv run python main.py
14
- ```
15
-
16
- The agent starts on port 9000.
17
-
18
- ## Deploy
19
-
20
- ```bash
21
- agentcore deploy
22
- ```
@@ -1,107 +0,0 @@
1
- from strands import Agent, tool
2
- from strands.multiagent.a2a.executor import StrandsA2AExecutor
3
- from bedrock_agentcore.runtime import serve_a2a
4
- from model.load import load_model
5
- {{#if hasMemory}}
6
- from memory.session import get_memory_session_manager
7
- {{/if}}
8
- {{#if sessionStorageMountPath}}
9
- import os
10
- {{/if}}
11
-
12
-
13
- @tool
14
- def add_numbers(a: int, b: int) -> int:
15
- """Return the sum of two numbers."""
16
- return a + b
17
-
18
-
19
- tools = [add_numbers]
20
-
21
- {{#if sessionStorageMountPath}}
22
- SESSION_STORAGE_PATH = "{{sessionStorageMountPath}}"
23
-
24
- def _safe_resolve(path: str) -> str:
25
- """Resolve path safely within the storage boundary."""
26
- resolved = os.path.realpath(os.path.join(SESSION_STORAGE_PATH, path.lstrip("/")))
27
- if not resolved.startswith(os.path.realpath(SESSION_STORAGE_PATH)):
28
- raise ValueError(f"Path '{path}' is outside the storage boundary")
29
- return resolved
30
-
31
- @tool
32
- def file_read(path: str) -> str:
33
- """Read a file from persistent storage. The path is relative to the storage root."""
34
- try:
35
- full_path = _safe_resolve(path)
36
- with open(full_path) as f:
37
- return f.read()
38
- except ValueError as e:
39
- return str(e)
40
- except OSError as e:
41
- return f"Error reading '{path}': {e.strerror}"
42
-
43
- @tool
44
- def file_write(path: str, content: str) -> str:
45
- """Write content to a file in persistent storage. The path is relative to the storage root."""
46
- try:
47
- full_path = _safe_resolve(path)
48
- parent = os.path.dirname(full_path)
49
- if parent:
50
- os.makedirs(parent, exist_ok=True)
51
- with open(full_path, "w") as f:
52
- f.write(content)
53
- return f"Written to {path}"
54
- except ValueError as e:
55
- return str(e)
56
- except OSError as e:
57
- return f"Error writing '{path}': {e.strerror}"
58
-
59
- @tool
60
- def list_files(directory: str = "") -> str:
61
- """List files in persistent storage. The directory is relative to the storage root."""
62
- try:
63
- target = _safe_resolve(directory)
64
- entries = os.listdir(target)
65
- return "\n".join(entries) if entries else "(empty directory)"
66
- except ValueError as e:
67
- return str(e)
68
- except OSError as e:
69
- return f"Error listing '{directory}': {e.strerror}"
70
-
71
- tools.extend([file_read, file_write, list_files])
72
- {{/if}}
73
-
74
- SYSTEM_PROMPT = """
75
- You are a helpful assistant. Use tools when appropriate.
76
- {{#if sessionStorageMountPath}}
77
- You have persistent storage at {{sessionStorageMountPath}}. Use file tools to read and write files. Data persists across sessions.
78
- {{/if}}
79
- """
80
-
81
- {{#if hasMemory}}
82
- def agent_factory():
83
- cache = {}
84
- def get_or_create_agent(session_id, user_id):
85
- key = f"{session_id}/{user_id}"
86
- if key not in cache:
87
- cache[key] = Agent(
88
- model=load_model(),
89
- session_manager=get_memory_session_manager(session_id, user_id),
90
- system_prompt=SYSTEM_PROMPT,
91
- tools=tools,
92
- )
93
- return cache[key]
94
- return get_or_create_agent
95
-
96
- get_or_create_agent = agent_factory()
97
- agent = get_or_create_agent("default-session", "default-user")
98
- {{else}}
99
- agent = Agent(
100
- model=load_model(),
101
- system_prompt=SYSTEM_PROMPT,
102
- tools=tools,
103
- )
104
- {{/if}}
105
-
106
- if __name__ == "__main__":
107
- serve_a2a(StrandsA2AExecutor(agent))
@@ -1,123 +0,0 @@
1
- {{#if (eq modelProvider "Bedrock")}}
2
- from strands.models.bedrock import BedrockModel
3
-
4
-
5
- def load_model() -> BedrockModel:
6
- """Get Bedrock model client using IAM credentials."""
7
- return BedrockModel(model_id="global.anthropic.claude-sonnet-4-5-20250929-v1:0")
8
- {{/if}}
9
- {{#if (eq modelProvider "Anthropic")}}
10
- import os
11
-
12
- from strands.models.anthropic import AnthropicModel
13
- from bedrock_agentcore.identity.auth import requires_api_key
14
-
15
- IDENTITY_PROVIDER_NAME = "{{identityProviders.[0].name}}"
16
- IDENTITY_ENV_VAR = "{{identityProviders.[0].envVarName}}"
17
-
18
-
19
- @requires_api_key(provider_name=IDENTITY_PROVIDER_NAME)
20
- def _agentcore_identity_api_key_provider(api_key: str) -> str:
21
- """Fetch API key from AgentCore Identity."""
22
- return api_key
23
-
24
-
25
- def _get_api_key() -> str:
26
- """
27
- Uses AgentCore Identity for API key management in deployed environments.
28
- For local development, run via 'agentcore dev' which loads agentcore/.env.
29
- """
30
- if os.getenv("LOCAL_DEV") == "1":
31
- api_key = os.getenv(IDENTITY_ENV_VAR)
32
- if not api_key:
33
- raise RuntimeError(
34
- f"{IDENTITY_ENV_VAR} not found. Add {IDENTITY_ENV_VAR}=your-key to .env.local"
35
- )
36
- return api_key
37
- return _agentcore_identity_api_key_provider()
38
-
39
-
40
- def load_model() -> AnthropicModel:
41
- """Get authenticated Anthropic model client."""
42
- return AnthropicModel(
43
- client_args={"api_key": _get_api_key()},
44
- model_id="claude-sonnet-4-5-20250929",
45
- max_tokens=5000,
46
- )
47
- {{/if}}
48
- {{#if (eq modelProvider "OpenAI")}}
49
- import os
50
-
51
- from strands.models.openai import OpenAIModel
52
- from bedrock_agentcore.identity.auth import requires_api_key
53
-
54
- IDENTITY_PROVIDER_NAME = "{{identityProviders.[0].name}}"
55
- IDENTITY_ENV_VAR = "{{identityProviders.[0].envVarName}}"
56
-
57
-
58
- @requires_api_key(provider_name=IDENTITY_PROVIDER_NAME)
59
- def _agentcore_identity_api_key_provider(api_key: str) -> str:
60
- """Fetch API key from AgentCore Identity."""
61
- return api_key
62
-
63
-
64
- def _get_api_key() -> str:
65
- """
66
- Uses AgentCore Identity for API key management in deployed environments.
67
- For local development, run via 'agentcore dev' which loads agentcore/.env.
68
- """
69
- if os.getenv("LOCAL_DEV") == "1":
70
- api_key = os.getenv(IDENTITY_ENV_VAR)
71
- if not api_key:
72
- raise RuntimeError(
73
- f"{IDENTITY_ENV_VAR} not found. Add {IDENTITY_ENV_VAR}=your-key to .env.local"
74
- )
75
- return api_key
76
- return _agentcore_identity_api_key_provider()
77
-
78
-
79
- def load_model() -> OpenAIModel:
80
- """Get authenticated OpenAI model client."""
81
- return OpenAIModel(
82
- client_args={"api_key": _get_api_key()},
83
- model_id="gpt-4.1",
84
- )
85
- {{/if}}
86
- {{#if (eq modelProvider "Gemini")}}
87
- import os
88
-
89
- from strands.models.gemini import GeminiModel
90
- from bedrock_agentcore.identity.auth import requires_api_key
91
-
92
- IDENTITY_PROVIDER_NAME = "{{identityProviders.[0].name}}"
93
- IDENTITY_ENV_VAR = "{{identityProviders.[0].envVarName}}"
94
-
95
-
96
- @requires_api_key(provider_name=IDENTITY_PROVIDER_NAME)
97
- def _agentcore_identity_api_key_provider(api_key: str) -> str:
98
- """Fetch API key from AgentCore Identity."""
99
- return api_key
100
-
101
-
102
- def _get_api_key() -> str:
103
- """
104
- Uses AgentCore Identity for API key management in deployed environments.
105
- For local development, run via 'agentcore dev' which loads agentcore/.env.
106
- """
107
- if os.getenv("LOCAL_DEV") == "1":
108
- api_key = os.getenv(IDENTITY_ENV_VAR)
109
- if not api_key:
110
- raise RuntimeError(
111
- f"{IDENTITY_ENV_VAR} not found. Add {IDENTITY_ENV_VAR}=your-key to .env.local"
112
- )
113
- return api_key
114
- return _agentcore_identity_api_key_provider()
115
-
116
-
117
- def load_model() -> GeminiModel:
118
- """Get authenticated Gemini model client."""
119
- return GeminiModel(
120
- client_args={"api_key": _get_api_key()},
121
- model_id="gemini-2.5-flash",
122
- )
123
- {{/if}}
@@ -1,23 +0,0 @@
1
- [build-system]
2
- requires = ["hatchling"]
3
- build-backend = "hatchling.build"
4
-
5
- [project]
6
- name = "{{ name }}"
7
- version = "0.1.0"
8
- description = "AgentCore A2A Agent using Strands SDK"
9
- readme = "README.md"
10
- requires-python = ">=3.10"
11
- dependencies = [
12
- {{#if (eq modelProvider "Anthropic")}}"anthropic >= 0.30.0",
13
- {{/if}}"a2a-sdk[all] >= 0.2.0",
14
- "aws-opentelemetry-distro",
15
- "bedrock-agentcore[a2a] >= 1.0.3",
16
- "botocore[crt] >= 1.35.0",
17
- {{#if (eq modelProvider "Gemini")}}"google-genai >= 1.0.0",
18
- {{/if}}{{#if (eq modelProvider "OpenAI")}}"openai >= 1.0.0",
19
- {{/if}}"strands-agents >= 1.13.0",
20
- ]
21
-
22
- [tool.hatch.build.targets.wheel]
23
- packages = ["."]
@@ -1,43 +0,0 @@
1
- import os
2
- import uuid
3
- from typing import Optional
4
-
5
- from bedrock_agentcore.memory.integrations.strands.config import AgentCoreMemoryConfig{{#if memoryProviders.[0].strategies.length}}, RetrievalConfig{{/if}}
6
- from bedrock_agentcore.memory.integrations.strands.session_manager import AgentCoreMemorySessionManager
7
-
8
- MEMORY_ID = os.getenv("{{memoryProviders.[0].envVarName}}")
9
- REGION = os.getenv("AWS_REGION")
10
-
11
- def get_memory_session_manager(session_id: Optional[str], actor_id: str) -> Optional[AgentCoreMemorySessionManager]:
12
- if not MEMORY_ID:
13
- return None
14
-
15
- # AgentCoreMemoryConfig rejects None; OAuth/CUSTOM_JWT callers can reach us
16
- # without a runtime session header, so synthesize one when absent.
17
- session_id = session_id or uuid.uuid4().hex
18
-
19
- {{#if memoryProviders.[0].strategies.length}}
20
- retrieval_config = {
21
- {{#if (includes memoryProviders.[0].strategies "SEMANTIC")}}
22
- f"/users/{actor_id}/facts": RetrievalConfig(top_k=3, relevance_score=0.5),
23
- {{/if}}
24
- {{#if (includes memoryProviders.[0].strategies "USER_PREFERENCE")}}
25
- f"/users/{actor_id}/preferences": RetrievalConfig(top_k=3, relevance_score=0.5),
26
- {{/if}}
27
- {{#if (includes memoryProviders.[0].strategies "SUMMARIZATION")}}
28
- f"/summaries/{actor_id}/{session_id}": RetrievalConfig(top_k=3, relevance_score=0.5),
29
- {{/if}}
30
- }
31
- {{/if}}
32
-
33
- return AgentCoreMemorySessionManager(
34
- AgentCoreMemoryConfig(
35
- memory_id=MEMORY_ID,
36
- session_id=session_id,
37
- actor_id=actor_id,
38
- {{#if memoryProviders.[0].strategies.length}}
39
- retrieval_config=retrieval_config,
40
- {{/if}}
41
- ),
42
- REGION
43
- )
@@ -1,30 +0,0 @@
1
- # {{ name }}
2
-
3
- An AG-UI agent deployed on Amazon Bedrock AgentCore using Google ADK.
4
-
5
- ## Overview
6
-
7
- This agent implements the AG-UI protocol using Google's Agent Development Kit, enabling rich agent-user interaction via the AG-UI event stream.
8
-
9
- ## Local Development
10
-
11
- ```bash
12
- uv sync
13
- uv run python main.py
14
- ```
15
-
16
- The agent starts on port 8080 and serves requests at `/invocations`.
17
-
18
- ## Health Check
19
-
20
- ```
21
- GET /ping
22
- ```
23
-
24
- Returns `{"status": "healthy"}`.
25
-
26
- ## Deploy
27
-
28
- ```bash
29
- agentcore deploy
30
- ```
@@ -1,41 +0,0 @@
1
- # Environment variables
2
- .env
3
-
4
- # Python
5
- __pycache__/
6
- *.py[cod]
7
- *$py.class
8
- *.so
9
- .Python
10
- build/
11
- develop-eggs/
12
- dist/
13
- downloads/
14
- eggs/
15
- .eggs/
16
- lib/
17
- lib64/
18
- parts/
19
- sdist/
20
- var/
21
- wheels/
22
- *.egg-info/
23
- .installed.cfg
24
- *.egg
25
-
26
- # Virtual environments
27
- .venv/
28
- venv/
29
- ENV/
30
- env/
31
-
32
- # IDE
33
- .vscode/
34
- .idea/
35
- *.swp
36
- *.swo
37
- *~
38
-
39
- # OS
40
- .DS_Store
41
- Thumbs.db
@@ -1,31 +0,0 @@
1
- import os
2
- import uvicorn
3
- from google.adk.agents import LlmAgent
4
- from ag_ui_adk import ADKAgent, AGUIToolset, create_adk_app
5
- from model.load import load_model
6
-
7
- load_model()
8
-
9
- agent = LlmAgent(
10
- name="{{ name }}",
11
- model="gemini-2.5-flash",
12
- instruction="You are a helpful assistant.",
13
- tools=[AGUIToolset()],
14
- )
15
-
16
- adk_agent = ADKAgent(
17
- adk_agent=agent,
18
- app_name="{{ name }}",
19
- use_in_memory_services=True,
20
- )
21
-
22
- app = create_adk_app(adk_agent, path="/invocations")
23
-
24
-
25
- @app.get("/ping")
26
- async def ping():
27
- return {"status": "healthy"}
28
-
29
-
30
- if __name__ == "__main__":
31
- uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080")))
@@ -1,41 +0,0 @@
1
- import os
2
- from bedrock_agentcore.identity.auth import requires_api_key
3
-
4
- IDENTITY_PROVIDER_NAME = "{{identityProviders.[0].name}}"
5
- IDENTITY_ENV_VAR = "{{identityProviders.[0].envVarName}}"
6
-
7
-
8
- @requires_api_key(provider_name=IDENTITY_PROVIDER_NAME)
9
- def _agentcore_identity_api_key_provider(api_key: str) -> str:
10
- """Fetch API key from AgentCore Identity."""
11
- return api_key
12
-
13
-
14
- def _get_api_key() -> str:
15
- """
16
- Uses AgentCore Identity for API key management in deployed environments.
17
- For local development, run via 'agentcore dev' which loads agentcore/.env.
18
- """
19
- if os.getenv("LOCAL_DEV") == "1":
20
- api_key = os.getenv(IDENTITY_ENV_VAR)
21
- if not api_key:
22
- raise RuntimeError(
23
- f"{IDENTITY_ENV_VAR} not found. Add {IDENTITY_ENV_VAR}=your-key to .env.local"
24
- )
25
- return api_key
26
- return _agentcore_identity_api_key_provider()
27
-
28
-
29
- def load_model() -> None:
30
- """
31
- Set up Gemini API key authentication.
32
- Uses AgentCore Identity for API key management in deployed environments,
33
- and falls back to .env file for local development.
34
- Sets the GOOGLE_API_KEY environment variable for the Google ADK.
35
- """
36
- api_key = _get_api_key()
37
- # Use Google AI Studios API Key Authentication.
38
- # https://google.github.io/adk-docs/agents/models/#google-ai-studio
39
- os.environ["GOOGLE_API_KEY"] = api_key
40
- # Set to TRUE is using Google Vertex AI, Set to FALSE for Google AI Studio
41
- os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "FALSE"
@@ -1,24 +0,0 @@
1
- [build-system]
2
- requires = ["hatchling"]
3
- build-backend = "hatchling.build"
4
-
5
- [project]
6
- name = "{{ name }}"
7
- version = "0.1.0"
8
- description = "AgentCore AG-UI Agent using Google ADK"
9
- readme = "README.md"
10
- requires-python = ">=3.10"
11
- dependencies = [
12
- "ag-ui-adk >= 0.6.0",
13
- "ag-ui-protocol >= 0.1.10",
14
- "bedrock-agentcore >= 1.0.3",
15
- "fastapi >= 0.115.12",
16
- "google-adk >= 1.16.0",
17
- "google-genai >= 1.0.0",
18
- "opentelemetry-distro",
19
- "opentelemetry-exporter-otlp",
20
- "uvicorn >= 0.34.3",
21
- ]
22
-
23
- [tool.hatch.build.targets.wheel]
24
- packages = ["."]
@@ -1,22 +0,0 @@
1
- # {{ name }}
2
-
3
- An AG-UI agent deployed on Amazon Bedrock AgentCore using LangChain + LangGraph.
4
-
5
- ## Overview
6
-
7
- This agent implements the AG-UI protocol using LangGraph, enabling seamless frontend-to-agent communication with support for streaming, tool calls, and frontend-injected tools.
8
-
9
- ## Local Development
10
-
11
- ```bash
12
- uv sync
13
- uv run python main.py
14
- ```
15
-
16
- The agent starts on port 8080.
17
-
18
- ## Deploy
19
-
20
- ```bash
21
- agentcore deploy
22
- ```
@@ -1,41 +0,0 @@
1
- # Environment variables
2
- .env
3
-
4
- # Python
5
- __pycache__/
6
- *.py[cod]
7
- *$py.class
8
- *.so
9
- .Python
10
- build/
11
- develop-eggs/
12
- dist/
13
- downloads/
14
- eggs/
15
- .eggs/
16
- lib/
17
- lib64/
18
- parts/
19
- sdist/
20
- var/
21
- wheels/
22
- *.egg-info/
23
- .installed.cfg
24
- *.egg
25
-
26
- # Virtual environments
27
- .venv/
28
- venv/
29
- ENV/
30
- env/
31
-
32
- # IDE
33
- .vscode/
34
- .idea/
35
- *.swp
36
- *.swo
37
- *~
38
-
39
- # OS
40
- .DS_Store
41
- Thumbs.db
@@ -1,74 +0,0 @@
1
- import os
2
-
3
- os.environ["LANGGRAPH_FAST_API"] = "true"
4
-
5
- import uvicorn
6
- from typing import Any, List
7
- from fastapi import FastAPI
8
- from fastapi.middleware.cors import CORSMiddleware
9
- from langgraph.graph import StateGraph, START
10
- from langgraph.graph.message import MessagesState
11
- from langgraph.checkpoint.memory import MemorySaver
12
- from langgraph.prebuilt import ToolNode, tools_condition
13
- from langchain_core.tools import tool
14
- from opentelemetry.instrumentation.langchain import LangchainInstrumentor
15
- from ag_ui_langgraph import LangGraphAgent, add_langgraph_fastapi_endpoint
16
- from model.load import load_model
17
-
18
- LangchainInstrumentor().instrument()
19
-
20
-
21
- @tool
22
- def add_numbers(a: int, b: int) -> int:
23
- """Return the sum of two numbers."""
24
- return a + b
25
-
26
-
27
- backend_tools = [add_numbers]
28
- model = load_model()
29
-
30
-
31
- class AgentState(MessagesState):
32
- tools: List[Any]
33
-
34
-
35
- def chat_node(state: AgentState):
36
- bound_model = model.bind_tools(
37
- [*state.get("tools", []), *backend_tools],
38
- )
39
- response = bound_model.invoke(state["messages"])
40
- return {"messages": [response]}
41
-
42
-
43
- builder = StateGraph(AgentState)
44
- builder.add_node("chat", chat_node)
45
- builder.add_node("tools", ToolNode(tools=backend_tools))
46
- builder.add_edge(START, "chat")
47
- builder.add_conditional_edges("chat", tools_condition)
48
- builder.add_edge("tools", "chat")
49
- graph = builder.compile(checkpointer=MemorySaver())
50
-
51
- agent = LangGraphAgent(
52
- name="{{ name }}",
53
- graph=graph,
54
- description="A helpful assistant",
55
- )
56
-
57
- app = FastAPI()
58
- app.add_middleware(
59
- CORSMiddleware,
60
- allow_origins=["*"],
61
- allow_methods=["*"],
62
- allow_headers=["*"],
63
- )
64
-
65
- add_langgraph_fastapi_endpoint(app=app, agent=agent, path="/invocations")
66
-
67
-
68
- @app.get("/ping")
69
- async def ping():
70
- return {"status": "healthy"}
71
-
72
-
73
- if __name__ == "__main__":
74
- uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080")))