Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
fcee7d8
1
Parent(s):
5f0d6a0
pin requirements
Browse files- baseline/requirements.txt +6 -6
- baseline/run.py +2 -0
baseline/requirements.txt
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
-i https://pypi.org/simple
|
| 2 |
-
datasets
|
| 3 |
-
huggingface-hub
|
| 4 |
-
smolagents
|
| 5 |
-
litellm
|
| 6 |
-
tenacity
|
| 7 |
-
arize-phoenix
|
|
|
|
| 1 |
-i https://pypi.org/simple
|
| 2 |
+
datasets==3.2.0
|
| 3 |
+
huggingface-hub==0.27.1
|
| 4 |
+
smolagents==1.3.0
|
| 5 |
+
litellm==1.58.2
|
| 6 |
+
tenacity==9.0.0
|
| 7 |
+
arize-phoenix==7.8.1
|
baseline/run.py
CHANGED
|
@@ -76,6 +76,8 @@ def run_single_task(
|
|
| 76 |
guidelines=task["guidelines"]
|
| 77 |
)
|
| 78 |
agent = create_code_agent_with_reasoning_llm(model_id, api_base, api_key, max_steps, ctx_path)
|
|
|
|
|
|
|
| 79 |
else:
|
| 80 |
prompt = chat_llm_task_prompt.format(
|
| 81 |
ctx_path=ctx_path,
|
|
|
|
| 76 |
guidelines=task["guidelines"]
|
| 77 |
)
|
| 78 |
agent = create_code_agent_with_reasoning_llm(model_id, api_base, api_key, max_steps, ctx_path)
|
| 79 |
+
prompt = agent.system_prompt + "\n" + prompt
|
| 80 |
+
agent.system_prompt = ""
|
| 81 |
else:
|
| 82 |
prompt = chat_llm_task_prompt.format(
|
| 83 |
ctx_path=ctx_path,
|