style: Fix linting errors reported by Ruff

Author: PryskaS

lab_reflection_agent.ipynb

@@ -19,7 +19,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import os\n",
     "from dotenv import load_dotenv\n",
     "\n",
     "# Import our custom agent class\n",

main.py

@@ -4,12 +4,12 @@
 from pydantic import BaseModel, Field
 from typing import List
 
-# Load environment variables from a .env file
-load_dotenv()
-
 # Import core logic
 from reflection_pattern_agent.reflection_agent import ReflectionAgent
 
+# Load environment variables from a .env file
+load_dotenv()
+
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)

tests/test_reflection_agent.py

@@ -1,4 +1,3 @@
-import pytest
 from reflection_pattern_agent.reflection_agent import ReflectionAgent
 
 # Create a fake response object that looks exactly like the real OpenAI response.