@aws/agentcore 1.0.0-preview.8 → 1.0.0-rc.1

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 (389) hide show
  1. package/README.md +978 -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 +3 -0
  5. package/dist/assets/cdk/bin/cdk.ts +109 -59
  6. package/dist/assets/cdk/lib/cdk-stack.ts +30 -19
  7. package/dist/assets/cdk/package.json +9 -9
  8. package/dist/assets/cdk/test/cdk.test.ts +94 -2
  9. package/dist/assets/evaluators/autoevals-lambda/README.md +23 -0
  10. package/dist/assets/evaluators/autoevals-lambda/execution-role-policy.json +15 -0
  11. package/dist/assets/evaluators/autoevals-lambda/lambda_function.py +37 -0
  12. package/dist/assets/evaluators/autoevals-lambda/pyproject.toml +23 -0
  13. package/dist/assets/evaluators/deepeval-lambda/README.md +23 -0
  14. package/dist/assets/evaluators/deepeval-lambda/execution-role-policy.json +15 -0
  15. package/dist/assets/evaluators/deepeval-lambda/lambda_function.py +29 -0
  16. package/dist/assets/evaluators/deepeval-lambda/pyproject.toml +22 -0
  17. package/dist/assets/evaluators/python-lambda/README.md +28 -0
  18. package/dist/assets/evaluators/python-lambda/pyproject.toml +1 -1
  19. package/dist/assets/templates/a2a-python-strands/README.md +31 -0
  20. package/dist/assets/templates/a2a-python-strands/main.py +100 -0
  21. package/dist/assets/templates/a2a-python-strands/memory/session.py +35 -0
  22. package/dist/assets/templates/a2a-python-strands/model/load.py +6 -0
  23. package/dist/assets/templates/a2a-python-strands/pyproject.toml +20 -0
  24. package/dist/assets/templates/agent-python-langchain/README.md +55 -0
  25. package/dist/assets/{python/a2a/googleadk/base → templates/agent-python-langchain}/gitignore.template +1 -1
  26. package/dist/assets/templates/agent-python-langchain/main.py +54 -0
  27. package/dist/assets/templates/agent-python-langchain/model/load.py +9 -0
  28. package/dist/assets/templates/agent-python-langchain/pyproject.toml +21 -0
  29. package/dist/assets/templates/agent-python-minimal/README.md +31 -0
  30. package/dist/assets/templates/agent-python-minimal/main.py +13 -0
  31. package/dist/assets/templates/agent-python-minimal/pyproject.toml +18 -0
  32. package/dist/assets/{container/python/Dockerfile → templates/agent-python-strands/Dockerfile.template} +3 -1
  33. package/dist/assets/templates/agent-python-strands/README.md +46 -0
  34. package/dist/assets/{python/a2a/langchain_langgraph/base → templates/agent-python-strands}/gitignore.template +1 -1
  35. package/dist/assets/templates/agent-python-strands/main.py +106 -0
  36. package/dist/assets/templates/agent-python-strands/memory/__init__.py +0 -0
  37. package/dist/assets/templates/agent-python-strands/memory/session.py +35 -0
  38. package/dist/assets/{python/http/strands/base → templates/agent-python-strands}/model/load.py +46 -0
  39. package/dist/assets/templates/agent-python-strands/pyproject.toml +24 -0
  40. package/dist/assets/templates/agent-typescript-strands/README.md +29 -0
  41. package/dist/assets/templates/agent-typescript-strands/gitignore.template +22 -0
  42. package/dist/assets/templates/agent-typescript-strands/main.ts +86 -0
  43. package/dist/assets/templates/agent-typescript-strands/mcp_client/client.ts +11 -0
  44. package/dist/assets/templates/agent-typescript-strands/memory/memory.ts +44 -0
  45. package/dist/assets/templates/agent-typescript-strands/model/load.ts +5 -0
  46. package/dist/assets/templates/agent-typescript-strands/package.json.template +29 -0
  47. package/dist/assets/templates/agent-typescript-strands/tsconfig.json +19 -0
  48. package/dist/assets/templates/agent-typescript-vercel/README.md +22 -0
  49. package/dist/assets/templates/agent-typescript-vercel/gitignore.template +22 -0
  50. package/dist/assets/templates/agent-typescript-vercel/main.ts +35 -0
  51. package/dist/assets/templates/agent-typescript-vercel/package.json.template +24 -0
  52. package/dist/assets/templates/agent-typescript-vercel/tsconfig.json +19 -0
  53. package/dist/assets/templates/agui-python-strands/README.md +39 -0
  54. package/dist/assets/{python/a2a/strands/base → templates/agui-python-strands}/gitignore.template +1 -1
  55. package/dist/assets/{python/agui/strands/base → templates/agui-python-strands}/main.py +11 -14
  56. package/dist/assets/templates/agui-python-strands/memory/__init__.py +0 -0
  57. package/dist/assets/templates/agui-python-strands/memory/session.py +35 -0
  58. package/dist/assets/templates/agui-python-strands/model/load.py +6 -0
  59. package/dist/assets/templates/agui-python-strands/pyproject.toml +22 -0
  60. package/dist/assets/{python/http/strands/base → templates/export-harness-python}/README.md +15 -8
  61. package/dist/assets/templates/export-harness-python/gitignore.template +41 -0
  62. package/dist/assets/templates/export-harness-python/main.py +584 -0
  63. package/dist/assets/templates/export-harness-python/mcp_client/client.py +63 -0
  64. package/dist/assets/templates/export-harness-python/memory/__init__.py +0 -0
  65. package/dist/assets/templates/export-harness-python/memory/session.py +47 -0
  66. package/dist/assets/templates/export-harness-python/model/load.py +249 -0
  67. package/dist/assets/templates/export-harness-python/model/mantle_compat.py +21 -0
  68. package/dist/assets/templates/export-harness-python/pyproject.toml +22 -0
  69. package/dist/assets/templates/export-harness-python/skills/fetcher.py +279 -0
  70. package/dist/assets/templates/mcp-python-fastmcp/README.md +32 -0
  71. package/dist/assets/templates/mcp-python-fastmcp/gitignore.template +41 -0
  72. package/dist/assets/templates/mcp-python-fastmcp/main.py +83 -0
  73. package/dist/assets/{python/mcp/standalone/base → templates/mcp-python-fastmcp}/pyproject.toml +3 -2
  74. package/dist/assets/templates/shared/env.local.template +11 -0
  75. package/dist/assets/templates/shared/gitignore.template +19 -0
  76. package/dist/index.js +4 -27
  77. package/dist/main.js +1190 -0
  78. package/package.json +75 -149
  79. package/LICENSE +0 -175
  80. package/dist/agent-inspector/index.css +0 -1
  81. package/dist/agent-inspector/index.js +0 -279
  82. package/dist/assets/README.md +0 -104
  83. package/dist/assets/__tests__/__snapshots__/assets.snapshot.test.ts.snap +0 -5673
  84. package/dist/assets/__tests__/__snapshots__/dockerfile-render.test.ts.snap +0 -77
  85. package/dist/assets/__tests__/assets.snapshot.test.ts +0 -125
  86. package/dist/assets/__tests__/dockerfile-render.test.ts +0 -24
  87. package/dist/assets/agents/AGENTS.md +0 -141
  88. package/dist/assets/harness/invoke.py.template +0 -74
  89. package/dist/assets/mcp/python/README.md +0 -27
  90. package/dist/assets/mcp/python/pyproject.toml +0 -22
  91. package/dist/assets/mcp/python/server.py +0 -117
  92. package/dist/assets/mcp/python-lambda/README.md +0 -22
  93. package/dist/assets/mcp/python-lambda/handler.py +0 -144
  94. package/dist/assets/mcp/python-lambda/pyproject.toml +0 -15
  95. package/dist/assets/python/a2a/googleadk/base/README.md +0 -22
  96. package/dist/assets/python/a2a/googleadk/base/main.py +0 -110
  97. package/dist/assets/python/a2a/googleadk/base/model/load.py +0 -41
  98. package/dist/assets/python/a2a/googleadk/base/pyproject.toml +0 -20
  99. package/dist/assets/python/a2a/langchain_langgraph/base/README.md +0 -22
  100. package/dist/assets/python/a2a/langchain_langgraph/base/main.py +0 -130
  101. package/dist/assets/python/a2a/langchain_langgraph/base/model/load.py +0 -123
  102. package/dist/assets/python/a2a/langchain_langgraph/base/pyproject.toml +0 -25
  103. package/dist/assets/python/a2a/strands/base/README.md +0 -22
  104. package/dist/assets/python/a2a/strands/base/main.py +0 -107
  105. package/dist/assets/python/a2a/strands/base/model/load.py +0 -123
  106. package/dist/assets/python/a2a/strands/base/pyproject.toml +0 -23
  107. package/dist/assets/python/a2a/strands/capabilities/memory/session.py +0 -43
  108. package/dist/assets/python/agui/googleadk/base/README.md +0 -30
  109. package/dist/assets/python/agui/googleadk/base/gitignore.template +0 -41
  110. package/dist/assets/python/agui/googleadk/base/main.py +0 -31
  111. package/dist/assets/python/agui/googleadk/base/model/load.py +0 -41
  112. package/dist/assets/python/agui/googleadk/base/pyproject.toml +0 -24
  113. package/dist/assets/python/agui/langchain_langgraph/base/README.md +0 -22
  114. package/dist/assets/python/agui/langchain_langgraph/base/gitignore.template +0 -41
  115. package/dist/assets/python/agui/langchain_langgraph/base/main.py +0 -74
  116. package/dist/assets/python/agui/langchain_langgraph/base/model/load.py +0 -123
  117. package/dist/assets/python/agui/langchain_langgraph/base/pyproject.toml +0 -30
  118. package/dist/assets/python/agui/strands/base/README.md +0 -22
  119. package/dist/assets/python/agui/strands/base/gitignore.template +0 -41
  120. package/dist/assets/python/agui/strands/base/model/__init__.py +0 -1
  121. package/dist/assets/python/agui/strands/base/model/load.py +0 -123
  122. package/dist/assets/python/agui/strands/base/pyproject.toml +0 -27
  123. package/dist/assets/python/agui/strands/capabilities/memory/__init__.py +0 -1
  124. package/dist/assets/python/agui/strands/capabilities/memory/session.py +0 -43
  125. package/dist/assets/python/http/autogen/base/README.md +0 -39
  126. package/dist/assets/python/http/autogen/base/gitignore.template +0 -41
  127. package/dist/assets/python/http/autogen/base/main.py +0 -112
  128. package/dist/assets/python/http/autogen/base/mcp_client/__init__.py +0 -1
  129. package/dist/assets/python/http/autogen/base/mcp_client/client.py +0 -28
  130. package/dist/assets/python/http/autogen/base/model/__init__.py +0 -1
  131. package/dist/assets/python/http/autogen/base/model/load.py +0 -136
  132. package/dist/assets/python/http/autogen/base/pyproject.toml +0 -34
  133. package/dist/assets/python/http/googleadk/base/README.md +0 -39
  134. package/dist/assets/python/http/googleadk/base/gitignore.template +0 -41
  135. package/dist/assets/python/http/googleadk/base/main.py +0 -160
  136. package/dist/assets/python/http/googleadk/base/mcp_client/__init__.py +0 -1
  137. package/dist/assets/python/http/googleadk/base/mcp_client/client.py +0 -75
  138. package/dist/assets/python/http/googleadk/base/model/__init__.py +0 -1
  139. package/dist/assets/python/http/googleadk/base/model/load.py +0 -41
  140. package/dist/assets/python/http/googleadk/base/pyproject.toml +0 -22
  141. package/dist/assets/python/http/langchain_langgraph/base/README.md +0 -39
  142. package/dist/assets/python/http/langchain_langgraph/base/gitignore.template +0 -41
  143. package/dist/assets/python/http/langchain_langgraph/base/main.py +0 -176
  144. package/dist/assets/python/http/langchain_langgraph/base/mcp_client/__init__.py +0 -1
  145. package/dist/assets/python/http/langchain_langgraph/base/mcp_client/client.py +0 -77
  146. package/dist/assets/python/http/langchain_langgraph/base/model/__init__.py +0 -1
  147. package/dist/assets/python/http/langchain_langgraph/base/model/load.py +0 -123
  148. package/dist/assets/python/http/langchain_langgraph/base/pyproject.toml +0 -37
  149. package/dist/assets/python/http/openaiagents/base/README.md +0 -39
  150. package/dist/assets/python/http/openaiagents/base/gitignore.template +0 -41
  151. package/dist/assets/python/http/openaiagents/base/main.py +0 -169
  152. package/dist/assets/python/http/openaiagents/base/mcp_client/__init__.py +0 -1
  153. package/dist/assets/python/http/openaiagents/base/mcp_client/client.py +0 -74
  154. package/dist/assets/python/http/openaiagents/base/model/__init__.py +0 -1
  155. package/dist/assets/python/http/openaiagents/base/model/load.py +0 -37
  156. package/dist/assets/python/http/openaiagents/base/pyproject.toml +0 -21
  157. package/dist/assets/python/http/strands/base/main.py +0 -217
  158. package/dist/assets/python/http/strands/base/mcp_client/__init__.py +0 -1
  159. package/dist/assets/python/http/strands/base/mcp_client/client.py +0 -73
  160. package/dist/assets/python/http/strands/base/model/__init__.py +0 -1
  161. package/dist/assets/python/http/strands/base/pyproject.toml +0 -25
  162. package/dist/assets/python/http/strands/capabilities/memory/__init__.py +0 -1
  163. package/dist/assets/python/http/strands/capabilities/memory/session.py +0 -44
  164. package/dist/assets/python/mcp/standalone/base/README.md +0 -36
  165. package/dist/assets/python/mcp/standalone/base/gitignore.template +0 -41
  166. package/dist/assets/python/mcp/standalone/base/main.py +0 -25
  167. package/dist/cli/index.mjs +0 -1718
  168. package/dist/index.d.ts +0 -9
  169. package/dist/index.d.ts.map +0 -1
  170. package/dist/index.js.map +0 -1
  171. package/dist/lib/constants.d.ts +0 -37
  172. package/dist/lib/constants.d.ts.map +0 -1
  173. package/dist/lib/constants.js +0 -69
  174. package/dist/lib/constants.js.map +0 -1
  175. package/dist/lib/errors/config.d.ts +0 -49
  176. package/dist/lib/errors/config.d.ts.map +0 -1
  177. package/dist/lib/errors/config.js +0 -170
  178. package/dist/lib/errors/config.js.map +0 -1
  179. package/dist/lib/errors/index.d.ts +0 -2
  180. package/dist/lib/errors/index.d.ts.map +0 -1
  181. package/dist/lib/errors/index.js +0 -18
  182. package/dist/lib/errors/index.js.map +0 -1
  183. package/dist/lib/index.d.ts +0 -7
  184. package/dist/lib/index.d.ts.map +0 -1
  185. package/dist/lib/index.js +0 -44
  186. package/dist/lib/index.js.map +0 -1
  187. package/dist/lib/packaging/build-args.d.ts +0 -6
  188. package/dist/lib/packaging/build-args.d.ts.map +0 -1
  189. package/dist/lib/packaging/build-args.js +0 -18
  190. package/dist/lib/packaging/build-args.js.map +0 -1
  191. package/dist/lib/packaging/container.d.ts +0 -10
  192. package/dist/lib/packaging/container.d.ts.map +0 -1
  193. package/dist/lib/packaging/container.js +0 -78
  194. package/dist/lib/packaging/container.js.map +0 -1
  195. package/dist/lib/packaging/errors.d.ts +0 -16
  196. package/dist/lib/packaging/errors.d.ts.map +0 -1
  197. package/dist/lib/packaging/errors.js +0 -36
  198. package/dist/lib/packaging/errors.js.map +0 -1
  199. package/dist/lib/packaging/helpers.d.ts +0 -54
  200. package/dist/lib/packaging/helpers.d.ts.map +0 -1
  201. package/dist/lib/packaging/helpers.js +0 -463
  202. package/dist/lib/packaging/helpers.js.map +0 -1
  203. package/dist/lib/packaging/index.d.ts +0 -40
  204. package/dist/lib/packaging/index.d.ts.map +0 -1
  205. package/dist/lib/packaging/index.js +0 -98
  206. package/dist/lib/packaging/index.js.map +0 -1
  207. package/dist/lib/packaging/node.d.ts +0 -22
  208. package/dist/lib/packaging/node.d.ts.map +0 -1
  209. package/dist/lib/packaging/node.js +0 -109
  210. package/dist/lib/packaging/node.js.map +0 -1
  211. package/dist/lib/packaging/python.d.ts +0 -23
  212. package/dist/lib/packaging/python.d.ts.map +0 -1
  213. package/dist/lib/packaging/python.js +0 -165
  214. package/dist/lib/packaging/python.js.map +0 -1
  215. package/dist/lib/packaging/types/index.d.ts +0 -2
  216. package/dist/lib/packaging/types/index.d.ts.map +0 -1
  217. package/dist/lib/packaging/types/index.js +0 -3
  218. package/dist/lib/packaging/types/index.js.map +0 -1
  219. package/dist/lib/packaging/types/packaging.d.ts +0 -57
  220. package/dist/lib/packaging/types/packaging.d.ts.map +0 -1
  221. package/dist/lib/packaging/types/packaging.js +0 -3
  222. package/dist/lib/packaging/types/packaging.js.map +0 -1
  223. package/dist/lib/packaging/uv.d.ts +0 -7
  224. package/dist/lib/packaging/uv.d.ts.map +0 -1
  225. package/dist/lib/packaging/uv.js +0 -40
  226. package/dist/lib/packaging/uv.js.map +0 -1
  227. package/dist/lib/schemas/io/config-io.d.ts +0 -119
  228. package/dist/lib/schemas/io/config-io.d.ts.map +0 -1
  229. package/dist/lib/schemas/io/config-io.js +0 -323
  230. package/dist/lib/schemas/io/config-io.js.map +0 -1
  231. package/dist/lib/schemas/io/global-config.d.ts +0 -33
  232. package/dist/lib/schemas/io/global-config.d.ts.map +0 -1
  233. package/dist/lib/schemas/io/global-config.js +0 -89
  234. package/dist/lib/schemas/io/global-config.js.map +0 -1
  235. package/dist/lib/schemas/io/index.d.ts +0 -3
  236. package/dist/lib/schemas/io/index.d.ts.map +0 -1
  237. package/dist/lib/schemas/io/index.js +0 -18
  238. package/dist/lib/schemas/io/index.js.map +0 -1
  239. package/dist/lib/schemas/io/path-resolver.d.ts +0 -119
  240. package/dist/lib/schemas/io/path-resolver.d.ts.map +0 -1
  241. package/dist/lib/schemas/io/path-resolver.js +0 -207
  242. package/dist/lib/schemas/io/path-resolver.js.map +0 -1
  243. package/dist/lib/utils/aws-account.d.ts +0 -7
  244. package/dist/lib/utils/aws-account.d.ts.map +0 -1
  245. package/dist/lib/utils/aws-account.js +0 -24
  246. package/dist/lib/utils/aws-account.js.map +0 -1
  247. package/dist/lib/utils/credentials.d.ts +0 -86
  248. package/dist/lib/utils/credentials.d.ts.map +0 -1
  249. package/dist/lib/utils/credentials.js +0 -153
  250. package/dist/lib/utils/credentials.js.map +0 -1
  251. package/dist/lib/utils/env.d.ts +0 -22
  252. package/dist/lib/utils/env.d.ts.map +0 -1
  253. package/dist/lib/utils/env.js +0 -65
  254. package/dist/lib/utils/env.js.map +0 -1
  255. package/dist/lib/utils/index.d.ts +0 -9
  256. package/dist/lib/utils/index.d.ts.map +0 -1
  257. package/dist/lib/utils/index.js +0 -27
  258. package/dist/lib/utils/index.js.map +0 -1
  259. package/dist/lib/utils/json-rpc.d.ts +0 -3
  260. package/dist/lib/utils/json-rpc.d.ts.map +0 -1
  261. package/dist/lib/utils/json-rpc.js +0 -27
  262. package/dist/lib/utils/json-rpc.js.map +0 -1
  263. package/dist/lib/utils/platform.d.ts +0 -63
  264. package/dist/lib/utils/platform.d.ts.map +0 -1
  265. package/dist/lib/utils/platform.js +0 -88
  266. package/dist/lib/utils/platform.js.map +0 -1
  267. package/dist/lib/utils/subprocess.d.ts +0 -29
  268. package/dist/lib/utils/subprocess.d.ts.map +0 -1
  269. package/dist/lib/utils/subprocess.js +0 -116
  270. package/dist/lib/utils/subprocess.js.map +0 -1
  271. package/dist/lib/utils/time-parser.d.ts +0 -11
  272. package/dist/lib/utils/time-parser.d.ts.map +0 -1
  273. package/dist/lib/utils/time-parser.js +0 -47
  274. package/dist/lib/utils/time-parser.js.map +0 -1
  275. package/dist/lib/utils/zod.d.ts +0 -14
  276. package/dist/lib/utils/zod.d.ts.map +0 -1
  277. package/dist/lib/utils/zod.js +0 -32
  278. package/dist/lib/utils/zod.js.map +0 -1
  279. package/dist/schema/constants.d.ts +0 -119
  280. package/dist/schema/constants.d.ts.map +0 -1
  281. package/dist/schema/constants.js +0 -170
  282. package/dist/schema/constants.js.map +0 -1
  283. package/dist/schema/index.d.ts +0 -4
  284. package/dist/schema/index.d.ts.map +0 -1
  285. package/dist/schema/index.js +0 -21
  286. package/dist/schema/index.js.map +0 -1
  287. package/dist/schema/schemas/agent-env.d.ts +0 -186
  288. package/dist/schema/schemas/agent-env.d.ts.map +0 -1
  289. package/dist/schema/schemas/agent-env.js +0 -240
  290. package/dist/schema/schemas/agent-env.js.map +0 -1
  291. package/dist/schema/schemas/agentcore-project.d.ts +0 -838
  292. package/dist/schema/schemas/agentcore-project.d.ts.map +0 -1
  293. package/dist/schema/schemas/agentcore-project.js +0 -335
  294. package/dist/schema/schemas/agentcore-project.js.map +0 -1
  295. package/dist/schema/schemas/auth.d.ts +0 -140
  296. package/dist/schema/schemas/auth.d.ts.map +0 -1
  297. package/dist/schema/schemas/auth.js +0 -114
  298. package/dist/schema/schemas/auth.js.map +0 -1
  299. package/dist/schema/schemas/aws-targets.d.ts +0 -71
  300. package/dist/schema/schemas/aws-targets.d.ts.map +0 -1
  301. package/dist/schema/schemas/aws-targets.js +0 -57
  302. package/dist/schema/schemas/aws-targets.js.map +0 -1
  303. package/dist/schema/schemas/deployed-state.d.ts +0 -725
  304. package/dist/schema/schemas/deployed-state.d.ts.map +0 -1
  305. package/dist/schema/schemas/deployed-state.js +0 -218
  306. package/dist/schema/schemas/deployed-state.js.map +0 -1
  307. package/dist/schema/schemas/index.d.ts +0 -9
  308. package/dist/schema/schemas/index.d.ts.map +0 -1
  309. package/dist/schema/schemas/index.js +0 -26
  310. package/dist/schema/schemas/index.js.map +0 -1
  311. package/dist/schema/schemas/mcp-defs.d.ts +0 -52
  312. package/dist/schema/schemas/mcp-defs.d.ts.map +0 -1
  313. package/dist/schema/schemas/mcp-defs.js +0 -50
  314. package/dist/schema/schemas/mcp-defs.js.map +0 -1
  315. package/dist/schema/schemas/mcp.d.ts +0 -820
  316. package/dist/schema/schemas/mcp.d.ts.map +0 -1
  317. package/dist/schema/schemas/mcp.js +0 -564
  318. package/dist/schema/schemas/mcp.js.map +0 -1
  319. package/dist/schema/schemas/primitives/ab-test.d.ts +0 -141
  320. package/dist/schema/schemas/primitives/ab-test.d.ts.map +0 -1
  321. package/dist/schema/schemas/primitives/ab-test.js +0 -97
  322. package/dist/schema/schemas/primitives/ab-test.js.map +0 -1
  323. package/dist/schema/schemas/primitives/config-bundle.d.ts +0 -31
  324. package/dist/schema/schemas/primitives/config-bundle.d.ts.map +0 -1
  325. package/dist/schema/schemas/primitives/config-bundle.js +0 -38
  326. package/dist/schema/schemas/primitives/config-bundle.js.map +0 -1
  327. package/dist/schema/schemas/primitives/evaluator.d.ts +0 -102
  328. package/dist/schema/schemas/primitives/evaluator.d.ts.map +0 -1
  329. package/dist/schema/schemas/primitives/evaluator.js +0 -84
  330. package/dist/schema/schemas/primitives/evaluator.js.map +0 -1
  331. package/dist/schema/schemas/primitives/harness.d.ts +0 -363
  332. package/dist/schema/schemas/primitives/harness.d.ts.map +0 -1
  333. package/dist/schema/schemas/primitives/harness.js +0 -261
  334. package/dist/schema/schemas/primitives/harness.js.map +0 -1
  335. package/dist/schema/schemas/primitives/http-gateway.d.ts +0 -21
  336. package/dist/schema/schemas/primitives/http-gateway.d.ts.map +0 -1
  337. package/dist/schema/schemas/primitives/http-gateway.js +0 -34
  338. package/dist/schema/schemas/primitives/http-gateway.js.map +0 -1
  339. package/dist/schema/schemas/primitives/index.d.ts +0 -15
  340. package/dist/schema/schemas/primitives/index.d.ts.map +0 -1
  341. package/dist/schema/schemas/primitives/index.js +0 -61
  342. package/dist/schema/schemas/primitives/index.js.map +0 -1
  343. package/dist/schema/schemas/primitives/memory.d.ts +0 -51
  344. package/dist/schema/schemas/primitives/memory.d.ts.map +0 -1
  345. package/dist/schema/schemas/primitives/memory.js +0 -73
  346. package/dist/schema/schemas/primitives/memory.js.map +0 -1
  347. package/dist/schema/schemas/primitives/online-eval-config.d.ts +0 -14
  348. package/dist/schema/schemas/primitives/online-eval-config.d.ts.map +0 -1
  349. package/dist/schema/schemas/primitives/online-eval-config.js +0 -30
  350. package/dist/schema/schemas/primitives/online-eval-config.js.map +0 -1
  351. package/dist/schema/schemas/primitives/policy.d.ts +0 -49
  352. package/dist/schema/schemas/primitives/policy.d.ts.map +0 -1
  353. package/dist/schema/schemas/primitives/policy.js +0 -62
  354. package/dist/schema/schemas/primitives/policy.js.map +0 -1
  355. package/dist/schema/schemas/primitives/tags.d.ts +0 -6
  356. package/dist/schema/schemas/primitives/tags.d.ts.map +0 -1
  357. package/dist/schema/schemas/primitives/tags.js +0 -27
  358. package/dist/schema/schemas/primitives/tags.js.map +0 -1
  359. package/dist/schema/schemas/zod-util.d.ts +0 -10
  360. package/dist/schema/schemas/zod-util.d.ts.map +0 -1
  361. package/dist/schema/schemas/zod-util.js +0 -23
  362. package/dist/schema/schemas/zod-util.js.map +0 -1
  363. package/dist/schema/types/index.d.ts +0 -2
  364. package/dist/schema/types/index.d.ts.map +0 -1
  365. package/dist/schema/types/index.js +0 -18
  366. package/dist/schema/types/index.js.map +0 -1
  367. package/dist/schema/types/path.d.ts +0 -27
  368. package/dist/schema/types/path.d.ts.map +0 -1
  369. package/dist/schema/types/path.js +0 -13
  370. package/dist/schema/types/path.js.map +0 -1
  371. package/scripts/bump-version.ts +0 -393
  372. package/scripts/bundle.mjs +0 -219
  373. package/scripts/check-old-cli.lib.mjs +0 -102
  374. package/scripts/check-old-cli.mjs +0 -10
  375. package/scripts/copy-assets.mjs +0 -64
  376. package/scripts/generate-schema.mjs +0 -40
  377. package/scripts/run-e2e-local.sh +0 -112
  378. package/scripts/start-tui-harness.sh +0 -90
  379. /package/dist/{agent-inspector/favicon.svg → assets/agent-inspector/favicon.svg.asset} +0 -0
  380. /package/dist/{agent-inspector/index.html → assets/agent-inspector/index.html.asset} +0 -0
  381. /package/dist/assets/{python/http/strands/base → templates/a2a-python-strands}/gitignore.template +0 -0
  382. /package/dist/assets/{typescript/.gitkeep → templates/a2a-python-strands/memory/__init__.py} +0 -0
  383. /package/dist/assets/{python/a2a/googleadk/base → templates/a2a-python-strands}/model/__init__.py +0 -0
  384. /package/dist/assets/{python/a2a/langchain_langgraph/base → templates/agent-python-langchain}/model/__init__.py +0 -0
  385. /package/dist/assets/{container/python → templates/agent-python-strands}/dockerignore.template +0 -0
  386. /package/dist/assets/{python/a2a/strands/base → templates/agent-python-strands}/model/__init__.py +0 -0
  387. /package/dist/assets/{python/a2a/strands/capabilities/memory → templates/agui-python-strands/model}/__init__.py +0 -0
  388. /package/dist/assets/{python/agui/googleadk/base/model → templates/export-harness-python/mcp_client}/__init__.py +0 -0
  389. /package/dist/assets/{python/agui/langchain_langgraph/base → templates/export-harness-python}/model/__init__.py +0 -0
@@ -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")))