`agents.components.llm`#

Module Contents#

Classes#

LLM

This component utilizes large language models (e.g Qwen) that can be used to process text data.

API#

class agents.components.llm.LLM(*, inputs: List[Union[agents.ros.Topic, agents.ros.FixedInput]], outputs: List[agents.ros.Topic], model_client: Optional[agents.clients.model_base.ModelClient] = None, config: Optional[agents.config.LLMConfig] = None, db_client: Optional[agents.clients.db_base.DBClient] = None, trigger: Union[agents.ros.Topic, List[agents.ros.Topic], float, agents.ros.Event] = 1.0, component_name: str, **kwargs)#

Bases: agents.components.model_component.ModelComponent

This component utilizes large language models (e.g Qwen) that can be used to process text data.

Parameters:

inputs (list[Topic | FixedInput]) – The input topics or fixed inputs for the LLM component. This should be a list of Topic objects or FixedInput instances.
outputs (list[Topic]) – The output topics for the LLM component. This should be a list of Topic objects. String type is handled automatically.
model_client (Optional[ModelClient]) – The model client for the LLM component. This should be an instance of ModelClient. Optional if enable_local_model is set to True in the config.
config (LLMConfig) – The configuration for the LLM component. This should be an instance of LLMConfig. If not provided, defaults to LLMConfig().
db_client (Optional[DBClient]) – An optional database client for the LLM component. If provided, this should be an instance of DBClient. Otherwise, it defaults to None.
trigger (Union[Topic, list[Topic], float]) – The trigger value or topic for the LLM component. This can be a single Topic object, a list of Topic objects, or a float value for a timed component. Defaults to 1.
component_name (str) – The name of the LLM component. This should be a string.
kwargs – Additional keyword arguments for the LLM.

Example usage:

text0 = Topic(name="text0", msg_type="String")
text1 = Topic(name="text1", msg_type="String")
config = LLMConfig()
model = OllamaModel(name='llama', checkpoint='llama3.2:3b')
model_client = ModelClient(model=model)
llm_component = LLM(inputs=[text0],
                    outputs=[text1],
                    model_client=model_client,
                    config=config,
                    component_name='llama_component')

Example usage with local model:

text0 = Topic(name="text0", msg_type="String")
text1 = Topic(name="text1", msg_type="String")
config = LLMConfig(enable_local_model=True)
llm_component = LLM(inputs=[text0],
                    outputs=[text1],
                    config=config,
                    component_name='local_llm')

custom_on_configure()#: Create model client if provided and initialize model.

custom_on_deactivate()#: Destroy model client if it exists

create_all_service_clients()#: Override create all clients in the LLM

add_documents(ids: List[str], metadatas: List[Dict], documents: List[str]) → None#

Add documents to vector DB for Retrieval Augmented Generation (RAG).

Important

Documents can be provided after parsing them using a document parser. Checkout various document parsers, available in packages like langchain_community

Parameters:

ids (list[str]) – List of unique string ids for each document
metadatas (list[dict]) – List of metadata dicts for each document
documents (list[str]) – List of documents which are to be store in the vector DB

Return type:

None

reset_history() → agents.ros.ActionReturnType#

Forget the conversation, keeping the system prompt. The same as the reset phrase, for events, recipes and the planner.

Returns:: Whether the history was reset, and a message
Return type:: ActionReturnType

get_history() → agents.ros.ActionReturnType#

The conversation as kept, as text.

Returns:: Whether there is a conversation, and its text
Return type:: ActionReturnType

set_topic_prompt(input_topic: agents.ros.Topic, template: Union[str, pathlib.Path]) → None#

Set prompt template on any input topic of type string.

Parameters:

input_topic (Topic) – Name of the input topic on which the prompt template is to be applied
template (Union[str, Path]) – Template in the form of a valid jinja2 string or a path to a file containing the jinja2 string.

Return type:

None

Example usage:

llm_component = LLM(inputs=[text0],
                    outputs=[text1],
                    model_client=model_client,
                    config=config,
                    component_name='llama_component')
llm_component.set_topic_prompt(text0, template="Please answer the following: {{ text0 }}")

set_component_prompt(template: Union[str, pathlib.Path]) → None#

Set component level prompt template which can use multiple input topics.

Parameters:: template (Union[str, Path]) – Template in the form of a valid jinja2 string or a path to a file containing the jinja2 string.
Return type:: None

Example usage:

llm_component = LLM(inputs=[text0],
                    outputs=[text1],
                    model_client=model_client,
                    config=config,
                    component_name='llama_component')
llm_component.set_component_prompt(template="You can see the following items: {{ detections }}. Please answer the following: {{ text0 }}")

set_system_prompt(prompt: str) → None#

Set system prompt for the model, which defines the models ‘personality’.

Parameters:: prompt – string or a path to a file containing the string.
Return type:: None

Example usage:

llm_component = LLM(inputs=[text0],
                    outputs=[text1],
                    model_client=model_client,
                    config=config,
                    component_name='llama_component')
llm_component.set_system_prompt(prompt="You are an amazing and funny robot. You answer all questions with short and concise answers.")

register_tool(tool: Callable, tool_description: Dict, send_tool_response_to_model: bool = False) → None#

Register a tool with the component which can be called by the model. If the send_tool_response_to_model flag is set to True than the output of the tool is sent back to the model and final output of the model is sent to component publishers (i.e. the model “uses” the tool to give a more accurate response.). If the flag is set to False than the output of the tool is sent to publishers of the component.

Parameters:

tool (Callable) – An arbitrary function that needs to be called. The model response will describe a call to this function.
tool_description (dict) – A dictionary describing the function. This dictionary needs to be made in the format shown here. Also see usage example.
send_tool_response_to_model – Whether the model should be called with the tool response. If set to false the tool response will be sent to component publishers. If set to true, the response will be sent back to the model and the final response from the model will be sent to the publishers. Default is False.
send_tool_response_to_model – bool

Return type:

None

Example usage:

def my_arbitrary_function(first_param: str, second_param: int) -> str:
    return f"{first_param}, {second_param}"

my_func_description = {
          'type': 'function',
          'function': {
            'name': 'my_arbitrary_function',
            'description': 'Description of my arbitrary function',
            'parameters': {
              'type': 'object',
              'properties': {
                'first_param': {
                  'type': 'string',
                  'description': 'Description of the first param',
                },
                'second_param': {
                  'type': 'int',
                  'description': 'Description of the second param',
                },
              },
              'required': ['first_param', 'second_param'],
            },
          },
        }

my_component.register_tool(tool=my_arbitrary_function, tool_description=my_func_description, send_tool_response_to_model=False)

property additional_model_clients: Optional[Dict[str, agents.clients.model_base.ModelClient]]#

Get the dictionary of additional model clients registered to this component.

Returns:: A dictionary mapping client names (str) to ModelClient instances, or None if not set.
Return type:: Optional[Dict[str, ModelClient]]

fallback_to_local() → agents.ros.ActionReturnType#

Switch from remote model_client to the built-in local model at runtime.

The local model is deployed on first call (lazy initialization) to avoid consuming GPU memory until actually needed. If enable_local_model is not already set in config, it is enabled automatically.

This is commonly used as a target for Actions in the Event system.

Returns:: Whether the switch happened, with a confirmation message or why the local model could not be deployed.
Return type:: ActionReturnType
Example:

    from agents.ros import Action

    # Define an action to switch to the 'local model' available in each component
    switch_to_local = Action(
        method=brain.fallback_to_local,
    )

    # Trigger this action if the component fails (e.g. internet outage)
    brain.on_component_fail(action=switch_to_local, max_retries=3)

change_model_client(model_client_name: str) → agents.ros.ActionReturnType#

Hot-swap the active model client at runtime.

This method replaces the component’s current model_client with one from the registered additional_model_clients. It handles the safe de-initialization of the old client and initialization of the new one.

This is commonly used as a target for Actions in the Event system.

Parameters:: model_client_name (str) – The key corresponding to the desired client in additional_model_clients.
Returns:: Whether the swap happened, with a confirmation message or why not: no additional clients are registered, the requested client name is not found, or its initialization failed.
Return type:: ActionReturnType
Example:

    from agents.ros import Action

    # Define an action to switch to the 'remote_backup' client defined previously
    switch_to_backup = Action(
        method=brain.change_model_client,
        args=("remote_backup",)
    )

    # Trigger this action if the component fails (e.g. server down)
    brain.on_component_fail(action=switch_to_backup, max_retries=3)

inspect_component() → str#: Return component info including additional model clients.

property warmup: bool#: Enable warmup of the model.

destroy_all_service_clients()#: Destroy the clients for other components’ actions with the rest

custom_on_activate()#: Custom configuration for creating triggers.

init_variables()#

Partition trigger inputs out of the callbacks dict, at activation.

This runs AFTER a robot plugin adapted the callbacks, so the objects moved into trig_callbacks are the adapted ones. Idempotent across reactivations.

create_all_subscribers()#: Override to handle trigger topics and fixed inputs. Called by parent BaseComponent

activate_all_triggers() → None#: Activates component triggers by attaching execution step to callbacks

destroy_all_subscribers() → None#: Destroys all node subscribers

agents.components.llm#

Module Contents#

Classes#

API#

`agents.components.llm`#