More seed data updates

Author: pamelafox

evals/generate_ground_truth.py

@@ -12,7 +12,7 @@
 from sqlalchemy import create_engine, select
 from sqlalchemy.orm import Session
 
-from fastapi_app.postgres_models import Restaurant
+from fastapi_app.postgres_models import Item
 
 logger = logging.getLogger("ragapp")
 

src/backend/fastapi_app/postgres_models.py

@@ -1,9 +1,8 @@
 from __future__ import annotations
 
 from pgvector.sqlalchemy import Vector
-from sqlalchemy import Index, VARCHAR
+from sqlalchemy import VARCHAR, Index
 from sqlalchemy.dialects import postgresql
-
 from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
 
 
@@ -43,10 +42,11 @@ def to_dict(self, include_embedding: bool = False):
         return model_dict
 
     def to_str_for_rag(self):
-        return f"Name:{self.name} Description:{self.description} Location:{self.location} Cuisine:{self.cuisine} Rating:{self.rating} Price Level:{self.price_level} Review Count:{self.review_count} Hours:{self.hours} Tags:{self.tags} Menu Summary:{self.menu_summary} Top Reviews:{self.top_reviews} Vibe:{self.vibe}"
-    
+        return f"Name:{self.name} Description:{self.description} Location:{self.location} Cuisine:{self.cuisine} Rating:{self.rating} Price Level:{self.price_level} Review Count:{self.review_count} Hours:{self.hours} Tags:{self.tags} Menu Summary:{self.menu_summary} Top Reviews:{self.top_reviews} Vibe:{self.vibe}"  # noqa: E501
+
     def to_str_for_embedding(self):
-        return f"Name: {self.name} Description: {self.description} Cuisine: {self.cuisine} Tags: {self.tags} Menu Summary: {self.menu_summary} Top Reviews: {self.top_reviews} Vibe: {self.vibe}"
+        return f"Name: {self.name} Description: {self.description} Cuisine: {self.cuisine} Tags: {self.tags} Menu Summary: {self.menu_summary} Top Reviews: {self.top_reviews} Vibe: {self.vibe}"  # noqa: E501
+
 
 """
 **Define HNSW index to support vector similarity search**

src/backend/fastapi_app/prompts/answer.txt

@@ -6,4 +6,4 @@ Do not generate answers that don't use the sources below.
 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].
+Don't combine references, cite each restaurant separately, for example [27][51].

src/backend/fastapi_app/query_rewriter.py

@@ -22,11 +22,11 @@ def build_search_function() -> list[ChatCompletionToolParam]:
                         },
                         "price_level_filter": {
                             "type": "object",
-                            "description": "Filter search results to a certain price level (from 1 $ to 4 $$$$, with 4 being most costly)",
+                            "description": "Filter search results to a certain price level (from 1 $ to 4 $$$$, with 4 being most costly)",  # noqa: E501
                             "properties": {
                                 "comparison_operator": {
                                     "type": "string",
-                                    "description": "Operator to compare the column value, either '>', '<', '>=', '<=', '='",  # noqa
+                                    "description": "Operator to compare the column value, either '>', '<', '>=', '<=', '='",  # noqa: E501
                                 },
                                 "value": {
                                     "type": "number",
@@ -36,11 +36,11 @@ def build_search_function() -> list[ChatCompletionToolParam]:
                         },
                         "rating_filter": {
                             "type": "object",
-                            "description": "Filter search results based on ratings of restaurant (from 1 to 5 stars, with 5 the best)",
+                            "description": "Filter search results based on ratings of restaurant (from 1 to 5 stars, with 5 the best)",  # noqa: E501
                             "properties": {
                                 "comparison_operator": {
                                     "type": "string",
-                                    "description": "Operator to compare the column value, either '>', '<', '>=', '<=', '='",
+                                    "description": "Operator to compare the column value, either '>', '<', '>=', '<=', '='",  # noqa: E501
                                 },
                                 "value": {
                                     "type": "string",
@@ -69,7 +69,11 @@ 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_level_filter" in arg and arg["price_level_filter"] and isinstance(arg["price_level_filter"], dict):
+                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(
                         {

src/backend/fastapi_app/seed_data.json

@@ -90799,4 +90799,4 @@
             -0.057345152
         ]
     }
-]
+]