Commit Graph

9 Commits

Author SHA1 Message Date
acano f747c140c8 Enhance generate_mbap_v2.py with new reward mechanism and GoldPool integration
- Added GoldPool class to manage a top-K pool of high-reward examples.
- Implemented compute_reward function to calculate composite rewards based on execution coverage, novelty, and test quality.
- Introduced call_api_reward function for API calls in the new reward mode.
- Updated main function to support new reward mode with adjustable weights for ECS, novelty, and test quality.
- Enhanced dataset saving functionality to include reward statistics.
- Refactored existing code for improved readability and consistency.
2026-03-27 14:04:21 +01:00
acano c6b57849cd Merge branch 'online' of github.com:BRUNIX-AI/assistance-engine into mrh-online-dev 2026-03-26 17:02:27 +01:00
rafa-ruiz fe43cd6fa9 scripts documentation 2026-03-26 07:51:01 -07:00
acano d50f33c707 Merge branch 'online' of github.com:BRUNIX-AI/assistance-engine into mrh-online-dev 2026-03-26 09:37:57 +01:00
rafa-ruiz ccd9073a52 feat(dataset): add ADR-0006 and scaffold reward algorithm pipeline 2026-03-25 22:19:19 -07:00
rafa-ruiz 90857e1b0a UPDATE: Modified LRM and generate_mbap.py to ensure better samples 2026-03-11 20:09:05 -07:00
rafa-ruiz b5167b71e3 UPDATE: Sample generator now includes a new key in each item. 2026-03-11 12:22:08 -07:00
acano 5f21544e0b Refactor Elasticsearch ingestion pipeline and add MBPP generation script
- Updated `elasticsearch_ingestion.py` to streamline document processing and ingestion into Elasticsearch.
- Introduced `generate_mbap.py` for generating benchmark problems in AVAP language from a provided LRM.
- Created `prompts.py` to define prompts for converting Python problems to AVAP.
- Enhanced chunk processing in `chunk.py` to support markdown and AVAP documents.
- Added `OllamaEmbeddings` class in `embeddings.py` for handling embeddings with Ollama model.
- Updated dependencies in `uv.lock` to include new packages and versions.
2026-03-11 17:17:44 +01:00
rafa-ruiz 35ca56118d feat: add MBPP-style dataset generator and evaluation docs 2026-03-10 13:37:19 -07:00