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.
- pygeai/__init__.py +11 -2
- pygeai/_docs/Makefile +20 -0
- pygeai/_docs/make.bat +35 -0
- pygeai/_docs/source/conf.py +117 -0
- pygeai/_docs/source/content/ai_lab/cli.rst +747 -0
- pygeai/_docs/source/content/ai_lab/models.rst +1734 -0
- pygeai/_docs/source/content/ai_lab/runner.rst +253 -0
- pygeai/_docs/source/content/ai_lab/spec.rst +431 -0
- pygeai/_docs/source/content/ai_lab/usage.rst +1011 -0
- pygeai/_docs/source/content/ai_lab.rst +102 -0
- pygeai/_docs/source/content/analytics.rst +598 -0
- pygeai/_docs/source/content/api_reference/admin.rst +161 -0
- pygeai/_docs/source/content/api_reference/assistant.rst +326 -0
- pygeai/_docs/source/content/api_reference/auth.rst +379 -0
- pygeai/_docs/source/content/api_reference/chat.rst +754 -0
- pygeai/_docs/source/content/api_reference/embeddings.rst +154 -0
- pygeai/_docs/source/content/api_reference/evaluation.rst +590 -0
- pygeai/_docs/source/content/api_reference/feedback.rst +237 -0
- pygeai/_docs/source/content/api_reference/files.rst +592 -0
- pygeai/_docs/source/content/api_reference/gam.rst +401 -0
- pygeai/_docs/source/content/api_reference/health.rst +58 -0
- pygeai/_docs/source/content/api_reference/project.rst +738 -0
- pygeai/_docs/source/content/api_reference/proxy.rst +318 -0
- pygeai/_docs/source/content/api_reference/rag.rst +710 -0
- pygeai/_docs/source/content/api_reference/rerank.rst +94 -0
- pygeai/_docs/source/content/api_reference/secrets.rst +495 -0
- pygeai/_docs/source/content/api_reference/usage_limits.rst +390 -0
- pygeai/_docs/source/content/api_reference.rst +58 -0
- pygeai/_docs/source/content/authentication.rst +295 -0
- pygeai/_docs/source/content/chat_gui.rst +121 -0
- pygeai/_docs/source/content/cli.rst +203 -0
- pygeai/_docs/source/content/debugger.rst +651 -0
- pygeai/_docs/source/content/intro.rst +67 -0
- pygeai/_docs/source/content/migration.rst +929 -0
- pygeai/_docs/source/content/modules.rst +7 -0
- pygeai/_docs/source/content/quickstart.rst +143 -0
- pygeai/_docs/source/content/samples.rst +394 -0
- pygeai/_docs/source/index.rst +75 -0
- pygeai/_docs/source/modules.rst +7 -0
- pygeai/_docs/source/pygeai.admin.rst +29 -0
- pygeai/_docs/source/pygeai.analytics.rst +53 -0
- pygeai/_docs/source/pygeai.assistant.data.rst +21 -0
- pygeai/_docs/source/pygeai.assistant.data_analyst.rst +29 -0
- pygeai/_docs/source/pygeai.assistant.rag.rst +53 -0
- pygeai/_docs/source/pygeai.assistant.rst +55 -0
- pygeai/_docs/source/pygeai.auth.rst +29 -0
- pygeai/_docs/source/pygeai.chat.rst +69 -0
- pygeai/_docs/source/pygeai.cli.commands.flows.rst +10 -0
- pygeai/_docs/source/pygeai.cli.commands.lab.rst +53 -0
- pygeai/_docs/source/pygeai.cli.commands.rst +222 -0
- pygeai/_docs/source/pygeai.cli.rst +62 -0
- pygeai/_docs/source/pygeai.cli.texts.rst +21 -0
- pygeai/_docs/source/pygeai.core.base.rst +53 -0
- pygeai/_docs/source/pygeai.core.common.rst +37 -0
- pygeai/_docs/source/pygeai.core.embeddings.rst +61 -0
- pygeai/_docs/source/pygeai.core.feedback.rst +37 -0
- pygeai/_docs/source/pygeai.core.files.rst +61 -0
- pygeai/_docs/source/pygeai.core.llm.rst +29 -0
- pygeai/_docs/source/pygeai.core.plugins.rst +37 -0
- pygeai/_docs/source/pygeai.core.rerank.rst +53 -0
- pygeai/_docs/source/pygeai.core.rst +63 -0
- pygeai/_docs/source/pygeai.core.secrets.rst +29 -0
- pygeai/_docs/source/pygeai.core.services.llm.rst +29 -0
- pygeai/_docs/source/pygeai.core.services.rst +37 -0
- pygeai/_docs/source/pygeai.core.utils.rst +37 -0
- pygeai/_docs/source/pygeai.dbg.rst +21 -0
- pygeai/_docs/source/pygeai.evaluation.dataset.rst +29 -0
- pygeai/_docs/source/pygeai.evaluation.plan.rst +29 -0
- pygeai/_docs/source/pygeai.evaluation.result.rst +29 -0
- pygeai/_docs/source/pygeai.evaluation.rst +31 -0
- pygeai/_docs/source/pygeai.flows.rst +29 -0
- pygeai/_docs/source/pygeai.gam.rst +29 -0
- pygeai/_docs/source/pygeai.health.rst +29 -0
- pygeai/_docs/source/pygeai.lab.agents.rst +37 -0
- pygeai/_docs/source/pygeai.lab.processes.rst +37 -0
- pygeai/_docs/source/pygeai.lab.rst +65 -0
- pygeai/_docs/source/pygeai.lab.spec.rst +29 -0
- pygeai/_docs/source/pygeai.lab.strategies.rst +37 -0
- pygeai/_docs/source/pygeai.lab.tools.rst +37 -0
- pygeai/_docs/source/pygeai.man.man1.rst +10 -0
- pygeai/_docs/source/pygeai.man.rst +18 -0
- pygeai/_docs/source/pygeai.migration.rst +29 -0
- pygeai/_docs/source/pygeai.organization.limits.rst +45 -0
- pygeai/_docs/source/pygeai.organization.rst +61 -0
- pygeai/_docs/source/pygeai.proxy.rst +53 -0
- pygeai/_docs/source/pygeai.rst +35 -0
- pygeai/_docs/source/pygeai.tests.admin.rst +21 -0
- pygeai/_docs/source/pygeai.tests.analytics.rst +45 -0
- pygeai/_docs/source/pygeai.tests.assistants.rag.rst +37 -0
- pygeai/_docs/source/pygeai.tests.assistants.rst +45 -0
- pygeai/_docs/source/pygeai.tests.auth.rst +29 -0
- pygeai/_docs/source/pygeai.tests.chat.rst +45 -0
- pygeai/_docs/source/pygeai.tests.cli.commands.lab.rst +37 -0
- pygeai/_docs/source/pygeai.tests.cli.commands.rst +165 -0
- pygeai/_docs/source/pygeai.tests.cli.docker.rst +10 -0
- pygeai/_docs/source/pygeai.tests.cli.rst +46 -0
- pygeai/_docs/source/pygeai.tests.core.base.data.rst +29 -0
- pygeai/_docs/source/pygeai.tests.core.base.rst +45 -0
- pygeai/_docs/source/pygeai.tests.core.common.data.rst +10 -0
- pygeai/_docs/source/pygeai.tests.core.common.rst +37 -0
- pygeai/_docs/source/pygeai.tests.core.embeddings.rst +37 -0
- pygeai/_docs/source/pygeai.tests.core.feedback.rst +21 -0
- pygeai/_docs/source/pygeai.tests.core.files.rst +53 -0
- pygeai/_docs/source/pygeai.tests.core.llm.rst +21 -0
- pygeai/_docs/source/pygeai.tests.core.plugins.rst +21 -0
- pygeai/_docs/source/pygeai.tests.core.rerank.rst +37 -0
- pygeai/_docs/source/pygeai.tests.core.rst +39 -0
- pygeai/_docs/source/pygeai.tests.core.secrets.rst +21 -0
- pygeai/_docs/source/pygeai.tests.core.services.rst +21 -0
- pygeai/_docs/source/pygeai.tests.core.utils.rst +21 -0
- pygeai/_docs/source/pygeai.tests.dbg.rst +21 -0
- pygeai/_docs/source/pygeai.tests.evaluation.dataset.rst +21 -0
- pygeai/_docs/source/pygeai.tests.evaluation.plan.rst +21 -0
- pygeai/_docs/source/pygeai.tests.evaluation.result.rst +21 -0
- pygeai/_docs/source/pygeai.tests.evaluation.rst +20 -0
- pygeai/_docs/source/pygeai.tests.gam.rst +21 -0
- pygeai/_docs/source/pygeai.tests.health.rst +21 -0
- pygeai/_docs/source/pygeai.tests.integration.assistants.rag.rst +21 -0
- pygeai/_docs/source/pygeai.tests.integration.assistants.rst +18 -0
- pygeai/_docs/source/pygeai.tests.integration.chat.rst +21 -0
- pygeai/_docs/source/pygeai.tests.integration.lab.agents.rst +69 -0
- pygeai/_docs/source/pygeai.tests.integration.lab.processes.rst +77 -0
- pygeai/_docs/source/pygeai.tests.integration.lab.reasoning_strategies.rst +37 -0
- pygeai/_docs/source/pygeai.tests.integration.lab.rst +21 -0
- pygeai/_docs/source/pygeai.tests.integration.lab.tools.rst +77 -0
- pygeai/_docs/source/pygeai.tests.integration.rst +20 -0
- pygeai/_docs/source/pygeai.tests.lab.agents.rst +29 -0
- pygeai/_docs/source/pygeai.tests.lab.processes.rst +29 -0
- pygeai/_docs/source/pygeai.tests.lab.rst +49 -0
- pygeai/_docs/source/pygeai.tests.lab.spec.rst +29 -0
- pygeai/_docs/source/pygeai.tests.lab.strategies.rst +29 -0
- pygeai/_docs/source/pygeai.tests.lab.tools.rst +29 -0
- pygeai/_docs/source/pygeai.tests.migration.rst +29 -0
- pygeai/_docs/source/pygeai.tests.organization.limits.rst +29 -0
- pygeai/_docs/source/pygeai.tests.organization.rst +53 -0
- pygeai/_docs/source/pygeai.tests.proxy.rst +61 -0
- pygeai/_docs/source/pygeai.tests.rst +33 -0
- pygeai/admin/clients.py +14 -11
- pygeai/admin/endpoints.py +2 -2
- pygeai/analytics/clients.py +505 -0
- pygeai/analytics/endpoints.py +35 -0
- pygeai/analytics/managers.py +606 -0
- pygeai/analytics/mappers.py +207 -0
- pygeai/analytics/responses.py +240 -0
- pygeai/assistant/clients.py +48 -57
- pygeai/assistant/data/__init__.py +0 -0
- pygeai/assistant/data/clients.py +15 -0
- pygeai/assistant/data_analyst/__init__.py +0 -0
- pygeai/assistant/data_analyst/clients.py +75 -0
- pygeai/assistant/data_analyst/endpoints.py +2 -0
- pygeai/assistant/endpoints.py +0 -2
- pygeai/assistant/managers.py +738 -0
- pygeai/assistant/mappers.py +153 -0
- pygeai/assistant/rag/clients.py +132 -21
- pygeai/assistant/rag/mappers.py +228 -0
- pygeai/assistant/rag/models.py +396 -0
- pygeai/assistant/rag/responses.py +10 -0
- pygeai/auth/__init__.py +0 -0
- pygeai/auth/clients.py +129 -0
- pygeai/auth/endpoints.py +6 -0
- pygeai/chat/clients.py +406 -31
- pygeai/chat/endpoints.py +3 -0
- pygeai/chat/iris.py +17 -0
- pygeai/chat/managers.py +64 -0
- pygeai/chat/session.py +38 -0
- pygeai/chat/settings.py +6 -0
- pygeai/chat/ui.py +678 -0
- pygeai/cli/__init__.py +0 -1
- pygeai/cli/commands/admin.py +9 -12
- pygeai/cli/commands/analytics.py +533 -0
- pygeai/cli/commands/assistant.py +11 -11
- pygeai/cli/commands/auth.py +299 -0
- pygeai/cli/commands/base.py +201 -7
- pygeai/cli/commands/chat.py +875 -14
- pygeai/cli/commands/common.py +30 -26
- pygeai/cli/commands/configuration.py +84 -9
- pygeai/cli/commands/docs.py +105 -0
- pygeai/cli/commands/embeddings.py +187 -0
- pygeai/cli/commands/evaluation.py +2069 -0
- pygeai/cli/commands/feedback.py +93 -0
- pygeai/cli/commands/files.py +312 -0
- pygeai/cli/commands/flows/__init__.py +0 -0
- pygeai/cli/commands/gam.py +349 -0
- pygeai/cli/commands/lab/__init__.py +0 -0
- pygeai/cli/commands/lab/ai_lab.py +4110 -0
- pygeai/cli/commands/lab/common.py +135 -0
- pygeai/cli/commands/lab/options.py +8 -0
- pygeai/cli/commands/lab/spec.py +273 -0
- pygeai/cli/commands/lab/utils.py +13 -0
- pygeai/cli/commands/llm.py +164 -0
- pygeai/cli/commands/migrate.py +1198 -0
- pygeai/cli/commands/options.py +86 -0
- pygeai/cli/commands/organization.py +560 -98
- pygeai/cli/commands/rag.py +306 -10
- pygeai/cli/commands/rerank.py +108 -0
- pygeai/cli/commands/secrets.py +357 -0
- pygeai/cli/commands/usage_limits.py +583 -0
- pygeai/cli/commands/validators.py +209 -0
- pygeai/cli/commands/version.py +44 -0
- pygeai/cli/error_handler.py +151 -0
- pygeai/cli/geai.py +171 -30
- pygeai/cli/geai_proxy.py +318 -0
- pygeai/cli/install_man.py +107 -0
- pygeai/cli/parsers.py +78 -25
- pygeai/cli/texts/help.py +712 -55
- pygeai/core/__init__.py +9 -1
- pygeai/core/base/clients.py +61 -10
- pygeai/core/base/mappers.py +208 -30
- pygeai/core/base/models.py +8 -308
- pygeai/core/base/responses.py +18 -1
- pygeai/core/base/session.py +110 -17
- pygeai/core/common/config.py +98 -16
- pygeai/core/common/decorators.py +44 -0
- pygeai/core/common/exceptions.py +104 -4
- pygeai/core/embeddings/__init__.py +19 -0
- pygeai/core/embeddings/clients.py +93 -0
- pygeai/core/embeddings/endpoints.py +1 -0
- pygeai/core/embeddings/managers.py +62 -0
- pygeai/core/embeddings/mappers.py +52 -0
- pygeai/core/embeddings/models.py +14 -0
- pygeai/core/embeddings/responses.py +31 -0
- pygeai/core/feedback/__init__.py +0 -0
- pygeai/core/feedback/clients.py +50 -0
- pygeai/core/feedback/endpoints.py +1 -0
- pygeai/core/feedback/models.py +10 -0
- pygeai/core/files/__init__.py +0 -0
- pygeai/core/files/clients.py +156 -0
- pygeai/core/files/endpoints.py +5 -0
- pygeai/core/files/managers.py +224 -0
- pygeai/core/files/mappers.py +44 -0
- pygeai/core/files/models.py +24 -0
- pygeai/core/files/responses.py +19 -0
- pygeai/core/handlers.py +32 -0
- pygeai/core/llm/__init__.py +0 -0
- pygeai/core/llm/clients.py +53 -0
- pygeai/core/llm/endpoints.py +4 -0
- pygeai/core/models.py +799 -0
- pygeai/core/plugins/__init__.py +0 -0
- pygeai/core/plugins/clients.py +32 -0
- pygeai/core/plugins/endpoints.py +1 -0
- pygeai/core/plugins/models.py +86 -0
- pygeai/core/rerank/__init__.py +0 -0
- pygeai/core/rerank/clients.py +35 -0
- pygeai/core/rerank/endpoints.py +1 -0
- pygeai/core/rerank/managers.py +47 -0
- pygeai/core/rerank/mappers.py +23 -0
- pygeai/core/rerank/models.py +27 -0
- pygeai/core/responses.py +104 -0
- pygeai/core/secrets/__init__.py +0 -0
- pygeai/core/secrets/clients.py +212 -0
- pygeai/core/secrets/endpoints.py +7 -0
- pygeai/core/services/llm/__init__.py +0 -0
- pygeai/core/services/llm/model.py +186 -0
- pygeai/core/services/llm/providers.py +15 -0
- pygeai/core/services/response.py +18 -0
- pygeai/core/services/rest.py +311 -89
- pygeai/core/utils/__init__.py +0 -0
- pygeai/core/utils/console.py +83 -0
- pygeai/core/utils/parsers.py +32 -0
- pygeai/core/utils/validators.py +10 -0
- pygeai/dbg/__init__.py +3 -0
- pygeai/dbg/debugger.py +870 -0
- pygeai/evaluation/__init__.py +0 -0
- pygeai/evaluation/clients.py +19 -0
- pygeai/evaluation/dataset/__init__.py +0 -0
- pygeai/evaluation/dataset/clients.py +514 -0
- pygeai/evaluation/dataset/endpoints.py +26 -0
- pygeai/evaluation/plan/__init__.py +0 -0
- pygeai/evaluation/plan/clients.py +302 -0
- pygeai/evaluation/plan/endpoints.py +16 -0
- pygeai/evaluation/result/__init__.py +0 -0
- pygeai/evaluation/result/clients.py +70 -0
- pygeai/evaluation/result/endpoints.py +2 -0
- pygeai/flows/__init__.py +0 -0
- pygeai/flows/endpoints.py +362 -0
- pygeai/flows/models.py +1304 -0
- pygeai/gam/__init__.py +0 -0
- pygeai/gam/clients.py +178 -0
- pygeai/gam/endpoints.py +4 -0
- pygeai/health/__init__.py +0 -0
- pygeai/health/clients.py +24 -0
- pygeai/health/endpoints.py +1 -0
- pygeai/lab/__init__.py +0 -0
- pygeai/lab/agents/__init__.py +0 -0
- pygeai/lab/agents/clients.py +426 -0
- pygeai/lab/agents/endpoints.py +12 -0
- pygeai/lab/agents/mappers.py +319 -0
- pygeai/lab/clients.py +24 -0
- pygeai/lab/constants.py +3 -0
- pygeai/lab/managers.py +1558 -0
- pygeai/lab/models.py +1719 -0
- pygeai/lab/processes/__init__.py +0 -0
- pygeai/lab/processes/clients.py +1051 -0
- pygeai/lab/processes/endpoints.py +26 -0
- pygeai/lab/processes/mappers.py +395 -0
- pygeai/lab/runners.py +90 -0
- pygeai/lab/spec/__init__.py +0 -0
- pygeai/lab/spec/loader.py +24 -0
- pygeai/lab/spec/parsers.py +39 -0
- pygeai/lab/strategies/__init__.py +0 -0
- pygeai/lab/strategies/clients.py +212 -0
- pygeai/lab/strategies/endpoints.py +5 -0
- pygeai/lab/strategies/mappers.py +58 -0
- pygeai/lab/tools/__init__.py +0 -0
- pygeai/lab/tools/clients.py +465 -0
- pygeai/lab/tools/endpoints.py +13 -0
- pygeai/lab/tools/mappers.py +131 -0
- pygeai/man/__init__.py +1 -0
- pygeai/man/man1/__init__.py +1 -0
- pygeai/man/man1/geai-proxy.1 +246 -0
- pygeai/man/man1/geai.1 +2615 -0
- pygeai/migration/__init__.py +33 -0
- pygeai/migration/strategies.py +603 -0
- pygeai/migration/tools.py +180 -0
- pygeai/organization/clients.py +246 -18
- pygeai/organization/endpoints.py +17 -8
- pygeai/organization/limits/__init__.py +0 -0
- pygeai/organization/limits/clients.py +281 -0
- pygeai/organization/limits/endpoints.py +15 -0
- pygeai/organization/limits/managers.py +331 -0
- pygeai/organization/limits/mappers.py +21 -0
- pygeai/organization/managers.py +537 -0
- pygeai/organization/mappers.py +111 -46
- pygeai/organization/responses.py +61 -11
- pygeai/proxy/__init__.py +0 -0
- pygeai/proxy/clients.py +216 -0
- pygeai/proxy/config.py +128 -0
- pygeai/proxy/managers.py +232 -0
- pygeai/proxy/servers.py +304 -0
- pygeai/proxy/tool.py +69 -0
- pygeai/tests/admin/__init__.py +0 -0
- pygeai/tests/admin/test_clients.py +148 -0
- pygeai/tests/analytics/__init__.py +0 -0
- pygeai/tests/analytics/test_clients.py +86 -0
- pygeai/tests/analytics/test_managers.py +94 -0
- pygeai/tests/analytics/test_mappers.py +84 -0
- pygeai/tests/analytics/test_responses.py +73 -0
- pygeai/tests/assistants/rag/__init__.py +0 -0
- pygeai/tests/assistants/rag/test_clients.py +346 -0
- pygeai/tests/assistants/rag/test_mappers.py +189 -0
- pygeai/tests/assistants/rag/test_models.py +292 -0
- pygeai/tests/assistants/test_clients.py +176 -80
- pygeai/tests/assistants/test_managers.py +198 -0
- pygeai/tests/assistants/test_mappers.py +111 -0
- pygeai/tests/auth/__init__.py +0 -0
- pygeai/tests/auth/test_clients.py +289 -0
- pygeai/tests/auth/test_oauth.py +172 -0
- pygeai/tests/auth/test_session_logging.py +150 -0
- pygeai/tests/chat/__init__.py +0 -0
- pygeai/tests/chat/test_clients.py +393 -0
- pygeai/tests/chat/test_iris.py +38 -0
- pygeai/tests/chat/test_session.py +62 -0
- pygeai/tests/chat/test_ui.py +224 -0
- pygeai/tests/cli/__init__.py +0 -0
- pygeai/tests/cli/commands/__init__.py +0 -0
- pygeai/tests/cli/commands/lab/__init__.py +0 -0
- pygeai/tests/cli/commands/lab/test_ai_lab.py +786 -0
- pygeai/tests/cli/commands/lab/test_common.py +208 -0
- pygeai/tests/cli/commands/lab/test_spec.py +246 -0
- pygeai/tests/cli/commands/test_assistant.py +202 -0
- pygeai/tests/cli/commands/test_chat.py +130 -0
- pygeai/tests/cli/commands/test_common.py +350 -0
- pygeai/tests/cli/commands/test_embeddings.py +132 -0
- pygeai/tests/cli/commands/test_evaluation.py +656 -0
- pygeai/tests/cli/commands/test_feedback.py +65 -0
- pygeai/tests/cli/commands/test_files.py +161 -0
- pygeai/tests/cli/commands/test_gam.py +201 -0
- pygeai/tests/cli/commands/test_llm.py +114 -0
- pygeai/tests/cli/commands/test_migrate.py +176 -0
- pygeai/tests/cli/commands/test_organization.py +276 -0
- pygeai/tests/cli/commands/test_rag.py +266 -0
- pygeai/tests/cli/commands/test_rerank.py +110 -0
- pygeai/tests/cli/commands/test_secrets.py +171 -0
- pygeai/tests/cli/commands/test_show_help.py +41 -0
- pygeai/tests/cli/commands/test_usage_limits.py +412 -0
- pygeai/tests/cli/commands/test_validators.py +160 -0
- pygeai/tests/cli/commands/test_version.py +81 -0
- pygeai/tests/cli/docker/__init__.py +0 -0
- pygeai/tests/cli/test_credentials_flag.py +316 -0
- pygeai/tests/cli/test_error_handler.py +225 -0
- pygeai/tests/cli/test_geai_driver.py +154 -0
- pygeai/tests/cli/test_parsers.py +154 -0
- pygeai/tests/core/base/__init__.py +0 -0
- pygeai/tests/core/base/data/__init__.py +0 -0
- pygeai/tests/core/base/data/mappers.py +117 -0
- pygeai/tests/core/base/data/models.py +312 -0
- pygeai/tests/core/base/test_mappers.py +569 -0
- pygeai/tests/core/base/test_models.py +261 -0
- pygeai/tests/core/base/test_responses.py +53 -0
- pygeai/tests/core/common/__init__.py +0 -0
- pygeai/tests/core/common/data/__init__.py +0 -0
- pygeai/tests/core/common/test_config.py +186 -0
- pygeai/tests/core/common/test_decorators.py +69 -0
- pygeai/tests/core/embeddings/__init__.py +0 -0
- pygeai/tests/core/embeddings/test_clients.py +225 -0
- pygeai/tests/core/embeddings/test_managers.py +171 -0
- pygeai/tests/core/embeddings/test_mappers.py +142 -0
- pygeai/tests/core/feedback/__init__.py +0 -0
- pygeai/tests/core/feedback/test_clients.py +64 -0
- pygeai/tests/core/files/__init__.py +0 -0
- pygeai/tests/core/files/test_clients.py +128 -0
- pygeai/tests/core/files/test_managers.py +219 -0
- pygeai/tests/core/files/test_mappers.py +137 -0
- pygeai/tests/core/files/test_models.py +103 -0
- pygeai/tests/core/files/test_responses.py +122 -0
- pygeai/tests/core/llm/__init__.py +0 -0
- pygeai/tests/core/llm/test_clients.py +142 -0
- pygeai/tests/core/plugins/__init__.py +0 -0
- pygeai/tests/core/plugins/test_clients.py +66 -0
- pygeai/tests/core/rerank/__init__.py +0 -0
- pygeai/tests/core/rerank/test_clients.py +76 -0
- pygeai/tests/core/rerank/test_managers.py +99 -0
- pygeai/tests/core/rerank/test_mappers.py +54 -0
- pygeai/tests/core/secrets/__init__.py +0 -0
- pygeai/tests/core/secrets/test_clients.py +264 -0
- pygeai/tests/core/services/__init__.py +0 -0
- pygeai/tests/core/services/test_rest.py +273 -0
- pygeai/tests/core/test_handlers.py +66 -0
- pygeai/tests/core/utils/__init__.py +0 -0
- pygeai/tests/core/utils/test_console.py +80 -0
- pygeai/tests/dbg/__init__.py +0 -0
- pygeai/tests/dbg/test_debugger.py +591 -0
- pygeai/tests/evaluation/__init__.py +0 -0
- pygeai/tests/evaluation/dataset/__init__.py +0 -0
- pygeai/tests/evaluation/dataset/test_clients.py +265 -0
- pygeai/tests/evaluation/plan/__init__.py +0 -0
- pygeai/tests/evaluation/plan/test_clients.py +195 -0
- pygeai/tests/evaluation/result/__init__.py +0 -0
- pygeai/tests/evaluation/result/test_clients.py +66 -0
- pygeai/tests/gam/__init__.py +0 -0
- pygeai/tests/gam/test_clients.py +195 -0
- pygeai/tests/health/__init__.py +0 -0
- pygeai/tests/health/test_clients.py +41 -0
- pygeai/tests/integration/__init__.py +0 -0
- pygeai/tests/integration/assistants/__init__.py +0 -0
- pygeai/tests/integration/assistants/rag/__init__.py +0 -0
- pygeai/tests/integration/assistants/rag/test_create_rag.py +91 -0
- pygeai/tests/integration/chat/__init__.py +0 -0
- pygeai/tests/integration/chat/test_generate_image.py +158 -0
- pygeai/tests/integration/lab/__init__.py +0 -0
- pygeai/tests/integration/lab/agents/__init__.py +0 -0
- pygeai/tests/integration/lab/agents/test_agents_list.py +106 -0
- pygeai/tests/integration/lab/agents/test_create_agent.py +319 -0
- pygeai/tests/integration/lab/agents/test_create_sharing_link.py +70 -0
- pygeai/tests/integration/lab/agents/test_delete_agent.py +75 -0
- pygeai/tests/integration/lab/agents/test_get_agent.py +94 -0
- pygeai/tests/integration/lab/agents/test_publish_agent_revision.py +127 -0
- pygeai/tests/integration/lab/agents/test_update_agent.py +250 -0
- pygeai/tests/integration/lab/processes/__init__.py +0 -0
- pygeai/tests/integration/lab/processes/test_create_process.py +345 -0
- pygeai/tests/integration/lab/processes/test_create_task.py +211 -0
- pygeai/tests/integration/lab/processes/test_delete_process.py +111 -0
- pygeai/tests/integration/lab/processes/test_get_process.py +201 -0
- pygeai/tests/integration/lab/processes/test_list_process_instances.py +91 -0
- pygeai/tests/integration/lab/processes/test_list_processes.py +138 -0
- pygeai/tests/integration/lab/processes/test_publish_process_revision.py +232 -0
- pygeai/tests/integration/lab/processes/test_update_process.py +289 -0
- pygeai/tests/integration/lab/reasoning_strategies/__init__.py +0 -0
- pygeai/tests/integration/lab/reasoning_strategies/test_get_reasoning_strategy.py +70 -0
- pygeai/tests/integration/lab/reasoning_strategies/test_list_reasoning_strategies.py +93 -0
- pygeai/tests/integration/lab/reasoning_strategies/test_update_reasoning_strategy.py +149 -0
- pygeai/tests/integration/lab/tools/__init__.py +0 -0
- pygeai/tests/integration/lab/tools/test_create_tool.py +288 -0
- pygeai/tests/integration/lab/tools/test_delete_tool.py +87 -0
- pygeai/tests/integration/lab/tools/test_get_parameter.py +98 -0
- pygeai/tests/integration/lab/tools/test_get_tool.py +91 -0
- pygeai/tests/integration/lab/tools/test_list_tools.py +106 -0
- pygeai/tests/integration/lab/tools/test_publish_tool_revision.py +119 -0
- pygeai/tests/integration/lab/tools/test_set_parameter.py +114 -0
- pygeai/tests/integration/lab/tools/test_update_tool.py +267 -0
- pygeai/tests/lab/__init__.py +0 -0
- pygeai/tests/lab/agents/__init__.py +0 -0
- pygeai/tests/lab/agents/test_clients.py +481 -0
- pygeai/tests/lab/agents/test_mappers.py +440 -0
- pygeai/tests/lab/processes/__init__.py +0 -0
- pygeai/tests/lab/processes/test_clients.py +1416 -0
- pygeai/tests/lab/processes/test_mappers.py +1092 -0
- pygeai/tests/lab/spec/__init__.py +0 -0
- pygeai/tests/lab/spec/test_loader.py +59 -0
- pygeai/tests/lab/spec/test_parsers.py +182 -0
- pygeai/tests/lab/strategies/__init__.py +0 -0
- pygeai/tests/lab/strategies/test_clients.py +241 -0
- pygeai/tests/lab/strategies/test_mappers.py +132 -0
- pygeai/tests/lab/test_managers.py +553 -0
- pygeai/tests/lab/test_mappers.py +245 -0
- pygeai/tests/lab/test_models.py +1154 -0
- pygeai/tests/lab/tools/__init__.py +0 -0
- pygeai/tests/lab/tools/test_clients.py +521 -0
- pygeai/tests/lab/tools/test_mappers.py +198 -0
- pygeai/tests/migration/__init__.py +0 -0
- pygeai/tests/migration/test_strategies.py +405 -0
- pygeai/tests/migration/test_tools.py +159 -0
- pygeai/tests/organization/limits/__init__.py +0 -0
- pygeai/tests/organization/limits/test_clients.py +567 -0
- pygeai/tests/organization/limits/test_managers.py +402 -0
- pygeai/tests/organization/test_clients.py +615 -64
- pygeai/tests/organization/test_managers.py +424 -0
- pygeai/tests/organization/test_mappers.py +153 -0
- pygeai/tests/organization/test_responses.py +137 -0
- pygeai/tests/proxy/__init__.py +1 -0
- pygeai/tests/proxy/test_clients.py +397 -0
- pygeai/tests/proxy/test_config.py +171 -0
- pygeai/tests/proxy/test_integration.py +305 -0
- pygeai/tests/proxy/test_managers.py +312 -0
- pygeai/tests/proxy/test_servers.py +387 -0
- pygeai/tests/proxy/test_tool.py +176 -0
- pygeai/tests/snippets/__init__.py +0 -0
- pygeai/tests/snippets/analytics/__init__.py +0 -0
- pygeai/tests/snippets/analytics/get_agent_usage_per_user.py +16 -0
- pygeai/tests/snippets/analytics/get_agents_created_and_modified.py +11 -0
- pygeai/tests/snippets/analytics/get_average_cost_per_request.py +10 -0
- pygeai/tests/snippets/analytics/get_overall_error_rate.py +10 -0
- pygeai/tests/snippets/analytics/get_top_10_agents_by_requests.py +12 -0
- pygeai/tests/snippets/analytics/get_total_active_users.py +10 -0
- pygeai/tests/snippets/analytics/get_total_cost.py +10 -0
- pygeai/tests/snippets/analytics/get_total_requests_per_day.py +12 -0
- pygeai/tests/snippets/analytics/get_total_tokens.py +12 -0
- pygeai/tests/snippets/assistants/__init__.py +0 -0
- pygeai/tests/snippets/assistants/create_chat_assistant.py +54 -0
- pygeai/tests/snippets/assistants/create_text_assistant.py +51 -0
- pygeai/tests/snippets/assistants/data_analyst/__init__.py +0 -0
- pygeai/tests/snippets/assistants/data_analyst/extend_and_check.py +100 -0
- pygeai/tests/snippets/assistants/data_analyst/extend_dataset.py +9 -0
- pygeai/tests/snippets/assistants/data_analyst/get_status.py +9 -0
- pygeai/tests/snippets/assistants/file_summarizer_assistant.py +149 -0
- pygeai/tests/snippets/assistants/get_assistant_data.py +8 -0
- pygeai/tests/snippets/assistants/get_assistant_list.py +7 -0
- pygeai/tests/snippets/assistants/rag/__init__.py +0 -0
- pygeai/tests/snippets/assistants/rag/create_rag_assistant.py +65 -0
- pygeai/tests/snippets/assistants/rag/delete_al_documents.py +7 -0
- pygeai/tests/snippets/assistants/rag/delete_document.py +10 -0
- pygeai/tests/snippets/assistants/rag/delete_rag_assistant.py +8 -0
- pygeai/tests/snippets/assistants/rag/get_document.py +10 -0
- pygeai/tests/snippets/assistants/rag/get_documents.py +7 -0
- pygeai/tests/snippets/assistants/rag/get_rag_assistant_data.py +8 -0
- pygeai/tests/snippets/assistants/rag/update_rag_assistant.py +48 -0
- pygeai/tests/snippets/assistants/rag/upload_document.py +19 -0
- pygeai/tests/snippets/assistants/send_feedback.py +14 -0
- pygeai/tests/snippets/assistants/update_chat_assistant.py +63 -0
- pygeai/tests/snippets/auth/__init__.py +0 -0
- pygeai/tests/snippets/chat/__init__.py +0 -0
- pygeai/tests/snippets/chat/cancel_request.py +7 -0
- pygeai/tests/snippets/chat/chat_completion.py +28 -0
- pygeai/tests/snippets/chat/chat_completion_1.py +40 -0
- pygeai/tests/snippets/chat/chat_completion_2.py +60 -0
- pygeai/tests/snippets/chat/chat_completion_3.py +27 -0
- pygeai/tests/snippets/chat/chat_completion_4.py +67 -0
- pygeai/tests/snippets/chat/chat_completion_streaming.py +63 -0
- pygeai/tests/snippets/chat/chat_completion_with_reasoning_effort.py +18 -0
- pygeai/tests/snippets/chat/get_request_status.py +7 -0
- pygeai/tests/snippets/chat/get_response.py +15 -0
- pygeai/tests/snippets/chat/get_response_complete_example.py +67 -0
- pygeai/tests/snippets/chat/get_response_streaming.py +20 -0
- pygeai/tests/snippets/chat/get_response_with_files.py +16 -0
- pygeai/tests/snippets/chat/get_response_with_instructions.py +19 -0
- pygeai/tests/snippets/chat/get_response_with_metadata.py +24 -0
- pygeai/tests/snippets/chat/get_response_with_parallel_tools.py +58 -0
- pygeai/tests/snippets/chat/get_response_with_reasoning.py +21 -0
- pygeai/tests/snippets/chat/get_response_with_store.py +38 -0
- pygeai/tests/snippets/chat/get_response_with_tools.py +36 -0
- pygeai/tests/snippets/chat/get_response_with_truncation.py +24 -0
- pygeai/tests/snippets/chat/send_chat_request.py +33 -0
- pygeai/tests/snippets/dbg/__init__.py +0 -0
- pygeai/tests/snippets/dbg/basic_debugging.py +32 -0
- pygeai/tests/snippets/dbg/breakpoint_management.py +48 -0
- pygeai/tests/snippets/dbg/file_debugging.py +72 -0
- pygeai/tests/snippets/dbg/module_debugging.py +61 -0
- pygeai/tests/snippets/dbg/stack_navigation.py +45 -0
- pygeai/tests/snippets/dbg/stepping_example.py +40 -0
- pygeai/tests/snippets/embeddings/__init__.py +0 -0
- pygeai/tests/snippets/embeddings/cache_example.py +31 -0
- pygeai/tests/snippets/embeddings/cohere_example.py +41 -0
- pygeai/tests/snippets/embeddings/generate_embeddings.py +26 -0
- pygeai/tests/snippets/embeddings/openai_base64_example.py +27 -0
- pygeai/tests/snippets/embeddings/openai_example.py +30 -0
- pygeai/tests/snippets/embeddings/similarity_example.py +42 -0
- pygeai/tests/snippets/evaluation/__init__.py +0 -0
- pygeai/tests/snippets/evaluation/dataset/__init__.py +0 -0
- pygeai/tests/snippets/evaluation/dataset/complete_workflow_example.py +195 -0
- pygeai/tests/snippets/evaluation/dataset/create_dataset.py +26 -0
- pygeai/tests/snippets/evaluation/dataset/create_dataset_from_file.py +11 -0
- pygeai/tests/snippets/evaluation/dataset/create_dataset_row.py +17 -0
- pygeai/tests/snippets/evaluation/dataset/create_expected_source.py +18 -0
- pygeai/tests/snippets/evaluation/dataset/create_filter_variable.py +19 -0
- pygeai/tests/snippets/evaluation/dataset/delete_dataset.py +9 -0
- pygeai/tests/snippets/evaluation/dataset/delete_dataset_row.py +10 -0
- pygeai/tests/snippets/evaluation/dataset/delete_expected_source.py +15 -0
- pygeai/tests/snippets/evaluation/dataset/delete_filter_variable.py +15 -0
- pygeai/tests/snippets/evaluation/dataset/get_dataset.py +9 -0
- pygeai/tests/snippets/evaluation/dataset/get_dataset_row.py +10 -0
- pygeai/tests/snippets/evaluation/dataset/get_expected_source.py +15 -0
- pygeai/tests/snippets/evaluation/dataset/get_filter_variable.py +15 -0
- pygeai/tests/snippets/evaluation/dataset/list_dataset_rows.py +9 -0
- pygeai/tests/snippets/evaluation/dataset/list_datasets.py +6 -0
- pygeai/tests/snippets/evaluation/dataset/list_expected_sources.py +10 -0
- pygeai/tests/snippets/evaluation/dataset/list_filter_variables.py +10 -0
- pygeai/tests/snippets/evaluation/dataset/update_dataset.py +15 -0
- pygeai/tests/snippets/evaluation/dataset/update_dataset_row.py +20 -0
- pygeai/tests/snippets/evaluation/dataset/update_expected_source.py +18 -0
- pygeai/tests/snippets/evaluation/dataset/update_filter_variable.py +19 -0
- pygeai/tests/snippets/evaluation/dataset/upload_dataset_rows_file.py +10 -0
- pygeai/tests/snippets/evaluation/plan/__init__.py +0 -0
- pygeai/tests/snippets/evaluation/plan/add_plan_system_metric.py +13 -0
- pygeai/tests/snippets/evaluation/plan/complete_workflow_example.py +136 -0
- pygeai/tests/snippets/evaluation/plan/create_evaluation_plan.py +24 -0
- pygeai/tests/snippets/evaluation/plan/create_rag_evaluation_plan.py +22 -0
- pygeai/tests/snippets/evaluation/plan/delete_evaluation_plan.py +9 -0
- pygeai/tests/snippets/evaluation/plan/delete_plan_system_metric.py +13 -0
- pygeai/tests/snippets/evaluation/plan/execute_evaluation_plan.py +11 -0
- pygeai/tests/snippets/evaluation/plan/get_evaluation_plan.py +9 -0
- pygeai/tests/snippets/evaluation/plan/get_plan_system_metric.py +13 -0
- pygeai/tests/snippets/evaluation/plan/get_system_metric.py +9 -0
- pygeai/tests/snippets/evaluation/plan/list_evaluation_plans.py +7 -0
- pygeai/tests/snippets/evaluation/plan/list_plan_system_metrics.py +9 -0
- pygeai/tests/snippets/evaluation/plan/list_system_metrics.py +7 -0
- pygeai/tests/snippets/evaluation/plan/update_evaluation_plan.py +22 -0
- pygeai/tests/snippets/evaluation/plan/update_plan_system_metric.py +14 -0
- pygeai/tests/snippets/evaluation/result/__init__.py +0 -0
- pygeai/tests/snippets/evaluation/result/complete_workflow_example.py +150 -0
- pygeai/tests/snippets/evaluation/result/get_evaluation_result.py +26 -0
- pygeai/tests/snippets/evaluation/result/list_evaluation_results.py +17 -0
- pygeai/tests/snippets/files/__init__.py +0 -0
- pygeai/tests/snippets/files/delete_file.py +9 -0
- pygeai/tests/snippets/files/get_file_content.py +10 -0
- pygeai/tests/snippets/files/get_file_data.py +9 -0
- pygeai/tests/snippets/files/get_file_list.py +6 -0
- pygeai/tests/snippets/files/upload_file.py +13 -0
- pygeai/tests/snippets/gam/__init__.py +0 -0
- pygeai/tests/snippets/gam/gam_access_token.py +87 -0
- pygeai/tests/snippets/lab/__init__.py +0 -0
- pygeai/tests/snippets/lab/agentic_flow_example_1.py +326 -0
- pygeai/tests/snippets/lab/agentic_flow_example_2.py +206 -0
- pygeai/tests/snippets/lab/agentic_flow_example_3.py +486 -0
- pygeai/tests/snippets/lab/agentic_flow_example_4.py +446 -0
- pygeai/tests/snippets/lab/agents/__init__.py +0 -0
- pygeai/tests/snippets/lab/agents/create_agent.py +48 -0
- pygeai/tests/snippets/lab/agents/create_agent_2.py +48 -0
- pygeai/tests/snippets/lab/agents/create_agent_edge_case.py +48 -0
- pygeai/tests/snippets/lab/agents/create_agent_with_permissions.py +39 -0
- pygeai/tests/snippets/lab/agents/create_agent_with_properties.py +46 -0
- pygeai/tests/snippets/lab/agents/create_agent_without_instructions.py +48 -0
- pygeai/tests/snippets/lab/agents/delete_agent.py +12 -0
- pygeai/tests/snippets/lab/agents/get_agent.py +24 -0
- pygeai/tests/snippets/lab/agents/get_agent_with_new_fields.py +62 -0
- pygeai/tests/snippets/lab/agents/get_sharing_link.py +13 -0
- pygeai/tests/snippets/lab/agents/list_agents.py +18 -0
- pygeai/tests/snippets/lab/agents/publish_agent_revision.py +12 -0
- pygeai/tests/snippets/lab/agents/update_agent.py +50 -0
- pygeai/tests/snippets/lab/agents/update_agent_properties.py +50 -0
- pygeai/tests/snippets/lab/assistant_to_agent.py +191 -0
- pygeai/tests/snippets/lab/crud_ui.py +462 -0
- pygeai/tests/snippets/lab/processes/__init__.py +0 -0
- pygeai/tests/snippets/lab/processes/create_process.py +24 -0
- pygeai/tests/snippets/lab/processes/create_task.py +8 -0
- pygeai/tests/snippets/lab/processes/jobs/__init__.py +0 -0
- pygeai/tests/snippets/lab/processes/jobs/list_jobs.py +21 -0
- pygeai/tests/snippets/lab/processes/kbs/__init__.py +0 -0
- pygeai/tests/snippets/lab/processes/kbs/create_kb.py +18 -0
- pygeai/tests/snippets/lab/processes/kbs/get_kb.py +26 -0
- pygeai/tests/snippets/lab/processes/kbs/list_kbs.py +30 -0
- pygeai/tests/snippets/lab/processes/kbs/try_all.py +73 -0
- pygeai/tests/snippets/lab/processes/list_processes.py +10 -0
- pygeai/tests/snippets/lab/runner_1.py +212 -0
- pygeai/tests/snippets/lab/samples/__init__.py +0 -0
- pygeai/tests/snippets/lab/samples/summarize_files.py +162 -0
- pygeai/tests/snippets/lab/strategies/__init__.py +0 -0
- pygeai/tests/snippets/lab/strategies/create_reasoning_strategy.py +22 -0
- pygeai/tests/snippets/lab/strategies/get_reasoning_strategy.py +10 -0
- pygeai/tests/snippets/lab/strategies/list_reasoning_strategies.py +16 -0
- pygeai/tests/snippets/lab/strategies/update_reasoning_strategy.py +26 -0
- pygeai/tests/snippets/lab/tools/__init__.py +0 -0
- pygeai/tests/snippets/lab/tools/create_tool.py +48 -0
- pygeai/tests/snippets/lab/tools/create_tool_edge_case.py +50 -0
- pygeai/tests/snippets/lab/tools/delete_tool.py +21 -0
- pygeai/tests/snippets/lab/tools/get_parameter.py +21 -0
- pygeai/tests/snippets/lab/tools/get_tool.py +22 -0
- pygeai/tests/snippets/lab/tools/list_tools.py +23 -0
- pygeai/tests/snippets/lab/tools/publish_tool_revision.py +13 -0
- pygeai/tests/snippets/lab/tools/set_parameters.py +33 -0
- pygeai/tests/snippets/lab/tools/update_tool.py +52 -0
- pygeai/tests/snippets/lab/use_cases/__init__.py +0 -0
- pygeai/tests/snippets/lab/use_cases/c_code_fixer_agent_flow.py +238 -0
- pygeai/tests/snippets/lab/use_cases/create_cli_expert.py +1640 -0
- pygeai/tests/snippets/lab/use_cases/create_lab_expert.py +4541 -0
- pygeai/tests/snippets/lab/use_cases/create_tool_headless_web_browser.py +133 -0
- pygeai/tests/snippets/lab/use_cases/create_web_designer.py +189 -0
- pygeai/tests/snippets/lab/use_cases/create_web_reader.py +185 -0
- pygeai/tests/snippets/lab/use_cases/file_summarizer_example.py +157 -0
- pygeai/tests/snippets/lab/use_cases/file_summarizer_example_2.py +157 -0
- pygeai/tests/snippets/lab/use_cases/update_cli_expert.py +1773 -0
- pygeai/tests/snippets/lab/use_cases/update_lab_expert.py +4541 -0
- pygeai/tests/snippets/lab/use_cases/update_web_designer.py +188 -0
- pygeai/tests/snippets/lab/use_cases/update_web_reader.py +195 -0
- pygeai/tests/snippets/lab/use_cases/update_web_reader_with_tool.py +210 -0
- pygeai/tests/snippets/migrate/__init__.py +45 -0
- pygeai/tests/snippets/migrate/agent_migration.py +110 -0
- pygeai/tests/snippets/migrate/assistant_migration.py +64 -0
- pygeai/tests/snippets/migrate/orchestrator_examples.py +179 -0
- pygeai/tests/snippets/migrate/process_migration.py +64 -0
- pygeai/tests/snippets/migrate/project_migration.py +42 -0
- pygeai/tests/snippets/migrate/tool_migration.py +64 -0
- pygeai/tests/snippets/organization/__init__.py +0 -0
- pygeai/tests/snippets/organization/add_project_member.py +10 -0
- pygeai/tests/snippets/organization/add_project_member_batch.py +44 -0
- pygeai/tests/snippets/organization/create_project.py +23 -0
- pygeai/tests/snippets/organization/delete_project.py +7 -0
- pygeai/tests/snippets/organization/export_request_data.py +7 -0
- pygeai/tests/snippets/organization/get_memberships.py +12 -0
- pygeai/tests/snippets/organization/get_organization_members.py +6 -0
- pygeai/tests/snippets/organization/get_project_data.py +7 -0
- pygeai/tests/snippets/organization/get_project_list.py +8 -0
- pygeai/tests/snippets/organization/get_project_members.py +6 -0
- pygeai/tests/snippets/organization/get_project_memberships.py +12 -0
- pygeai/tests/snippets/organization/get_project_roles.py +6 -0
- pygeai/tests/snippets/organization/get_project_tokens.py +7 -0
- pygeai/tests/snippets/organization/update_project.py +14 -0
- pygeai/tests/snippets/rerank/__init__.py +0 -0
- pygeai/tests/snippets/rerank/rerank_chunks.py +19 -0
- pygeai/tests/snippets/secrets/__init__.py +0 -0
- pygeai/tests/snippets/usage_limit/__init__.py +0 -0
- pygeai/tests/snippets/usage_limit/delete_usage_limit.py +16 -0
- pygeai/tests/snippets/usage_limit/get_all_usage_limit_from_organization.py +12 -0
- pygeai/tests/snippets/usage_limit/get_usage_limit_from_organization.py +11 -0
- pygeai/tests/snippets/usage_limit/get_usage_limit_from_project.py +13 -0
- pygeai/tests/snippets/usage_limit/set_usage_limit_organization.py +22 -0
- pygeai/tests/snippets/usage_limit/set_usage_limit_project.py +23 -0
- pygeai/tests/snippets/usage_limit/update_usage_limit_organization.py +23 -0
- pygeai/tests/snippets/usage_limit/update_usage_limit_project.py +24 -0
- pygeai/vendor/a2a/__init__.py +1 -0
- pygeai/vendor/a2a/auth/__init__.py +0 -0
- pygeai/vendor/a2a/auth/user.py +31 -0
- pygeai/vendor/a2a/client/__init__.py +19 -0
- pygeai/vendor/a2a/client/client.py +425 -0
- pygeai/vendor/a2a/client/errors.py +33 -0
- pygeai/vendor/a2a/client/helpers.py +22 -0
- pygeai/vendor/a2a/py.typed +0 -0
- pygeai/vendor/a2a/server/__init__.py +1 -0
- pygeai/vendor/a2a/server/agent_execution/__init__.py +18 -0
- pygeai/vendor/a2a/server/agent_execution/agent_executor.py +44 -0
- pygeai/vendor/a2a/server/agent_execution/context.py +155 -0
- pygeai/vendor/a2a/server/agent_execution/request_context_builder.py +20 -0
- pygeai/vendor/a2a/server/agent_execution/simple_request_context_builder.py +77 -0
- pygeai/vendor/a2a/server/apps/__init__.py +16 -0
- pygeai/vendor/a2a/server/apps/jsonrpc/__init__.py +16 -0
- pygeai/vendor/a2a/server/apps/jsonrpc/fastapi_app.py +88 -0
- pygeai/vendor/a2a/server/apps/jsonrpc/jsonrpc_app.py +426 -0
- pygeai/vendor/a2a/server/apps/jsonrpc/starlette_app.py +123 -0
- pygeai/vendor/a2a/server/context.py +23 -0
- pygeai/vendor/a2a/server/events/__init__.py +21 -0
- pygeai/vendor/a2a/server/events/event_consumer.py +149 -0
- pygeai/vendor/a2a/server/events/event_queue.py +156 -0
- pygeai/vendor/a2a/server/events/in_memory_queue_manager.py +85 -0
- pygeai/vendor/a2a/server/events/queue_manager.py +35 -0
- pygeai/vendor/a2a/server/request_handlers/__init__.py +20 -0
- pygeai/vendor/a2a/server/request_handlers/default_request_handler.py +435 -0
- pygeai/vendor/a2a/server/request_handlers/jsonrpc_handler.py +327 -0
- pygeai/vendor/a2a/server/request_handlers/request_handler.py +161 -0
- pygeai/vendor/a2a/server/request_handlers/response_helpers.py +133 -0
- pygeai/vendor/a2a/server/tasks/__init__.py +20 -0
- pygeai/vendor/a2a/server/tasks/inmemory_push_notifier.py +62 -0
- pygeai/vendor/a2a/server/tasks/inmemory_task_store.py +51 -0
- pygeai/vendor/a2a/server/tasks/push_notifier.py +25 -0
- pygeai/vendor/a2a/server/tasks/result_aggregator.py +151 -0
- pygeai/vendor/a2a/server/tasks/task_manager.py +253 -0
- pygeai/vendor/a2a/server/tasks/task_store.py +22 -0
- pygeai/vendor/a2a/server/tasks/task_updater.py +155 -0
- pygeai/vendor/a2a/types.py +1624 -0
- pygeai/vendor/a2a/utils/__init__.py +40 -0
- pygeai/vendor/a2a/utils/artifact.py +72 -0
- pygeai/vendor/a2a/utils/errors.py +69 -0
- pygeai/vendor/a2a/utils/helpers.py +176 -0
- pygeai/vendor/a2a/utils/message.py +83 -0
- pygeai/vendor/a2a/utils/task.py +57 -0
- pygeai/vendor/a2a/utils/telemetry.py +299 -0
- pygeai-0.6.0b15.dist-info/METADATA +205 -0
- pygeai-0.6.0b15.dist-info/RECORD +799 -0
- {pygeai-0.1.6.dist-info → pygeai-0.6.0b15.dist-info}/WHEEL +1 -1
- pygeai-0.6.0b15.dist-info/entry_points.txt +5 -0
- {pygeai-0.1.6.dist-info → pygeai-0.6.0b15.dist-info/licenses}/LICENSE +13 -1
- {pygeai-0.1.6.dist-info → pygeai-0.6.0b15.dist-info}/top_level.txt +0 -1
- docs/source/conf.py +0 -45
- pygeai/core/clients.py +0 -240
- pygeai/tests/core/test_clients.py +0 -49
- pygeai-0.1.6.dist-info/METADATA +0 -92
- pygeai-0.1.6.dist-info/RECORD +0 -65
- pygeai-0.1.6.dist-info/SOURCES.sync-conflict-20241223-145950-3QD4F42.txt +0 -41
- pygeai-0.1.6.dist-info/entry_points.txt +0 -2
- /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")
|