From 56edf1f95f8e0b1a2421ae1caf73b7925a6682e3 Mon Sep 17 00:00:00 2001
From: Mani <mani.srinivasan@tavily.com>
Date: Mon, 6 Jul 2026 17:20:59 -0400
Subject: [PATCH] Added client name header (#6413)

- added client_name header to the 4 tavily tools to classify incoming requests as 'crewai' requests.

- This is for internal analysis

Co-authored-by: Lorenze Jay <63378463+lorenzejay@users.noreply.github.com>
---
 .../tools/tavily_extractor_tool/tavily_extractor_tool.py    | 6 ++++--
 .../tavily_get_research_tool/tavily_get_research_tool.py    | 6 ++++--
 .../tools/tavily_research_tool/tavily_research_tool.py      | 6 ++++--
 .../tools/tavily_search_tool/tavily_search_tool.py          | 6 ++++--
 4 files changed, 16 insertions(+), 8 deletions(-)

diff --git a/lib/crewai-tools/src/crewai_tools/tools/tavily_extractor_tool/tavily_extractor_tool.py b/lib/crewai-tools/src/crewai_tools/tools/tavily_extractor_tool/tavily_extractor_tool.py
index a89689e0a..57f3f89aa 100644
--- a/lib/crewai-tools/src/crewai_tools/tools/tavily_extractor_tool/tavily_extractor_tool.py
+++ b/lib/crewai-tools/src/crewai_tools/tools/tavily_extractor_tool/tavily_extractor_tool.py
@@ -87,9 +87,11 @@ class TavilyExtractorTool(BaseTool):
         """
         super().__init__(**kwargs)
         if TAVILY_AVAILABLE:
-            self.client = TavilyClient(api_key=self.api_key, proxies=self.proxies)
+            self.client = TavilyClient(
+                api_key=self.api_key, proxies=self.proxies, client_name="crewai"
+            )
             self.async_client = AsyncTavilyClient(
-                api_key=self.api_key, proxies=self.proxies
+                api_key=self.api_key, proxies=self.proxies, client_name="crewai"
             )
         else:
             try:
diff --git a/lib/crewai-tools/src/crewai_tools/tools/tavily_get_research_tool/tavily_get_research_tool.py b/lib/crewai-tools/src/crewai_tools/tools/tavily_get_research_tool/tavily_get_research_tool.py
index c3d6787b3..1b9139137 100644
--- a/lib/crewai-tools/src/crewai_tools/tools/tavily_get_research_tool/tavily_get_research_tool.py
+++ b/lib/crewai-tools/src/crewai_tools/tools/tavily_get_research_tool/tavily_get_research_tool.py
@@ -54,8 +54,10 @@ class TavilyGetResearchTool(BaseTool):
         super().__init__(**kwargs)
         if TAVILY_AVAILABLE:
             api_key = os.getenv("TAVILY_API_KEY")
-            self._client = TavilyClient(api_key=api_key)
-            self._async_client = AsyncTavilyClient(api_key=api_key)
+            self._client = TavilyClient(api_key=api_key, client_name="crewai")
+            self._async_client = AsyncTavilyClient(
+                api_key=api_key, client_name="crewai"
+            )
         else:
             try:
                 import subprocess
diff --git a/lib/crewai-tools/src/crewai_tools/tools/tavily_research_tool/tavily_research_tool.py b/lib/crewai-tools/src/crewai_tools/tools/tavily_research_tool/tavily_research_tool.py
index 084fefdf1..b4bb8e0af 100644
--- a/lib/crewai-tools/src/crewai_tools/tools/tavily_research_tool/tavily_research_tool.py
+++ b/lib/crewai-tools/src/crewai_tools/tools/tavily_research_tool/tavily_research_tool.py
@@ -90,8 +90,10 @@ class TavilyResearchTool(BaseTool):
         super().__init__(**kwargs)
         if TAVILY_AVAILABLE:
             api_key = os.getenv("TAVILY_API_KEY")
-            self._client = TavilyClient(api_key=api_key)
-            self._async_client = AsyncTavilyClient(api_key=api_key)
+            self._client = TavilyClient(api_key=api_key, client_name="crewai")
+            self._async_client = AsyncTavilyClient(
+                api_key=api_key, client_name="crewai"
+            )
         else:
             try:
                 import subprocess
diff --git a/lib/crewai-tools/src/crewai_tools/tools/tavily_search_tool/tavily_search_tool.py b/lib/crewai-tools/src/crewai_tools/tools/tavily_search_tool/tavily_search_tool.py
index ef8d7412d..1b5ebfe79 100644
--- a/lib/crewai-tools/src/crewai_tools/tools/tavily_search_tool/tavily_search_tool.py
+++ b/lib/crewai-tools/src/crewai_tools/tools/tavily_search_tool/tavily_search_tool.py
@@ -115,9 +115,11 @@ class TavilySearchTool(BaseTool):
     def __init__(self, **kwargs: Any):
         super().__init__(**kwargs)
         if TAVILY_AVAILABLE:
-            self.client = TavilyClient(api_key=self.api_key, proxies=self.proxies)
+            self.client = TavilyClient(
+                api_key=self.api_key, proxies=self.proxies, client_name="crewai"
+            )
             self.async_client = AsyncTavilyClient(
-                api_key=self.api_key, proxies=self.proxies
+                api_key=self.api_key, proxies=self.proxies, client_name="crewai"
             )
         else:
             try: