agent_studio.llm.huggingface

Contents

`agent_studio.llm.huggingface`#

Module Contents#

Classes#

HuggingFaceProvider

Base class for models.

Attributes#

`config`
`logger`

agent_studio.llm.huggingface.config[source]#

agent_studio.llm.huggingface.logger[source]#

class agent_studio.llm.huggingface.HuggingFaceProvider(**kwargs)[source]#

Bases: agent_studio.llm.base_model.BaseModel

Base class for models.

name = 'huggingface'[source]#

compose_messages(intermedia_msg: list[dict[str, Any]]) → Any[source]#

generate_response(messages: list[dict[str, Any]], **kwargs) → tuple[str, dict[str, int]][source]#: Creates a chat completion using the Gemini API.