Girish Jeswani commited on
Commit
c3523ce
·
1 Parent(s): 4a9d22d

add new imports

Browse files
multi_llm_chatbot_backend/app/api/routes/__init__.py CHANGED
@@ -5,6 +5,7 @@ from .sessions import router as session_router
5
  from .provider import router as provider_router
6
  from .debug import router as debug_router
7
  from .root import router as root_router
 
8
 
9
  router = APIRouter()
10
  router.include_router(chat_router)
@@ -13,3 +14,4 @@ router.include_router(session_router)
13
  router.include_router(provider_router)
14
  router.include_router(debug_router)
15
  router.include_router(root_router)
 
 
5
  from .provider import router as provider_router
6
  from .debug import router as debug_router
7
  from .root import router as root_router
8
+ from .phd_canvas import router as canvas_router
9
 
10
  router = APIRouter()
11
  router.include_router(chat_router)
 
14
  router.include_router(provider_router)
15
  router.include_router(debug_router)
16
  router.include_router(root_router)
17
+ router.include_router(canvas_router)
multi_llm_chatbot_backend/app/core/database.py CHANGED
@@ -27,10 +27,19 @@ async def connect_to_mongo():
27
 
28
  # Test connection
29
  await db.client.admin.command('ping')
 
 
 
30
  logger.info(f"Successfully connected to MongoDB database: {db_name}")
31
 
32
  # Create indexes for better performance
33
  await create_indexes()
 
 
 
 
 
 
34
 
35
  except ConnectionFailure as e:
36
  logger.error(f"Failed to connect to MongoDB: {e}")
 
27
 
28
  # Test connection
29
  await db.client.admin.command('ping')
30
+
31
+ # Create indexes for better performance
32
+
33
  logger.info(f"Successfully connected to MongoDB database: {db_name}")
34
 
35
  # Create indexes for better performance
36
  await create_indexes()
37
+ try:
38
+ from app.core.canvas_database import setup_canvas_collections
39
+ await setup_canvas_collections(db) # Pass db directly
40
+ logger.info("Canvas database initialization completed")
41
+ except Exception as canvas_error:
42
+ logger.error(f"Canvas database initialization failed: {canvas_error}")
43
 
44
  except ConnectionFailure as e:
45
  logger.error(f"Failed to connect to MongoDB: {e}")