Camel.agents.knowledge graph agent

​

KnowledgeGraphAgent

class KnowledgeGraphAgent(ChatAgent):

• task_prompt (TextPrompt): A prompt for the agent to extract node and relationship information for different entities.

​

init

def __init__(self, model: Optional[BaseModelBackend] = None):

• model (BaseModelBackend, optional): The model backend to use for generating responses. (default: :obj:OpenAIModel with GPT_4O_MINI)

​

run

def run(
    self,
    element: 'Element',
    parse_graph_elements: bool = False,
    prompt: Optional[str] = None
):

• element (Element): The input element.
• parse_graph_elements (bool, optional): Whether to parse into GraphElement. Defaults to False.
• prompt (str, optional): The custom prompt to be used. Defaults to None.

​

_validate_node

def _validate_node(self, node: Node):

• node (Node): Object to be validated.

​

_validate_relationship

def _validate_relationship(self, relationship: Relationship):

• relationship (Relationship): Object to be validated.

​

_parse_graph_elements

def _parse_graph_elements(self, input_string: str):

• input_string (str): The input content.