feat: support latest SDK parameters

pyproject.toml

@@ -18,7 +18,7 @@ classifiers = [
 
 dependencies = [
   "langchain-core",
-  "linkup-sdk>=0.2.8",
+  "linkup-sdk>=0.11.0",
 ]
 
 [dependency-groups]

src/langchain_linkup/search_retriever.py

@@ -42,6 +42,17 @@ class LinkupSearchRetriever(BaseRetriever):
             The list of domains to exclude from the search.
         include_image: bool = False
             If set to True, image results will be included alongside text results.
+        max_results: Optional[int] = None
+            The maximum number of results to return.
+        include_inline_citations: Optional[bool] = None
+            If output_type is "sourcedAnswer", indicate whether the answer should include inline
+            citations.
+        include_sources: Optional[bool] = None
+            If output_type is "structured", indicate whether the answer should include sources.
+            This will modify the schema of the structured response.
+        timeout: Optional[float] = None
+            The timeout for the HTTP request, in seconds. If None, the request will have
+            no timeout.
 
     Instantiate:
         .. code-block:: python
@@ -137,6 +148,16 @@ def format_docs(docs):
     """The list of domains to exclude from the search."""
     include_image: bool = False
     """If set to True, image results will be included alongside text results."""
+    max_results: Optional[int] = None
+    """The maximum number of results to return."""
+    include_inline_citations: Optional[bool] = None
+    """If output_type is "sourcedAnswer", indicate whether the answer should include inline
+    citations."""
+    include_sources: Optional[bool] = None
+    """If output_type is "structured", indicate whether the answer should include sources. This
+    will modify the schema of the structured response."""
+    timeout: Optional[float] = None
+    """The timeout for the HTTP request, in seconds. If None, the request will have no timeout."""
 
     def _get_relevant_documents(
         self,
@@ -154,6 +175,10 @@ def _get_relevant_documents(
             include_domains=self.include_domains,
             exclude_domains=self.exclude_domains,
             include_images=self.include_image,
+            max_results=self.max_results,
+            include_inline_citations=self.include_inline_citations,
+            include_sources=self.include_sources,
+            timeout=self.timeout,
         )
 
         return [
@@ -183,6 +208,10 @@ async def _aget_relevant_documents(
             include_domains=self.include_domains,
             exclude_domains=self.exclude_domains,
             include_images=self.include_image,
+            max_results=self.max_results,
+            include_inline_citations=self.include_inline_citations,
+            include_sources=self.include_sources,
+            timeout=self.timeout,
         )
 
         return [

src/langchain_linkup/search_tool.py

@@ -51,6 +51,17 @@ class LinkupSearchTool(BaseTool):
             The list of domains to exclude from the search.
         include_image: bool = False
             If set to True, image results will be included alongside text results.
+        max_results: Optional[int] = None
+            The maximum number of results to return.
+        include_inline_citations: Optional[bool] = None
+            If output_type is "sourcedAnswer", indicate whether the answer should include inline
+            citations.
+        include_sources: Optional[bool] = None
+            If output_type is "structured", indicate whether the answer should include sources.
+            This will modify the schema of the structured response.
+        timeout: Optional[float] = None
+            The timeout for the HTTP request, in seconds. If None, the request will have
+            no timeout.
 
     Instantiate:
         .. code-block:: python
@@ -67,6 +78,10 @@ class LinkupSearchTool(BaseTool):
                 include_domains=None,
                 exclude_domains=None,
                 include_image=False,
+                max_results=None,
+                include_inline_citations=None,
+                include_sources=None,
+                timeout=None,
             )
 
     Usage:
@@ -143,6 +158,16 @@ class LinkupSearchTool(BaseTool):
     """The list of domains to exclude from the search."""
     include_image: bool = False
     """If set to True, image results will be included alongside text results."""
+    max_results: Optional[int] = None
+    """The maximum number of results to return."""
+    include_inline_citations: Optional[bool] = None
+    """If output_type is "sourcedAnswer", indicate whether the answer should include inline
+    citations."""
+    include_sources: Optional[bool] = None
+    """If output_type is "structured", indicate whether the answer should include sources. This
+    will modify the schema of the structured response."""
+    timeout: Optional[float] = None
+    """The timeout for the HTTP request, in seconds. If None, the request will have no timeout."""
 
     # Fields used by the agent to describe how to use the tool under the hood
     name: str = "linkup"
@@ -169,6 +194,10 @@ def _run(
             include_domains=self.include_domains,
             exclude_domains=self.exclude_domains,
             include_images=self.include_image,
+            max_results=self.max_results,
+            include_inline_citations=self.include_inline_citations,
+            include_sources=self.include_sources,
+            timeout=self.timeout,
         )
 
     async def _arun(
@@ -187,4 +216,8 @@ async def _arun(
             include_domains=self.include_domains,
             exclude_domains=self.exclude_domains,
             include_images=self.include_image,
+            max_results=self.max_results,
+            include_inline_citations=self.include_inline_citations,
+            include_sources=self.include_sources,
+            timeout=self.timeout,
         )

tests/unit_tests/test_search_retriever.py

@@ -11,7 +11,7 @@
 
 def test_get_relevant_document(mocker: MockerFixture, linkup_api_key: str) -> None:
     mocker.patch(
-        "linkup.client.LinkupClient._request",
+        "linkup.LinkupClient._request",
         return_value=Response(
             status_code=200,
             content=b"""
@@ -41,7 +41,7 @@ def test_get_relevant_document(mocker: MockerFixture, linkup_api_key: str) -> No
 @pytest.mark.asyncio
 async def test_aget_relevant_documents(mocker: MockerFixture, linkup_api_key: str) -> None:
     mocker.patch(
-        "linkup.client.LinkupClient._async_request",
+        "linkup.LinkupClient._async_request",
         return_value=Response(
             status_code=200,
             content=b"""