Commit Graph

1460 Commits

Author SHA1 Message Date
you-n-g ee8c119f31 fix: set requires_documentation_search to None to disable feature in eval (#1245) 2025-09-15 07:45:08 +08:00
you-n-g 5ba5e8356c feat: init pydantic ai agent & context 7 mcp (#1240)
* feat: init pydantic ai agent & context 7 mcp

* feat: integrate MCP documentation search into data science pipeline evaluation

* fix: disable MCP documentation search and update related docstrings and defaults

* lint

* fix: correct prompt formatting and conditional blocks in pipeline_eval section

* lint

* feat: add query method to PAIAgent for synchronous agent execution

* fix: apply nest_asyncio for agent and update context7 query method

* lint

* lint

* lint

* lint

* docs: update MCP folder docstring and rename test class in test_pydantic.py

* refactor: centralize completion kwargs logic and update pydantic_ai integration

* fixbug

* typo

* fix: bug triggered by padantic-ai version backtracking.

---------

Co-authored-by: Linlang <Lv.Linlang@hotmail.com>
2025-09-13 10:25:02 +08:00
amstrongzyf 7e996055f2 fix: fix chat_max_tokens calculation method to show true input_max_tokens (#1241)
* fix: use real max_input_length

* lint

* Update rdagent/oai/backend/litellm.py

Co-authored-by: you-n-g <you-n-g@users.noreply.github.com>

* fix lint

* lint

* lint

---------

Co-authored-by: you-n-g <you-n-g@users.noreply.github.com>
2025-09-12 15:09:20 +08:00
amstrongzyf 2201a47623 fix: revert 2 commits (#1239)
* revert commit:c51a3008f897416a7416f92d1286cbbf6a2291ae. PR 1179

* Revert "fix: change runner prompts (#1223)"

This reverts commit db2ebc27d70663d6aa85e0b05d9b81f5f6443c17.

* fix ruff check error

* fix: change switch place for fix_seed_and_data_split
2025-09-11 16:14:48 +08:00
you-n-g a984f69f68 fix: move task cancellation to finally block and fix subprocess kill typo (#1234) 2025-09-10 10:05:08 +08:00
you-n-g 0daeb82d63 feat: add stdout into workspace for easier debugging (#1236)
* refactor: use get_truncated_stdout for consistent stdout handling across modules

* lint

* feat: add dump_stdout_type to DSRunnerCoSTEERSettings and use in eval

* fix: avoid circular import by moving DSRunnerEvaluator import inside method
2025-09-10 10:02:49 +08:00
Tim d4c539912a feat: offline selector (#1231)
* offline selector test
* fix score with tensor
* sort sota list

---------

Co-authored-by: Xu Yang <peteryang@vip.qq.com>
2025-09-09 11:56:42 +08:00
Star dust fc18942339 fix: add a switch for ensemble_time_upper_bound and fix some bug in main (#1226)
* change runner prompts

* v1

* ensemble_time_upper_bound

* lint

* fix inf bug in function prob_dis_torch() and ensemble prompts

* lint

* lint

---------

Co-authored-by: amstrongzyf <amstrongzyf@126.com>
2025-09-08 16:35:42 +08:00
you-n-g 86ce4f135d fix: increase retry count in hypothesis_gen decorator to 10 (#1230) 2025-09-08 12:23:58 +08:00
you-n-g 6fc3877a39 fix: handle ValueError in stdout shrinking and refactor shrink logic (#1228) 2025-09-07 23:24:45 +08:00
Tim d68b0f4223 chore: make sure token size below limit (#1225)
* chore: make sure token size below limit

* refactor: refactor and add import

---------

Co-authored-by: amstrongzyf <amstrongzyf@126.com>
2025-09-04 15:07:42 +08:00
Star dust be3433f26b fix: change runner prompts (#1223)
* change runner prompts

* v1
2025-09-04 11:31:42 +08:00
amstrongzyf 51f5bc9e11 fix: revert to v10 setting (#1220)
* feat: add runner_patience

* task gen prompts and torch version

* fix ensemble time bug

* small bug

* lint

* add torch

* fiix: remove useless code and change formula

* fix: rename parameter

---------

Co-authored-by: jingyuanlm <842442862@qq.com>
2025-09-03 16:37:33 +08:00
XianBW beab473b40 fix: summary page bug (#1219)
* fix summary bug

* fix CI
2025-09-02 17:06:54 +08:00
you-n-g 70fd91cd05 feat: ui, support disable cache (#1217)
* fix: (drop) add cache_enable for functions in ui.utils

* fix: (drop) handle no logging, rename sota_exp to last_sota_exp for clarity in running_win
2025-09-01 15:58:05 +08:00
amstrongzyf 6725f15f30 fix: jinja problem of enumerate (#1216) 2025-09-01 14:50:26 +08:00
XianBW 5bad04b439 for ui, a simple fix (#1215) 2025-09-01 14:46:51 +08:00
you-n-g f02dc5f47d fix: allow prev_out keys to be None in workspace cleanup assertion (#1214)
* fix: allow prev_out keys to be None in workspace cleanup assertion

* fix: clean workspace only for non-None DSExperiment instances
2025-08-30 07:27:12 +08:00
amstrongzyf c8bf617aca fix: add missing self parameter to instance methods in DSProposalV2ExpGen (#1213) 2025-08-30 00:45:55 +08:00
you-n-g 9de8d60669 fix: handle None output and conditional step dump in LoopBase execution (#1212) 2025-08-29 20:19:09 +08:00
you-n-g dbbe374ac8 fix: update fallback criterion (#1210)
* fix: update fallback criterion

* fix: ensure evo_fb is initialized and used correctly in fallback logic

* refactor: rename use_new_evo to should_use_new_evo for clarity
2025-08-29 17:07:41 +08:00
you-n-g f82de4a380 feat: add option to enable hyperparameter tuning only in first eval loop (#1211)
* feat: add option to enable hyperparameter tuning only in first eval loop

* fix: use total_seconds() for accurate time calculations in evolution and tracking
2025-08-29 16:59:22 +08:00
XianBW 215db05656 chore: ui updates (#1209)
* show a workspace path string for easier copy, only percent existing columns when percent summary dataframe

* catch summary exception

* fix a bug

* fix ci
2025-08-29 15:05:15 +08:00
amstrongzyf d759ca95e7 fix: fix bug for hypo_select_with_llm when not support response_schema (#1208) 2025-08-28 14:36:23 +08:00
XianBW 70fdfc89b0 fix ui bug (#1207) 2025-08-28 14:18:58 +08:00
you-n-g 774346d92e fix: move snapshot saving after step index update in loop execution (#1206) 2025-08-28 10:58:57 +08:00
amstrongzyf 90dd2f7b9b feat: enable LLM‑based hypothesis selection with time‑aware prompt & colored logging (#1122)
* add hypo select by llm (without time)

* add time_info and log color

* select no smooth

* 2 hypo

* change select

* small change

* fix bug

* fix bug and add hypothesis router and begin flag

* fix bug v1

* fix bug v2

* fix feedback

* add new model

* add filter

* fix bug v3

* fix bug v4

* change prompts v2

* fix bug v5

* fix bug v6

* fix hypo

* fix some bug(sota socre, prompts, ensemble prompts  ) and add path legth.

* fix bug v7

* fix bug v8

* fix bug v9

* fix bug v10

* reset to v10 and refine

* fix: translate to english

* fix bug

* fix: use differnet increase_stage for coder & runner

* fix: use different timeout_increase_stage for coder and runner.

* fix: revert logger

* fix: remove duplicate content

* feat: implement LLM-driven extra hypothesis selection and adjust logic

* refactor: relocate Hypothesis models below TraceChallenges

* remove torch fix bug

* fix small bug

* refactor: prefix internal methods with underscore for llm-based hypothesis selection

* add fix_seed_and_data_split and enable_simple_hypothesis

* lint

* refine proposal config order

* code review; comments

* lint

* merge int to float

---------

Co-authored-by: jingyuanlm <842442862@qq.com>
Co-authored-by: Young <afe.young@gmail.com>
2025-08-27 18:36:51 +08:00
XianBW 3df94e920c chore: refine get_summary_df and sota exp stat in log utils (#1201)
* add best valid selector to get sota exp stat

* remove unused in summary & refine get_summary_df

* fix hypothesis table bug in trace
2025-08-26 18:31:21 +08:00
Linlang a829b2aaea style: convert CLI flags from snake_case to kebab-case (#1198)
Co-authored-by: Young <afe.young@gmail.com>
2025-08-25 20:07:01 +08:00
Xu Yang e4bd647d1b fix: increase time default not controlled by LLM (#1196)
* fix: add longer timeout configuration for data science scenarios

* fix CI
2025-08-21 11:27:17 +08:00
XianBW 04878f9e70 fix: kaggle competition metric direction (#1195)
* fix leaderboard and competition direction

* ui fix
2025-08-20 16:40:53 +08:00
amstrongzyf c150d33603 fix bug for embedding caused by PR 1188 (#1194) 2025-08-18 20:15:18 +08:00
Tim 3260c0c3ff chore: refine env command (#1193) 2025-08-18 18:16:16 +08:00
XianBW 2f8261f82b fix: ui bug (#1192)
* small bug

* fix ci
2025-08-18 17:48:49 +08:00
amstrongzyf 880a6c70c4 fix: enable embedding truncation (#1188)
* fix: enable embedding auto truncation

* move embedding_utils to utils.embedding

* fix bug

* fix(embedding): improve text truncation with retry strategies and binary search optimization

* new way for embedding truncate

* fix ci errors
2025-08-18 17:20:41 +08:00
xuangu-fang b05a530126 feat: enable to inject diversity cross async multi-trace (#1173)
* init multi_trace_async_ diversity inject

* lint

* add task in context for diversity injection, add abstract class for diversity injection

* lint

* feat: update diversity injection strategy and enhance sibling context handling in experiment generation

* add always inject

---------

Co-authored-by: Xu Yang <peteryang@vip.qq.com>
2025-08-18 12:43:21 +08:00
you-n-g d493d33603 change the trace number from 3 to 1 (simpler default value) (#1191) 2025-08-18 12:13:23 +08:00
XianBW 2c0702a4ac fix ui bugs (#1190) 2025-08-18 10:56:51 +08:00
you-n-g dbe214282e fix: skip res_ratio check if timer or res_time is None (#1189) 2025-08-17 16:08:55 +08:00
Tim 4a516949b4 chore: add inference mode for model dump (#1182)
* chore: add inference mode for model dump
* check runtime environment
* update opened_trace_lines
---------

Co-authored-by: you-n-g <you-n-g@users.noreply.github.com>
2025-08-15 15:34:49 +08:00
you-n-g e0453e0058 fix: insert await asyncio.sleep(0) to yield control in loop (#1186) 2025-08-15 13:27:29 +08:00
you-n-g 481e226844 refactor: use UI_SETTING.amlt_path for combined log folder path (#1184) 2025-08-14 21:51:05 +08:00
Tim 85ebc4889d chore: catch log_workflow error (#1183) 2025-08-14 16:12:37 +08:00
amstrongzyf 5353bd31f2 fix: refine prompts and add additional package info (#1179)
* refine prompts and add additional package info

* refine prompts to be specific for GBDT models

* minor refine prompts

* use include to replace duplicate info

* refine prompts

* refactor: import DSTrace from base and remove exp_gen __init__

* lint

---------

Co-authored-by: Young <afe.young@gmail.com>
2025-08-13 23:00:23 +08:00
Tim e0f713c7d3 chore: add merge statistics (#1176)
* chore: add stat for merge
* use best selector
* use trace.hist
2025-08-13 22:59:34 +08:00
Chaos Yu b6fae75cde fix(graph): using assignment expression to avoid repeated function call (#1174)
* using assignment expression to avoid repeated function call

* format
2025-08-13 17:16:58 +08:00
Roland Minrui e77572fb53 feat: refine the logic of enabling hyperparameter tuning and add criteira (#1175)
* add 4 tuning criteria

* fix equation direction

* fix ci

* add comment

Co-authored-by: you-n-g <you-n-g@users.noreply.github.com>

---------

Co-authored-by: Xu <v-xuminrui@microsoft.com>
Co-authored-by: you-n-g <you-n-g@users.noreply.github.com>
2025-08-12 21:12:54 +08:00
you-n-g 0e3f4cf08f fix: cancel tasks on resume and kill subprocesses on termination (#1166)
* fix: cancel tasks on resume and kill subprocesses on termination

* lint

* lint
2025-08-09 19:39:13 +08:00
XianBW 02069767e1 fix exec_time in summary because of parallel (#1171) 2025-08-08 17:14:02 +08:00
XianBW 6f1a8c63e0 add auto sota selector llm log in trace (#1169) 2025-08-08 16:26:48 +08:00