Spaces:
Runtime error
Runtime error
File size: 2,536 Bytes
7baafc3 d9ea4be 7baafc3 d9ea4be 7baafc3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
from typing import Dict, Any
def generate_mongodb_query(filters: Dict[str, Any]) -> Dict[str, Any]:
query = {}
# # makeModel
# if filters.get('makeModel') != "NULL":
# query['makeModel'] = filters['makeModel']
# # Model
# if filters.get('model') != "NULL":
# query['model'] = filters['model']
# # Make
# if filters.get('make') != "NULL":
# query['make'] = filters['make']
# # Color
# if filters.get('color') != "NULL":
# query['color'] = filters['color']
# # Features
# if filters.get('features') != "NULL" and filters.get('features'):
# query['features'] = {'$all': filters['features']}
# price
if filters.get('minprice') != "NULL" or filters.get('maxprice') != "NULL":
price_filter = {}
if filters.get('minprice') != "NULL":
price_filter['$gte'] = filters['minprice']
if filters.get('maxprice') != "NULL":
price_filter['$lte'] = filters['maxprice']
if price_filter:
query['msrp'] = price_filter
# # Year
# if filters.get('minyear') != "NULL" or filters.get('maxyear') != "NULL":
# year_filter = {}
# if filters.get('minyear') != "NULL":
# year_filter['$gte'] = filters['minyear']
# if filters.get('maxyear') != "NULL":
# year_filter['$lte'] = filters['maxyear']
# if year_filter:
# query['year'] = year_filter
return query
def get_prompt(query):
prompt = f"""
You are a helpful assistant that extracts the price and features of a car from a given user query.
You will extract the following fields:
- makeModel (string datatype)
- minprice (int datatype)
- maxprice (int datatype)
- minyear (int datatype)
- maxyear (int datatype)
- features (list of strings)
- color (string datatype)
- make (string datatype)
- model (string datatype)
price should only be a numerical number. dont write words or text or abbreviations. Just numbers like 40000 or 50000.
return "NULL" with proper quotation marks in the field if it is not mentioned
You goal is to extract the price and features of a car from a given user query.
return a json only and ensure the output is a valid JSON response
example output:
{{
"makeModel": "MINI HARDTOP 4 DOOR",
"price": "40000",
"features": ["Leather Seats", "Navigation", "Sunroof"]
}}
Here is the user query: {query}
"""
return prompt |