Pular para conteúdo

Tables metadata prompt

generate_tables_metadata_prompt(metadata)

Generates a descriptive string for table metadata.

Parameters:

Name Type Description Default
metadata List[TableMetadata]

List[TableMetadata]: A list of TableMetadata objects containing table information.

required

Returns:

Name Type Description
str str

A formatted string describing the tables and their columns.

Source code in api/utils/tables_metadata_prompt.py
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
def generate_tables_metadata_prompt(metadata: List[TableMetadata]) -> str:
    """
    Generates a descriptive string for table metadata.

    Args:
        metadata: List[TableMetadata]: A list of TableMetadata objects containing table information.

    Returns:
        str: A formatted string describing the tables and their columns.
    """
    table_descriptions = []
    for table in metadata:
        column_descriptions = "\n".join(
            [f"- {col.name}: {col.description}" for col in table.columns]
        )
        table_descriptions.append(
            f"Table: {table.name}\nDescription: {table.description}\nColumns:\n{column_descriptions}"
        )
    return "\n\n".join(table_descriptions)