agentchat.contrib.llava_agent

LLaVAAgent

class LLaVAAgent(MultimodalConversableAgent)

__init__

def __init__(name: str,
             system_message: Optional[Tuple[str,
                                            List]] = DEFAULT_LLAVA_SYS_MSG,
             *args,
             **kwargs)

Arguments:

• name str - agent name.
• system_message str - system message for the ChatCompletion inference. Please override this attribute if you want to reprogram the agent.
• **kwargs dict - Please refer to other kwargs in ConversableAgent.

llava_call

def llava_call(prompt: str, llm_config: dict) -> str

Makes a call to the LLaVA service to generate text based on a given prompt