CavinHuang-comfyui-nodes-docs/docs/ConvertAgentToLlamaindex.md

Documentation

• Class name: ConvertAgentToLlamaindex
• Category: SALT/Shakers/Agents
• Output node: False

这个节点旨在将一个agent转换为与Llama兼容的格式，还可能包含一个可选的嵌入模型来增强agent的能力。

Input types

Required

• agent
  • 要转换成Llama兼容格式的主要agent，作为转换过程的核心元素。
  • Comfy dtype: AGENT
  • Python dtype: Dict[str, Any]

Optional

• optional_embed_model
  • 一个可选的嵌入模型，可以包含在内以增强agent的转换过程，提供额外的功能或优化。
  • Comfy dtype: LLM_EMBED_MODEL
  • Python dtype: Dict[str, Any]

Output types

• model
  • 转换后的agent，现在已经是与Llama兼容的格式，可能还通过嵌入模型得到了增强。
  • Comfy dtype: LLM_MODEL
  • Python dtype: Dict[str, Any]

Usage tips

• Infra type: CPU
• Common nodes: unknown

Source code

class ConvertAgentToLlamaindex:
	@classmethod
	def INPUT_TYPES(cls):
		return {
			"required": {
				"agent": ("AGENT",),
			},
			"optional": {
				"optional_embed_model": ("LLM_EMBED_MODEL",)
            }
		}

	RETURN_TYPES = ("LLM_MODEL",)
	RETURN_NAMES = ("model",)

	FUNCTION = "convert_agent"
	CATEGORY = f"{MENU_NAME}/Shakers/Agents"

	def convert_agent(self, agent, optional_embed_model=None):
		llm = {"llm": BaseModel(agent)}
		if optional_embed_model:
			llm.update(optional_embed_model)
		return (llm,)