diff --git a/lib/crewai-tools/src/crewai_tools/tools/nl2sql/nl2sql_tool.py b/lib/crewai-tools/src/crewai_tools/tools/nl2sql/nl2sql_tool.py index 84b4bd772..2566b7825 100644 --- a/lib/crewai-tools/src/crewai_tools/tools/nl2sql/nl2sql_tool.py +++ b/lib/crewai-tools/src/crewai_tools/tools/nl2sql/nl2sql_tool.py @@ -1,6 +1,6 @@ import logging import os -from typing import Any +from typing import Any, Self from crewai.tools import BaseTool from pydantic import BaseModel, Field, model_validator @@ -83,7 +83,7 @@ class NL2SQLTool(BaseTool): args_schema: type[BaseModel] = NL2SQLToolInput @model_validator(mode="after") - def _apply_env_override(self) -> "NL2SQLTool": + def _apply_env_override(self) -> Self: """Allow CREWAI_NL2SQL_ALLOW_DML=true to override allow_dml at runtime.""" if os.environ.get("CREWAI_NL2SQL_ALLOW_DML", "").strip().lower() == "true": if not self.allow_dml: diff --git a/lib/crewai-tools/tests/tools/test_nl2sql_security.py b/lib/crewai-tools/tests/tools/test_nl2sql_security.py index f13a7c8ea..91838c2ab 100644 --- a/lib/crewai-tools/tests/tools/test_nl2sql_security.py +++ b/lib/crewai-tools/tests/tools/test_nl2sql_security.py @@ -13,7 +13,6 @@ import pytest pytest.importorskip("sqlalchemy") from sqlalchemy import create_engine, text # noqa: E402 -from sqlalchemy.orm import sessionmaker # noqa: E402 from crewai_tools.tools.nl2sql.nl2sql_tool import NL2SQLTool # noqa: E402 diff --git a/lib/crewai/src/crewai/cli/cli.py b/lib/crewai/src/crewai/cli/cli.py index b0483d570..c40fe656f 100644 --- a/lib/crewai/src/crewai/cli/cli.py +++ b/lib/crewai/src/crewai/cli/cli.py @@ -609,7 +609,6 @@ def env() -> None: @env.command("view") def env_view() -> None: """View tracing-related environment variables.""" - import os from pathlib import Path from rich.console import Console @@ -738,7 +737,6 @@ def traces_disable() -> None: @traces.command("status") def traces_status() -> None: """Show current trace collection status.""" - import os from rich.console import Console from rich.panel import Panel diff --git a/lib/crewai/src/crewai/tasks/llm_guardrail.py b/lib/crewai/src/crewai/tasks/llm_guardrail.py index 3cbd20c65..754596ab7 100644 --- a/lib/crewai/src/crewai/tasks/llm_guardrail.py +++ b/lib/crewai/src/crewai/tasks/llm_guardrail.py @@ -1,6 +1,6 @@ import asyncio -import concurrent.futures from collections.abc import Coroutine +import concurrent.futures import contextvars import inspect from typing import Any