AI fixes tests

tests/test_core_helpers.py

@@ -1,10 +1,16 @@
 """Unit tests for prompt/token/engine helper utilities."""
 from __future__ import annotations
 
+import sys
 import types
+from pathlib import Path
 
 import pytest
 
+ROOT = Path(__file__).resolve().parents[1]
+if str(ROOT) not in sys.path:
+    sys.path.insert(0, str(ROOT))
+
 from app.core import engine, tokens
 from app.core.prompting import DEFAULT_SYSTEM_PROMPT, render_chat_prompt
 from app.schemas.chat import ChatMessage

tests/test_main_behavior.py

@@ -3,10 +3,16 @@ from __future__ import annotations
 
 import asyncio
 import json
+import sys
+from pathlib import Path
 
 from fastapi import HTTPException
 from starlette.requests import Request
 
+ROOT = Path(__file__).resolve().parents[1]
+if str(ROOT) not in sys.path:
+    sys.path.insert(0, str(ROOT))
+
 from app import main