Update README.md

README.md

@@ -56,6 +56,64 @@ CREATE TABLE generalinfo\n(\n    id_restaurant INTEGER         not null\n
 Generate the SQL after thinking step by step:\n  <br>
 ```
 
+````python
+def bird_gpt_template_no_format(question, commonsense, schema):
+    return f'''{schema}
+
+-- External Knowledge: {commonsense}
+-- Using valid SQLite and understanding External Knowledge, answer the following question for the tables provided above.
+-- {question}
+Generate the SQL after thinking step by step:
+'''
+```
+
+### Generate Schema:
+
+```python
+def generate_schema_prompt(db_path, num_rows=None):
+    # extract create ddls
+    '''
+    :param root_place:
+    :param db_name:
+    :return:
+    '''
+    full_schema_prompt_list = []
+    conn = sqlite3.connect(db_path)
+    # Create a cursor object
+    cursor = conn.cursor()
+    cursor.execute("SELECT name FROM sqlite_master WHERE type='table'")
+    tables = cursor.fetchall()
+    schemas = {}
+    for table in tables:
+        if table == 'sqlite_sequence':
+            continue
+        cursor.execute("SELECT sql FROM sqlite_master WHERE type='table' AND name='{}';".format(table[0]))
+        create_prompt = cursor.fetchone()[0]
+        schemas[table[0]] = create_prompt
+        if num_rows:
+            cur_table = table[0]
+            if cur_table in ['order', 'by', 'group','transaction'] or ' ' in str(cur_table).strip() or '-' in str(cur_table).strip():
+                cur_table = '"{}"'.format(cur_table)
+
+
+            cursor.execute("SELECT * FROM {} LIMIT {}".format(cur_table, num_rows))
+            column_names = [description[0] for description in cursor.description]
+            values = cursor.fetchall()
+            rows_prompt = nice_look_table(column_names=column_names, values=values)
+            verbose_prompt = "/* \n {} example rows: \n SELECT * FROM {} LIMIT {}; \n {} \n */".format(num_rows,
+                                                                                                       cur_table,
+                                                                                                       num_rows,
+                                                                                                       rows_prompt)
+            schemas[table[0]] = "{} \n {}".format(create_prompt, verbose_prompt)
+
+    for k, v in schemas.items():
+        full_schema_prompt_list.append(v)
+
+    schema_prompt = "\n\n".join(full_schema_prompt_list)
+
+    return schema_prompt
+```
+
 
 ### System Prompt: