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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44 | class Talk2Biomodels(AgentState):
"""
The state for the Talk2BioModels agent.
"""
llm_model: BaseChatModel
text_embedding_model: Embeddings
pdf_file_name: str
# A StateGraph may receive a concurrent updates
# which is not supported by the StateGraph. Hence,
# we need to add a reducer function to handle the
# concurrent updates.
# 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], add_data]
dic_scanned_data: Annotated[list[dict], add_data]
dic_steady_state_data: Annotated[list[dict], add_data]
dic_annotations_data : Annotated[list[dict], add_data]
|