kiln-ai 0.5.5__py3-none-any.whl → 0.6.1__py3-none-any.whl

kiln_ai/datamodel/test_models.py

@@ -3,7 +3,14 @@ import json
 import pytest
 from pydantic import ValidationError
 
-from kiln_ai.datamodel import Priority, Project, Task, TaskDeterminism
+from kiln_ai.datamodel import (
+    DataSource,
+    DataSourceType,
+    Project,
+    Task,
+    TaskOutput,
+    TaskRun,
+)
 from kiln_ai.datamodel.test_json_schema import json_joke_schema
 
 
@@ -62,9 +69,7 @@ def test_save_to_file(test_project_file):
 
 def test_task_defaults():
     task = Task(name="Test Task", instruction="Test Instruction")
-    assert task.description == ""
-    assert task.priority == Priority.p2
-    assert task.determinism == TaskDeterminism.flexible
+    assert task.description is None
 
 
 def test_task_serialization(test_project_file):
@@ -73,9 +78,8 @@ def test_task_serialization(test_project_file):
         parent=project,
         name="Test Task",
         description="Test Description",
-        determinism=TaskDeterminism.semantic_match,
-        priority=Priority.p0,
         instruction="Test Base Task Instruction",
+        thinking_instruction="Test Thinking Instruction",
     )
 
     task.save_to_file()
@@ -84,8 +88,7 @@ def test_task_serialization(test_project_file):
     assert parsed_task.name == "Test Task"
     assert parsed_task.description == "Test Description"
     assert parsed_task.instruction == "Test Base Task Instruction"
-    assert parsed_task.determinism == TaskDeterminism.semantic_match
-    assert parsed_task.priority == Priority.p0
+    assert parsed_task.thinking_instruction == "Test Thinking Instruction"
 
 
 def test_save_to_file_without_path():
@@ -189,3 +192,36 @@ def test_task_output_schema(tmp_path):
         task = Task(name="Test Task", output_json_schema="{'asdf':{}}", path=path)
     with pytest.raises(ValidationError):
         task = Task(name="Test Task", input_json_schema="{asdf", path=path)
+
+
+def test_task_run_intermediate_outputs():
+    # Create a basic task output
+    output = TaskOutput(
+        output="test output",
+        source=DataSource(
+            type=DataSourceType.synthetic,
+            properties={
+                "model_name": "test-model",
+                "model_provider": "test-provider",
+                "adapter_name": "test-adapter",
+            },
+        ),
+    )
+
+    # Test valid intermediate outputs
+    task_run = TaskRun(
+        input="test input",
+        input_source=DataSource(
+            type=DataSourceType.human,
+            properties={"created_by": "test-user"},
+        ),
+        output=output,
+        intermediate_outputs={
+            "cot": "chain of thought output",
+            "draft": "draft output",
+        },
+    )
+    assert task_run.intermediate_outputs == {
+        "cot": "chain of thought output",
+        "draft": "draft output",
+    }

kiln_ai/utils/config.py

@@ -97,7 +97,8 @@ class Config:
 
         # Check if the value is in settings
         if name in self._settings:
-            return property_config.type(self._settings[name])
+            value = self._settings[name]
+            return value if value is None else property_config.type(value)
 
         # Check environment variable
         if property_config.env_var and property_config.env_var in os.environ:
@@ -110,7 +111,7 @@ class Config:
         else:
             value = property_config.default
 
-        return property_config.type(value)
+        return None if value is None else property_config.type(value)
 
     def __setattr__(self, name, value):
         if name in ("_properties", "_settings"):

kiln_ai/utils/test_config.py

@@ -26,6 +26,7 @@ def config_with_yaml(mock_yaml_file):
                     str, default="default_value", env_var="EXAMPLE_PROPERTY"
                 ),
                 "int_property": ConfigProperty(int, default=0),
+                "empty_property": ConfigProperty(str),
             }
         )
 
@@ -67,6 +68,12 @@ def test_nonexistent_property(config_with_yaml):
         config.nonexistent_property
 
 
+def test_nonexistent_property_get_value(config_with_yaml):
+    config = config_with_yaml
+    assert config.get_value("nonexistent_property") is None
+    assert config.get_value("empty_property") is None
+
+
 def test_property_type_conversion(config_with_yaml):
     config = config_with_yaml
     config = Config(properties={"int_property": ConfigProperty(int, default="42")})

{kiln_ai-0.5.5.dist-info → kiln_ai-0.6.1.dist-info}/METADATA

@@ -1,13 +1,12 @@
 Metadata-Version: 2.3
 Name: kiln-ai
-Version: 0.5.5
+Version: 0.6.1
 Summary: Kiln AI
 Project-URL: Homepage, https://getkiln.ai
 Project-URL: Repository, https://github.com/Kiln-AI/kiln
 Project-URL: Documentation, https://kiln-ai.github.io/Kiln/kiln_core_docs/kiln_ai.html
 Project-URL: Issues, https://github.com/Kiln-AI/kiln/issues
 Author-email: "Steve Cosman, Chesterfield Laboratories Inc" <scosman@users.noreply.github.com>
-License-File: LICENSE.txt
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3.10

kiln_ai-0.6.1.dist-info/RECORD

@@ -0,0 +1,37 @@
+kiln_ai/__init__.py,sha256=Sc4z8LRVFMwJUoc_DPVUriSXTZ6PO9MaJ80PhRbKyB8,34
+kiln_ai/adapters/__init__.py,sha256=o4zubg6oHEVS15rhRHiWPmrxdgr_IXwQZ4Cr_XYT7zo,938
+kiln_ai/adapters/base_adapter.py,sha256=E_RfXxzEhW-i066xOhZdPuTM7OPKQv70hDpfMsxfYEs,6145
+kiln_ai/adapters/langchain_adapters.py,sha256=iN26w-BTM14AtnaTU_maq57j7Lm2uKfrOlVronyHsno,6712
+kiln_ai/adapters/ml_model_list.py,sha256=maEawe9BQ3T-2qSssbQ6tkB7mUzi0YhHP5R7uwmF6iA,26084
+kiln_ai/adapters/prompt_builders.py,sha256=Mdu-f1mC9hWIDwoF7Qwd9F99GDx6oNGvtEZN-SrOsNM,10325
+kiln_ai/adapters/test_langchain_adapter.py,sha256=sQU_ZDF2p022wU9eYg4QFKQva8yQvw8QRb6DoCvfywk,4703
+kiln_ai/adapters/test_ml_model_list.py,sha256=3O0T86yLOXDF1XTc9BxdK-pYPRN0eRV_rgB_AfFCwbM,6375
+kiln_ai/adapters/test_prompt_adaptors.py,sha256=T8ecAzQ9HMcEkxJIAKvF-Om6d0eIHT_FTIgZAJ_r9x0,7414
+kiln_ai/adapters/test_prompt_builders.py,sha256=sU0bSBZa9Y4Q-mmkDf3HbQ0MNSWk5o9bC9sNgtnBokk,14598
+kiln_ai/adapters/test_saving_adapter_results.py,sha256=SYYh2xY1zmeKhFHfWAuEY4pEiLd8SitSV5ewGOTmaOI,6447
+kiln_ai/adapters/test_structured_output.py,sha256=o3w7r7pSjFxVUh4AIRflkbLRByLAG7A20M6nvdTwHkI,10035
+kiln_ai/adapters/data_gen/__init__.py,sha256=QTZWaf7kq5BorhPvexJfwDEKmjRmIbhwW9ei8LW2SIs,276
+kiln_ai/adapters/data_gen/data_gen_prompts.py,sha256=kudjHnAz7L3q0k_NLyTlaIV7M0uRFrxXNcfcnjOE2uc,5810
+kiln_ai/adapters/data_gen/data_gen_task.py,sha256=vwjC47YDrsl4GtBJpK6FWh07TGd8CalhZOX4p4YBX8w,5904
+kiln_ai/adapters/data_gen/test_data_gen_task.py,sha256=vym6qOTLeFrG54wh3qLFPH0qZtop2ZaGX8AbzlUtlp8,9429
+kiln_ai/adapters/repair/__init__.py,sha256=dOO9MEpEhjiwzDVFg3MNfA2bKMPlax9iekDatpTkX8E,217
+kiln_ai/adapters/repair/repair_task.py,sha256=VXvX1l9AYDE_GV0i3S_vPThltJoQlCFVCCHV9m-QA7k,3297
+kiln_ai/adapters/repair/test_repair_task.py,sha256=kYCEIgRj_-b0AUqXokzheg7RzloWsRHWbUU_J6EQWDM,7938
+kiln_ai/datamodel/__init__.py,sha256=bWuJnJl6whe-kTJEp9yQorzCbweEOjkHnIQHYBEARpY,15406
+kiln_ai/datamodel/basemodel.py,sha256=lNSxZQO2isNP1JhDpc-bK6VqKWz1oY_Cc9RUT3T-Pz8,17535
+kiln_ai/datamodel/json_schema.py,sha256=l4BIq1ItLHgcSHqsqDOchegLLHY48U4yR0SP2aMb4i0,2449
+kiln_ai/datamodel/test_basemodel.py,sha256=679aT4UplglHtTVYmHGgT4v5uyJd7J2w8dBicZlLomk,9439
+kiln_ai/datamodel/test_datasource.py,sha256=GAiZz31qezVVPwFqnt8wHMu15WvtlV89jw8C1Ue6YNI,3165
+kiln_ai/datamodel/test_example_models.py,sha256=PxDtC8H3EfZem4kAQ-waFlRMlqnV4YlLEFKzz0Pq4tM,20013
+kiln_ai/datamodel/test_json_schema.py,sha256=vdLnTQxxrcmuSrf6iOmkrmpfh7JnxqIw4B4dbDAAcZ4,3199
+kiln_ai/datamodel/test_models.py,sha256=4VwggmkHEpTi8yHN48guVwtRUuBmDo7D_OQpA6DDXMo,6731
+kiln_ai/datamodel/test_nested_save.py,sha256=xciCddqvPyKyoyjC5Lx_3Kh1t4LJv1xYRAPazR3SRcs,5588
+kiln_ai/datamodel/test_output_rating.py,sha256=iw7fVUAPORA-0-VFiikZV3NDycGFaFMHSX1a38t_aQA,2647
+kiln_ai/utils/__init__.py,sha256=PTD0MwBCKAMIOGsTAwsFaJOusTJJoRFTfOGqRvCaU-E,142
+kiln_ai/utils/config.py,sha256=hjfshIWKLNMvNfpmgLJfermC3G6bgEgCXOSFogHG9FQ,5505
+kiln_ai/utils/formatting.py,sha256=VtB9oag0lOGv17dwT7OPX_3HzBfaU9GsLH-iLete0yM,97
+kiln_ai/utils/test_config.py,sha256=pTYItz5WD15rTRdxKE7vszXF_mb-dik2qrFWzkVemEY,7671
+kiln_ai-0.6.1.dist-info/METADATA,sha256=kBG2FmTo2mz0myHGIBimYMuzv5jOYUD8qXCgT2J-zkI,2979
+kiln_ai-0.6.1.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+kiln_ai-0.6.1.dist-info/licenses/LICENSE.txt,sha256=_NA5pnTYgRRr4qH6lE3X-TuZJ8iRcMUi5ASoGr-lEx8,1209
+kiln_ai-0.6.1.dist-info/RECORD,,

{kiln_ai-0.5.5.dist-info → kiln_ai-0.6.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.25.0
+Generator: hatchling 1.26.3
 Root-Is-Purelib: true
 Tag: py3-none-any

kiln_ai-0.5.5.dist-info/RECORD

@@ -1,33 +0,0 @@
-kiln_ai/__init__.py,sha256=Sc4z8LRVFMwJUoc_DPVUriSXTZ6PO9MaJ80PhRbKyB8,34
-kiln_ai/adapters/__init__.py,sha256=3NC1lE_Sg1bF4IsKCoUgje2GL0IwTd1atw1BcDLI8IA,883
-kiln_ai/adapters/base_adapter.py,sha256=xXCISAJHaPCYHad28CS0wZEUlx711FZ_6AwW4rJx4jk,6688
-kiln_ai/adapters/langchain_adapters.py,sha256=WNxhuTdjGCsCyqmXJNLe7HJ-MzJ08yagGV-eAHPZF-E,5411
-kiln_ai/adapters/ml_model_list.py,sha256=ueh2jUqCmgGg-jMv0exn5siOU_6p0rGeJs3jy8ZWvuE,23821
-kiln_ai/adapters/prompt_builders.py,sha256=nfZnEr1E30ZweQhEzIP21rNrL2Or1ILajyX8gU3B7w0,7796
-kiln_ai/adapters/test_langchain_adapter.py,sha256=_xHpVAkkoGh0PRO3BFFqvVj95SVtYZPOdFbYGYfzvQ0,1876
-kiln_ai/adapters/test_ml_model_list.py,sha256=XHbwEFFb7WvZ6UkArqIiQ_yhS_urezHtgvJOSnaricY,4660
-kiln_ai/adapters/test_prompt_adaptors.py,sha256=W3TeacWs5iPA3BE1OJ6VkIftrHWzXd3edBoUgFaQAek,6389
-kiln_ai/adapters/test_prompt_builders.py,sha256=WmTR59tnKnKQ5gnX1X9EqvEUdQr0PQ8OvadYtRQR5sQ,11483
-kiln_ai/adapters/test_saving_adapter_results.py,sha256=tQvpLawo8mR2scPwmRCIz9Sp0ZkerS3kVJKBzlcjwRE,6041
-kiln_ai/adapters/test_structured_output.py,sha256=Z9A2R-TC-2atsdr8sGVGDlJhfa7uytW8Xi8PKBdEEAw,9033
-kiln_ai/adapters/repair/__init__.py,sha256=dOO9MEpEhjiwzDVFg3MNfA2bKMPlax9iekDatpTkX8E,217
-kiln_ai/adapters/repair/repair_task.py,sha256=VXvX1l9AYDE_GV0i3S_vPThltJoQlCFVCCHV9m-QA7k,3297
-kiln_ai/adapters/repair/test_repair_task.py,sha256=12PHb4SgBvVdLUzjZz31M8OTa8D8QjHD0Du4s7ij-i8,7819
-kiln_ai/datamodel/__init__.py,sha256=zkqabC2tdK3ia7qp-pur5lo3LFRYeszsA6SdjCdxerE,14761
-kiln_ai/datamodel/basemodel.py,sha256=lNSxZQO2isNP1JhDpc-bK6VqKWz1oY_Cc9RUT3T-Pz8,17535
-kiln_ai/datamodel/json_schema.py,sha256=mrb_oXf6gT-ahjBlzfIw2lT7iTmmowmMYxHnmVqPTgE,2334
-kiln_ai/datamodel/test_basemodel.py,sha256=679aT4UplglHtTVYmHGgT4v5uyJd7J2w8dBicZlLomk,9439
-kiln_ai/datamodel/test_datasource.py,sha256=GAiZz31qezVVPwFqnt8wHMu15WvtlV89jw8C1Ue6YNI,3165
-kiln_ai/datamodel/test_example_models.py,sha256=PxDtC8H3EfZem4kAQ-waFlRMlqnV4YlLEFKzz0Pq4tM,20013
-kiln_ai/datamodel/test_json_schema.py,sha256=vdLnTQxxrcmuSrf6iOmkrmpfh7JnxqIw4B4dbDAAcZ4,3199
-kiln_ai/datamodel/test_models.py,sha256=FS0G4HRrE9SqX98ZO4SnEwire4AcXw5V96JIPu_Wi9Q,5930
-kiln_ai/datamodel/test_nested_save.py,sha256=xciCddqvPyKyoyjC5Lx_3Kh1t4LJv1xYRAPazR3SRcs,5588
-kiln_ai/datamodel/test_output_rating.py,sha256=iw7fVUAPORA-0-VFiikZV3NDycGFaFMHSX1a38t_aQA,2647
-kiln_ai/utils/__init__.py,sha256=PTD0MwBCKAMIOGsTAwsFaJOusTJJoRFTfOGqRvCaU-E,142
-kiln_ai/utils/config.py,sha256=jXUB8lwFkxLNEaizwIsoeFLg1BwjWr39-5KdEGF37Bg,5424
-kiln_ai/utils/formatting.py,sha256=VtB9oag0lOGv17dwT7OPX_3HzBfaU9GsLH-iLete0yM,97
-kiln_ai/utils/test_config.py,sha256=lbN0NhgKPEZ0idaS-zTn6mWsSAV6omo32JcIy05h2-M,7411
-kiln_ai-0.5.5.dist-info/METADATA,sha256=rD2UKYBIVHUrfsPP7-BhaUXGdLXVkcJDIUs8i75GSX8,3005
-kiln_ai-0.5.5.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-kiln_ai-0.5.5.dist-info/licenses/LICENSE.txt,sha256=_NA5pnTYgRRr4qH6lE3X-TuZJ8iRcMUi5ASoGr-lEx8,1209
-kiln_ai-0.5.5.dist-info/RECORD,,