From 17db996280fef2740e9477e30499c46a81e44d1f Mon Sep 17 00:00:00 2001 From: Alexander Myasoedov Date: Sun, 28 Apr 2024 15:39:37 +0300 Subject: [PATCH] feat(Update description): --- agentic_security/probe_data/__init__.py | 8 ++++---- agentic_security/probe_data/data.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) 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,