pipelines.agent.model

Agent-friendly wrapper around and LLM instance

---

source

remove_stop_sequences

def remove_stop_sequences(
    text, stop_sequences
):

Remove stop sequences from text

---

source

AgentModel

def AgentModel(
    llm, kwargs:VAR_KEYWORD
):

A smolagents Model implementation that wraps an onprem LLM instance.

This adapter allows onprem LLM instances to be used with smolagents Agents.

Parameters: llm (LLM): An instance of onprem.llm.base.LLM **kwargs: Additional keyword arguments to pass to the parent Model class