chore: show message to user if building block requested is not supported yet

Author: runemalm

src/codius/domain/model/intents/intent_type.py

@@ -1,5 +1,8 @@
 from enum import Enum
 
+from codius.infrastructure.services.code_scanner.model.building_block_type import \
+    BuildingBlockType
+
 
 class IntentType(str, Enum):
     ADD_AGGREGATE = "add_aggregate"
@@ -22,3 +25,17 @@ class IntentType(str, Enum):
     REMOVE_REPOSITORY_METHOD = "remove_repository_method"
 
     UNSURE = "unsure"
+
+    @property
+    def building_block(self) -> BuildingBlockType:
+        if self.name.startswith("ADD_AGGREGATE") or self.name.startswith(
+            "REMOVE_AGGREGATE"):
+            return BuildingBlockType.AGGREGATE_ROOT
+        elif self.name.startswith("ADD_VALUE_OBJECT") or self.name.startswith(
+            "REMOVE_VALUE_OBJECT"):
+            return BuildingBlockType.VALUE_OBJECT
+        elif self.name.startswith("ADD_REPOSITORY") or self.name.startswith(
+            "REMOVE_REPOSITORY"):
+            return BuildingBlockType.REPOSITORY
+        else:
+            raise NotImplementedError(f"Building block not mapped for intent {self}")

src/codius/domain/model/prompts/distill_intent_prompt.py

@@ -22,6 +22,8 @@
 from codius.domain.model.intents.value_object.add_value_object_property_intent import AddValueObjectPropertyIntent
 from codius.domain.model.intents.value_object.remove_value_object_property_intent import RemoveValueObjectPropertyIntent
 from codius.domain.model.intents.repository.add_repository_intent import AddRepositoryIntent
+from codius.infrastructure.services.code_scanner.model.building_block_type import \
+    BuildingBlockType
 
 
 @dataclass(frozen=True)
@@ -49,6 +51,14 @@ def as_prompt(self) -> str:
             RemoveRepositoryMethodIntent,
         ]
 
+        all_blocks = {bb.value for bb in BuildingBlockType}
+        supported_blocks = {intent.building_block.value for intent in IntentType if
+                            intent != IntentType.UNSURE}
+        unsupported_blocks = all_blocks - supported_blocks
+
+        supported_blocks_text = "\n".join(f"- {b}" for b in sorted(supported_blocks))
+        unsupported_blocks_text = "\n".join(f"- {b}" for b in sorted(unsupported_blocks))
+
         example_blocks = "\n".join(
             f"### {cls.intent.value}\n```json\n{cls.to_example_json()}\n```"
             for cls in example_intents
@@ -88,6 +98,29 @@ def as_prompt(self) -> str:
 **Database Providers**:
 {database_text}
 
+---
+
+### Known DDD Building Blocks
+
+These are the known building blocks in Domain-Driven Design:
+
+**Supported:**
+{supported_blocks_text}
+
+**Not yet supported:**
+{unsupported_blocks_text}
+
+If the user refers to a known DDD building block that is **not yet supported**, return:
+
+```json
+{{ "intent": "unsupported", "building_block": "<block_name>" }}
+```
+
+If the user's intent is unclear or unsupported, respond only with:
+
+```json
+{{ "intent": "unsure" }}
+
 ### Instructions
 
 - Break complex modeling instructions into small, **granular intents**.
@@ -109,9 +142,4 @@ def as_prompt(self) -> str:
 ### Examples:
 
 {example_blocks}
-
-If the user's intent is unclear or unsupported, respond only with:
-
-```json
-{{ "intent": "unsure" }}
 """

src/codius/graph/nodes/handle_unsupported_intent.py

@@ -0,0 +1,34 @@
+from typing import cast
+
+
+def handle_unsupported_intent(state: dict) -> dict:
+    from codius.domain.model.intents.intent_type import IntentType
+
+    unsupported_blocks = {
+        intent.get("building_block")
+        for intent in state.get("intent", [])
+        if intent.get("intent") == "unsupported"
+    }
+    block_list = ", ".join(sorted(unsupported_blocks))
+
+    # Dynamically get supported building blocks
+    supported_blocks = {
+        intent.building_block.value
+        for intent in IntentType
+        if intent != IntentType.UNSURE
+    }
+    supported_blocks_list = "\n".join(f"- {b}" for b in sorted(supported_blocks))
+
+    state["final_output"] = (
+        f"⚠️ The assistant understood your request, but the following building block(s) aren't supported yet:\n"
+        f"- {block_list}\n\n"
+        "We're shipping updates frequently, so it might already be available in a newer version.\n\n"
+        "👉 Try updating Codius:\n"
+        "```bash\npip install --upgrade codius\n```\n"
+        "Then rerun your request.\n\n"
+        "In the meantime, you can try working with one of the currently supported building blocks:\n"
+        f"{supported_blocks_list}\n\n"
+        "💡 Need help or want to request this feature? Let us know on GitHub!"
+    )
+
+    return state

src/codius/graph/routers/intent_router.py

@@ -16,6 +16,8 @@ def route_by_intent(state: dict) -> str:
 
             if intent_type == "error":
                 return "error"
+            elif intent_type == "unsupported":
+                return "unsupported"
             elif intent_type and intent_type not in {"none", "greeting", "unsure"}:
                 return "valid"
 

src/codius/infrastructure/services/graph_service.py

@@ -9,6 +9,7 @@
 from codius.graph.nodes.generate_code import generate_code
 from codius.graph.nodes.handle_intent_error import handle_intent_error
 from codius.graph.nodes.handle_unclear_intent import handle_unclear_intent
+from codius.graph.nodes.handle_unsupported_intent import handle_unsupported_intent
 from codius.graph.nodes.plan_changes import plan_changes
 from codius.graph.nodes.revise_intent import revise_intent
 from codius.graph.routers.approval_router import route_by_user_approval
@@ -69,6 +70,7 @@ def _build_graph(self):
         # Add nodes
         graph.add_node("DistillIntent", distill_intent)
         graph.add_node("HandleUnclearIntent", handle_unclear_intent)
+        graph.add_node("HandleUnsupportedIntent", handle_unsupported_intent)
         graph.add_node("HandleIntentError", handle_intent_error)
         graph.add_node("ExtractProjectMetadata", extract_project_metadata)
         graph.add_node("ExtractBuildingBlocks", extract_building_blocks)
@@ -84,7 +86,8 @@ def _build_graph(self):
         graph.add_conditional_edges("DistillIntent", route_by_intent, {
             "valid": "ExtractProjectMetadata",
             "unclear": "HandleUnclearIntent",
-            "error": "HandleIntentError"
+            "error": "HandleIntentError",
+            "unsupported": "HandleUnsupportedIntent",
         })
         graph.add_conditional_edges("Preview", route_by_user_approval, {
             "apply": "ApplyChanges",