mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 00:28:31 +00:00
fix: resolve circular import in CLI authentication module
- Move ToolCommand import to be local inside _poll_for_token method - Update test mock to patch ToolCommand at correct location - Resolves Python 3.11 test collection failure in CI Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
@@ -5,8 +5,6 @@ from typing import Any, Dict
|
|||||||
import requests
|
import requests
|
||||||
from rich.console import Console
|
from rich.console import Console
|
||||||
|
|
||||||
from crewai.cli.tools.main import ToolCommand
|
|
||||||
|
|
||||||
from .constants import AUTH0_AUDIENCE, AUTH0_CLIENT_ID, AUTH0_DOMAIN
|
from .constants import AUTH0_AUDIENCE, AUTH0_CLIENT_ID, AUTH0_DOMAIN
|
||||||
from .utils import TokenManager, validate_token
|
from .utils import TokenManager, validate_token
|
||||||
|
|
||||||
@@ -67,6 +65,7 @@ class AuthenticationCommand:
|
|||||||
self.token_manager.save_tokens(token_data["access_token"], expires_in)
|
self.token_manager.save_tokens(token_data["access_token"], expires_in)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
from crewai.cli.tools.main import ToolCommand
|
||||||
ToolCommand().login()
|
ToolCommand().login()
|
||||||
except Exception:
|
except Exception:
|
||||||
console.print(
|
console.print(
|
||||||
|
|||||||
@@ -44,7 +44,7 @@ class TestAuthenticationCommand(unittest.TestCase):
|
|||||||
mock_print.assert_any_call("2. Enter the following code: ", "ABCDEF")
|
mock_print.assert_any_call("2. Enter the following code: ", "ABCDEF")
|
||||||
mock_open.assert_called_once_with("https://example.com")
|
mock_open.assert_called_once_with("https://example.com")
|
||||||
|
|
||||||
@patch("crewai.cli.authentication.main.ToolCommand")
|
@patch("crewai.cli.tools.main.ToolCommand")
|
||||||
@patch("crewai.cli.authentication.main.requests.post")
|
@patch("crewai.cli.authentication.main.requests.post")
|
||||||
@patch("crewai.cli.authentication.main.validate_token")
|
@patch("crewai.cli.authentication.main.validate_token")
|
||||||
@patch("crewai.cli.authentication.main.console.print")
|
@patch("crewai.cli.authentication.main.console.print")
|
||||||
|
|||||||
Reference in New Issue
Block a user