Move off v1

src/crewai_tools/tools/exa_tools/exa_base_tool.py

@@ -1,36 +1,49 @@
 import os
 from typing import Type
-from pydantic.v1 import BaseModel, Field
+
+from pydantic import BaseModel, Field
+
 from crewai_tools.tools.base_tool import BaseTool
 
+
 class EXABaseToolToolSchema(BaseModel):
-	"""Input for EXABaseTool."""
-	search_query: str = Field(..., description="Mandatory search query you want to use to search the internet")
+    """Input for EXABaseTool."""
+
+    search_query: str = Field(
+        ..., description="Mandatory search query you want to use to search the internet"
+    )
+
 
 class EXABaseTool(BaseTool):
-	name: str = "Search the internet"
-	description: str = "A tool that can be used to search the internet from a search_query"
-	args_schema: Type[BaseModel] = EXABaseToolToolSchema
-	search_url: str = "https://api.exa.ai/search"
-	n_results: int = None
-	headers: dict = {
-			"accept": "application/json",
-			"content-type": "application/json",
-		}
+    name: str = "Search the internet"
+    description: str = (
+        "A tool that can be used to search the internet from a search_query"
+    )
+    args_schema: Type[BaseModel] = EXABaseToolToolSchema
+    search_url: str = "https://api.exa.ai/search"
+    n_results: int = None
+    headers: dict = {
+        "accept": "application/json",
+        "content-type": "application/json",
+    }
 
-	def _parse_results(self, results):
-		stirng = []
-		for result in results:
-			try:
-				stirng.append('\n'.join([
-						f"Title: {result['title']}",
-						f"Score: {result['score']}",
-						f"Url: {result['url']}",
-						f"ID: {result['id']}",
-						"---"
-				]))
-			except KeyError:
-				next
+    def _parse_results(self, results):
+        stirng = []
+        for result in results:
+            try:
+                stirng.append(
+                    "\n".join(
+                        [
+                            f"Title: {result['title']}",
+                            f"Score: {result['score']}",
+                            f"Url: {result['url']}",
+                            f"ID: {result['id']}",
+                            "---",
+                        ]
+                    )
+                )
+            except KeyError:
+                next
 
-		content = '\n'.join(stirng)
-		return f"\nSearch results: {content}\n"
+        content = "\n".join(stirng)
+        return f"\nSearch results: {content}\n"