Обновить new_parser_test.py

This commit is contained in:
2025-10-30 23:08:10 +03:00
parent 3f07cb558f
commit 883ce0bc88

View File

@@ -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 которые явно подходят!!! Примеры могут быть неполными!