lvwerra HF Staff commited on
Commit
131f73e
·
1 Parent(s): 962509d

smol fixes

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -37,8 +37,8 @@ with open(TMP_DIR+"jupyter-agent.ipynb", 'w', encoding='utf-8') as f:
37
 
38
  with open("ds-system-prompt.txt", "r") as f:
39
  DEFAULT_SYSTEM_PROMPT = f.read()
40
- DEFAULT_SYSTEM_PROMPT = """You are a coding agent with access to a Jupyter Kernel.\
41
- The following files are available:
42
  {}
43
  """
44
 
@@ -75,7 +75,7 @@ def execute_jupyter_agent(
75
  message_history.append(
76
  {
77
  "role": "system",
78
- "content": sytem_prompt.format("- " + "\n- ".join(filenames)),
79
  }
80
  )
81
  message_history.append({"role": "user", "content": user_input})
 
37
 
38
  with open("ds-system-prompt.txt", "r") as f:
39
  DEFAULT_SYSTEM_PROMPT = f.read()
40
+ DEFAULT_SYSTEM_PROMPT = """You are a coding agent with access to a Jupyter Kernel. \
41
+ The following files are available (if any):
42
  {}
43
  """
44
 
 
75
  message_history.append(
76
  {
77
  "role": "system",
78
+ "content": sytem_prompt.format("- " + "\n- ".join(filenames)) if files is not None else "None",
79
  }
80
  )
81
  message_history.append({"role": "user", "content": user_input})