veadk-python 0.2.12__tar.gz → 0.2.14__tar.gz

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 veadk-python might be problematic. Click here for more details.

Files changed (205) hide show
  1. {veadk_python-0.2.12 → veadk_python-0.2.14}/PKG-INFO +1 -1
  2. {veadk_python-0.2.12 → veadk_python-0.2.14}/pyproject.toml +1 -1
  3. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli.py +2 -0
  4. veadk_python-0.2.14/veadk/cli/cli_create.py +122 -0
  5. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/consts.py +0 -3
  6. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_tos/ve_tos.py +5 -0
  7. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/utils.py +2 -1
  8. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/vikingdb_knowledge_backend.py +47 -19
  9. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory.py +9 -0
  10. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py +24 -5
  11. veadk_python-0.2.12/veadk/tools/builtin_tools/generate_image.py → veadk_python-0.2.14/veadk/tools/builtin_tools/image_generate.py +27 -11
  12. veadk_python-0.2.14/veadk/tools/builtin_tools/run_code.py +81 -0
  13. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/version.py +1 -1
  14. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk_python.egg-info/PKG-INFO +1 -1
  15. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk_python.egg-info/SOURCES.txt +2 -1
  16. veadk_python-0.2.12/veadk/tools/builtin_tools/image_generate.py +0 -296
  17. {veadk_python-0.2.12 → veadk_python-0.2.14}/LICENSE +0 -0
  18. {veadk_python-0.2.12 → veadk_python-0.2.14}/README.md +0 -0
  19. {veadk_python-0.2.12 → veadk_python-0.2.14}/setup.cfg +0 -0
  20. {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_agent.py +0 -0
  21. {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_cloud.py +0 -0
  22. {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_evaluator.py +0 -0
  23. {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_knowledgebase.py +0 -0
  24. {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_long_term_memory.py +0 -0
  25. {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_multiple_agents.py +0 -0
  26. {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_runner.py +0 -0
  27. {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_runtime_data_collecting.py +0 -0
  28. {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_short_term_memory.py +0 -0
  29. {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_tracing.py +0 -0
  30. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/__init__.py +0 -0
  31. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/a2a/__init__.py +0 -0
  32. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/a2a/agent_card.py +0 -0
  33. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/a2a/remote_ve_agent.py +0 -0
  34. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/a2a/ve_a2a_server.py +0 -0
  35. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/a2a/ve_agent_executor.py +0 -0
  36. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/a2a/ve_task_store.py +0 -0
  37. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/agent.py +0 -0
  38. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/agent_builder.py +0 -0
  39. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/agents/loop_agent.py +0 -0
  40. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/agents/parallel_agent.py +0 -0
  41. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/agents/sequential_agent.py +0 -0
  42. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/__init__.py +0 -0
  43. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/base_auth.py +0 -0
  44. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/__init__.py +0 -0
  45. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/apmplus_veauth.py +0 -0
  46. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/ark_veauth.py +0 -0
  47. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/base_veauth.py +0 -0
  48. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/cozeloop_veauth.py +0 -0
  49. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/opensearch_veauth.py +0 -0
  50. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/postgresql_veauth.py +0 -0
  51. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/prompt_pilot_veauth.py +0 -0
  52. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/utils.py +0 -0
  53. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/vesearch_veauth.py +0 -0
  54. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/__init__.py +0 -0
  55. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_deploy.py +0 -0
  56. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_eval.py +0 -0
  57. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_init.py +0 -0
  58. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_kb.py +0 -0
  59. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_pipeline.py +0 -0
  60. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_prompt.py +0 -0
  61. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_uploadevalset.py +0 -0
  62. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_web.py +0 -0
  63. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cloud/__init__.py +0 -0
  64. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cloud/cloud_agent_engine.py +0 -0
  65. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cloud/cloud_app.py +0 -0
  66. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/config.py +0 -0
  67. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/configs/__init__.py +0 -0
  68. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/configs/database_configs.py +0 -0
  69. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/configs/model_configs.py +0 -0
  70. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/configs/tool_configs.py +0 -0
  71. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/configs/tracing_configs.py +0 -0
  72. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/__init__.py +0 -0
  73. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/adk_evaluator/__init__.py +0 -0
  74. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/adk_evaluator/adk_evaluator.py +0 -0
  75. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/base_evaluator.py +0 -0
  76. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/deepeval_evaluator/__init__.py +0 -0
  77. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/deepeval_evaluator/deepeval_evaluator.py +0 -0
  78. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/eval_set_file_loader.py +0 -0
  79. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/eval_set_recorder.py +0 -0
  80. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/types.py +0 -0
  81. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/utils/prometheus.py +0 -0
  82. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/__init__.py +0 -0
  83. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_apig/__init__.py +0 -0
  84. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_apig/ve_apig.py +0 -0
  85. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_apig/ve_apig_utils.py +0 -0
  86. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_code_pipeline/__init__.py +0 -0
  87. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_code_pipeline/ve_code_pipeline.py +0 -0
  88. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_cozeloop/__init__.py +0 -0
  89. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_cozeloop/ve_cozeloop.py +0 -0
  90. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_cr/__init__.py +0 -0
  91. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_cr/ve_cr.py +0 -0
  92. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/__init__.py +0 -0
  93. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/cookiecutter.json +0 -0
  94. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/__init__.py +0 -0
  95. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/clean.py +0 -0
  96. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/config.yaml.example +0 -0
  97. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/deploy.py +0 -0
  98. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/__init__.py +0 -0
  99. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/agent.py +0 -0
  100. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/app.py +0 -0
  101. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/requirements.txt +0 -0
  102. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/run.sh +0 -0
  103. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/{{ cookiecutter.app_name }}/__init__.py +0 -0
  104. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/{{ cookiecutter.app_name }}/agent.py +0 -0
  105. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/ve_faas.py +0 -0
  106. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/ve_faas_utils.py +0 -0
  107. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/cookiecutter.json +0 -0
  108. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/__init__.py +0 -0
  109. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/clean.py +0 -0
  110. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/config.yaml.example +0 -0
  111. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/deploy.py +0 -0
  112. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/Dockerfile +0 -0
  113. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/app.py +0 -0
  114. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/init_db.py +0 -0
  115. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/models.py +0 -0
  116. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/requirements.txt +0 -0
  117. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/run.sh +0 -0
  118. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/static/css/style.css +0 -0
  119. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/static/js/admin.js +0 -0
  120. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/templates/admin/dashboard.html +0 -0
  121. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/templates/admin/edit_post.html +0 -0
  122. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/templates/admin/login.html +0 -0
  123. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/templates/admin/posts.html +0 -0
  124. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/templates/base.html +0 -0
  125. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/templates/index.html +0 -0
  126. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/templates/post.html +0 -0
  127. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_prompt_pilot/__init__.py +0 -0
  128. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_prompt_pilot/ve_prompt_pilot.py +0 -0
  129. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_tls/__init__.py +0 -0
  130. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_tls/utils.py +0 -0
  131. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_tls/ve_tls.py +0 -0
  132. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_viking_db_memory/__init__.py +0 -0
  133. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_viking_db_memory/ve_viking_db_memory.py +0 -0
  134. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/__init__.py +0 -0
  135. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/__init__.py +0 -0
  136. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/base_backend.py +0 -0
  137. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/in_memory_backend.py +0 -0
  138. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/opensearch_backend.py +0 -0
  139. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/redis_backend.py +0 -0
  140. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/entry.py +0 -0
  141. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/knowledgebase.py +0 -0
  142. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/__init__.py +0 -0
  143. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/__init__.py +0 -0
  144. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/base_backend.py +0 -0
  145. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/in_memory_backend.py +0 -0
  146. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/mem0_backend.py +0 -0
  147. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/opensearch_backend.py +0 -0
  148. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/redis_backend.py +0 -0
  149. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory.py +0 -0
  150. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory_backends/__init__.py +0 -0
  151. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory_backends/base_backend.py +0 -0
  152. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory_backends/mysql_backend.py +0 -0
  153. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory_backends/postgresql_backend.py +0 -0
  154. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory_backends/sqlite_backend.py +0 -0
  155. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory_processor.py +0 -0
  156. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/prompts/__init__.py +0 -0
  157. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/prompts/agent_default_prompt.py +0 -0
  158. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/prompts/prompt_evaluator.py +0 -0
  159. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/prompts/prompt_memory_processor.py +0 -0
  160. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/prompts/prompt_optimization.py +0 -0
  161. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/runner.py +0 -0
  162. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/__init__.py +0 -0
  163. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/__init__.py +0 -0
  164. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/image_edit.py +0 -0
  165. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/lark.py +0 -0
  166. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/las.py +0 -0
  167. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/load_knowledgebase.py +0 -0
  168. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/mcp_router.py +0 -0
  169. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/vesearch.py +0 -0
  170. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/video_generate.py +0 -0
  171. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/web_scraper.py +0 -0
  172. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/web_search.py +0 -0
  173. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/demo_tools.py +0 -0
  174. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/load_knowledgebase_tool.py +0 -0
  175. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/sandbox/__init__.py +0 -0
  176. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/sandbox/browser_sandbox.py +0 -0
  177. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/sandbox/code_sandbox.py +0 -0
  178. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/sandbox/computer_sandbox.py +0 -0
  179. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/__init__.py +0 -0
  180. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/base_tracer.py +0 -0
  181. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/__init__.py +0 -0
  182. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/attributes/attributes.py +0 -0
  183. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/attributes/extractors/common_attributes_extractors.py +0 -0
  184. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/attributes/extractors/llm_attributes_extractors.py +0 -0
  185. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/attributes/extractors/tool_attributes_extractors.py +0 -0
  186. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/attributes/extractors/types.py +0 -0
  187. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/exporters/__init__.py +0 -0
  188. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/exporters/apmplus_exporter.py +0 -0
  189. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/exporters/base_exporter.py +0 -0
  190. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/exporters/cozeloop_exporter.py +0 -0
  191. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/exporters/inmemory_exporter.py +0 -0
  192. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/exporters/tls_exporter.py +0 -0
  193. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/opentelemetry_tracer.py +0 -0
  194. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/telemetry.py +0 -0
  195. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/types.py +0 -0
  196. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/utils/__init__.py +0 -0
  197. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/utils/logger.py +0 -0
  198. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/utils/mcp_utils.py +0 -0
  199. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/utils/misc.py +0 -0
  200. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/utils/patches.py +0 -0
  201. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/utils/volcengine_sign.py +0 -0
  202. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk_python.egg-info/dependency_links.txt +0 -0
  203. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk_python.egg-info/entry_points.txt +0 -0
  204. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk_python.egg-info/requires.txt +0 -0
  205. {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk_python.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: veadk-python
3
- Version: 0.2.12
3
+ Version: 0.2.14
4
4
  Summary: Volcengine agent development kit, integrations with Volcengine cloud services.
5
5
  Author-email: Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>, Zhi Han <sliverydayday@gmail.com>, Meng Wang <mengwangwm@gmail.com>
6
6
  License: Apache License
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "veadk-python"
3
- version = "0.2.12"
3
+ version = "0.2.14"
4
4
  description = "Volcengine agent development kit, integrations with Volcengine cloud services."
5
5
  readme = "README.md"
6
6
  requires-python = ">=3.10"
@@ -18,6 +18,7 @@ import click
18
18
  from veadk.cli.cli_deploy import deploy
19
19
  from veadk.cli.cli_eval import eval
20
20
  from veadk.cli.cli_init import init
21
+ from veadk.cli.cli_create import create
21
22
  from veadk.cli.cli_kb import kb
22
23
  from veadk.cli.cli_pipeline import pipeline
23
24
  from veadk.cli.cli_prompt import prompt
@@ -37,6 +38,7 @@ def veadk():
37
38
 
38
39
  veadk.add_command(deploy)
39
40
  veadk.add_command(init)
41
+ veadk.add_command(create)
40
42
  veadk.add_command(prompt)
41
43
  veadk.add_command(web)
42
44
  veadk.add_command(pipeline)
@@ -0,0 +1,122 @@
1
+ # Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+
15
+ import click
16
+ import shutil
17
+ from pathlib import Path
18
+
19
+ _CONFIG_YAML_TEMPLATE = """\
20
+ model:
21
+ agent:
22
+ name: doubao-seed-1-6-251015
23
+ api_key: {ark_api_key}
24
+ video:
25
+ name: doubao-seedance-1-0-pro-250528
26
+ # if you want to use different api_key, just uncomment following line and complete api_key
27
+ # api_key:
28
+ image:
29
+ name: doubao-seedream-4-0-250828
30
+ # if you want to use different api_key, just uncomment following line and complete api_key
31
+ # api_key:
32
+
33
+ logging:
34
+ # ERROR
35
+ # WARNING
36
+ # INFO
37
+ # DEBUG
38
+ level: DEBUG
39
+ """
40
+
41
+ _INIT_PY_TEMPLATE = """\
42
+ from . import agent
43
+ """
44
+
45
+ _AGENT_PY_TEMPLATE = """\
46
+ from veadk import Agent
47
+
48
+ root_agent = Agent(
49
+ name="root_agent",
50
+ description="A helpful assistant for user questions.",
51
+ instruction="Answer user questions to the best of your knowledge",
52
+ )
53
+ """
54
+
55
+ _SUCCESS_MSG = """\
56
+ Agent '{agent_name}' created successfully at '{agent_folder}':
57
+ - config.yaml
58
+ - {agent_name}/__init__.py
59
+ - {agent_name}/agent.py
60
+
61
+ You can run the agent by executing: cd {agent_name} && veadk web
62
+ """
63
+
64
+
65
+ def _prompt_for_ark_api_key() -> str:
66
+ click.secho(
67
+ "An API key is required to run the agent. See https://www.volcengine.com/docs/82379/1541594 for details.",
68
+ fg="green",
69
+ )
70
+ click.echo("You have two options:")
71
+ click.echo(" 1. Enter the API key now.")
72
+ click.echo(" 2. Configure it later in the generated config.yaml file.")
73
+ choice = click.prompt("Please select an option", type=click.Choice(["1", "2"]))
74
+ if choice == "1":
75
+ return click.prompt("Please enter your ARK API key")
76
+ else:
77
+ click.secho(
78
+ "You can set the `api_key` in the config.yaml file later.", fg="yellow"
79
+ )
80
+ return ""
81
+
82
+
83
+ def _generate_files(agent_name: str, ark_api_key: str, target_dir_path: Path) -> None:
84
+ agent_dir_path = target_dir_path / agent_name
85
+ agent_dir_path.mkdir(parents=True, exist_ok=True)
86
+ config_yaml_path = target_dir_path / "config.yaml"
87
+ init_file_path = agent_dir_path / "__init__.py"
88
+ agent_file_path = agent_dir_path / "agent.py"
89
+
90
+ config_yaml_content = _CONFIG_YAML_TEMPLATE.format(ark_api_key=ark_api_key)
91
+ config_yaml_path.write_text(config_yaml_content)
92
+ init_file_path.write_text(_INIT_PY_TEMPLATE)
93
+ agent_file_path.write_text(_AGENT_PY_TEMPLATE)
94
+
95
+ click.secho(
96
+ _SUCCESS_MSG.format(agent_name=agent_name, agent_folder=target_dir_path),
97
+ fg="green",
98
+ )
99
+
100
+
101
+ @click.command()
102
+ @click.option("--agent-name", help="The name of the agent.")
103
+ @click.option("--ark-api-key", help="The ARK API key.")
104
+ def create(agent_name: str, ark_api_key: str) -> None:
105
+ """Creates a new agent in the current folder with prepopulated agent template."""
106
+ if not agent_name:
107
+ agent_name = click.prompt("Enter the agent name")
108
+ if not ark_api_key:
109
+ ark_api_key = _prompt_for_ark_api_key()
110
+
111
+ cwd = Path.cwd()
112
+ target_dir_path = cwd / agent_name
113
+
114
+ if target_dir_path.exists() and any(target_dir_path.iterdir()):
115
+ if not click.confirm(
116
+ f"Directory '{target_dir_path}' already exists and is not empty. Do you want to overwrite it?"
117
+ ):
118
+ click.secho("Operation cancelled.", fg="red")
119
+ return
120
+ shutil.rmtree(target_dir_path)
121
+
122
+ _generate_files(agent_name, ark_api_key, target_dir_path)
@@ -62,9 +62,6 @@ DEFAULT_TOS_BUCKET_NAME = "ark-tutorial"
62
62
 
63
63
  DEFAULT_COZELOOP_SPACE_NAME = "VeADK Space"
64
64
 
65
- DEFAULT_TEXT_TO_IMAGE_MODEL_NAME = "doubao-seedream-3-0-t2i-250415"
66
- DEFAULT_TEXT_TO_IMAGE_MODEL_API_BASE = "https://ark.cn-beijing.volces.com/api/v3/"
67
-
68
65
  DEFAULT_IMAGE_EDIT_MODEL_NAME = "doubao-seededit-3-0-i2i-250628"
69
66
  DEFAULT_IMAGE_EDIT_MODEL_API_BASE = "https://ark.cn-beijing.volces.com/api/v3/"
70
67
 
@@ -36,11 +36,14 @@ class VeTOS:
36
36
  self,
37
37
  ak: str = "",
38
38
  sk: str = "",
39
+ session_token: str = "",
39
40
  region: str = "cn-beijing",
40
41
  bucket_name: str = DEFAULT_TOS_BUCKET_NAME,
41
42
  ) -> None:
42
43
  self.ak = ak if ak else os.getenv("VOLCENGINE_ACCESS_KEY", "")
43
44
  self.sk = sk if sk else os.getenv("VOLCENGINE_SECRET_KEY", "")
45
+ self.session_token = session_token
46
+
44
47
  # Add empty value validation
45
48
  if not self.ak or not self.sk:
46
49
  raise ValueError(
@@ -71,6 +74,7 @@ class VeTOS:
71
74
  self._client = self._tos_module.TosClientV2(
72
75
  ak=self.ak,
73
76
  sk=self.sk,
77
+ security_token=self.session_token,
74
78
  endpoint=f"tos-{self.region}.volces.com",
75
79
  region=self.region,
76
80
  )
@@ -85,6 +89,7 @@ class VeTOS:
85
89
  self._client = self._tos_module.TosClientV2(
86
90
  self.ak,
87
91
  self.sk,
92
+ security_token=self.session_token,
88
93
  endpoint=f"tos-{self.region}.volces.com",
89
94
  region=self.region,
90
95
  )
@@ -46,6 +46,7 @@ def build_vikingdb_knowledgebase_request(
46
46
  path: str,
47
47
  volcengine_access_key: str,
48
48
  volcengine_secret_key: str,
49
+ session_token: str = "",
49
50
  method: Literal["GET", "POST", "PUT", "DELETE"] = "POST",
50
51
  region: str = "cn-beijing",
51
52
  params=None,
@@ -85,7 +86,7 @@ def build_vikingdb_knowledgebase_request(
85
86
  r.set_body(json.dumps(data))
86
87
 
87
88
  credentials = Credentials(
88
- volcengine_access_key, volcengine_secret_key, "air", region
89
+ volcengine_access_key, volcengine_secret_key, "air", region, session_token
89
90
  )
90
91
  SignerV4.sign(r, credentials)
91
92
  return r
@@ -14,6 +14,7 @@
14
14
 
15
15
  import asyncio
16
16
  import json
17
+ import os
17
18
  import re
18
19
  from pathlib import Path
19
20
  from typing import Any, Literal
@@ -23,7 +24,7 @@ from pydantic import Field
23
24
  from typing_extensions import override
24
25
 
25
26
  import veadk.config # noqa E401
26
- from veadk.config import getenv
27
+ from veadk.auth.veauth.utils import get_credential_from_vefaas_iam
27
28
  from veadk.configs.database_configs import NormalTOSConfig, TOSConfig
28
29
  from veadk.knowledgebase.backends.base_backend import BaseKnowledgebaseBackend
29
30
  from veadk.knowledgebase.backends.utils import build_vikingdb_knowledgebase_request
@@ -58,14 +59,16 @@ def get_files_in_directory(directory: str):
58
59
 
59
60
 
60
61
  class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend):
61
- volcengine_access_key: str = Field(
62
- default_factory=lambda: getenv("VOLCENGINE_ACCESS_KEY")
62
+ volcengine_access_key: str | None = Field(
63
+ default_factory=lambda: os.getenv("VOLCENGINE_ACCESS_KEY")
63
64
  )
64
65
 
65
- volcengine_secret_key: str = Field(
66
- default_factory=lambda: getenv("VOLCENGINE_SECRET_KEY")
66
+ volcengine_secret_key: str | None = Field(
67
+ default_factory=lambda: os.getenv("VOLCENGINE_SECRET_KEY")
67
68
  )
68
69
 
70
+ session_token: str = ""
71
+
69
72
  volcengine_project: str = "default"
70
73
  """VikingDB knowledgebase project in Volcengine console platform. Default by `default`"""
71
74
 
@@ -75,6 +78,15 @@ class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend):
75
78
  tos_config: TOSConfig | NormalTOSConfig = Field(default_factory=TOSConfig)
76
79
  """TOS config, used to upload files to TOS"""
77
80
 
81
+ def model_post_init(self, __context: Any) -> None:
82
+ self.precheck_index_naming()
83
+
84
+ # check whether collection exist, if not, create it
85
+ if not self.collection_status()["existed"]:
86
+ logger.warning(
87
+ f"VikingDB knowledgebase collection {self.index} does not exist, please create it first..."
88
+ )
89
+
78
90
  def precheck_index_naming(self):
79
91
  if not (
80
92
  isinstance(self.index, str)
@@ -86,18 +98,21 @@ class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend):
86
98
  "it must start with an English letter, contain only letters, numbers, and underscores, and have a length of 1-128."
87
99
  )
88
100
 
89
- def model_post_init(self, __context: Any) -> None:
90
- self.precheck_index_naming()
91
-
92
- # check whether collection exist, if not, create it
93
- if not self.collection_status()["existed"]:
94
- logger.warning(
95
- f"VikingDB knowledgebase collection {self.index} does not exist, please create it first..."
96
- )
97
-
98
- self._tos_client = VeTOS(
99
- ak=self.volcengine_access_key,
100
- sk=self.volcengine_secret_key,
101
+ def _get_tos_client(self) -> VeTOS:
102
+ volcengine_access_key = self.volcengine_access_key
103
+ volcengine_secret_key = self.volcengine_secret_key
104
+ session_token = self.session_token
105
+
106
+ if not (volcengine_access_key and volcengine_secret_key):
107
+ cred = get_credential_from_vefaas_iam()
108
+ volcengine_access_key = cred.access_key_id
109
+ volcengine_secret_key = cred.secret_access_key
110
+ session_token = cred.session_token
111
+
112
+ return VeTOS(
113
+ ak=volcengine_access_key,
114
+ sk=volcengine_secret_key,
115
+ session_token=session_token,
101
116
  region=self.tos_config.region,
102
117
  bucket_name=self.tos_config.bucket,
103
118
  )
@@ -404,6 +419,8 @@ class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend):
404
419
  metadata: dict | None = None,
405
420
  ) -> str:
406
421
  # Here, we set the metadata via the TOS object, ref: https://www.volcengine.com/docs/84313/1254624
422
+ self._tos_client = self._get_tos_client()
423
+
407
424
  self._tos_client.bucket_name = tos_bucket_name
408
425
  coro = self._tos_client.upload(
409
426
  object_key=object_key,
@@ -504,10 +521,21 @@ class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend):
504
521
  ) -> dict:
505
522
  VIKINGDB_KNOWLEDGEBASE_BASE_URL = "api-knowledgebase.mlp.cn-beijing.volces.com"
506
523
 
524
+ volcengine_access_key = self.volcengine_access_key
525
+ volcengine_secret_key = self.volcengine_secret_key
526
+ session_token = self.session_token
527
+
528
+ if not (volcengine_access_key and volcengine_secret_key):
529
+ cred = get_credential_from_vefaas_iam()
530
+ volcengine_access_key = cred.access_key_id
531
+ volcengine_secret_key = cred.secret_access_key
532
+ session_token = cred.session_token
533
+
507
534
  request = build_vikingdb_knowledgebase_request(
508
535
  path=path,
509
- volcengine_access_key=self.volcengine_access_key,
510
- volcengine_secret_key=self.volcengine_secret_key,
536
+ volcengine_access_key=volcengine_access_key,
537
+ volcengine_secret_key=volcengine_secret_key,
538
+ session_token=session_token,
511
539
  method=method,
512
540
  data=body,
513
541
  )
@@ -104,6 +104,15 @@ class LongTermMemory(BaseMemoryService, BaseModel):
104
104
 
105
105
  # Once user define backend config, use it directly
106
106
  if self.backend_config:
107
+ if "index" not in self.backend_config:
108
+ logger.warning(
109
+ "Attribute `index` not provided in backend_config, use `index` or `app_name` instead."
110
+ )
111
+ self.backend_config["index"] = self.index or self.app_name
112
+
113
+ logger.debug(
114
+ f"Init {self.backend}, Use provided backend config: {self.backend_config}"
115
+ )
107
116
  self._backend = _get_backend_cls(self.backend)(**self.backend_config)
108
117
  return
109
118
 
@@ -49,7 +49,23 @@ class VikingDBLTMBackend(BaseLongTermMemoryBackend):
49
49
  region: str = "cn-beijing"
50
50
  """VikingDB memory region"""
51
51
 
52
+ memory_type: list[str] = Field(default_factory=list)
53
+
52
54
  def model_post_init(self, __context: Any) -> None:
55
+ # We get memory type from:
56
+ # 1. user input
57
+ # 2. environment variable
58
+ # 3. default value
59
+ if not self.memory_type:
60
+ env_memory_type = os.getenv("DATABASE_VIKINGMEM_MEMORY_TYPE")
61
+ if env_memory_type:
62
+ # "event_1, event_2" -> ["event_1", "event_2"]
63
+ self.memory_type = [x.strip() for x in env_memory_type.split(",")]
64
+ else:
65
+ self.memory_type = ["sys_event_v1", "event_v1"]
66
+
67
+ logger.info(f"Using memory type: {self.memory_type}")
68
+
53
69
  # check whether collection exist, if not, create it
54
70
  if not self._collection_exist():
55
71
  self._create_collection()
@@ -68,17 +84,23 @@ class VikingDBLTMBackend(BaseLongTermMemoryBackend):
68
84
  try:
69
85
  client = self._get_client()
70
86
  client.get_collection(collection_name=self.index)
87
+ logger.info(f"Collection {self.index} exist.")
71
88
  return True
72
89
  except Exception:
90
+ logger.info(f"Collection {self.index} not exist.")
73
91
  return False
74
92
 
75
93
  def _create_collection(self) -> None:
94
+ logger.info(
95
+ f"Create collection with collection_name={self.index}, builtin_event_types={self.memory_type}"
96
+ )
76
97
  client = self._get_client()
77
98
  response = client.create_collection(
78
99
  collection_name=self.index,
79
100
  description="Created by Volcengine Agent Development Kit VeADK",
80
- builtin_event_types=["sys_event_v1"],
101
+ builtin_event_types=self.memory_type,
81
102
  )
103
+ logger.debug(f"Create collection with response {response}")
82
104
  return response
83
105
 
84
106
  def _get_client(self) -> VikingDBMemoryClient:
@@ -135,10 +157,7 @@ class VikingDBLTMBackend(BaseLongTermMemoryBackend):
135
157
  def search_memory(
136
158
  self, user_id: str, query: str, top_k: int, **kwargs
137
159
  ) -> list[str]:
138
- filter = {
139
- "user_id": user_id,
140
- "memory_type": ["sys_event_v1", "event_v1"],
141
- }
160
+ filter = {"user_id": user_id, "memory_type": self.memory_type}
142
161
 
143
162
  logger.debug(
144
163
  f"Request for search memory in VikingDB: filter={filter}, collection_name={self.index}, query={query}, limit={top_k}"
@@ -12,7 +12,10 @@
12
12
  # See the License for the specific language governing permissions and
13
13
  # limitations under the License.
14
14
 
15
+ import asyncio
15
16
  import base64
17
+ import concurrent.futures
18
+ import contextvars
16
19
  import json
17
20
  import mimetypes
18
21
  import traceback
@@ -27,16 +30,12 @@ from volcenginesdkarkruntime.types.images.images import SequentialImageGeneratio
27
30
 
28
31
  from veadk.config import getenv, settings
29
32
  from veadk.consts import (
30
- DEFAULT_IMAGE_GENERATE_MODEL_NAME,
31
33
  DEFAULT_IMAGE_GENERATE_MODEL_API_BASE,
34
+ DEFAULT_IMAGE_GENERATE_MODEL_NAME,
32
35
  )
33
36
  from veadk.utils.logger import get_logger
34
37
  from veadk.utils.misc import formatted_timestamp, read_file_to_bytes
35
38
  from veadk.version import VERSION
36
- import asyncio
37
- import concurrent.futures
38
- import contextvars
39
-
40
39
 
41
40
  logger = get_logger(__name__)
42
41
 
@@ -48,7 +47,7 @@ client = Ark(
48
47
  )
49
48
 
50
49
  executor = concurrent.futures.ThreadPoolExecutor(max_workers=8)
51
- tracer = trace.get_tracer("gcp.vertex.agent")
50
+ tracer = trace.get_tracer("veadk")
52
51
 
53
52
 
54
53
  def _build_input_parts(item: dict, task_type: str, image_field):
@@ -205,8 +204,10 @@ def handle_single_task_sync(
205
204
 
206
205
 
207
206
  async def image_generate(tasks: list[dict], tool_context) -> Dict:
208
- """
209
- Seedream 4.0: batch image generation via tasks.
207
+ """Generate images with Seedream 4.0.
208
+
209
+ Commit batch image generation requests via tasks.
210
+
210
211
  Args:
211
212
  tasks (list[dict]):
212
213
  A list of image-generation tasks. Each task is a dict.
@@ -284,12 +285,27 @@ async def image_generate(tasks: list[dict], tool_context) -> Dict:
284
285
  - 如果想要指定生成组图的数量,请在prompt里添加数量说明,例如:"生成3张图片"。
285
286
  - size 推荐使用 2048x2048 或表格里的标准比例,确保生成质量。
286
287
  """
287
- logger.debug(
288
- f"Using model: {getenv('MODEL_IMAGE_NAME', DEFAULT_IMAGE_GENERATE_MODEL_NAME)}"
289
- )
288
+ model = getenv("MODEL_IMAGE_NAME", DEFAULT_IMAGE_GENERATE_MODEL_NAME)
289
+
290
+ if model.startswith("doubao-seedream-3-0"):
291
+ logger.error(
292
+ f"Image generation by Doubao Seedream 3.0 ({model}) is depracated. Please use Doubao Seedream 4.0 (e.g., doubao-seedream-4-0-250828) instead."
293
+ )
294
+ return {
295
+ "status": "failed",
296
+ "success_list": [],
297
+ "error_list": [
298
+ "Image generation by Doubao Seedream 3.0 ({model}) is depracated. Please use Doubao Seedream 4.0 (e.g., doubao-seedream-4-0-250828) instead."
299
+ ],
300
+ }
301
+
302
+ logger.debug(f"Using model to generate image: {model}")
303
+
290
304
  success_list: list[dict] = []
291
305
  error_list: list[str] = []
306
+
292
307
  logger.debug(f"image_generate tasks: {tasks}")
308
+
293
309
  with tracer.start_as_current_span("image_generate"):
294
310
  base_ctx = contextvars.copy_context()
295
311
 
@@ -0,0 +1,81 @@
1
+ # Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+
15
+ import json
16
+
17
+ from google.adk.tools import ToolContext
18
+
19
+ from veadk.config import getenv
20
+ from veadk.utils.logger import get_logger
21
+ from veadk.utils.volcengine_sign import ve_request
22
+
23
+ logger = get_logger(__name__)
24
+
25
+
26
+ def run_code(code: str, language: str, tool_context: ToolContext) -> str:
27
+ """Run code in a code sandbox and return the output.
28
+
29
+ Args:
30
+ code (str): The code to run.
31
+ language (str): The programming language of the code. Language must be one of the supported languages: python3.
32
+
33
+ Returns:
34
+ str: The output of the code execution.
35
+ """
36
+
37
+ tool_id = getenv("AGENTKIT_TOOL_ID")
38
+ host = getenv("AGENTKIT_TOOL_HOST") # temporary host for code run tool
39
+ service = getenv(
40
+ "AGENTKIT_TOOL_SERVICE_CODE"
41
+ ) # temporary service for code run tool
42
+ region = getenv("AGENTKIT_TOOL_REGION", "cn-beijing")
43
+
44
+ session_id = tool_context._invocation_context.session.id
45
+
46
+ logger.debug(
47
+ f"Running code in language: {language}, session_id={session_id}, code={code}, tool_id={tool_id}, host={host}, service={service}, region={region}"
48
+ )
49
+
50
+ access_key = getenv("VOLCENGINE_ACCESS_KEY")
51
+ secret_key = getenv("VOLCENGINE_SECRET_KEY")
52
+
53
+ res = ve_request(
54
+ request_body={
55
+ "ToolId": tool_id,
56
+ "UserSessionId": session_id,
57
+ "OperationType": "RunCode",
58
+ "OperationPayload": json.dumps(
59
+ {
60
+ "code": code,
61
+ "timeout": 30,
62
+ "kernel_name": language,
63
+ }
64
+ ),
65
+ },
66
+ action="InvokeTool",
67
+ ak=access_key,
68
+ sk=secret_key,
69
+ service=service,
70
+ version="2025-10-30",
71
+ region=region,
72
+ host=host,
73
+ )
74
+
75
+ logger.debug(f"Invoke run code response: {res}")
76
+
77
+ try:
78
+ return res["Result"]["Result"]
79
+ except KeyError as e:
80
+ logger.error(f"Error occurred while running code: {e}, response is {res}")
81
+ return res
@@ -12,4 +12,4 @@
12
12
  # See the License for the specific language governing permissions and
13
13
  # limitations under the License.
14
14
 
15
- VERSION = "0.2.12"
15
+ VERSION = "0.2.14"
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: veadk-python
3
- Version: 0.2.12
3
+ Version: 0.2.14
4
4
  Summary: Volcengine agent development kit, integrations with Volcengine cloud services.
5
5
  Author-email: Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>, Zhi Han <sliverydayday@gmail.com>, Meng Wang <mengwangwm@gmail.com>
6
6
  License: Apache License
@@ -42,6 +42,7 @@ veadk/auth/veauth/utils.py
42
42
  veadk/auth/veauth/vesearch_veauth.py
43
43
  veadk/cli/__init__.py
44
44
  veadk/cli/cli.py
45
+ veadk/cli/cli_create.py
45
46
  veadk/cli/cli_deploy.py
46
47
  veadk/cli/cli_eval.py
47
48
  veadk/cli/cli_init.py
@@ -156,13 +157,13 @@ veadk/tools/__init__.py
156
157
  veadk/tools/demo_tools.py
157
158
  veadk/tools/load_knowledgebase_tool.py
158
159
  veadk/tools/builtin_tools/__init__.py
159
- veadk/tools/builtin_tools/generate_image.py
160
160
  veadk/tools/builtin_tools/image_edit.py
161
161
  veadk/tools/builtin_tools/image_generate.py
162
162
  veadk/tools/builtin_tools/lark.py
163
163
  veadk/tools/builtin_tools/las.py
164
164
  veadk/tools/builtin_tools/load_knowledgebase.py
165
165
  veadk/tools/builtin_tools/mcp_router.py
166
+ veadk/tools/builtin_tools/run_code.py
166
167
  veadk/tools/builtin_tools/vesearch.py
167
168
  veadk/tools/builtin_tools/video_generate.py
168
169
  veadk/tools/builtin_tools/web_scraper.py