diff --git a/new_parser_test.py b/new_parser_test.py index 20bd502..8dd9155 100644 --- a/new_parser_test.py +++ b/new_parser_test.py @@ -15,7 +15,7 @@ TAG_LIST = [ "Архитектура", "Площадь", "Набережная", "Канатная дорога", "Сквер", "Музей", "Собор", "Церковь", "Храм", "Театр", "Выставка", "Мост", "Художественное пространство", "Вокзал", "Фонтан", "Лестница", "Кинотеатр", - "Дом Культуры", "Планетарий", "Галерея", "Ресторан", "Бар", "Бистро", "Кафе" + "Дом Культуры", "Планетарий", "Галерея", "Ресторан", "Бар", "Бистро", "Кафе", "Кофейня" ] @@ -37,14 +37,14 @@ def build_extraction_prompt(conversation_history: List[Dict[str, str]], tag_list } example_user_3 = "Я хочу культурно провести время потом попить кофе, я свободен до 6 вечера, я сейчас на б-р 60 лет Октября, 9" example_json_3 = { - "tags": [['Музей', 'Галерея', 'Театр', 'Выставка', 'Художественное пространство', 'Планетарий', 'ANY', 'MULTI'], ["Кафе"]], + "tags": [['Музей', 'Галерея', 'Театр', 'Выставка', 'Художественное пространство', 'Планетарий', 'ANY', 'MULTI'], ["Кофейня"]], "user_location": "бульвар 60 лет Октября, дом 9", "time": 'until 18:00' } example_user_4 = "Я хочу обойти места в которых можно попить кофе и зайти в музей, у меня 4 часа, я на дальней 8" example_json_4 = { - "tags": [['Кафе', 'MULTI'], + "tags": [['Кофейня', 'MULTI'], ["Музей"]], "user_location": "улица Дальняя, дом 8", "time": 240 @@ -60,10 +60,10 @@ def build_extraction_prompt(conversation_history: List[Dict[str, str]], tag_list Работа с тегами: Группируй теги по ПРИОРИТЕТУ посещения: -- Если пользователь говорит "сначала музей, потом кофе" -> [['Музей'], ['Кафе']] -- Если "хочу в музей и кофе" (без явного порядка) -> [['Музей', 'Кафе']] +- Если пользователь говорит "сначала музей, потом кафе" -> [['Музей'], ['Кафе']] +- Если "хочу в музей и кафе" (без явного порядка) -> [['Музей', 'Кафе']] - Если неопределённые предпочтения "погулять по интересным местам", добавь ключевое слово 'ANY' к группе подходящих тегов. -- Если пользователь хочет посетить несколько мест одного типа, добавь ключевое слово 'MULTI' в соответсвующую группу. "Хочу походить по музеям, потом попить кофе." -> [['Музей','MULTI'], ['Кафе']] +- Если пользователь хочет посетить несколько мест одного типа, добавь ключевое слово 'MULTI' в соответсвующую группу. "Хочу походить по музеям, потом попить кофе." -> [['Музей','MULTI'], ['Кофейня']] - Запрос пользователя может быть абстрактным "я хочу погулять по улице". Подбери релевантные теги на открытом воздухе (т.к. погулять можно интерпретировать как походить по улице не заходя куда-либо) например: [['Архитектура','Сквер','Площадь', 'ANY']]. Запрос может быть абстрактным, но не может быть полностью неопределенным. Примеры неопределенных ответов: "Хочу куда-нибудь сходить", "Я не знаю куда сходить", "Куда мне сходить?" - Указывай все теги из tag_list которые явно подходят!!! Примеры могут быть неполными!