diff --git a/examples/deployment/analysis_flow.py b/examples/deployment/analysis_flow.py index 194f16e04..30827fc11 100644 --- a/examples/deployment/analysis_flow.py +++ b/examples/deployment/analysis_flow.py @@ -1,7 +1,8 @@ from pydantic import BaseModel from crewai import Agent, Crew, Task -from crewai.flow import Flow, start, listen +from crewai.flow import Flow, listen, start + class AnalysisState(BaseModel): topic: str = "" diff --git a/src/crewai/deployment/cli.py b/src/crewai/deployment/cli.py index 24177a9e3..60ed9a1a5 100644 --- a/src/crewai/deployment/cli.py +++ b/src/crewai/deployment/cli.py @@ -1,4 +1,5 @@ import os + import click from rich.console import Console diff --git a/src/crewai/deployment/config.py b/src/crewai/deployment/config.py index 3b67056d5..f473f629b 100644 --- a/src/crewai/deployment/config.py +++ b/src/crewai/deployment/config.py @@ -1,7 +1,9 @@ from pathlib import Path from typing import Any, Dict, List, Optional + import yaml + class Config: """ Configuration for CrewAI deployments. diff --git a/src/crewai/deployment/docker/container.py b/src/crewai/deployment/docker/container.py index f8855e888..63a71ede2 100644 --- a/src/crewai/deployment/docker/container.py +++ b/src/crewai/deployment/docker/container.py @@ -4,6 +4,7 @@ import subprocess from pathlib import Path from typing import Dict, List, Optional + class DockerContainer: """ Manages Docker containers for CrewAI deployments. diff --git a/src/crewai/deployment/main.py b/src/crewai/deployment/main.py index ffed76118..eb0949759 100644 --- a/src/crewai/deployment/main.py +++ b/src/crewai/deployment/main.py @@ -1,5 +1,5 @@ -import os import json +import os import shutil from pathlib import Path from typing import Any, Dict, List, Optional @@ -7,6 +7,7 @@ from typing import Any, Dict, List, Optional from crewai.deployment.config import Config from crewai.deployment.docker.container import DockerContainer + class Deployment: """ Handles the deployment of CrewAI crews and flows. diff --git a/tests/deployment/test_deployment.py b/tests/deployment/test_deployment.py index 0ddaec239..41a335c68 100644 --- a/tests/deployment/test_deployment.py +++ b/tests/deployment/test_deployment.py @@ -1,12 +1,13 @@ import os import tempfile import unittest -from unittest import mock from pathlib import Path +from unittest import mock from crewai.deployment.config import Config from crewai.deployment.main import Deployment + class TestDeployment(unittest.TestCase): def setUp(self): self.temp_dir = tempfile.TemporaryDirectory()