From 14d8b18f1daec3a2f673ca59300089d0e9a2a4ba Mon Sep 17 00:00:00 2001
From: Yiorgis Gozadinos <ggozadinos@gmail.com>
Date: Fri, 1 Aug 2025 10:13:46 +0200
Subject: [PATCH] Wrap code files in md code blocks

---
 src/haiku/rag/reader.py | 36 +++++++++++++++++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

diff --git a/src/haiku/rag/reader.py b/src/haiku/rag/reader.py
index 9e7f7d42..0e6608c6 100644
--- a/src/haiku/rag/reader.py
+++ b/src/haiku/rag/reader.py
@@ -53,6 +53,33 @@ class FileReader:
         ".yml",
     ]
 
+    # Code file extensions with their markdown language identifiers for syntax highlighting
+    code_markdown_identifier: ClassVar[dict[str, str]] = {
+        ".astro": "astro",
+        ".c": "c",
+        ".cpp": "cpp",
+        ".css": "css",
+        ".go": "go",
+        ".h": "c",
+        ".hpp": "cpp",
+        ".java": "java",
+        ".js": "javascript",
+        ".json": "json",
+        ".kt": "kotlin",
+        ".mjs": "javascript",
+        ".php": "php",
+        ".py": "python",
+        ".rb": "ruby",
+        ".rs": "rust",
+        ".svelte": "svelte",
+        ".swift": "swift",
+        ".ts": "typescript",
+        ".tsx": "tsx",
+        ".vue": "vue",
+        ".yaml": "yaml",
+        ".yml": "yaml",
+    }
+
     extensions: ClassVar[list[str]] = docling_extensions + text_extensions
 
     @staticmethod
@@ -67,7 +94,14 @@ class FileReader:
                 return result.document.export_to_markdown()
             elif file_extension in FileReader.text_extensions:
                 # Read plain text files directly
-                return path.read_text(encoding="utf-8")
+                content = path.read_text(encoding="utf-8")
+
+                # Wrap code files (but not plain txt) in markdown code blocks for better presentation
+                if file_extension in FileReader.code_markdown_identifier:
+                    language = FileReader.code_markdown_identifier[file_extension]
+                    return f"```{language}\n{content}\n```"
+
+                return content
             else:
                 # Fallback: try to read as text
                 return path.read_text(encoding="utf-8")