aiagents4pharma 1.7.0__tar.gz → 1.8.0__tar.gz

{aiagents4pharma-1.7.0 → aiagents4pharma-1.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: aiagents4pharma
-Version: 1.7.0
+Version: 1.8.0
 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.7.0 → aiagents4pharma-1.8.0}/aiagents4pharma/talk2biomodels/tools/custom_plotter.py

@@ -6,11 +6,14 @@ Tool for plotting a custom figure.
 
 import logging
 from typing import Type, List, TypedDict, Annotated, Tuple, Union, Literal
+from typing import Type, List, TypedDict, Annotated, Tuple, Union, Literal
 from pydantic import BaseModel, Field
 import pandas as pd
+import pandas as pd
 from langchain_openai import ChatOpenAI
 from langchain_core.tools import BaseTool
 from langgraph.prebuilt import InjectedState
+from langgraph.prebuilt import InjectedState
 
 # Initialize logger
 logging.basicConfig(level=logging.INFO)
@@ -22,18 +25,23 @@ class CustomPlotterInput(BaseModel):
     """
     question: str = Field(description="Description of the plot")
     state: Annotated[dict, InjectedState]
+    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 plotting a custom plot.
+    Tool for making custom plots
     """
     name: str = "custom_plotter"
-    description: str = "A tool to plot a custom figure."
+    description: str = "A tool to make custom plots of the simulation results"
     args_schema: Type[BaseModel] = CustomPlotterInput
     response_format: str = "content_and_artifact"
+    response_format: str = "content_and_artifact"
 
     def _run(self,
              question: str,
@@ -45,6 +53,8 @@ class CustomPlotterTool(BaseTool):
         Args:
             question (str): The question about the custom plot.
             state (dict): The state of the graph.
+            question (str): The question about the custom plot.
+            state (dict): The state of the graph.
 
         Returns:
             str: The answer to the question
@@ -66,7 +76,7 @@ class CustomPlotterTool(BaseTool):
             A list of species based on user question.
             """
             relevant_species: Union[None, List[Literal[*species_names]]] = Field(
-                                        description="List of species based on user question.")
+                    description="List of species based on user question. If no relevant species are found, it will be None.")
         # Create an instance of the LLM model
         llm = ChatOpenAI(model=state['llm_model'], temperature=0)
         llm_with_structured_output = llm.with_structured_output(CustomHeader)

{aiagents4pharma-1.7.0 → aiagents4pharma-1.8.0}/aiagents4pharma/talk2biomodels/tools/search_models.py

@@ -27,7 +27,7 @@ class SearchModelsTool(BaseTool):
     Tool for returning the search results based on the search query.
     """
     name: str = "search_models"
-    description: str = "Search models based on search query."
+    description: str = "Search models in the BioMmodels database based on keywords."
     args_schema: Type[BaseModel] = SearchModelsInput
     return_direct: bool = True
 

{aiagents4pharma-1.7.0 → aiagents4pharma-1.8.0}/aiagents4pharma/talk2biomodels/tools/simulate_model.py

@@ -99,7 +99,7 @@ class SimulateModelTool(BaseTool):
     Tool for simulating a model.
     """
     name: str = "simulate_model"
-    description: str = "A tool to simulate a model."
+    description: str = "A tool to simulate a biomodel"
     args_schema: Type[BaseModel] = SimulateModelInput
 
     def _run(self,

{aiagents4pharma-1.7.0 → aiagents4pharma-1.8.0}/aiagents4pharma.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: aiagents4pharma
-Version: 1.7.0
+Version: 1.8.0
 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.8.0/release_version.txt

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

aiagents4pharma-1.7.0/release_version.txt

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