Skip to content

Commit 93d6246

Browse files
committed
Add additional fields to Item model for RAG conversion
1 parent 7461416 commit 93d6246

File tree

1 file changed

+41
-17
lines changed

1 file changed

+41
-17
lines changed

src/fastapi_app/postgres_models.py

Lines changed: 41 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -179,23 +179,47 @@ def to_dict(self, include_embedding: bool = False):
179179
return model_dict
180180

181181
def to_str_for_rag(self):
182-
return f"""package_name: {self.package_name}
183-
package_picture: {self.package_picture}
184-
url: {self.url}
185-
price: {self.price}
186-
cash_discount: {self.cash_discount}
187-
installment_month: {self.installment_month}
188-
installment_limit: {self.installment_limit}
189-
price_to_reserve_for_this_package: {self.price_to_reserve_for_this_package}
190-
brand: {self.brand}
191-
price_details: {self.price_details}
192-
package_details: {self.package_details}
193-
important_info: {self.important_info}
194-
payment_booking_info: {self.payment_booking_info}
195-
general_info: {self.general_info}
196-
brand_ranking_position: {self.brand_ranking_position}
197-
faq: {self.faq}
198-
"""
182+
return f"""
183+
package_name: {self.package_name}
184+
package_picture: {self.package_picture}
185+
url: {self.url}
186+
price: {self.price}
187+
cash_discount: {self.cash_discount}
188+
installment_month: {self.installment_month}
189+
installment_limit: {self.installment_limit}
190+
price_to_reserve_for_this_package: {self.price_to_reserve_for_this_package}
191+
shop_name: {self.shop_name}
192+
category: {self.category}
193+
category_tags: {self.category_tags}
194+
preview_1_10: {self.preview_1_10}
195+
selling_point: {self.selling_point}
196+
meta_keywords: {self.meta_keywords}
197+
brand: {self.brand}
198+
min_max_age: {self.min_max_age}
199+
locations: {self.locations}
200+
meta_description: {self.meta_description}
201+
price_details: {self.price_details}
202+
package_details: {self.package_details}
203+
important_info: {self.important_info}
204+
payment_booking_info: {self.payment_booking_info}
205+
general_info: {self.general_info}
206+
early_signs_for_diagnosis: {self.early_signs_for_diagnosis}
207+
how_to_diagnose: {self.how_to_diagnose}
208+
hdcare_summary: {self.hdcare_summary}
209+
common_question: {self.common_question}
210+
know_this_disease: {self.know_this_disease}
211+
courses_of_action: {self.courses_of_action}
212+
signals_to_proceed_surgery: {self.signals_to_proceed_surgery}
213+
get_to_know_this_surgery: {self.get_to_know_this_surgery}
214+
comparisons: {self.comparisons}
215+
getting_ready: {self.getting_ready}
216+
recovery: {self.recovery}
217+
side_effects: {self.side_effects}
218+
review_4_5_stars: {self.review_4_5_stars}
219+
brand_option_in_thai_name: {self.brand_option_in_thai_name}
220+
brand_ranking_position: {self.brand_ranking_position}
221+
faq: {self.faq}
222+
"""
199223

200224
def to_str_for_embedding_package_name(self):
201225
return f"Package Name: {self.package_name}" if self.package_name else ""

0 commit comments

Comments
 (0)