Skip to content

Commit 4cdc09b

Browse files
committed
modify the test for updated schema function
1 parent a01404f commit 4cdc09b

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/test_prompter.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ def test_construct_prompt_basic(self):
2020
"tables": {
2121
"users": {
2222
"columns": ["id", "name", "email", "status"],
23-
"samples": [{"id": "1", "name": "John Doe", "email": "john@example.com", "status": "active"}]
2423
}
2524
}
2625
}
@@ -29,8 +28,8 @@ def test_construct_prompt_basic(self):
2928
Act as a data analyst and SQL expert. Translate this natural language input into a SQL query for a Postgres DB:
3029
"{natural_language_input}"
3130
32-
Schema and sample data:
33-
Table: users,Columns: id, name, email, status,Sample Data: id: 1, name: John Doe, email: john@example.com, status: active
31+
Schema:
32+
Table: users,Columns: id, name, email, status
3433
3534
Reference examples:
3635
- Fetch active users: SELECT * FROM users WHERE status = 'active'

0 commit comments

Comments
 (0)