More seed data updates

Author: pamelafox

data.csv

@@ -85,7 +85,7 @@ class ItemPublic(BaseModel):
     vibe: str
 
 
-class ItemPublicWithDistance(RestaurantPublic):
+class ItemWithDistance(ItemPublic):
     distance: float
 
     def __init__(self, **data):

data.json

@@ -1,7 +1,9 @@
 from __future__ import annotations
 
 from pgvector.sqlalchemy import Vector
-from sqlalchemy import Index
+from sqlalchemy import Index, VARCHAR
+from sqlalchemy.dialects import postgresql
+
 from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
 
 
@@ -20,7 +22,7 @@ class Item(Base):
     price_level: Mapped[int] = mapped_column()
     review_count: Mapped[int] = mapped_column()
     hours: Mapped[str] = mapped_column()
-    tags: Mapped[str] = mapped_column()  # JSON array
+    tags: Mapped[list[str]] = mapped_column(postgresql.ARRAY(VARCHAR))  # Array of strings
     description: Mapped[str] = mapped_column()
     menu_summary: Mapped[str] = mapped_column()
     top_reviews: Mapped[str] = mapped_column()

src/backend/fastapi_app/api_models.py

@@ -1,9 +1,9 @@
-Assistant helps customers with questions about products.
-Respond as if you are a salesperson helping a customer in a store. Do NOT respond with tables.
-Answer ONLY with the product details listed in the products.
+Assistant helps Pycon attendees with questions about restaurants.
+Respond as if you are a conference volunteer. Do NOT respond with tables.
+Answer ONLY with the restaurant details listed in the sources.
 If there isn't enough information below, say you don't know.
 Do not generate answers that don't use the sources below.
-Each product has an ID in brackets followed by colon and the product details.
-Always include the product ID for each product you use in the response.
-Use square brackets to reference the source, for example [52].
-Don't combine citations, list each product separately, for example [27][51].
+Each restaurant has an ID in brackets followed by colon and the restaurant details.
+Always include the restaurant ID for each restaurant you reference in the response.
+Use square brackets to reference the restaurant, for example [52].
+Don't combine references, cite each restaurant separately, for example [27][51].

src/backend/fastapi_app/postgres_models.py

@@ -1,5 +1,5 @@
 Below is a history of the conversation so far, and a new question asked by the user that needs to be answered by searching database rows.
-You have access to an Azure PostgreSQL database with an items table that has columns for title, description, brand, price, and type.
+You have access to an Azure PostgreSQL database with a restaurants table that has name, description, menu summary, vibe, ratings, etc.
 Generate a search query based on the conversation and the new question.
 If the question is not in English, translate the question to English before generating the search query.
 If you cannot generate a search query, return the original user question.

src/backend/fastapi_app/prompts/answer.txt

@@ -1,34 +1,34 @@
 [
-    {"role": "user", "content": "good options for climbing gear that can be used outside?"},
+    {"role": "user", "content": "good options for ethiopian restaurants?"},
     {"role": "assistant", "tool_calls": [
         {
             "id": "call_abc123",
             "type": "function",
             "function": {
-                "arguments": "{\"search_query\":\"climbing gear outside\"}",
+                "arguments": "{\"search_query\":\"ethiopian\"}",
                 "name": "search_database"
             }
         }
     ]},
     {
         "role": "tool",
         "tool_call_id": "call_abc123",
-        "content": "Search results for climbing gear that can be used outside: ..."
+        "content": "Search results for ethiopian: ..."
     },
-    {"role": "user", "content": "are there any shoes less than $50?"},
+    {"role": "user", "content": "are there any inexpensive chinese restaurants?"},
     {"role": "assistant", "tool_calls": [
         {
             "id": "call_abc456",
             "type": "function",
             "function": {
-                "arguments": "{\"search_query\":\"shoes\",\"price_filter\":{\"comparison_operator\":\"<\",\"value\":50}}",
+                "arguments": "{\"search_query\":\"chinese\",\"price_level_filter\":{\"comparison_operator\":\"<\",\"value\":3}}",
                 "name": "search_database"
             }
         }
     ]},
     {
         "role": "tool",
         "tool_call_id": "call_abc456",
-        "content": "Search results for shoes cheaper than 50: ..."
+        "content": "Search results for chinese: ..."
     }
 ]

src/backend/fastapi_app/prompts/query.txt

@@ -12,39 +12,39 @@ def build_search_function() -> list[ChatCompletionToolParam]:
             "type": "function",
             "function": {
                 "name": "search_database",
-                "description": "Search PostgreSQL database for relevant products based on user query",
+                "description": "Search PostgreSQL database for relevant restaurants based on user query",
                 "parameters": {
                     "type": "object",
                     "properties": {
                         "search_query": {
                             "type": "string",
                             "description": "Query string to use for full text search, e.g. 'red shoes'",
                         },
-                        "price_filter": {
+                        "price_level_filter": {
                             "type": "object",
-                            "description": "Filter search results based on price of the product",
+                            "description": "Filter search results to a certain price level (from 1 $ to 4 $$$$, with 4 being most costly)",
                             "properties": {
                                 "comparison_operator": {
                                     "type": "string",
                                     "description": "Operator to compare the column value, either '>', '<', '>=', '<=', '='",  # noqa
                                 },
                                 "value": {
                                     "type": "number",
-                                    "description": "Value to compare against, e.g. 30",
+                                    "description": "Value to compare against, either 1, 2, 3, 4",
                                 },
                             },
                         },
-                        "brand_filter": {
+                        "rating_filter": {
                             "type": "object",
-                            "description": "Filter search results based on brand of the product",
+                            "description": "Filter search results based on ratings of restaurant (from 1 to 5 stars, with 5 the best)",
                             "properties": {
                                 "comparison_operator": {
                                     "type": "string",
-                                    "description": "Operator to compare the column value, either '=' or '!='",
+                                    "description": "Operator to compare the column value, either '>', '<', '>=', '<=', '='",
                                 },
                                 "value": {
                                     "type": "string",
-                                    "description": "Value to compare against, e.g. AirStrider",
+                                    "description": "Value to compare against, either 0 1 2 3 4 5",
                                 },
                             },
                         },
@@ -69,22 +69,22 @@ def extract_search_arguments(original_user_query: str, chat_completion: ChatComp
                 arg = json.loads(function.arguments)
                 # Even though its required, search_query is not always specified
                 search_query = arg.get("search_query", original_user_query)
-                if "price_filter" in arg and arg["price_filter"] and isinstance(arg["price_filter"], dict):
-                    price_filter = arg["price_filter"]
+                if "price_level_filter" in arg and arg["price_level_filter"] and isinstance(arg["price_level_filter"], dict):
+                    price_level_filter = arg["price_level_filter"]
                     filters.append(
                         {
-                            "column": "price",
-                            "comparison_operator": price_filter["comparison_operator"],
-                            "value": price_filter["value"],
+                            "column": "price_level",
+                            "comparison_operator": price_level_filter["comparison_operator"],
+                            "value": price_level_filter["value"],
                         }
                     )
-                if "brand_filter" in arg and arg["brand_filter"] and isinstance(arg["brand_filter"], dict):
-                    brand_filter = arg["brand_filter"]
+                if "rating_filter" in arg and arg["rating_filter"] and isinstance(arg["rating_filter"], dict):
+                    rating_filter = arg["rating_filter"]
                     filters.append(
                         {
-                            "column": "brand",
-                            "comparison_operator": brand_filter["comparison_operator"],
-                            "value": brand_filter["value"],
+                            "column": "rating",
+                            "comparison_operator": rating_filter["comparison_operator"],
+                            "value": rating_filter["value"],
                         }
                     )
     elif query_text := response_message.content: