aiagents4pharma 1.30.2__py3-none-any.whl → 1.30.4__py3-none-any.whl

aiagents4pharma/talk2scholars/__init__.py

@@ -3,3 +3,5 @@ This file is used to import all the modules in the package.
 """
 
 from . import agents, configs, state, tests, tools
+
+__all__ = ["agents", "configs", "state", "tests", "tools"]

aiagents4pharma/talk2scholars/agents/__init__.py

@@ -7,3 +7,11 @@ from . import s2_agent
 from . import paper_download_agent
 from . import zotero_agent
 from . import pdf_agent
+
+__all__ = [
+    "main_agent",
+    "s2_agent",
+    "paper_download_agent",
+    "zotero_agent",
+    "pdf_agent",
+]

aiagents4pharma/talk2scholars/configs/__init__.py

@@ -5,3 +5,5 @@ This package contains configuration settings and prompts used by various AI agen
 from . import agents
 from . import tools
 from . import app
+
+__all__ = ["agents", "tools", "app"]

aiagents4pharma/talk2scholars/configs/agents/__init__.py

@@ -3,3 +3,5 @@ Import all the modules in the package
 """
 
 from . import talk2scholars
+
+__all__ = ["talk2scholars"]

aiagents4pharma/talk2scholars/configs/agents/talk2scholars/__init__.py

@@ -6,3 +6,5 @@ from . import s2_agent
 from . import main_agent
 from . import pdf_agent
 from . import zotero_agent
+
+__all__ = ["s2_agent", "main_agent", "pdf_agent", "zotero_agent"]

aiagents4pharma/talk2scholars/configs/app/__init__.py

@@ -3,3 +3,5 @@ Import all the modules in the package
 """
 
 from . import frontend
+
+__all__ = ["frontend"]

aiagents4pharma/talk2scholars/configs/tools/__init__.py

@@ -8,3 +8,12 @@ from . import multi_paper_recommendation
 from . import question_and_answer
 from . import zotero_read
 from . import zotero_write
+
+__all__ = [
+    "search",
+    "single_paper_recommendation",
+    "multi_paper_recommendation",
+    "question_and_answer",
+    "zotero_read",
+    "zotero_write",
+]

aiagents4pharma/talk2scholars/state/__init__.py

@@ -1,5 +1,7 @@
-'''
+"""
 This file is used to import all the modules in the package.
-'''
+"""
 
 from . import state_talk2scholars
+
+__all__ = ["state_talk2scholars"]

aiagents4pharma/talk2scholars/tests/test_s2_multi.py

@@ -12,6 +12,7 @@ import hydra
 from aiagents4pharma.talk2scholars.tools.s2.multi_paper_rec import (
     get_multi_paper_recommendations,
 )
+from aiagents4pharma.talk2scholars.tools.s2.utils import multi_helper
 
 # --- Dummy Hydra Config Setup ---
 
@@ -285,20 +286,21 @@ def test_multi_paper_rec_requests_exception(monkeypatch):
 
 def test_multi_paper_rec_no_response(monkeypatch):
     """
-    Test that get_multi_paper_recommendations raises a RuntimeError when no response is obtained.
-    This is simulated by patching 'range' in the underlying function's globals to
-    return an empty iterator,
-    so that the for loop does not iterate and response remains None.
+    Test that get_multi_paper_recommendations raises a RuntimeError
+    when no response is obtained. This is simulated by patching 'range'
+    in the module namespace of multi_helper to return an empty iterator,
+    so that the for loop in _fetch_recommendations never iterates and
+    self.response remains None.
     """
-    # Patch 'range' in the underlying function's globals (accessed via .func.__globals__)
-    monkeypatch.setitem(
-        get_multi_paper_recommendations.func.__globals__, "range", lambda x: iter([])
-    )
+    # Inject a patched 'range' into the multi_helper module's dictionary.
+    monkeypatch.setitem(multi_helper.__dict__, "range", lambda x: iter([]))
+
     tool_call_id = "test_tool_call_id"
     input_data = {
         "paper_ids": ["p1", "p2"],
         "tool_call_id": tool_call_id,
     }
+
     with pytest.raises(
         RuntimeError, match="Failed to obtain a response from the Semantic Scholar API."
     ):

aiagents4pharma/talk2scholars/tests/test_s2_search.py

@@ -9,6 +9,7 @@ import requests
 from langgraph.types import Command
 from langchain_core.messages import ToolMessage
 from aiagents4pharma.talk2scholars.tools.s2.search import search_tool
+from aiagents4pharma.talk2scholars.tools.s2.utils import search_helper
 
 # --- Dummy Hydra Config Setup ---
 
@@ -269,12 +270,15 @@ def test_search_tool_requests_exception(monkeypatch):
 
 def test_search_tool_no_response(monkeypatch):
     """
-    Test that search_tool raises a RuntimeError when no response is obtained.
-    This is simulated by patching 'range' in the original function's globals (a dict)
-    so that it returns an empty iterator, leaving response as None.
+    Test that search_tool raises a RuntimeError when no response
+    is obtained. This is simulated by patching 'range' in the
+    module namespace of search_helper to return an empty iterator,
+    so that the for-loop in _fetch_papers never iterates and self.response
+    remains None.
     """
-    # Patch 'range' in the original function's globals using setitem.
-    monkeypatch.setitem(search_tool.func.__globals__, "range", lambda x: iter([]))
+    # Patch 'range' in the module globals of search_helper.
+    monkeypatch.setitem(search_helper.__dict__, "range", lambda x: iter([]))
+
     tool_call_id = "test_tool_call_id"
     with pytest.raises(
         RuntimeError, match="Failed to obtain a response from the Semantic Scholar API."

aiagents4pharma/talk2scholars/tests/test_s2_single.py

@@ -11,6 +11,7 @@ from langchain_core.messages import ToolMessage
 from aiagents4pharma.talk2scholars.tools.s2.single_paper_rec import (
     get_single_paper_recommendations,
 )
+from aiagents4pharma.talk2scholars.tools.s2.utils import single_helper
 
 # --- Dummy Hydra Config Setup ---
 
@@ -279,14 +280,13 @@ def test_single_paper_rec_no_response(monkeypatch):
     """
     Test that get_single_paper_recommendations raises a RuntimeError
     when no response is obtained from the API.
-
-    This is simulated by patching 'range' in the underlying function's globals
-    to return an empty iterator, so the for-loop never iterates and response remains None.
+    This is simulated by patching 'range' in the module namespace
+    of single_helper to return an empty iterator, so that the for-loop
+    in _fetch_recommendations never iterates and response remains None.
     """
-    # Patch 'range' in the underlying function's globals (accessed via .func.__globals__)
-    monkeypatch.setitem(
-        get_single_paper_recommendations.func.__globals__, "range", lambda x: iter([])
-    )
+    # Patch 'range' in the module globals of single_helper.
+    monkeypatch.setitem(single_helper.__dict__, "range", lambda x: iter([]))
+
     tool_call_id = "test_tool_call_id"
     input_data = {
         "paper_id": "12345",

aiagents4pharma/talk2scholars/tests/test_zotero_path.py

@@ -292,7 +292,9 @@ class TestZoteroWrite:
     @pytest.fixture
     def mock_hydra(self):
         """Fixture to mock hydra configuration."""
-        with patch("hydra.compose") as mock_compose:
+        with patch(
+            "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.hydra.compose"
+        ) as mock_compose:
             cfg = MagicMock()
             cfg.tools.zotero_write.user_id = "test_user"
             cfg.tools.zotero_write.library_type = "user"
@@ -305,13 +307,15 @@ class TestZoteroWrite:
     @pytest.fixture
     def mock_zotero(self):
         """Fixture to mock Zotero client."""
-        with patch("pyzotero.zotero.Zotero") as mock_zot_class:
+        with patch(
+            "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.zotero.Zotero"
+        ) as mock_zot_class:
             mock_zot = MagicMock()
             mock_zot_class.return_value = mock_zot
             yield mock_zot
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.fetch_papers_for_save"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.fetch_papers_for_save"
     )
     def test_zotero_write_no_papers(self, mock_fetch):
         """When no papers exist (even after approval), the function raises a ValueError."""
@@ -335,12 +339,14 @@ class TestZoteroWrite:
         assert "No fetched papers were found to save" in str(excinfo.value)
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.fetch_papers_for_save"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.fetch_papers_for_save"
     )
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.find_or_create_collection"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.find_or_create_collection"
     )
-    def test_zotero_write_invalid_collection(self, mock_find, mock_fetch, mock_zotero):
+    def test_zotero_write_invalid_collection(
+        self, mock_find, mock_fetch, mock_zotero
+    ):
         """Saving to a nonexistent Zotero collection returns an error Command."""
         sample = {"paper1": {"Title": "Test Paper"}}
         mock_fetch.return_value = sample
@@ -372,12 +378,14 @@ class TestZoteroWrite:
         assert "Curiosity, Random" in msg
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.fetch_papers_for_save"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.fetch_papers_for_save"
     )
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.find_or_create_collection"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.find_or_create_collection"
     )
-    def test_zotero_write_success(self, mock_find, mock_fetch, mock_hydra, mock_zotero):
+    def test_zotero_write_success(
+        self, mock_find, mock_fetch, mock_hydra, mock_zotero
+    ):
         """A valid approved save returns a success Command with summary."""
         sample = {"paper1": {"Title": "Test Paper", "Authors": ["Test Author"]}}
         mock_fetch.return_value = sample
@@ -418,7 +426,11 @@ class TestZoteroRead:
     @pytest.fixture
     def mock_hydra(self):
         """Fixture to mock hydra configuration."""
-        with patch("hydra.initialize"), patch("hydra.compose") as mock_compose:
+        with patch(
+            "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.hydra.initialize"
+        ), patch(
+            "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.hydra.compose"
+        ) as mock_compose:
             cfg = MagicMock()
             cfg.tools.zotero_read.user_id = "test_user"
             cfg.tools.zotero_read.library_type = "user"
@@ -435,7 +447,9 @@ class TestZoteroRead:
     @pytest.fixture
     def mock_zotero(self):
         """Fixture to mock Zotero client."""
-        with patch("pyzotero.zotero.Zotero") as mock_zot_class:
+        with patch(
+            "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.zotero.Zotero"
+        ) as mock_zot_class:
             mock_zot = MagicMock()
             mock_zot_class.return_value = mock_zot
             yield mock_zot

aiagents4pharma/talk2scholars/tests/test_zotero_read.py

@@ -6,9 +6,7 @@ from types import SimpleNamespace
 import unittest
 from unittest.mock import patch, MagicMock
 from langgraph.types import Command
-from aiagents4pharma.talk2scholars.tools.zotero.zotero_read import (
-    zotero_read,
-)
+from aiagents4pharma.talk2scholars.tools.zotero.zotero_read import zotero_read
 
 
 # Dummy Hydra configuration to be used in tests
@@ -29,11 +27,13 @@ class TestZoteroSearchTool(unittest.TestCase):
     """Tests for Zotero search tool."""
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_read.get_item_collections"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.get_item_collections"
+    )
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.zotero.Zotero")
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.compose")
+    @patch(
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.initialize"
     )
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.zotero.Zotero")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.compose")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.initialize")
     def test_valid_query(
         self,
         mock_hydra_init,
@@ -104,11 +104,13 @@ class TestZoteroSearchTool(unittest.TestCase):
         self.assertIn("Number of papers found: 2", message_content)
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_read.get_item_collections"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.get_item_collections"
+    )
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.zotero.Zotero")
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.compose")
+    @patch(
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.initialize"
     )
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.zotero.Zotero")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.compose")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.initialize")
     def test_empty_query_fetch_all_items(
         self,
         mock_hydra_init,
@@ -154,11 +156,13 @@ class TestZoteroSearchTool(unittest.TestCase):
         )
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_read.get_item_collections"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.get_item_collections"
+    )
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.zotero.Zotero")
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.compose")
+    @patch(
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.initialize"
     )
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.zotero.Zotero")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.compose")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.initialize")
     def test_no_items_returned(
         self,
         mock_hydra_init,
@@ -187,11 +191,13 @@ class TestZoteroSearchTool(unittest.TestCase):
         self.assertIn("No items returned from Zotero", str(context.exception))
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_read.get_item_collections"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.get_item_collections"
+    )
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.zotero.Zotero")
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.compose")
+    @patch(
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.initialize"
     )
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.zotero.Zotero")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.compose")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.initialize")
     def test_filtering_no_matching_papers(
         self,
         mock_hydra_init,
@@ -252,11 +258,13 @@ class TestZoteroSearchTool(unittest.TestCase):
         self.assertEqual(len(filtered_papers), 2)
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_read.get_item_collections"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.get_item_collections"
+    )
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.zotero.Zotero")
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.compose")
+    @patch(
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.initialize"
     )
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.zotero.Zotero")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.compose")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.initialize")
     def test_items_api_exception(
         self,
         mock_hydra_init,
@@ -285,11 +293,13 @@ class TestZoteroSearchTool(unittest.TestCase):
         self.assertIn("Failed to fetch items from Zotero", str(context.exception))
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_read.get_item_collections"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.get_item_collections"
+    )
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.zotero.Zotero")
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.compose")
+    @patch(
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.initialize"
     )
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.zotero.Zotero")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.compose")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.initialize")
     def test_missing_key_in_item(
         self,
         mock_hydra_init,
@@ -344,11 +354,13 @@ class TestZoteroSearchTool(unittest.TestCase):
         self.assertEqual(len(filtered_papers), 1)
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_read.get_item_collections"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.get_item_collections"
+    )
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.zotero.Zotero")
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.compose")
+    @patch(
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.initialize"
     )
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.zotero.Zotero")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.compose")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.initialize")
     def test_item_not_dict(
         self,
         mock_hydra_init,
@@ -382,11 +394,13 @@ class TestZoteroSearchTool(unittest.TestCase):
         self.assertIn("No matching papers returned from Zotero", str(context.exception))
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_read.get_item_collections"
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.zotero_path.get_item_collections"
+    )
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.zotero.Zotero")
+    @patch("aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.compose")
+    @patch(
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.read_helper.hydra.initialize"
     )
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.zotero.Zotero")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.compose")
-    @patch("aiagents4pharma.talk2scholars.tools.zotero.zotero_read.hydra.initialize")
     def test_data_not_dict(
         self,
         mock_hydra_init,

aiagents4pharma/talk2scholars/tests/test_zotero_write.py

@@ -22,14 +22,14 @@ class TestZoteroSaveTool(unittest.TestCase):
     def setUp(self):
         """Patch Hydra and Zotero client globally"""
         self.hydra_init = patch(
-            "aiagents4pharma.talk2scholars.tools.zotero.zotero_write.hydra.initialize"
+            "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.hydra.initialize"
         ).start()
         self.hydra_compose = patch(
-            "aiagents4pharma.talk2scholars.tools.zotero.zotero_write.hydra.compose",
+            "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.hydra.compose",
             return_value=dummy_cfg,
         ).start()
         self.zotero_class = patch(
-            "aiagents4pharma.talk2scholars.tools.zotero.zotero_write.zotero.Zotero"
+            "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.zotero.Zotero"
         ).start()
 
         self.fake_zot = MagicMock()
@@ -57,7 +57,7 @@ class TestZoteroSaveTool(unittest.TestCase):
         return state
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_write.fetch_papers_for_save",
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.fetch_papers_for_save",
         return_value=None,
     )
     def test_no_papers_after_approval(self, mock_fetch):
@@ -74,11 +74,11 @@ class TestZoteroSaveTool(unittest.TestCase):
         mock_fetch.assert_called_once()
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_write.fetch_papers_for_save",
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.fetch_papers_for_save",
         return_value={"p1": {"Title": "X"}},
     )
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_write.find_or_create_collection",
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.find_or_create_collection",
         return_value=None,
     )
     def test_invalid_collection(self, mock_find, mock_fetch):
@@ -104,11 +104,11 @@ class TestZoteroSaveTool(unittest.TestCase):
         mock_find.return_value = None
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_write.fetch_papers_for_save",
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.fetch_papers_for_save",
         return_value={"p1": {"Title": "X"}},
     )
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_write.find_or_create_collection",
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.find_or_create_collection",
         return_value="colKey",
     )
     def test_save_failure(self, mock_find, mock_fetch):
@@ -131,11 +131,11 @@ class TestZoteroSaveTool(unittest.TestCase):
         mock_find.return_value = "colKey"
 
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_write.fetch_papers_for_save",
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.fetch_papers_for_save",
         return_value={"p1": {"Title": "X"}},
     )
     @patch(
-        "aiagents4pharma.talk2scholars.tools.zotero.zotero_write.find_or_create_collection",
+        "aiagents4pharma.talk2scholars.tools.zotero.utils.write_helper.find_or_create_collection",
         return_value="colKey",
     )
     def test_successful_save(self, mock_find, mock_fetch):

aiagents4pharma/talk2scholars/tools/__init__.py

@@ -7,3 +7,6 @@ Import statements
 from . import s2
 from . import pdf
 from . import zotero
+from . import paper_download
+
+__all__ = ["s2", "pdf", "zotero", "paper_download"]

aiagents4pharma/talk2scholars/tools/pdf/__init__.py

@@ -1,5 +1,7 @@
-'''
+"""
 This file is used to import all the modules in the package.
-'''
+"""
 
 from . import question_and_answer
+
+__all__ = ["question_and_answer"]

aiagents4pharma/talk2scholars/tools/s2/__init__.py

@@ -8,3 +8,12 @@ from . import search
 from . import single_paper_rec
 from . import query_results
 from . import retrieve_semantic_scholar_paper_id
+
+__all__ = [
+    "display_results",
+    "multi_paper_rec",
+    "search",
+    "single_paper_rec",
+    "query_results",
+    "retrieve_semantic_scholar_paper_id",
+]