Class QueryWriter

QueryWriter class Generates Nunjucks SQL query templates using AI with few-shot learning

Index

Constructors

Methods

Constructors

constructor

new QueryWriter(contextUser: UserInfo, config: QueryGenConfig): QueryWriter
Parameters
- contextUser: UserInfo
- config: QueryGenConfig
Returns QueryWriter
- Defined in QueryGen/src/core/QueryWriter.ts:22

Methods

generateQuery

generateQuery(
    businessQuestion: BusinessQuestion,
    entityMetadata: EntityMetadataForPrompt[],
    fewShotExamples: GoldenQuery[],
): Promise<GeneratedQuery>
Generate SQL query template for a business question Uses few-shot learning with similar golden query examples
Parameters
- businessQuestion: BusinessQuestion
  Business question to answer with SQL
- entityMetadata: EntityMetadataForPrompt[]
  Available entity metadata for query
- fewShotExamples: GoldenQuery[]
  Similar golden queries for few-shot learning
Returns Promise<GeneratedQuery>
Generated SQL query with parameters and output schema
- Defined in QueryGen/src/core/QueryWriter.ts:36