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.
- {veadk_python-0.2.12 → veadk_python-0.2.14}/PKG-INFO +1 -1
- {veadk_python-0.2.12 → veadk_python-0.2.14}/pyproject.toml +1 -1
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli.py +2 -0
- veadk_python-0.2.14/veadk/cli/cli_create.py +122 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/consts.py +0 -3
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_tos/ve_tos.py +5 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/utils.py +2 -1
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/vikingdb_knowledge_backend.py +47 -19
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory.py +9 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py +24 -5
- 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
- veadk_python-0.2.14/veadk/tools/builtin_tools/run_code.py +81 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/version.py +1 -1
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk_python.egg-info/PKG-INFO +1 -1
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk_python.egg-info/SOURCES.txt +2 -1
- veadk_python-0.2.12/veadk/tools/builtin_tools/image_generate.py +0 -296
- {veadk_python-0.2.12 → veadk_python-0.2.14}/LICENSE +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/README.md +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/setup.cfg +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_agent.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_cloud.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_evaluator.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_knowledgebase.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_long_term_memory.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_multiple_agents.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_runner.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_runtime_data_collecting.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_short_term_memory.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/tests/test_tracing.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/a2a/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/a2a/agent_card.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/a2a/remote_ve_agent.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/a2a/ve_a2a_server.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/a2a/ve_agent_executor.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/a2a/ve_task_store.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/agent.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/agent_builder.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/agents/loop_agent.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/agents/parallel_agent.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/agents/sequential_agent.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/base_auth.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/apmplus_veauth.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/ark_veauth.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/base_veauth.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/cozeloop_veauth.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/opensearch_veauth.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/postgresql_veauth.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/prompt_pilot_veauth.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/utils.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/auth/veauth/vesearch_veauth.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_deploy.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_eval.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_init.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_kb.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_pipeline.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_prompt.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_uploadevalset.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cli/cli_web.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cloud/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cloud/cloud_agent_engine.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/cloud/cloud_app.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/config.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/configs/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/configs/database_configs.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/configs/model_configs.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/configs/tool_configs.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/configs/tracing_configs.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/adk_evaluator/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/adk_evaluator/adk_evaluator.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/base_evaluator.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/deepeval_evaluator/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/deepeval_evaluator/deepeval_evaluator.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/eval_set_file_loader.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/eval_set_recorder.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/types.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/evaluation/utils/prometheus.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_apig/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_apig/ve_apig.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_apig/ve_apig_utils.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_code_pipeline/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_code_pipeline/ve_code_pipeline.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_cozeloop/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_cozeloop/ve_cozeloop.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_cr/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_cr/ve_cr.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/cookiecutter.json +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/clean.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/config.yaml.example +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/deploy.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/agent.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/app.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/requirements.txt +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/run.sh +0 -0
- {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
- {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
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/ve_faas.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/ve_faas_utils.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/cookiecutter.json +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/clean.py +0 -0
- {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
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/deploy.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/Dockerfile +0 -0
- {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
- {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
- {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
- {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
- {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
- {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
- {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
- {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
- {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
- {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
- {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
- {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
- {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
- {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
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_prompt_pilot/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_prompt_pilot/ve_prompt_pilot.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_tls/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_tls/utils.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_tls/ve_tls.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_viking_db_memory/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/integrations/ve_viking_db_memory/ve_viking_db_memory.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/base_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/in_memory_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/opensearch_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/backends/redis_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/entry.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/knowledgebase/knowledgebase.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/base_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/in_memory_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/mem0_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/opensearch_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/long_term_memory_backends/redis_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory_backends/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory_backends/base_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory_backends/mysql_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory_backends/postgresql_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory_backends/sqlite_backend.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/memory/short_term_memory_processor.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/prompts/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/prompts/agent_default_prompt.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/prompts/prompt_evaluator.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/prompts/prompt_memory_processor.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/prompts/prompt_optimization.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/runner.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/image_edit.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/lark.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/las.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/load_knowledgebase.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/mcp_router.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/vesearch.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/video_generate.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/web_scraper.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/builtin_tools/web_search.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/demo_tools.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/load_knowledgebase_tool.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/sandbox/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/sandbox/browser_sandbox.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/sandbox/code_sandbox.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tools/sandbox/computer_sandbox.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/base_tracer.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/attributes/attributes.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/attributes/extractors/common_attributes_extractors.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/attributes/extractors/llm_attributes_extractors.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/attributes/extractors/tool_attributes_extractors.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/attributes/extractors/types.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/exporters/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/exporters/apmplus_exporter.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/exporters/base_exporter.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/exporters/cozeloop_exporter.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/exporters/inmemory_exporter.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/exporters/tls_exporter.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/opentelemetry_tracer.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/tracing/telemetry/telemetry.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/types.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/utils/__init__.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/utils/logger.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/utils/mcp_utils.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/utils/misc.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/utils/patches.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk/utils/volcengine_sign.py +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk_python.egg-info/dependency_links.txt +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk_python.egg-info/entry_points.txt +0 -0
- {veadk_python-0.2.12 → veadk_python-0.2.14}/veadk_python.egg-info/requires.txt +0 -0
- {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.
|
|
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
|
|
@@ -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.
|
|
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
|
|
90
|
-
self.
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
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=
|
|
510
|
-
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=
|
|
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("
|
|
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
|
-
|
|
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
|
-
|
|
288
|
-
|
|
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
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: veadk-python
|
|
3
|
-
Version: 0.2.
|
|
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
|