ailabs/puaros
1
0
Fork 0
mirror of https://github.com/samiyev/puaros.git synced 2025-12-27 23:06:54 +05:00
Code Issues Packages Projects Releases Wiki Activity

feat(ipuaro): add enum value definitions to initial context

Browse Source 
Extract enum declarations with member names and values from TypeScript
AST and display them in the initial LLM context. This allows the LLM
to know valid enum values without making tool calls.

Features:
- Numeric values (Active=1)
- String values (Admin="admin")
- Implicit values (Up, Down)
- Negative numbers (Cold=-10)
- const enum modifier
- export enum modifier
- Long enum truncation (>100 chars)

Adds EnumInfo and EnumMemberInfo interfaces, extractEnum() method in
ASTParser, formatEnumSignature() in prompts.ts, and 17 new unit tests.
This commit is contained in:
imfozilbek
2025-12-05 13:14:51 +05:00
parent 806c9281b0
commit 5a22cd5c9b
7 changed files with 575 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

285
packages/ipuaro/tests/unit/infrastructure/llm/prompts.test.ts
View File

@@ -1191,9 +1191,7 @@ describe("prompts", () => {
const context = buildInitialContext(structure, asts)
expect(context).toContain(
"interface AdminUser extends User { role: string }",
)
expect(context).toContain("interface AdminUser extends User { role: string }")
})
it("should format interface with readonly fields", () => {
@@ -1494,4 +1492,285 @@ describe("prompts", () => {
expect(context).toContain("- type Handler = (event: Event) => void")
})
})
describe("enum definitions (0.24.3)", () => {
it("should format enum with numeric values", () => {
const structure: ProjectStructure = {
name: "test",
rootPath: "/test",
files: ["enums.ts"],
directories: [],
}
const asts = new Map<string, FileAST>([
[
"enums.ts",
{
imports: [],
exports: [],
functions: [],
classes: [],
interfaces: [],
typeAliases: [],
enums: [
{
name: "Status",
lineStart: 1,
lineEnd: 5,
members: [
{ name: "Active", value: 1 },
{ name: "Inactive", value: 0 },
{ name: "Pending", value: 2 },
],
isExported: true,
isConst: false,
},
],
parseError: false,
},
],
])
const context = buildInitialContext(structure, asts)
expect(context).toContain("- enum Status { Active=1, Inactive=0, Pending=2 }")
})
it("should format enum with string values", () => {
const structure: ProjectStructure = {
name: "test",
rootPath: "/test",
files: ["enums.ts"],
directories: [],
}
const asts = new Map<string, FileAST>([
[
"enums.ts",
{
imports: [],
exports: [],
functions: [],
classes: [],
interfaces: [],
typeAliases: [],
enums: [
{
name: "Role",
lineStart: 1,
lineEnd: 5,
members: [
{ name: "Admin", value: "admin" },
{ name: "User", value: "user" },
],
isExported: true,
isConst: false,
},
],
parseError: false,
},
],
])
const context = buildInitialContext(structure, asts)
expect(context).toContain('- enum Role { Admin="admin", User="user" }')
})
it("should format const enum", () => {
const structure: ProjectStructure = {
name: "test",
rootPath: "/test",
files: ["enums.ts"],
directories: [],
}
const asts = new Map<string, FileAST>([
[
"enums.ts",
{
imports: [],
exports: [],
functions: [],
classes: [],
interfaces: [],
typeAliases: [],
enums: [
{
name: "HttpStatus",
lineStart: 1,
lineEnd: 5,
members: [
{ name: "OK", value: 200 },
{ name: "NotFound", value: 404 },
],
isExported: true,
isConst: true,
},
],
parseError: false,
},
],
])
const context = buildInitialContext(structure, asts)
expect(context).toContain("- const enum HttpStatus { OK=200, NotFound=404 }")
})
it("should format enum without explicit values", () => {
const structure: ProjectStructure = {
name: "test",
rootPath: "/test",
files: ["enums.ts"],
directories: [],
}
const asts = new Map<string, FileAST>([
[
"enums.ts",
{
imports: [],
exports: [],
functions: [],
classes: [],
interfaces: [],
typeAliases: [],
enums: [
{
name: "Direction",
lineStart: 1,
lineEnd: 5,
members: [
{ name: "Up", value: undefined },
{ name: "Down", value: undefined },
],
isExported: true,
isConst: false,
},
],
parseError: false,
},
],
])
const context = buildInitialContext(structure, asts)
expect(context).toContain("- enum Direction { Up, Down }")
})
it("should format empty enum", () => {
const structure: ProjectStructure = {
name: "test",
rootPath: "/test",
files: ["enums.ts"],
directories: [],
}
const asts = new Map<string, FileAST>([
[
"enums.ts",
{
imports: [],
exports: [],
functions: [],
classes: [],
interfaces: [],
typeAliases: [],
enums: [
{
name: "Empty",
lineStart: 1,
lineEnd: 1,
members: [],
isExported: true,
isConst: false,
},
],
parseError: false,
},
],
])
const context = buildInitialContext(structure, asts)
expect(context).toContain("- enum Empty")
})
it("should include enum in compact format", () => {
const structure: ProjectStructure = {
name: "test",
rootPath: "/test",
files: ["enums.ts"],
directories: [],
}
const asts = new Map<string, FileAST>([
[
"enums.ts",
{
imports: [],
exports: [],
functions: [],
classes: [],
interfaces: [],
typeAliases: [],
enums: [
{
name: "Status",
lineStart: 1,
lineEnd: 5,
members: [{ name: "Active", value: 1 }],
isExported: true,
isConst: false,
},
],
parseError: false,
},
],
])
const context = buildInitialContext(structure, asts, undefined, {
includeSignatures: false,
})
expect(context).toContain("enum: Status")
})
it("should truncate long enum definitions", () => {
const structure: ProjectStructure = {
name: "test",
rootPath: "/test",
files: ["enums.ts"],
directories: [],
}
const asts = new Map<string, FileAST>([
[
"enums.ts",
{
imports: [],
exports: [],
functions: [],
classes: [],
interfaces: [],
typeAliases: [],
enums: [
{
name: "VeryLongEnumName",
lineStart: 1,
lineEnd: 20,
members: [
{ name: "FirstVeryLongMemberName", value: "first_value" },
{ name: "SecondVeryLongMemberName", value: "second_value" },
{ name: "ThirdVeryLongMemberName", value: "third_value" },
],
isExported: true,
isConst: false,
},
],
parseError: false,
},
],
])
const context = buildInitialContext(structure, asts)
expect(context).toContain("...")
expect(context).toContain("- enum VeryLongEnumName")
})
})
})