initial commit

venv/Lib/site-packages/langchain_community/tools/pubmed/__init__.py

@@ -0,0 +1 @@
+"""PubMed API toolkit."""

venv/Lib/site-packages/langchain_community/tools/pubmed/tool.py

@@ -0,0 +1,29 @@
+from typing import Optional
+
+from langchain_core.callbacks import CallbackManagerForToolRun
+from langchain_core.tools import BaseTool
+from pydantic import Field
+
+from langchain_community.utilities.pubmed import PubMedAPIWrapper
+
+
+class PubmedQueryRun(BaseTool):
+    """Tool that searches the PubMed API."""
+
+    name: str = "pub_med"
+    description: str = (
+        "A wrapper around PubMed. "
+        "Useful for when you need to answer questions about medicine, health, "
+        "and biomedical topics "
+        "from biomedical literature, MEDLINE, life science journals, and online books. "
+        "Input should be a search query."
+    )
+    api_wrapper: PubMedAPIWrapper = Field(default_factory=PubMedAPIWrapper)  # type: ignore[arg-type]
+
+    def _run(
+        self,
+        query: str,
+        run_manager: Optional[CallbackManagerForToolRun] = None,
+    ) -> str:
+        """Use the PubMed tool."""
+        return self.api_wrapper.run(query)