Update app/main.py

app/main.py

@@ -5,7 +5,9 @@ from sse_starlette.sse import EventSourceResponse
 from sqlalchemy.orm import Session
 from langserve.serialization import WellKnownLCSerializer
 from typing import Any, List
-import crud, models, schemas
+import crud
+import models
+import schemas
 from database import SessionLocal, engine
 from chains import simple_chain, formatted_chain, history_chain, rag_chain
 from prompts import format_chat_history
@@ -87,10 +89,6 @@ async def rag_stream(request: Request, db: Session = Depends(get_db)):
         chat_history=format_chat_history(chat_history),
     )
 
-    stream = rag_chain.stream(
-        rag_input.dict(),
-        config={"callbacks": [Callback()]}
-    )
     callback = LogResponseCallback(user_request, db)
     return EventSourceResponse(generate_stream(rag_input, callback))