andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: additional linting fixes from pre-commit hooks

Browse Source 
- Fixed import error handling in bedrock invoke agent tool
- Fixed timeout in contextual AI rerank tool
This commit is contained in:
Greyson LaLonde
2025-09-19 00:09:44 -04:00
parent c5c07331bb
commit 080a7d753f
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
packages/tools/src/crewai_tools/aws/bedrock/agents/invoke_agent_tool.py
View File

@@ -96,7 +96,9 @@ class BedrockInvokeAgentTool(BaseTool):
import boto3 import boto3
from botocore.exceptions import ClientError from botocore.exceptions import ClientError
except ImportError as e: except ImportError as e:
raise ImportError("`boto3` package not found, please run `uv add boto3`") from e raise ImportError(
"`boto3` package not found, please run `uv add boto3`"
) from e
try: try:
# Initialize the Bedrock Agent Runtime client # Initialize the Bedrock Agent Runtime client

4
packages/tools/src/crewai_tools/tools/contextualai_rerank_tool/contextual_rerank_tool.py
View File

@@ -66,7 +66,9 @@ class ContextualAIRerankTool(BaseTool):
payload["metadata"] = metadata payload["metadata"] = metadata
rerank_url = f"{base_url}/rerank" rerank_url = f"{base_url}/rerank"
result = requests.post(rerank_url, json=payload, headers=headers, timeout=30) result = requests.post(
rerank_url, json=payload, headers=headers, timeout=30
)
if result.status_code != 200: if result.status_code != 200:
raise RuntimeError( raise RuntimeError(