mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-07-01 13:18:10 +00:00
bc248c2b9c
- Updated `create_json_crew.py` to require `crewai[tools]>=1.14.7`. - Enhanced `git.py` with improved repository initialization, including automatic initial commit creation and exclusion patterns for initial commits. - Modified `install_crew.py` to allow error handling during installation with an optional `raise_on_error` parameter. - Expanded `plus_api.py` to include methods for creating and updating crews from ZIP files. - Introduced a new `archive.py` for creating deployable ZIP archives of CrewAI projects, ensuring local artifacts are excluded. - Updated `run_crew.py` to manage JSON crew dependencies and run crews in the project's environment. - Enhanced deployment logic in `main.py` to handle ZIP uploads and improve user feedback during deployment processes. - Added tests for new functionalities and ensured existing tests reflect recent changes in behavior and requirements.
93 lines
2.6 KiB
Python
93 lines
2.6 KiB
Python
"""CrewAI CLI API client extensions."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from typing import Any, Literal, cast
|
|
from urllib.parse import urljoin
|
|
|
|
from crewai_core.plus_api import PlusAPI as _CorePlusAPI
|
|
import httpx
|
|
|
|
|
|
HttpMethod = Literal["GET", "POST", "PATCH", "DELETE"]
|
|
|
|
|
|
class PlusAPI(_CorePlusAPI):
|
|
"""CLI API client.
|
|
|
|
The ZIP deployment methods live here as well as in newer crewai-core
|
|
versions so editable CLI installs still work when an older crewai-core is
|
|
present in the runtime environment.
|
|
"""
|
|
|
|
def _make_multipart_request(
|
|
self,
|
|
method: HttpMethod,
|
|
endpoint: str,
|
|
*,
|
|
zip_file_path: str | Path,
|
|
data: dict[str, str] | None = None,
|
|
timeout: float | None = None,
|
|
verify: bool = True,
|
|
) -> httpx.Response:
|
|
url = urljoin(self.base_url, endpoint)
|
|
headers = dict(cast(dict[str, str], self.headers))
|
|
headers.pop("Content-Type", None)
|
|
path = Path(zip_file_path)
|
|
request_kwargs: dict[str, Any] = {"headers": headers}
|
|
if data is not None:
|
|
request_kwargs["data"] = data
|
|
if timeout is not None:
|
|
request_kwargs["timeout"] = timeout
|
|
|
|
with (
|
|
path.open("rb") as file_handle,
|
|
httpx.Client(trust_env=False, verify=verify) as client,
|
|
):
|
|
files = {
|
|
"zip_file": (path.name, file_handle, "application/zip"),
|
|
}
|
|
return client.request(method, url, files=files, **request_kwargs)
|
|
|
|
def create_crew_from_zip(
|
|
self,
|
|
zip_file_path: str | Path,
|
|
*,
|
|
name: str | None = None,
|
|
env: dict[str, str] | None = None,
|
|
) -> httpx.Response:
|
|
data: dict[str, str] = {}
|
|
if name:
|
|
data["name"] = name
|
|
if env:
|
|
data.update({f"env[{key}]": value for key, value in env.items()})
|
|
return self._make_multipart_request(
|
|
"POST",
|
|
f"{self.CREWS_RESOURCE}/zip",
|
|
zip_file_path=zip_file_path,
|
|
data=data or None,
|
|
timeout=300,
|
|
)
|
|
|
|
def update_crew_from_zip(
|
|
self,
|
|
uuid: str,
|
|
zip_file_path: str | Path,
|
|
*,
|
|
env: dict[str, str] | None = None,
|
|
) -> httpx.Response:
|
|
data: dict[str, str] = {}
|
|
if env:
|
|
data.update({f"env[{key}]": value for key, value in env.items()})
|
|
return self._make_multipart_request(
|
|
"POST",
|
|
f"{self.CREWS_RESOURCE}/{uuid}/zip_update",
|
|
zip_file_path=zip_file_path,
|
|
data=data or None,
|
|
timeout=300,
|
|
)
|
|
|
|
|
|
__all__ = ["PlusAPI"]
|