datarobot-genai 0.1.65__tar.gz → 0.1.66__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.
Files changed (101) hide show
  1. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/PKG-INFO +2 -2
  2. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/pyproject.toml +2 -2
  3. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/mcp_server_tools.py +2 -2
  4. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/.gitignore +0 -0
  5. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/AUTHORS +0 -0
  6. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/LICENSE +0 -0
  7. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/README.md +0 -0
  8. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/__init__.py +0 -0
  9. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/__init__.py +0 -0
  10. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/agents/__init__.py +0 -0
  11. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/agents/base.py +0 -0
  12. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/chat/__init__.py +0 -0
  13. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/chat/auth.py +0 -0
  14. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/chat/client.py +0 -0
  15. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/chat/responses.py +0 -0
  16. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/cli/__init__.py +0 -0
  17. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/cli/agent_environment.py +0 -0
  18. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/cli/agent_kernel.py +0 -0
  19. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/custom_model.py +0 -0
  20. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/mcp/__init__.py +0 -0
  21. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/mcp/common.py +0 -0
  22. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/telemetry_agent.py +0 -0
  23. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/utils/__init__.py +0 -0
  24. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/utils/auth.py +0 -0
  25. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/core/utils/urls.py +0 -0
  26. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/crewai/__init__.py +0 -0
  27. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/crewai/agent.py +0 -0
  28. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/crewai/base.py +0 -0
  29. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/crewai/events.py +0 -0
  30. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/crewai/mcp.py +0 -0
  31. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/__init__.py +0 -0
  32. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/__init__.py +0 -0
  33. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/auth.py +0 -0
  34. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/clients.py +0 -0
  35. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/config.py +0 -0
  36. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/config_utils.py +0 -0
  37. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/constants.py +0 -0
  38. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/credentials.py +0 -0
  39. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dr_mcp_server.py +0 -0
  40. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dr_mcp_server_logo.py +0 -0
  41. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_prompts/__init__.py +0 -0
  42. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_prompts/controllers.py +0 -0
  43. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_prompts/dr_lib.py +0 -0
  44. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_prompts/register.py +0 -0
  45. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_prompts/utils.py +0 -0
  46. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/__init__.py +0 -0
  47. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/deployment/__init__.py +0 -0
  48. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/deployment/adapters/__init__.py +0 -0
  49. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/deployment/adapters/base.py +0 -0
  50. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/deployment/adapters/default.py +0 -0
  51. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/deployment/adapters/drum.py +0 -0
  52. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/deployment/config.py +0 -0
  53. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/deployment/controllers.py +0 -0
  54. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/deployment/metadata.py +0 -0
  55. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/deployment/register.py +0 -0
  56. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/deployment/schemas/drum_agentic_fallback_schema.json +0 -0
  57. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/deployment/schemas/drum_prediction_fallback_schema.json +0 -0
  58. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/register.py +0 -0
  59. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/dynamic_tools/schema.py +0 -0
  60. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/exceptions.py +0 -0
  61. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/logging.py +0 -0
  62. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/mcp_instance.py +0 -0
  63. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/memory_management/__init__.py +0 -0
  64. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/memory_management/manager.py +0 -0
  65. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/memory_management/memory_tools.py +0 -0
  66. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/routes.py +0 -0
  67. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/routes_utils.py +0 -0
  68. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/server_life_cycle.py +0 -0
  69. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/telemetry.py +0 -0
  70. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/tool_filter.py +0 -0
  71. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/core/utils.py +0 -0
  72. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/server.py +0 -0
  73. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/test_utils/__init__.py +0 -0
  74. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/test_utils/integration_mcp_server.py +0 -0
  75. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/test_utils/mcp_utils_ete.py +0 -0
  76. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/test_utils/mcp_utils_integration.py +0 -0
  77. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/test_utils/openai_llm_mcp_client.py +0 -0
  78. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/test_utils/tool_base_ete.py +0 -0
  79. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/test_utils/utils.py +0 -0
  80. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/tools/__init__.py +0 -0
  81. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/tools/predictive/__init__.py +0 -0
  82. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/tools/predictive/data.py +0 -0
  83. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/tools/predictive/deployment.py +0 -0
  84. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/tools/predictive/deployment_info.py +0 -0
  85. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/tools/predictive/model.py +0 -0
  86. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/tools/predictive/predict.py +0 -0
  87. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/tools/predictive/predict_realtime.py +0 -0
  88. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/tools/predictive/project.py +0 -0
  89. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/drmcp/tools/predictive/training.py +0 -0
  90. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/langgraph/__init__.py +0 -0
  91. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/langgraph/agent.py +0 -0
  92. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/langgraph/mcp.py +0 -0
  93. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/llama_index/__init__.py +0 -0
  94. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/llama_index/agent.py +0 -0
  95. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/llama_index/base.py +0 -0
  96. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/llama_index/mcp.py +0 -0
  97. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/nat/__init__.py +0 -0
  98. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/nat/agent.py +0 -0
  99. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/nat/datarobot_llm_clients.py +0 -0
  100. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/nat/datarobot_llm_providers.py +0 -0
  101. {datarobot_genai-0.1.65 → datarobot_genai-0.1.66}/src/datarobot_genai/py.typed +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: datarobot-genai
3
- Version: 0.1.65
3
+ Version: 0.1.66
4
4
  Summary: Generic helpers for GenAI
5
5
  Project-URL: Homepage, https://github.com/datarobot-oss/datarobot-genai
6
6
  Author: DataRobot, Inc.
@@ -32,7 +32,7 @@ Requires-Dist: aiohttp<4.0.0,>=3.9.0; extra == 'drmcp'
32
32
  Requires-Dist: aiosignal<2.0.0,>=1.3.1; extra == 'drmcp'
33
33
  Requires-Dist: boto3<2.0.0,>=1.34.0; extra == 'drmcp'
34
34
  Requires-Dist: datarobot-asgi-middleware<1.0.0,>=0.2.0; extra == 'drmcp'
35
- Requires-Dist: fastmcp==2.13.0.2; extra == 'drmcp'
35
+ Requires-Dist: fastmcp<3.0.0,>=2.13.0.2; extra == 'drmcp'
36
36
  Requires-Dist: httpx<1.0.0,>=0.28.1; extra == 'drmcp'
37
37
  Requires-Dist: opentelemetry-api<2.0.0,>=1.22.0; extra == 'drmcp'
38
38
  Requires-Dist: opentelemetry-exporter-otlp-proto-http<2.0.0,>=1.22.0; extra == 'drmcp'
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
4
4
 
5
5
  [project]
6
6
  name = "datarobot-genai"
7
- version = "0.1.65"
7
+ version = "0.1.66"
8
8
  description = "Generic helpers for GenAI"
9
9
  readme = "README.md"
10
10
  requires-python = ">=3.10, <3.13"
@@ -84,7 +84,7 @@ drmcp = [
84
84
  "aiohttp>=3.9.0,<4.0.0",
85
85
  "aiohttp-retry>=2.8.3,<3.0.0",
86
86
  "aiosignal>=1.3.1,<2.0.0",
87
- "fastmcp==2.13.0.2",
87
+ "fastmcp>=2.13.0.2,<3.0.0",
88
88
  ]
89
89
 
90
90
  [tool.hatch.build.targets.wheel]
@@ -51,7 +51,7 @@ async def list_tools_by_tags(tags: list[str] | None = None, match_all: bool = Fa
51
51
  -------
52
52
  A formatted string listing tools that match the tag criteria.
53
53
  """
54
- tools = await mcp._list_tools_mcp(tags=tags, match_all=match_all)
54
+ tools = await mcp.list_tools(tags=tags, match_all=match_all)
55
55
 
56
56
  if not tools:
57
57
  if tags:
@@ -95,7 +95,7 @@ async def get_tool_info_by_name(tool_name: str) -> str:
95
95
  -------
96
96
  A formatted string with detailed information about the tool.
97
97
  """
98
- all_tools = await mcp._list_tools_mcp()
98
+ all_tools = await mcp.list_tools()
99
99
 
100
100
  for tool in all_tools:
101
101
  if tool.name == tool_name: