Mustehson commited on
Commit
d43019d
·
1 Parent(s): 8856e7f

Update Prompt

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. prompt.py +1 -0
app.py CHANGED
@@ -157,7 +157,7 @@ def main(table):
157
  tests = run_llm(df)
158
  print(tests)
159
  if isinstance(tests, Exception):
160
- tests = pd.DataFrame([{"error": f"❌ Unable to get the SQL query based on the text. {tests}"}])
161
  return df.head(10), df_statistics, df_alerts, describe_cat, describe_num, tests, pd.DataFrame([])
162
 
163
  tests_df = pd.DataFrame(tests)
 
157
  tests = run_llm(df)
158
  print(tests)
159
  if isinstance(tests, Exception):
160
+ tests = pd.DataFrame([{"error": f"❌ Unable to generate tests. {tests}"}])
161
  return df.head(10), df_statistics, df_alerts, describe_cat, describe_num, tests, pd.DataFrame([])
162
 
163
  tests_df = pd.DataFrame(tests)
prompt.py CHANGED
@@ -7,6 +7,7 @@ Follow this process:
7
  1. **Observe the sample data.**
8
  2. **For each column**, create a validation rule using Pandera syntax.
9
  Here are the valid pandera check class methods DO NOT USE ANYOTHER METHODS OTHER THAN THE BELOW GIVEN METHODS:
 
10
  [
11
  'pa.Check.between(min_value, max_value, include_min=True, include_max=True, **kwargs)',
12
  'pa.Check.eq(value, **kwargs)',
 
7
  1. **Observe the sample data.**
8
  2. **For each column**, create a validation rule using Pandera syntax.
9
  Here are the valid pandera check class methods DO NOT USE ANYOTHER METHODS OTHER THAN THE BELOW GIVEN METHODS:
10
+ DO NOT USE SINGLE backslashes \ BUT USE DOUBLE backslashes \\ IN PATTERN
11
  [
12
  'pa.Check.between(min_value, max_value, include_min=True, include_max=True, **kwargs)',
13
  'pa.Check.eq(value, **kwargs)',