kiln-ai 0.21.0__py3-none-any.whl → 0.22.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of kiln-ai might be problematic. Click here for more details.

Files changed (53) hide show
  1. kiln_ai/adapters/extractors/litellm_extractor.py +52 -32
  2. kiln_ai/adapters/extractors/test_litellm_extractor.py +169 -71
  3. kiln_ai/adapters/ml_embedding_model_list.py +330 -28
  4. kiln_ai/adapters/ml_model_list.py +503 -23
  5. kiln_ai/adapters/model_adapters/litellm_adapter.py +39 -8
  6. kiln_ai/adapters/model_adapters/test_litellm_adapter.py +78 -0
  7. kiln_ai/adapters/model_adapters/test_litellm_adapter_tools.py +119 -5
  8. kiln_ai/adapters/model_adapters/test_saving_adapter_results.py +9 -3
  9. kiln_ai/adapters/model_adapters/test_structured_output.py +6 -9
  10. kiln_ai/adapters/test_ml_embedding_model_list.py +89 -279
  11. kiln_ai/adapters/test_ml_model_list.py +0 -10
  12. kiln_ai/adapters/vector_store/lancedb_adapter.py +24 -70
  13. kiln_ai/adapters/vector_store/lancedb_helpers.py +101 -0
  14. kiln_ai/adapters/vector_store/test_lancedb_adapter.py +9 -16
  15. kiln_ai/adapters/vector_store/test_lancedb_helpers.py +142 -0
  16. kiln_ai/adapters/vector_store_loaders/__init__.py +0 -0
  17. kiln_ai/adapters/vector_store_loaders/test_lancedb_loader.py +282 -0
  18. kiln_ai/adapters/vector_store_loaders/test_vector_store_loader.py +544 -0
  19. kiln_ai/adapters/vector_store_loaders/vector_store_loader.py +91 -0
  20. kiln_ai/datamodel/basemodel.py +31 -3
  21. kiln_ai/datamodel/external_tool_server.py +206 -54
  22. kiln_ai/datamodel/extraction.py +14 -0
  23. kiln_ai/datamodel/task.py +5 -0
  24. kiln_ai/datamodel/task_output.py +41 -11
  25. kiln_ai/datamodel/test_attachment.py +3 -3
  26. kiln_ai/datamodel/test_basemodel.py +269 -13
  27. kiln_ai/datamodel/test_datasource.py +50 -0
  28. kiln_ai/datamodel/test_external_tool_server.py +534 -152
  29. kiln_ai/datamodel/test_extraction_model.py +31 -0
  30. kiln_ai/datamodel/test_task.py +35 -1
  31. kiln_ai/datamodel/test_tool_id.py +106 -1
  32. kiln_ai/datamodel/tool_id.py +49 -0
  33. kiln_ai/tools/base_tool.py +30 -6
  34. kiln_ai/tools/built_in_tools/math_tools.py +12 -4
  35. kiln_ai/tools/kiln_task_tool.py +162 -0
  36. kiln_ai/tools/mcp_server_tool.py +7 -5
  37. kiln_ai/tools/mcp_session_manager.py +50 -24
  38. kiln_ai/tools/rag_tools.py +17 -6
  39. kiln_ai/tools/test_kiln_task_tool.py +527 -0
  40. kiln_ai/tools/test_mcp_server_tool.py +4 -15
  41. kiln_ai/tools/test_mcp_session_manager.py +186 -226
  42. kiln_ai/tools/test_rag_tools.py +86 -5
  43. kiln_ai/tools/test_tool_registry.py +199 -5
  44. kiln_ai/tools/tool_registry.py +49 -17
  45. kiln_ai/utils/filesystem.py +4 -4
  46. kiln_ai/utils/open_ai_types.py +19 -2
  47. kiln_ai/utils/pdf_utils.py +21 -0
  48. kiln_ai/utils/test_open_ai_types.py +88 -12
  49. kiln_ai/utils/test_pdf_utils.py +14 -1
  50. {kiln_ai-0.21.0.dist-info → kiln_ai-0.22.1.dist-info}/METADATA +79 -1
  51. {kiln_ai-0.21.0.dist-info → kiln_ai-0.22.1.dist-info}/RECORD +53 -45
  52. {kiln_ai-0.21.0.dist-info → kiln_ai-0.22.1.dist-info}/WHEEL +0 -0
  53. {kiln_ai-0.21.0.dist-info → kiln_ai-0.22.1.dist-info}/licenses/LICENSE.txt +0 -0
kiln_ai/tools/test_mcp_server_tool.py CHANGED
@@ -10,7 +10,10 @@ from mcp.types import (
10
10
  Tool,
11
11
  )
12
12
 
13
- from kiln_ai.datamodel.external_tool_server import ExternalToolServer, ToolServerType
13
+ from kiln_ai.datamodel.external_tool_server import (
14
+ ExternalToolServer,
15
+ ToolServerType,
16
+ )
14
17
  from kiln_ai.datamodel.tool_id import MCP_REMOTE_TOOL_ID_PREFIX
15
18
  from kiln_ai.tools.mcp_server_tool import MCPServerTool
16
19
 
@@ -27,7 +30,6 @@ class TestMCPServerTool:
27
30
  description="Test server",
28
31
  properties={
29
32
  "server_url": "https://example.com",
30
- "headers": {},
31
33
  },
32
34
  )
33
35
 
@@ -60,7 +62,6 @@ class TestMCPServerTool:
60
62
  type=ToolServerType.remote_mcp,
61
63
  properties={
62
64
  "server_url": "https://example.com",
63
- "headers": {},
64
65
  },
65
66
  )
66
67
  tool = MCPServerTool(server, "test_tool")
@@ -90,7 +91,6 @@ class TestMCPServerTool:
90
91
  type=ToolServerType.remote_mcp,
91
92
  properties={
92
93
  "server_url": "https://example.com",
93
- "headers": {},
94
94
  },
95
95
  )
96
96
  tool = MCPServerTool(server, "test_tool")
@@ -116,7 +116,6 @@ class TestMCPServerTool:
116
116
  type=ToolServerType.remote_mcp,
117
117
  properties={
118
118
  "server_url": "https://example.com",
119
- "headers": {},
120
119
  },
121
120
  )
122
121
  tool = MCPServerTool(server, "test_tool")
@@ -143,7 +142,6 @@ class TestMCPServerTool:
143
142
  type=ToolServerType.remote_mcp,
144
143
  properties={
145
144
  "server_url": "https://example.com",
146
- "headers": {},
147
145
  },
148
146
  )
149
147
  tool = MCPServerTool(server, "test_tool")
@@ -170,7 +168,6 @@ class TestMCPServerTool:
170
168
  type=ToolServerType.remote_mcp,
171
169
  properties={
172
170
  "server_url": "https://example.com",
173
- "headers": {},
174
171
  },
175
172
  )
176
173
  tool = MCPServerTool(server, "test_tool")
@@ -196,7 +193,6 @@ class TestMCPServerTool:
196
193
  type=ToolServerType.remote_mcp,
197
194
  properties={
198
195
  "server_url": "https://example.com",
199
- "headers": {},
200
196
  },
201
197
  )
202
198
  tool = MCPServerTool(server, "test_tool")
@@ -231,7 +227,6 @@ class TestMCPServerTool:
231
227
  type=ToolServerType.remote_mcp,
232
228
  properties={
233
229
  "server_url": "https://example.com",
234
- "headers": {},
235
230
  },
236
231
  )
237
232
  tool = MCPServerTool(server, "target_tool")
@@ -258,7 +253,6 @@ class TestMCPServerTool:
258
253
  type=ToolServerType.remote_mcp,
259
254
  properties={
260
255
  "server_url": "https://example.com",
261
- "headers": {},
262
256
  },
263
257
  )
264
258
  tool = MCPServerTool(server, "missing_tool")
@@ -287,7 +281,6 @@ class TestMCPServerTool:
287
281
  type=ToolServerType.remote_mcp,
288
282
  properties={
289
283
  "server_url": "https://example.com",
290
- "headers": {},
291
284
  },
292
285
  )
293
286
  tool = MCPServerTool(server, "test_tool")
@@ -321,7 +314,6 @@ class TestMCPServerTool:
321
314
  type=ToolServerType.remote_mcp,
322
315
  properties={
323
316
  "server_url": "https://example.com",
324
- "headers": {},
325
317
  },
326
318
  )
327
319
  tool = MCPServerTool(server, "test_tool")
@@ -347,7 +339,6 @@ class TestMCPServerTool:
347
339
  type=ToolServerType.remote_mcp,
348
340
  properties={
349
341
  "server_url": "https://example.com",
350
- "headers": {},
351
342
  },
352
343
  )
353
344
  tool = MCPServerTool(server, "test_tool")
@@ -365,7 +356,6 @@ class TestMCPServerTool:
365
356
  type=ToolServerType.remote_mcp,
366
357
  properties={
367
358
  "server_url": "https://example.com",
368
- "headers": {},
369
359
  },
370
360
  )
371
361
  tool = MCPServerTool(server, "test_tool")
@@ -415,7 +405,6 @@ class TestMCPServerToolIntegration:
415
405
  description="Postman Echo MCP Server for testing",
416
406
  properties={
417
407
  "server_url": "https://postman-echo-mcp.fly.dev/",
418
- "headers": {},
419
408
  },
420
409
  )
421
410