feat(LAB-3168): remove status from CreateLLMAsset resolver

kili-technology · Nov 5, 2024 · c589aff · c589aff
1 parent 530828a
commit c589aff
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/src/kili/adapters/kili_api_gateway/llm/operations_mixin.py b/src/kili/adapters/kili_api_gateway/llm/operations_mixin.py
@@ -171,12 +171,11 @@ def create_llm_asset(
         self,
         project_id: str,
         author_id: str,
-        status: Optional[str] = None,
         label_type: Optional[str] = None,
     ) -> Dict:
         """Create an LLM asset in a project, with optional status and label_type."""
         where = map_project_where(project_id)
-        data = {"author_id": author_id, "status": status, "label_type": label_type}
+        data = {"author_id": author_id, "label_type": label_type}
         data_mapped = map_create_llm_asset_input(data)
         variables = {"where": where, "data": data_mapped}
         fragment = fragment_builder(["id", "latestLabel.id"])

diff --git a/src/kili/llm/presentation/client/llm.py b/src/kili/llm/presentation/client/llm.py
@@ -370,13 +370,12 @@ def create_conversation(self, project_id: str, prompt: str) -> List[ChatItemDict
         Notes:
             - The first chat item corresponds to the user's prompt.
             - The subsequent chat items are assistant responses generated by the project's models.
-            - An LLM asset and a label are created in the project with status "TODO" and labelType "PREDICTION".
+            - An LLM asset and a label are created in the project with labelType "PREDICTION".
         """
         user_id = self.kili_api_gateway.get_current_user(["id"])["id"]
         llm_asset = self.kili_api_gateway.create_llm_asset(
             project_id=project_id,
             author_id=user_id,
-            status="TODO",
             label_type="PREDICTION",
         )
         asset_id = llm_asset["id"]

diff --git a/tests/unit/llm/test_create_conversation.py b/tests/unit/llm/test_create_conversation.py
@@ -24,7 +24,7 @@ def test_create_conversation(mocker):
 
     kili_api_gateway.get_current_user.assert_called_once_with(["id"])
     kili_api_gateway.create_llm_asset.assert_called_once_with(
-        project_id="project_id", author_id="user_id", status="TODO", label_type="PREDICTION"
+        project_id="project_id", author_id="user_id", label_type="PREDICTION"
     )
     kili_api_gateway.create_chat_item.assert_called_once_with(
         asset_id="asset_id", label_id="label_id", prompt="prompt text"