Merge pull request #5 from UABGH-Emerging-Technologies/feature/azure-ready

Feature/azure ready
merged on incoming

Author: rcg-uab

UMLBot/__init__.py

@@ -1,2 +1 @@
 """UMLBot core package."""
-

UMLBot/config/config.py

@@ -72,28 +72,18 @@ class UMLBotConfig:
     DEFAULT_DIAGRAM_TYPE = "Use Case"
 
     FALLBACK_PLANTUML_TEMPLATE = "@startuml\n' {diagram_type} diagram\n' {description}\n@enduml"
-    FALLBACK_MINDMAP_TEMPLATE = (
-        "@startmindmap\n* {diagram_type}\n** {description}\n@endmindmap"
-    )
-    FALLBACK_SALT_TEMPLATE = (
-        "@startsalt\n{diagram_type}\n{description}\n@endsalt"
-    )
-    FALLBACK_GANTT_TEMPLATE = (
-        "@startgantt\n[Task] lasts 1 day\n@endgantt"
-    )
-    FALLBACK_ERD_TEMPLATE = (
-        "@startuml\nentity \"Entity\" as E {\n  *id : int\n}\n@enduml"
-    )
-    FALLBACK_JSON_TEMPLATE = (
-        "@startjson\n{\n  \"sample\": true\n}\n@endjson"
-    )
+    FALLBACK_MINDMAP_TEMPLATE = "@startmindmap\n* {diagram_type}\n** {description}\n@endmindmap"
+    FALLBACK_SALT_TEMPLATE = "@startsalt\n{diagram_type}\n{description}\n@endsalt"
+    FALLBACK_GANTT_TEMPLATE = "@startgantt\n[Task] lasts 1 day\n@endgantt"
+    FALLBACK_ERD_TEMPLATE = '@startuml\nentity "Entity" as E {\n  *id : int\n}\n@enduml'
+    FALLBACK_JSON_TEMPLATE = '@startjson\n{\n  "sample": true\n}\n@endjson'
     FALLBACK_C4_TEMPLATE = (
         "@startuml\n"
         "!include https://raw.githubusercontent.com/plantuml-stdlib/C4-PlantUML/master/C4_Context.puml\n"
         "title {diagram_type}\n"
-        "Person(user, \"User\", \"\")\n"
-        "System(system, \"System\", \"{description}\")\n"
-        "Rel(user, system, \"Uses\")\n"
+        'Person(user, "User", "")\n'
+        'System(system, "System", "{description}")\n'
+        'Rel(user, system, "Uses")\n'
         "@enduml"
     )
     DIAGRAM_SUCCESS_MSG = "Diagram generated successfully using LLM."

UMLBot/diagram_handlers/__init__.py

@@ -1,12 +1,12 @@
 """Diagram handler package exports."""
 
-from UMLBot.diagram_handlers.uml_draft_handler import UMLDraftHandler, UMLRetryManager
-from UMLBot.diagram_handlers.mindmap_draft_handler import MindmapDraftHandler
-from UMLBot.diagram_handlers.ui_mockup_draft_handler import UIMockupDraftHandler
-from UMLBot.diagram_handlers.gantt_draft_handler import GanttDraftHandler
+from UMLBot.diagram_handlers.c4_draft_handler import C4DraftHandler
 from UMLBot.diagram_handlers.er_draft_handler import ERDraftHandler
+from UMLBot.diagram_handlers.gantt_draft_handler import GanttDraftHandler
 from UMLBot.diagram_handlers.json_draft_handler import JsonDraftHandler
-from UMLBot.diagram_handlers.c4_draft_handler import C4DraftHandler
+from UMLBot.diagram_handlers.mindmap_draft_handler import MindmapDraftHandler
+from UMLBot.diagram_handlers.ui_mockup_draft_handler import UIMockupDraftHandler
+from UMLBot.diagram_handlers.uml_draft_handler import UMLDraftHandler, UMLRetryManager
 
 __all__ = [
     "C4DraftHandler",

UMLBot/diagram_handlers/gantt_draft_handler.py

@@ -20,6 +20,4 @@ class GanttDraftHandler(UMLDraftHandler):
     def __init__(self, config: Optional[UMLBotConfig] = None):
         """Initialize the handler with optional configuration overrides."""
         super().__init__(config=config)
-        self.prompty_path = (
-            Path(__file__).resolve().parents[2] / "assets" / "gantt_chart.prompty"
-        )
+        self.prompty_path = Path(__file__).resolve().parents[2] / "assets" / "gantt_chart.prompty"

UMLBot/diagram_handlers/mindmap_draft_handler.py

@@ -20,6 +20,4 @@ class MindmapDraftHandler(UMLDraftHandler):
     def __init__(self, config: Optional[UMLBotConfig] = None):
         """Initialize the handler with optional configuration overrides."""
         super().__init__(config=config)
-        self.prompty_path = (
-            Path(__file__).resolve().parents[2] / "assets" / "mindmap.prompty"
-        )
+        self.prompty_path = Path(__file__).resolve().parents[2] / "assets" / "mindmap.prompty"

UMLBot/diagram_handlers/ui_mockup_draft_handler.py

@@ -20,6 +20,4 @@ class UIMockupDraftHandler(UMLDraftHandler):
     def __init__(self, config: Optional[UMLBotConfig] = None):
         """Initialize the handler with optional configuration overrides."""
         super().__init__(config=config)
-        self.prompty_path = (
-            Path(__file__).resolve().parents[2] / "assets" / "ui_mockup.prompty"
-        )
+        self.prompty_path = Path(__file__).resolve().parents[2] / "assets" / "ui_mockup.prompty"

UMLBot/diagram_handlers/uml_draft_handler.py

@@ -45,7 +45,7 @@ def error_context(self) -> str:
 
 import logging
 from pathlib import Path
-from typing import Optional, Any
+from typing import Any, Optional
 
 
 def escape_curly_braces(val: Optional[str]) -> Optional[str]:
@@ -62,6 +62,7 @@ def escape_curly_braces(val: Optional[str]) -> Optional[str]:
 
 
 from aiweb_common.WorkflowHandler import WorkflowHandler
+
 from UMLBot.config.config import UMLBotConfig
 
 
@@ -76,9 +77,7 @@ class UMLDraftHandler(WorkflowHandler):
     def __init__(self, config: Optional[UMLBotConfig] = None):
         """Initialize the handler with optional configuration overrides."""
         super().__init__()
-        self.prompty_path = (
-            Path(__file__).resolve().parents[2] / "assets" / "uml_diagram.prompty"
-        )
+        self.prompty_path = Path(__file__).resolve().parents[2] / "assets" / "uml_diagram.prompty"
         self.config = config or UMLBotConfig()
 
     def _validate_prompt_template(self, template: str) -> None:

UMLBot/exceptions.py

@@ -1,4 +1,5 @@
 """Custom exceptions for UMLBot modules."""
 
+
 class LLMError(Exception):
     """Raised when an LLM invocation fails."""

UMLBot/llm_interface.py

@@ -51,4 +51,3 @@ async def invoke_async(self, prompt: str) -> str:
 
 
 __all__ = ["LangchainLLMAdapter", "LLMInterface"]
-

UMLBot/services/diagram_service.py

@@ -29,6 +29,7 @@
 @dataclass
 class DiagramGenerationResult:
     """Result container for a UML generation request."""
+
     plantuml_code: str
     pil_image: Image.Image | None
     status_message: str