Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 6 additions & 6 deletions src/creative_agent/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,14 +14,14 @@
from mcp.types import TextContent
from pydantic import AnyUrl

from .data.standard_formats import (
from creative_agent.data.standard_formats import (
AGENT_CAPABILITIES,
AGENT_NAME,
AGENT_URL,
filter_formats,
get_format_by_id,
)
from .schemas import (
from creative_agent.schemas import (
ListCreativeFormatsResponse,
PreviewCreativeRequest,
)
Expand Down Expand Up @@ -223,7 +223,7 @@ def _handle_single_preview(
output_format: str,
) -> ToolResult:
"""Handle a single preview request."""
from .schemas.manifest import PreviewInput
from creative_agent.schemas.manifest import PreviewInput

# Parse inputs if provided
inputs_obj: list[PreviewInput] | None = None
Expand Down Expand Up @@ -268,7 +268,7 @@ def _handle_single_preview(
)

# Validate manifest assets
from .validation import validate_manifest_assets
from creative_agent.validation import validate_manifest_assets

validation_errors = validate_manifest_assets(
request.creative_manifest,
Expand All @@ -294,7 +294,7 @@ def _handle_single_preview(
]

# Generate previews for each input set
from .storage import generate_preview_html, upload_preview_html
from creative_agent.storage import generate_preview_html, upload_preview_html

previews = []
for input_set in request.inputs:
Expand Down Expand Up @@ -720,7 +720,7 @@ def build_creative(
output_manifest = json.loads(manifest_json)

# Validate against output format
from .validation import validate_manifest_assets
from creative_agent.validation import validate_manifest_assets

validation_errors = validate_manifest_assets(
output_manifest,
Expand Down