pygeai 0.1.6__py3-none-any.whl → 0.6.0b15__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.

Potentially problematic release.


This version of pygeai might be problematic. Click here for more details.

Files changed (788) hide show
  1. pygeai/__init__.py +11 -2
  2. pygeai/_docs/Makefile +20 -0
  3. pygeai/_docs/make.bat +35 -0
  4. pygeai/_docs/source/conf.py +117 -0
  5. pygeai/_docs/source/content/ai_lab/cli.rst +747 -0
  6. pygeai/_docs/source/content/ai_lab/models.rst +1734 -0
  7. pygeai/_docs/source/content/ai_lab/runner.rst +253 -0
  8. pygeai/_docs/source/content/ai_lab/spec.rst +431 -0
  9. pygeai/_docs/source/content/ai_lab/usage.rst +1011 -0
  10. pygeai/_docs/source/content/ai_lab.rst +102 -0
  11. pygeai/_docs/source/content/analytics.rst +598 -0
  12. pygeai/_docs/source/content/api_reference/admin.rst +161 -0
  13. pygeai/_docs/source/content/api_reference/assistant.rst +326 -0
  14. pygeai/_docs/source/content/api_reference/auth.rst +379 -0
  15. pygeai/_docs/source/content/api_reference/chat.rst +754 -0
  16. pygeai/_docs/source/content/api_reference/embeddings.rst +154 -0
  17. pygeai/_docs/source/content/api_reference/evaluation.rst +590 -0
  18. pygeai/_docs/source/content/api_reference/feedback.rst +237 -0
  19. pygeai/_docs/source/content/api_reference/files.rst +592 -0
  20. pygeai/_docs/source/content/api_reference/gam.rst +401 -0
  21. pygeai/_docs/source/content/api_reference/health.rst +58 -0
  22. pygeai/_docs/source/content/api_reference/project.rst +738 -0
  23. pygeai/_docs/source/content/api_reference/proxy.rst +318 -0
  24. pygeai/_docs/source/content/api_reference/rag.rst +710 -0
  25. pygeai/_docs/source/content/api_reference/rerank.rst +94 -0
  26. pygeai/_docs/source/content/api_reference/secrets.rst +495 -0
  27. pygeai/_docs/source/content/api_reference/usage_limits.rst +390 -0
  28. pygeai/_docs/source/content/api_reference.rst +58 -0
  29. pygeai/_docs/source/content/authentication.rst +295 -0
  30. pygeai/_docs/source/content/chat_gui.rst +121 -0
  31. pygeai/_docs/source/content/cli.rst +203 -0
  32. pygeai/_docs/source/content/debugger.rst +651 -0
  33. pygeai/_docs/source/content/intro.rst +67 -0
  34. pygeai/_docs/source/content/migration.rst +929 -0
  35. pygeai/_docs/source/content/modules.rst +7 -0
  36. pygeai/_docs/source/content/quickstart.rst +143 -0
  37. pygeai/_docs/source/content/samples.rst +394 -0
  38. pygeai/_docs/source/index.rst +75 -0
  39. pygeai/_docs/source/modules.rst +7 -0
  40. pygeai/_docs/source/pygeai.admin.rst +29 -0
  41. pygeai/_docs/source/pygeai.analytics.rst +53 -0
  42. pygeai/_docs/source/pygeai.assistant.data.rst +21 -0
  43. pygeai/_docs/source/pygeai.assistant.data_analyst.rst +29 -0
  44. pygeai/_docs/source/pygeai.assistant.rag.rst +53 -0
  45. pygeai/_docs/source/pygeai.assistant.rst +55 -0
  46. pygeai/_docs/source/pygeai.auth.rst +29 -0
  47. pygeai/_docs/source/pygeai.chat.rst +69 -0
  48. pygeai/_docs/source/pygeai.cli.commands.flows.rst +10 -0
  49. pygeai/_docs/source/pygeai.cli.commands.lab.rst +53 -0
  50. pygeai/_docs/source/pygeai.cli.commands.rst +222 -0
  51. pygeai/_docs/source/pygeai.cli.rst +62 -0
  52. pygeai/_docs/source/pygeai.cli.texts.rst +21 -0
  53. pygeai/_docs/source/pygeai.core.base.rst +53 -0
  54. pygeai/_docs/source/pygeai.core.common.rst +37 -0
  55. pygeai/_docs/source/pygeai.core.embeddings.rst +61 -0
  56. pygeai/_docs/source/pygeai.core.feedback.rst +37 -0
  57. pygeai/_docs/source/pygeai.core.files.rst +61 -0
  58. pygeai/_docs/source/pygeai.core.llm.rst +29 -0
  59. pygeai/_docs/source/pygeai.core.plugins.rst +37 -0
  60. pygeai/_docs/source/pygeai.core.rerank.rst +53 -0
  61. pygeai/_docs/source/pygeai.core.rst +63 -0
  62. pygeai/_docs/source/pygeai.core.secrets.rst +29 -0
  63. pygeai/_docs/source/pygeai.core.services.llm.rst +29 -0
  64. pygeai/_docs/source/pygeai.core.services.rst +37 -0
  65. pygeai/_docs/source/pygeai.core.utils.rst +37 -0
  66. pygeai/_docs/source/pygeai.dbg.rst +21 -0
  67. pygeai/_docs/source/pygeai.evaluation.dataset.rst +29 -0
  68. pygeai/_docs/source/pygeai.evaluation.plan.rst +29 -0
  69. pygeai/_docs/source/pygeai.evaluation.result.rst +29 -0
  70. pygeai/_docs/source/pygeai.evaluation.rst +31 -0
  71. pygeai/_docs/source/pygeai.flows.rst +29 -0
  72. pygeai/_docs/source/pygeai.gam.rst +29 -0
  73. pygeai/_docs/source/pygeai.health.rst +29 -0
  74. pygeai/_docs/source/pygeai.lab.agents.rst +37 -0
  75. pygeai/_docs/source/pygeai.lab.processes.rst +37 -0
  76. pygeai/_docs/source/pygeai.lab.rst +65 -0
  77. pygeai/_docs/source/pygeai.lab.spec.rst +29 -0
  78. pygeai/_docs/source/pygeai.lab.strategies.rst +37 -0
  79. pygeai/_docs/source/pygeai.lab.tools.rst +37 -0
  80. pygeai/_docs/source/pygeai.man.man1.rst +10 -0
  81. pygeai/_docs/source/pygeai.man.rst +18 -0
  82. pygeai/_docs/source/pygeai.migration.rst +29 -0
  83. pygeai/_docs/source/pygeai.organization.limits.rst +45 -0
  84. pygeai/_docs/source/pygeai.organization.rst +61 -0
  85. pygeai/_docs/source/pygeai.proxy.rst +53 -0
  86. pygeai/_docs/source/pygeai.rst +35 -0
  87. pygeai/_docs/source/pygeai.tests.admin.rst +21 -0
  88. pygeai/_docs/source/pygeai.tests.analytics.rst +45 -0
  89. pygeai/_docs/source/pygeai.tests.assistants.rag.rst +37 -0
  90. pygeai/_docs/source/pygeai.tests.assistants.rst +45 -0
  91. pygeai/_docs/source/pygeai.tests.auth.rst +29 -0
  92. pygeai/_docs/source/pygeai.tests.chat.rst +45 -0
  93. pygeai/_docs/source/pygeai.tests.cli.commands.lab.rst +37 -0
  94. pygeai/_docs/source/pygeai.tests.cli.commands.rst +165 -0
  95. pygeai/_docs/source/pygeai.tests.cli.docker.rst +10 -0
  96. pygeai/_docs/source/pygeai.tests.cli.rst +46 -0
  97. pygeai/_docs/source/pygeai.tests.core.base.data.rst +29 -0
  98. pygeai/_docs/source/pygeai.tests.core.base.rst +45 -0
  99. pygeai/_docs/source/pygeai.tests.core.common.data.rst +10 -0
  100. pygeai/_docs/source/pygeai.tests.core.common.rst +37 -0
  101. pygeai/_docs/source/pygeai.tests.core.embeddings.rst +37 -0
  102. pygeai/_docs/source/pygeai.tests.core.feedback.rst +21 -0
  103. pygeai/_docs/source/pygeai.tests.core.files.rst +53 -0
  104. pygeai/_docs/source/pygeai.tests.core.llm.rst +21 -0
  105. pygeai/_docs/source/pygeai.tests.core.plugins.rst +21 -0
  106. pygeai/_docs/source/pygeai.tests.core.rerank.rst +37 -0
  107. pygeai/_docs/source/pygeai.tests.core.rst +39 -0
  108. pygeai/_docs/source/pygeai.tests.core.secrets.rst +21 -0
  109. pygeai/_docs/source/pygeai.tests.core.services.rst +21 -0
  110. pygeai/_docs/source/pygeai.tests.core.utils.rst +21 -0
  111. pygeai/_docs/source/pygeai.tests.dbg.rst +21 -0
  112. pygeai/_docs/source/pygeai.tests.evaluation.dataset.rst +21 -0
  113. pygeai/_docs/source/pygeai.tests.evaluation.plan.rst +21 -0
  114. pygeai/_docs/source/pygeai.tests.evaluation.result.rst +21 -0
  115. pygeai/_docs/source/pygeai.tests.evaluation.rst +20 -0
  116. pygeai/_docs/source/pygeai.tests.gam.rst +21 -0
  117. pygeai/_docs/source/pygeai.tests.health.rst +21 -0
  118. pygeai/_docs/source/pygeai.tests.integration.assistants.rag.rst +21 -0
  119. pygeai/_docs/source/pygeai.tests.integration.assistants.rst +18 -0
  120. pygeai/_docs/source/pygeai.tests.integration.chat.rst +21 -0
  121. pygeai/_docs/source/pygeai.tests.integration.lab.agents.rst +69 -0
  122. pygeai/_docs/source/pygeai.tests.integration.lab.processes.rst +77 -0
  123. pygeai/_docs/source/pygeai.tests.integration.lab.reasoning_strategies.rst +37 -0
  124. pygeai/_docs/source/pygeai.tests.integration.lab.rst +21 -0
  125. pygeai/_docs/source/pygeai.tests.integration.lab.tools.rst +77 -0
  126. pygeai/_docs/source/pygeai.tests.integration.rst +20 -0
  127. pygeai/_docs/source/pygeai.tests.lab.agents.rst +29 -0
  128. pygeai/_docs/source/pygeai.tests.lab.processes.rst +29 -0
  129. pygeai/_docs/source/pygeai.tests.lab.rst +49 -0
  130. pygeai/_docs/source/pygeai.tests.lab.spec.rst +29 -0
  131. pygeai/_docs/source/pygeai.tests.lab.strategies.rst +29 -0
  132. pygeai/_docs/source/pygeai.tests.lab.tools.rst +29 -0
  133. pygeai/_docs/source/pygeai.tests.migration.rst +29 -0
  134. pygeai/_docs/source/pygeai.tests.organization.limits.rst +29 -0
  135. pygeai/_docs/source/pygeai.tests.organization.rst +53 -0
  136. pygeai/_docs/source/pygeai.tests.proxy.rst +61 -0
  137. pygeai/_docs/source/pygeai.tests.rst +33 -0
  138. pygeai/admin/clients.py +14 -11
  139. pygeai/admin/endpoints.py +2 -2
  140. pygeai/analytics/clients.py +505 -0
  141. pygeai/analytics/endpoints.py +35 -0
  142. pygeai/analytics/managers.py +606 -0
  143. pygeai/analytics/mappers.py +207 -0
  144. pygeai/analytics/responses.py +240 -0
  145. pygeai/assistant/clients.py +48 -57
  146. pygeai/assistant/data/__init__.py +0 -0
  147. pygeai/assistant/data/clients.py +15 -0
  148. pygeai/assistant/data_analyst/__init__.py +0 -0
  149. pygeai/assistant/data_analyst/clients.py +75 -0
  150. pygeai/assistant/data_analyst/endpoints.py +2 -0
  151. pygeai/assistant/endpoints.py +0 -2
  152. pygeai/assistant/managers.py +738 -0
  153. pygeai/assistant/mappers.py +153 -0
  154. pygeai/assistant/rag/clients.py +132 -21
  155. pygeai/assistant/rag/mappers.py +228 -0
  156. pygeai/assistant/rag/models.py +396 -0
  157. pygeai/assistant/rag/responses.py +10 -0
  158. pygeai/auth/__init__.py +0 -0
  159. pygeai/auth/clients.py +129 -0
  160. pygeai/auth/endpoints.py +6 -0
  161. pygeai/chat/clients.py +406 -31
  162. pygeai/chat/endpoints.py +3 -0
  163. pygeai/chat/iris.py +17 -0
  164. pygeai/chat/managers.py +64 -0
  165. pygeai/chat/session.py +38 -0
  166. pygeai/chat/settings.py +6 -0
  167. pygeai/chat/ui.py +678 -0
  168. pygeai/cli/__init__.py +0 -1
  169. pygeai/cli/commands/admin.py +9 -12
  170. pygeai/cli/commands/analytics.py +533 -0
  171. pygeai/cli/commands/assistant.py +11 -11
  172. pygeai/cli/commands/auth.py +299 -0
  173. pygeai/cli/commands/base.py +201 -7
  174. pygeai/cli/commands/chat.py +875 -14
  175. pygeai/cli/commands/common.py +30 -26
  176. pygeai/cli/commands/configuration.py +84 -9
  177. pygeai/cli/commands/docs.py +105 -0
  178. pygeai/cli/commands/embeddings.py +187 -0
  179. pygeai/cli/commands/evaluation.py +2069 -0
  180. pygeai/cli/commands/feedback.py +93 -0
  181. pygeai/cli/commands/files.py +312 -0
  182. pygeai/cli/commands/flows/__init__.py +0 -0
  183. pygeai/cli/commands/gam.py +349 -0
  184. pygeai/cli/commands/lab/__init__.py +0 -0
  185. pygeai/cli/commands/lab/ai_lab.py +4110 -0
  186. pygeai/cli/commands/lab/common.py +135 -0
  187. pygeai/cli/commands/lab/options.py +8 -0
  188. pygeai/cli/commands/lab/spec.py +273 -0
  189. pygeai/cli/commands/lab/utils.py +13 -0
  190. pygeai/cli/commands/llm.py +164 -0
  191. pygeai/cli/commands/migrate.py +1198 -0
  192. pygeai/cli/commands/options.py +86 -0
  193. pygeai/cli/commands/organization.py +560 -98
  194. pygeai/cli/commands/rag.py +306 -10
  195. pygeai/cli/commands/rerank.py +108 -0
  196. pygeai/cli/commands/secrets.py +357 -0
  197. pygeai/cli/commands/usage_limits.py +583 -0
  198. pygeai/cli/commands/validators.py +209 -0
  199. pygeai/cli/commands/version.py +44 -0
  200. pygeai/cli/error_handler.py +151 -0
  201. pygeai/cli/geai.py +171 -30
  202. pygeai/cli/geai_proxy.py +318 -0
  203. pygeai/cli/install_man.py +107 -0
  204. pygeai/cli/parsers.py +78 -25
  205. pygeai/cli/texts/help.py +712 -55
  206. pygeai/core/__init__.py +9 -1
  207. pygeai/core/base/clients.py +61 -10
  208. pygeai/core/base/mappers.py +208 -30
  209. pygeai/core/base/models.py +8 -308
  210. pygeai/core/base/responses.py +18 -1
  211. pygeai/core/base/session.py +110 -17
  212. pygeai/core/common/config.py +98 -16
  213. pygeai/core/common/decorators.py +44 -0
  214. pygeai/core/common/exceptions.py +104 -4
  215. pygeai/core/embeddings/__init__.py +19 -0
  216. pygeai/core/embeddings/clients.py +93 -0
  217. pygeai/core/embeddings/endpoints.py +1 -0
  218. pygeai/core/embeddings/managers.py +62 -0
  219. pygeai/core/embeddings/mappers.py +52 -0
  220. pygeai/core/embeddings/models.py +14 -0
  221. pygeai/core/embeddings/responses.py +31 -0
  222. pygeai/core/feedback/__init__.py +0 -0
  223. pygeai/core/feedback/clients.py +50 -0
  224. pygeai/core/feedback/endpoints.py +1 -0
  225. pygeai/core/feedback/models.py +10 -0
  226. pygeai/core/files/__init__.py +0 -0
  227. pygeai/core/files/clients.py +156 -0
  228. pygeai/core/files/endpoints.py +5 -0
  229. pygeai/core/files/managers.py +224 -0
  230. pygeai/core/files/mappers.py +44 -0
  231. pygeai/core/files/models.py +24 -0
  232. pygeai/core/files/responses.py +19 -0
  233. pygeai/core/handlers.py +32 -0
  234. pygeai/core/llm/__init__.py +0 -0
  235. pygeai/core/llm/clients.py +53 -0
  236. pygeai/core/llm/endpoints.py +4 -0
  237. pygeai/core/models.py +799 -0
  238. pygeai/core/plugins/__init__.py +0 -0
  239. pygeai/core/plugins/clients.py +32 -0
  240. pygeai/core/plugins/endpoints.py +1 -0
  241. pygeai/core/plugins/models.py +86 -0
  242. pygeai/core/rerank/__init__.py +0 -0
  243. pygeai/core/rerank/clients.py +35 -0
  244. pygeai/core/rerank/endpoints.py +1 -0
  245. pygeai/core/rerank/managers.py +47 -0
  246. pygeai/core/rerank/mappers.py +23 -0
  247. pygeai/core/rerank/models.py +27 -0
  248. pygeai/core/responses.py +104 -0
  249. pygeai/core/secrets/__init__.py +0 -0
  250. pygeai/core/secrets/clients.py +212 -0
  251. pygeai/core/secrets/endpoints.py +7 -0
  252. pygeai/core/services/llm/__init__.py +0 -0
  253. pygeai/core/services/llm/model.py +186 -0
  254. pygeai/core/services/llm/providers.py +15 -0
  255. pygeai/core/services/response.py +18 -0
  256. pygeai/core/services/rest.py +311 -89
  257. pygeai/core/utils/__init__.py +0 -0
  258. pygeai/core/utils/console.py +83 -0
  259. pygeai/core/utils/parsers.py +32 -0
  260. pygeai/core/utils/validators.py +10 -0
  261. pygeai/dbg/__init__.py +3 -0
  262. pygeai/dbg/debugger.py +870 -0
  263. pygeai/evaluation/__init__.py +0 -0
  264. pygeai/evaluation/clients.py +19 -0
  265. pygeai/evaluation/dataset/__init__.py +0 -0
  266. pygeai/evaluation/dataset/clients.py +514 -0
  267. pygeai/evaluation/dataset/endpoints.py +26 -0
  268. pygeai/evaluation/plan/__init__.py +0 -0
  269. pygeai/evaluation/plan/clients.py +302 -0
  270. pygeai/evaluation/plan/endpoints.py +16 -0
  271. pygeai/evaluation/result/__init__.py +0 -0
  272. pygeai/evaluation/result/clients.py +70 -0
  273. pygeai/evaluation/result/endpoints.py +2 -0
  274. pygeai/flows/__init__.py +0 -0
  275. pygeai/flows/endpoints.py +362 -0
  276. pygeai/flows/models.py +1304 -0
  277. pygeai/gam/__init__.py +0 -0
  278. pygeai/gam/clients.py +178 -0
  279. pygeai/gam/endpoints.py +4 -0
  280. pygeai/health/__init__.py +0 -0
  281. pygeai/health/clients.py +24 -0
  282. pygeai/health/endpoints.py +1 -0
  283. pygeai/lab/__init__.py +0 -0
  284. pygeai/lab/agents/__init__.py +0 -0
  285. pygeai/lab/agents/clients.py +426 -0
  286. pygeai/lab/agents/endpoints.py +12 -0
  287. pygeai/lab/agents/mappers.py +319 -0
  288. pygeai/lab/clients.py +24 -0
  289. pygeai/lab/constants.py +3 -0
  290. pygeai/lab/managers.py +1558 -0
  291. pygeai/lab/models.py +1719 -0
  292. pygeai/lab/processes/__init__.py +0 -0
  293. pygeai/lab/processes/clients.py +1051 -0
  294. pygeai/lab/processes/endpoints.py +26 -0
  295. pygeai/lab/processes/mappers.py +395 -0
  296. pygeai/lab/runners.py +90 -0
  297. pygeai/lab/spec/__init__.py +0 -0
  298. pygeai/lab/spec/loader.py +24 -0
  299. pygeai/lab/spec/parsers.py +39 -0
  300. pygeai/lab/strategies/__init__.py +0 -0
  301. pygeai/lab/strategies/clients.py +212 -0
  302. pygeai/lab/strategies/endpoints.py +5 -0
  303. pygeai/lab/strategies/mappers.py +58 -0
  304. pygeai/lab/tools/__init__.py +0 -0
  305. pygeai/lab/tools/clients.py +465 -0
  306. pygeai/lab/tools/endpoints.py +13 -0
  307. pygeai/lab/tools/mappers.py +131 -0
  308. pygeai/man/__init__.py +1 -0
  309. pygeai/man/man1/__init__.py +1 -0
  310. pygeai/man/man1/geai-proxy.1 +246 -0
  311. pygeai/man/man1/geai.1 +2615 -0
  312. pygeai/migration/__init__.py +33 -0
  313. pygeai/migration/strategies.py +603 -0
  314. pygeai/migration/tools.py +180 -0
  315. pygeai/organization/clients.py +246 -18
  316. pygeai/organization/endpoints.py +17 -8
  317. pygeai/organization/limits/__init__.py +0 -0
  318. pygeai/organization/limits/clients.py +281 -0
  319. pygeai/organization/limits/endpoints.py +15 -0
  320. pygeai/organization/limits/managers.py +331 -0
  321. pygeai/organization/limits/mappers.py +21 -0
  322. pygeai/organization/managers.py +537 -0
  323. pygeai/organization/mappers.py +111 -46
  324. pygeai/organization/responses.py +61 -11
  325. pygeai/proxy/__init__.py +0 -0
  326. pygeai/proxy/clients.py +216 -0
  327. pygeai/proxy/config.py +128 -0
  328. pygeai/proxy/managers.py +232 -0
  329. pygeai/proxy/servers.py +304 -0
  330. pygeai/proxy/tool.py +69 -0
  331. pygeai/tests/admin/__init__.py +0 -0
  332. pygeai/tests/admin/test_clients.py +148 -0
  333. pygeai/tests/analytics/__init__.py +0 -0
  334. pygeai/tests/analytics/test_clients.py +86 -0
  335. pygeai/tests/analytics/test_managers.py +94 -0
  336. pygeai/tests/analytics/test_mappers.py +84 -0
  337. pygeai/tests/analytics/test_responses.py +73 -0
  338. pygeai/tests/assistants/rag/__init__.py +0 -0
  339. pygeai/tests/assistants/rag/test_clients.py +346 -0
  340. pygeai/tests/assistants/rag/test_mappers.py +189 -0
  341. pygeai/tests/assistants/rag/test_models.py +292 -0
  342. pygeai/tests/assistants/test_clients.py +176 -80
  343. pygeai/tests/assistants/test_managers.py +198 -0
  344. pygeai/tests/assistants/test_mappers.py +111 -0
  345. pygeai/tests/auth/__init__.py +0 -0
  346. pygeai/tests/auth/test_clients.py +289 -0
  347. pygeai/tests/auth/test_oauth.py +172 -0
  348. pygeai/tests/auth/test_session_logging.py +150 -0
  349. pygeai/tests/chat/__init__.py +0 -0
  350. pygeai/tests/chat/test_clients.py +393 -0
  351. pygeai/tests/chat/test_iris.py +38 -0
  352. pygeai/tests/chat/test_session.py +62 -0
  353. pygeai/tests/chat/test_ui.py +224 -0
  354. pygeai/tests/cli/__init__.py +0 -0
  355. pygeai/tests/cli/commands/__init__.py +0 -0
  356. pygeai/tests/cli/commands/lab/__init__.py +0 -0
  357. pygeai/tests/cli/commands/lab/test_ai_lab.py +786 -0
  358. pygeai/tests/cli/commands/lab/test_common.py +208 -0
  359. pygeai/tests/cli/commands/lab/test_spec.py +246 -0
  360. pygeai/tests/cli/commands/test_assistant.py +202 -0
  361. pygeai/tests/cli/commands/test_chat.py +130 -0
  362. pygeai/tests/cli/commands/test_common.py +350 -0
  363. pygeai/tests/cli/commands/test_embeddings.py +132 -0
  364. pygeai/tests/cli/commands/test_evaluation.py +656 -0
  365. pygeai/tests/cli/commands/test_feedback.py +65 -0
  366. pygeai/tests/cli/commands/test_files.py +161 -0
  367. pygeai/tests/cli/commands/test_gam.py +201 -0
  368. pygeai/tests/cli/commands/test_llm.py +114 -0
  369. pygeai/tests/cli/commands/test_migrate.py +176 -0
  370. pygeai/tests/cli/commands/test_organization.py +276 -0
  371. pygeai/tests/cli/commands/test_rag.py +266 -0
  372. pygeai/tests/cli/commands/test_rerank.py +110 -0
  373. pygeai/tests/cli/commands/test_secrets.py +171 -0
  374. pygeai/tests/cli/commands/test_show_help.py +41 -0
  375. pygeai/tests/cli/commands/test_usage_limits.py +412 -0
  376. pygeai/tests/cli/commands/test_validators.py +160 -0
  377. pygeai/tests/cli/commands/test_version.py +81 -0
  378. pygeai/tests/cli/docker/__init__.py +0 -0
  379. pygeai/tests/cli/test_credentials_flag.py +316 -0
  380. pygeai/tests/cli/test_error_handler.py +225 -0
  381. pygeai/tests/cli/test_geai_driver.py +154 -0
  382. pygeai/tests/cli/test_parsers.py +154 -0
  383. pygeai/tests/core/base/__init__.py +0 -0
  384. pygeai/tests/core/base/data/__init__.py +0 -0
  385. pygeai/tests/core/base/data/mappers.py +117 -0
  386. pygeai/tests/core/base/data/models.py +312 -0
  387. pygeai/tests/core/base/test_mappers.py +569 -0
  388. pygeai/tests/core/base/test_models.py +261 -0
  389. pygeai/tests/core/base/test_responses.py +53 -0
  390. pygeai/tests/core/common/__init__.py +0 -0
  391. pygeai/tests/core/common/data/__init__.py +0 -0
  392. pygeai/tests/core/common/test_config.py +186 -0
  393. pygeai/tests/core/common/test_decorators.py +69 -0
  394. pygeai/tests/core/embeddings/__init__.py +0 -0
  395. pygeai/tests/core/embeddings/test_clients.py +225 -0
  396. pygeai/tests/core/embeddings/test_managers.py +171 -0
  397. pygeai/tests/core/embeddings/test_mappers.py +142 -0
  398. pygeai/tests/core/feedback/__init__.py +0 -0
  399. pygeai/tests/core/feedback/test_clients.py +64 -0
  400. pygeai/tests/core/files/__init__.py +0 -0
  401. pygeai/tests/core/files/test_clients.py +128 -0
  402. pygeai/tests/core/files/test_managers.py +219 -0
  403. pygeai/tests/core/files/test_mappers.py +137 -0
  404. pygeai/tests/core/files/test_models.py +103 -0
  405. pygeai/tests/core/files/test_responses.py +122 -0
  406. pygeai/tests/core/llm/__init__.py +0 -0
  407. pygeai/tests/core/llm/test_clients.py +142 -0
  408. pygeai/tests/core/plugins/__init__.py +0 -0
  409. pygeai/tests/core/plugins/test_clients.py +66 -0
  410. pygeai/tests/core/rerank/__init__.py +0 -0
  411. pygeai/tests/core/rerank/test_clients.py +76 -0
  412. pygeai/tests/core/rerank/test_managers.py +99 -0
  413. pygeai/tests/core/rerank/test_mappers.py +54 -0
  414. pygeai/tests/core/secrets/__init__.py +0 -0
  415. pygeai/tests/core/secrets/test_clients.py +264 -0
  416. pygeai/tests/core/services/__init__.py +0 -0
  417. pygeai/tests/core/services/test_rest.py +273 -0
  418. pygeai/tests/core/test_handlers.py +66 -0
  419. pygeai/tests/core/utils/__init__.py +0 -0
  420. pygeai/tests/core/utils/test_console.py +80 -0
  421. pygeai/tests/dbg/__init__.py +0 -0
  422. pygeai/tests/dbg/test_debugger.py +591 -0
  423. pygeai/tests/evaluation/__init__.py +0 -0
  424. pygeai/tests/evaluation/dataset/__init__.py +0 -0
  425. pygeai/tests/evaluation/dataset/test_clients.py +265 -0
  426. pygeai/tests/evaluation/plan/__init__.py +0 -0
  427. pygeai/tests/evaluation/plan/test_clients.py +195 -0
  428. pygeai/tests/evaluation/result/__init__.py +0 -0
  429. pygeai/tests/evaluation/result/test_clients.py +66 -0
  430. pygeai/tests/gam/__init__.py +0 -0
  431. pygeai/tests/gam/test_clients.py +195 -0
  432. pygeai/tests/health/__init__.py +0 -0
  433. pygeai/tests/health/test_clients.py +41 -0
  434. pygeai/tests/integration/__init__.py +0 -0
  435. pygeai/tests/integration/assistants/__init__.py +0 -0
  436. pygeai/tests/integration/assistants/rag/__init__.py +0 -0
  437. pygeai/tests/integration/assistants/rag/test_create_rag.py +91 -0
  438. pygeai/tests/integration/chat/__init__.py +0 -0
  439. pygeai/tests/integration/chat/test_generate_image.py +158 -0
  440. pygeai/tests/integration/lab/__init__.py +0 -0
  441. pygeai/tests/integration/lab/agents/__init__.py +0 -0
  442. pygeai/tests/integration/lab/agents/test_agents_list.py +106 -0
  443. pygeai/tests/integration/lab/agents/test_create_agent.py +319 -0
  444. pygeai/tests/integration/lab/agents/test_create_sharing_link.py +70 -0
  445. pygeai/tests/integration/lab/agents/test_delete_agent.py +75 -0
  446. pygeai/tests/integration/lab/agents/test_get_agent.py +94 -0
  447. pygeai/tests/integration/lab/agents/test_publish_agent_revision.py +127 -0
  448. pygeai/tests/integration/lab/agents/test_update_agent.py +250 -0
  449. pygeai/tests/integration/lab/processes/__init__.py +0 -0
  450. pygeai/tests/integration/lab/processes/test_create_process.py +345 -0
  451. pygeai/tests/integration/lab/processes/test_create_task.py +211 -0
  452. pygeai/tests/integration/lab/processes/test_delete_process.py +111 -0
  453. pygeai/tests/integration/lab/processes/test_get_process.py +201 -0
  454. pygeai/tests/integration/lab/processes/test_list_process_instances.py +91 -0
  455. pygeai/tests/integration/lab/processes/test_list_processes.py +138 -0
  456. pygeai/tests/integration/lab/processes/test_publish_process_revision.py +232 -0
  457. pygeai/tests/integration/lab/processes/test_update_process.py +289 -0
  458. pygeai/tests/integration/lab/reasoning_strategies/__init__.py +0 -0
  459. pygeai/tests/integration/lab/reasoning_strategies/test_get_reasoning_strategy.py +70 -0
  460. pygeai/tests/integration/lab/reasoning_strategies/test_list_reasoning_strategies.py +93 -0
  461. pygeai/tests/integration/lab/reasoning_strategies/test_update_reasoning_strategy.py +149 -0
  462. pygeai/tests/integration/lab/tools/__init__.py +0 -0
  463. pygeai/tests/integration/lab/tools/test_create_tool.py +288 -0
  464. pygeai/tests/integration/lab/tools/test_delete_tool.py +87 -0
  465. pygeai/tests/integration/lab/tools/test_get_parameter.py +98 -0
  466. pygeai/tests/integration/lab/tools/test_get_tool.py +91 -0
  467. pygeai/tests/integration/lab/tools/test_list_tools.py +106 -0
  468. pygeai/tests/integration/lab/tools/test_publish_tool_revision.py +119 -0
  469. pygeai/tests/integration/lab/tools/test_set_parameter.py +114 -0
  470. pygeai/tests/integration/lab/tools/test_update_tool.py +267 -0
  471. pygeai/tests/lab/__init__.py +0 -0
  472. pygeai/tests/lab/agents/__init__.py +0 -0
  473. pygeai/tests/lab/agents/test_clients.py +481 -0
  474. pygeai/tests/lab/agents/test_mappers.py +440 -0
  475. pygeai/tests/lab/processes/__init__.py +0 -0
  476. pygeai/tests/lab/processes/test_clients.py +1416 -0
  477. pygeai/tests/lab/processes/test_mappers.py +1092 -0
  478. pygeai/tests/lab/spec/__init__.py +0 -0
  479. pygeai/tests/lab/spec/test_loader.py +59 -0
  480. pygeai/tests/lab/spec/test_parsers.py +182 -0
  481. pygeai/tests/lab/strategies/__init__.py +0 -0
  482. pygeai/tests/lab/strategies/test_clients.py +241 -0
  483. pygeai/tests/lab/strategies/test_mappers.py +132 -0
  484. pygeai/tests/lab/test_managers.py +553 -0
  485. pygeai/tests/lab/test_mappers.py +245 -0
  486. pygeai/tests/lab/test_models.py +1154 -0
  487. pygeai/tests/lab/tools/__init__.py +0 -0
  488. pygeai/tests/lab/tools/test_clients.py +521 -0
  489. pygeai/tests/lab/tools/test_mappers.py +198 -0
  490. pygeai/tests/migration/__init__.py +0 -0
  491. pygeai/tests/migration/test_strategies.py +405 -0
  492. pygeai/tests/migration/test_tools.py +159 -0
  493. pygeai/tests/organization/limits/__init__.py +0 -0
  494. pygeai/tests/organization/limits/test_clients.py +567 -0
  495. pygeai/tests/organization/limits/test_managers.py +402 -0
  496. pygeai/tests/organization/test_clients.py +615 -64
  497. pygeai/tests/organization/test_managers.py +424 -0
  498. pygeai/tests/organization/test_mappers.py +153 -0
  499. pygeai/tests/organization/test_responses.py +137 -0
  500. pygeai/tests/proxy/__init__.py +1 -0
  501. pygeai/tests/proxy/test_clients.py +397 -0
  502. pygeai/tests/proxy/test_config.py +171 -0
  503. pygeai/tests/proxy/test_integration.py +305 -0
  504. pygeai/tests/proxy/test_managers.py +312 -0
  505. pygeai/tests/proxy/test_servers.py +387 -0
  506. pygeai/tests/proxy/test_tool.py +176 -0
  507. pygeai/tests/snippets/__init__.py +0 -0
  508. pygeai/tests/snippets/analytics/__init__.py +0 -0
  509. pygeai/tests/snippets/analytics/get_agent_usage_per_user.py +16 -0
  510. pygeai/tests/snippets/analytics/get_agents_created_and_modified.py +11 -0
  511. pygeai/tests/snippets/analytics/get_average_cost_per_request.py +10 -0
  512. pygeai/tests/snippets/analytics/get_overall_error_rate.py +10 -0
  513. pygeai/tests/snippets/analytics/get_top_10_agents_by_requests.py +12 -0
  514. pygeai/tests/snippets/analytics/get_total_active_users.py +10 -0
  515. pygeai/tests/snippets/analytics/get_total_cost.py +10 -0
  516. pygeai/tests/snippets/analytics/get_total_requests_per_day.py +12 -0
  517. pygeai/tests/snippets/analytics/get_total_tokens.py +12 -0
  518. pygeai/tests/snippets/assistants/__init__.py +0 -0
  519. pygeai/tests/snippets/assistants/create_chat_assistant.py +54 -0
  520. pygeai/tests/snippets/assistants/create_text_assistant.py +51 -0
  521. pygeai/tests/snippets/assistants/data_analyst/__init__.py +0 -0
  522. pygeai/tests/snippets/assistants/data_analyst/extend_and_check.py +100 -0
  523. pygeai/tests/snippets/assistants/data_analyst/extend_dataset.py +9 -0
  524. pygeai/tests/snippets/assistants/data_analyst/get_status.py +9 -0
  525. pygeai/tests/snippets/assistants/file_summarizer_assistant.py +149 -0
  526. pygeai/tests/snippets/assistants/get_assistant_data.py +8 -0
  527. pygeai/tests/snippets/assistants/get_assistant_list.py +7 -0
  528. pygeai/tests/snippets/assistants/rag/__init__.py +0 -0
  529. pygeai/tests/snippets/assistants/rag/create_rag_assistant.py +65 -0
  530. pygeai/tests/snippets/assistants/rag/delete_al_documents.py +7 -0
  531. pygeai/tests/snippets/assistants/rag/delete_document.py +10 -0
  532. pygeai/tests/snippets/assistants/rag/delete_rag_assistant.py +8 -0
  533. pygeai/tests/snippets/assistants/rag/get_document.py +10 -0
  534. pygeai/tests/snippets/assistants/rag/get_documents.py +7 -0
  535. pygeai/tests/snippets/assistants/rag/get_rag_assistant_data.py +8 -0
  536. pygeai/tests/snippets/assistants/rag/update_rag_assistant.py +48 -0
  537. pygeai/tests/snippets/assistants/rag/upload_document.py +19 -0
  538. pygeai/tests/snippets/assistants/send_feedback.py +14 -0
  539. pygeai/tests/snippets/assistants/update_chat_assistant.py +63 -0
  540. pygeai/tests/snippets/auth/__init__.py +0 -0
  541. pygeai/tests/snippets/chat/__init__.py +0 -0
  542. pygeai/tests/snippets/chat/cancel_request.py +7 -0
  543. pygeai/tests/snippets/chat/chat_completion.py +28 -0
  544. pygeai/tests/snippets/chat/chat_completion_1.py +40 -0
  545. pygeai/tests/snippets/chat/chat_completion_2.py +60 -0
  546. pygeai/tests/snippets/chat/chat_completion_3.py +27 -0
  547. pygeai/tests/snippets/chat/chat_completion_4.py +67 -0
  548. pygeai/tests/snippets/chat/chat_completion_streaming.py +63 -0
  549. pygeai/tests/snippets/chat/chat_completion_with_reasoning_effort.py +18 -0
  550. pygeai/tests/snippets/chat/get_request_status.py +7 -0
  551. pygeai/tests/snippets/chat/get_response.py +15 -0
  552. pygeai/tests/snippets/chat/get_response_complete_example.py +67 -0
  553. pygeai/tests/snippets/chat/get_response_streaming.py +20 -0
  554. pygeai/tests/snippets/chat/get_response_with_files.py +16 -0
  555. pygeai/tests/snippets/chat/get_response_with_instructions.py +19 -0
  556. pygeai/tests/snippets/chat/get_response_with_metadata.py +24 -0
  557. pygeai/tests/snippets/chat/get_response_with_parallel_tools.py +58 -0
  558. pygeai/tests/snippets/chat/get_response_with_reasoning.py +21 -0
  559. pygeai/tests/snippets/chat/get_response_with_store.py +38 -0
  560. pygeai/tests/snippets/chat/get_response_with_tools.py +36 -0
  561. pygeai/tests/snippets/chat/get_response_with_truncation.py +24 -0
  562. pygeai/tests/snippets/chat/send_chat_request.py +33 -0
  563. pygeai/tests/snippets/dbg/__init__.py +0 -0
  564. pygeai/tests/snippets/dbg/basic_debugging.py +32 -0
  565. pygeai/tests/snippets/dbg/breakpoint_management.py +48 -0
  566. pygeai/tests/snippets/dbg/file_debugging.py +72 -0
  567. pygeai/tests/snippets/dbg/module_debugging.py +61 -0
  568. pygeai/tests/snippets/dbg/stack_navigation.py +45 -0
  569. pygeai/tests/snippets/dbg/stepping_example.py +40 -0
  570. pygeai/tests/snippets/embeddings/__init__.py +0 -0
  571. pygeai/tests/snippets/embeddings/cache_example.py +31 -0
  572. pygeai/tests/snippets/embeddings/cohere_example.py +41 -0
  573. pygeai/tests/snippets/embeddings/generate_embeddings.py +26 -0
  574. pygeai/tests/snippets/embeddings/openai_base64_example.py +27 -0
  575. pygeai/tests/snippets/embeddings/openai_example.py +30 -0
  576. pygeai/tests/snippets/embeddings/similarity_example.py +42 -0
  577. pygeai/tests/snippets/evaluation/__init__.py +0 -0
  578. pygeai/tests/snippets/evaluation/dataset/__init__.py +0 -0
  579. pygeai/tests/snippets/evaluation/dataset/complete_workflow_example.py +195 -0
  580. pygeai/tests/snippets/evaluation/dataset/create_dataset.py +26 -0
  581. pygeai/tests/snippets/evaluation/dataset/create_dataset_from_file.py +11 -0
  582. pygeai/tests/snippets/evaluation/dataset/create_dataset_row.py +17 -0
  583. pygeai/tests/snippets/evaluation/dataset/create_expected_source.py +18 -0
  584. pygeai/tests/snippets/evaluation/dataset/create_filter_variable.py +19 -0
  585. pygeai/tests/snippets/evaluation/dataset/delete_dataset.py +9 -0
  586. pygeai/tests/snippets/evaluation/dataset/delete_dataset_row.py +10 -0
  587. pygeai/tests/snippets/evaluation/dataset/delete_expected_source.py +15 -0
  588. pygeai/tests/snippets/evaluation/dataset/delete_filter_variable.py +15 -0
  589. pygeai/tests/snippets/evaluation/dataset/get_dataset.py +9 -0
  590. pygeai/tests/snippets/evaluation/dataset/get_dataset_row.py +10 -0
  591. pygeai/tests/snippets/evaluation/dataset/get_expected_source.py +15 -0
  592. pygeai/tests/snippets/evaluation/dataset/get_filter_variable.py +15 -0
  593. pygeai/tests/snippets/evaluation/dataset/list_dataset_rows.py +9 -0
  594. pygeai/tests/snippets/evaluation/dataset/list_datasets.py +6 -0
  595. pygeai/tests/snippets/evaluation/dataset/list_expected_sources.py +10 -0
  596. pygeai/tests/snippets/evaluation/dataset/list_filter_variables.py +10 -0
  597. pygeai/tests/snippets/evaluation/dataset/update_dataset.py +15 -0
  598. pygeai/tests/snippets/evaluation/dataset/update_dataset_row.py +20 -0
  599. pygeai/tests/snippets/evaluation/dataset/update_expected_source.py +18 -0
  600. pygeai/tests/snippets/evaluation/dataset/update_filter_variable.py +19 -0
  601. pygeai/tests/snippets/evaluation/dataset/upload_dataset_rows_file.py +10 -0
  602. pygeai/tests/snippets/evaluation/plan/__init__.py +0 -0
  603. pygeai/tests/snippets/evaluation/plan/add_plan_system_metric.py +13 -0
  604. pygeai/tests/snippets/evaluation/plan/complete_workflow_example.py +136 -0
  605. pygeai/tests/snippets/evaluation/plan/create_evaluation_plan.py +24 -0
  606. pygeai/tests/snippets/evaluation/plan/create_rag_evaluation_plan.py +22 -0
  607. pygeai/tests/snippets/evaluation/plan/delete_evaluation_plan.py +9 -0
  608. pygeai/tests/snippets/evaluation/plan/delete_plan_system_metric.py +13 -0
  609. pygeai/tests/snippets/evaluation/plan/execute_evaluation_plan.py +11 -0
  610. pygeai/tests/snippets/evaluation/plan/get_evaluation_plan.py +9 -0
  611. pygeai/tests/snippets/evaluation/plan/get_plan_system_metric.py +13 -0
  612. pygeai/tests/snippets/evaluation/plan/get_system_metric.py +9 -0
  613. pygeai/tests/snippets/evaluation/plan/list_evaluation_plans.py +7 -0
  614. pygeai/tests/snippets/evaluation/plan/list_plan_system_metrics.py +9 -0
  615. pygeai/tests/snippets/evaluation/plan/list_system_metrics.py +7 -0
  616. pygeai/tests/snippets/evaluation/plan/update_evaluation_plan.py +22 -0
  617. pygeai/tests/snippets/evaluation/plan/update_plan_system_metric.py +14 -0
  618. pygeai/tests/snippets/evaluation/result/__init__.py +0 -0
  619. pygeai/tests/snippets/evaluation/result/complete_workflow_example.py +150 -0
  620. pygeai/tests/snippets/evaluation/result/get_evaluation_result.py +26 -0
  621. pygeai/tests/snippets/evaluation/result/list_evaluation_results.py +17 -0
  622. pygeai/tests/snippets/files/__init__.py +0 -0
  623. pygeai/tests/snippets/files/delete_file.py +9 -0
  624. pygeai/tests/snippets/files/get_file_content.py +10 -0
  625. pygeai/tests/snippets/files/get_file_data.py +9 -0
  626. pygeai/tests/snippets/files/get_file_list.py +6 -0
  627. pygeai/tests/snippets/files/upload_file.py +13 -0
  628. pygeai/tests/snippets/gam/__init__.py +0 -0
  629. pygeai/tests/snippets/gam/gam_access_token.py +87 -0
  630. pygeai/tests/snippets/lab/__init__.py +0 -0
  631. pygeai/tests/snippets/lab/agentic_flow_example_1.py +326 -0
  632. pygeai/tests/snippets/lab/agentic_flow_example_2.py +206 -0
  633. pygeai/tests/snippets/lab/agentic_flow_example_3.py +486 -0
  634. pygeai/tests/snippets/lab/agentic_flow_example_4.py +446 -0
  635. pygeai/tests/snippets/lab/agents/__init__.py +0 -0
  636. pygeai/tests/snippets/lab/agents/create_agent.py +48 -0
  637. pygeai/tests/snippets/lab/agents/create_agent_2.py +48 -0
  638. pygeai/tests/snippets/lab/agents/create_agent_edge_case.py +48 -0
  639. pygeai/tests/snippets/lab/agents/create_agent_with_permissions.py +39 -0
  640. pygeai/tests/snippets/lab/agents/create_agent_with_properties.py +46 -0
  641. pygeai/tests/snippets/lab/agents/create_agent_without_instructions.py +48 -0
  642. pygeai/tests/snippets/lab/agents/delete_agent.py +12 -0
  643. pygeai/tests/snippets/lab/agents/get_agent.py +24 -0
  644. pygeai/tests/snippets/lab/agents/get_agent_with_new_fields.py +62 -0
  645. pygeai/tests/snippets/lab/agents/get_sharing_link.py +13 -0
  646. pygeai/tests/snippets/lab/agents/list_agents.py +18 -0
  647. pygeai/tests/snippets/lab/agents/publish_agent_revision.py +12 -0
  648. pygeai/tests/snippets/lab/agents/update_agent.py +50 -0
  649. pygeai/tests/snippets/lab/agents/update_agent_properties.py +50 -0
  650. pygeai/tests/snippets/lab/assistant_to_agent.py +191 -0
  651. pygeai/tests/snippets/lab/crud_ui.py +462 -0
  652. pygeai/tests/snippets/lab/processes/__init__.py +0 -0
  653. pygeai/tests/snippets/lab/processes/create_process.py +24 -0
  654. pygeai/tests/snippets/lab/processes/create_task.py +8 -0
  655. pygeai/tests/snippets/lab/processes/jobs/__init__.py +0 -0
  656. pygeai/tests/snippets/lab/processes/jobs/list_jobs.py +21 -0
  657. pygeai/tests/snippets/lab/processes/kbs/__init__.py +0 -0
  658. pygeai/tests/snippets/lab/processes/kbs/create_kb.py +18 -0
  659. pygeai/tests/snippets/lab/processes/kbs/get_kb.py +26 -0
  660. pygeai/tests/snippets/lab/processes/kbs/list_kbs.py +30 -0
  661. pygeai/tests/snippets/lab/processes/kbs/try_all.py +73 -0
  662. pygeai/tests/snippets/lab/processes/list_processes.py +10 -0
  663. pygeai/tests/snippets/lab/runner_1.py +212 -0
  664. pygeai/tests/snippets/lab/samples/__init__.py +0 -0
  665. pygeai/tests/snippets/lab/samples/summarize_files.py +162 -0
  666. pygeai/tests/snippets/lab/strategies/__init__.py +0 -0
  667. pygeai/tests/snippets/lab/strategies/create_reasoning_strategy.py +22 -0
  668. pygeai/tests/snippets/lab/strategies/get_reasoning_strategy.py +10 -0
  669. pygeai/tests/snippets/lab/strategies/list_reasoning_strategies.py +16 -0
  670. pygeai/tests/snippets/lab/strategies/update_reasoning_strategy.py +26 -0
  671. pygeai/tests/snippets/lab/tools/__init__.py +0 -0
  672. pygeai/tests/snippets/lab/tools/create_tool.py +48 -0
  673. pygeai/tests/snippets/lab/tools/create_tool_edge_case.py +50 -0
  674. pygeai/tests/snippets/lab/tools/delete_tool.py +21 -0
  675. pygeai/tests/snippets/lab/tools/get_parameter.py +21 -0
  676. pygeai/tests/snippets/lab/tools/get_tool.py +22 -0
  677. pygeai/tests/snippets/lab/tools/list_tools.py +23 -0
  678. pygeai/tests/snippets/lab/tools/publish_tool_revision.py +13 -0
  679. pygeai/tests/snippets/lab/tools/set_parameters.py +33 -0
  680. pygeai/tests/snippets/lab/tools/update_tool.py +52 -0
  681. pygeai/tests/snippets/lab/use_cases/__init__.py +0 -0
  682. pygeai/tests/snippets/lab/use_cases/c_code_fixer_agent_flow.py +238 -0
  683. pygeai/tests/snippets/lab/use_cases/create_cli_expert.py +1640 -0
  684. pygeai/tests/snippets/lab/use_cases/create_lab_expert.py +4541 -0
  685. pygeai/tests/snippets/lab/use_cases/create_tool_headless_web_browser.py +133 -0
  686. pygeai/tests/snippets/lab/use_cases/create_web_designer.py +189 -0
  687. pygeai/tests/snippets/lab/use_cases/create_web_reader.py +185 -0
  688. pygeai/tests/snippets/lab/use_cases/file_summarizer_example.py +157 -0
  689. pygeai/tests/snippets/lab/use_cases/file_summarizer_example_2.py +157 -0
  690. pygeai/tests/snippets/lab/use_cases/update_cli_expert.py +1773 -0
  691. pygeai/tests/snippets/lab/use_cases/update_lab_expert.py +4541 -0
  692. pygeai/tests/snippets/lab/use_cases/update_web_designer.py +188 -0
  693. pygeai/tests/snippets/lab/use_cases/update_web_reader.py +195 -0
  694. pygeai/tests/snippets/lab/use_cases/update_web_reader_with_tool.py +210 -0
  695. pygeai/tests/snippets/migrate/__init__.py +45 -0
  696. pygeai/tests/snippets/migrate/agent_migration.py +110 -0
  697. pygeai/tests/snippets/migrate/assistant_migration.py +64 -0
  698. pygeai/tests/snippets/migrate/orchestrator_examples.py +179 -0
  699. pygeai/tests/snippets/migrate/process_migration.py +64 -0
  700. pygeai/tests/snippets/migrate/project_migration.py +42 -0
  701. pygeai/tests/snippets/migrate/tool_migration.py +64 -0
  702. pygeai/tests/snippets/organization/__init__.py +0 -0
  703. pygeai/tests/snippets/organization/add_project_member.py +10 -0
  704. pygeai/tests/snippets/organization/add_project_member_batch.py +44 -0
  705. pygeai/tests/snippets/organization/create_project.py +23 -0
  706. pygeai/tests/snippets/organization/delete_project.py +7 -0
  707. pygeai/tests/snippets/organization/export_request_data.py +7 -0
  708. pygeai/tests/snippets/organization/get_memberships.py +12 -0
  709. pygeai/tests/snippets/organization/get_organization_members.py +6 -0
  710. pygeai/tests/snippets/organization/get_project_data.py +7 -0
  711. pygeai/tests/snippets/organization/get_project_list.py +8 -0
  712. pygeai/tests/snippets/organization/get_project_members.py +6 -0
  713. pygeai/tests/snippets/organization/get_project_memberships.py +12 -0
  714. pygeai/tests/snippets/organization/get_project_roles.py +6 -0
  715. pygeai/tests/snippets/organization/get_project_tokens.py +7 -0
  716. pygeai/tests/snippets/organization/update_project.py +14 -0
  717. pygeai/tests/snippets/rerank/__init__.py +0 -0
  718. pygeai/tests/snippets/rerank/rerank_chunks.py +19 -0
  719. pygeai/tests/snippets/secrets/__init__.py +0 -0
  720. pygeai/tests/snippets/usage_limit/__init__.py +0 -0
  721. pygeai/tests/snippets/usage_limit/delete_usage_limit.py +16 -0
  722. pygeai/tests/snippets/usage_limit/get_all_usage_limit_from_organization.py +12 -0
  723. pygeai/tests/snippets/usage_limit/get_usage_limit_from_organization.py +11 -0
  724. pygeai/tests/snippets/usage_limit/get_usage_limit_from_project.py +13 -0
  725. pygeai/tests/snippets/usage_limit/set_usage_limit_organization.py +22 -0
  726. pygeai/tests/snippets/usage_limit/set_usage_limit_project.py +23 -0
  727. pygeai/tests/snippets/usage_limit/update_usage_limit_organization.py +23 -0
  728. pygeai/tests/snippets/usage_limit/update_usage_limit_project.py +24 -0
  729. pygeai/vendor/a2a/__init__.py +1 -0
  730. pygeai/vendor/a2a/auth/__init__.py +0 -0
  731. pygeai/vendor/a2a/auth/user.py +31 -0
  732. pygeai/vendor/a2a/client/__init__.py +19 -0
  733. pygeai/vendor/a2a/client/client.py +425 -0
  734. pygeai/vendor/a2a/client/errors.py +33 -0
  735. pygeai/vendor/a2a/client/helpers.py +22 -0
  736. pygeai/vendor/a2a/py.typed +0 -0
  737. pygeai/vendor/a2a/server/__init__.py +1 -0
  738. pygeai/vendor/a2a/server/agent_execution/__init__.py +18 -0
  739. pygeai/vendor/a2a/server/agent_execution/agent_executor.py +44 -0
  740. pygeai/vendor/a2a/server/agent_execution/context.py +155 -0
  741. pygeai/vendor/a2a/server/agent_execution/request_context_builder.py +20 -0
  742. pygeai/vendor/a2a/server/agent_execution/simple_request_context_builder.py +77 -0
  743. pygeai/vendor/a2a/server/apps/__init__.py +16 -0
  744. pygeai/vendor/a2a/server/apps/jsonrpc/__init__.py +16 -0
  745. pygeai/vendor/a2a/server/apps/jsonrpc/fastapi_app.py +88 -0
  746. pygeai/vendor/a2a/server/apps/jsonrpc/jsonrpc_app.py +426 -0
  747. pygeai/vendor/a2a/server/apps/jsonrpc/starlette_app.py +123 -0
  748. pygeai/vendor/a2a/server/context.py +23 -0
  749. pygeai/vendor/a2a/server/events/__init__.py +21 -0
  750. pygeai/vendor/a2a/server/events/event_consumer.py +149 -0
  751. pygeai/vendor/a2a/server/events/event_queue.py +156 -0
  752. pygeai/vendor/a2a/server/events/in_memory_queue_manager.py +85 -0
  753. pygeai/vendor/a2a/server/events/queue_manager.py +35 -0
  754. pygeai/vendor/a2a/server/request_handlers/__init__.py +20 -0
  755. pygeai/vendor/a2a/server/request_handlers/default_request_handler.py +435 -0
  756. pygeai/vendor/a2a/server/request_handlers/jsonrpc_handler.py +327 -0
  757. pygeai/vendor/a2a/server/request_handlers/request_handler.py +161 -0
  758. pygeai/vendor/a2a/server/request_handlers/response_helpers.py +133 -0
  759. pygeai/vendor/a2a/server/tasks/__init__.py +20 -0
  760. pygeai/vendor/a2a/server/tasks/inmemory_push_notifier.py +62 -0
  761. pygeai/vendor/a2a/server/tasks/inmemory_task_store.py +51 -0
  762. pygeai/vendor/a2a/server/tasks/push_notifier.py +25 -0
  763. pygeai/vendor/a2a/server/tasks/result_aggregator.py +151 -0
  764. pygeai/vendor/a2a/server/tasks/task_manager.py +253 -0
  765. pygeai/vendor/a2a/server/tasks/task_store.py +22 -0
  766. pygeai/vendor/a2a/server/tasks/task_updater.py +155 -0
  767. pygeai/vendor/a2a/types.py +1624 -0
  768. pygeai/vendor/a2a/utils/__init__.py +40 -0
  769. pygeai/vendor/a2a/utils/artifact.py +72 -0
  770. pygeai/vendor/a2a/utils/errors.py +69 -0
  771. pygeai/vendor/a2a/utils/helpers.py +176 -0
  772. pygeai/vendor/a2a/utils/message.py +83 -0
  773. pygeai/vendor/a2a/utils/task.py +57 -0
  774. pygeai/vendor/a2a/utils/telemetry.py +299 -0
  775. pygeai-0.6.0b15.dist-info/METADATA +205 -0
  776. pygeai-0.6.0b15.dist-info/RECORD +799 -0
  777. {pygeai-0.1.6.dist-info → pygeai-0.6.0b15.dist-info}/WHEEL +1 -1
  778. pygeai-0.6.0b15.dist-info/entry_points.txt +5 -0
  779. {pygeai-0.1.6.dist-info → pygeai-0.6.0b15.dist-info/licenses}/LICENSE +13 -1
  780. {pygeai-0.1.6.dist-info → pygeai-0.6.0b15.dist-info}/top_level.txt +0 -1
  781. docs/source/conf.py +0 -45
  782. pygeai/core/clients.py +0 -240
  783. pygeai/tests/core/test_clients.py +0 -49
  784. pygeai-0.1.6.dist-info/METADATA +0 -92
  785. pygeai-0.1.6.dist-info/RECORD +0 -65
  786. pygeai-0.1.6.dist-info/SOURCES.sync-conflict-20241223-145950-3QD4F42.txt +0 -41
  787. pygeai-0.1.6.dist-info/entry_points.txt +0 -2
  788. /pygeai/{agent → analytics}/__init__.py +0 -0
@@ -0,0 +1,1154 @@
1
+ import unittest
2
+ from unittest import TestCase
3
+
4
+ from pydantic_core._pydantic_core import ValidationError
5
+
6
+ from pygeai.lab.models import (
7
+ FilterSettings, Sampling, LlmConfig, Model, PromptExample, PromptOutput, Prompt, ModelList,
8
+ AgentData, Agent, AgentList, SharingLink, ToolParameter, ToolMessage, Tool, ToolList,
9
+ LocalizedDescription, ReasoningStrategy, ReasoningStrategyList, KnowledgeBase, AgenticActivity,
10
+ ArtifactSignal, UserSignal, Event, SequenceFlow, AgenticProcess, Task, AgenticProcessList,
11
+ TaskList, Variable, VariableList, ProcessInstance, ProcessInstanceList, ArtifactType, ArtifactTypeList,
12
+ JobParameter, Job, JobList, ResourcePoolTool, ResourcePoolAgent, ResourcePool, ResourcePoolList, KnowledgeBaseList
13
+ )
14
+
15
+
16
+ class TestLabModels(TestCase):
17
+ """
18
+ python -m unittest pygeai.tests.lab.test_models.TestLabModels
19
+ """
20
+
21
+ def test_filter_settings_model_validate(self):
22
+ filter_data = {
23
+ "id": "agent123",
24
+ "name": "test-agent",
25
+ "accessScope": "private",
26
+ "isActive": False,
27
+ "allowDrafts": False,
28
+ "allowExternal": False
29
+ }
30
+ filter_settings = FilterSettings.model_validate(filter_data)
31
+ self.assertEqual(filter_settings.id, filter_data["id"])
32
+ self.assertEqual(filter_settings.name, filter_data["name"])
33
+ self.assertEqual(filter_settings.access_scope, filter_data["accessScope"])
34
+ self.assertEqual(filter_settings.allow_drafts, filter_data["allowDrafts"])
35
+ self.assertIsNone(filter_settings.status)
36
+ self.assertTrue(filter_settings.to_dict() == filter_data)
37
+
38
+ def test_sampling_model_validate(self):
39
+ sampling_data = {
40
+ "temperature": 0.7,
41
+ "topK": 40,
42
+ "topP": 0.9
43
+ }
44
+ sampling = Sampling.model_validate(sampling_data)
45
+ self.assertEqual(sampling.temperature, sampling_data["temperature"])
46
+ self.assertEqual(sampling.top_k, sampling_data["topK"])
47
+ self.assertEqual(sampling.top_p, sampling_data["topP"])
48
+ self.assertTrue(sampling.to_dict() == sampling_data)
49
+
50
+ def test_llm_config_model_validate(self):
51
+ llm_config_data = {
52
+ "maxTokens": 1000,
53
+ "timeout": 30,
54
+ "sampling": {"temperature": 0.7, "topK": 40, "topP": 0.9}
55
+ }
56
+ llm_config = LlmConfig.model_validate(llm_config_data)
57
+ self.assertEqual(llm_config.max_tokens, llm_config_data["maxTokens"])
58
+ self.assertEqual(llm_config.timeout, llm_config_data["timeout"])
59
+ self.assertEqual(llm_config.sampling.temperature, llm_config_data["sampling"]["temperature"])
60
+ self.assertEqual(llm_config.sampling.top_k, llm_config_data["sampling"]["topK"])
61
+ self.assertEqual(llm_config.sampling.top_p, llm_config_data["sampling"]["topP"])
62
+ self.assertTrue(llm_config.to_dict() == llm_config_data)
63
+
64
+ def test_model_model_validate(self):
65
+ model_data = {
66
+ "name": "gpt-4",
67
+ "llmConfig": {"maxTokens": 500, "timeout": 20, "sampling": {"temperature": 0.5, "topK": 30, "topP": 0.8}}
68
+ }
69
+ model = Model.model_validate(model_data)
70
+ self.assertEqual(model.name, model_data["name"])
71
+ self.assertEqual(model.llm_config.max_tokens, model_data["llmConfig"]["maxTokens"])
72
+ self.assertEqual(model.llm_config.timeout, model_data["llmConfig"]["timeout"])
73
+ self.assertEqual(model.llm_config.sampling.temperature, model_data["llmConfig"]["sampling"]["temperature"])
74
+ self.assertIsNone(model.prompt)
75
+ self.assertTrue(model.to_dict() == model_data)
76
+
77
+ def test_prompt_example_model_validate(self):
78
+ example_data = {
79
+ "inputData": "Hello world",
80
+ "output": '{"result": "Hi"}'
81
+ }
82
+ prompt_example = PromptExample.model_validate(example_data)
83
+ self.assertEqual(prompt_example.input_data, example_data["inputData"])
84
+ self.assertEqual(prompt_example.output, example_data["output"])
85
+ self.assertTrue(prompt_example.to_dict() == example_data)
86
+
87
+ def test_prompt_output_model_validate(self):
88
+ output_data = {
89
+ "key": "summary",
90
+ "description": "Summary of text"
91
+ }
92
+ prompt_output = PromptOutput.model_validate(output_data)
93
+ self.assertEqual(prompt_output.key, output_data["key"])
94
+ self.assertEqual(prompt_output.description, output_data["description"])
95
+ self.assertTrue(prompt_output.to_dict() == output_data)
96
+
97
+ def test_prompt_model_validate(self):
98
+ prompt_data = {
99
+ "instructions": "Summarize text",
100
+ "inputs": ["text"],
101
+ "outputs": [{"key": "summary", "description": "Summary of text"}],
102
+ "examples": [{"inputData": "Hello world", "output": '{"summary": "Hi"}'}]
103
+ }
104
+ prompt = Prompt.model_validate(prompt_data)
105
+ self.assertEqual(prompt.instructions, prompt_data["instructions"])
106
+ self.assertEqual(prompt.inputs, prompt_data["inputs"])
107
+ self.assertEqual(prompt.outputs[0].key, prompt_data["outputs"][0]["key"])
108
+ self.assertEqual(prompt.examples[0].input_data, prompt_data["examples"][0]["inputData"])
109
+ self.assertTrue(prompt.to_dict() == prompt_data)
110
+
111
+ def test_model_list_model_validate(self):
112
+ model_list_data = [
113
+ {"name": "gpt-4"},
114
+ {"name": "gpt-3.5-turbo"}
115
+ ]
116
+ model_list = ModelList.model_validate({"models": model_list_data})
117
+ self.assertEqual(len(model_list.models), 2)
118
+ self.assertEqual(model_list.models[0].name, model_list_data[0]["name"])
119
+ self.assertEqual(model_list.models[1].name, model_list_data[1]["name"])
120
+ self.assertTrue(model_list.to_dict() == model_list_data)
121
+
122
+ def test_agent_data_model_validate(self):
123
+ agent_data_data = {
124
+ "prompt": {"instructions": "Summarize", "inputs": ["text"], "outputs": [{"key": "summary", "description": "Summary"}]},
125
+ "llmConfig": {"maxTokens": 1000, "timeout": 30, "sampling": {"temperature": 0.7, "topK": 40, "topP": 0.9}},
126
+ "strategyName": "Dynamic Prompting",
127
+ "models": [{"name": "gpt-4"}]
128
+ }
129
+ agent_data = AgentData.model_validate(agent_data_data)
130
+ self.assertEqual(agent_data.prompt.instructions, agent_data_data["prompt"]["instructions"])
131
+ self.assertEqual(agent_data.llm_config.max_tokens, agent_data_data["llmConfig"]["maxTokens"])
132
+ self.assertEqual(agent_data.models.models[0].name, agent_data_data["models"][0]["name"])
133
+ self.assertTrue(agent_data.to_dict() == agent_data_data)
134
+
135
+ def test_agent_model_validate(self):
136
+ agent_data = {
137
+ "id": "agent123",
138
+ "status": "active",
139
+ "name": "TestAgent",
140
+ "isDraft": False,
141
+ "isReadonly": False,
142
+ "accessScope": "public",
143
+ "publicName": "test-agent",
144
+ "agentData": {
145
+ "prompt": {"instructions": "Summarize", "inputs": ["text"], "outputs": [{"key": "summary", "description": "Summary"}]},
146
+ "llmConfig": {"maxTokens": 1000, "timeout": 30, "sampling": {"temperature": 0.7, "topK": 40, "topP": 0.9}},
147
+ "strategyName": "Dynamic Prompting",
148
+ "models": [{"name": "gpt-4"}]
149
+ }
150
+ }
151
+ agent = Agent.model_validate(agent_data)
152
+ self.assertEqual(agent.id, agent_data["id"])
153
+ self.assertEqual(agent.name, agent_data["name"])
154
+ self.assertEqual(agent.access_scope, agent_data["accessScope"])
155
+ self.assertEqual(agent.public_name, agent_data["publicName"])
156
+ self.assertEqual(agent.agent_data.prompt.instructions, agent_data["agentData"]["prompt"]["instructions"])
157
+ self.assertTrue(agent.to_dict() == agent_data)
158
+
159
+ def test_agent_list_model_validate(self):
160
+ agent_list_data = {
161
+ "agents": [
162
+ {"id": "agent1", "name": "Agent1", "isReadonly": False, "isDraft": True, "accessScope": "private", "status": "active"},
163
+ {"id": "agent2", "name": "Agent2", "isReadonly": False, "isDraft": True, "accessScope": "private", "status": "active"}
164
+ ]
165
+ }
166
+ agent_list = AgentList.model_validate(agent_list_data)
167
+ self.assertEqual(len(agent_list.agents), 2)
168
+ self.assertEqual(agent_list.agents[0].id, agent_list_data["agents"][0]["id"])
169
+ self.assertEqual(agent_list.agents[1].name, agent_list_data["agents"][1]["name"])
170
+ self.assertTrue(agent_list.to_list() == agent_list_data["agents"])
171
+
172
+ def test_sharing_link_model_validate(self):
173
+ sharing_link_data = {
174
+ "agentId": "agent123",
175
+ "apiToken": "token456",
176
+ "sharedLink": "https://example.com/share/agent123"
177
+ }
178
+ sharing_link = SharingLink.model_validate(sharing_link_data)
179
+ self.assertEqual(sharing_link.agent_id, sharing_link_data["agentId"])
180
+ self.assertEqual(sharing_link.api_token, sharing_link_data["apiToken"])
181
+ self.assertEqual(sharing_link.shared_link, sharing_link_data["sharedLink"])
182
+ self.assertTrue(sharing_link.to_dict() == sharing_link_data)
183
+
184
+ def test_tool_parameter_model_validate(self):
185
+ param_data = {
186
+ "key": "input",
187
+ "dataType": "String",
188
+ "description": "Input text",
189
+ "isRequired": True,
190
+ "type": "app"
191
+ }
192
+ tool_param = ToolParameter.model_validate(param_data)
193
+ self.assertEqual(tool_param.key, param_data["key"])
194
+ self.assertEqual(tool_param.data_type, param_data["dataType"])
195
+ self.assertEqual(tool_param.description, param_data["description"])
196
+ self.assertEqual(tool_param.is_required, param_data["isRequired"])
197
+ self.assertTrue(tool_param.to_dict() == param_data)
198
+
199
+ def test_tool_message_model_validate(self):
200
+ message_data = {
201
+ "description": "Warning: deprecated",
202
+ "type": "warning"
203
+ }
204
+ tool_message = ToolMessage.model_validate(message_data)
205
+ self.assertEqual(tool_message.description, message_data["description"])
206
+ self.assertEqual(tool_message.type, message_data["type"])
207
+ self.assertTrue(tool_message.to_dict() == message_data)
208
+
209
+ def test_tool_model_validate(self):
210
+ tool_data = {
211
+ "name": "TextTool",
212
+ "description": "Processes text",
213
+ "scope": "builtin",
214
+ "parameters": [{"key": "input", "dataType": "String", "description": "Input text", "isRequired": True, "type": "app"}],
215
+ "reportEvents": "None"
216
+
217
+ }
218
+ tool = Tool.model_validate(tool_data)
219
+ self.assertEqual(tool.name, tool_data["name"])
220
+ self.assertEqual(tool.description, tool_data["description"])
221
+ self.assertEqual(tool.scope, tool_data["scope"])
222
+ self.assertEqual(tool.parameters[0].key, tool_data["parameters"][0]["key"])
223
+ self.assertTrue(tool.to_dict() == tool_data)
224
+
225
+ def test_tool_list_model_validate(self):
226
+ tool_list_data = {
227
+ "tools": [
228
+ {"name": "Tool1", "description": "Tool 1", "scope": "builtin", 'reportEvents': 'None'},
229
+ {"name": "Tool2", "description": "Tool 2", "scope": "builtin", 'reportEvents': 'None'}
230
+ ]
231
+ }
232
+ tool_list = ToolList.model_validate(tool_list_data)
233
+ self.assertEqual(len(tool_list.tools), 2)
234
+ self.assertEqual(tool_list.tools[0].name, tool_list_data["tools"][0]["name"])
235
+ self.assertEqual(tool_list.tools[1].description, tool_list_data["tools"][1]["description"])
236
+ self.assertTrue(tool_list.to_dict() == tool_list_data)
237
+
238
+ def test_localized_description_model_validate(self):
239
+ localized_data = {
240
+ "language": "english",
241
+ "description": "Test strategy"
242
+ }
243
+ localized_desc = LocalizedDescription.model_validate(localized_data)
244
+ self.assertEqual(localized_desc.language, localized_data["language"])
245
+ self.assertEqual(localized_desc.description, localized_data["description"])
246
+ self.assertTrue(localized_desc.to_dict() == localized_data)
247
+
248
+ def test_reasoning_strategy_model_validate(self):
249
+ strategy_data = {
250
+ "name": "TestStrategy",
251
+ "accessScope": "public",
252
+ "type": "addendum",
253
+ "localizedDescriptions": [{"language": "english", "description": "Test strategy"}]
254
+ }
255
+ strategy = ReasoningStrategy.model_validate(strategy_data)
256
+ self.assertEqual(strategy.name, strategy_data["name"])
257
+ self.assertEqual(strategy.access_scope, strategy_data["accessScope"])
258
+ self.assertEqual(strategy.type, strategy_data["type"])
259
+ self.assertEqual(strategy.localized_descriptions[0].language, strategy_data["localizedDescriptions"][0]["language"])
260
+ self.assertTrue(strategy.to_dict() == strategy_data)
261
+
262
+ @unittest.skip("Skip for now. Fix later")
263
+ def test_reasoning_strategy_list_model_validate(self):
264
+ strategy_list_data = [
265
+ {"name": "Strategy1", "accessScope": "public", "type": "addendum"},
266
+ {"name": "Strategy2", "accessScope": "private", "type": "base"}
267
+ ]
268
+ strategy_list = ReasoningStrategyList.model_validate(strategy_list_data)
269
+ self.assertEqual(len(strategy_list.strategies), 2)
270
+ self.assertEqual(strategy_list.strategies[0].name, strategy_list_data[0]["name"])
271
+ self.assertEqual(strategy_list.strategies[1].access_scope, strategy_list_data[1]["accessScope"])
272
+ self.assertTrue(strategy_list.to_dict() == strategy_list_data)
273
+
274
+ def test_knowledge_base_model_validate(self):
275
+ kb_data = {
276
+ "name": "TestKB",
277
+ "artifactTypeName": ["doc"]
278
+ }
279
+ kb = KnowledgeBase.model_validate(kb_data)
280
+ self.assertEqual(kb.name, kb_data["name"])
281
+ self.assertEqual(kb.artifact_type_name, kb_data["artifactTypeName"])
282
+ self.assertTrue(kb.to_dict() == kb_data)
283
+
284
+ def test_agentic_activity_model_validate(self):
285
+ activity_data = {
286
+ "key": "act1",
287
+ "name": "TestActivity",
288
+ "taskName": "TestTask",
289
+ "agentName": "TestAgent",
290
+ "agentRevisionId": 1
291
+ }
292
+ activity = AgenticActivity.model_validate(activity_data)
293
+ self.assertEqual(activity.key, activity_data["key"])
294
+ self.assertEqual(activity.name, activity_data["name"])
295
+ self.assertEqual(activity.task_name, activity_data["taskName"])
296
+ self.assertEqual(activity.agent_name, activity_data["agentName"])
297
+ self.assertEqual(activity.agent_revision_id, activity_data["agentRevisionId"])
298
+ self.assertTrue(activity.to_dict() == activity_data)
299
+
300
+ def test_artifact_signal_model_validate(self):
301
+ signal_data = {
302
+ "key": "sig1",
303
+ "name": "TestSignal",
304
+ "handlingType": "C",
305
+ "artifactTypeName": ["text"]
306
+ }
307
+ signal = ArtifactSignal.model_validate(signal_data)
308
+ self.assertEqual(signal.key, signal_data["key"])
309
+ self.assertEqual(signal.name, signal_data["name"])
310
+ self.assertEqual(signal.handling_type, signal_data["handlingType"])
311
+ self.assertEqual(signal.artifact_type_name, signal_data["artifactTypeName"])
312
+ self.assertTrue(signal.to_dict() == signal_data)
313
+
314
+ def test_user_signal_model_validate(self):
315
+ user_signal_data = {
316
+ "key": "user1",
317
+ "name": "UserDone"
318
+ }
319
+ user_signal = UserSignal.model_validate(user_signal_data)
320
+ self.assertEqual(user_signal.key, user_signal_data["key"])
321
+ self.assertEqual(user_signal.name, user_signal_data["name"])
322
+ self.assertTrue(user_signal.to_dict() == user_signal_data)
323
+
324
+ def test_event_model_validate(self):
325
+ event_data = {
326
+ "key": "evt1",
327
+ "name": "StartEvent"
328
+ }
329
+ event = Event.model_validate(event_data)
330
+ self.assertEqual(event.key, event_data["key"])
331
+ self.assertEqual(event.name, event_data["name"])
332
+ self.assertTrue(event.to_dict() == event_data)
333
+
334
+ def test_sequence_flow_model_validate(self):
335
+ flow_data = {
336
+ "key": "flow1",
337
+ "sourceKey": "start",
338
+ "targetKey": "end"
339
+ }
340
+ sequence_flow = SequenceFlow.model_validate(flow_data)
341
+ self.assertEqual(sequence_flow.key, flow_data["key"])
342
+ self.assertEqual(sequence_flow.source_key, flow_data["sourceKey"])
343
+ self.assertEqual(sequence_flow.target_key, flow_data["targetKey"])
344
+ self.assertTrue(sequence_flow.to_dict() == flow_data)
345
+
346
+ def test_agentic_process_model_validate(self):
347
+ process_data = {
348
+ "name": "TestProcess",
349
+ "kb": {"name": "TestKB", "artifactTypeName": ["doc"]},
350
+ "agenticActivities": [{"key": "act1", "name": "Act1", "taskName": "Task1", "agentName": "Agent1", "agentRevisionId": 1}]
351
+ }
352
+ process = AgenticProcess.model_validate(process_data)
353
+ self.assertEqual(process.name, process_data["name"])
354
+ self.assertEqual(process.kb.name, process_data["kb"]["name"])
355
+ self.assertEqual(process.agentic_activities[0].key, process_data["agenticActivities"][0]["key"])
356
+ self.assertTrue(process.to_dict() == process_data)
357
+
358
+ def test_task_model_validate(self):
359
+ task_data = {
360
+ "name": "TestTask"
361
+ }
362
+ task = Task.model_validate(task_data)
363
+ self.assertEqual(task.name, task_data["name"])
364
+ self.assertIsNone(task.description)
365
+ self.assertTrue(task.to_dict() == task_data)
366
+
367
+ def test_agentic_process_list_model_validate(self):
368
+ process_list_data = [
369
+ {"name": "Process1"},
370
+ {"name": "Process2"}
371
+ ]
372
+ process_list = AgenticProcessList.model_validate({"processes": process_list_data})
373
+ self.assertEqual(len(process_list.processes), 2)
374
+ self.assertEqual(process_list.processes[0].name, process_list_data[0]["name"])
375
+ self.assertEqual(process_list.processes[1].name, process_list_data[1]["name"])
376
+ self.assertTrue(process_list.to_dict() == {"processes": process_list_data})
377
+
378
+ def test_task_list_model_validate(self):
379
+ task_list_data = [
380
+ {"name": "Task1"},
381
+ {"name": "Task2"}
382
+ ]
383
+ task_list = TaskList.model_validate({"tasks": task_list_data})
384
+ self.assertEqual(len(task_list.tasks), 2)
385
+ self.assertEqual(task_list.tasks[0].name, task_list_data[0]["name"])
386
+ self.assertEqual(task_list.tasks[1].name, task_list_data[1]["name"])
387
+ self.assertTrue(task_list.to_dict() == task_list_data)
388
+
389
+ def test_variable_model_validate(self):
390
+ variable_data = {
391
+ "key": "var1",
392
+ "value": "value1"
393
+ }
394
+ variable = Variable.model_validate(variable_data)
395
+ self.assertEqual(variable.key, variable_data["key"])
396
+ self.assertEqual(variable.value, variable_data["value"])
397
+ self.assertTrue(variable.to_dict() == variable_data)
398
+
399
+ def test_variable_list_model_validate(self):
400
+ variable_list_data = [
401
+ {"key": "var1", "value": "value1"},
402
+ {"key": "var2", "value": "value2"}
403
+ ]
404
+ variable_list = VariableList.model_validate({"variables": variable_list_data})
405
+ self.assertEqual(len(variable_list.variables), 2)
406
+ self.assertEqual(variable_list.variables[0].key, variable_list_data[0]["key"])
407
+ self.assertEqual(variable_list.variables[1].value, variable_list_data[1]["value"])
408
+ self.assertTrue(variable_list.to_dict() == variable_list_data)
409
+
410
+ def test_process_instance_model_validate(self):
411
+ instance_data = {
412
+ "id": "inst123",
413
+ "process": {"name": "TestProcess"},
414
+ "createdAt": "2023-01-01T12:00:00",
415
+ "subject": "TestSubject",
416
+ "variables": [{"key": "var1", "value": "val1"}]
417
+ }
418
+ instance = ProcessInstance.model_validate(instance_data)
419
+ self.assertEqual(instance.id, instance_data["id"])
420
+ self.assertEqual(instance.process.name, instance_data["process"]["name"])
421
+ self.assertEqual(instance.created_at, instance_data["createdAt"])
422
+ self.assertEqual(instance.variables.variables[0].key, instance_data["variables"][0]["key"])
423
+ self.assertTrue(instance.to_dict() == instance_data)
424
+
425
+ def test_process_instance_list_model_validate(self):
426
+ instance_list_data = [
427
+ {"id": "inst1", "process": {"name": "Process1"}, "subject": "ProcessSubject"},
428
+ {"id": "inst2", "process": {"name": "Process2"}, "subject": "ProcessSubject"}
429
+ ]
430
+ instance_list = ProcessInstanceList.model_validate({"instances": instance_list_data})
431
+ self.assertEqual(len(instance_list.instances), 2)
432
+ self.assertEqual(instance_list.instances[0].id, instance_list_data[0]["id"])
433
+ self.assertEqual(instance_list.instances[1].process.name, instance_list_data[1]["process"]["name"])
434
+ self.assertTrue(instance_list.to_dict() == instance_list_data)
435
+
436
+ def test_filter_settings_instantiation(self):
437
+
438
+ filter_attr = FilterSettings(id="agent123", name="test-agent", access_scope="private", allow_drafts=False)
439
+
440
+ filter_dict = FilterSettings(**{
441
+ "id": "agent123",
442
+ "name": "test-agent",
443
+ "accessScope": "private",
444
+ "allowDrafts": False
445
+ })
446
+ self.assertEqual(filter_attr.id, filter_dict.id)
447
+ self.assertEqual(filter_attr.name, filter_dict.name)
448
+ self.assertEqual(filter_attr.access_scope, filter_dict.access_scope)
449
+ self.assertEqual(filter_attr.allow_drafts, filter_dict.allow_drafts)
450
+ self.assertEqual(filter_attr.to_dict(), filter_dict.to_dict())
451
+
452
+ def test_sampling_instantiation(self):
453
+
454
+ sampling_attr = Sampling(temperature=0.7, top_k=40, top_p=0.9)
455
+
456
+ sampling_dict = Sampling(**{
457
+ "temperature": 0.7,
458
+ "topK": 40,
459
+ "topP": 0.9
460
+ })
461
+ self.assertEqual(sampling_attr.temperature, sampling_dict.temperature)
462
+ self.assertEqual(sampling_attr.top_k, sampling_dict.top_k)
463
+ self.assertEqual(sampling_attr.top_p, sampling_dict.top_p)
464
+ self.assertEqual(sampling_attr.to_dict(), sampling_dict.to_dict())
465
+
466
+ def test_llm_config_instantiation(self):
467
+
468
+ sampling = Sampling(temperature=0.7, top_k=40, top_p=0.9)
469
+ llm_attr = LlmConfig(max_tokens=1000, timeout=30, sampling=sampling)
470
+
471
+ llm_dict = LlmConfig(**{
472
+ "maxTokens": 1000,
473
+ "timeout": 30,
474
+ "sampling": {
475
+ "temperature": 0.7,
476
+ "topK": 40,
477
+ "topP": 0.9
478
+ }
479
+ })
480
+ self.assertEqual(llm_attr.max_tokens, llm_dict.max_tokens)
481
+ self.assertEqual(llm_attr.timeout, llm_dict.timeout)
482
+ self.assertEqual(llm_attr.sampling.temperature, llm_dict.sampling.temperature)
483
+ self.assertEqual(llm_attr.to_dict(), llm_dict.to_dict())
484
+
485
+ def test_model_instantiation(self):
486
+
487
+ sampling = Sampling(temperature=0.5, top_k=30, top_p=0.8)
488
+ llm_config = LlmConfig(max_tokens=500, timeout=20, sampling=sampling)
489
+ model_attr = Model(name="gpt-4", llm_config=llm_config)
490
+
491
+ model_dict = Model(**{
492
+ "name": "gpt-4",
493
+ "llmConfig": {
494
+ "maxTokens": 500,
495
+ "timeout": 20,
496
+ "sampling": {
497
+ "temperature": 0.5,
498
+ "topK": 30,
499
+ "topP": 0.8
500
+ }
501
+ }
502
+ })
503
+ self.assertEqual(model_attr.name, model_dict.name)
504
+ self.assertEqual(model_attr.llm_config.max_tokens, model_dict.llm_config.max_tokens)
505
+ self.assertEqual(model_attr.to_dict(), model_dict.to_dict())
506
+
507
+ def test_prompt_instantiation(self):
508
+
509
+ output = PromptOutput(key="summary", description="Summary of text")
510
+ example = PromptExample(input_data="Hello world", output='{"summary": "Hi"}')
511
+ prompt_attr = Prompt(instructions="Summarize text", inputs=["text"], outputs=[output], examples=[example])
512
+
513
+ prompt_dict = Prompt(**{
514
+ "instructions": "Summarize text",
515
+ "inputs": ["text"],
516
+ "outputs": [{"key": "summary", "description": "Summary of text"}],
517
+ "examples": [{"inputData": "Hello world", "output": '{"summary": "Hi"}'}]
518
+ })
519
+ self.assertEqual(prompt_attr.instructions, prompt_dict.instructions)
520
+ self.assertEqual(prompt_attr.inputs, prompt_dict.inputs)
521
+ self.assertEqual(prompt_attr.outputs[0].key, prompt_dict.outputs[0].key)
522
+ self.assertEqual(prompt_attr.examples[0].input_data, prompt_dict.examples[0].input_data)
523
+ self.assertEqual(prompt_attr.to_dict(), prompt_dict.to_dict())
524
+
525
+ def test_model_list_instantiation(self):
526
+
527
+ model = Model(name="gpt-4")
528
+ model_list_attr = ModelList(models=[model])
529
+
530
+ model_list_dict = ModelList(**{
531
+ "models": [{"name": "gpt-4"}]
532
+ })
533
+ self.assertEqual(len(model_list_attr.models), len(model_list_dict.models))
534
+ self.assertEqual(model_list_attr.models[0].name, model_list_dict.models[0].name)
535
+ self.assertEqual(model_list_attr.to_dict(), model_list_dict.to_dict())
536
+
537
+ def test_agent_data_instantiation(self):
538
+
539
+ output = PromptOutput(key="summary", description="Summary")
540
+ prompt = Prompt(instructions="Summarize", inputs=["text"], outputs=[output])
541
+ sampling = Sampling(temperature=0.7, top_k=40, top_p=0.9)
542
+ llm_config = LlmConfig(max_tokens=1000, timeout=30, sampling=sampling)
543
+ model_list = ModelList(models=[Model(name="gpt-4")])
544
+ agent_data_attr = AgentData(prompt=prompt, llm_config=llm_config, models=model_list)
545
+
546
+ agent_data_dict = AgentData(**{
547
+ "prompt": {
548
+ "instructions": "Summarize",
549
+ "inputs": ["text"],
550
+ "outputs": [{"key": "summary", "description": "Summary"}]
551
+ },
552
+ "llmConfig": {
553
+ "maxTokens": 1000,
554
+ "timeout": 30,
555
+ "sampling": {"temperature": 0.7, "topK": 40, "topP": 0.9}
556
+ },
557
+ "models": [{"name": "gpt-4"}]
558
+ })
559
+ self.assertEqual(agent_data_attr.prompt.instructions, agent_data_dict.prompt.instructions)
560
+ self.assertEqual(agent_data_attr.llm_config.max_tokens, agent_data_dict.llm_config.max_tokens)
561
+ self.assertEqual(agent_data_attr.models.models[0].name, agent_data_dict.models.models[0].name)
562
+ self.assertEqual(agent_data_attr.to_dict(), agent_data_dict.to_dict())
563
+
564
+ def test_agent_instantiation(self):
565
+
566
+ output = PromptOutput(key="summary", description="Summary")
567
+ prompt = Prompt(instructions="Summarize", inputs=["text"], outputs=[output])
568
+ sampling = Sampling(temperature=0.7, top_k=40, top_p=0.9)
569
+ llm_config = LlmConfig(max_tokens=1000, timeout=30, sampling=sampling)
570
+ model_list = ModelList(models=[Model(name="gpt-4")])
571
+ agent_data = AgentData(prompt=prompt, llm_config=llm_config, models=model_list)
572
+ agent_attr = Agent(id="agent123", name="TestAgent", access_scope="public", public_name="test-agent", agent_data=agent_data)
573
+
574
+ agent_dict = Agent(**{
575
+ "id": "agent123",
576
+ "name": "TestAgent",
577
+ "accessScope": "public",
578
+ "publicName": "test-agent",
579
+ "agentData": {
580
+ "prompt": {
581
+ "instructions": "Summarize",
582
+ "inputs": ["text"],
583
+ "outputs": [{"key": "summary", "description": "Summary"}]
584
+ },
585
+ "llmConfig": {
586
+ "maxTokens": 1000,
587
+ "timeout": 30,
588
+ "sampling": {"temperature": 0.7, "topK": 40, "topP": 0.9}
589
+ },
590
+ "models": [{"name": "gpt-4"}]
591
+ }
592
+ })
593
+ self.assertEqual(agent_attr.id, agent_dict.id)
594
+ self.assertEqual(agent_attr.name, agent_dict.name)
595
+ self.assertEqual(agent_attr.access_scope, agent_dict.access_scope)
596
+ self.assertEqual(agent_attr.public_name, agent_dict.public_name)
597
+ self.assertEqual(agent_attr.agent_data.prompt.instructions, agent_dict.agent_data.prompt.instructions)
598
+ self.assertEqual(agent_attr.to_dict(), agent_dict.to_dict())
599
+
600
+ def test_tool_instantiation(self):
601
+
602
+ param = ToolParameter(key="input", data_type="String", description="Input text", is_required=True)
603
+ tool_attr = Tool(name="TextTool", description="Processes text", scope="builtin", parameters=[param])
604
+
605
+ tool_dict = Tool(**{
606
+ "name": "TextTool",
607
+ "description": "Processes text",
608
+ "scope": "builtin",
609
+ "parameters": [
610
+ {"key": "input", "dataType": "String", "description": "Input text", "isRequired": True}
611
+ ]
612
+ })
613
+ self.assertEqual(tool_attr.name, tool_dict.name)
614
+ self.assertEqual(tool_attr.description, tool_dict.description)
615
+ self.assertEqual(tool_attr.scope, tool_dict.scope)
616
+ self.assertEqual(tool_attr.parameters[0].key, tool_dict.parameters[0].key)
617
+ self.assertEqual(tool_attr.to_dict(), tool_dict.to_dict())
618
+
619
+ def test_agentic_process_instantiation(self):
620
+
621
+ activity = AgenticActivity(key="act1", name="Act1", task_name="Task1", agent_name="Agent1", agent_revision_id=1)
622
+ start_event = Event(key="start1", name="Start")
623
+ flow = SequenceFlow(key="flow1", source_key="start1", target_key="act1")
624
+ process_attr = AgenticProcess(name="TestProcess", agentic_activities=[activity], start_event=start_event, sequence_flows=[flow])
625
+
626
+ process_dict = AgenticProcess(**{
627
+ "name": "TestProcess",
628
+ "agenticActivities": [
629
+ {"key": "act1", "name": "Act1", "taskName": "Task1", "agentName": "Agent1", "agentRevisionId": 1}
630
+ ],
631
+ "startEvent": {"key": "start1", "name": "Start"},
632
+ "sequenceFlows": [
633
+ {"key": "flow1", "sourceKey": "start1", "targetKey": "act1"}
634
+ ]
635
+ })
636
+ self.assertEqual(process_attr.name, process_dict.name)
637
+ self.assertEqual(process_attr.agentic_activities[0].key, process_dict.agentic_activities[0].key)
638
+ self.assertEqual(process_attr.start_event.key, process_dict.start_event.key)
639
+ self.assertEqual(process_attr.sequence_flows[0].key, process_dict.sequence_flows[0].key)
640
+ self.assertEqual(process_attr.to_dict(), process_dict.to_dict())
641
+
642
+ def test_task_instantiation(self):
643
+
644
+ output = PromptOutput(key="summary", description="Summary")
645
+ prompt = Prompt(instructions="Summarize", inputs=["text"], outputs=[output])
646
+ artifact = ArtifactType(name="document", usage_type="input")
647
+ artifact_list = ArtifactTypeList(artifact_types=[artifact])
648
+ task_attr = Task(name="TestTask", prompt_data=prompt, artifact_types=artifact_list)
649
+
650
+ task_dict = Task(**{
651
+ "name": "TestTask",
652
+ "promptData": {
653
+ "instructions": "Summarize",
654
+ "inputs": ["text"],
655
+ "outputs": [{"key": "summary", "description": "Summary"}]
656
+ },
657
+ "artifactTypes": [
658
+ {"name": "document", "usageType": "input"}
659
+ ]
660
+ })
661
+ self.assertEqual(task_attr.name, task_dict.name)
662
+ self.assertEqual(task_attr.prompt_data.instructions, task_dict.prompt_data.instructions)
663
+ self.assertEqual(task_attr.artifact_types.artifact_types[0].name, task_dict.artifact_types.artifact_types[0].name)
664
+ self.assertEqual(task_attr.to_dict(), task_dict.to_dict())
665
+
666
+ def test_agent_invalid_public_name_access_scope(self):
667
+ agent_data = {
668
+ "id": "agent123",
669
+ "status": "active",
670
+ "name": "TestAgent",
671
+ "isDraft": False,
672
+ "isReadonly": False,
673
+ "accessScope": "public"
674
+ # Missing publicName to trigger validation error
675
+ }
676
+ with self.assertRaises(ValueError) as context:
677
+ Agent.model_validate(agent_data)
678
+ self.assertIn("public_name is required if access_scope is", str(context.exception).lower())
679
+
680
+ def test_agent_invalid_public_name_format(self):
681
+ agent_data = {
682
+ "id": "agent123",
683
+ "status": "active",
684
+ "name": "TestAgent",
685
+ "isDraft": False,
686
+ "isReadonly": False,
687
+ "accessScope": "public",
688
+ "publicName": "invalid@name!" # Invalid characters
689
+ }
690
+ with self.assertRaises(ValueError) as context:
691
+ Agent.model_validate(agent_data)
692
+ self.assertIn("public_name must contain only letters, numbers, periods, dashes, or underscores",
693
+ str(context.exception))
694
+
695
+ def test_agent_data_publication_validation_no_models(self):
696
+ agent_data = {
697
+ "id": "agent123",
698
+ "status": "active",
699
+ "name": "TestAgent",
700
+ "isDraft": False, # Not a draft, triggers validation
701
+ "isReadonly": False,
702
+ "accessScope": "private",
703
+ "agentData": {
704
+ "prompt": {"instructions": "Summarize", "inputs": ["text"],
705
+ "outputs": [{"key": "summary", "description": "Summary"}]},
706
+ "llmConfig": {"maxTokens": 1000, "timeout": 30,
707
+ "sampling": {"temperature": 0.7, "topK": 40, "topP": 0.9}},
708
+ "models": [] # Empty models list, should trigger validation error
709
+ }
710
+ }
711
+ with self.assertRaises(ValueError) as context:
712
+ Agent.model_validate(agent_data)
713
+ self.assertIn("at least one valid model must be provided in agent_data.models for publication",
714
+ str(context.exception).lower())
715
+
716
+ def test_agent_data_publication_validation_no_instructions(self):
717
+ agent_data = {
718
+ "id": "agent123",
719
+ "status": "active",
720
+ "name": "TestAgent",
721
+ "isDraft": False, # Not a draft, triggers validation
722
+ "isReadonly": False,
723
+ "accessScope": "private",
724
+ "agentData": {
725
+ "prompt": {"instructions": "", "inputs": ["text"],
726
+ "outputs": [{"key": "summary", "description": "Summary"}]},
727
+ "llmConfig": {"maxTokens": 1000, "timeout": 30,
728
+ "sampling": {"temperature": 0.7, "topK": 40, "topP": 0.9}},
729
+ "models": [{"name": "gpt-4"}] # Valid models, but no instructions
730
+ }
731
+ }
732
+ with self.assertRaises(ValueError) as context:
733
+ Agent.model_validate(agent_data)
734
+ self.assertIn("agent_data.prompt must have at least instructions for publication",
735
+ str(context.exception).lower())
736
+
737
+ def test_tool_invalid_name(self):
738
+ tool_data = {
739
+ "name": "Invalid:Tool", # Contains forbidden character ':'
740
+ "description": "Invalid tool",
741
+ "scope": "builtin"
742
+ }
743
+ with self.assertRaises(ValueError) as context:
744
+ Tool.model_validate(tool_data)
745
+ self.assertIn("name cannot contain ':' or '/'", str(context.exception).lower())
746
+
747
+ def test_tool_invalid_public_name_access_scope(self):
748
+ tool_data = {
749
+ "name": "TestTool",
750
+ "description": "Test tool",
751
+ "scope": "builtin",
752
+ "accessScope": "public"
753
+ # Missing publicName to trigger validation error
754
+ }
755
+ with self.assertRaises(ValueError) as context:
756
+ Tool.model_validate(tool_data)
757
+ self.assertIn("public_name is required if access_scope is 'public'", str(context.exception).lower())
758
+
759
+ def test_tool_invalid_public_name_format(self):
760
+ tool_data = {
761
+ "name": "TestTool",
762
+ "description": "Test tool",
763
+ "scope": "builtin",
764
+ "accessScope": "public",
765
+ "publicName": "invalid@name!" # Invalid characters
766
+ }
767
+ with self.assertRaises(ValueError) as context:
768
+ Tool.model_validate(tool_data)
769
+ self.assertIn("public_name must contain only letters, numbers, periods, dashes, or underscores",
770
+ str(context.exception))
771
+
772
+ def test_tool_api_scope_validation_no_open_api(self):
773
+ tool_data = {
774
+ "name": "ApiTool",
775
+ "description": "API tool",
776
+ "scope": "api"
777
+ # Missing open_api or open_api_json - validation currently commented out
778
+ }
779
+ # Validation is currently commented out in the model, so this should not raise
780
+ tool = Tool.model_validate(tool_data)
781
+ self.assertEqual(tool.name, "ApiTool")
782
+ self.assertEqual(tool.scope, "api")
783
+
784
+ def test_tool_duplicate_parameter_keys(self):
785
+ tool_data = {
786
+ "name": "DuplicateTool",
787
+ "description": "Tool with duplicate params",
788
+ "scope": "builtin",
789
+ "parameters": [
790
+ {"key": "param1", "dataType": "String", "description": "Param 1", "isRequired": True},
791
+ {"key": "param1", "dataType": "Integer", "description": "Param 1 duplicate", "isRequired": False}
792
+ ]
793
+ }
794
+ # Validation is currently commented out in the model, so this should not raise
795
+ tool = Tool.model_validate(tool_data)
796
+ self.assertEqual(tool.name, "DuplicateTool")
797
+ self.assertEqual(len(tool.parameters), 2)
798
+
799
+ def test_knowledge_base_invalid_artifacts(self):
800
+ kb_data = {
801
+ "name": "InvalidKB",
802
+ "artifacts": ["", "valid"] # Empty string should trigger validation error
803
+ }
804
+ with self.assertRaises(ValueError) as context:
805
+ KnowledgeBase.model_validate(kb_data)
806
+ self.assertIn("artifact identifiers cannot be empty", str(context.exception).lower())
807
+
808
+ def test_knowledge_base_invalid_metadata(self):
809
+ kb_data = {
810
+ "name": "InvalidKB",
811
+ "metadata": ["", "valid"] # Empty string should trigger validation error
812
+ }
813
+ with self.assertRaises(ValueError) as context:
814
+ KnowledgeBase.model_validate(kb_data)
815
+ self.assertIn("metadata identifiers cannot be empty", str(context.exception).lower())
816
+
817
+ def test_artifact_type_list_model_validate(self):
818
+ artifact_list_data = {
819
+ "artifact_types": [
820
+ {"name": "doc", "usageType": "input"},
821
+ {"name": "image", "usageType": "output"}
822
+ ]
823
+ }
824
+ artifact_list = ArtifactTypeList.model_validate(artifact_list_data)
825
+ self.assertEqual(len(artifact_list.artifact_types), 2)
826
+ self.assertEqual(artifact_list.artifact_types[0].name, artifact_list_data["artifact_types"][0]["name"])
827
+ self.assertEqual(artifact_list.artifact_types[1].usage_type,
828
+ artifact_list_data["artifact_types"][1]["usageType"])
829
+ # Adjusted to check for content rather than direct equality due to potential nested structure issues
830
+ result_dict = artifact_list.to_dict()
831
+ self.assertEqual(len(result_dict), len(artifact_list_data["artifact_types"]))
832
+ self.assertEqual(result_dict[0]["name"], artifact_list_data["artifact_types"][0]["name"])
833
+
834
+ def test_job_parameter_invalid_name(self):
835
+ job_param_data = {
836
+ "Name": "",
837
+ "Value": "value1"
838
+ }
839
+ with self.assertRaises(ValueError) as context:
840
+ JobParameter.model_validate(job_param_data)
841
+ self.assertIn("parameter name cannot be blank", str(context.exception).lower())
842
+
843
+ def test_job_invalid_name(self):
844
+ job_data = {
845
+ "caption": "Job completed",
846
+ "name": "",
847
+ "parameters": [{"Name": "param1", "Value": "value1"}],
848
+ "request": "2023-01-01T12:00:00",
849
+ "token": "token123",
850
+ "topic": "Default"
851
+ }
852
+ with self.assertRaises(ValueError) as context:
853
+ Job.model_validate(job_data)
854
+ self.assertIn("job name cannot be blank", str(context.exception).lower())
855
+
856
+ def test_job_invalid_token(self):
857
+ job_data = {
858
+ "caption": "Job completed",
859
+ "name": "execute_job",
860
+ "parameters": [{"Name": "param1", "Value": "value1"}],
861
+ "request": "2023-01-01T12:00:00",
862
+ "token": "",
863
+ "topic": "Default"
864
+ }
865
+ with self.assertRaises(ValueError) as context:
866
+ Job.model_validate(job_data)
867
+ self.assertIn("token cannot be blank", str(context.exception).lower())
868
+
869
+ def test_job_list_model_validate(self):
870
+ job_list_data = {
871
+ "jobs": [
872
+ {"caption": "Job 1", "name": "job1", "parameters": [], "request": "2023-01-01T12:00:00",
873
+ "token": "token1", "topic": "Default"},
874
+ {"caption": "Job 2", "name": "job2", "parameters": [], "request": "2023-01-02T12:00:00",
875
+ "token": "token2", "topic": "Event"}
876
+ ]
877
+ }
878
+ job_list = JobList.model_validate(job_list_data)
879
+ self.assertEqual(len(job_list.jobs), 2)
880
+ self.assertEqual(job_list.jobs[0].name, job_list_data["jobs"][0]["name"])
881
+ self.assertEqual(job_list.jobs[1].token, job_list_data["jobs"][1]["token"])
882
+ # Adjusted to check for content rather than direct equality due to potential nested structure issues
883
+ result_dict = job_list.to_dict()
884
+ self.assertEqual(len(result_dict), len(job_list_data["jobs"]))
885
+ self.assertEqual(result_dict[0]["name"], job_list_data["jobs"][0]["name"])
886
+
887
+ def test_resource_pool_tool_model_validate(self):
888
+ tool_data = {
889
+ "name": "TestTool",
890
+ "revision": 2
891
+ }
892
+ resource_tool = ResourcePoolTool.model_validate(tool_data)
893
+ self.assertEqual(resource_tool.name, tool_data["name"])
894
+ self.assertEqual(resource_tool.revision, tool_data["revision"])
895
+ self.assertEqual(resource_tool.to_dict(), tool_data)
896
+
897
+ def test_resource_pool_tool_invalid_name(self):
898
+ tool_data = {
899
+ "name": "Invalid:Tool", # Contains forbidden character ':'
900
+ "revision": 1
901
+ }
902
+ with self.assertRaises(ValueError) as context:
903
+ ResourcePoolTool.model_validate(tool_data)
904
+ self.assertIn("name cannot contain ':' or '/'", str(context.exception).lower())
905
+
906
+ def test_resource_pool_agent_model_validate(self):
907
+ agent_data = {
908
+ "name": "TestAgent",
909
+ "revision": 3
910
+ }
911
+ resource_agent = ResourcePoolAgent.model_validate(agent_data)
912
+ self.assertEqual(resource_agent.name, agent_data["name"])
913
+ self.assertEqual(resource_agent.revision, agent_data["revision"])
914
+ self.assertEqual(resource_agent.to_dict(), agent_data)
915
+
916
+ def test_resource_pool_agent_invalid_name(self):
917
+ agent_data = {
918
+ "name": "Invalid/Agent", # Contains forbidden character '/'
919
+ "revision": 1
920
+ }
921
+ with self.assertRaises(ValueError) as context:
922
+ ResourcePoolAgent.model_validate(agent_data)
923
+ self.assertIn("name cannot contain ':' or '/'", str(context.exception).lower())
924
+
925
+ def test_resource_pool_model_validate(self):
926
+ pool_data = {
927
+ "name": "TestPool",
928
+ "tools": [{"name": "Tool1", "revision": 1}],
929
+ "agents": [{"name": "Agent1", "revision": 2}]
930
+ }
931
+ resource_pool = ResourcePool.model_validate(pool_data)
932
+ self.assertEqual(resource_pool.name, pool_data["name"])
933
+ self.assertEqual(resource_pool.tools[0].name, pool_data["tools"][0]["name"])
934
+ self.assertEqual(resource_pool.agents[0].name, pool_data["agents"][0]["name"])
935
+ self.assertEqual(resource_pool.to_dict(), pool_data)
936
+
937
+ def test_resource_pool_invalid_name(self):
938
+ pool_data = {
939
+ "name": "Invalid:Pool", # Contains forbidden character ':'
940
+ "tools": [],
941
+ "agents": []
942
+ }
943
+ with self.assertRaises(ValueError) as context:
944
+ ResourcePool.model_validate(pool_data)
945
+ self.assertIn("name cannot contain ':' or '/'", str(context.exception).lower())
946
+
947
+ def test_resource_pool_list_model_validate(self):
948
+ pool_list_data = {
949
+ "resourcePools": [
950
+ {"name": "Pool1", "tools": [{"name": "Tool1", "revision": 1}]},
951
+ {"name": "Pool2", "agents": [{"name": "Agent1", "revision": 2}]}
952
+ ]
953
+ }
954
+ resource_pool_list = ResourcePoolList.model_validate(pool_list_data)
955
+ self.assertEqual(len(resource_pool_list.resource_pools), 2)
956
+ self.assertEqual(resource_pool_list.resource_pools[0].name, pool_list_data["resourcePools"][0]["name"])
957
+ self.assertEqual(resource_pool_list.resource_pools[1].name, pool_list_data["resourcePools"][1]["name"])
958
+ self.assertEqual(resource_pool_list.to_dict(),
959
+ [pool_list_data["resourcePools"][0], pool_list_data["resourcePools"][1]])
960
+
961
+ def test_agent_data_unique_resource_pool_names_validation(self):
962
+ agent_data_data = {
963
+ "prompt": {"instructions": "Summarize", "inputs": ["text"],
964
+ "outputs": [{"key": "summary", "description": "Summary"}]},
965
+ "llmConfig": {"maxTokens": 1000, "timeout": 30, "sampling": {"temperature": 0.7, "topK": 40, "topP": 0.9}},
966
+ "models": [{"name": "gpt-4"}],
967
+ "resourcePools": [
968
+ {"name": "Pool1", "tools": [{"name": "Tool1", "revision": 1}]},
969
+ {"name": "Pool1", "agents": [{"name": "Agent1", "revision": 2}]} # Duplicate name
970
+ ]
971
+ }
972
+ with self.assertRaises(ValueError) as context:
973
+ AgentData.model_validate(agent_data_data)
974
+ self.assertIn("resource pool names must be unique within agentdata", str(context.exception).lower())
975
+
976
+ def test_knowledge_base_list_model_validate(self):
977
+ kb_list_data = {
978
+ "knowledgeBases": [
979
+ {"name": "KB1", "artifactTypeName": ["doc"]},
980
+ {"name": "KB2", "artifactTypeName": ["image"]}
981
+ ]
982
+ }
983
+ kb_list = KnowledgeBaseList.model_validate(kb_list_data)
984
+ self.assertEqual(len(kb_list.knowledge_bases), 2)
985
+ self.assertEqual(kb_list.knowledge_bases[0].name, kb_list_data["knowledgeBases"][0]["name"])
986
+ self.assertEqual(kb_list.knowledge_bases[1].name, kb_list_data["knowledgeBases"][1]["name"])
987
+ self.assertEqual(kb_list.to_dict(), [kb_list_data["knowledgeBases"][0], kb_list_data["knowledgeBases"][1]])
988
+
989
+ def test_task_artifact_type_invalid_usage_type(self):
990
+ task_data = {
991
+ "name": "TestTask",
992
+ "artifactTypes": [
993
+ {"name": "doc", "usageType": "invalid"} # Invalid usage type
994
+ ]
995
+ }
996
+ with self.assertRaises(ValidationError) as context:
997
+ Task.model_validate(task_data)
998
+ self.assertIn("usagetype must be 'input' or 'output'", str(context.exception).lower())
999
+
1000
+ def test_task_artifact_type_long_variable_key(self):
1001
+ task_data = {
1002
+ "name": "TestTask",
1003
+ "artifactTypes": [
1004
+ {"name": "doc", "usageType": "input", "artifactVariableKey": "this_key_is_way_too_long_to_be_valid"}
1005
+ # Too long
1006
+ ]
1007
+ }
1008
+ with self.assertRaises(ValueError) as context:
1009
+ Task.model_validate(task_data)
1010
+ self.assertIn("artifactvariablekey", str(context.exception).lower())
1011
+ self.assertIn("must be 20 characters or less", str(context.exception).lower())
1012
+
1013
+ def test_agentic_process_invalid_name(self):
1014
+ process_data = {
1015
+ "name": "Invalid:Process", # Contains forbidden character ':'
1016
+ "agenticActivities": []
1017
+ }
1018
+ with self.assertRaises(ValueError) as context:
1019
+ AgenticProcess.model_validate(process_data)
1020
+ self.assertIn("name cannot contain ':' or '/'", str(context.exception).lower())
1021
+
1022
+ def test_model_list_iterable_and_append(self):
1023
+ model_list = ModelList(models=[])
1024
+ self.assertEqual(len(model_list), 0)
1025
+ model = Model(name="gpt-4")
1026
+ model_list.append(model)
1027
+ self.assertEqual(len(model_list), 1)
1028
+ self.assertEqual(model_list[0].name, "gpt-4")
1029
+ iterated_models = [m for m in model_list]
1030
+ self.assertEqual(len(iterated_models), 1)
1031
+ self.assertEqual(iterated_models[0].name, "gpt-4")
1032
+
1033
+ def test_resource_pool_list_iterable_and_append(self):
1034
+ pool_list = ResourcePoolList(resource_pools=[])
1035
+ self.assertEqual(len(pool_list), 0)
1036
+ pool = ResourcePool(name="TestPool")
1037
+ pool_list.append(pool)
1038
+ self.assertEqual(len(pool_list), 1)
1039
+ self.assertEqual(pool_list[0].name, "TestPool")
1040
+ iterated_pools = [p for p in pool_list]
1041
+ self.assertEqual(len(iterated_pools), 1)
1042
+ self.assertEqual(iterated_pools[0].name, "TestPool")
1043
+
1044
+ def test_agent_list_iterable_and_append(self):
1045
+ agent_list = AgentList(agents=[])
1046
+ self.assertEqual(len(agent_list), 0)
1047
+ agent = Agent(name="TestAgent", access_scope="private")
1048
+ agent_list.append(agent)
1049
+ self.assertEqual(len(agent_list), 1)
1050
+ self.assertEqual(agent_list[0].name, "TestAgent")
1051
+ iterated_agents = [a for a in agent_list]
1052
+ self.assertEqual(len(iterated_agents), 1)
1053
+ self.assertEqual(iterated_agents[0].name, "TestAgent")
1054
+
1055
+ def test_tool_list_iterable_and_append(self):
1056
+ tool_list = ToolList(tools=[])
1057
+ self.assertEqual(len(tool_list), 0)
1058
+ tool = Tool(name="TestTool", description="Test", scope="builtin")
1059
+ tool_list.append(tool)
1060
+ self.assertEqual(len(tool_list), 1)
1061
+ self.assertEqual(tool_list[0].name, "TestTool")
1062
+ iterated_tools = [t for t in tool_list]
1063
+ self.assertEqual(len(iterated_tools), 1)
1064
+ self.assertEqual(iterated_tools[0].name, "TestTool")
1065
+
1066
+ def test_reasoning_strategy_list_iterable_and_append(self):
1067
+ strategy_list = ReasoningStrategyList(strategies=[])
1068
+ self.assertEqual(len(strategy_list), 0)
1069
+ strategy = ReasoningStrategy(name="TestStrategy", access_scope="private", type="addendum")
1070
+ strategy_list.append(strategy)
1071
+ self.assertEqual(len(strategy_list), 1)
1072
+ self.assertEqual(strategy_list[0].name, "TestStrategy")
1073
+ iterated_strategies = [s for s in strategy_list]
1074
+ self.assertEqual(len(iterated_strategies), 1)
1075
+ self.assertEqual(iterated_strategies[0].name, "TestStrategy")
1076
+
1077
+ def test_knowledge_base_list_iterable_and_append(self):
1078
+ kb_list = KnowledgeBaseList(knowledge_bases=[])
1079
+ self.assertEqual(len(kb_list), 0)
1080
+ kb = KnowledgeBase(name="TestKB")
1081
+ kb_list.append(kb)
1082
+ self.assertEqual(len(kb_list), 1)
1083
+ self.assertEqual(kb_list[0].name, "TestKB")
1084
+ iterated_kbs = [k for k in kb_list]
1085
+ self.assertEqual(len(iterated_kbs), 1)
1086
+ self.assertEqual(iterated_kbs[0].name, "TestKB")
1087
+
1088
+ def test_variable_list_iterable_and_append(self):
1089
+ var_list = VariableList(variables=[])
1090
+ self.assertEqual(len(var_list), 0)
1091
+ var = Variable(key="var1", value="value1")
1092
+ var_list.append(var)
1093
+ self.assertEqual(len(var_list), 1)
1094
+ self.assertEqual(var_list[0].key, "var1")
1095
+ iterated_vars = [v for v in var_list]
1096
+ self.assertEqual(len(iterated_vars), 1)
1097
+ self.assertEqual(iterated_vars[0].key, "var1")
1098
+
1099
+ def test_artifact_type_list_iterable_and_append(self):
1100
+ artifact_list = ArtifactTypeList(artifact_types=[])
1101
+ self.assertEqual(len(artifact_list), 0)
1102
+ artifact = ArtifactType(name="doc", usage_type="input")
1103
+ artifact_list.append(artifact)
1104
+ self.assertEqual(len(artifact_list), 1)
1105
+ self.assertEqual(artifact_list[0].name, "doc")
1106
+ iterated_artifacts = [a for a in artifact_list]
1107
+ self.assertEqual(len(iterated_artifacts), 1)
1108
+ self.assertEqual(iterated_artifacts[0].name, "doc")
1109
+
1110
+ def test_agentic_process_list_iterable_and_append(self):
1111
+ process_list = AgenticProcessList(processes=[])
1112
+ self.assertEqual(len(process_list), 0)
1113
+ process = AgenticProcess(name="TestProcess")
1114
+ process_list.append(process)
1115
+ self.assertEqual(len(process_list), 1)
1116
+ self.assertEqual(process_list[0].name, "TestProcess")
1117
+ iterated_processes = [p for p in process_list]
1118
+ self.assertEqual(len(iterated_processes), 1)
1119
+ self.assertEqual(iterated_processes[0].name, "TestProcess")
1120
+
1121
+ def test_task_list_iterable_and_append(self):
1122
+ task_list = TaskList(tasks=[])
1123
+ self.assertEqual(len(task_list), 0)
1124
+ task = Task(name="TestTask")
1125
+ task_list.append(task)
1126
+ self.assertEqual(len(task_list), 1)
1127
+ self.assertEqual(task_list[0].name, "TestTask")
1128
+ iterated_tasks = [t for t in task_list]
1129
+ self.assertEqual(len(iterated_tasks), 1)
1130
+ self.assertEqual(iterated_tasks[0].name, "TestTask")
1131
+
1132
+ def test_process_instance_list_iterable_and_append(self):
1133
+ instance_list = ProcessInstanceList(instances=[])
1134
+ self.assertEqual(len(instance_list), 0)
1135
+ process = AgenticProcess(name="TestProcess")
1136
+ instance = ProcessInstance(process=process, subject="TestSubject")
1137
+ instance_list.append(instance)
1138
+ self.assertEqual(len(instance_list), 1)
1139
+ self.assertEqual(instance_list[0].process.name, "TestProcess")
1140
+ iterated_instances = [i for i in instance_list]
1141
+ self.assertEqual(len(iterated_instances), 1)
1142
+ self.assertEqual(iterated_instances[0].process.name, "TestProcess")
1143
+
1144
+ def test_job_list_iterable_and_append(self):
1145
+ job_list = JobList(jobs=[])
1146
+ self.assertEqual(len(job_list), 0)
1147
+ job = Job(caption="Job completed", name="execute_job", request="2023-01-01T12:00:00", token="token123",
1148
+ topic="Default")
1149
+ job_list.append(job)
1150
+ self.assertEqual(len(job_list), 1)
1151
+ self.assertEqual(job_list[0].name, "execute_job")
1152
+ iterated_jobs = [j for j in job_list]
1153
+ self.assertEqual(len(iterated_jobs), 1)
1154
+ self.assertEqual(iterated_jobs[0].name, "execute_job")