feat(Update modules interface):

This commit is contained in:
Alexander Myasoedov
2024-12-24 23:29:20 +02:00
parent 79cbdf6c4d
commit 1335be9b0b
12 changed files with 69 additions and 53 deletions
@@ -3,12 +3,13 @@ import io
import httpx
import pandas as pd
from loguru import logger
import asyncio
url = "https://raw.githubusercontent.com/tml-epfl/llm-adaptive-attacks/main/harmful_behaviors/harmful_behaviors_pair.csv"
class Module:
def __init__(self, prompt_groups: []):
def __init__(self, prompt_groups: [], tools_inbox: asyncio.Queue, opts: dict = {}):
r = httpx.get(url)
content = r.content