jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor get_table_schema_as_dict function to remove index key

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-21 18:57:51 -03:00
commit de15017469
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/services/monitor/utils.py
View file

@ -14,7 +14,9 @@ INDEX_KEY = "index"
def get_table_schema_as_dict(conn: duckdb.DuckDBPyConnection, table_name: str) -> dict: def get_table_schema_as_dict(conn: duckdb.DuckDBPyConnection, table_name: str) -> dict:
result = conn.execute(f"PRAGMA table_info('{table_name}')").fetchall() result = conn.execute(f"PRAGMA table_info('{table_name}')").fetchall()
return {row[1]: row[2].upper() for row in result} schema = {row[1]: row[2].upper() for row in result}
schema.pop(INDEX_KEY, None)
return schema
def model_to_sql_column_definitions(model: Type[BaseModel]) -> dict: def model_to_sql_column_definitions(model: Type[BaseModel]) -> dict:
@ -56,6 +58,7 @@ def drop_and_create_table_if_schema_mismatch(
desired_schema = model_to_sql_column_definitions(model) desired_schema = model_to_sql_column_definitions(model)
# Compare the current and desired schemas # Compare the current and desired schemas
if current_schema != desired_schema: if current_schema != desired_schema:
# If they don't match, drop the existing table and create a new one # If they don't match, drop the existing table and create a new one
conn.execute(f"DROP TABLE IF EXISTS {table_name}") conn.execute(f"DROP TABLE IF EXISTS {table_name}")