Update utils.py

utils.py

@@ -272,6 +272,14 @@ def clean_sql(s: str) -> str:
         s = s.replace("SELECT","SELECT TOP 1")
     return s
 
+def get_metadata_info(metadata_df, table_names):
+    str = ""
+    for table in table_names:
+        str += "The following metadata is for the table " + table + "\n"
+        str += metadata_df[metadata_df.table == table].desc.iloc[0]
+        
+    return str
+    
 class SQLDatabaseChainPatched(SQLDatabaseChain):
     intermediate_steps: List[Any] = Field(default_factory=list)
     llms: Dict[str, Any] = Field(default_factory=dict)