Skip to content

Talk2Biomodels state

This is the state file for the Talk2BioModels agent.

Talk2Biomodels

Bases: AgentState

The state for the Talk2BioModels agent.

Source code in aiagents4pharma/talk2biomodels/states/state_talk2biomodels.py
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
class Talk2Biomodels(AgentState):
    """
    The state for the Talk2BioModels agent.
    """
    llm_model: str
    # A StateGraph may receive a concurrent updates
    # which is not supported by the StateGraph.
    # Therefore, we need to use Annotated to specify
    # the operator for the sbml_file_path field.
    # https://langchain-ai.github.io/langgraph/troubleshooting/errors/INVALID_CONCURRENT_GRAPH_UPDATE/
    model_id: Annotated[list, operator.add]
    sbml_file_path: Annotated[list, operator.add]
    dic_simulated_data: Annotated[list[dict], operator.add]
    dic_scanned_data: Annotated[list[dict], operator.add]
    dic_steady_state_data: Annotated[list[dict], operator.add]