Fix query rewriting extraction

pamelafox · pamelafox · commit 4226309d56b1 · 2024-05-14T23:01:48.000Z
diff --git a/src/fastapi_app/query_rewriter.py b/src/fastapi_app/query_rewriter.py
@@ -68,7 +68,7 @@ def extract_search_arguments(chat_completion: ChatCompletion):
             if function.name == "search_database":
                 arg = json.loads(function.arguments)
                 search_query = arg.get("search_query")
-                if "price_filter" in arg:
+                if "price_filter" in arg and arg["price_filter"]:
                     price_filter = arg["price_filter"]
                     filters.append(
                         {
@@ -77,7 +77,7 @@ def extract_search_arguments(chat_completion: ChatCompletion):
                             "value": price_filter["value"],
                         }
                     )
-                if "brand_filter" in arg:
+                if "brand_filter" in arg and arg["brand_filter"]:
                     brand_filter = arg["brand_filter"]
                     filters.append(
                         {

Original file line number	Diff line number	Diff line change
`@@ -68,7 +68,7 @@ def extract_search_arguments(chat_completion: ChatCompletion):`
`68`	`68`	`if function.name == "search_database":`
`69`	`69`	`arg = json.loads(function.arguments)`
`70`	`70`	`search_query = arg.get("search_query")`
`71`		`- if "price_filter" in arg:`
	`71`	`+ if "price_filter" in arg and arg["price_filter"]:`
`72`	`72`	`price_filter = arg["price_filter"]`
`73`	`73`	`filters.append(`
`74`	`74`	`{`
`@@ -77,7 +77,7 @@ def extract_search_arguments(chat_completion: ChatCompletion):`
`77`	`77`	`"value": price_filter["value"],`
`78`	`78`	`}`
`79`	`79`	`)`
`80`		`- if "brand_filter" in arg:`
	`80`	`+ if "brand_filter" in arg and arg["brand_filter"]:`
`81`	`81`	`brand_filter = arg["brand_filter"]`
`82`	`82`	`filters.append(`
`83`	`83`	`{`