[Haik]: ckpt, swapped the HaikFix folder to be parallel to the agents folder bc ima start the full replacement process i think

backend/apps/HaikFix/Agent/shared_structs/Message/Message.py

@@ -0,0 +1,78 @@
+# Message.py
+
+from typing import List, Literal, Dict, Annotated, Union
+from pydantic import BaseModel, Field
+from datetime import datetime
+from uuid import uuid4
+from typeguard import typechecked
+
+from backend.apps.HaikFix.Agent.shared_structs.Message.agent_inputs import (
+    PromptBlock, TextPromptBlock, ImagePromptBlock, ImageSource, ContextPath, SkillMeta
+)
+from backend.apps.HaikFix.Agent.shared_structs.Message.agent_outputs import (
+    ToolCallContent, ToolResultContent
+)
+
+class Message(BaseModel):
+    id: str = Field(default_factory=lambda: uuid4().hex)
+    timestamp: datetime = Field(default_factory=datetime.now)
+    branch_id: str = "main"
+    hidden: bool = False
+
+
+PromptMsgDict = Dict[
+    Literal["type", "message"], 
+    Dict[
+        Literal["role", "content"], 
+        List[PromptBlock]
+        ]
+    ]
+
+class UserMessage(Message):
+    role: Literal["user"] = "user"
+    content: str
+    images: List[str] = Field(default_factory=list)       # base64 strings
+    image_media_types: List[str] = Field(default_factory=list)
+    context_paths: List[ContextPath] = Field(default_factory=list)
+    attached_skills: List[SkillMeta] = Field(default_factory=list)
+    forced_tools: List[str] = Field(default_factory=list)
+
+    @typechecked
+    def to_prompt(self) -> PromptMsgDict:
+        blocks: List[PromptBlock] = [TextPromptBlock(type="text", text=self.content)]
+        for data, media_type in zip[tuple[str, str]](self.images, self.image_media_types):
+            blocks.append(ImagePromptBlock(
+                type="image",
+                source=ImageSource(type="base64", media_type=media_type, data=data),
+            ))
+        return {
+            "type": self.role, 
+            "message": {
+                "role": self.role, 
+                "content": blocks
+            }
+        }
+
+class AssistantMessage(Message):
+    role: Literal["assistant"] = "assistant"
+    content: str
+
+
+class ToolCallMessage(Message):
+    role: Literal["tool_call"] = "tool_call"
+    content: ToolCallContent
+
+
+class ToolResultMessage(Message):
+    role: Literal["tool_result"] = "tool_result"
+    content: ToolResultContent
+
+
+class SystemMessage(Message):
+    role: Literal["system"] = "system"
+    content: str
+
+AnyMessage = Annotated[
+    Union[UserMessage, AssistantMessage, ToolCallMessage, ToolResultMessage, SystemMessage],
+    Field(discriminator="role"),
+]

backend/apps/HaikFix/Agent/shared_structs/Message/agent_inputs.py

@@ -0,0 +1,33 @@
+# sub_types.py
+
+from typing import Literal, Union, TypedDict
+from pydantic import BaseModel
+
+
+# --- Prompt content blocks (sent to SDK via to_prompt) ---
+
+class TextPromptBlock(TypedDict):
+    type: Literal["text"]
+    text: str
+
+class ImageSource(TypedDict):
+    type: Literal["base64"]
+    media_type: str
+    data: str
+
+class ImagePromptBlock(TypedDict):
+    type: Literal["image"]
+    source: "ImageSource"
+
+
+PromptBlock = Union[TextPromptBlock, ImagePromptBlock]
+
+# --- Attachments (user message only) ---
+
+class ContextPath(BaseModel):
+    path: str
+    type: Literal["file", "directory"]
+
+class SkillMeta(BaseModel):
+    id: str
+    name: str

backend/apps/HaikFix/Agent/shared_structs/Message/agent_outputs.py

@@ -0,0 +1,26 @@
+from typing import Dict, Any, TypedDict, List
+from typing_extensions import NotRequired
+from pydantic import BaseModel
+
+
+# --- Stored content types ---
+class ToolCallContent(BaseModel):
+    id: str
+    tool: str
+    input: Dict[str, Any]
+
+class ToolResultContent(BaseModel):
+    tool_use_id: str
+    text: str
+    is_error: bool = False
+
+
+# --- Tool response types ---
+
+class TextContent(TypedDict):
+    type: str
+    text: str
+
+class ToolResponse(TypedDict):
+    content: List[TextContent]
+    is_error: NotRequired[bool]