Skip to content

Commit e2ea6db

Browse files
committed
fix(argilla): add **kwargs to tools for pipeline compatibility
- create_user, create_dataset, push_records, annotation_progress now accept **kwargs - This prevents errors when Zem automatically injects fields from previous steps
1 parent 7f268f6 commit e2ea6db

2 files changed

Lines changed: 5 additions & 1 deletion

File tree

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[project]
22
name = "xfmr-zem"
3-
version = "0.3.7"
3+
version = "0.3.8"
44
description = "Zem: Unified Data Pipeline Framework (ZenML + NeMo Curator + DataJuicer) for multi-domain processing"
55
readme = "README.md"
66
requires-python = ">=3.10,<3.13"

src/xfmr_zem/servers/argilla/server.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,7 @@ def create_dataset(
7070
guidelines: str = "",
7171
api_url: Optional[str] = None,
7272
api_key: Optional[str] = None,
73+
**kwargs: Any,
7374
) -> Dict[str, Any]:
7475
"""
7576
Tạo hoặc lấy dataset trên Argilla server.
@@ -135,6 +136,7 @@ def push_records(
135136
batch_size: int = 200,
136137
api_url: Optional[str] = None,
137138
api_key: Optional[str] = None,
139+
**kwargs: Any,
138140
) -> Dict[str, Any]:
139141
"""
140142
Đẩy records từ pipeline vào Argilla dataset.
@@ -454,6 +456,7 @@ def annotation_progress(
454456
workspace: str = "admin",
455457
api_url: Optional[str] = None,
456458
api_key: Optional[str] = None,
459+
**kwargs: Any,
457460
) -> Dict[str, Any]:
458461
"""
459462
Thống kê tiến độ annotation của dataset.
@@ -619,6 +622,7 @@ def create_user(
619622
workspace: Optional[str] = None,
620623
api_url: Optional[str] = None,
621624
api_key: Optional[str] = None,
625+
**kwargs: Any,
622626
) -> Dict[str, Any]:
623627
"""
624628
Tạo user mới trên Argilla server.

0 commit comments

Comments
 (0)