Skip to content

Commit 935a239

Browse files
committed
fix(argilla): add data param to create_dataset and create_user for Zem auto-chain
- create_dataset: missing data: Any = None caused KeyError when Zem injects previous_output from chained step - create_user: same issue, also relaxed username/password to optional str so tool can be called without previous data context
1 parent 910706f commit 935a239

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

src/xfmr_zem/servers/argilla/server.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ def _get_client(api_url: Optional[str] = None, api_key: Optional[str] = None):
6262
# ─────────────────────────────────────────────────────────────────────────────
6363
@server.tool()
6464
def create_dataset(
65+
data: Any = None,
6566
name: str = "zem_dataset",
6667
workspace: str = "admin",
6768
fields: Optional[List[Dict[str, Any]]] = None,
@@ -74,6 +75,7 @@ def create_dataset(
7475
Tạo hoặc lấy dataset trên Argilla server.
7576
7677
Args:
78+
data: Không dùng, cho phép Zem auto-chain từ step trước
7779
name: Tên dataset
7880
workspace: Workspace chứa dataset
7981
fields: List cấu hình fields. Mỗi field là dict:
@@ -608,8 +610,9 @@ def agreement_score(
608610
# ─────────────────────────────────────────────────────────────────────────────
609611
@server.tool()
610612
def create_user(
611-
username: str,
612-
password: str,
613+
data: Any = None,
614+
username: str = "",
615+
password: str = "",
613616
role: str = "annotator",
614617
first_name: str = "",
615618
last_name: str = "",
@@ -621,6 +624,7 @@ def create_user(
621624
Tạo user mới trên Argilla server.
622625
623626
Args:
627+
data: Không dùng, cho phép Zem auto-chain từ step trước
624628
username: Tên đăng nhập (unique)
625629
password: Mật khẩu (min 8 ký tự)
626630
role: "annotator" (mặc định) | "admin"

0 commit comments

Comments
 (0)