reverse to original file

app/main.py

@@ -17,16 +17,12 @@ models.Base.metadata.create_all(bind=engine)
 
 app = FastAPI()
 
-# def get_db():
-#     db = SessionLocal()
-#     try:
-#         yield db
-#     finally:
-#         db.close()
-
-# @app.get("/")
-# async def read_root():
-#     return {"message": "API is running!"}
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
 
 
 async def generate_stream(input_data: schemas.BaseModel, runnable: Runnable, callbacks: List[BaseCallbackHandler]=[]):
@@ -49,13 +45,6 @@ async def simple_stream(request: Request):
 #     raise NotImplemented
 
 
-# def get_db():
-#     db = SessionLocal()
-#     try:
-#         yield db
-#     finally:
-#         db.close()
-
 # @app.post("/history/stream")
 # async def history_stream(request: Request, db: Session = Depends(get_db)):  
 #     # TODO: Let's implement the "/history/stream" endpoint. The endpoint should follow those steps: