amd-gaia 0.15.0__py3-none-any.whl → 0.15.2__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (185) hide show
  1. {amd_gaia-0.15.0.dist-info → amd_gaia-0.15.2.dist-info}/METADATA +222 -223
  2. amd_gaia-0.15.2.dist-info/RECORD +182 -0
  3. {amd_gaia-0.15.0.dist-info → amd_gaia-0.15.2.dist-info}/WHEEL +1 -1
  4. {amd_gaia-0.15.0.dist-info → amd_gaia-0.15.2.dist-info}/entry_points.txt +1 -0
  5. {amd_gaia-0.15.0.dist-info → amd_gaia-0.15.2.dist-info}/licenses/LICENSE.md +20 -20
  6. gaia/__init__.py +29 -29
  7. gaia/agents/__init__.py +19 -19
  8. gaia/agents/base/__init__.py +9 -9
  9. gaia/agents/base/agent.py +2132 -2177
  10. gaia/agents/base/api_agent.py +119 -120
  11. gaia/agents/base/console.py +1967 -1841
  12. gaia/agents/base/errors.py +237 -237
  13. gaia/agents/base/mcp_agent.py +86 -86
  14. gaia/agents/base/tools.py +88 -83
  15. gaia/agents/blender/__init__.py +7 -0
  16. gaia/agents/blender/agent.py +553 -556
  17. gaia/agents/blender/agent_simple.py +133 -135
  18. gaia/agents/blender/app.py +211 -211
  19. gaia/agents/blender/app_simple.py +41 -41
  20. gaia/agents/blender/core/__init__.py +16 -16
  21. gaia/agents/blender/core/materials.py +506 -506
  22. gaia/agents/blender/core/objects.py +316 -316
  23. gaia/agents/blender/core/rendering.py +225 -225
  24. gaia/agents/blender/core/scene.py +220 -220
  25. gaia/agents/blender/core/view.py +146 -146
  26. gaia/agents/chat/__init__.py +9 -9
  27. gaia/agents/chat/agent.py +809 -835
  28. gaia/agents/chat/app.py +1065 -1058
  29. gaia/agents/chat/session.py +508 -508
  30. gaia/agents/chat/tools/__init__.py +15 -15
  31. gaia/agents/chat/tools/file_tools.py +96 -96
  32. gaia/agents/chat/tools/rag_tools.py +1744 -1729
  33. gaia/agents/chat/tools/shell_tools.py +437 -436
  34. gaia/agents/code/__init__.py +7 -7
  35. gaia/agents/code/agent.py +549 -549
  36. gaia/agents/code/cli.py +377 -0
  37. gaia/agents/code/models.py +135 -135
  38. gaia/agents/code/orchestration/__init__.py +24 -24
  39. gaia/agents/code/orchestration/checklist_executor.py +1763 -1763
  40. gaia/agents/code/orchestration/checklist_generator.py +713 -713
  41. gaia/agents/code/orchestration/factories/__init__.py +9 -9
  42. gaia/agents/code/orchestration/factories/base.py +63 -63
  43. gaia/agents/code/orchestration/factories/nextjs_factory.py +118 -118
  44. gaia/agents/code/orchestration/factories/python_factory.py +106 -106
  45. gaia/agents/code/orchestration/orchestrator.py +841 -841
  46. gaia/agents/code/orchestration/project_analyzer.py +391 -391
  47. gaia/agents/code/orchestration/steps/__init__.py +67 -67
  48. gaia/agents/code/orchestration/steps/base.py +188 -188
  49. gaia/agents/code/orchestration/steps/error_handler.py +314 -314
  50. gaia/agents/code/orchestration/steps/nextjs.py +828 -828
  51. gaia/agents/code/orchestration/steps/python.py +307 -307
  52. gaia/agents/code/orchestration/template_catalog.py +469 -469
  53. gaia/agents/code/orchestration/workflows/__init__.py +14 -14
  54. gaia/agents/code/orchestration/workflows/base.py +80 -80
  55. gaia/agents/code/orchestration/workflows/nextjs.py +186 -186
  56. gaia/agents/code/orchestration/workflows/python.py +94 -94
  57. gaia/agents/code/prompts/__init__.py +11 -11
  58. gaia/agents/code/prompts/base_prompt.py +77 -77
  59. gaia/agents/code/prompts/code_patterns.py +2034 -2036
  60. gaia/agents/code/prompts/nextjs_prompt.py +40 -40
  61. gaia/agents/code/prompts/python_prompt.py +109 -109
  62. gaia/agents/code/schema_inference.py +365 -365
  63. gaia/agents/code/system_prompt.py +41 -41
  64. gaia/agents/code/tools/__init__.py +42 -42
  65. gaia/agents/code/tools/cli_tools.py +1138 -1138
  66. gaia/agents/code/tools/code_formatting.py +319 -319
  67. gaia/agents/code/tools/code_tools.py +769 -769
  68. gaia/agents/code/tools/error_fixing.py +1347 -1347
  69. gaia/agents/code/tools/external_tools.py +180 -180
  70. gaia/agents/code/tools/file_io.py +845 -845
  71. gaia/agents/code/tools/prisma_tools.py +190 -190
  72. gaia/agents/code/tools/project_management.py +1016 -1016
  73. gaia/agents/code/tools/testing.py +321 -321
  74. gaia/agents/code/tools/typescript_tools.py +122 -122
  75. gaia/agents/code/tools/validation_parsing.py +461 -461
  76. gaia/agents/code/tools/validation_tools.py +806 -806
  77. gaia/agents/code/tools/web_dev_tools.py +1758 -1758
  78. gaia/agents/code/validators/__init__.py +16 -16
  79. gaia/agents/code/validators/antipattern_checker.py +241 -241
  80. gaia/agents/code/validators/ast_analyzer.py +197 -197
  81. gaia/agents/code/validators/requirements_validator.py +145 -145
  82. gaia/agents/code/validators/syntax_validator.py +171 -171
  83. gaia/agents/docker/__init__.py +7 -7
  84. gaia/agents/docker/agent.py +643 -642
  85. gaia/agents/emr/__init__.py +8 -8
  86. gaia/agents/emr/agent.py +1504 -1506
  87. gaia/agents/emr/cli.py +1322 -1322
  88. gaia/agents/emr/constants.py +475 -475
  89. gaia/agents/emr/dashboard/__init__.py +4 -4
  90. gaia/agents/emr/dashboard/server.py +1972 -1974
  91. gaia/agents/jira/__init__.py +11 -11
  92. gaia/agents/jira/agent.py +894 -894
  93. gaia/agents/jira/jql_templates.py +299 -299
  94. gaia/agents/routing/__init__.py +7 -7
  95. gaia/agents/routing/agent.py +567 -570
  96. gaia/agents/routing/system_prompt.py +75 -75
  97. gaia/agents/summarize/__init__.py +11 -0
  98. gaia/agents/summarize/agent.py +885 -0
  99. gaia/agents/summarize/prompts.py +129 -0
  100. gaia/api/__init__.py +23 -23
  101. gaia/api/agent_registry.py +238 -238
  102. gaia/api/app.py +305 -305
  103. gaia/api/openai_server.py +575 -575
  104. gaia/api/schemas.py +186 -186
  105. gaia/api/sse_handler.py +373 -373
  106. gaia/apps/__init__.py +4 -4
  107. gaia/apps/llm/__init__.py +6 -6
  108. gaia/apps/llm/app.py +184 -169
  109. gaia/apps/summarize/app.py +116 -633
  110. gaia/apps/summarize/html_viewer.py +133 -133
  111. gaia/apps/summarize/pdf_formatter.py +284 -284
  112. gaia/audio/__init__.py +2 -2
  113. gaia/audio/audio_client.py +439 -439
  114. gaia/audio/audio_recorder.py +269 -269
  115. gaia/audio/kokoro_tts.py +599 -599
  116. gaia/audio/whisper_asr.py +432 -432
  117. gaia/chat/__init__.py +16 -16
  118. gaia/chat/app.py +428 -430
  119. gaia/chat/prompts.py +522 -522
  120. gaia/chat/sdk.py +1228 -1225
  121. gaia/cli.py +5659 -5632
  122. gaia/database/__init__.py +10 -10
  123. gaia/database/agent.py +176 -176
  124. gaia/database/mixin.py +290 -290
  125. gaia/database/testing.py +64 -64
  126. gaia/eval/batch_experiment.py +2332 -2332
  127. gaia/eval/claude.py +542 -542
  128. gaia/eval/config.py +37 -37
  129. gaia/eval/email_generator.py +512 -512
  130. gaia/eval/eval.py +3179 -3179
  131. gaia/eval/groundtruth.py +1130 -1130
  132. gaia/eval/transcript_generator.py +582 -582
  133. gaia/eval/webapp/README.md +167 -167
  134. gaia/eval/webapp/package-lock.json +875 -875
  135. gaia/eval/webapp/package.json +20 -20
  136. gaia/eval/webapp/public/app.js +3402 -3402
  137. gaia/eval/webapp/public/index.html +87 -87
  138. gaia/eval/webapp/public/styles.css +3661 -3661
  139. gaia/eval/webapp/server.js +415 -415
  140. gaia/eval/webapp/test-setup.js +72 -72
  141. gaia/installer/__init__.py +23 -0
  142. gaia/installer/init_command.py +1275 -0
  143. gaia/installer/lemonade_installer.py +619 -0
  144. gaia/llm/__init__.py +10 -2
  145. gaia/llm/base_client.py +60 -0
  146. gaia/llm/exceptions.py +12 -0
  147. gaia/llm/factory.py +70 -0
  148. gaia/llm/lemonade_client.py +3421 -3221
  149. gaia/llm/lemonade_manager.py +294 -294
  150. gaia/llm/providers/__init__.py +9 -0
  151. gaia/llm/providers/claude.py +108 -0
  152. gaia/llm/providers/lemonade.py +118 -0
  153. gaia/llm/providers/openai_provider.py +79 -0
  154. gaia/llm/vlm_client.py +382 -382
  155. gaia/logger.py +189 -189
  156. gaia/mcp/agent_mcp_server.py +245 -245
  157. gaia/mcp/blender_mcp_client.py +138 -138
  158. gaia/mcp/blender_mcp_server.py +648 -648
  159. gaia/mcp/context7_cache.py +332 -332
  160. gaia/mcp/external_services.py +518 -518
  161. gaia/mcp/mcp_bridge.py +811 -550
  162. gaia/mcp/servers/__init__.py +6 -6
  163. gaia/mcp/servers/docker_mcp.py +83 -83
  164. gaia/perf_analysis.py +361 -0
  165. gaia/rag/__init__.py +10 -10
  166. gaia/rag/app.py +293 -293
  167. gaia/rag/demo.py +304 -304
  168. gaia/rag/pdf_utils.py +235 -235
  169. gaia/rag/sdk.py +2194 -2194
  170. gaia/security.py +183 -163
  171. gaia/talk/app.py +287 -289
  172. gaia/talk/sdk.py +538 -538
  173. gaia/testing/__init__.py +87 -87
  174. gaia/testing/assertions.py +330 -330
  175. gaia/testing/fixtures.py +333 -333
  176. gaia/testing/mocks.py +493 -493
  177. gaia/util.py +46 -46
  178. gaia/utils/__init__.py +33 -33
  179. gaia/utils/file_watcher.py +675 -675
  180. gaia/utils/parsing.py +223 -223
  181. gaia/version.py +100 -100
  182. amd_gaia-0.15.0.dist-info/RECORD +0 -168
  183. gaia/agents/code/app.py +0 -266
  184. gaia/llm/llm_client.py +0 -723
  185. {amd_gaia-0.15.0.dist-info → amd_gaia-0.15.2.dist-info}/top_level.txt +0 -0
@@ -1,469 +1,469 @@
1
- # Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
2
- # SPDX-License-Identifier: MIT
3
- """Template Catalog for LLM-Driven Checklist Generation.
4
-
5
- This module defines the catalog of available templates that the LLM sees
6
- during checklist generation. Each template definition includes:
7
- - Name: The template identifier used in checklist items
8
- - Description: Human-readable description for the LLM
9
- - Parameters: Expected parameters with their types and descriptions
10
- - Dependencies: Other templates that should be executed first
11
- - Produces: Files/artifacts this template creates
12
-
13
- The catalog is used by ChecklistGenerator to build the system prompt
14
- that tells the LLM what templates are available and how to use them.
15
- """
16
-
17
- from dataclasses import dataclass, field
18
- from enum import Enum
19
- from typing import Any, Dict, List, Optional
20
-
21
-
22
- class ParameterType(Enum):
23
- """Supported parameter types for template parameters."""
24
-
25
- STRING = "string"
26
- NUMBER = "number"
27
- BOOLEAN = "boolean"
28
- LIST = "list"
29
- DICT = "dict"
30
-
31
-
32
- @dataclass
33
- class ParameterSpec:
34
- """Specification for a template parameter."""
35
-
36
- type: ParameterType
37
- description: str
38
- required: bool = True
39
- default: Optional[Any] = None
40
- example: Optional[Any] = None
41
-
42
- def to_prompt(self) -> str:
43
- """Generate prompt-friendly description of this parameter."""
44
- type_str = self.type.value
45
- req_str = "required" if self.required else "optional"
46
- desc = f"{type_str}, {req_str}: {self.description}"
47
- if self.example is not None:
48
- desc += f" (e.g., {self.example})"
49
- return desc
50
-
51
-
52
- @dataclass
53
- class TemplateDefinition:
54
- """Definition of a template for the catalog.
55
-
56
- Each template represents an executable unit of code generation
57
- that the LLM can include in its checklist.
58
- """
59
-
60
- name: str
61
- description: str
62
- parameters: Dict[str, ParameterSpec]
63
- dependencies: List[str] = field(default_factory=list)
64
- produces: List[str] = field(default_factory=list)
65
- category: str = "general"
66
- semantic_hints: List[str] = field(default_factory=list)
67
-
68
- def to_prompt(self) -> str:
69
- """Generate prompt-friendly description of this template."""
70
- lines = [f"### {self.name}"]
71
- lines.append(f"**Description**: {self.description}")
72
-
73
- if self.semantic_hints:
74
- lines.append(f"**When to use**: {', '.join(self.semantic_hints)}")
75
-
76
- lines.append("**Parameters**:")
77
- for param_name, param_spec in self.parameters.items():
78
- lines.append(f" - `{param_name}`: {param_spec.to_prompt()}")
79
-
80
- if self.dependencies:
81
- lines.append(f"**Requires**: {', '.join(self.dependencies)}")
82
-
83
- if self.produces:
84
- lines.append(f"**Creates**: {', '.join(self.produces)}")
85
-
86
- return "\n".join(lines)
87
-
88
-
89
- # ============================================================================
90
- # Template Catalog Definition
91
- # ============================================================================
92
-
93
- TEMPLATE_CATALOG: Dict[str, TemplateDefinition] = {
94
- # ========== Project Setup Templates ==========
95
- "create_next_app": TemplateDefinition(
96
- name="create_next_app",
97
- description="Initialize a new Next.js project with TypeScript, Tailwind CSS, and App Router",
98
- parameters={
99
- "project_name": ParameterSpec(
100
- type=ParameterType.STRING,
101
- description="Name of the project (used for directory)",
102
- example="my-todo-app",
103
- ),
104
- },
105
- dependencies=[],
106
- produces=[
107
- "package.json",
108
- "next.config.ts",
109
- "src/app/layout.tsx",
110
- "src/app/page.tsx",
111
- ],
112
- category="setup",
113
- semantic_hints=["Always use this first for new Next.js projects"],
114
- ),
115
- "setup_app_styling": TemplateDefinition(
116
- name="setup_app_styling",
117
- description="Configure app-wide styling with modern dark theme design system",
118
- parameters={
119
- "app_title": ParameterSpec(
120
- type=ParameterType.STRING,
121
- description="Application title for metadata",
122
- example="Todo App",
123
- ),
124
- "app_description": ParameterSpec(
125
- type=ParameterType.STRING,
126
- description="Application description for metadata",
127
- required=False,
128
- default="A modern web application",
129
- ),
130
- },
131
- dependencies=["create_next_app"],
132
- produces=["src/app/layout.tsx", "src/app/globals.css"],
133
- category="setup",
134
- semantic_hints=["Sets up dark theme with glass morphism effects"],
135
- ),
136
- "setup_prisma": TemplateDefinition(
137
- name="setup_prisma",
138
- description="Install Prisma dependencies, initialize database with SQLite, and create client singleton",
139
- parameters={
140
- "database_url": ParameterSpec(
141
- type=ParameterType.STRING,
142
- description="Database connection URL",
143
- required=False,
144
- default="file:./dev.db",
145
- ),
146
- },
147
- dependencies=["create_next_app"],
148
- produces=["prisma/schema.prisma", "src/lib/prisma.ts", ".env"],
149
- category="setup",
150
- semantic_hints=[
151
- "Installs Prisma 5.x, @prisma/client, and Zod dependencies",
152
- "Runs 'npx prisma init --datasource-provider sqlite' to create schema.prisma",
153
- "Creates src/lib/prisma.ts singleton for database client access",
154
- "MUST complete before creating any Prisma models or database operations",
155
- ],
156
- ),
157
- "setup_testing": TemplateDefinition(
158
- name="setup_testing",
159
- description="Set up Vitest testing infrastructure with React Testing Library",
160
- parameters={
161
- "resource_name": ParameterSpec(
162
- type=ParameterType.STRING,
163
- description="Optional resource name for customized Prisma mocks",
164
- required=False,
165
- ),
166
- },
167
- dependencies=["create_next_app"],
168
- produces=["vitest.config.ts", "tests/setup.ts"],
169
- category="setup",
170
- semantic_hints=["Enables running tests with npm test"],
171
- ),
172
- # ========== Data Model Templates ==========
173
- "generate_prisma_model": TemplateDefinition(
174
- name="generate_prisma_model",
175
- description="Define a database model with fields in Prisma schema",
176
- parameters={
177
- "model_name": ParameterSpec(
178
- type=ParameterType.STRING,
179
- description="Model name in PascalCase (singular)",
180
- example="Todo",
181
- ),
182
- "fields": ParameterSpec(
183
- type=ParameterType.DICT,
184
- description="Field definitions as {name: type} where type is string|number|boolean|date|email|url",
185
- example={"title": "string", "completed": "boolean"},
186
- ),
187
- },
188
- dependencies=["setup_prisma"],
189
- produces=["prisma/schema.prisma (updated)"],
190
- category="data",
191
- semantic_hints=[
192
- "Use 'boolean' for todo completion, checklist items",
193
- "Use 'date' for blog posts, events with dates",
194
- "id, createdAt, updatedAt are auto-generated",
195
- ],
196
- ),
197
- "prisma_db_sync": TemplateDefinition(
198
- name="prisma_db_sync",
199
- description="Generate Prisma client and push schema to database (REQUIRED after generate_prisma_model)",
200
- parameters={},
201
- dependencies=["generate_prisma_model"],
202
- produces=["node_modules/.prisma/client", "prisma/dev.db"],
203
- category="data",
204
- semantic_hints=[
205
- "MUST run after generate_prisma_model before any API routes",
206
- "Runs 'prisma generate' to create TypeScript types",
207
- "Runs 'prisma db push' to create database tables",
208
- ],
209
- ),
210
- # ========== API Templates ==========
211
- "generate_api_route": TemplateDefinition(
212
- name="generate_api_route",
213
- description="Create REST API endpoints with Prisma queries and Zod validation",
214
- parameters={
215
- "resource": ParameterSpec(
216
- type=ParameterType.STRING,
217
- description="Resource name in lowercase (singular)",
218
- example="todo",
219
- ),
220
- "operations": ParameterSpec(
221
- type=ParameterType.LIST,
222
- description="HTTP methods to implement",
223
- example=["GET", "POST"],
224
- ),
225
- "type": ParameterSpec(
226
- type=ParameterType.STRING,
227
- description="Route type: 'collection' for /api/todos or 'item' for /api/todos/[id]",
228
- example="collection",
229
- ),
230
- "enable_pagination": ParameterSpec(
231
- type=ParameterType.BOOLEAN,
232
- description="Whether to add pagination to GET endpoint",
233
- required=False,
234
- default=False,
235
- ),
236
- },
237
- dependencies=["prisma_db_sync"],
238
- produces=[
239
- "src/app/api/{resource}s/route.ts",
240
- "src/app/api/{resource}s/[id]/route.ts",
241
- ],
242
- category="api",
243
- semantic_hints=[
244
- "Collection routes handle GET (list) and POST (create)",
245
- "Item routes handle GET (single), PATCH (update), DELETE",
246
- ],
247
- ),
248
- # ========== UI Component Templates ==========
249
- "generate_react_component": TemplateDefinition(
250
- name="generate_react_component",
251
- description="Create React components for displaying and managing resources",
252
- parameters={
253
- "resource": ParameterSpec(
254
- type=ParameterType.STRING,
255
- description="Resource name in lowercase (singular)",
256
- example="todo",
257
- ),
258
- "variant": ParameterSpec(
259
- type=ParameterType.STRING,
260
- description="Component variant: list|form|new|detail|actions|artifact-timer",
261
- example="list",
262
- ),
263
- "component_name": ParameterSpec(
264
- type=ParameterType.STRING,
265
- description="Optional explicit component name (e.g., CountdownTimer)",
266
- required=False,
267
- ),
268
- "with_checkboxes": ParameterSpec(
269
- type=ParameterType.BOOLEAN,
270
- description="Add checkbox UI for boolean fields (e.g., todo completion)",
271
- required=False,
272
- default=False,
273
- ),
274
- },
275
- dependencies=["generate_api_route"],
276
- produces=["src/app/{resource}s/page.tsx", "src/components/{Resource}Form.tsx"],
277
- category="ui",
278
- semantic_hints=[
279
- "Use 'list' for main page showing all items",
280
- "Use 'form' for reusable create/edit form component",
281
- "Use 'new' for /resource/new page",
282
- "Use 'detail' for /resource/[id] EDIT page with pre-populated form",
283
- "Use 'artifact-timer' when the user requests a countdown; supply component_name (e.g., CountdownTimer) so pages can import the client-side timer widget",
284
- "Add with_checkboxes=true for todo apps",
285
- ],
286
- ),
287
- "update_landing_page": TemplateDefinition(
288
- name="update_landing_page",
289
- description="Update the home page with navigation links to resource pages",
290
- parameters={
291
- "resource": ParameterSpec(
292
- type=ParameterType.STRING,
293
- description="Resource name to link to",
294
- example="todo",
295
- ),
296
- "description": ParameterSpec(
297
- type=ParameterType.STRING,
298
- description="Description text for the link",
299
- required=False,
300
- ),
301
- },
302
- dependencies=["generate_react_component"],
303
- produces=["src/app/page.tsx (updated)"],
304
- category="ui",
305
- semantic_hints=["Add after creating resource pages so users can navigate"],
306
- ),
307
- # ========== Validation Templates ==========
308
- "run_typescript_check": TemplateDefinition(
309
- name="run_typescript_check",
310
- description="Run TypeScript compiler to check for type errors",
311
- parameters={},
312
- dependencies=[],
313
- produces=[],
314
- category="validation",
315
- semantic_hints=["Use after generating code to catch type errors early"],
316
- ),
317
- "validate_styles": TemplateDefinition(
318
- name="validate_styles",
319
- description="Validate CSS files for content integrity and design system consistency",
320
- parameters={
321
- "_resource_name": ParameterSpec(
322
- type=ParameterType.STRING,
323
- description="Optional resource name for component class checks",
324
- required=False,
325
- ),
326
- },
327
- dependencies=["setup_app_styling", "run_typescript_check"],
328
- produces=[],
329
- category="validation",
330
- semantic_hints=[
331
- "Run AFTER run_typescript_check to validate styling",
332
- "Catches TypeScript code accidentally written to CSS files (Issue #1002)",
333
- "Validates globals.css has Tailwind directives",
334
- "Checks layout.tsx imports globals.css",
335
- ],
336
- ),
337
- "generate_style_tests": TemplateDefinition(
338
- name="generate_style_tests",
339
- description="Generate CSS and styling tests for the project",
340
- parameters={
341
- "resource_name": ParameterSpec(
342
- type=ParameterType.STRING,
343
- description="Resource name for component styling tests",
344
- required=True,
345
- ),
346
- },
347
- dependencies=["setup_testing", "setup_app_styling"],
348
- produces=["tests/styles.test.ts", "tests/styling/{Resource}Styling.test.tsx"],
349
- category="testing",
350
- semantic_hints=[
351
- "Generate after setup_testing to create CSS validation tests",
352
- "Tests check CSS file integrity (no TypeScript in CSS)",
353
- "Tests validate design system class definitions",
354
- "Tests verify layout.tsx imports globals.css",
355
- ],
356
- ),
357
- # ========== Remediation Templates ==========
358
- "fix_code": TemplateDefinition(
359
- name="fix_code",
360
- description="Use the LLM fixer to repair an existing source file based on an error description.",
361
- parameters={
362
- "file_path": ParameterSpec(
363
- type=ParameterType.STRING,
364
- description="Path to the file that needs to be fixed.",
365
- example="src/app/api/todos/route.ts",
366
- ),
367
- "error_description": ParameterSpec(
368
- type=ParameterType.STRING,
369
- description="Short summary of the failure or lint error that needs to be resolved.",
370
- required=False,
371
- ),
372
- },
373
- dependencies=[],
374
- produces=["Updated file with fixes applied"],
375
- category="remediation",
376
- semantic_hints=[
377
- "Use when a validation log references a specific file with errors.",
378
- "Provide the exact error message (TypeScript, lint, or runtime) to guide the fixer.",
379
- ],
380
- ),
381
- }
382
-
383
-
384
- def get_template(name: str) -> Optional[TemplateDefinition]:
385
- """Get a template definition by name.
386
-
387
- Args:
388
- name: Template name to look up
389
-
390
- Returns:
391
- TemplateDefinition if found, None otherwise
392
- """
393
- return TEMPLATE_CATALOG.get(name)
394
-
395
-
396
- def get_templates_by_category(category: str) -> List[TemplateDefinition]:
397
- """Get all templates in a specific category.
398
-
399
- Args:
400
- category: Category name (setup, data, api, ui, testing, validation, remediation)
401
-
402
- Returns:
403
- List of templates in the category
404
- """
405
- return [t for t in TEMPLATE_CATALOG.values() if t.category == category]
406
-
407
-
408
- def get_catalog_prompt() -> str:
409
- """Generate the complete template catalog prompt for LLM.
410
-
411
- This is included in the system prompt so the LLM knows what
412
- templates are available and how to use them.
413
-
414
- Returns:
415
- Formatted markdown string describing all templates
416
- """
417
- lines = ["# Available Templates", ""]
418
- lines.append(
419
- "Use these templates to generate a checklist for the user's request. "
420
- "Each template has specific parameters and dependencies."
421
- )
422
- lines.append("")
423
-
424
- # Group by category
425
- categories = ["setup", "data", "api", "ui", "testing", "validation", "remediation"]
426
-
427
- for category in categories:
428
- templates = get_templates_by_category(category)
429
- if templates:
430
- lines.append(f"## {category.title()} Templates")
431
- lines.append("")
432
- for template in templates:
433
- lines.append(template.to_prompt())
434
- lines.append("")
435
-
436
- return "\n".join(lines)
437
-
438
-
439
- def validate_checklist_item(template_name: str, params: Dict[str, Any]) -> List[str]:
440
- """Validate a checklist item against the template definition.
441
-
442
- Args:
443
- template_name: Name of the template
444
- params: Parameters provided for the template
445
-
446
- Returns:
447
- List of validation error messages (empty if valid)
448
- """
449
- errors = []
450
-
451
- template = get_template(template_name)
452
- if not template:
453
- errors.append(f"Unknown template: {template_name}")
454
- return errors
455
-
456
- # Check required parameters
457
- for param_name, param_spec in template.parameters.items():
458
- if param_spec.required and param_name not in params:
459
- errors.append(
460
- f"Missing required parameter '{param_name}' for {template_name}"
461
- )
462
-
463
- # Check for unknown parameters
464
- valid_params = set(template.parameters.keys())
465
- for param_name in params:
466
- if param_name not in valid_params:
467
- errors.append(f"Unknown parameter '{param_name}' for {template_name}")
468
-
469
- return errors
1
+ # Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
2
+ # SPDX-License-Identifier: MIT
3
+ """Template Catalog for LLM-Driven Checklist Generation.
4
+
5
+ This module defines the catalog of available templates that the LLM sees
6
+ during checklist generation. Each template definition includes:
7
+ - Name: The template identifier used in checklist items
8
+ - Description: Human-readable description for the LLM
9
+ - Parameters: Expected parameters with their types and descriptions
10
+ - Dependencies: Other templates that should be executed first
11
+ - Produces: Files/artifacts this template creates
12
+
13
+ The catalog is used by ChecklistGenerator to build the system prompt
14
+ that tells the LLM what templates are available and how to use them.
15
+ """
16
+
17
+ from dataclasses import dataclass, field
18
+ from enum import Enum
19
+ from typing import Any, Dict, List, Optional
20
+
21
+
22
+ class ParameterType(Enum):
23
+ """Supported parameter types for template parameters."""
24
+
25
+ STRING = "string"
26
+ NUMBER = "number"
27
+ BOOLEAN = "boolean"
28
+ LIST = "list"
29
+ DICT = "dict"
30
+
31
+
32
+ @dataclass
33
+ class ParameterSpec:
34
+ """Specification for a template parameter."""
35
+
36
+ type: ParameterType
37
+ description: str
38
+ required: bool = True
39
+ default: Optional[Any] = None
40
+ example: Optional[Any] = None
41
+
42
+ def to_prompt(self) -> str:
43
+ """Generate prompt-friendly description of this parameter."""
44
+ type_str = self.type.value
45
+ req_str = "required" if self.required else "optional"
46
+ desc = f"{type_str}, {req_str}: {self.description}"
47
+ if self.example is not None:
48
+ desc += f" (e.g., {self.example})"
49
+ return desc
50
+
51
+
52
+ @dataclass
53
+ class TemplateDefinition:
54
+ """Definition of a template for the catalog.
55
+
56
+ Each template represents an executable unit of code generation
57
+ that the LLM can include in its checklist.
58
+ """
59
+
60
+ name: str
61
+ description: str
62
+ parameters: Dict[str, ParameterSpec]
63
+ dependencies: List[str] = field(default_factory=list)
64
+ produces: List[str] = field(default_factory=list)
65
+ category: str = "general"
66
+ semantic_hints: List[str] = field(default_factory=list)
67
+
68
+ def to_prompt(self) -> str:
69
+ """Generate prompt-friendly description of this template."""
70
+ lines = [f"### {self.name}"]
71
+ lines.append(f"**Description**: {self.description}")
72
+
73
+ if self.semantic_hints:
74
+ lines.append(f"**When to use**: {', '.join(self.semantic_hints)}")
75
+
76
+ lines.append("**Parameters**:")
77
+ for param_name, param_spec in self.parameters.items():
78
+ lines.append(f" - `{param_name}`: {param_spec.to_prompt()}")
79
+
80
+ if self.dependencies:
81
+ lines.append(f"**Requires**: {', '.join(self.dependencies)}")
82
+
83
+ if self.produces:
84
+ lines.append(f"**Creates**: {', '.join(self.produces)}")
85
+
86
+ return "\n".join(lines)
87
+
88
+
89
+ # ============================================================================
90
+ # Template Catalog Definition
91
+ # ============================================================================
92
+
93
+ TEMPLATE_CATALOG: Dict[str, TemplateDefinition] = {
94
+ # ========== Project Setup Templates ==========
95
+ "create_next_app": TemplateDefinition(
96
+ name="create_next_app",
97
+ description="Initialize a new Next.js project with TypeScript, Tailwind CSS, and App Router",
98
+ parameters={
99
+ "project_name": ParameterSpec(
100
+ type=ParameterType.STRING,
101
+ description="Name of the project (used for directory)",
102
+ example="my-todo-app",
103
+ ),
104
+ },
105
+ dependencies=[],
106
+ produces=[
107
+ "package.json",
108
+ "next.config.ts",
109
+ "src/app/layout.tsx",
110
+ "src/app/page.tsx",
111
+ ],
112
+ category="setup",
113
+ semantic_hints=["Always use this first for new Next.js projects"],
114
+ ),
115
+ "setup_app_styling": TemplateDefinition(
116
+ name="setup_app_styling",
117
+ description="Configure app-wide styling with modern dark theme design system",
118
+ parameters={
119
+ "app_title": ParameterSpec(
120
+ type=ParameterType.STRING,
121
+ description="Application title for metadata",
122
+ example="Todo App",
123
+ ),
124
+ "app_description": ParameterSpec(
125
+ type=ParameterType.STRING,
126
+ description="Application description for metadata",
127
+ required=False,
128
+ default="A modern web application",
129
+ ),
130
+ },
131
+ dependencies=["create_next_app"],
132
+ produces=["src/app/layout.tsx", "src/app/globals.css"],
133
+ category="setup",
134
+ semantic_hints=["Sets up dark theme with glass morphism effects"],
135
+ ),
136
+ "setup_prisma": TemplateDefinition(
137
+ name="setup_prisma",
138
+ description="Install Prisma dependencies, initialize database with SQLite, and create client singleton",
139
+ parameters={
140
+ "database_url": ParameterSpec(
141
+ type=ParameterType.STRING,
142
+ description="Database connection URL",
143
+ required=False,
144
+ default="file:./dev.db",
145
+ ),
146
+ },
147
+ dependencies=["create_next_app"],
148
+ produces=["prisma/schema.prisma", "src/lib/prisma.ts", ".env"],
149
+ category="setup",
150
+ semantic_hints=[
151
+ "Installs Prisma 5.x, @prisma/client, and Zod dependencies",
152
+ "Runs 'npx prisma init --datasource-provider sqlite' to create schema.prisma",
153
+ "Creates src/lib/prisma.ts singleton for database client access",
154
+ "MUST complete before creating any Prisma models or database operations",
155
+ ],
156
+ ),
157
+ "setup_testing": TemplateDefinition(
158
+ name="setup_testing",
159
+ description="Set up Vitest testing infrastructure with React Testing Library",
160
+ parameters={
161
+ "resource_name": ParameterSpec(
162
+ type=ParameterType.STRING,
163
+ description="Optional resource name for customized Prisma mocks",
164
+ required=False,
165
+ ),
166
+ },
167
+ dependencies=["create_next_app"],
168
+ produces=["vitest.config.ts", "tests/setup.ts"],
169
+ category="setup",
170
+ semantic_hints=["Enables running tests with npm test"],
171
+ ),
172
+ # ========== Data Model Templates ==========
173
+ "generate_prisma_model": TemplateDefinition(
174
+ name="generate_prisma_model",
175
+ description="Define a database model with fields in Prisma schema",
176
+ parameters={
177
+ "model_name": ParameterSpec(
178
+ type=ParameterType.STRING,
179
+ description="Model name in PascalCase (singular)",
180
+ example="Todo",
181
+ ),
182
+ "fields": ParameterSpec(
183
+ type=ParameterType.DICT,
184
+ description="Field definitions as {name: type} where type is string|number|boolean|date|email|url",
185
+ example={"title": "string", "completed": "boolean"},
186
+ ),
187
+ },
188
+ dependencies=["setup_prisma"],
189
+ produces=["prisma/schema.prisma (updated)"],
190
+ category="data",
191
+ semantic_hints=[
192
+ "Use 'boolean' for todo completion, checklist items",
193
+ "Use 'date' for blog posts, events with dates",
194
+ "id, createdAt, updatedAt are auto-generated",
195
+ ],
196
+ ),
197
+ "prisma_db_sync": TemplateDefinition(
198
+ name="prisma_db_sync",
199
+ description="Generate Prisma client and push schema to database (REQUIRED after generate_prisma_model)",
200
+ parameters={},
201
+ dependencies=["generate_prisma_model"],
202
+ produces=["node_modules/.prisma/client", "prisma/dev.db"],
203
+ category="data",
204
+ semantic_hints=[
205
+ "MUST run after generate_prisma_model before any API routes",
206
+ "Runs 'prisma generate' to create TypeScript types",
207
+ "Runs 'prisma db push' to create database tables",
208
+ ],
209
+ ),
210
+ # ========== API Templates ==========
211
+ "generate_api_route": TemplateDefinition(
212
+ name="generate_api_route",
213
+ description="Create REST API endpoints with Prisma queries and Zod validation",
214
+ parameters={
215
+ "resource": ParameterSpec(
216
+ type=ParameterType.STRING,
217
+ description="Resource name in lowercase (singular)",
218
+ example="todo",
219
+ ),
220
+ "operations": ParameterSpec(
221
+ type=ParameterType.LIST,
222
+ description="HTTP methods to implement",
223
+ example=["GET", "POST"],
224
+ ),
225
+ "type": ParameterSpec(
226
+ type=ParameterType.STRING,
227
+ description="Route type: 'collection' for /api/todos or 'item' for /api/todos/[id]",
228
+ example="collection",
229
+ ),
230
+ "enable_pagination": ParameterSpec(
231
+ type=ParameterType.BOOLEAN,
232
+ description="Whether to add pagination to GET endpoint",
233
+ required=False,
234
+ default=False,
235
+ ),
236
+ },
237
+ dependencies=["prisma_db_sync"],
238
+ produces=[
239
+ "src/app/api/{resource}s/route.ts",
240
+ "src/app/api/{resource}s/[id]/route.ts",
241
+ ],
242
+ category="api",
243
+ semantic_hints=[
244
+ "Collection routes handle GET (list) and POST (create)",
245
+ "Item routes handle GET (single), PATCH (update), DELETE",
246
+ ],
247
+ ),
248
+ # ========== UI Component Templates ==========
249
+ "generate_react_component": TemplateDefinition(
250
+ name="generate_react_component",
251
+ description="Create React components for displaying and managing resources",
252
+ parameters={
253
+ "resource": ParameterSpec(
254
+ type=ParameterType.STRING,
255
+ description="Resource name in lowercase (singular)",
256
+ example="todo",
257
+ ),
258
+ "variant": ParameterSpec(
259
+ type=ParameterType.STRING,
260
+ description="Component variant: list|form|new|detail|actions|artifact-timer",
261
+ example="list",
262
+ ),
263
+ "component_name": ParameterSpec(
264
+ type=ParameterType.STRING,
265
+ description="Optional explicit component name (e.g., CountdownTimer)",
266
+ required=False,
267
+ ),
268
+ "with_checkboxes": ParameterSpec(
269
+ type=ParameterType.BOOLEAN,
270
+ description="Add checkbox UI for boolean fields (e.g., todo completion)",
271
+ required=False,
272
+ default=False,
273
+ ),
274
+ },
275
+ dependencies=["generate_api_route"],
276
+ produces=["src/app/{resource}s/page.tsx", "src/components/{Resource}Form.tsx"],
277
+ category="ui",
278
+ semantic_hints=[
279
+ "Use 'list' for main page showing all items",
280
+ "Use 'form' for reusable create/edit form component",
281
+ "Use 'new' for /resource/new page",
282
+ "Use 'detail' for /resource/[id] EDIT page with pre-populated form",
283
+ "Use 'artifact-timer' when the user requests a countdown; supply component_name (e.g., CountdownTimer) so pages can import the client-side timer widget",
284
+ "Add with_checkboxes=true for todo apps",
285
+ ],
286
+ ),
287
+ "update_landing_page": TemplateDefinition(
288
+ name="update_landing_page",
289
+ description="Update the home page with navigation links to resource pages",
290
+ parameters={
291
+ "resource": ParameterSpec(
292
+ type=ParameterType.STRING,
293
+ description="Resource name to link to",
294
+ example="todo",
295
+ ),
296
+ "description": ParameterSpec(
297
+ type=ParameterType.STRING,
298
+ description="Description text for the link",
299
+ required=False,
300
+ ),
301
+ },
302
+ dependencies=["generate_react_component"],
303
+ produces=["src/app/page.tsx (updated)"],
304
+ category="ui",
305
+ semantic_hints=["Add after creating resource pages so users can navigate"],
306
+ ),
307
+ # ========== Validation Templates ==========
308
+ "run_typescript_check": TemplateDefinition(
309
+ name="run_typescript_check",
310
+ description="Run TypeScript compiler to check for type errors",
311
+ parameters={},
312
+ dependencies=[],
313
+ produces=[],
314
+ category="validation",
315
+ semantic_hints=["Use after generating code to catch type errors early"],
316
+ ),
317
+ "validate_styles": TemplateDefinition(
318
+ name="validate_styles",
319
+ description="Validate CSS files for content integrity and design system consistency",
320
+ parameters={
321
+ "_resource_name": ParameterSpec(
322
+ type=ParameterType.STRING,
323
+ description="Optional resource name for component class checks",
324
+ required=False,
325
+ ),
326
+ },
327
+ dependencies=["setup_app_styling", "run_typescript_check"],
328
+ produces=[],
329
+ category="validation",
330
+ semantic_hints=[
331
+ "Run AFTER run_typescript_check to validate styling",
332
+ "Catches TypeScript code accidentally written to CSS files (Issue #1002)",
333
+ "Validates globals.css has Tailwind directives",
334
+ "Checks layout.tsx imports globals.css",
335
+ ],
336
+ ),
337
+ "generate_style_tests": TemplateDefinition(
338
+ name="generate_style_tests",
339
+ description="Generate CSS and styling tests for the project",
340
+ parameters={
341
+ "resource_name": ParameterSpec(
342
+ type=ParameterType.STRING,
343
+ description="Resource name for component styling tests",
344
+ required=True,
345
+ ),
346
+ },
347
+ dependencies=["setup_testing", "setup_app_styling"],
348
+ produces=["tests/styles.test.ts", "tests/styling/{Resource}Styling.test.tsx"],
349
+ category="testing",
350
+ semantic_hints=[
351
+ "Generate after setup_testing to create CSS validation tests",
352
+ "Tests check CSS file integrity (no TypeScript in CSS)",
353
+ "Tests validate design system class definitions",
354
+ "Tests verify layout.tsx imports globals.css",
355
+ ],
356
+ ),
357
+ # ========== Remediation Templates ==========
358
+ "fix_code": TemplateDefinition(
359
+ name="fix_code",
360
+ description="Use the LLM fixer to repair an existing source file based on an error description.",
361
+ parameters={
362
+ "file_path": ParameterSpec(
363
+ type=ParameterType.STRING,
364
+ description="Path to the file that needs to be fixed.",
365
+ example="src/app/api/todos/route.ts",
366
+ ),
367
+ "error_description": ParameterSpec(
368
+ type=ParameterType.STRING,
369
+ description="Short summary of the failure or lint error that needs to be resolved.",
370
+ required=False,
371
+ ),
372
+ },
373
+ dependencies=[],
374
+ produces=["Updated file with fixes applied"],
375
+ category="remediation",
376
+ semantic_hints=[
377
+ "Use when a validation log references a specific file with errors.",
378
+ "Provide the exact error message (TypeScript, lint, or runtime) to guide the fixer.",
379
+ ],
380
+ ),
381
+ }
382
+
383
+
384
+ def get_template(name: str) -> Optional[TemplateDefinition]:
385
+ """Get a template definition by name.
386
+
387
+ Args:
388
+ name: Template name to look up
389
+
390
+ Returns:
391
+ TemplateDefinition if found, None otherwise
392
+ """
393
+ return TEMPLATE_CATALOG.get(name)
394
+
395
+
396
+ def get_templates_by_category(category: str) -> List[TemplateDefinition]:
397
+ """Get all templates in a specific category.
398
+
399
+ Args:
400
+ category: Category name (setup, data, api, ui, testing, validation, remediation)
401
+
402
+ Returns:
403
+ List of templates in the category
404
+ """
405
+ return [t for t in TEMPLATE_CATALOG.values() if t.category == category]
406
+
407
+
408
+ def get_catalog_prompt() -> str:
409
+ """Generate the complete template catalog prompt for LLM.
410
+
411
+ This is included in the system prompt so the LLM knows what
412
+ templates are available and how to use them.
413
+
414
+ Returns:
415
+ Formatted markdown string describing all templates
416
+ """
417
+ lines = ["# Available Templates", ""]
418
+ lines.append(
419
+ "Use these templates to generate a checklist for the user's request. "
420
+ "Each template has specific parameters and dependencies."
421
+ )
422
+ lines.append("")
423
+
424
+ # Group by category
425
+ categories = ["setup", "data", "api", "ui", "testing", "validation", "remediation"]
426
+
427
+ for category in categories:
428
+ templates = get_templates_by_category(category)
429
+ if templates:
430
+ lines.append(f"## {category.title()} Templates")
431
+ lines.append("")
432
+ for template in templates:
433
+ lines.append(template.to_prompt())
434
+ lines.append("")
435
+
436
+ return "\n".join(lines)
437
+
438
+
439
+ def validate_checklist_item(template_name: str, params: Dict[str, Any]) -> List[str]:
440
+ """Validate a checklist item against the template definition.
441
+
442
+ Args:
443
+ template_name: Name of the template
444
+ params: Parameters provided for the template
445
+
446
+ Returns:
447
+ List of validation error messages (empty if valid)
448
+ """
449
+ errors = []
450
+
451
+ template = get_template(template_name)
452
+ if not template:
453
+ errors.append(f"Unknown template: {template_name}")
454
+ return errors
455
+
456
+ # Check required parameters
457
+ for param_name, param_spec in template.parameters.items():
458
+ if param_spec.required and param_name not in params:
459
+ errors.append(
460
+ f"Missing required parameter '{param_name}' for {template_name}"
461
+ )
462
+
463
+ # Check for unknown parameters
464
+ valid_params = set(template.parameters.keys())
465
+ for param_name in params:
466
+ if param_name not in valid_params:
467
+ errors.append(f"Unknown parameter '{param_name}' for {template_name}")
468
+
469
+ return errors