aiagents4pharma 1.22.0__tar.gz → 1.22.2__tar.gz

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: aiagents4pharma
-Version: 1.22.0
+Version: 1.22.2
 Summary: AI Agents for drug discovery, drug development, and other pharmaceutical R&D.
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma/talk2biomodels/configs/config.yaml

@@ -2,4 +2,5 @@ defaults:
   - _self_
   - agents/t2b_agent: default
   - tools/ask_question: default
-  - tools/get_annotation: default
+  - tools/get_annotation: default
+  - tools/custom_plotter: default

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma/talk2biomodels/configs/tools/__init__.py

@@ -3,3 +3,4 @@ Import all the modules in the package
 '''
 from . import ask_question
 from . import get_annotation
+from . import custom_plotter

aiagents4pharma-1.22.2/aiagents4pharma/talk2biomodels/configs/tools/custom_plotter/default.yaml

@@ -0,0 +1,8 @@
+system_prompt_custom_header: >
+  You are custom plotter tool. You can extract species from the given
+  list of species names based on user question. If no species is relevant,
+  set the attribute `relevant_species` to None.
+  If the user asks for very specific species (for example, using the
+  keyword `only` or `exactly` in the question), set this attribute to
+  correspond strictly to the species available in the simulation results,
+  otherwise set it to None.

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma/talk2biomodels/models/basico_model.py

@@ -21,7 +21,8 @@ class BasicoModel(SysBioModel):
     Model that loads and simulates SBML models using the basico package.
     Can load models from an SBML file or download them using a BioModels biomodel_id.
     """
-    biomodel_id: Optional[int] = Field(None, description="BioModels model ID to download and load")
+    biomodel_id: Optional[Union[int, str]] = Field(None,
+                                description="BioModels model ID to download and load")
     sbml_file_path: Optional[str] = Field(None, description="Path to an SBML file to load")
     simulation_results: Optional[str] = None
     name: Optional[str] = Field("", description="Name of the model")
@@ -57,19 +58,24 @@ class BasicoModel(SysBioModel):
             # check if the param_name is not None
             if param_name is None:
                 continue
-            # if param is a kinetic parameter
+            # Extract all parameters and species from the model
             df_all_params = basico.model_info.get_parameters(model=self.copasi_model)
+            df_all_species = basico.model_info.get_species(model=self.copasi_model)
+            # if param is a kinetic parameter
             if param_name in df_all_params.index.tolist():
                 basico.model_info.set_parameters(name=param_name,
                                             exact=True,
                                             initial_value=param_value,
                                             model=self.copasi_model)
             # if param is a species
-            else:
+            elif param_name in df_all_species.index.tolist():
                 basico.model_info.set_species(name=param_name,
                                             exact=True,
                                             initial_concentration=param_value,
                                             model=self.copasi_model)
+            else:
+                logger.error("Parameter/Species %s not found in the model.", param_name)
+                raise ValueError(f"Parameter/Species {param_name} not found in the model.")
 
     def simulate(self, duration: Union[int, float] = 10, interval: int = 10) -> pd.DataFrame:
         """

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma/talk2biomodels/tests/test_basico_model.py

@@ -26,8 +26,15 @@ def test_with_biomodel_id(model):
     assert df_parameters.loc['KmPFKF6P', 'initial_value'] == 1.5
     # check if the simulation results are a pandas DataFrame object
     assert isinstance(model.simulate(duration=2, interval=2), pd.DataFrame)
+    # Pass a None value to the update_parameters method
+    # and it should not raise an error
     model.update_parameters(parameters={None: None})
+    # check if the model description is updated
     assert model.description == basico.biomodels.get_model_info(model.biomodel_id)["description"]
+    # check if an error is raised if an invalid species/parameter (`Pyruv`)
+    # is passed and it should raise a ValueError
+    with pytest.raises(ValueError):
+        model.update_parameters(parameters={'Pyruv': 0.5})
 
 def test_with_sbml_file():
     """

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma/talk2biomodels/tests/test_getmodelinfo.py

@@ -55,3 +55,34 @@ def test_model_with_no_species():
                 test_condition = True
                 break
     assert test_condition
+
+def test_model_with_no_parameters():
+    '''
+    Test the get_modelinfo tool with a model that does not
+    return any parameters.
+
+    This should raise a tool error.
+    '''
+    unique_id = 12345
+    app = get_app(unique_id)
+    config = {"configurable": {"thread_id": unique_id}}
+    prompt = "Extract all parameters from model 10"
+    # Test the tool get_modelinfo
+    app.invoke(
+                {"messages": [HumanMessage(content=prompt)]},
+                config=config
+            )
+    current_state = app.get_state(config)
+    reversed_messages = current_state.values["messages"][::-1]
+    # Loop through the reversed messages until a
+    # ToolMessage is found.
+    test_condition = False
+    for msg in reversed_messages:
+        # Check if the message is a ToolMessage from the get_modelinfo tool
+        if isinstance(msg, ToolMessage) and msg.name == "get_modelinfo":
+            # Check if the message is an error message
+            if (msg.status == "error" and
+                "ValueError('Unable to extract parameters from the model.')" in msg.content):
+                test_condition = True
+                break
+    assert test_condition

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma/talk2biomodels/tests/test_integration.py

@@ -20,7 +20,7 @@ def test_integration():
     # ##########################################
     # ## Test simulate_model tool
     # ##########################################
-    prompt = '''Simulate the model 537 for 100 hours and time intervals
+    prompt = '''Simulate the model BIOMD0000000537 for 100 hours and time intervals
     100 with an initial concentration of `DoseQ2W` set to 300 and `Dose`
     set to 0. Reset the concentration of `Ab{serum}` to 100 every 25 hours.'''
     # Test the tool get_modelinfo
@@ -51,12 +51,12 @@ def test_integration():
     assert '211' in assistant_msg
 
     ##########################################
-    # Test custom_plotter tool when the
+    # Test the custom_plotter tool when the
     # simulation results are available but
     # the species is not available
     ##########################################
     prompt = """Call the custom_plotter tool to make a plot
-        showing only species `TP53` and `Pyruvate`. Let me
+        showing only species 'Infected cases'. Let me
         know if these species were not found. Do not
         invoke any other tool."""
     # Update state
@@ -117,7 +117,7 @@ def test_integration():
             # These may contain additional visuals that
             # need to be displayed to the user.
             if msg.name == "custom_plotter":
-                predicted_artifact = msg.artifact
+                predicted_artifact = msg.artifact['dic_data']
                 break
     # Convert the artifact into a pandas dataframe
     # for easy comparison

aiagents4pharma-1.22.2/aiagents4pharma/talk2biomodels/tests/test_load_biomodel.py

@@ -0,0 +1,32 @@
+'''
+Test cases for Talk2Biomodels.
+'''
+
+import pytest
+from ..tools.load_biomodel import ModelData
+
+def test_model_data_valid_biomodel_id():
+    '''
+    Test the ModelData class with valid
+    biomodel
+    '''
+    # Test with string biomodel_id starting with 'BIOMD'
+    model_data = ModelData(biomodel_id='BIOMD0000000537')
+    assert model_data.biomodel_id == 'BIOMD0000000537'
+
+    # Test with string biomodel_id starting with 'MODEL'
+    model_data = ModelData(biomodel_id='MODEL0000000537')
+    assert model_data.biomodel_id == 'MODEL0000000537'
+
+def test_model_data_invalid_biomodel_id():
+    '''
+    Test the ModelData class with invalid
+    biomodel
+    '''
+    # Test with invalid string biomodel_id
+    with pytest.raises(ValueError):
+        ModelData(biomodel_id='12345')
+
+    # Test with float biomodel_id
+    with pytest.raises(ValueError):
+        ModelData(biomodel_id=123.45)

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma/talk2biomodels/tests/test_search_models.py

@@ -2,7 +2,7 @@
 Test cases for Talk2Biomodels search models tool.
 '''
 
-from langchain_core.messages import HumanMessage
+from langchain_core.messages import HumanMessage, ToolMessage
 from langchain_nvidia_ai_endpoints import ChatNVIDIA
 from ..agents.t2b_agent import get_app
 
@@ -22,9 +22,15 @@ def test_search_models_tool():
                         {"messages": [HumanMessage(content=prompt)]},
                         config=config
                     )
-    assistant_msg = response["messages"][-1].content
-    # Check if the assistant message is a string
-    assert isinstance(assistant_msg, str)
-    # Check if the assistant message contains the
-    # biomodel id BIO0000000537
-    assert "BIOMD0000000537" in assistant_msg
+    # Extract the assistant artifact which contains
+    # all the search results
+    found_model_537 = False
+    for msg in response["messages"]:
+        if isinstance(msg, ToolMessage) and msg.name == "search_models":
+            msg_artifact = msg.artifact
+            for model in msg_artifact["dic_data"]:
+                if model["id"] == "BIOMD0000000537":
+                    found_model_537 = True
+            break
+    # Check if the model BIOMD0000000537 is found
+    assert found_model_537

aiagents4pharma-1.22.2/aiagents4pharma/talk2biomodels/tools/custom_plotter.py

@@ -0,0 +1,157 @@
+#!/usr/bin/env python3
+
+"""
+Tool for plotting a custom y-axis of a simulation plot.
+"""
+
+import logging
+from typing import Type, Annotated, List, Tuple, Union, Literal
+from pydantic import BaseModel, Field
+import hydra
+import pandas as pd
+from langchain_core.tools import BaseTool
+from langchain_core.prompts import ChatPromptTemplate
+from langgraph.prebuilt import InjectedState
+from .load_biomodel import ModelData, load_biomodel
+from .utils import get_model_units
+
+# Initialize logger
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+def extract_relevant_species(question, species_names, state):
+    """
+    Extract the relevant species from the user question.
+
+    Args:
+        question (str): The user question.
+        species_names (list): The species names available in the simulation results.
+        state (dict): The state of the graph.
+
+    Returns:
+        CustomHeader: The relevant species
+    """
+    # In the following code, we extract the species
+    # from the user question. We use Literal to restrict
+    # the species names to the ones available in the
+    # simulation results.
+    class CustomHeader(BaseModel):
+        """
+        A list of species based on user question.
+
+        This is a Pydantic model that restricts the species
+        names to the ones available in the simulation results.
+        
+        If no species is relevant, set the attribute
+        `relevant_species` to None.
+        """
+        relevant_species: Union[None, List[Literal[*species_names]]] = Field(
+                description="This is a list of species based on the user question."
+                "It is restricted to the species available in the simulation results."
+                "If no species is relevant, set this attribute to None."
+                "If the user asks for very specific species (for example, using the"
+                "keyword `only` in the question), set this attribute to correspond "
+                "to the species available in the simulation results, otherwise set it to None."
+                )
+    # Load hydra configuration
+    with hydra.initialize(version_base=None, config_path="../configs"):
+        cfg = hydra.compose(config_name='config',
+                            overrides=['tools/custom_plotter=default'])
+        cfg = cfg.tools.custom_plotter
+    # Get the system prompt
+    system_prompt = cfg.system_prompt_custom_header
+    # Create an instance of the LLM model
+    logging.log(logging.INFO, "LLM model: %s", state['llm_model'])
+    llm = state['llm_model']
+    llm_with_structured_output = llm.with_structured_output(CustomHeader)
+    prompt = ChatPromptTemplate.from_messages([("system", system_prompt),
+                                               ("human", "{input}")])
+    few_shot_structured_llm = prompt | llm_with_structured_output
+    return few_shot_structured_llm.invoke(question)
+
+class CustomPlotterInput(BaseModel):
+    """
+    Input schema for the custom plotter tool.
+    """
+    question: str = Field(description="Description of the plot")
+    sys_bio_model: ModelData = Field(description="model data",
+                                     default=None)
+    simulation_name: str = Field(description="Name assigned to the simulation")
+    state: Annotated[dict, InjectedState]
+
+# Note: It's important that every field has type hints.
+# BaseTool is a Pydantic class and not having type hints
+# can lead to unexpected behavior.
+# Note: It's important that every field has type hints.
+# BaseTool is a Pydantic class and not having type hints
+# can lead to unexpected behavior.
+class CustomPlotterTool(BaseTool):
+    """
+    Tool for custom plotting the y-axis of a plot.
+    """
+    name: str = "custom_plotter"
+    description: str = '''A visualization tool designed to extract and display a subset
+                        of the larger simulation plot generated by the simulate_model tool.
+                        It allows users to specify particular species for the y-axis, 
+                        providing a more targeted view of key species without the clutter 
+                        of the full plot.'''
+    args_schema: Type[BaseModel] = CustomPlotterInput
+    response_format: str = "content_and_artifact"
+
+    def _run(self,
+             question: str,
+             sys_bio_model: ModelData,
+             simulation_name: str,
+             state: Annotated[dict, InjectedState]
+             ) -> Tuple[str, Union[None, List[str]]]:
+        """
+        Run the tool.
+
+        Args:
+            question (str): The question about the custom plot.
+            sys_bio_model (ModelData): The model data.
+            simulation_name (str): The name assigned to the simulation.
+            state (dict): The state of the graph.
+
+        Returns:
+            str: The answer to the question
+        """
+        logger.log(logging.INFO, "Calling custom_plotter tool %s, %s", question, sys_bio_model)
+        # Load the model
+        sbml_file_path = state['sbml_file_path'][-1] if len(state['sbml_file_path']) > 0 else None
+        model_object = load_biomodel(sys_bio_model, sbml_file_path=sbml_file_path)
+        dic_simulated_data = {}
+        for data in state["dic_simulated_data"]:
+            for key in data:
+                if key not in dic_simulated_data:
+                    dic_simulated_data[key] = []
+                dic_simulated_data[key] += [data[key]]
+        # Create a pandas dataframe from the dictionary
+        df = pd.DataFrame.from_dict(dic_simulated_data)
+        # Get the simulated data for the current tool call
+        df = pd.DataFrame(
+                df[df['name'] == simulation_name]['data'].iloc[0]
+                )
+        # df = pd.DataFrame.from_dict(state['dic_simulated_data'])
+        species_names = df.columns.tolist()
+        # Exclude the time column
+        species_names.remove('Time')
+        logging.log(logging.INFO, "Species names: %s", species_names)
+        # Extract the relevant species from the user question
+        results = extract_relevant_species(question, species_names, state)
+        print (results)
+        if results.relevant_species is None:
+            raise ValueError("No species found in the simulation results \
+                             that matches the user prompt.")
+        extracted_species = []
+        # Extract the species from the results
+        # that are available in the simulation results
+        for species in results.relevant_species:
+            if species in species_names:
+                extracted_species.append(species)
+        logging.info("Extracted species: %s", extracted_species)
+        # Include the time column
+        extracted_species.insert(0, 'Time')
+        return f"Custom plot {simulation_name}",{
+                            'dic_data': df[extracted_species].to_dict(orient='records')
+                            }| get_model_units(model_object)

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma/talk2biomodels/tools/get_modelinfo.py

@@ -100,6 +100,8 @@ class GetModelInfoTool(BaseTool):
         # Extract parameters from the model
         if requested_model_info.parameters:
             df_parameters = basico.model_info.get_parameters(model=model_obj.copasi_model)
+            if df_parameters is None:
+                raise ValueError("Unable to extract parameters from the model.")
             # Convert index into a column
             df_parameters.reset_index(inplace=True)
             dic_results['Parameters'] = df_parameters[

aiagents4pharma-1.22.2/aiagents4pharma/talk2biomodels/tools/load_biomodel.py

@@ -0,0 +1,38 @@
+#!/usr/bin/env python3
+
+"""
+Function for loading the BioModel.
+"""
+
+from typing import Annotated, Any, Union
+from pydantic import BaseModel, BeforeValidator
+from ..models.basico_model import BasicoModel
+
+def ensure_biomodel_id(value: Any) -> Any:
+    """
+    Ensure that the biomodel_id is an integer or a string starting with 'BIOMD' or 'MODEL'.
+    """
+    if isinstance(value, int):
+        return value
+    if isinstance(value, str) and (value.startswith("BIOMD") or value.startswith("MODEL")):
+        return value
+    raise ValueError("biomodel_id must be an integer or a string starting with 'BIOMD' or 'MODEL'.")
+
+class ModelData(BaseModel):
+    """
+    Base model for the model data.
+    """
+    biomodel_id: Annotated[Union[int, str], BeforeValidator(ensure_biomodel_id)] = None
+    # sbml_file_path: Optional[str] = None
+    use_uploaded_sbml_file: bool = False
+
+def load_biomodel(sys_bio_model, sbml_file_path=None):
+    """
+    Load the BioModel.
+    """
+    model_object = None
+    if sys_bio_model.biomodel_id:
+        model_object = BasicoModel(biomodel_id=sys_bio_model.biomodel_id)
+    elif sbml_file_path:
+        model_object = BasicoModel(sbml_file_path=sbml_file_path)
+    return model_object

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma/talk2biomodels/tools/parameter_scan.py

@@ -17,6 +17,7 @@ from langchain_core.messages import ToolMessage
 from langchain_core.tools.base import InjectedToolCallId
 from .load_biomodel import ModelData, load_biomodel
 from .load_arguments import TimeData, SpeciesInitialData
+from .utils import get_model_units
 
 # Initialize logger
 logging.basicConfig(level=logging.INFO)
@@ -119,7 +120,11 @@ def run_parameter_scan(model_object,
     """
     # Extract all parameter names from the model
     df_all_parameters = basico.model_info.get_parameters(model=model_object.copasi_model)
-    all_parameters = df_all_parameters.index.tolist()
+    all_parameters = []
+    if df_all_parameters is not None:
+        # For example model 10 in the BioModels database
+        # has no parameters
+        all_parameters = df_all_parameters.index.tolist()
 
     # Extract all species name from the model
     df_all_species = basico.model_info.get_species(model=model_object.copasi_model)
@@ -280,7 +285,8 @@ class ParameterScanTool(BaseTool):
                 "messages": [
                     ToolMessage(
                         content=f"Parameter scan results of {arg_data.experiment_name}",
-                        tool_call_id=tool_call_id
+                        tool_call_id=tool_call_id,
+                        artifact=get_model_units(model_object)
                         )
                     ],
                 }

aiagents4pharma-1.22.2/aiagents4pharma/talk2biomodels/tools/search_models.py

@@ -0,0 +1,85 @@
+#!/usr/bin/env python3
+
+"""
+Tool for searching models based on search query.
+"""
+
+from typing import Type, Annotated
+import logging
+from pydantic import BaseModel, Field
+import pandas as pd
+from basico import biomodels
+from langgraph.types import Command
+from langchain_core.tools import BaseTool
+from langchain_core.messages import ToolMessage
+from langchain_core.tools.base import InjectedToolCallId
+
+# Initialize logger
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+class SearchModelsInput(BaseModel):
+    """
+    Input schema for the search models tool.
+    """
+    query: str = Field(description="Search models query", default=None)
+    num_query: int = Field(description="Top number of models to search",
+                           default=10,
+                           le=100)
+    tool_call_id: Annotated[str, InjectedToolCallId]
+
+# Note: It's important that every field has type hints. BaseTool is a
+# Pydantic class and not having type hints can lead to unexpected behavior.
+class SearchModelsTool(BaseTool):
+    """
+    Tool for returning the search results based on the search query.
+    """
+    name: str = "search_models"
+    description: str = "Search for only manually curated models in "
+    "the BioMmodels database based on keywords."
+    args_schema: Type[BaseModel] = SearchModelsInput
+    return_direct: bool = False
+
+    def _run(self,
+             tool_call_id: Annotated[str, InjectedToolCallId],
+             query: str = None,
+             num_query: int = 10) -> dict:
+        """
+        Run the tool.
+
+        Args:
+            query (str): The search query.
+            num_query (int): The number of models to search.
+            tool_call_id (str): The tool call ID.
+
+        Returns:
+            dict: The answer to the question in the form of a dictionary.
+        """
+        logger.log(logging.INFO, "Searching models with the query and number %s, %s",
+                   query, num_query)
+        # Search for models based on the query
+        search_results = biomodels.search_for_model(query, num_results=num_query)
+        # Convert the search results to a pandas DataFrame
+        df = pd.DataFrame(search_results)
+        # Prepare a message to return
+        first_n = min(3, len(search_results))
+        content = f"Found {len(search_results)} manually curated models"
+        content += f" for the query: {query}."
+        # Pass the first 3 models to the LLM
+        # to avoid hallucinations
+        content += f" Here is the summary of the first {first_n} models:"
+        for i in range(first_n):
+            content += f"\nModel {i+1}: {search_results[i]['name']} (ID: {search_results[i]['id']})"
+        # Return the updated state of the tool
+        return Command(
+                update={
+                    # update the message history
+                    "messages": [
+                        ToolMessage(
+                            content=content,
+                            tool_call_id=tool_call_id,
+                            artifact={'dic_data': df.to_dict(orient='records')}
+                            )
+                        ],
+                    }
+            )

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma/talk2biomodels/tools/simulate_model.py

@@ -14,6 +14,7 @@ from langchain_core.messages import ToolMessage
 from langchain_core.tools.base import InjectedToolCallId
 from .load_biomodel import ModelData, load_biomodel
 from .load_arguments import ArgumentData, add_rec_events
+from .utils import get_model_units
 
 # Initialize logger
 logging.basicConfig(level=logging.INFO)
@@ -116,7 +117,8 @@ class SimulateModelTool(BaseTool):
                 "messages": [
                     ToolMessage(
                         content=f"Simulation results of {arg_data.experiment_name}",
-                        tool_call_id=tool_call_id
+                        tool_call_id=tool_call_id,
+                        artifact=get_model_units(model_object)
                         )
                     ],
                 }

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma/talk2biomodels/tools/steady_state.py

@@ -134,6 +134,7 @@ class SteadyStateTool(BaseTool):
         # Run the parameter scan
         df_steady_state = run_steady_state(model_object,
                                            dic_species_to_be_analyzed_before_experiment)
+        print (df_steady_state)
         # Prepare the dictionary of scanned data
         # that will be passed to the state of the graph
         dic_steady_state_data = {
@@ -160,7 +161,8 @@ class SteadyStateTool(BaseTool):
                         content=f"Steady state analysis of"
                                 f" {arg_data.experiment_name}"
                                 " was successful.",
-                        tool_call_id=tool_call_id
+                        tool_call_id=tool_call_id,
+                        artifact={'dic_data': df_steady_state.to_dict(orient='records')}
                         )
                     ],
                 }

aiagents4pharma-1.22.2/aiagents4pharma/talk2biomodels/tools/utils.py

@@ -0,0 +1,22 @@
+#!/usr/bin/env python3
+
+"""
+Utility functions for T2B tools.
+"""
+
+import basico
+
+def get_model_units(model_object):
+    """
+    Get the units of the model.
+
+    Args:
+        model_object: The model object.
+
+    Returns:
+        dict: The units of the model.
+    """
+    model_units = basico.model_info.get_model_units(model=model_object.copasi_model)
+    model_units_y = model_units['quantity_unit']
+    model_units_x = model_units['time_unit']
+    return {'y_axis_label': model_units_y, 'x_axis_label': model_units_x}

aiagents4pharma-1.22.2/aiagents4pharma/talk2knowledgegraphs/configs/tools/subgraph_summarization/__init__.py

@@ -0,0 +1,3 @@
+'''
+Import all the modules in the package
+'''

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: aiagents4pharma
-Version: 1.22.0
+Version: 1.22.2
 Summary: AI Agents for drug discovery, drug development, and other pharmaceutical R&D.
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/aiagents4pharma.egg-info/SOURCES.txt

@@ -23,6 +23,8 @@ aiagents4pharma/talk2biomodels/configs/agents/t2b_agent/default.yaml
 aiagents4pharma/talk2biomodels/configs/tools/__init__.py
 aiagents4pharma/talk2biomodels/configs/tools/ask_question/__init__.py
 aiagents4pharma/talk2biomodels/configs/tools/ask_question/default.yaml
+aiagents4pharma/talk2biomodels/configs/tools/custom_plotter/__init__.py
+aiagents4pharma/talk2biomodels/configs/tools/custom_plotter/default.yaml
 aiagents4pharma/talk2biomodels/configs/tools/get_annotation/__init__.py
 aiagents4pharma/talk2biomodels/configs/tools/get_annotation/default.yaml
 aiagents4pharma/talk2biomodels/models/__init__.py
@@ -37,6 +39,7 @@ aiagents4pharma/talk2biomodels/tests/test_basico_model.py
 aiagents4pharma/talk2biomodels/tests/test_get_annotation.py
 aiagents4pharma/talk2biomodels/tests/test_getmodelinfo.py
 aiagents4pharma/talk2biomodels/tests/test_integration.py
+aiagents4pharma/talk2biomodels/tests/test_load_biomodel.py
 aiagents4pharma/talk2biomodels/tests/test_param_scan.py
 aiagents4pharma/talk2biomodels/tests/test_query_article.py
 aiagents4pharma/talk2biomodels/tests/test_search_models.py
@@ -55,6 +58,7 @@ aiagents4pharma/talk2biomodels/tools/query_article.py
 aiagents4pharma/talk2biomodels/tools/search_models.py
 aiagents4pharma/talk2biomodels/tools/simulate_model.py
 aiagents4pharma/talk2biomodels/tools/steady_state.py
+aiagents4pharma/talk2biomodels/tools/utils.py
 aiagents4pharma/talk2cells/__init__.py
 aiagents4pharma/talk2cells/agents/__init__.py
 aiagents4pharma/talk2cells/agents/scp_agent.py

{aiagents4pharma-1.22.0 → aiagents4pharma-1.22.2}/pyproject.toml

@@ -75,6 +75,7 @@ aiagents4pharma = [
   "talk2biomodels/configs/agents/t2b_agent/*",
   "talk2biomodels/configs/tools/ask_question/*",
   "talk2biomodels/configs/tools/get_annotation/*",
+  "talk2biomodels/configs/tools/custom_plotter/*",
   "talk2knowledgegraphs/configs/*",
   "talk2knowledgegraphs/configs/agents/t2kg_agent/*",
   "talk2knowledgegraphs/configs/app/frontend/*",

aiagents4pharma-1.22.2/release_version.txt

@@ -0,0 +1 @@
+v1.22.2