Fix import statements

test_agent.py

@@ -7,7 +7,8 @@ from vectara_agent.agent import Agent
 import sqlite3
 from datasets import load_dataset
 
-from app import initialize_agent, setup_db
+from app import setup_db
+from agent import initialize_agent
 
 from dotenv import load_dotenv
 load_dotenv(override=True)