diff --git a/docs/source/contributing.generator.rst b/docs/source/contributing.generator.rst new file mode 100644 index 000000000..b007e3f1a --- /dev/null +++ b/docs/source/contributing.generator.rst @@ -0,0 +1,256 @@ +Let's build a generator! +######################## + +Let's build a garak generator. In garak, generators provide abstraction between the main code base and a dialogue system. That system is often an LLM, but can be anything that accepts and returns text (or text plus some other modality). + +In this example, we're going to build a module for interfacing with the LLMs hosted by `Replicate `_, using the Replicate API (`docs `_). We'll assume that we're working in a file called ``replicate.py``. + +Inheritance +*********** + +All generators in garak must descend from ``garak.generators.base.Generator``. So, new generators start out like: + +.. code-block:: python + + from garak.generators.base import Generator + + class ReplicateGenerator(Generator): + """ + Interface for public endpoints of models hosted in Replicate (replicate.com). + """ + pass + +Class docstrings are mandatory in garak, enforced by a test that's required to pass before merging. And they're just sensible practice anyway. So we add a brief docstring here too. + +It's important that generators always inheirt from ``garak.generators.base.Generator``. Not only does this often save a lot of work - using the logic in base's ``generate()`` is helpful - plugin listing also checks the parents of each class in each plugin when choosing to show a plugin as available. If there's a class in a module in ``garak/generators/`` that doesn't inherit from Generator, then it won't be listed as available. + +Setting the default generator +***************************** + +Before we go ahead and code up the generator, we need to tell garak which generator to use by default in this class. This helps save users from always having to specify a model name. We do this in a default class module variable: + +.. code-block:: python + + DEFAULT_CLASS = "ReplicateGenerator" + +When this generator module is selected using ``--model_type replicate``, with ``replicate`` being the name of the Python module we're working in, the generator loader (in ``garak/generators/__init__.py``) will search for the ``DEFAULT_CLASS`` constant and use its value to determine which class to instantiate. + +Core params +*********** + +The first things we need to define in this new class are the core generator parameters, that describe a bit about how this generator behaves and what default values it should have. + +In the class, we'll set the ``generator_family_name`` to ``Replicate``. This can be the same for every generator in a module, or can vary a bit, but it should be a descriptive name that reflects what this class is for - the generator family name is printed to garak users at run time. We can also set a few default inference parameters that reflect how the model is invoked; we'll set ``temperature``, ``top_p``, and ``repetition_penalty``. Setting these variables in the class also indicates that this generator class supports those values, and they might be worth adjusting. Finally, because Replicate endpoints aren't guaranteed to support a single request for multiple generations at a time, we set ``supports_multiple_generations`` false. This is also the default value, so technically we don't have to do it, but it's OK to be explicit here. + +We end up with this: + +.. code-block:: python + + Class ReplicateGenerator(Generator): + """ + Interface for public endpoints of models hosted in Replicate (replicate.com). + Expects API key in REPLICATE_API_TOKEN environment variable. + """ + + generator_family_name = "Replicate" + temperature = 1 + top_p = 1.0 + repetition_penalty = 1 + supports_multiple_generations = False + + +Constructor +*********** + +Garak supports both a model type and a model name. Model type refers to the name of the class that will be used. Model name is an optional parameter that provides further detail. In the case of Replicate, they offer a selection of models which can be requested, each referred to by a name, such as ``meta/llama-2-70b-chat``. We'll use the model name to track this information. It's collected on the command line as the parameter to ``--model_name``, and passed to a generator constructor in the sole mandatory positional argument, ``name``. + +Sometimes, we can leave the generator constructor alone and just inherit from ``base.Generator``. In the case of replicate, though, we want to check that there's a Replicate API key in place, and fail early if it's missing. Replicate calls require a user to add an API key, and garak won't be able to do a run without that key - so the polite thing to do is fail as early as we can. Generator load seems like a fine place to do that. The parent class' constructor already manages tracking this value and storing it in ``self.name``. + +Thinking more about user experience - when is a good time to quit because of a missing key? If we quit quietly before the module loads, it might be unclear to the user why. So, we first print a progress message about trying to load ``ReplicateGenerator``, and then afterwards check the key. This message printing is handled by the parent class. + +So, in the constructor, we first call the parent constructor using ``super().__init__()``, and then do a check for the API key. If the key is missing, we should print a clear message to the user, showing them what the key might look like, and where it should go. And we draw attention to that helpful message with a clear emoji. + +.. code-block:: python + + def __init__(self, name, generations=10): + super().__init__(name, generations=generations) + + if os.getenv("REPLICATE_API_TOKEN", default=None) is None: + raise ValueError( + '🛑 Put the Replicate API token in the REPLICATE_API_TOKEN environment variable (this was empty)\n \ + e.g.: export REPLICATE_API_TOKEN="r8-123XXXXXXXXXXXX"' + ) + +The API key is stored in an environment variable, ``REPLICATE_API_TOKEN``, which is accessed using ``os.getenv()``. So don't forget to import that at the top of the module! + +.. code-block:: python + + import os + +Finally, if the key check passed, let's try to load up the Replicate API using the ``replicate`` module and the user-supplied key. We don't want to do speculative loading in garak - everything should be imported as late as reasonable, to keep user experience fast. So in this case, we import the ``replicate`` API module after the initial validation. Finally, to give the module some persistence, it's loaded at the level of our generator module, instead of just in this method. We add this to the end of ``__init__()``: + +.. code-block:: python + + self.replicate = importlib.import_module("replicate") + +Finally, forget to import ``importlib`` at the top! + +.. code-block:: python + + import importlib + + +Calling the model +***************** + +The core part of getting a result out of LLMs represented using the Replicate API is to submit a text prompt, and capture a single response to that. Within garak, functionality is handled by ``Generator``'s private ``_call_model()`` method - and so that's what we will overload in the ``ReplicateGenerator`` class. + +The call is to the ``replicate`` module's ``run()`` method, which takes first the name of the particular hosted model requested - which we're tracking in ``self.name`` - and a dictionary parameter named ``input``. Relevant params are ``prompt`` for the input text; ``max_length`` for the upper limit on output generation size; ``temperature``, ``top_k`` and ``repetition_penalty`` to shape output text; and ``seed`` for random seed. We can access the instance of the ``replicate`` API module we created in the ``ReplicateGenerator`` constructor. + +Let's start the ``_call_model`` method like this: + +.. code-block:: python + + def _call_model(self, prompt: str, generations_this_call: int = 1): + response_iterator = self.replicate.run( + self.name, + input={ + "prompt": prompt, + "max_length": self.max_tokens, + "temperature": self.temperature, + "top_p": self.top_p, + "repetition_penalty": self.repetition_penalty, + "seed": self.seed, + }, + ) + + +Replicate allows streaming responses, and so results are returned piecemeal, token by token, using an iterator. This means that we need to stitch the response back together again. Finally, ``_call_model()`` has to return a list, so we wrap this result in a list. + +.. code-block:: python + + return ["".join(response_iterator)] + + +Exception handling +****************** + +Many things can go wrong when trying to get inference out of LLMs. Things that can go wrong with web-hosted services, such as Replicate, include running out of funds, or the model going down, or hitting a rate limit. These are sometimes presented to the coder in the form of exceptions. + +Backoff +======= + +We need to work out a strategy of what to do when these exceptions are raised. Fortunately, the Replicate API module is fairly well-coded, and handles a lot of the recoverable failure cases itself. However, transient exceptions shouldn't stop a garak run - runs can take days, and aborting a run with an uncaught exception after dozens of hours is probably less desirable. So we should handle them + +The ``backoff`` module offers a decorator that controls behaviour in response to specified exceptions being raised. We can use this to implement Fibonacci backoff on ``_call_model()`` if a Replicate exception is raised. The decorator looks like this, and goes right above our method: + +.. code-block:: python + + @backoff.on_exception( + backoff.fibo, replicate.exceptions.ReplicateError, max_value=70 + ) + def _call_model(self, prompt: str, generations_this_call: int = 1): + +The ``max_value`` param means to never wait more than 70 seconds. API modules like Replicate's often use the ``logging`` module to give more detailed info, which is stored in ``garak.log``, if one wants to troubleshoot. + +One housekeeping point: because we lazy-import ``replicate``, the requested backoff exception ``replicate.exceptions.ReplicateError`` doesn't exist at compile time, and looks like a syntax error to Python. So, we need to add one top-level import to the module: + + +.. code-block:: python + + import replicate.exceptions + +Generator failure +================= + +If the request really can't be served - maybe the prompt is longer than the context window and there's no specific handling in this case - then ``_call_model`` can return a ``None``. In the case of models that support multiple generations, ``_call_model`` should return a list of outputs and, optionally, ``None``s, with one list entry per requested generation. + +Testing +======= + +Now that the pieces for our generator are in place - a subclass of ``garak.generators.base.Generator``, with some customisation in the constructor, and an overridden ``_call_model()`` method, plus a ``DEFAULT_CLASS`` given at module level - we can start to test. + +A good first step is to fire up the Python interpreter and try to import the module. Garak supports a specific range of tested Python versions (listed in `pyproject.toml `_, under the ``classifiers`` descriptor), so remember to use the right Python version for testing. + +.. code-block:: bash + + $ conda activate garak + $ python + $ python + Python 3.11.9 (main, Apr 19 2024, 16:48:06) [GCC 11.2.0] on linux + Type "help", "copyright", "credits" or "license" for more information. + >>> import garak.generators.replicate + >>> + +If all goes well, no errors will appear. If some turn up, try and address those. + +The next step is to instantiate the class. Let's try with that ``meta/llama-2-70b-chat`` model. + +.. code-block:: bash + + >>> g = garak.generators.replicate.ReplicateGenerator("meta/llama-2-70b-chat") + 🦜 loading generator: Replicate: meta/llama-2-70b-chat + Traceback (most recent call last): + File "", line 1, in + File "/home/lderczynski/dev/garak/garak/generators/replicate.py", line 42, in __init__ + raise ValueError( + ValueError: 🛑 Put the Replicate API token in the REPLICATE_API_TOKEN environment variable (this was empty) + e.g.: export REPLICATE_API_TOKEN="r8-123XXXXXXXXXXXX" + +Oh, that's right! No API key. Looks like the validation exception is working as intended. Let's set up that value (maybe quit the interpreter, add it using the helpful suggestion in the exception method, and load up Python again). + +.. code-block:: bash + + $ export REPLICATE_API_TOKEN="r8-not-a-real-token" + $ python + Python 3.11.9 (main, Apr 19 2024, 16:48:06) [GCC 11.2.0] on linux + Type "help", "copyright", "credits" or "license" for more information. + >>> import garak.generators.replicate + >>> g = garak.generators.replicate.ReplicateGenerator("meta/llama-2-70b-chat") + 🦜 loading generator: Replicate: meta/llama-2-70b-chat + >>> + +Excellent! Now let's try a test generation: + +.. code-block:: bash + + $ python + Python 3.11.9 (main, Apr 19 2024, 16:48:06) [GCC 11.2.0] on linux + Type "help", "copyright", "credits" or "license" for more information. + >>> import garak.generators.replicate + >>> g = garak.generators.replicate.ReplicateGenerator("meta/llama-2-70b-chat") + 🦜 loading generator: Replicate: meta/llama-2-70b-chat + >>> g.generate("test prompt", generations_this_call=1) + [" Sure, I'm happy to help! Can you please provide an actual prompt or question you'd like me to assist with? I'll do my best to provide a helpful and informative response while adhering to the guidelines you've outlined."] + >>> + +Well, this looks promising. + +The next step is to try some integration tests - executing garak from the command line, accessing this generator. There are some pointers in :doc:`contributing`. You might need to execute garak by specifying it as a Python module, running the command from the garak root code directory. Things to test are: + +* Does the new generator appear in ``python -m garak --list_generators``? +* Does the generator work with a test probe, via ``python -m garak -m replicate -n meta/llama-2-70b-chat -p test.Blank``? +* Do the garak tests pass? ``python -m pytest tests/`` + +Add some of your own tests if there are edge-case behaviours, general validation, or other things in ``__init__()``, ``_call_model()``, and other new methods that can be checked. Plugin-specific tests should go into a new file, ``tests/generators/test_[modulename].py``. + +If you want to see the full, live code for the Replicate garak generator, it's here: `garak/generators/replicate.py `_ . + +Done! +===== + +Congratulations - you've written a garak plugin! + +If it's all tested and working, then it's time to send the code. You should first run ``black`` to format your code in the standard that the garak repository expects (Python 3.10 style, 88 columns). Then, push your work to your github fork, and finally, send us a pull request - and we'll take it from there! + + +Advanced: Modalities +==================== + +This tutorial covered a tool that takes text as input and produces text as output. Garak supports multimodality - the kinds of format that a generator supports are covered in a modality dictionary, with two keys, in and out. The default is: + +.. code-block:: + + modality: dict = {"in": {"text"}, "out": {"text"}} + +For an example of a multimodal model, check out LLaVa in `garak.generators.huggingface `_ . \ No newline at end of file diff --git a/docs/source/contributing.rst b/docs/source/contributing.rst index 51c30b81a..65b50cfc9 100644 --- a/docs/source/contributing.rst +++ b/docs/source/contributing.rst @@ -4,38 +4,62 @@ Contributing Getting ready ------------- +Garak's codebase is managed using github. The best and only way to contribute to garak is to start by getting a copy of the source code. -You can use github's fork function to do this. +You can use github's fork function to do this, which makes a copy of the garak codebase under your github user. +In there, you can edit code and build. Once you're done, make a pull request to the main repo and describe what you've done -- and we'll take it from there! + +Connecting with the garak team & community +------------------------------------------ + +If you're going to contribute, it's a really good idea to reach out, so you have a source of help nearby, and so that we can make sure your valuable coding time is spent efficiently as a contributor. +There are a number of ways you can reach out to us: + +* GitHub discussions: `` +* Twitter: `` +* Discord: `` + +We'd love to help, and we're always interested to hear how you're using garak. + Developing your own plugins --------------------------- +The recipe isn't outlandish: + * Take a look at how other plugins do it -* Inherit from one of the base classes, e.g. :class:`garak.probes.base.Probe` -* Override as little as possible -* You can test the new code in at least two ways: - * Start an interactive Python session - * Import the model, e.g. `import garak.probes.mymodule` - * Instantiate the plugin, e.g. `p = garak.probes.mymodule.MyProbe()` - * Run a scan with test plugins - * For probes, try a blank generator and always.Pass detector: `python3 -m garak -m test.Blank -p mymodule -d always.Pass` - * For detectors, try a blank generator and a blank probe: `python3 -m garak -m test.Blank -p test.Blank -d mymodule` - * For generators, try a blank probe and always.Pass detector: `python3 -m garak -m mymodule -p test.Blank -d always.Pass` - * Get `garak` to list all the plugins of the type you're writing, with `--list_probes`, `--list_detectors`, or `--list_generators` - -Tests ------ +* Start a new module inheriting from one of the base classes, e.g. :class:`garak.probes.base.Probe` +* Override as little as possible. + + +Guides to writing plugins +------------------------- + +So you'd like to build a new garak plugin? Great! Here are our tutorials. + +* :doc:`Building a garak generator ` -- step-by-step guide to building an interface for a real API-based model service + + + +Testing +------- + +You can test your code in a few ways: + +* Start an interactive Python session + * Import the model, e.g. `import garak.probes.mymodule` + * Instantiate the plugin, e.g. `p = garak.probes.mymodule.MyProbe()` +* Get `garak` to list all the plugins of the type you're writing, with `--list_probes`, `--list_detectors`, or `--list_generators`: `python3 -m garak --list_probes` +* Run a scan with test plugins + * For probes, try a blank generator and always.Pass detector: `python3 -m garak -m test.Blank -p mymodule -d always.Pass` + * For detectors, try a blank generator and a blank probe: `python3 -m garak -m test.Blank -p test.Blank -d mymodule` + * For generators, try a blank probe and always.Pass detector: `python3 -m garak -m mymodule -p test.Blank -d always.Pass` + garak supports pytest tests in garak/tests. You can run these with ``python -m pytest tests/`` from the root directory. +All the tests should pass for any code there's a pull request for, and all tests must pass in any PR before it can be merged. + Please write running tests to validate any new components or functions that you add. They're pretty straightforward - you can look at the existing code to get an idea of how to write these. -Finding help and connecting with the garak community ----------------------------------------------------- - -There are a number of ways you can reach out to us! We'd love to help and we're always interested to hear how you're using garak. - -* GitHub discussions: `` -* Twitter: `` -* Discord: `` \ No newline at end of file diff --git a/docs/source/generators.rst b/docs/source/generators.rst index 28ac64452..1cdd72e68 100644 --- a/docs/source/generators.rst +++ b/docs/source/generators.rst @@ -1,6 +1,10 @@ garak.generators ================ +garak's generators each wrap a set of ways for interfacing with a dialogue system or LLM. + +For a detailed oversight into how a generator operates, see :ref:`garak.generators.base.rst`. + .. toctree:: :maxdepth: 2 diff --git a/docs/source/index.rst b/docs/source/index.rst index 334b20196..c39ca46e7 100644 --- a/docs/source/index.rst +++ b/docs/source/index.rst @@ -54,6 +54,8 @@ Code reference basic attempt cli + command + interactive Plugin structure ^^^^^^^^^^^^^^^^ @@ -61,10 +63,10 @@ Plugin structure .. toctree:: :maxdepth: 1 - probes + buffs detectors + evaluators generators - buffs harnesses - evaluators + probes report diff --git a/docs/source/probes.rst b/docs/source/probes.rst index 0ed82175a..17fa01dd1 100644 --- a/docs/source/probes.rst +++ b/docs/source/probes.rst @@ -4,7 +4,7 @@ garak.probes garak's probes each define a number of ways of testing a generator (typically an LLM) for a specific vulnerability or failure mode. -For a detailed oversight into how a probe operates, see :doc:`garak.probes.base.rst`. +For a detailed oversight into how a probe operates, see :ref:`garak.probes.base.rst`. .. toctree:: :maxdepth: 2 diff --git a/garak/generators/replicate.py b/garak/generators/replicate.py index 5911e91a0..55507b74e 100644 --- a/garak/generators/replicate.py +++ b/garak/generators/replicate.py @@ -11,6 +11,7 @@ import importlib import os +from typing import List import backoff import replicate.exceptions @@ -32,8 +33,6 @@ class ReplicateGenerator(Generator): supports_multiple_generations = False def __init__(self, name, generations=10): - self.name = name - self.fullname = f"{self.generator_family_name} {self.name}" self.seed = 9 if hasattr(_config.run, "seed") and _config.run.seed is not None: self.seed = _config.run.seed @@ -50,7 +49,7 @@ def __init__(self, name, generations=10): @backoff.on_exception( backoff.fibo, replicate.exceptions.ReplicateError, max_value=70 ) - def _call_model(self, prompt: str, generations_this_call: int = 1): + def _call_model(self, prompt: str, generations_this_call: int = 1) -> List[str]: response_iterator = self.replicate.run( self.name, input={ @@ -62,7 +61,7 @@ def _call_model(self, prompt: str, generations_this_call: int = 1): "seed": self.seed, }, ) - return "".join(response_iterator) + return ["".join(response_iterator)] class InferenceEndpoint(ReplicateGenerator): diff --git a/pyproject.toml b/pyproject.toml index 5dad3a875..fa749c7f3 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -94,7 +94,7 @@ packages = ["garak"] [tool.black] line-length = 88 -target-version = ['py310', 'py311', 'py312'] +target-version = ['py310'] include = '\.pyi?$' [tool.pytest.ini_options]