aiagents4pharma 1.14.1__tar.gz → 1.15.1__tar.gz

{aiagents4pharma-1.14.1 → aiagents4pharma-1.15.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: aiagents4pharma
-Version: 1.14.1
+Version: 1.15.1
 Summary: AI Agents for drug discovery, drug development, and other pharmaceutical R&D
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
@@ -29,6 +29,7 @@ Requires-Dist: pandas==2.2.3
 Requires-Dist: plotly==5.24.1
 Requires-Dist: pydantic==2.9.2
 Requires-Dist: pylint==3.3.1
+Requires-Dist: pypdf==5.2.0
 Requires-Dist: pytest==8.3.3
 Requires-Dist: pytest-asyncio==0.25.2
 Requires-Dist: streamlit==1.39.0

aiagents4pharma-1.15.1/aiagents4pharma/configs/talk2biomodels/agents/t2b_agent/default.yaml

@@ -0,0 +1,14 @@
+_target_: talk2biomodels.agents.t2b_agent.get_app
+state_modifier: >
+  You are Talk2BioModels agent.
+  If the user asks for the uploaded model, 
+  then pass the use_uploaded_model argument
+  as True. If the user asks for simulation
+  or param_scan or steady state, suggest a
+  value for the `experiment_name` argument.
+
+  If the user asks question related to the
+  uploaded document/pdf/article/document,
+  use the tool `query_article` to answer the
+  question. Please note that the `experiment_name`
+  argument may be unrelated to the question asked.

aiagents4pharma-1.15.1/aiagents4pharma/configs/talk2biomodels/tools/ask_question/default.yaml

@@ -0,0 +1,39 @@
+_target_: talk2biomodels.tools.ask_question.AskQuestionTool
+steady_state_prompt: >
+  Following are header columns of the data:
+  `species_name`: Name of the species,
+  `compartment`: Compartment of the species,
+  `initial_concentration`: Concentration of the species at the start of steady state,
+  `steady_state_concentration`: Concentration of the species at steady state,
+  `steady_state_transition_time`: Time taken to reach steady state,
+  `display_name`: An alias for the species name.
+
+  Here are some instructions to help you answer questions:
+
+  1. Before you answer any question, follow the plan and solve
+  technique. Start by understanding the question, then plan your
+  approach to solve the question, and finally solve the question
+  by following the plan. Always give a brief explanation of your
+  answer to the user.
+
+  2. If the user wants to know the time taken by the model to reach
+  steady state, you should look at the steady_state_transition_time
+  column of the data for the model species. The highest value in
+  this column is the time taken by the model to reach steady state.
+
+  3. To get accurate results, trim the data to the relevant columns
+  before performing any calculations. This will help you avoid
+  errors in your calculations, and ignore irrelevant data.
+
+  4. Please use the units provided below to answer the questions.
+simulation_prompt: >
+  Following is the information about the data frame:
+  1. First column is the time column, and the rest of the columns
+  are the species names.
+
+  2. While the time column records the simulation time points, the
+  rest of the columns record the concentration of the species at
+  each time point.
+
+  3. Please use the units provided below to answer the questions.
+  

{aiagents4pharma-1.14.1 → aiagents4pharma-1.15.1}/aiagents4pharma/talk2biomodels/agents/t2b_agent.py

@@ -19,6 +19,7 @@ from ..tools.get_annotation import GetAnnotationTool
 from ..tools.ask_question import AskQuestionTool
 from ..tools.parameter_scan import ParameterScanTool
 from ..tools.steady_state import SteadyStateTool
+from ..tools.query_article import QueryArticle
 from ..states.state_talk2biomodels import Talk2Biomodels
 
 # Initialize logger
@@ -46,7 +47,8 @@ def get_app(uniq_id, llm_model='gpt-4o-mini'):
                     GetModelInfoTool(),
                     SteadyStateTool(),
                     ParameterScanTool(),
-                    GetAnnotationTool()
+                    GetAnnotationTool(),
+                    QueryArticle()
                 ])
 
     # Define the model

{aiagents4pharma-1.14.1 → aiagents4pharma-1.15.1}/aiagents4pharma/talk2biomodels/states/state_talk2biomodels.py

@@ -27,6 +27,7 @@ class Talk2Biomodels(AgentState):
     The state for the Talk2BioModels agent.
     """
     llm_model: str
+    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

aiagents4pharma-1.15.1/aiagents4pharma/talk2biomodels/tests/test_query_article.py

@@ -0,0 +1,76 @@
+'''
+Test cases for Talk2Biomodels query_article tool.
+'''
+
+from pydantic import BaseModel, Field
+from langchain_core.messages import HumanMessage, ToolMessage
+from langchain_openai import ChatOpenAI
+from ..agents.t2b_agent import get_app
+
+class Article(BaseModel):
+    '''
+    Article schema.
+    '''
+    title: str = Field(description="Title of the article.")
+
+def test_query_article_with_an_article():
+    '''
+    Test the query_article tool by providing an article.
+    '''
+    unique_id = 12345
+    app = get_app(unique_id)
+    config = {"configurable": {"thread_id": unique_id}}
+    # Update state by providing the pdf file name
+    app.update_state(config,
+      {"pdf_file_name": "aiagents4pharma/talk2biomodels/tests/article_on_model_537.pdf"})
+    prompt = "What is the title of the article?"
+    # Test the tool query_article
+    response = app.invoke(
+                        {"messages": [HumanMessage(content=prompt)]},
+                        config=config
+                    )
+    # Get the response from the tool
+    assistant_msg = response["messages"][-1].content
+    # Prepare a LLM that can be used as a judge
+    llm = ChatOpenAI(model='gpt-4o-mini', temperature=0)
+    # Make it return a structured output
+    structured_llm = llm.with_structured_output(Article)
+    # Prepare a prompt for the judge
+    prompt = "Given the text below, what is the title of the article?"
+    prompt += f"\n\n{assistant_msg}"
+    # Get the structured output
+    article = structured_llm.invoke(prompt)
+    # Check if the article title is correct
+    expected_title = "A Multiscale Model of IL-6–Mediated "
+    expected_title += "Immune Regulation in Crohn’s Disease"
+    # Check if the article title is correct
+    assert article.title == expected_title
+
+def test_query_article_without_an_article():
+    '''
+    Test the query_article tool without providing an article.
+    The status of the tool should be error.
+    '''
+    unique_id = 12345
+    app = get_app(unique_id)
+    config = {"configurable": {"thread_id": unique_id}}
+    prompt = "What is the title of the uploaded article?"
+    # Test the tool query_article
+    app.invoke(
+            {"messages": [HumanMessage(content=prompt)]},
+            config=config
+        )
+    current_state = app.get_state(config)
+    # Get the messages from the current state
+    # and reverse the order
+    reversed_messages = current_state.values["messages"][::-1]
+    # Loop through the reversed messages
+    # until a ToolMessage is found.
+    tool_status_is_error = False
+    for msg in reversed_messages:
+        if isinstance(msg, ToolMessage):
+            # Skip until it finds a ToolMessage
+            if msg.name == "query_article" and msg.status == "error":
+                tool_status_is_error = True
+                break
+    assert tool_status_is_error

{aiagents4pharma-1.14.1 → aiagents4pharma-1.15.1}/aiagents4pharma/talk2biomodels/tools/__init__.py

@@ -10,3 +10,4 @@ from . import parameter_scan
 from . import steady_state
 from . import load_biomodel
 from . import get_annotation
+from . import query_article

aiagents4pharma-1.15.1/aiagents4pharma/talk2biomodels/tools/query_article.py

@@ -0,0 +1,59 @@
+#!/usr/bin/env python3
+
+"""
+Tool for asking questions to the article.
+"""
+
+import logging
+from typing import Type, Annotated
+from pydantic import BaseModel, Field
+from langchain_core.tools import BaseTool
+from langchain_core.vectorstores import InMemoryVectorStore
+from langchain_openai.embeddings import OpenAIEmbeddings
+from langchain_community.document_loaders import PyPDFLoader
+from langgraph.prebuilt import InjectedState
+
+# Initialize logger
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+class QueryArticleInput(BaseModel):
+    """
+    Input schema for the query_articles tool.
+    """
+    question: Annotated[str, Field(description="User question to search articles.")]
+    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.
+class QueryArticle(BaseTool):
+    """
+    Tool to ask questions to the article.
+    """
+    name: str = "query_article"
+    description: str = "Ask questions to the article."
+    args_schema: Type[BaseModel] = QueryArticleInput
+
+    def _run(self,
+             question: str,
+             state: Annotated[dict, InjectedState]):
+        """
+        Run the tool.
+
+        Args:
+            query (str): The search query.
+        """
+        logger.log(logging.INFO, "loading the article from %s", state['pdf_file_name'])
+        logger.log(logging.INFO, "searching the article with the question: %s", question)
+        # Load the article
+        loader = PyPDFLoader(state['pdf_file_name'])
+        # Load the pages of the article
+        pages = []
+        for page in loader.lazy_load():
+            pages.append(page)
+        # Create a vector store from the pages
+        vector_store = InMemoryVectorStore.from_documents(pages, OpenAIEmbeddings())
+        # Search the article with the question
+        docs = vector_store.similarity_search(question)
+        # Return the content of the pages
+        return "\n".join([doc.page_content for doc in docs])

{aiagents4pharma-1.14.1 → aiagents4pharma-1.15.1}/aiagents4pharma.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: aiagents4pharma
-Version: 1.14.1
+Version: 1.15.1
 Summary: AI Agents for drug discovery, drug development, and other pharmaceutical R&D
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
@@ -29,6 +29,7 @@ Requires-Dist: pandas==2.2.3
 Requires-Dist: plotly==5.24.1
 Requires-Dist: pydantic==2.9.2
 Requires-Dist: pylint==3.3.1
+Requires-Dist: pypdf==5.2.0
 Requires-Dist: pytest==8.3.3
 Requires-Dist: pytest-asyncio==0.25.2
 Requires-Dist: streamlit==1.39.0

{aiagents4pharma-1.14.1 → aiagents4pharma-1.15.1}/aiagents4pharma.egg-info/SOURCES.txt

@@ -16,6 +16,7 @@ aiagents4pharma/configs/talk2biomodels/agents/t2b_agent/__init__.py
 aiagents4pharma/configs/talk2biomodels/agents/t2b_agent/default.yaml
 aiagents4pharma/configs/talk2biomodels/tools/__init__.py
 aiagents4pharma/configs/talk2biomodels/tools/ask_question/__init__.py
+aiagents4pharma/configs/talk2biomodels/tools/ask_question/default.yaml
 aiagents4pharma/talk2biomodels/__init__.py
 aiagents4pharma/talk2biomodels/agents/__init__.py
 aiagents4pharma/talk2biomodels/agents/t2b_agent.py
@@ -36,6 +37,7 @@ aiagents4pharma/talk2biomodels/tests/test_get_annotation.py
 aiagents4pharma/talk2biomodels/tests/test_getmodelinfo.py
 aiagents4pharma/talk2biomodels/tests/test_integration.py
 aiagents4pharma/talk2biomodels/tests/test_param_scan.py
+aiagents4pharma/talk2biomodels/tests/test_query_article.py
 aiagents4pharma/talk2biomodels/tests/test_search_models.py
 aiagents4pharma/talk2biomodels/tests/test_simulate_model.py
 aiagents4pharma/talk2biomodels/tests/test_steady_state.py
@@ -48,6 +50,7 @@ aiagents4pharma/talk2biomodels/tools/get_modelinfo.py
 aiagents4pharma/talk2biomodels/tools/load_arguments.py
 aiagents4pharma/talk2biomodels/tools/load_biomodel.py
 aiagents4pharma/talk2biomodels/tools/parameter_scan.py
+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-1.14.1 → aiagents4pharma-1.15.1}/aiagents4pharma.egg-info/requires.txt

@@ -19,6 +19,7 @@ pandas==2.2.3
 plotly==5.24.1
 pydantic==2.9.2
 pylint==3.3.1
+pypdf==5.2.0
 pytest==8.3.3
 pytest-asyncio==0.25.2
 streamlit==1.39.0

{aiagents4pharma-1.14.1 → aiagents4pharma-1.15.1}/pyproject.toml

@@ -34,6 +34,7 @@ dependencies = [
     "plotly==5.24.1",
     "pydantic==2.9.2",
     "pylint==3.3.1",
+    "pypdf==5.2.0",
     "pytest==8.3.3",
     "pytest-asyncio==0.25.2",
     "streamlit==1.39.0",
@@ -67,7 +68,9 @@ include = ["aiagents4pharma*"]
 [tool.setuptools.package-data]
 aiagents4pharma = [
             "configs/*",
-            "configs/talk2biomodels/agents/t2b_agent/*"]
+            "configs/talk2biomodels/agents/t2b_agent/*",
+            "configs/talk2biomodels/tools/ask_question/*",
+            "configs/talk2biomodels/tools/get_annotation/*"]
 
 [tool.pytest.ini_options]
 asyncio_default_fixture_loop_scope = "function"

aiagents4pharma-1.15.1/release_version.txt

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

aiagents4pharma-1.14.1/aiagents4pharma/configs/talk2biomodels/agents/t2b_agent/default.yaml

@@ -1,8 +0,0 @@
-_target_: talk2biomodels.agents.t2b_agent.get_app
-state_modifier: >
-  You are Talk2BioModels agent.
-  If the user asks for the uploaded model, 
-  then pass the use_uploaded_model argument
-  as True. If the user asks for simulation
-  or param_scan or steady state, suggest a
-  value for the `experiment_name` argument.

aiagents4pharma-1.14.1/release_version.txt

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