feat: add MongoDB test generation and update dependencies
- Added pymongo==3.13.0 to requirements.txt for MongoDB connectivity - Implemented generate_summarization_from_mongo.py script to generate summarization tests from MongoDB - Updated run.sh to support 'gen-mongo' command for MongoDB test generation - Enhanced scripts/README.md with documentation for new MongoDB functionality - Improved help text in run.sh to clarify available commands and usage examples ``` This commit adds MongoDB integration for test generation and updates the documentation and scripts accordingly.
This commit is contained in:
17
run.sh
17
run.sh
@@ -44,13 +44,24 @@ if [ -n "$1" ]; then
|
||||
echo "🤖 Генерирую тесты через Ollama..."
|
||||
python scripts/generate_tests.py --count 1 --category all --model second_constantine/t-lite-it-1.0:7b --ollama-url http://10.0.0.4:11434
|
||||
echo "✅ Тесты успешно сгенерированы"
|
||||
elif [[ "$1" == "gen-mongo" ]]; then
|
||||
activate
|
||||
echo "🔍 Генерирую тесты пересказов из MongoDB... "
|
||||
python scripts/generate_summarization_from_mongo.py --record-id "$2"
|
||||
echo "✅ Тесты из MongoDB успешно сгенерированы"
|
||||
fi
|
||||
else
|
||||
echo " Аргументом необходимо написать название скрипта (+опционально аргументы скрипта)"
|
||||
echo "Скрипты:"
|
||||
echo " * init - инициализация, устанавливает env"
|
||||
echo " * upd - обновление зависимостей"
|
||||
echo " * run - запуск бенчмарков"
|
||||
echo " * run - запуск бенчмарков (translation, summarization, codegen)"
|
||||
echo " * clean - очистка отчетов"
|
||||
echo " * gen - генерация тестов через Ollama"
|
||||
fi
|
||||
echo " * gen - генерация тестов через Ollama (translation, summarization, codegen)"
|
||||
echo " * gen-mongo - генерация тестов пересказов из MongoDB (использование: ./run.sh gen-mongo <record-id> [output-dir])"
|
||||
echo ""
|
||||
echo "Примеры использования:"
|
||||
echo " * ./run.sh run -m second_constantine/t-lite-it-1.0:7b -b translation summarization"
|
||||
echo " * ./run.sh gen"
|
||||
echo " * ./run.sh gen-mongo 507f1f77bcf86cd799439011"
|
||||
fi
|
||||
|
||||
Reference in New Issue
Block a user