Update main.py

main.py

@@ -119,10 +119,10 @@ class FollowupQueryModel(BaseModel):
 
 
 
-async def digiyatra_query_table(query: str, db: Annotated[Any, Depends(get_db_connection)], limit: Optional[int] = 5):
+async def digiyatra_query_table(query: str, db: Annotated[Any, Depends(get_db_connection)], limit: Optional[int] = 2):
     """Query the digiyatra table."""
     response = await query_table(
-        table_id="llama",
+        table_id="digiyatra",
         query=query,
         user_id="digiyatra", 
         limit=limit