Model config response count support

Author: lgluszek

libs/labelbox/src/labelbox/schema/project.py

@@ -1118,18 +1118,19 @@ def get_label_count(self) -> int:
         res = self.client.execute(query_str, {"projectId": self.uid})
         return res["project"]["labelCount"]
 
-    def add_model_config(self, model_config_id: str) -> str:
+    def add_model_config(self, model_config_id: str, response_count: Optional[int] = None) -> str:
         """Adds a model config to this project.
 
         Args:
             model_config_id (str): ID of a model config to add to this project.
+            response_count (Optional[int]): Number of responses to generate. If not provided, uses the default.
 
         Returns:
             str, ID of the project model config association. This is needed for updating and deleting associations.
         """
 
-        query = """mutation CreateProjectModelConfigPyApi($projectId: ID!, $modelConfigId: ID!)  {
-                    createProjectModelConfig(input: {projectId: $projectId, modelConfigId: $modelConfigId}) {
+        query = """mutation CreateProjectModelConfigPyApi($projectId: ID!, $modelConfigId: ID!, $responseCount: Int)  {
+                    createProjectModelConfig(input: {projectId: $projectId, modelConfigId: $modelConfigId, responseCount: $responseCount}) {
                         projectModelConfigId
                     }
                 }"""
@@ -1138,6 +1139,9 @@ def add_model_config(self, model_config_id: str) -> str:
             "projectId": self.uid,
             "modelConfigId": model_config_id,
         }
+        
+        if response_count is not None:
+            params["responseCount"] = response_count
         try:
             result = self.client.execute(query, params)
         except LabelboxError as e:

requirements-dev.lock

@@ -6,6 +6,8 @@
 #   features: []
 #   all-features: true
 #   with-sources: false
+#   generate-hashes: false
+#   universal: false
 
 -e file:libs/labelbox
 -e file:libs/lbox-clients

requirements.lock

@@ -6,6 +6,8 @@
 #   features: []
 #   all-features: true
 #   with-sources: false
+#   generate-hashes: false
+#   universal: false
 
 -e file:libs/labelbox
 -e file:libs/lbox-clients