Add Python path fix to all modules

api/main.py

@@ -1,4 +1,9 @@
 # api/main.py
+# Add project root to Python path
+import sys
+from pathlib import Path
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
 from typing import Optional
 from contextlib import asynccontextmanager
 from fastapi import FastAPI, Query, HTTPException

database/mongodb_client.py

@@ -1,3 +1,8 @@
+# Add project root to Python path
+import sys
+from pathlib import Path
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
 from pymongo import MongoClient, ReadPreference
 from pymongo.errors import ServerSelectionTimeoutError, ConnectionFailure
 from langchain_mongodb.vectorstores import MongoDBAtlasVectorSearch

database/postgres_checkpointer.py

@@ -1,4 +1,8 @@
 # database/postgres_checkpointer.py - CORRECT VERSION
+# Add project root to Python path
+import sys
+from pathlib import Path
+sys.path.insert(0, str(Path(__file__).parent.parent))
 from psycopg_pool import AsyncConnectionPool
 from psycopg.rows import dict_row
 from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver  # ✅ Correct import