@echo off
REM Test Novita AI connection using Anaconda environment
REM This script activates the conda environment and runs the test

echo ============================================================
echo Testing Novita AI Connection with Anaconda
echo ============================================================
echo.

REM Check if conda is available
where conda >nul 2>&1
if %ERRORLEVEL% NEQ 0 (
    echo ERROR: conda command not found
    echo Please activate Anaconda Prompt first or add conda to PATH
    goto :end
)

echo Step 1: Checking conda environments...
call conda env list

echo.
echo Step 2: Creating environment if it doesn't exist...
call conda env create -f environment.yml --name research-ai-assistant 2>nul
if %ERRORLEVEL% NEQ 0 (
    echo Environment may already exist, continuing...
)

echo.
echo Step 3: Activating environment and running test...
call conda activate research-ai-assistant
if %ERRORLEVEL% NEQ 0 (
    echo ERROR: Failed to activate environment
    echo Try: conda activate research-ai-assistant
    goto :end
)

echo.
echo Step 4: Installing openai package if needed...
python -c "import openai" 2>nul
if %ERRORLEVEL% NEQ 0 (
    echo Installing openai package...
    pip install openai>=1.0.0
)

echo.
echo Step 5: Running Novita AI connection test...
python test_novita_connection.py

:end
echo.
echo Test complete!
pause