codemie-sdk-python 0.1.292__tar.gz → 0.1.294__tar.gz

{codemie_sdk_python-0.1.292 → codemie_sdk_python-0.1.294}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.292
+Version: 0.1.294
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com

{codemie_sdk_python-0.1.292 → codemie_sdk_python-0.1.294}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codemie-sdk-python"
-version = "0.1.292"
+version = "0.1.294"
 description = "CodeMie SDK for Python"
 authors = [
     "Vadym Vlasenko <vadym_vlasenko@epam.com>",

{codemie_sdk_python-0.1.292 → codemie_sdk_python-0.1.294}/src/codemie_sdk/__init__.py

@@ -100,6 +100,13 @@ from .services.vendor_knowledgebase import VendorKnowledgeBaseService
 from .services.vendor_guardrail import VendorGuardrailService
 from .services.codemie_guardrails import CodemieGuardrailService
 from .services.analytics import AnalyticsService
+from .models.mermaid import (
+    MermaidDiagramRequest,
+    MermaidDiagramResponse,
+    ContentType,
+    ResponseType,
+)
+from .services.mermaid import MermaidService
 
 __version__ = "0.2.12"
 __all__ = [
@@ -174,4 +181,9 @@ __all__ = [
     "AnalyticsQueryParams",
     "PaginatedAnalyticsQueryParams",
     "AnalyticsService",
+    "MermaidDiagramRequest",
+    "MermaidDiagramResponse",
+    "ContentType",
+    "ResponseType",
+    "MermaidService",
 ]

{codemie_sdk_python-0.1.292 → codemie_sdk_python-0.1.294}/src/codemie_sdk/client/client.py

@@ -10,6 +10,7 @@ from ..services.conversation import ConversationService
 from ..services.datasource import DatasourceService
 from ..services.llm import LLMService
 from ..services.integration import IntegrationService
+from ..services.mermaid import MermaidService
 from ..services.task import TaskService
 from ..services.user import UserService
 from ..services.workflow import WorkflowService
@@ -80,6 +81,9 @@ class CodeMieClient:
             self._api_domain, self._token, verify_ssl=verify_ssl
         )
         self.llms = LLMService(self._api_domain, self._token, verify_ssl=verify_ssl)
+        self.mermaid = MermaidService(
+            self._api_domain, self._token, verify_ssl=verify_ssl
+        )
         self.integrations = IntegrationService(
             self._api_domain, self._token, verify_ssl=verify_ssl
         )
@@ -150,6 +154,9 @@ class CodeMieClient:
         self.llms = LLMService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.mermaid = MermaidService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
         self.integrations = IntegrationService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )

codemie_sdk_python-0.1.294/src/codemie_sdk/models/mermaid.py

@@ -0,0 +1,25 @@
+"""Models for mermaid diagram operations."""
+
+from typing import Literal
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class MermaidDiagramRequest(BaseModel):
+    """Request model for mermaid diagram generation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    code: str = Field(..., description="Mermaid diagram code")
+
+
+class MermaidDiagramResponse(BaseModel):
+    """Response model for mermaid diagram generation (FILE mode)."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    file_url: str = Field(..., description="URL to the generated diagram file")
+
+
+# Type aliases for clarity
+ContentType = Literal["svg", "png"]
+ResponseType = Literal["file", "raw"]

codemie_sdk_python-0.1.294/src/codemie_sdk/services/mermaid.py

@@ -0,0 +1,67 @@
+"""Mermaid diagram service implementation."""
+
+from typing import Union
+import requests
+
+from ..models.mermaid import (
+    MermaidDiagramRequest,
+    MermaidDiagramResponse,
+    ContentType,
+    ResponseType,
+)
+from ..utils import ApiRequestHandler
+
+
+class MermaidService:
+    """Service for generating mermaid diagrams."""
+
+    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+        """Initialize the mermaid service.
+
+        Args:
+            api_domain: Base URL for the CodeMie API
+            token: Authentication token
+            verify_ssl: Whether to verify SSL certificates
+        """
+        self._api = ApiRequestHandler(api_domain, token, verify_ssl)
+
+    def generate_diagram(
+        self,
+        code: str,
+        content_type: ContentType = "svg",
+        response_type: ResponseType = "file",
+    ) -> Union[MermaidDiagramResponse, bytes]:
+        """Generate a mermaid diagram from mermaid code.
+
+        Args:
+            code: Mermaid diagram code (including theme directives if needed)
+            content_type: Output format - "svg" or "png" (default: "svg")
+            response_type: Response format - "file" (returns URL) or "raw" (returns binary data) (default: "file")
+
+        Returns:
+            MermaidDiagramResponse with file_url if response_type="file",
+            or bytes with raw diagram data if response_type="raw"
+
+        Raises:
+            ApiError: If diagram generation fails or invalid mermaid code
+        """
+        request = MermaidDiagramRequest(code=code)
+
+        # Build endpoint with query parameters
+        endpoint = f"/v1/files/diagram/mermaid?content_type={content_type}&response_type={response_type}"
+
+        if response_type == "file":
+            return self._api.post(
+                endpoint,
+                response_model=MermaidDiagramResponse,
+                json_data=request.model_dump(),
+                wrap_response=False,
+            )
+        else:
+            # For raw mode, return binary content
+            response = self._api.post(
+                endpoint,
+                response_model=requests.Response,
+                json_data=request.model_dump(),
+            )
+            return response.content