Update app.py

app.py

@@ -90,11 +90,10 @@ async def chat_endpoint(request: ChatRequest):
     async def event_generator():
         print(f"--- 🚀 Starting stream for {request.thread_id} ---")
         
-        # Use 'v2' if you are on the latest LangGraph, but 'v1' is safer for compatibility
+        # We iterate over ALL events
         async for event in rag_app.astream_events(inputs, config=config, version="v1"):
             
-            # [DEBUG] Print the event type to your Hugging Face Logs
-            # This will show us if the events are firing but named differently
+            # [DEBUG] Print event types to your terminal so you can see what's happening
             event_type = event.get("event")
             
             # Logic: We don't care about the event name. 
@@ -103,11 +102,11 @@ async def chat_endpoint(request: ChatRequest):
             chunk = data.get("chunk")
             
             # Check if chunk exists and has .content attribute (standard LangChain message chunk)
-            if chunk and hasattr(chunk, "content") and chunk.content:
+            if chunk and hasattr(chunk, "content"):
                 content = chunk.content
                 
                 # Filter out empty strings or weird artifacts
-                if content.strip() != "":
+                if content:
                     # JSON encode the content
                     chunk_json = json.dumps({"content": content})
                     yield f"data: {chunk_json}\n\n"
@@ -122,7 +121,7 @@ async def chat_endpoint(request: ChatRequest):
             "Cache-Control": "no-cache",
             "Connection": "keep-alive",
             "Content-Type": "text/event-stream",
-            "X-Accel-Buffering": "no",
+            "X-Accel-Buffering": "no", # Critical for HF
         },
     )