second_constantine
2048e4e40d
- Updated summarization prompt to require Russian output and exclude non-textual elements - Upgraded ollama dependency to v0.6.1 - Enhanced run.sh script to support both single record and file-based ID input for MongoDB test generation - Updated documentation in scripts/README.md to reflect new functionality - Added verbose flag to generate_summarization_from_mongo.py for better debugging ``` This commit message follows the conventional commit format with a short title (50-72 characters) and provides a clear description of the changes made and their purpose.
81 lines
3.0 KiB
Bash
Executable File
81 lines
3.0 KiB
Bash
Executable File
#!/bin/bash
|
||
|
||
# Получаем имя ядра (Linux – Linux, macOS – Darwin, FreeBSD – FreeBSD …)
|
||
OS_NAME=$(uname -s)
|
||
|
||
init() {
|
||
if [[ "$OS_NAME" == "Darwin" ]]; then
|
||
python3.13 -m venv z
|
||
else
|
||
python3 -m venv z
|
||
fi
|
||
upd
|
||
}
|
||
|
||
upd() {
|
||
activate
|
||
pip install -r requirements.txt --upgrade
|
||
git submodule update --remote --merge
|
||
}
|
||
|
||
clean() {
|
||
rm -rf results/*
|
||
echo "Отчеты успешно очищены"
|
||
}
|
||
|
||
activate() {
|
||
source z/bin/activate
|
||
}
|
||
|
||
echo "_= Project Scripts =_"
|
||
if [ -n "$1" ]; then
|
||
if [[ "$1" == "init" ]]; then
|
||
init
|
||
elif [[ "$1" == "upd" ]]; then
|
||
upd
|
||
elif [[ "$1" == "run" ]]; then
|
||
activate
|
||
shift
|
||
python src/main.py "$@"
|
||
elif [[ "$1" == "clean" ]]; then
|
||
clean
|
||
elif [[ "$1" == "gen" ]]; then
|
||
activate
|
||
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... "
|
||
if [[ -n "$2" ]] && [[ "$2" != "--id-file" ]]; then
|
||
# Старый формат: ./run.sh gen-mongo <record-id>
|
||
python scripts/generate_summarization_from_mongo.py --record-id "$2"
|
||
elif [[ -n "$2" ]] && [[ "$2" == "--id-file" ]]; then
|
||
# Новй формат: ./run.sh gen-mongo --id-file <file-path>
|
||
shift 2
|
||
python scripts/generate_summarization_from_mongo.py --id-file "$1"
|
||
else
|
||
echo "❌ Ошибка: Укажите либо --record-id, либо --id-file"
|
||
echo "Использование: ./run.sh gen-mongo <record-id> или ./run.sh gen-mongo --id-file <file-path>"
|
||
exit 1
|
||
fi
|
||
echo "✅ Тесты из MongoDB успешно сгенерированы"
|
||
fi
|
||
else
|
||
echo " Аргументом необходимо написать название скрипта (+опционально аргументы скрипта)"
|
||
echo "Скрипты:"
|
||
echo " * init - инициализация, устанавливает env"
|
||
echo " * upd - обновление зависимостей"
|
||
echo " * run - запуск бенчмарков (translation, summarization, codegen)"
|
||
echo " * clean - очистка отчетов"
|
||
echo " * gen - генерация тестов через Ollama (translation, summarization, codegen)"
|
||
echo " * gen-mongo - генерация тестов пересказов из MongoDB (использование: ./run.sh gen-mongo <record-id> или ./run.sh gen-mongo --id-file <file-path>)"
|
||
echo ""
|
||
echo "Примеры использования:"
|
||
echo " * ./run.sh run -m second_constantine/t-lite-it-1.0:7b -b translation summarization"
|
||
echo " * ./run.sh run -m second_constantine/t-lite-it-1.0:7b --num-ctx 16000"
|
||
echo " * ./run.sh gen"
|
||
echo " * ./run.sh gen-mongo 507f1f77bcf86cd799439011"
|
||
echo " * ./run.sh gen-mongo --id-file ids.txt"
|
||
fi
|