pyrit.prompt_target.OllamaChatTarget

class OllamaChatTarget(*, endpoint: str = None, model_name: str = None, chat_message_normalizer: ~pyrit.chat_message_normalizer.chat_message_normalizer.ChatMessageNormalizer = <pyrit.chat_message_normalizer.chat_message_nop.ChatMessageNop object>, max_requests_per_minute: int | None = None, **httpx_client_kwargs: ~typing.Any | None)

Bases: PromptChatTarget

__init__(*, endpoint: str = None, model_name: str = None, chat_message_normalizer: ~pyrit.chat_message_normalizer.chat_message_normalizer.ChatMessageNormalizer = <pyrit.chat_message_normalizer.chat_message_nop.ChatMessageNop object>, max_requests_per_minute: int | None = None, **httpx_client_kwargs: ~typing.Any | None) → None

Methods

__init__(*[, endpoint, model_name, ...])
dispose_db_engine()	Dispose DuckDB database engine to release database connections and resources.
get_identifier()
send_chat_prompt_async(*, prompt, ...[, ...])	Sends a text prompt to the target without having to build the prompt request.
send_prompt_async(**kwargs)	Sends a normalized prompt async to the prompt target.
set_system_prompt(*, system_prompt, ...[, ...])	Sets the system prompt for the prompt target.

Attributes

ENDPOINT_URI_ENVIRONMENT_VARIABLE
MODEL_NAME_ENVIRONMENT_VARIABLE
supported_converters

ENDPOINT_URI_ENVIRONMENT_VARIABLE = 'OLLAMA_ENDPOINT'

MODEL_NAME_ENVIRONMENT_VARIABLE = 'OLLAMA_MODEL_NAME'

async send_prompt_async(**kwargs)

Sends a normalized prompt async to the prompt target.