@superblocksteam/vite-plugin-file-sync 2.0.26-next.11 → 2.0.26-next.13

package/dist/ai-service/integrations/metadata/graphql-based.d.ts

@@ -0,0 +1,14 @@
+import { IntegrationMetadata } from "./index.js";
+import type { LLMProvider } from "../../llm/types.js";
+export declare class GraphQLIntegrationMetadata extends IntegrationMetadata {
+    static refineMetadataForPrompt({ metadata, prompt, llmProvider, }: {
+        metadata: Record<string, any>;
+        prompt: string;
+        llmProvider: LLMProvider;
+    }): Promise<Record<string, any>>;
+    static refineDbSchema(metadata: Record<string, any>, prompt: string, llmProvider: LLMProvider): Promise<Record<string, any>>;
+    static prefilterSchemaForLLM(schema: any): any;
+    static filterGraphQLSchema(schema: any, typeNamesToKeep: string[]): any;
+    static simplifyTypeRef(typeRef: any): any;
+}
+//# sourceMappingURL=graphql-based.d.ts.map

package/dist/ai-service/integrations/metadata/graphql-based.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"graphql-based.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/integrations/metadata/graphql-based.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEtD,qBAAa,0BAA2B,SAAQ,mBAAmB;WACpC,uBAAuB,CAAC,EACnD,QAAQ,EACR,MAAM,EACN,WAAW,GACZ,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC9B,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,WAAW,CAAC;KAC1B,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;WAOnB,cAAc,CACzB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAC7B,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW,GACvB,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IA0E/B,MAAM,CAAC,qBAAqB,CAAC,MAAM,EAAE,GAAG,GAAG,GAAG;IA0G9C,MAAM,CAAC,mBAAmB,CAAC,MAAM,EAAE,GAAG,EAAE,eAAe,EAAE,MAAM,EAAE,GAAG,GAAG;IA8KvE,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,GAAG,GAAG,GAAG;CAgB1C"}

package/dist/ai-service/integrations/metadata/graphql-based.js

@@ -0,0 +1,325 @@
+import { generateObject } from "ai";
+import ddTrace from "dd-trace";
+import { buildGraphQlSchemaIntentPrompt } from "../../prompt-builder-service/index.js";
+import { IntegrationMetadata } from "./index.js";
+export class GraphQLIntegrationMetadata extends IntegrationMetadata {
+    static async refineMetadataForPrompt({ metadata, prompt, llmProvider, }) {
+        // 1. reduce size
+        // 2. ask llm to trim
+        // 3. profit
+        return await this.refineDbSchema(metadata, prompt, llmProvider);
+    }
+    static async refineDbSchema(metadata, prompt, llmProvider) {
+        const schema = metadata.graphql?.data?.__schema;
+        // skip if we have empty metadata
+        if (Object.keys(metadata).length === 0 ||
+            !schema ||
+            Object.keys(schema).length === 0) {
+            return {};
+        }
+        // get a minified version to send to the llm
+        const prefilteredSchema = this.prefilterSchemaForLLM(schema);
+        const { prompt: refineSchemaPrompt, schema: zodSchema } = buildGraphQlSchemaIntentPrompt(prompt, prefilteredSchema);
+        const { llmobs } = ddTrace;
+        const model = llmProvider.modelForTask("openApiIntent");
+        // TODO: @joeyagreco - annotate span with integration type?
+        const wrappedGenerateObject = llmobs.wrap({
+            kind: "llm",
+            name: "refineDbSchema",
+            modelName: model.modelId,
+            modelProvider: model.provider,
+        }, async (args) => {
+            const result = await generateObject(args);
+            // https://docs.datadoghq.com/llm_observability/sdk/?tab=nodejs#annotating-a-span
+            llmobs.annotate({
+                inputData: [{ role: "user", content: args.prompt }],
+                outputData: [
+                    { role: "assistant", content: JSON.stringify(result.object) },
+                ],
+                metrics: {
+                    input_tokens: result.usage?.promptTokens,
+                    output_tokens: result.usage?.completionTokens,
+                    total_tokens: result.usage?.totalTokens,
+                },
+                tags: {
+                    finishReason: result.finishReason,
+                },
+            });
+            return result;
+        });
+        const result = await wrappedGenerateObject({
+            model,
+            schema: zodSchema,
+            prompt: refineSchemaPrompt,
+        });
+        const object = result.object;
+        // filter the graphql schema to only include the refined types
+        return {
+            ...metadata,
+            graphql: {
+                ...metadata.graphql,
+                data: {
+                    ...metadata.graphql.data,
+                    __schema: this.filterGraphQLSchema(metadata.graphql.data.__schema, object.type_names),
+                },
+            },
+        };
+    }
+    static prefilterSchemaForLLM(schema) {
+        if (!schema?.types || !Array.isArray(schema.types)) {
+            return schema;
+        }
+        // extract only object type names and their descriptions (if short)
+        const objectTypes = schema.types
+            .filter((type) => type.kind === "OBJECT" &&
+            !type.name.startsWith("__") &&
+            type.name !== "Query" &&
+            type.name !== "Mutation" &&
+            type.name !== "Subscription")
+            .map((type) => ({
+            name: type.name,
+            ...(type.description &&
+                type.description.length < 80 && {
+                description: type.description,
+            }),
+        }));
+        // include query type with its fields and arguments to preserve argument context
+        const queryType = schema.types.find((type) => type.name === "Query");
+        let queryFields = [];
+        if (queryType?.fields) {
+            queryFields = queryType.fields.map((field) => ({
+                name: field.name,
+                ...(field.description &&
+                    field.description.length < 100 && {
+                    description: field.description,
+                }),
+                ...(field.args &&
+                    field.args.length > 0 && {
+                    args: field.args.map((arg) => ({
+                        name: arg.name,
+                        type: this.simplifyTypeRef(arg.type),
+                        ...(arg.description &&
+                            arg.description.length < 100 && {
+                            description: arg.description,
+                        }),
+                    })),
+                }),
+            }));
+        }
+        // include enum and input types that might be used in query arguments
+        const enumTypes = schema.types
+            .filter((type) => type.kind === "ENUM" && !type.name.startsWith("__"))
+            .map((type) => ({
+            name: type.name,
+            kind: type.kind,
+            ...(type.description &&
+                type.description.length < 80 && {
+                description: type.description,
+            }),
+            ...(type.enumValues && {
+                enumValues: type.enumValues.map((val) => ({
+                    name: val.name,
+                    ...(val.description &&
+                        val.description.length < 60 && {
+                        description: val.description,
+                    }),
+                })),
+            }),
+        }));
+        const inputTypes = schema.types
+            .filter((type) => type.kind === "INPUT_OBJECT" && !type.name.startsWith("__"))
+            .map((type) => ({
+            name: type.name,
+            kind: type.kind,
+            ...(type.description &&
+                type.description.length < 80 && {
+                description: type.description,
+            }),
+            ...(type.inputFields && {
+                inputFields: type.inputFields.map((field) => ({
+                    name: field.name,
+                    type: this.simplifyTypeRef(field.type),
+                    ...(field.description &&
+                        field.description.length < 60 && {
+                        description: field.description,
+                    }),
+                })),
+            }),
+        }));
+        return {
+            queryType: schema.queryType,
+            mutationType: schema.mutationType,
+            subscriptionType: schema.subscriptionType,
+            objectTypes,
+            queryFields,
+            enumTypes,
+            inputTypes,
+            totalTypesInOriginalSchema: schema.types.length,
+        };
+    }
+    static filterGraphQLSchema(schema, typeNamesToKeep) {
+        if (!schema?.types || !Array.isArray(schema.types)) {
+            return schema;
+        }
+        // create a set for faster lookups
+        const typesToKeep = new Set(typeNamesToKeep);
+        // always keep scalar types and introspection types as they're fundamental
+        const alwaysKeepTypes = new Set([
+            "String",
+            "Int",
+            "Float",
+            "Boolean",
+            "ID",
+        ]);
+        // find types that are referenced by the types we want to keep
+        const referencedTypes = new Set();
+        const findReferencedTypes = (type) => {
+            if (!type)
+                return;
+            if (type.name && !type.name.startsWith("__")) {
+                referencedTypes.add(type.name);
+            }
+            // check fields
+            if (type.fields) {
+                type.fields.forEach((field) => {
+                    if (field.type) {
+                        addReferencedTypesFromTypeRef(field.type);
+                    }
+                    // check args
+                    if (field.args) {
+                        field.args.forEach((arg) => {
+                            if (arg.type) {
+                                addReferencedTypesFromTypeRef(arg.type);
+                            }
+                        });
+                    }
+                });
+            }
+            // check input fields
+            if (type.inputFields) {
+                type.inputFields.forEach((inputField) => {
+                    if (inputField.type) {
+                        addReferencedTypesFromTypeRef(inputField.type);
+                    }
+                });
+            }
+            // check interfaces
+            if (type.interfaces) {
+                type.interfaces.forEach((iface) => {
+                    if (iface.name) {
+                        referencedTypes.add(iface.name);
+                    }
+                });
+            }
+            // check possible types (for unions)
+            if (type.possibleTypes) {
+                type.possibleTypes.forEach((possibleType) => {
+                    if (possibleType.name) {
+                        referencedTypes.add(possibleType.name);
+                    }
+                });
+            }
+        };
+        const addReferencedTypesFromTypeRef = (typeRef) => {
+            if (!typeRef)
+                return;
+            if (typeRef.name && !typeRef.name.startsWith("__")) {
+                referencedTypes.add(typeRef.name);
+            }
+            // handle wrapped types (list, nonnull)
+            if (typeRef.ofType) {
+                addReferencedTypesFromTypeRef(typeRef.ofType);
+            }
+        };
+        // first, find all directly referenced types from the types we want to keep
+        schema.types.forEach((type) => {
+            if (typesToKeep.has(type.name)) {
+                findReferencedTypes(type);
+            }
+        });
+        // also include query type and its referenced types since it's the entry point
+        const queryType = schema.types.find((type) => type.name === "Query");
+        if (queryType) {
+            findReferencedTypes(queryType);
+        }
+        // filter types to only include:
+        // 1. types that were explicitly requested
+        // 2. types that are referenced by requested types
+        // 3. scalar types that are always needed
+        // 4. introspection types (start with __)
+        const filteredTypes = schema.types.filter((type) => {
+            return (typesToKeep.has(type.name) ||
+                referencedTypes.has(type.name) ||
+                alwaysKeepTypes.has(type.name) ||
+                type.name.startsWith("__") ||
+                type.name === "Query"); // always keep query type
+        });
+        // further strip down the filtered types to reduce size
+        const minimalTypes = filteredTypes.map((type) => {
+            // keep only essential properties for final system prompt
+            const minimalType = {
+                kind: type.kind,
+                name: type.name,
+                ...(type.description &&
+                    type.description.length < 100 && { description: type.description }),
+            };
+            // for object types, include field names and basic type info
+            // for query type, preserve arguments to maintain context for api generation
+            if (type.kind === "OBJECT" && type.fields) {
+                minimalType.fields = type.fields.map((field) => {
+                    const minimalField = {
+                        name: field.name,
+                        type: this.simplifyTypeRef(field.type),
+                    };
+                    // preserve arguments for query type fields
+                    if (type.name === "Query" && field.args && field.args.length > 0) {
+                        minimalField.args = field.args.map((arg) => ({
+                            name: arg.name,
+                            type: this.simplifyTypeRef(arg.type),
+                            ...(arg.description &&
+                                arg.description.length < 100 && {
+                                description: arg.description,
+                            }),
+                        }));
+                    }
+                    return minimalField;
+                });
+            }
+            // for input types, include only input field names and basic type info
+            if (type.kind === "INPUT_OBJECT" && type.inputFields) {
+                minimalType.inputFields = type.inputFields.map((field) => ({
+                    name: field.name,
+                    type: this.simplifyTypeRef(field.type),
+                }));
+            }
+            // for enums, keep enum values but remove descriptions
+            if (type.kind === "ENUM" && type.enumValues) {
+                minimalType.enumValues = type.enumValues.map((enumValue) => ({
+                    name: enumValue.name,
+                }));
+            }
+            return minimalType;
+        });
+        return {
+            queryType: schema.queryType,
+            mutationType: schema.mutationType,
+            subscriptionType: schema.subscriptionType,
+            types: minimalTypes,
+        };
+    }
+    static simplifyTypeRef(typeRef) {
+        if (!typeRef)
+            return null;
+        // simplify type reference by keeping only essential info
+        const simplified = {
+            kind: typeRef.kind,
+            ...(typeRef.name && { name: typeRef.name }),
+        };
+        // recursively simplify wrapped types (nonnull, list)
+        if (typeRef.ofType) {
+            simplified.ofType = this.simplifyTypeRef(typeRef.ofType);
+        }
+        return simplified;
+    }
+}
+//# sourceMappingURL=graphql-based.js.map

package/dist/ai-service/integrations/metadata/graphql-based.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"graphql-based.js","sourceRoot":"","sources":["../../../../src/ai-service/integrations/metadata/graphql-based.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAA6B,MAAM,IAAI,CAAC;AAC/D,OAAO,OAAO,MAAM,UAAU,CAAC;AAC/B,OAAO,EAAE,8BAA8B,EAAE,MAAM,uCAAuC,CAAC;AACvF,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAGjD,MAAM,OAAO,0BAA2B,SAAQ,mBAAmB;IAC1D,MAAM,CAAU,KAAK,CAAC,uBAAuB,CAAC,EACnD,QAAQ,EACR,MAAM,EACN,WAAW,GAKZ;QACC,iBAAiB;QACjB,qBAAqB;QACrB,YAAY;QACZ,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;IAClE,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,cAAc,CACzB,QAA6B,EAC7B,MAAc,EACd,WAAwB;QAExB,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,QAAQ,CAAC;QAChD,iCAAiC;QACjC,IACE,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,KAAK,CAAC;YAClC,CAAC,MAAM;YACP,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC,EAChC,CAAC;YACD,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,4CAA4C;QAC5C,MAAM,iBAAiB,GAAG,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAC7D,MAAM,EAAE,MAAM,EAAE,kBAAkB,EAAE,MAAM,EAAE,SAAS,EAAE,GACrD,8BAA8B,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC;QAE5D,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAE3B,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC;QAExD,2DAA2D;QAC3D,MAAM,qBAAqB,GAAG,MAAM,CAAC,IAAI,CACvC;YACE,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE,KAAK,CAAC,OAAO;YACxB,aAAa,EAAE,KAAK,CAAC,QAAQ;SAC9B,EACD,KAAK,EAAE,IAAI,EAAsC,EAAE;YACjD,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,CAAC;YAE1C,iFAAiF;YACjF,MAAM,CAAC,QAAQ,CAAC;gBACd,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;gBACnD,UAAU,EAAE;oBACV,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE;iBAC9D;gBACD,OAAO,EAAE;oBACP,YAAY,EAAE,MAAM,CAAC,KAAK,EAAE,YAAY;oBACxC,aAAa,EAAE,MAAM,CAAC,KAAK,EAAE,gBAAgB;oBAC7C,YAAY,EAAE,MAAM,CAAC,KAAK,EAAE,WAAW;iBACxC;gBACD,IAAI,EAAE;oBACJ,YAAY,EAAE,MAAM,CAAC,YAAY;iBAClC;aACF,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC,CACF,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC;YACzC,KAAK;YACL,MAAM,EAAE,SAAS;YACjB,MAAM,EAAE,kBAAkB;SAC3B,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAE7B,8DAA8D;QAC9D,OAAO;YACL,GAAG,QAAQ;YACX,OAAO,EAAE;gBACP,GAAG,QAAQ,CAAC,OAAO;gBACnB,IAAI,EAAE;oBACJ,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI;oBACxB,QAAQ,EAAE,IAAI,CAAC,mBAAmB,CAChC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,EAC9B,MAAM,CAAC,UAAU,CAClB;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,qBAAqB,CAAC,MAAW;QACtC,IAAI,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;YACnD,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,mEAAmE;QACnE,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK;aAC7B,MAAM,CACL,CAAC,IAAS,EAAE,EAAE,CACZ,IAAI,CAAC,IAAI,KAAK,QAAQ;YACtB,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;YAC3B,IAAI,CAAC,IAAI,KAAK,OAAO;YACrB,IAAI,CAAC,IAAI,KAAK,UAAU;YACxB,IAAI,CAAC,IAAI,KAAK,cAAc,CAC/B;aACA,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,CAAC;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,GAAG,CAAC,IAAI,CAAC,WAAW;gBAClB,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,EAAE,IAAI;gBAC9B,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC;SACL,CAAC,CAAC,CAAC;QAEN,gFAAgF;QAChF,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;QAC1E,IAAI,WAAW,GAAU,EAAE,CAAC;QAC5B,IAAI,SAAS,EAAE,MAAM,EAAE,CAAC;YACtB,WAAW,GAAG,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,CAAC;gBAClD,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,GAAG,CAAC,KAAK,CAAC,WAAW;oBACnB,KAAK,CAAC,WAAW,CAAC,MAAM,GAAG,GAAG,IAAI;oBAChC,WAAW,EAAE,KAAK,CAAC,WAAW;iBAC/B,CAAC;gBACJ,GAAG,CAAC,KAAK,CAAC,IAAI;oBACZ,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI;oBACvB,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,CAAC;wBAClC,IAAI,EAAE,GAAG,CAAC,IAAI;wBACd,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC;wBACpC,GAAG,CAAC,GAAG,CAAC,WAAW;4BACjB,GAAG,CAAC,WAAW,CAAC,MAAM,GAAG,GAAG,IAAI;4BAC9B,WAAW,EAAE,GAAG,CAAC,WAAW;yBAC7B,CAAC;qBACL,CAAC,CAAC;iBACJ,CAAC;aACL,CAAC,CAAC,CAAC;QACN,CAAC;QAED,qEAAqE;QACrE,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK;aAC3B,MAAM,CACL,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CACnE;aACA,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,CAAC;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,GAAG,CAAC,IAAI,CAAC,WAAW;gBAClB,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,EAAE,IAAI;gBAC9B,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC;YACJ,GAAG,CAAC,IAAI,CAAC,UAAU,IAAI;gBACrB,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,CAAC;oBAC7C,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,GAAG,CAAC,GAAG,CAAC,WAAW;wBACjB,GAAG,CAAC,WAAW,CAAC,MAAM,GAAG,EAAE,IAAI;wBAC7B,WAAW,EAAE,GAAG,CAAC,WAAW;qBAC7B,CAAC;iBACL,CAAC,CAAC;aACJ,CAAC;SACH,CAAC,CAAC,CAAC;QAEN,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK;aAC5B,MAAM,CACL,CAAC,IAAS,EAAE,EAAE,CACZ,IAAI,CAAC,IAAI,KAAK,cAAc,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAC9D;aACA,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,CAAC;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,GAAG,CAAC,IAAI,CAAC,WAAW;gBAClB,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,EAAE,IAAI;gBAC9B,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC;YACJ,GAAG,CAAC,IAAI,CAAC,WAAW,IAAI;gBACtB,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,CAAC;oBACjD,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC;oBACtC,GAAG,CAAC,KAAK,CAAC,WAAW;wBACnB,KAAK,CAAC,WAAW,CAAC,MAAM,GAAG,EAAE,IAAI;wBAC/B,WAAW,EAAE,KAAK,CAAC,WAAW;qBAC/B,CAAC;iBACL,CAAC,CAAC;aACJ,CAAC;SACH,CAAC,CAAC,CAAC;QAEN,OAAO;YACL,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;YACzC,WAAW;YACX,WAAW;YACX,SAAS;YACT,UAAU;YACV,0BAA0B,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM;SAChD,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,mBAAmB,CAAC,MAAW,EAAE,eAAyB;QAC/D,IAAI,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;YACnD,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,kCAAkC;QAClC,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,eAAe,CAAC,CAAC;QAE7C,0EAA0E;QAC1E,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC;YAC9B,QAAQ;YACR,KAAK;YACL,OAAO;YACP,SAAS;YACT,IAAI;SACL,CAAC,CAAC;QAEH,8DAA8D;QAC9D,MAAM,eAAe,GAAG,IAAI,GAAG,EAAU,CAAC;QAE1C,MAAM,mBAAmB,GAAG,CAAC,IAAS,EAAE,EAAE;YACxC,IAAI,CAAC,IAAI;gBAAE,OAAO;YAElB,IAAI,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC7C,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,CAAC;YAED,eAAe;YACf,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAU,EAAE,EAAE;oBACjC,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;wBACf,6BAA6B,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAC5C,CAAC;oBACD,aAAa;oBACb,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;wBACf,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAQ,EAAE,EAAE;4BAC9B,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC;gCACb,6BAA6B,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;4BAC1C,CAAC;wBACH,CAAC,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC;YAED,qBAAqB;YACrB,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;gBACrB,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,UAAe,EAAE,EAAE;oBAC3C,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;wBACpB,6BAA6B,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;oBACjD,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC;YAED,mBAAmB;YACnB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACpB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,KAAU,EAAE,EAAE;oBACrC,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;wBACf,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAClC,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC;YAED,oCAAoC;YACpC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;gBACvB,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,YAAiB,EAAE,EAAE;oBAC/C,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC;wBACtB,eAAe,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;oBACzC,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,6BAA6B,GAAG,CAAC,OAAY,EAAE,EAAE;YACrD,IAAI,CAAC,OAAO;gBAAE,OAAO;YAErB,IAAI,OAAO,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnD,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACpC,CAAC;YAED,uCAAuC;YACvC,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,6BAA6B,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,CAAC;QACH,CAAC,CAAC;QAEF,2EAA2E;QAC3E,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAS,EAAE,EAAE;YACjC,IAAI,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/B,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,8EAA8E;QAC9E,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;QAC1E,IAAI,SAAS,EAAE,CAAC;YACd,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAED,gCAAgC;QAChC,0CAA0C;QAC1C,kDAAkD;QAClD,yCAAyC;QACzC,yCAAyC;QACzC,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAS,EAAE,EAAE;YACtD,OAAO,CACL,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC1B,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC9B,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC9B,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;gBAC1B,IAAI,CAAC,IAAI,KAAK,OAAO,CACtB,CAAC,CAAC,yBAAyB;QAC9B,CAAC,CAAC,CAAC;QAEH,uDAAuD;QACvD,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE;YACnD,yDAAyD;YACzD,MAAM,WAAW,GAAG;gBAClB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,GAAG,CAAC,IAAI,CAAC,WAAW;oBAClB,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,GAAG,IAAI,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC;aACtE,CAAC;YAEF,4DAA4D;YAC5D,4EAA4E;YAC5E,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC1C,WAAW,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAU,EAAE,EAAE;oBAClD,MAAM,YAAY,GAAQ;wBACxB,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC;qBACvC,CAAC;oBAEF,2CAA2C;oBAC3C,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACjE,YAAY,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,CAAC;4BAChD,IAAI,EAAE,GAAG,CAAC,IAAI;4BACd,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC;4BACpC,GAAG,CAAC,GAAG,CAAC,WAAW;gCACjB,GAAG,CAAC,WAAW,CAAC,MAAM,GAAG,GAAG,IAAI;gCAC9B,WAAW,EAAE,GAAG,CAAC,WAAW;6BAC7B,CAAC;yBACL,CAAC,CAAC,CAAC;oBACN,CAAC;oBAED,OAAO,YAAY,CAAC;gBACtB,CAAC,CAAC,CAAC;YACL,CAAC;YAED,sEAAsE;YACtE,IAAI,IAAI,CAAC,IAAI,KAAK,cAAc,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;gBACrD,WAAW,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,CAAC;oBAC9D,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC;iBACvC,CAAC,CAAC,CAAC;YACN,CAAC;YAED,sDAAsD;YACtD,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC5C,WAAW,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAc,EAAE,EAAE,CAAC,CAAC;oBAChE,IAAI,EAAE,SAAS,CAAC,IAAI;iBACrB,CAAC,CAAC,CAAC;YACN,CAAC;YAED,OAAO,WAAW,CAAC;QACrB,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;YACzC,KAAK,EAAE,YAAY;SACpB,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,eAAe,CAAC,OAAY;QACjC,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAE1B,yDAAyD;QACzD,MAAM,UAAU,GAAG;YACjB,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC;SAC5C,CAAC;QAEF,qDAAqD;QACrD,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3D,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;CACF"}

package/dist/ai-service/integrations/metadata/open-api.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"open-api.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/integrations/metadata/open-api.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,gBAAgB,EAEhB,KAAK,WAAW,EACjB,MAAM,2BAA2B,CAAC;AAEnC,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEtD,MAAM,MAAM,aAAa,GAAG,UAAU,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAEhE,qBAAa,0BAA2B,SAAQ,mBAAmB;WACpC,uBAAuB,CAAC,EACnD,QAAQ,EACR,MAAM,EACN,WAAW,GACZ,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC9B,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,WAAW,CAAC;KAC1B,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;WAaZ,uBAAuB,CAAC,EAC1C,WAAW,EACX,MAAM,EACN,WAAW,GACZ,EAAE;QACD,WAAW,EAAE,WAAW,CAAC;QACzB,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,WAAW,CAAC;KAC1B,GAAG,OAAO,CAAC,aAAa,CAAC;CAuD3B"}
+{"version":3,"file":"open-api.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/integrations/metadata/open-api.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,gBAAgB,EAEhB,KAAK,WAAW,EACjB,MAAM,2BAA2B,CAAC;AAEnC,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEtD,MAAM,MAAM,aAAa,GAAG,UAAU,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAEhE,qBAAa,0BAA2B,SAAQ,mBAAmB;WACpC,uBAAuB,CAAC,EACnD,QAAQ,EACR,MAAM,EACN,WAAW,GACZ,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC9B,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,WAAW,CAAC;KAC1B,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;WAaZ,uBAAuB,CAAC,EAC1C,WAAW,EACX,MAAM,EACN,WAAW,GACZ,EAAE;QACD,WAAW,EAAE,WAAW,CAAC;QACzB,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,WAAW,CAAC;KAC1B,GAAG,OAAO,CAAC,aAAa,CAAC;CAwD3B"}

package/dist/ai-service/integrations/metadata/open-api.js

@@ -31,7 +31,9 @@ export class OpenApiIntegrationMetadata extends IntegrationMetadata {
             name: "getRelevantOpenApiPaths",
             modelName: model.modelId,
             modelProvider: model.provider,
-        }, async (args) => {
+        }, 
+        // NOTE: @joeyagreco - should we be using args.schema here ?
+        async (args) => {
             const result = await generateObject(args);
             // https://docs.datadoghq.com/llm_observability/sdk/?tab=nodejs#annotating-a-span
             llmobs.annotate({

package/dist/ai-service/integrations/metadata/open-api.js.map

@@ -1 +1 @@
-{"version":3,"file":"open-api.js","sourceRoot":"","sources":["../../../../src/ai-service/integrations/metadata/open-api.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAA6B,MAAM,IAAI,CAAC;AAC/D,OAAO,OAAO,MAAM,UAAU,CAAC;AAC/B,OAAO,EACL,gBAAgB,EAChB,gBAAgB,GAEjB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,wBAAwB,EAAE,MAAM,uCAAuC,CAAC;AACjF,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAKjD,MAAM,OAAO,0BAA2B,SAAQ,mBAAmB;IAC1D,MAAM,CAAU,KAAK,CAAC,uBAAuB,CAAC,EACnD,QAAQ,EACR,MAAM,EACN,WAAW,GAKZ;QACC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC;YACrD,MAAM;YACN,WAAW,EAAE,QAAQ,CAAC,WAAW;YACjC,WAAW;SACZ,CAAC,CAAC;QAEH,OAAO;YACL,GAAG,QAAQ;YACX,WAAW;SACZ,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,EAC1C,WAAW,EACX,MAAM,EACN,WAAW,GAKZ;QACC,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO;gBACL,KAAK,EAAE,EAAE;gBACT,UAAU,EAAE,EAAE;aACf,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAC;QACpD,MAAM,EAAE,MAAM,EAAE,mBAAmB,EAAE,MAAM,EAAE,GAAG,wBAAwB,CACtE,MAAM,EACN,aAAa,CACd,CAAC;QACF,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAE3B,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC;QAExD,MAAM,qBAAqB,GAAG,MAAM,CAAC,IAAI,CACvC;YACE,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,yBAAyB;YAC/B,SAAS,EAAE,KAAK,CAAC,OAAO;YACxB,aAAa,EAAE,KAAK,CAAC,QAAQ;SAC9B,EACD,KAAK,EAAE,IAAI,EAAsC,EAAE;YACjD,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,CAAC;YAE1C,iFAAiF;YACjF,MAAM,CAAC,QAAQ,CAAC;gBACd,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;gBACnD,UAAU,EAAE;oBACV,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE;iBAC9D;gBACD,OAAO,EAAE;oBACP,YAAY,EAAE,MAAM,CAAC,KAAK,EAAE,YAAY;oBACxC,aAAa,EAAE,MAAM,CAAC,KAAK,EAAE,gBAAgB;oBAC7C,YAAY,EAAE,MAAM,CAAC,KAAK,EAAE,WAAW;iBACxC;gBACD,IAAI,EAAE;oBACJ,YAAY,EAAE,MAAM,CAAC,YAAY;iBAClC;aACF,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC,CACF,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,qBAAqB,CAAC;YAC7C,KAAK;YACL,MAAM;YACN,MAAM,EAAE,mBAAmB;SAC5B,CAAC,CAAC;QAEH,OAAO,gBAAgB,CAAC,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;IACrD,CAAC;CACF"}
+{"version":3,"file":"open-api.js","sourceRoot":"","sources":["../../../../src/ai-service/integrations/metadata/open-api.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAA6B,MAAM,IAAI,CAAC;AAC/D,OAAO,OAAO,MAAM,UAAU,CAAC;AAC/B,OAAO,EACL,gBAAgB,EAChB,gBAAgB,GAEjB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,wBAAwB,EAAE,MAAM,uCAAuC,CAAC;AACjF,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAKjD,MAAM,OAAO,0BAA2B,SAAQ,mBAAmB;IAC1D,MAAM,CAAU,KAAK,CAAC,uBAAuB,CAAC,EACnD,QAAQ,EACR,MAAM,EACN,WAAW,GAKZ;QACC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC;YACrD,MAAM;YACN,WAAW,EAAE,QAAQ,CAAC,WAAW;YACjC,WAAW;SACZ,CAAC,CAAC;QAEH,OAAO;YACL,GAAG,QAAQ;YACX,WAAW;SACZ,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,EAC1C,WAAW,EACX,MAAM,EACN,WAAW,GAKZ;QACC,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO;gBACL,KAAK,EAAE,EAAE;gBACT,UAAU,EAAE,EAAE;aACf,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAC;QACpD,MAAM,EAAE,MAAM,EAAE,mBAAmB,EAAE,MAAM,EAAE,GAAG,wBAAwB,CACtE,MAAM,EACN,aAAa,CACd,CAAC;QACF,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAE3B,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC;QAExD,MAAM,qBAAqB,GAAG,MAAM,CAAC,IAAI,CACvC;YACE,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,yBAAyB;YAC/B,SAAS,EAAE,KAAK,CAAC,OAAO;YACxB,aAAa,EAAE,KAAK,CAAC,QAAQ;SAC9B;QACD,4DAA4D;QAC5D,KAAK,EAAE,IAAI,EAAsC,EAAE;YACjD,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,CAAC;YAE1C,iFAAiF;YACjF,MAAM,CAAC,QAAQ,CAAC;gBACd,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;gBACnD,UAAU,EAAE;oBACV,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE;iBAC9D;gBACD,OAAO,EAAE;oBACP,YAAY,EAAE,MAAM,CAAC,KAAK,EAAE,YAAY;oBACxC,aAAa,EAAE,MAAM,CAAC,KAAK,EAAE,gBAAgB;oBAC7C,YAAY,EAAE,MAAM,CAAC,KAAK,EAAE,WAAW;iBACxC;gBACD,IAAI,EAAE;oBACJ,YAAY,EAAE,MAAM,CAAC,YAAY;iBAClC;aACF,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC,CACF,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,qBAAqB,CAAC;YAC7C,KAAK;YACL,MAAM;YACN,MAAM,EAAE,mBAAmB;SAC5B,CAAC,CAAC;QAEH,OAAO,gBAAgB,CAAC,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;IACrD,CAAC;CACF"}

package/dist/ai-service/integrations/store.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../../../src/ai-service/integrations/store.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAG1D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,KAAK,EACV,KAAK,EACL,uBAAuB,EACxB,MAAM,+BAA+B,CAAC;AACvC,OAAO,KAAK,EACV,cAAc,EAGf,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,uCAAuC,CAAC;AAEtF,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,YAAY,CAAmC;IAEjD,8BAA8B,CAClC,WAAW,EAAE,wBAAwB,EACrC,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,uBAAuB,GAC9B,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC;IA6B3B,mBAAmB,CAAC,EACxB,KAAK,EACL,WAAW,EACX,MAAM,EACN,WAAW,GACZ,EAAE;QACD,KAAK,EAAE,KAAK,CAAC;QACb,WAAW,EAAE,wBAAwB,CAAC;QACtC,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,WAAW,CAAC;KAC1B,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAqBhC,8BAA8B,CAC5B,WAAW,EAAE,wBAAwB,GACpC,OAAO,mBAAmB;IAavB,cAAc,CAClB,WAAW,EAAE,wBAAwB,EACrC,KAAK,EAAE,KAAK,GACX,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC;CAyBpC"}
+{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../../../src/ai-service/integrations/store.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAG1D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,KAAK,EACV,KAAK,EACL,uBAAuB,EACxB,MAAM,+BAA+B,CAAC;AACvC,OAAO,KAAK,EACV,cAAc,EAGf,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,uCAAuC,CAAC;AAEtF,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,YAAY,CAAmC;IAEjD,8BAA8B,CAClC,WAAW,EAAE,wBAAwB,EACrC,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,uBAAuB,GAC9B,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC;IA6B3B,mBAAmB,CAAC,EACxB,KAAK,EACL,WAAW,EACX,MAAM,EACN,WAAW,GACZ,EAAE;QACD,KAAK,EAAE,KAAK,CAAC;QACb,WAAW,EAAE,wBAAwB,CAAC;QACtC,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,WAAW,CAAC;KAC1B,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAqBhC,8BAA8B,CAC5B,WAAW,EAAE,wBAAwB,GACpC,OAAO,mBAAmB;IAgBvB,cAAc,CAClB,WAAW,EAAE,wBAAwB,EACrC,KAAK,EAAE,KAAK,GACX,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC;CAyBpC"}

package/dist/ai-service/integrations/store.js

@@ -2,6 +2,7 @@ import { isEmpty } from "lodash-es";
 import { datasourceSdkClassByType } from "../const.js";
 import { databaseIntegrationTypes } from "../const.js";
 import { DatabaseIntegrationMetadata } from "./metadata/database.js";
+import { GraphQLIntegrationMetadata } from "./metadata/graphql-based.js";
 import { IntegrationMetadata } from "./metadata/index.js";
 import { OpenApiIntegrationMetadata } from "./metadata/open-api.js";
 import { TableBasedIntegrationMetadata } from "./metadata/table-based.js";
@@ -42,7 +43,7 @@ export class IntegrationStore {
             };
         }
         const klass = this.getMetadataClassForIntegration(integration);
-        return klass.refineMetadataForPrompt({
+        return await klass.refineMetadataForPrompt({
             metadata,
             prompt,
             llmProvider,
@@ -58,6 +59,9 @@ export class IntegrationStore {
         if (isTableBasedIntegration(integration)) {
             return TableBasedIntegrationMetadata;
         }
+        if (isGraphQlIntegration(integration)) {
+            return GraphQLIntegrationMetadata;
+        }
         return IntegrationMetadata;
     }
     async getOpenApiSpec(integration, clark) {
@@ -100,4 +104,7 @@ function isTableBasedIntegration(integration) {
     }
     return false;
 }
+function isGraphQlIntegration(integration) {
+    return integration.type === "graphqlintegration";
+}
 //# sourceMappingURL=store.js.map

package/dist/ai-service/integrations/store.js.map

@@ -1 +1 @@
-{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../src/ai-service/integrations/store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,2BAA2B,EAAE,MAAM,wBAAwB,CAAC;AACrE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,0BAA0B,EAAE,MAAM,wBAAwB,CAAC;AACpE,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAc1E,MAAM,OAAO,gBAAgB;IACnB,YAAY,GAAgC,EAAE,CAAC;IAEvD,KAAK,CAAC,8BAA8B,CAClC,WAAqC,EACrC,MAAc,EACd,KAAY,EACZ,MAA+B;QAE/B,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,WAAW,CAAC;QACjC,MAAM,IAAI,GAAG,WAAW,CAAC,IAA0B,CAAC;QAEpD,IACE,CAAC,IAAI;YACL,CAAC,EAAE;YACH,CAAC,gBAAgB,CAAC;gBAChB,IAAI;aACL,CAAC,EACF,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC;YAC9C,KAAK;YACL,WAAW;YACX,MAAM;YACN,WAAW,EAAE,MAAM,CAAC,WAAW;SAChC,CAAC,CAAC;QAEH,OAAO;YACL,IAAI;YACJ,EAAE;YACF,IAAI;YACJ,QAAQ;SACT,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,EACxB,KAAK,EACL,WAAW,EACX,MAAM,EACN,WAAW,GAMZ;QACC,gFAAgF;QAChF,gGAAgG;QAChG,IAAI,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC;QACpC,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;YACtC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;YAElE,QAAQ,GAAG;gBACT,GAAG,CAAC,WAAW,CAAC,QAAQ,IAAI,EAAE,CAAC;gBAC/B,WAAW;aACZ,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,8BAA8B,CAAC,WAAW,CAAC,CAAC;QAC/D,OAAO,KAAK,CAAC,uBAAuB,CAAC;YACnC,QAAQ;YACR,MAAM;YACN,WAAW;SACZ,CAAC,CAAC;IACL,CAAC;IAED,8BAA8B,CAC5B,WAAqC;QAErC,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;YACtC,OAAO,0BAA0B,CAAC;QACpC,CAAC;QACD,IAAI,qBAAqB,CAAC,WAAW,CAAC,EAAE,CAAC;YACvC,OAAO,2BAA2B,CAAC;QACrC,CAAC;QACD,IAAI,uBAAuB,CAAC,WAAW,CAAC,EAAE,CAAC;YACzC,OAAO,6BAA6B,CAAC;QACvC,CAAC;QACD,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,WAAqC,EACrC,KAAY;QAEZ,IAAI,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YACtC,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;QACnC,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;YACzC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,oKAAoK;QACpK,iHAAiH;QACjH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC;YAC5D,aAAa,EAAE,WAAW,CAAC,EAAE;YAC7B,GAAG,EAAE,WAAW,CAAC,QAAQ,CAAC,cAAc;SACzC,CAAC,CAAC;QACH,IAAI,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACpB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,GAAG,MAAqB,CAAC;QAC1D,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;IAC3C,CAAC;CACF;AAED,SAAS,oBAAoB,CAAC,WAAqC;IACjE,OAAO,WAAW,CAAC,QAAQ,EAAE,cAAc,KAAK,SAAS,CAAC;AAC5D,CAAC;AAED,SAAS,gBAAgB,CACvB,WAAoC;IAEpC,OAAO,CACL,OAAO,WAAW,CAAC,IAAI,KAAK,QAAQ;QACpC,WAAW,CAAC,IAAI,IAAI,wBAAwB,CAC7C,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,WAAqC;IAClE,OAAO,wBAAwB,CAAC,QAAQ,CACtC,WAAW,CAAC,IAAkC,CAC/C,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,WAAqC;IACpE,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,IAAI,WAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC9E,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC5D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC"}
+{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../src/ai-service/integrations/store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,2BAA2B,EAAE,MAAM,wBAAwB,CAAC;AACrE,OAAO,EAAE,0BAA0B,EAAE,MAAM,6BAA6B,CAAC;AACzE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,0BAA0B,EAAE,MAAM,wBAAwB,CAAC;AACpE,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAc1E,MAAM,OAAO,gBAAgB;IACnB,YAAY,GAAgC,EAAE,CAAC;IAEvD,KAAK,CAAC,8BAA8B,CAClC,WAAqC,EACrC,MAAc,EACd,KAAY,EACZ,MAA+B;QAE/B,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,WAAW,CAAC;QACjC,MAAM,IAAI,GAAG,WAAW,CAAC,IAA0B,CAAC;QAEpD,IACE,CAAC,IAAI;YACL,CAAC,EAAE;YACH,CAAC,gBAAgB,CAAC;gBAChB,IAAI;aACL,CAAC,EACF,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC;YAC9C,KAAK;YACL,WAAW;YACX,MAAM;YACN,WAAW,EAAE,MAAM,CAAC,WAAW;SAChC,CAAC,CAAC;QAEH,OAAO;YACL,IAAI;YACJ,EAAE;YACF,IAAI;YACJ,QAAQ;SACT,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,EACxB,KAAK,EACL,WAAW,EACX,MAAM,EACN,WAAW,GAMZ;QACC,gFAAgF;QAChF,gGAAgG;QAChG,IAAI,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC;QACpC,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;YACtC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;YAElE,QAAQ,GAAG;gBACT,GAAG,CAAC,WAAW,CAAC,QAAQ,IAAI,EAAE,CAAC;gBAC/B,WAAW;aACZ,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,8BAA8B,CAAC,WAAW,CAAC,CAAC;QAC/D,OAAO,MAAM,KAAK,CAAC,uBAAuB,CAAC;YACzC,QAAQ;YACR,MAAM;YACN,WAAW;SACZ,CAAC,CAAC;IACL,CAAC;IAED,8BAA8B,CAC5B,WAAqC;QAErC,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;YACtC,OAAO,0BAA0B,CAAC;QACpC,CAAC;QACD,IAAI,qBAAqB,CAAC,WAAW,CAAC,EAAE,CAAC;YACvC,OAAO,2BAA2B,CAAC;QACrC,CAAC;QACD,IAAI,uBAAuB,CAAC,WAAW,CAAC,EAAE,CAAC;YACzC,OAAO,6BAA6B,CAAC;QACvC,CAAC;QACD,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;YACtC,OAAO,0BAA0B,CAAC;QACpC,CAAC;QACD,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,WAAqC,EACrC,KAAY;QAEZ,IAAI,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YACtC,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;QACnC,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;YACzC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,oKAAoK;QACpK,iHAAiH;QACjH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC;YAC5D,aAAa,EAAE,WAAW,CAAC,EAAE;YAC7B,GAAG,EAAE,WAAW,CAAC,QAAQ,CAAC,cAAc;SACzC,CAAC,CAAC;QACH,IAAI,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACpB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,GAAG,MAAqB,CAAC;QAC1D,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;IAC3C,CAAC;CACF;AAED,SAAS,oBAAoB,CAAC,WAAqC;IACjE,OAAO,WAAW,CAAC,QAAQ,EAAE,cAAc,KAAK,SAAS,CAAC;AAC5D,CAAC;AAED,SAAS,gBAAgB,CACvB,WAAoC;IAEpC,OAAO,CACL,OAAO,WAAW,CAAC,IAAI,KAAK,QAAQ;QACpC,WAAW,CAAC,IAAI,IAAI,wBAAwB,CAC7C,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,WAAqC;IAClE,OAAO,wBAAwB,CAAC,QAAQ,CACtC,WAAW,CAAC,IAAkC,CAC/C,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,WAAqC;IACpE,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,IAAI,WAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC9E,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC5D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,oBAAoB,CAAC,WAAqC;IACjE,OAAO,WAAW,CAAC,IAAI,KAAK,oBAAoB,CAAC;AACnD,CAAC"}

package/dist/ai-service/prompt-builder-service/builders/dbschema-intent.d.ts

@@ -3,6 +3,16 @@ import type { SdkIntegrationTable } from "../../types.js";
 type JsonDbSchema = {
     tables: SdkIntegrationTable[];
 };
+export declare function buildGraphQlSchemaIntentPrompt(userPrompt: string, schema: any): {
+    prompt: string;
+    schema: z.ZodObject<{
+        type_names: z.ZodArray<z.ZodString, "many">;
+    }, "strip", z.ZodTypeAny, {
+        type_names: string[];
+    }, {
+        type_names: string[];
+    }>;
+};
 export declare function buildDbSchemaIntentPrompt(userPrompt: string, schema: JsonDbSchema): {
     prompt: string;
     schema: z.ZodObject<{

package/dist/ai-service/prompt-builder-service/builders/dbschema-intent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"dbschema-intent.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/prompt-builder-service/builders/dbschema-intent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAI1D,KAAK,YAAY,GAAG;IAClB,MAAM,EAAE,mBAAmB,EAAE,CAAC;CAC/B,CAAC;AAqBF,wBAAgB,yBAAyB,CACvC,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,YAAY;;;;;;;;;;;;;;;EAyBrB"}
+{"version":3,"file":"dbschema-intent.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/prompt-builder-service/builders/dbschema-intent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAW1D,KAAK,YAAY,GAAG;IAClB,MAAM,EAAE,mBAAmB,EAAE,CAAC;CAC/B,CAAC;AA4FF,wBAAgB,8BAA8B,CAC5C,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,GAAG;;;;;;;;;EAiCZ;AAED,wBAAgB,yBAAyB,CACvC,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,YAAY;;;;;;;;;;;;;;;EAyBrB"}

package/dist/ai-service/prompt-builder-service/builders/dbschema-intent.js

@@ -15,6 +15,96 @@ function getSchema(fullSchema) {
             .describe(description),
     });
 }
+function getGraphQLSchema(_) {
+    const description = "The GraphQL types that the user wants to access based on their query intent";
+    // use a flexible string array that allows the llm to return any type names
+    return z.object({
+        type_names: z.array(z.string().min(1)).describe(description),
+    });
+}
+function createMinimalSchemaDigest(schema) {
+    if (!schema?.types)
+        return { objectTypes: [], totalTypes: 0 };
+    // extract only the essential information for each type
+    const digest = schema.types
+        .filter((type) => !type.name.startsWith("__")) // skip introspection types
+        .map((type) => {
+        const basicInfo = {
+            name: type.name,
+            kind: type.kind,
+        };
+        // add description if it exists and is short
+        if (type.description && type.description.length < 100) {
+            basicInfo.description = type.description;
+        }
+        // for object types, include field names that reference other objects
+        if (type.kind === "OBJECT" && type.fields) {
+            const objectFields = type.fields
+                .filter((field) => {
+                // only include fields that reference other object types
+                const fieldType = getBaseType(field.type);
+                return (fieldType.kind === "OBJECT" && !fieldType.name.startsWith("__"));
+            })
+                .map((field) => ({
+                name: field.name,
+                type: getBaseType(field.type).name,
+            }));
+            if (objectFields.length > 0) {
+                basicInfo.relationships = objectFields;
+            }
+        }
+        return basicInfo;
+    });
+    return {
+        types: digest,
+        summary: {
+            totalTypes: schema.types.length,
+            objectTypes: digest.filter((t) => t.kind === "OBJECT")
+                .length,
+            interfaces: digest.filter((t) => t.kind === "INTERFACE")
+                .length,
+            enums: digest.filter((t) => t.kind === "ENUM").length,
+        },
+    };
+}
+function getBaseType(type) {
+    // unwrap nonnull and list wrappers to get the base type
+    if (type.ofType) {
+        return getBaseType(type.ofType);
+    }
+    return type;
+}
+export function buildGraphQlSchemaIntentPrompt(userPrompt, schema) {
+    // handle both prefiltered schema and raw schema formats
+    let objectTypeNames;
+    if (schema.objectTypes && Array.isArray(schema.objectTypes)) {
+        // prefiltered schema format
+        objectTypeNames = schema.objectTypes.map((t) => t.name);
+    }
+    else {
+        // raw schema format - create digest
+        const schemaDigest = createMinimalSchemaDigest(schema);
+        objectTypeNames = schemaDigest.types
+            .filter((t) => t.kind === "OBJECT")
+            .map((t) => t.name);
+    }
+    const prompt = `You are a GraphQL schema analyst. Your task is to identify which object types are needed for a user's request.
+
+**Available Object Types:** ${objectTypeNames.join(", ")}
+
+**User Request:** ${userPrompt}
+
+**Instructions:**
+- Return ALL object types that may be involved in fulfilling this request, including types used in nested fields or relationships (e.g. state, assignee, labels, etc.)
+- Be inclusive — when in doubt, include the type
+- Do NOT include scalar, input, or enum types (these are handled automatically)
+
+**Response format:** {"type_names": ["Type1", "Type2", "Type3", "Type4", "Type5"]}`;
+    return {
+        prompt,
+        schema: getGraphQLSchema(schema),
+    };
+}
 export function buildDbSchemaIntentPrompt(userPrompt, schema) {
     const prompt = `You are a classification agent that has the sole responsibility of understanding the user's intent based on a database schema and the user's prompt.
 Your sole task is to return a list of tables the user wants to access.

package/dist/ai-service/prompt-builder-service/builders/dbschema-intent.js.map

@@ -1 +1 @@
-{"version":3,"file":"dbschema-intent.js","sourceRoot":"","sources":["../../../../src/ai-service/prompt-builder-service/builders/dbschema-intent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAUxB,SAAS,SAAS,CAAC,UAAwB;IACzC,MAAM,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QAClD,OAAO,KAAK,CAAC,IAAI,CAAC;IACpB,CAAC,CAAC,CAAC;IACH,MAAM,WAAW,GAAG,0CAA0C,CAAC;IAE/D,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,CAAC,CAAC,MAAM,CAAC;YACd,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC;SAClD,CAAC,CAAC;IACL,CAAC;IAED,OAAO,CAAC,CAAC,MAAM,CAAC;QACd,MAAM,EAAE,CAAC;aACN,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,WAAoC,CAAC,CAAC;aACnD,QAAQ,CAAC,WAAW,CAAC;KACzB,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,UAAkB,EAClB,MAAoB;IAEpB,MAAM,MAAM,GAAG;;;;;EAKf,UAAU;;;EAGV,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;;;;;;;;;CAShC,CAAC;IAEA,OAAO;QACL,MAAM;QACN,MAAM,EAAE,SAAS,CAAC,MAAM,CAAC;KAC1B,CAAC;AACJ,CAAC"}
+{"version":3,"file":"dbschema-intent.js","sourceRoot":"","sources":["../../../../src/ai-service/prompt-builder-service/builders/dbschema-intent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAiBxB,SAAS,SAAS,CAAC,UAAwB;IACzC,MAAM,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QAClD,OAAO,KAAK,CAAC,IAAI,CAAC;IACpB,CAAC,CAAC,CAAC;IACH,MAAM,WAAW,GAAG,0CAA0C,CAAC;IAE/D,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,CAAC,CAAC,MAAM,CAAC;YACd,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC;SAClD,CAAC,CAAC;IACL,CAAC;IAED,OAAO,CAAC,CAAC,MAAM,CAAC;QACd,MAAM,EAAE,CAAC;aACN,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,WAAoC,CAAC,CAAC;aACnD,QAAQ,CAAC,WAAW,CAAC;KACzB,CAAC,CAAC;AACL,CAAC;AAED,SAAS,gBAAgB,CAAC,CAAM;IAC9B,MAAM,WAAW,GACf,6EAA6E,CAAC;IAEhF,2EAA2E;IAC3E,OAAO,CAAC,CAAC,MAAM,CAAC;QACd,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC;KAC7D,CAAC,CAAC;AACL,CAAC;AAED,SAAS,yBAAyB,CAAC,MAAW;IAC5C,IAAI,CAAC,MAAM,EAAE,KAAK;QAAE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;IAE9D,uDAAuD;IACvD,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK;SACxB,MAAM,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,2BAA2B;SAC9E,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE;QACjB,MAAM,SAAS,GAAkB;YAC/B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;SAChB,CAAC;QAEF,4CAA4C;QAC5C,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;YACtD,SAAS,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QAC3C,CAAC;QAED,qEAAqE;QACrE,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAC1C,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM;iBAC7B,MAAM,CAAC,CAAC,KAAU,EAAE,EAAE;gBACrB,wDAAwD;gBACxD,MAAM,SAAS,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC1C,OAAO,CACL,SAAS,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAChE,CAAC;YACJ,CAAC,CAAC;iBACD,GAAG,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,CAAC;gBACpB,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,IAAI,EAAE,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI;aACnC,CAAC,CAAC,CAAC;YAEN,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5B,SAAS,CAAC,aAAa,GAAG,YAAY,CAAC;YACzC,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC,CAAC;IAEL,OAAO;QACL,KAAK,EAAE,MAAM;QACb,OAAO,EAAE;YACP,UAAU,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM;YAC/B,WAAW,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAgB,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;iBAClE,MAAM;YACT,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAgB,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC;iBACpE,MAAM;YACT,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAgB,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,MAAM;SACrE;KACF,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,IAAS;IAC5B,wDAAwD;IACxD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QAChB,OAAO,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC5C,UAAkB,EAClB,MAAW;IAEX,wDAAwD;IACxD,IAAI,eAAyB,CAAC;IAE9B,IAAI,MAAM,CAAC,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;QAC5D,4BAA4B;QAC5B,eAAe,GAAG,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAC/D,CAAC;SAAM,CAAC;QACN,oCAAoC;QACpC,MAAM,YAAY,GAAG,yBAAyB,CAAC,MAAM,CAAC,CAAC;QACvD,eAAe,GAAG,YAAY,CAAC,KAAK;aACjC,MAAM,CAAC,CAAC,CAAgB,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aACjD,GAAG,CAAC,CAAC,CAAgB,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC;IAED,MAAM,MAAM,GAAG;;8BAEa,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC;;oBAEpC,UAAU;;;;;;;mFAOqD,CAAC;IAElF,OAAO;QACL,MAAM;QACN,MAAM,EAAE,gBAAgB,CAAC,MAAM,CAAC;KACjC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,UAAkB,EAClB,MAAoB;IAEpB,MAAM,MAAM,GAAG;;;;;EAKf,UAAU;;;EAGV,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;;;;;;;;;CAShC,CAAC;IAEA,OAAO;QACL,MAAM;QACN,MAAM,EAAE,SAAS,CAAC,MAAM,CAAC;KAC1B,CAAC;AACJ,CAAC"}

package/dist/ai-service/prompt-builder-service/index.d.ts

@@ -1,4 +1,4 @@
 export { buildCodeGenerationSystemPrompt } from "./builders/code-generation.js";
 export { buildOpenAPIIntentPrompt } from "./builders/openapi-intent.js";
-export { buildDbSchemaIntentPrompt } from "./builders/dbschema-intent.js";
+export { buildDbSchemaIntentPrompt, buildGraphQlSchemaIntentPrompt, } from "./builders/dbschema-intent.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/ai-service/prompt-builder-service/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/ai-service/prompt-builder-service/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,+BAA+B,EAAE,MAAM,+BAA+B,CAAC;AAChF,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AACxE,OAAO,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/ai-service/prompt-builder-service/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,+BAA+B,EAAE,MAAM,+BAA+B,CAAC;AAChF,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AACxE,OAAO,EACL,yBAAyB,EACzB,8BAA8B,GAC/B,MAAM,+BAA+B,CAAC"}

package/dist/ai-service/prompt-builder-service/index.js

@@ -1,4 +1,4 @@
 export { buildCodeGenerationSystemPrompt } from "./builders/code-generation.js";
 export { buildOpenAPIIntentPrompt } from "./builders/openapi-intent.js";
-export { buildDbSchemaIntentPrompt } from "./builders/dbschema-intent.js";
+export { buildDbSchemaIntentPrompt, buildGraphQlSchemaIntentPrompt, } from "./builders/dbschema-intent.js";
 //# sourceMappingURL=index.js.map

package/dist/ai-service/prompt-builder-service/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/ai-service/prompt-builder-service/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,+BAA+B,EAAE,MAAM,+BAA+B,CAAC;AAChF,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AACxE,OAAO,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/ai-service/prompt-builder-service/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,+BAA+B,EAAE,MAAM,+BAA+B,CAAC;AAChF,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AACxE,OAAO,EACL,yBAAyB,EACzB,8BAA8B,GAC/B,MAAM,+BAA+B,CAAC"}

package/dist/ai-service/prompt-builder-service/static-fragments/library-components/ButtonPropsDocs.js

@@ -1,6 +1,6 @@
 /* eslint-disable */
 // Auto-generated from ButtonPropsDocs.md
 // Do not edit this file directly
-// Generated at: 2025-08-22T20:22:54.392Z
+// Generated at: 2025-08-22T20:58:50.712Z
 export const content = "## Button\n\nThe following is the type definition for the Button component.\n\n```typescript\ninterface ButtonProps {\n  /** The text displayed on the button */\n  label?: string;\n  /** Sets the internal spacing around the component's content */\n  /** @default {\"left\":{\"mode\":\"px\",\"value\":10},\"right\":{\"mode\":\"px\",\"value\":10},\"top\":{\"mode\":\"px\",\"value\":8},\"bottom\":{\"mode\":\"px\",\"value\":8}} */\n  padding?: { left: Dim; right: Dim; top: Dim; bottom: Dim };\n  /** Sets the width of the component */\n  width?: Dim;\n  /** Sets the height of the component */\n  height?: Dim;\n  /** Controls the min width dimension of the component */\n  minWidth?: Dim;\n  /** Controls the max width dimension of the component */\n  maxWidth?: Dim;\n  /** Controls the min height dimension of the component */\n  minHeight?: Dim;\n  /** Controls the max height dimension of the component */\n  maxHeight?: Dim;\n  /** Sets the external spacing around the component */\n  /** @default {\"top\":{\"mode\":\"px\",\"value\":0},\"bottom\":{\"mode\":\"px\",\"value\":0},\"left\":{\"mode\":\"px\",\"value\":0},\"right\":{\"mode\":\"px\",\"value\":0}} */\n  margin?: { left: Dim; right: Dim; top: Dim; bottom: Dim };\n  /** Controls whether the component is visible on the page */\n  /** @default true */\n  isVisible?: boolean;\n  /** The visual style variant of the button (primary, secondary, or tertiary) */\n  /** @default \"primary\" */\n  variant?: \"primary\" | \"secondary\" | \"tertiary\";\n  /** The icon displayed on the button */\n  icon?: string;\n  /** The position of the icon relative to the button text (left or right) */\n  /** @default \"left\" */\n  iconPosition?: \"left\" | \"right\";\n  /** Configures the text styling including font family, size, weight, color, and other typography properties */\n  /** @default \"Computed at runtime\" */\n  textStyle?: TextStyleWithVariant;\n  /** Allows selection of a color value using a color picker interface */\n  /** @default \"Computed at runtime\" */\n  backgroundColor?: string;\n  /** Component border styling including color, width, and style for all sides */\n  /** @default {\"left\":{\"width\":{\"mode\":\"px\",\"value\":1},\"style\":\"solid\",\"color\":\"transparent\"},\"right\":{\"width\":{\"mode\":\"px\",\"value\":1},\"style\":\"solid\",\"color\":\"transparent\"},\"top\":{\"width\":{\"mode\":\"px\",\"value\":1},\"style\":\"solid\",\"color\":\"transparent\"},\"bottom\":{\"width\":{\"mode\":\"px\",\"value\":1},\"style\":\"solid\",\"color\":\"transparent\"}} */\n  border?: { left: Border; right: Border; top: Border; bottom: Border };\n  /** Component border radius settings for each corner (top-left, top-right, bottom-right, bottom-left) */\n  /** @default \"Computed at runtime\" */\n  borderRadius?: {\n    topLeft: Dim;\n    topRight: Dim;\n    bottomLeft: Dim;\n    bottomRight: Dim;\n  };\n  /** The horizontal alignment of the button content (left, center, or right) */\n  /** @default \"center\" */\n  horizontalAlign?: \"left\" | \"center\" | \"right\";\n  /** Controls the loading state of the component - off, on, or auto-detected based on data loading */\n  loading?: boolean;\n  /** Whether the button is disabled and cannot be clicked */\n  /** @default false */\n  isDisabled?: boolean;\n  onClick?: EventFlow;\n}\n```\n";
 //# sourceMappingURL=ButtonPropsDocs.js.map