feat: debuging

app 00-30-34-608/app.py

@@ -0,0 +1,78 @@
+from fastapi import FastAPI
+
+from interface import PlanRequest, PlanResponse, TripPlan , YoutubeLinkRequest, YoutubeLinkResponse, ChatRequest
+from data_importer import DataImporter
+from utils.llm_caller import LLMCaller
+import asyncio
+import time
+from datetime import datetime
+
+app = FastAPI()
+data_importer = DataImporter()
+agent = LLMCaller()
+
+
+@app.get("/v1")
+def greet_json():
+    start_time = time.time()
+    health_status = {
+        "status": "healthy",
+        "timestamp": datetime.utcnow().isoformat(),
+        "service": "SealionAI Travel Planning Service",
+        "version": "1.0.0",
+        "checks": {}
+    }
+    return health_status
+
+@app.post("/v1/generateTripPlan", response_model=PlanResponse)
+def generate_trip_plan(request: PlanRequest):
+    try:
+        trip_plan = asyncio.run(agent.query_with_rag(request))
+        return PlanResponse(tripOverview=trip_plan.tripOverview,
+                            query_params=request,
+                            retrieved_data=trip_plan.retrieved_data,
+                            trip_plan=trip_plan.trip_plan,
+                            meta={"status": "success", "timestamp": datetime.utcnow().isoformat()})
+    except Exception as e:
+        print(f"Error in generate_trip_plan: {e}")
+        # Return error response
+        return PlanResponse(
+            tripOverview=f"Error: {str(e)}",
+            query_params=request,
+            retrieved_data=[],
+            trip_plan=TripPlan(overview="Error occurred", total_estimated_cost=0.0, steps=[]),
+            meta={"status": "error", "error": str(e)}
+        )
+
+@app.post("/v1/addYoutubeLink", response_model=YoutubeLinkResponse)
+def add_youtube_link(request: YoutubeLinkRequest):
+    try:
+        data_importer.insert_from_youtube(request.video_id)
+    except Exception as e:
+        return YoutubeLinkResponse(
+            message="Failed to add YouTube link",
+            video_url= None
+        )
+    return YoutubeLinkResponse(
+        message="add successfully",
+        video_url=f"https://www.youtube.com/watch?v={request.video_id}"
+    )
+
+@app.post("/v1/searchSimilar", response_model=list[dict])
+def search_similar(request: YoutubeLinkRequest):
+    try:
+        results = data_importer.search_similar(query=request.video_id)
+        return results
+    except Exception as e:
+        print(f"Error during search: {e}")
+        return {"error": "Search failed"}
+    return []
+
+@app.post("/v1/basicChat", response_model=str)
+def basic_chat(request: ChatRequest):
+    user_message = request.message
+    llm_response = asyncio.run(agent.basic_query(
+        user_prompt=user_message
+    ))
+    return llm_response
+

app.py

@@ -85,19 +85,6 @@ def generate_trip_plan(request: PlanRequest):
             # Wait before retrying
             logger.info(f"Retrying in {RETRY_DELAY} seconds...")
             time.sleep(RETRY_DELAY)
-            
-        # except Exception as e:
-        #     # For non-timeout errors, don't retry
-        #     logger.error(f"Non-timeout error in generate_trip_plan: {e}")
-        #     raise HTTPException(
-        #         status_code=500,
-        #         detail={
-        #             "error": "Internal server error",
-        #             "message": "Failed to generate trip plan",
-        #             "details": str(e),
-        #             "attempt": attempt + 1
-        #         }
-        #     )
 
 
 @app.post("/v1/addYoutubeLink", response_model=YoutubeLinkResponse)

utils/llm_caller.py

@@ -104,7 +104,7 @@ class LLMCaller:
             retrieved_data = []
             context_text = ""
 
-            print(f"Search results: {search_results['result']}")
+            # print(f"Search results: {search_results['result']}")
             for result in search_results['result']:
                 retrieved_item = RetrievedItem(
                     place_id=result.get('id', 'Unknown'),
@@ -114,6 +114,7 @@ class LLMCaller:
                 retrieved_data.append(retrieved_item)
                 context_text += f"\n{result['payload'].get('text', '')}"
 
+            print(context_text)
             # 5. Create detailed prompt for LLM - updated with new fields
             llm_prompt = f"""Generate a travel plan in JSON format for:
             From: {plan_request.start_place} → To: {destination}