amd-gaia 0.14.3__py3-none-any.whl → 0.15.1__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 (181) hide show
  1. {amd_gaia-0.14.3.dist-info → amd_gaia-0.15.1.dist-info}/METADATA +223 -223
  2. amd_gaia-0.15.1.dist-info/RECORD +178 -0
  3. {amd_gaia-0.14.3.dist-info → amd_gaia-0.15.1.dist-info}/entry_points.txt +1 -0
  4. {amd_gaia-0.14.3.dist-info → amd_gaia-0.15.1.dist-info}/licenses/LICENSE.md +20 -20
  5. gaia/__init__.py +29 -29
  6. gaia/agents/__init__.py +19 -19
  7. gaia/agents/base/__init__.py +9 -9
  8. gaia/agents/base/agent.py +2177 -2177
  9. gaia/agents/base/api_agent.py +120 -120
  10. gaia/agents/base/console.py +1841 -1841
  11. gaia/agents/base/errors.py +237 -237
  12. gaia/agents/base/mcp_agent.py +86 -86
  13. gaia/agents/base/tools.py +83 -83
  14. gaia/agents/blender/agent.py +556 -556
  15. gaia/agents/blender/agent_simple.py +133 -135
  16. gaia/agents/blender/app.py +211 -211
  17. gaia/agents/blender/app_simple.py +41 -41
  18. gaia/agents/blender/core/__init__.py +16 -16
  19. gaia/agents/blender/core/materials.py +506 -506
  20. gaia/agents/blender/core/objects.py +316 -316
  21. gaia/agents/blender/core/rendering.py +225 -225
  22. gaia/agents/blender/core/scene.py +220 -220
  23. gaia/agents/blender/core/view.py +146 -146
  24. gaia/agents/chat/__init__.py +9 -9
  25. gaia/agents/chat/agent.py +835 -835
  26. gaia/agents/chat/app.py +1058 -1058
  27. gaia/agents/chat/session.py +508 -508
  28. gaia/agents/chat/tools/__init__.py +15 -15
  29. gaia/agents/chat/tools/file_tools.py +96 -96
  30. gaia/agents/chat/tools/rag_tools.py +1729 -1729
  31. gaia/agents/chat/tools/shell_tools.py +436 -436
  32. gaia/agents/code/__init__.py +7 -7
  33. gaia/agents/code/agent.py +549 -549
  34. gaia/agents/code/cli.py +377 -0
  35. gaia/agents/code/models.py +135 -135
  36. gaia/agents/code/orchestration/__init__.py +24 -24
  37. gaia/agents/code/orchestration/checklist_executor.py +1763 -1763
  38. gaia/agents/code/orchestration/checklist_generator.py +713 -713
  39. gaia/agents/code/orchestration/factories/__init__.py +9 -9
  40. gaia/agents/code/orchestration/factories/base.py +63 -63
  41. gaia/agents/code/orchestration/factories/nextjs_factory.py +118 -118
  42. gaia/agents/code/orchestration/factories/python_factory.py +106 -106
  43. gaia/agents/code/orchestration/orchestrator.py +841 -841
  44. gaia/agents/code/orchestration/project_analyzer.py +391 -391
  45. gaia/agents/code/orchestration/steps/__init__.py +67 -67
  46. gaia/agents/code/orchestration/steps/base.py +188 -188
  47. gaia/agents/code/orchestration/steps/error_handler.py +314 -314
  48. gaia/agents/code/orchestration/steps/nextjs.py +828 -828
  49. gaia/agents/code/orchestration/steps/python.py +307 -307
  50. gaia/agents/code/orchestration/template_catalog.py +469 -469
  51. gaia/agents/code/orchestration/workflows/__init__.py +14 -14
  52. gaia/agents/code/orchestration/workflows/base.py +80 -80
  53. gaia/agents/code/orchestration/workflows/nextjs.py +186 -186
  54. gaia/agents/code/orchestration/workflows/python.py +94 -94
  55. gaia/agents/code/prompts/__init__.py +11 -11
  56. gaia/agents/code/prompts/base_prompt.py +77 -77
  57. gaia/agents/code/prompts/code_patterns.py +2036 -2036
  58. gaia/agents/code/prompts/nextjs_prompt.py +40 -40
  59. gaia/agents/code/prompts/python_prompt.py +109 -109
  60. gaia/agents/code/schema_inference.py +365 -365
  61. gaia/agents/code/system_prompt.py +41 -41
  62. gaia/agents/code/tools/__init__.py +42 -42
  63. gaia/agents/code/tools/cli_tools.py +1138 -1138
  64. gaia/agents/code/tools/code_formatting.py +319 -319
  65. gaia/agents/code/tools/code_tools.py +769 -769
  66. gaia/agents/code/tools/error_fixing.py +1347 -1347
  67. gaia/agents/code/tools/external_tools.py +180 -180
  68. gaia/agents/code/tools/file_io.py +845 -845
  69. gaia/agents/code/tools/prisma_tools.py +190 -190
  70. gaia/agents/code/tools/project_management.py +1016 -1016
  71. gaia/agents/code/tools/testing.py +321 -321
  72. gaia/agents/code/tools/typescript_tools.py +122 -122
  73. gaia/agents/code/tools/validation_parsing.py +461 -461
  74. gaia/agents/code/tools/validation_tools.py +806 -806
  75. gaia/agents/code/tools/web_dev_tools.py +1758 -1758
  76. gaia/agents/code/validators/__init__.py +16 -16
  77. gaia/agents/code/validators/antipattern_checker.py +241 -241
  78. gaia/agents/code/validators/ast_analyzer.py +197 -197
  79. gaia/agents/code/validators/requirements_validator.py +145 -145
  80. gaia/agents/code/validators/syntax_validator.py +171 -171
  81. gaia/agents/docker/__init__.py +7 -7
  82. gaia/agents/docker/agent.py +642 -642
  83. gaia/agents/emr/__init__.py +8 -8
  84. gaia/agents/emr/agent.py +1506 -1506
  85. gaia/agents/emr/cli.py +1322 -1322
  86. gaia/agents/emr/constants.py +475 -475
  87. gaia/agents/emr/dashboard/__init__.py +4 -4
  88. gaia/agents/emr/dashboard/server.py +1974 -1974
  89. gaia/agents/jira/__init__.py +11 -11
  90. gaia/agents/jira/agent.py +894 -894
  91. gaia/agents/jira/jql_templates.py +299 -299
  92. gaia/agents/routing/__init__.py +7 -7
  93. gaia/agents/routing/agent.py +567 -570
  94. gaia/agents/routing/system_prompt.py +75 -75
  95. gaia/agents/summarize/__init__.py +11 -0
  96. gaia/agents/summarize/agent.py +885 -0
  97. gaia/agents/summarize/prompts.py +129 -0
  98. gaia/api/__init__.py +23 -23
  99. gaia/api/agent_registry.py +238 -238
  100. gaia/api/app.py +305 -305
  101. gaia/api/openai_server.py +575 -575
  102. gaia/api/schemas.py +186 -186
  103. gaia/api/sse_handler.py +373 -373
  104. gaia/apps/__init__.py +4 -4
  105. gaia/apps/llm/__init__.py +6 -6
  106. gaia/apps/llm/app.py +173 -169
  107. gaia/apps/summarize/app.py +116 -633
  108. gaia/apps/summarize/html_viewer.py +133 -133
  109. gaia/apps/summarize/pdf_formatter.py +284 -284
  110. gaia/audio/__init__.py +2 -2
  111. gaia/audio/audio_client.py +439 -439
  112. gaia/audio/audio_recorder.py +269 -269
  113. gaia/audio/kokoro_tts.py +599 -599
  114. gaia/audio/whisper_asr.py +432 -432
  115. gaia/chat/__init__.py +16 -16
  116. gaia/chat/app.py +430 -430
  117. gaia/chat/prompts.py +522 -522
  118. gaia/chat/sdk.py +1228 -1225
  119. gaia/cli.py +5481 -5621
  120. gaia/database/__init__.py +10 -10
  121. gaia/database/agent.py +176 -176
  122. gaia/database/mixin.py +290 -290
  123. gaia/database/testing.py +64 -64
  124. gaia/eval/batch_experiment.py +2332 -2332
  125. gaia/eval/claude.py +542 -542
  126. gaia/eval/config.py +37 -37
  127. gaia/eval/email_generator.py +512 -512
  128. gaia/eval/eval.py +3179 -3179
  129. gaia/eval/groundtruth.py +1130 -1130
  130. gaia/eval/transcript_generator.py +582 -582
  131. gaia/eval/webapp/README.md +167 -167
  132. gaia/eval/webapp/package-lock.json +875 -875
  133. gaia/eval/webapp/package.json +20 -20
  134. gaia/eval/webapp/public/app.js +3402 -3402
  135. gaia/eval/webapp/public/index.html +87 -87
  136. gaia/eval/webapp/public/styles.css +3661 -3661
  137. gaia/eval/webapp/server.js +415 -415
  138. gaia/eval/webapp/test-setup.js +72 -72
  139. gaia/llm/__init__.py +9 -2
  140. gaia/llm/base_client.py +60 -0
  141. gaia/llm/exceptions.py +12 -0
  142. gaia/llm/factory.py +70 -0
  143. gaia/llm/lemonade_client.py +3236 -3221
  144. gaia/llm/lemonade_manager.py +294 -294
  145. gaia/llm/providers/__init__.py +9 -0
  146. gaia/llm/providers/claude.py +108 -0
  147. gaia/llm/providers/lemonade.py +120 -0
  148. gaia/llm/providers/openai_provider.py +79 -0
  149. gaia/llm/vlm_client.py +382 -382
  150. gaia/logger.py +189 -189
  151. gaia/mcp/agent_mcp_server.py +245 -245
  152. gaia/mcp/blender_mcp_client.py +138 -138
  153. gaia/mcp/blender_mcp_server.py +648 -648
  154. gaia/mcp/context7_cache.py +332 -332
  155. gaia/mcp/external_services.py +518 -518
  156. gaia/mcp/mcp_bridge.py +811 -550
  157. gaia/mcp/servers/__init__.py +6 -6
  158. gaia/mcp/servers/docker_mcp.py +83 -83
  159. gaia/perf_analysis.py +361 -0
  160. gaia/rag/__init__.py +10 -10
  161. gaia/rag/app.py +293 -293
  162. gaia/rag/demo.py +304 -304
  163. gaia/rag/pdf_utils.py +235 -235
  164. gaia/rag/sdk.py +2194 -2194
  165. gaia/security.py +163 -163
  166. gaia/talk/app.py +289 -289
  167. gaia/talk/sdk.py +538 -538
  168. gaia/testing/__init__.py +87 -87
  169. gaia/testing/assertions.py +330 -330
  170. gaia/testing/fixtures.py +333 -333
  171. gaia/testing/mocks.py +493 -493
  172. gaia/util.py +46 -46
  173. gaia/utils/__init__.py +33 -33
  174. gaia/utils/file_watcher.py +675 -675
  175. gaia/utils/parsing.py +223 -223
  176. gaia/version.py +100 -100
  177. amd_gaia-0.14.3.dist-info/RECORD +0 -168
  178. gaia/agents/code/app.py +0 -266
  179. gaia/llm/llm_client.py +0 -729
  180. {amd_gaia-0.14.3.dist-info → amd_gaia-0.15.1.dist-info}/WHEEL +0 -0
  181. {amd_gaia-0.14.3.dist-info → amd_gaia-0.15.1.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