Update src/main.py

src/main.py

@@ -4,7 +4,7 @@ from fastapi import FastAPI, HTTPException, UploadFile, File, Form
 from pydantic import BaseModel, Field
 from dotenv import load_dotenv
 from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver
-from src.graph import build_agent, AgentContext
+from src.new_graph import build_agent, AgentContext
 from src.embedding import upload_file
 import asyncio
 from psycopg_pool import AsyncConnectionPool