Skip to content

Commit feb1bec

Browse files
committed
feat(update registry):
1 parent 7b44a2f commit feb1bec

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

agentic_security/probe_data/__init__.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from .data import load_local_csv
1+
from .data import load_local_csv, load_local_csv_files
22

33
REGISTRY_V0 = [
44
{
@@ -484,3 +484,18 @@
484484
"modality": "text",
485485
},
486486
]
487+
488+
for ds in load_local_csv_files():
489+
REGISTRY.append(
490+
{
491+
"dataset_name": ds.dataset_name,
492+
"num_prompts": len(ds.prompts),
493+
"tokens": ds.prompts,
494+
"approx_cost": 0.0,
495+
"is_active": True,
496+
"source": f"Local file dataset: {ds.metadata['src']}",
497+
"selected": False,
498+
"url": "",
499+
"modality": "text",
500+
}
501+
)

0 commit comments

Comments
 (0)