starsnatched commited on
Commit
4477cf4
·
1 Parent(s): e98e0c1

Update document upload logic and extend command execution timeout to 30 seconds

Browse files
Files changed (2) hide show
  1. run.py +3 -3
  2. src/tools.py +2 -2
run.py CHANGED
@@ -7,10 +7,10 @@ from src.chat import ChatSession
7
 
8
  async def _main() -> None:
9
  async with ChatSession(user="demo_user", session="demo_session") as chat:
10
- # doc_path = chat.upload_document("test.txt")
11
  # print(f"Document uploaded to VM at: {doc_path}")
12
- answer = await chat.chat(f"Remove all contents of test.txt and add the text 'Hello, World!' to it.")
13
- # answer = await chat.chat("What is in /data directory?")
14
  print("\n>>>", answer)
15
 
16
 
 
7
 
8
  async def _main() -> None:
9
  async with ChatSession(user="demo_user", session="demo_session") as chat:
10
+ doc_path = chat.upload_document("test.txt")
11
  # print(f"Document uploaded to VM at: {doc_path}")
12
+ # answer = await chat.chat(f"Remove all contents of test.txt and add the text 'Hello, World!' to it.")
13
+ answer = await chat.chat("What is in /data directory?")
14
  print("\n>>>", answer)
15
 
16
 
src/tools.py CHANGED
@@ -25,9 +25,9 @@ def execute_terminal(command: str) -> str:
25
 
26
  The command is executed with network access enabled. Output from both
27
  ``stdout`` and ``stderr`` is captured and returned. Commands are killed if
28
- they exceed 2 seconds.
29
  """
30
- timeout = 2
31
  if not command:
32
  return "No command provided."
33
 
 
25
 
26
  The command is executed with network access enabled. Output from both
27
  ``stdout`` and ``stderr`` is captured and returned. Commands are killed if
28
+ they exceed 30 seconds.
29
  """
30
+ timeout = 30
31
  if not command:
32
  return "No command provided."
33