From 98e1ad84f6d99440c48462bfcecaab26981f08a0 Mon Sep 17 00:00:00 2001 From: LeonceNsh Date: Sat, 8 Jun 2024 03:48:48 -0500 Subject: [PATCH] add defense specific notebook --- app/compare_evals.py | 20 +++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-) diff --git a/app/compare_evals.py b/app/compare_evals.py index 832a578..74e45cb 100644 --- a/app/compare_evals.py +++ b/app/compare_evals.py @@ -14,4 +14,22 @@ # must be a dict in these formats, depending # on your database type # https://github.com/defog-ai/defog-python/blob/63af5e3ded07da356365f20bc94a194c4f7c44fa/defog/__init__.py#L110 -) \ No newline at end of file +) + +results = defog.run_query( + "how many from San Francisco", + previous_context=results['previous_context'] +) +{ + 'columns': ['num_users'], + 'data': [(50,)], + 'query_generated': "SELECT COUNT(*) AS num_users FROM users WHERE city ILIKE '%San Francisco%';", + 'ran_successfully': True, + 'reason_for_query': "The user is asking for the number of users from San Francisco. The city of the user is stored in the 'city' column of the 'users' table. Therefore, we can use a simple COUNT query to count the number of users from San Francisco. We will use the ILIKE operator to perform a case-insensitive match on the city name, as the user may have typed it in different ways. Since the query only requires one table, we do not need to use a JOIN statement. ", + 'previous_context': [ + 'how many users do we have?', + 'SELECT COUNT(userid) AS num_users FROM users;', + 'how many from San Francisco?', + "SELECT COUNT(*) AS num_users FROM users WHERE city ILIKE '%San Francisco%';" + ] +} \ No newline at end of file