Skip to content

Commit b2d0231

Browse files
committed
Tighten refine prompt and expand AI config defaults
1 parent 3b8b490 commit b2d0231

2 files changed

Lines changed: 48 additions & 3 deletions

File tree

jumper_extension/adapters/ai_reviewer/agent/prompts.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@
5757
custom instruction.
5858
5959
Other options are shown as unified diffs vs the original cell code so you
60-
can incorporate their changes if the instruction refers to them.
60+
can incorporate their changes ONLY if the instruction explicitly refers to them.
6161
6262
Rewrite the suggested code so it follows the user's instruction while still
6363
addressing the original bottleneck analysis. Respond with the complete
Lines changed: 47 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,50 @@
11
base_url: https://llm.scads.ai/v1
22
model: MiniMaxAI/MiniMax-M2.7
3-
max_tokens: 8000
4-
timeout: 120.0
3+
max_tokens: 16000
4+
timeout: 520.0
5+
max_retries: 2
6+
streaming: true
7+
temperature: null
8+
top_p: null
9+
seed: null
10+
enable_thinking: false
11+
# arbitrary vLLM passthrough (top_k, min_p, repetition_penalty, guided_*, ...)
12+
extra_body: {}
513
api_key_env: JUMPER_AI_API_KEY
14+
15+
known_packages:
16+
# arrays / numerics
17+
- numpy
18+
- scipy
19+
- pandas
20+
- polars
21+
- pyarrow
22+
# ML / DL
23+
- torch
24+
- tensorflow
25+
- jax
26+
- jaxlib
27+
- scikit-learn
28+
- xgboost
29+
- lightgbm
30+
- catboost
31+
- transformers
32+
- accelerate
33+
- datasets
34+
- peft
35+
# GPU compute
36+
- cupy-cuda11x
37+
- cupy-cuda12x
38+
- cupy
39+
- cuml
40+
- cudf
41+
- pynvml
42+
- triton
43+
# parallelism / JIT
44+
- numba
45+
- joblib
46+
- dask
47+
- ray
48+
# viz
49+
- matplotlib
50+
- plotly

0 commit comments

Comments
 (0)