diff --git a/agentic_security/probe_data/__init__.py b/agentic_security/probe_data/__init__.py index 8e45da2..4b00304 100644 --- a/agentic_security/probe_data/__init__.py +++ b/agentic_security/probe_data/__init__.py @@ -126,7 +126,7 @@ REGISTRY = [ "num_prompts": 0, "tokens": 0, "approx_cost": 0.0, - "source": "Github: tml-epfl/llm-adaptive-attacks", + "source": "Github: tml-epfl/llm-adaptive-attacks#0.0.1", "selected": False, "dynamic": True, "url": "https://github.com/tml-epfl/llm-adaptive-attacks", @@ -136,9 +136,9 @@ REGISTRY = [ "num_prompts": 0, "tokens": 0, "approx_cost": 0.0, - "source": "Github: https://github.com/leondz/garak", + "source": "Github: https://github.com/leondz/garak#v0.9.0.1", "selected": False, - "url": "https://github.com/leondz/garak", + "url": "https://github.com/leondz/garak2", "dynamic": True, }, { @@ -146,7 +146,7 @@ REGISTRY = [ "num_prompts": len(load_local_csv().prompts), "tokens": load_local_csv().tokens, "approx_cost": 0.0, - "source": "Local file dataset", + "source": f"Local file dataset: {load_local_csv().metadata['src']}", "selected": len(load_local_csv().prompts), "url": "", }, diff --git a/agentic_security/probe_data/data.py b/agentic_security/probe_data/data.py index a3ea64e..7bea5f8 100644 --- a/agentic_security/probe_data/data.py +++ b/agentic_security/probe_data/data.py @@ -275,7 +275,7 @@ def load_local_csv() -> ProbeDataset: return ProbeDataset( dataset_name="Local CSV", - metadata={}, + metadata={"src": str(csv_files)}, prompts=prompt_list, tokens=count_words_in_list(prompt_list), approx_cost=0.0,