| import cmbagent | |
| def test_one_shot_engineer_oss(): | |
| task = r""" | |
| Compute the sum of the first 1000 natural numbers. | |
| Plot the histogram of the numbers. | |
| """ | |
| task = r""" | |
| Analyse /Users/boris/Desktop/data/Stocks.csv file using pandas. | |
| and then plot relevant statistics with matplotlib. | |
| """ | |
| results = cmbagent.one_shot( | |
| task, | |
| max_rounds=25, | |
| max_n_attempts=10, | |
| agent="engineer", | |
| engineer_model="gpt-oss-120b", | |
| default_formatter_model="gpt-oss-120b", | |
| default_llm_model="gpt-oss-120b", | |
| work_dir="/Users/boris/Desktop/one_shot_engineer_oss", | |
| clear_work_dir=True, | |
| ) | |
| assert results is not None | |
| test_one_shot_engineer_oss() | |