Merge pull request #113 from stainless-sdks/chandrudo/update-kbaas

link fix

Author: chandrudo

examples/knowledge_base_indexing_wait.py

@@ -43,8 +43,6 @@ def main() -> None:
         print("\n✅ Indexing job completed successfully!")
         if completed_job.job:
             print(f"Phase: {completed_job.job.phase}")
-            print(f"Total items indexed: {completed_job.job.total_items_indexed}")
-            print(f"Total items failed: {completed_job.job.total_items_failed}")
             print(f"Total datasources: {completed_job.job.total_datasources}")
             print(f"Completed datasources: {completed_job.job.completed_datasources}")
 

src/gradient/resources/knowledge_bases/indexing_jobs.py

@@ -367,9 +367,7 @@ def wait_for_completion(
                 # Failure states
                 if phase == "BATCH_JOB_PHASE_FAILED":
                     raise IndexingJobError(
-                        f"Indexing job {uuid} failed. "
-                        f"Total items indexed: {response.job.total_items_indexed}, "
-                        f"Total items failed: {response.job.total_items_failed}",
+                        f"Indexing job {uuid} failed. ",
                         uuid=uuid,
                         phase=phase,
                     )
@@ -737,9 +735,7 @@ async def wait_for_completion(
                 # Failure states
                 if phase == "BATCH_JOB_PHASE_FAILED":
                     raise IndexingJobError(
-                        f"Indexing job {uuid} failed. "
-                        f"Total items indexed: {response.job.total_items_indexed}, "
-                        f"Total items failed: {response.job.total_items_failed}",
+                        f"Indexing job {uuid} failed. ",
                         uuid=uuid,
                         phase=phase,
                     )