aixtools 0.1.4__tar.gz → 0.1.6__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 aixtools might be problematic. Click here for more details.

Files changed (139) hide show
  1. {aixtools-0.1.4 → aixtools-0.1.6}/.env_template +22 -6
  2. aixtools-0.1.4/.github/workflows/lint.yml → aixtools-0.1.6/.github/workflows/lint-and-test.yml +4 -0
  3. aixtools-0.1.6/PKG-INFO +668 -0
  4. aixtools-0.1.6/README.md +638 -0
  5. aixtools-0.1.4/PKG-INFO → aixtools-0.1.6/README.ori.md +41 -46
  6. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/_version.py +3 -3
  7. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/a2a/app.py +1 -1
  8. aixtools-0.1.6/aixtools/a2a/google_sdk/__init__.py +0 -0
  9. aixtools-0.1.6/aixtools/a2a/google_sdk/card.py +27 -0
  10. aixtools-0.1.6/aixtools/a2a/google_sdk/pydantic_ai_adapter/agent_executor.py +199 -0
  11. aixtools-0.1.6/aixtools/a2a/google_sdk/pydantic_ai_adapter/storage.py +26 -0
  12. aixtools-0.1.6/aixtools/a2a/google_sdk/remote_agent_connection.py +88 -0
  13. aixtools-0.1.6/aixtools/a2a/google_sdk/utils.py +59 -0
  14. aixtools-0.1.6/aixtools/agents/prompt.py +97 -0
  15. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/context.py +5 -0
  16. aixtools-0.1.6/aixtools/google/client.py +25 -0
  17. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/logging/logging_config.py +45 -0
  18. aixtools-0.1.6/aixtools/mcp/client.py +274 -0
  19. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/server/utils.py +3 -3
  20. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/utils/config.py +13 -0
  21. aixtools-0.1.6/aixtools/utils/files.py +17 -0
  22. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/utils/utils.py +7 -0
  23. aixtools-0.1.6/aixtools/vault/__init__.py +7 -0
  24. aixtools-0.1.6/aixtools/vault/vault.py +73 -0
  25. aixtools-0.1.6/aixtools.egg-info/PKG-INFO +668 -0
  26. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools.egg-info/SOURCES.txt +39 -3
  27. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools.egg-info/requires.txt +3 -0
  28. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools.egg-info/top_level.txt +1 -0
  29. {aixtools-0.1.4 → aixtools-0.1.6}/pyproject.toml +10 -2
  30. aixtools-0.1.6/scripts/test.sh +23 -0
  31. aixtools-0.1.6/tests/__init__.py +0 -0
  32. aixtools-0.1.6/tests/unit/__init__.py +0 -0
  33. aixtools-0.1.6/tests/unit/a2a/__init__.py +0 -0
  34. aixtools-0.1.6/tests/unit/a2a/google_sdk/__init__.py +0 -0
  35. aixtools-0.1.6/tests/unit/a2a/google_sdk/pydantic_ai_adapter/__init__.py +0 -0
  36. aixtools-0.1.6/tests/unit/a2a/google_sdk/pydantic_ai_adapter/test_agent_executor.py +188 -0
  37. aixtools-0.1.6/tests/unit/a2a/google_sdk/pydantic_ai_adapter/test_storage.py +156 -0
  38. aixtools-0.1.6/tests/unit/a2a/google_sdk/test_card.py +114 -0
  39. aixtools-0.1.6/tests/unit/a2a/google_sdk/test_remote_agent_connection.py +413 -0
  40. aixtools-0.1.6/tests/unit/a2a/google_sdk/test_utils.py +208 -0
  41. aixtools-0.1.6/tests/unit/agents/__init__.py +0 -0
  42. aixtools-0.1.6/tests/unit/agents/test_prompt.py +363 -0
  43. aixtools-0.1.6/tests/unit/google/__init__.py +1 -0
  44. aixtools-0.1.6/tests/unit/google/test_client.py +233 -0
  45. aixtools-0.1.6/tests/unit/mcp/__init__.py +0 -0
  46. aixtools-0.1.6/tests/unit/mcp/test_client.py +242 -0
  47. aixtools-0.1.6/tests/unit/server/__init__.py +0 -0
  48. aixtools-0.1.6/tests/unit/server/test_path.py +225 -0
  49. aixtools-0.1.6/tests/unit/server/test_utils.py +362 -0
  50. aixtools-0.1.6/tests/unit/utils/__init__.py +0 -0
  51. aixtools-0.1.6/tests/unit/utils/test_files.py +146 -0
  52. aixtools-0.1.6/tests/unit/vault/__init__.py +0 -0
  53. aixtools-0.1.6/tests/unit/vault/test_vault.py +114 -0
  54. {aixtools-0.1.4 → aixtools-0.1.6}/uv.lock +146 -1
  55. aixtools-0.1.4/README.md +0 -328
  56. aixtools-0.1.4/aixtools/a2a/__init__.py +0 -5
  57. aixtools-0.1.4/aixtools.egg-info/PKG-INFO +0 -355
  58. {aixtools-0.1.4 → aixtools-0.1.6}/.github/workflows/build_and_publish_docker.yml +0 -0
  59. {aixtools-0.1.4 → aixtools-0.1.6}/.github/workflows/release.yml +0 -0
  60. {aixtools-0.1.4 → aixtools-0.1.6}/.gitignore +0 -0
  61. {aixtools-0.1.4 → aixtools-0.1.6}/.python-version +0 -0
  62. {aixtools-0.1.4 → aixtools-0.1.6}/.roo/rules/rules-mcp.md +0 -0
  63. {aixtools-0.1.4 → aixtools-0.1.6}/.roo/rules/rules.md +0 -0
  64. {aixtools-0.1.4 → aixtools-0.1.6}/.vscode/settings.json +0 -0
  65. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/config.toml +0 -0
  66. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/bn.json +0 -0
  67. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/en-US.json +0 -0
  68. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/gu.json +0 -0
  69. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/he-IL.json +0 -0
  70. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/hi.json +0 -0
  71. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/ja.json +0 -0
  72. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/kn.json +0 -0
  73. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/ml.json +0 -0
  74. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/mr.json +0 -0
  75. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/nl.json +0 -0
  76. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/ta.json +0 -0
  77. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/te.json +0 -0
  78. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/.chainlit/translations/zh-CN.json +0 -0
  79. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/__init__.py +0 -0
  80. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/a2a/utils.py +0 -0
  81. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/agents/__init__.py +0 -0
  82. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/agents/agent.py +0 -0
  83. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/agents/agent_batch.py +0 -0
  84. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/app.py +0 -0
  85. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/chainlit.md +0 -0
  86. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/db/__init__.py +0 -0
  87. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/db/database.py +0 -0
  88. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/db/vector_db.py +0 -0
  89. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/log_view/__init__.py +0 -0
  90. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/log_view/app.py +0 -0
  91. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/log_view/display.py +0 -0
  92. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/log_view/export.py +0 -0
  93. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/log_view/filters.py +0 -0
  94. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/log_view/log_utils.py +0 -0
  95. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/log_view/node_summary.py +0 -0
  96. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/logfilters/__init__.py +0 -0
  97. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/logfilters/context_filter.py +0 -0
  98. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/logging/__init__.py +0 -0
  99. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/logging/log_objects.py +0 -0
  100. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/logging/mcp_log_models.py +0 -0
  101. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/logging/mcp_logger.py +0 -0
  102. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/logging/model_patch_logging.py +0 -0
  103. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/logging/open_telemetry.py +0 -0
  104. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/mcp/__init__.py +0 -0
  105. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/mcp/example_client.py +0 -0
  106. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/mcp/example_server.py +0 -0
  107. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/mcp/fast_mcp_log.py +0 -0
  108. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/mcp/faulty_mcp.py +0 -0
  109. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/model_patch/model_patch.py +0 -0
  110. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/server/__init__.py +0 -0
  111. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/server/app_mounter.py +0 -0
  112. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/server/path.py +0 -0
  113. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/testing/__init__.py +0 -0
  114. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/testing/aix_test_model.py +0 -0
  115. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/testing/mock_tool.py +0 -0
  116. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/testing/model_patch_cache.py +0 -0
  117. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/tools/doctor/__init__.py +0 -0
  118. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/tools/doctor/tool_doctor.py +0 -0
  119. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/tools/doctor/tool_recommendation.py +0 -0
  120. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/utils/__init__.py +0 -0
  121. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/utils/chainlit/cl_agent_show.py +0 -0
  122. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/utils/chainlit/cl_utils.py +0 -0
  123. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/utils/config_util.py +0 -0
  124. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/utils/enum_with_description.py +0 -0
  125. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools/utils/persisted_dict.py +0 -0
  126. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools.egg-info/dependency_links.txt +0 -0
  127. {aixtools-0.1.4 → aixtools-0.1.6}/aixtools.egg-info/entry_points.txt +0 -0
  128. {aixtools-0.1.4 → aixtools-0.1.6}/docker/mcp-base/Dockerfile +0 -0
  129. {aixtools-0.1.4 → aixtools-0.1.6}/notebooks/example_faulty_mcp_server.ipynb +0 -0
  130. {aixtools-0.1.4 → aixtools-0.1.6}/notebooks/example_mcp_server_stdio.ipynb +0 -0
  131. {aixtools-0.1.4 → aixtools-0.1.6}/notebooks/example_raw_mcp_client.ipynb +0 -0
  132. {aixtools-0.1.4 → aixtools-0.1.6}/notebooks/example_tool_doctor.ipynb +0 -0
  133. {aixtools-0.1.4 → aixtools-0.1.6}/scripts/config.sh +0 -0
  134. {aixtools-0.1.4 → aixtools-0.1.6}/scripts/lint.sh +0 -0
  135. {aixtools-0.1.4 → aixtools-0.1.6}/scripts/log_view.sh +0 -0
  136. {aixtools-0.1.4 → aixtools-0.1.6}/scripts/run_example_mcp_server.sh +0 -0
  137. {aixtools-0.1.4 → aixtools-0.1.6}/scripts/run_faulty_mcp_server.sh +0 -0
  138. {aixtools-0.1.4 → aixtools-0.1.6}/scripts/run_server.sh +0 -0
  139. {aixtools-0.1.4 → aixtools-0.1.6}/setup.cfg +0 -0
{aixtools-0.1.4 → aixtools-0.1.6}/.env_template RENAMED
@@ -4,13 +4,15 @@
4
4
  #
5
5
  #-----------------------------------------------------------------------------
6
6
 
7
-
7
+ # Select which credentials to use
8
+ # Options: azure, openai, bedrock, ollama, openrouter
8
9
  MODEL_FAMILY=azure
9
10
  VDB_EMBEDDINGS_MODEL_FAMILY=azure
10
11
 
11
12
  MODEL_TIMEOUT=120
12
13
 
13
- # Azure OpenAI
14
+ # Azure OpenAI credentials
15
+ # If 'MODEL_FAMILY=azure'
14
16
  AZURE_OPENAI_ENDPOINT=https://your_endpoint.openai.azure.com
15
17
  AZURE_OPENAI_API_VERSION=2024-06-01
16
18
  AZURE_OPENAI_API_KEY=your_secret_key
@@ -18,15 +20,29 @@ AZURE_MODEL_NAME=gpt-4o
18
20
  AZURE_OPENAI_PROVIDER_ID=azure
19
21
  AZURE_VDB_EMBEDDINGS_MODEL_NAME=text-embedding-3-small
20
22
 
21
-
22
- # Open AI
23
+ # OpenAI credentials
24
+ # If 'MODEL_FAMILY=openai'
23
25
  OPENAI_MODEL_NAME=gpt-4.5-preview
24
26
  OPENAI_API_KEY=openai_api_key
25
- OPENAI_VDB_EMBEDDINGS_MODEL_NAME=text-embedding-3-small
27
+
28
+ # Bedrock credentials
29
+ # If 'MODEL_FAMILY=bedrock'
30
+ BEDROCK_MODEL_NAME=us.anthropic.claude-sonnet-4-20250514-v1:0 # or any other enabled model
31
+ AWS_REGION=us-east-1
32
+
33
+ # OpenRouter credentials
34
+ # If 'MODEL_FAMILY=openrouter'
35
+ OPENROUTER_API_URL=https://openrouter.ai/api/v1
36
+ OPENROUTER_MODEL_NAME=anthropic/claude-3.7-sonnet
37
+ OPENROUTER_API_KEY=your_openrouter_api_key
26
38
 
27
39
  # Ollama models
40
+ # If 'MODEL_FAMILY=ollama'
28
41
  OLLAMA_MODEL_NAME=llama3.2:3b-instruct-fp16
29
- OLLAMA_LOCAL_URL=http://localhost:11434/v1
42
+ OLLAMA_URL=http://localhost:11434/v1
43
+
44
+ # Vector DB embeddings
45
+ OPENAI_VDB_EMBEDDINGS_MODEL_NAME=text-embedding-3-small
30
46
  OLLAMA_VDB_EMBEDDINGS_MODEL_NAME=snowflake-arctic-embed2:latest
31
47
 
32
48
  # Docker and CI/CD Configuration (optional)
aixtools-0.1.4/.github/workflows/lint.yml → aixtools-0.1.6/.github/workflows/lint-and-test.yml RENAMED
@@ -35,3 +35,7 @@ jobs:
35
35
  - name: Run linting
36
36
  run: |
37
37
  ./scripts/lint.sh
38
+
39
+ - name: Test
40
+ run: |
41
+ ./scripts/test.sh