Second-Me/scripts
JimmyZQX f5bb0dad59
Data Filtering with Gemma (#396)
* Add code for data filtering llm judge

* Ignore log file created on root (mainly for synthetic_data_generation.log)

* Fix metadata API compatibility issues by commenting out metadata tags in LLM API calls

- Commented out metadata.tags parameters in all LLM API calls across the codebase
- This fixes compatibility issues with custom LLM providers that don't support metadata
- Affects shades generation, topics generation, wiki generation, bio QA, and question generation
- Preserves the original code structure for future re-enabling if needed

* feat: add data filtering pipeline with Ollama integration

- Add MergedDataJudge class for intelligent data filtering using Ollama Gemma
- Integrate automatic Ollama CLI installation into project setup process
- Add DATA_FILTERING step to training pipeline with concurrent processing
- Include testing for MergedDataJudge in its local main() function
- Add Ollama dependency to pyproject.toml

* feat: add automatic Ollama model cleanup after data filtering

* Add logging for outputting data filtering parameters

* fix: adjust error handling for MergedDataJudge:
- Keep original merged.json unchanged when any error occurs
- Exit filtering process immediately on errors instead of continuing with defaults
- Ensure training pipeline continues safely even if data filtering fails

* Add frontend for data filtering pipeline

* resolve data filtering quality_level error by commenting out problematic fields, change TrainProcessService back to original class definition

* fix: quote unquoted shade icons to prevent JSON parsing errors

* Fixed wiki_res.json missing due to no database connection at wiki/base.py module import

* Added scoring reasoning as part of the merged data

* fix: filter ANSI escape sequences from Ollama logs in data filtering step

* fix: Add data filtering steps to cloud training to resolve KeyError

- Added 'Data Filtering' step to cloud training progress holder
- Added data filtering step execution in cloud training service
- Added data filtering parameters to cloud training routes
- Updated frontend to send data filtering parameters
- Fixed missing except clause in cloud training service

This resolves the KeyError: 'data_filtering' when switching from cloud to local training.
2025-08-15 11:19:12 +08:00
..
utils better symbol (#344) 2025-05-13 14:22:47 +08:00
downgrade_migration.py Feature/0416/add thinking mode (#264) 2025-04-24 14:19:23 +08:00
migrate_add_instance_password.py Initial commit 2025-03-20 00:37:54 +08:00
prompt_cuda.bat Added CUDA support (#228) 2025-04-25 10:20:36 +08:00
prompt_cuda.sh add execute right of prompt_cuda (#345) 2025-05-13 14:20:49 +08:00
restart-backend.sh Feat: Remove conda from setup (#195) 2025-04-10 19:39:26 +08:00
restart-force.sh fix: incorrect script paths in stop.sh and start.sh (#203) 2025-04-11 10:29:42 +08:00
restart.sh Feat: Remove conda from setup (#195) 2025-04-10 19:39:26 +08:00
run_migrations.py Feature/0416/add thinking mode (#264) 2025-04-24 14:19:23 +08:00
setup.sh Data Filtering with Gemma (#396) 2025-08-15 11:19:12 +08:00
start.sh Feat: Remove conda from setup (#195) 2025-04-10 19:39:26 +08:00
start_frontend.sh Feat: Remove conda from setup (#195) 2025-04-10 19:39:26 +08:00
start_local.sh Feature/0416/add thinking mode (#264) 2025-04-24 14:19:23 +08:00
status.sh Feat: Remove conda from setup (#195) 2025-04-10 19:39:26 +08:00
stop.sh Fix/cloud service stop (#384) 2025-06-17 15:07:39 +08:00