Skip to content

Talk2KG State

This is the state file for the Talk2KnowledgeGraphs agent.

Talk2KnowledgeGraphs

Bases: AgentState

The state for the Talk2KnowledgeGraphs agent.

Source code in aiagents4pharma/talk2knowledgegraphs/states/state_talk2knowledgegraphs.py 
27
28
29
30
31
32
33
34
35
36
37
38
class Talk2KnowledgeGraphs(AgentState):
    """
    The state for the Talk2KnowledgeGraphs agent.
    """

    llm_model: BaseChatModel
    embedding_model: Embeddings
    uploaded_files: list
    topk_nodes: int
    topk_edges: int
    dic_source_graph: Annotated[list[dict], add_data]
    dic_extracted_graph: Annotated[list[dict], add_data]