more emails

app.py

@@ -37,7 +37,7 @@ def get_df_from_workbook(sheet_name,
  url = f'https://docs.google.com/spreadsheets/d/{workbook_id}/gviz/tq?tqx=out:csv&sheet={sheet_name}'
  return pd.read_csv(url)
 
-docEmailSample = Document(
+docEmailSample1 = Document(
  text="Hey KD, let's grab dinner after our next game, Steph", 
  metadata={'from_to': 'Stephen Curry to Kevin Durant',}
 )
@@ -45,7 +45,28 @@ docEmailSample2 = Document(
  text="Yo Joker, you were a monster last year, can't wait to play against you in the opener! Draymond", 
  metadata={'from_to': 'Draymond Green to Nikola Jokic',}
 )
-docAdditionalSamples = [docEmailSample, docEmailSample2]
+docEmailSample3 = Document(
+ text="Hey LeBron, you ready for another showdown? Let's see if you can handle the Splash Bros again! 😜",
+ metadata={'from_to': 'Klay Thompson to LeBron James'}
+)
+docEmailSample4 = Document(
+ text="Yo Giannis, you sure you want to come to the Bay? We don't have any deer to hunt here! 😂",
+ metadata={'from_to': 'Draymond Green to Giannis Antetokounmpo'}
+)
+docEmailSample5 = Document(
+ text="Hey Luka, you're a beast on the court. Let's swap jerseys after our game and show some love! 💪",
+ metadata={'from_to': 'Andrew Wiggins to Luka Dončić'}
+)
+docEmailSample5 = Document(
+ text="Devin Booker, we could use your scoring in the Bay. Think about it, bro! 🏀",
+ metadata={'from_to': 'Klay Thompson to Devin Booker'}
+)
+docEmailSample6 = Document(
+ text="Tatum, you've got the skills. Let's team up and bring some championships to the Warriors! 💍",
+ metadata={'from_to': 'Draymond Green to Jayson Tatum'}
+)
+docAdditionalSamples = [docEmailSample1, docEmailSample2, docEmailSample3, 
+ docEmailSample4, docEmailSample5, docEmailSample6]
 
 class AutoRetrieveModel(BaseModel):
  query: str = Field(..., description="natural language query string")