Update main.py

main.py

@@ -132,7 +132,8 @@ async def digiyatra_query_table(query: FollowupQueryModel, db: Annotated[Any, De
         user_id=query.user_id, 
         limit=limit
     )
-    return response.results['data'][0]['text']
+    texts = [item['text'] for item in response.results['data']]
+    return str(texts)
 
 @app.post("/digiyatra-followup")
 async def followup_agent(query: FollowupQueryModel, background_tasks: BackgroundTasks):
@@ -149,7 +150,7 @@ async def followup_agent(query: FollowupQueryModel, background_tasks: Background
             ]
 
         digiyatra_response = await digiyatra_query_table(query, db=get_db_connection(), limit=5)
-        user_query_with_context = f"{query.query} \n\n FAQ Context for ANSWERING: {digiyatra_response}"
+        user_query_with_context = f"{query.query} \n\n ADDITIONAL CONTEXT: {digiyatra_response}"
         conversations[query.conversation_id].append({"role": "user", "content": user_query_with_context})
         last_activity[query.conversation_id] = time.time()