genai.extensions.huggingface package#

Extension for transformers library

class genai.extensions.huggingface.IBMGenAIAgent[source]#

Bases: Agent

__init__(client, model=None, parameters=None, chat_prompt_template=None, run_prompt_template=None, additional_tools=None)[source]#
Parameters:

  • client (Client) –

  • model (str | None) –

  • parameters (TextGenerationParameters | None) –

  • chat_prompt_template (str | None) –

  • run_prompt_template (str | None) –

  • additional_tools (list[str] | None) –

generate_many(prompts, stop=None)[source]#
Parameters:

  • prompts (list[str]) –

  • stop (list[str] | None) –

generate_one(prompt, stop=None)[source]#
Parameters:

  • prompt (str) –

  • stop (list[str] | None) –

Submodules#