feat: Add support for Protect project creation (#7)

pyproject.toml

@@ -8,7 +8,7 @@ packages = [{include = "galileo_protect", from = "src"}]
 
 [tool.poetry.dependencies]
 python = "^3.8.1,<3.13"
-galileo-core = "^0.9.3"
+galileo-core = "^0.10.0"
 
 
 [tool.poetry.group.test.dependencies]

src/galileo_protect/__init__.py

@@ -1,5 +1,6 @@
 # flake8: noqa: F401
 # ruff: noqa: F401
 from galileo_protect.invoke import invoke
+from galileo_protect.project import create_project
 
 __version__ = "0.1.0"

src/galileo_protect/project.py

@@ -0,0 +1,19 @@
+from typing import Optional
+
+from galileo_core.helpers.project import create_project as core_create_project
+from galileo_core.schemas.core.project import (
+    DEFAULT_PROJECT_NAME,
+    CreateProjectRequest,
+    ProjectResponse,
+    ProjectType,
+)
+
+from galileo_protect.helpers.config import ProtectConfig
+
+
+def create_project(name: str = DEFAULT_PROJECT_NAME, config: Optional[ProtectConfig] = None) -> ProjectResponse:
+    config = config or ProtectConfig.get()
+    project = core_create_project(CreateProjectRequest(name=name, type=ProjectType.protect), config)
+    config.project_id = project.id
+    config.write()
+    return project

tests/test_project.py

@@ -0,0 +1,23 @@
+from typing import Callable
+from uuid import uuid4
+
+from galileo_core.constants.routes import Routes as CoreRoutes
+from galileo_core.schemas.core.project import DEFAULT_PROJECT_NAME, ProjectType
+from requests_mock import POST
+
+from galileo_protect.project import create_project
+
+
+def test_create_project(set_validated_config: Callable, mock_request: Callable) -> None:
+    config = set_validated_config()
+    project_id = uuid4()
+    matcher = mock_request(
+        POST,
+        CoreRoutes.projects,
+        json={"id": str(project_id), "type": ProjectType.protect, "name": DEFAULT_PROJECT_NAME},
+    )
+    create_project(config=config)
+    assert matcher.called
+    # Verify that the project ID was set in the config.
+    assert config.project_id is not None
+    assert config.project_id == project_id