sohojoe commited on
Commit
0df6dd9
β€’
1 Parent(s): f249d6b

refactor charles_actor -> charles_app

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. charles_actor.py β†’ charles_app.py +3 -47
app.py CHANGED
@@ -51,7 +51,7 @@ def init_ray():
51
  # charles_actor_proc = subprocess.Popen([sys.executable, "charles_actor.py"])
52
  # while charles_actor_instance == None:
53
  # try:
54
- # charles_actor_instance = ray.get_actor("CharlesActor")
55
  # except ValueError as e:
56
  # time.sleep(0.1) # give the subprocess a chance to start
57
  # return charles_actor_instance
 
51
  # charles_actor_proc = subprocess.Popen([sys.executable, "charles_actor.py"])
52
  # while charles_actor_instance == None:
53
  # try:
54
+ # charles_actor_instance = ray.get_actor("CharlesApp")
55
  # except ValueError as e:
56
  # time.sleep(0.1) # give the subprocess a chance to start
57
  # return charles_actor_instance
charles_actor.py β†’ charles_app.py RENAMED
@@ -9,11 +9,11 @@ from respond_to_prompt_async import RespondToPromptAsync
9
  import asyncio
10
  import subprocess
11
 
12
- class CharlesActor:
13
  def __init__(self):
14
  self._needs_init = True
15
  self._charles_actor_debug_output = ""
16
- self._environment_state:EnvironmentState = EnvironmentState(episode=0, step=0) # Initialize as EnvironmentState
17
  self._state = "Initializing"
18
  self._clip_transform = CLIPTransform()
19
 
@@ -246,52 +246,8 @@ async def main():
246
  if not ray.is_initialized():
247
  init_ray()
248
 
249
- # charles_actor = CharlesActor.options(
250
- # name="CharlesActor",
251
- # get_if_exists=True,
252
- # ).remote()
253
- # future = charles_actor.start.remote()
254
- charles_actor = CharlesActor()
255
  await charles_actor.start()
256
-
257
- last_step = -1
258
- last_episode = -1
259
- try:
260
- while True:
261
- ready, _ = ray.wait([future], timeout=0)
262
- if ready:
263
- # The start method has terminated. You can fetch the result (if any) with ray.get().
264
- # If the method raised an exception, it will be re-raised here.
265
- try:
266
- result = ray.get(future)
267
- print(f"The start method has terminated with result: {result}")
268
- except Exception as e:
269
- print(f"The start method raised an exception: {e}")
270
- break
271
- else:
272
- # The start method is still running. You can poll for debug information here.
273
- await asyncio.sleep(1)
274
- state = await charles_actor.get_state.remote()
275
- env_state = await charles_actor.get_environment_state.remote()
276
- if (env_state.episode != last_episode) or (env_state.step != last_step):
277
- last_episode = env_state.episode
278
- last_step = env_state.step
279
- print(f"Charles is in state: {state}")
280
- # if len(env_state.llm_preview):
281
- # print (f"llm_preview: {env_state.llm_preview}")
282
- # if len(env_state.llm_responses):
283
- # print (f"llm_responses: {env_state.llm_responses}")
284
- # if len(env_state.tts_raw_chunk_ids):
285
- # for chunk_json in env_state.tts_raw_chunk_ids:
286
- # chunk = json.loads(chunk_json)
287
- # prompt = chunk['prompt']
288
- # line = chunk['llm_sentence_id']
289
- # chunk_id = chunk['chunk_count']
290
- # print(f"Prompt: {prompt}, Line: {line}, Chunk: {chunk_id}")
291
-
292
- except KeyboardInterrupt as e:
293
- print("Script was manually terminated")
294
- raise(e)
295
 
296
 
297
  if __name__ == "__main__":
 
9
  import asyncio
10
  import subprocess
11
 
12
+ class CharlesApp:
13
  def __init__(self):
14
  self._needs_init = True
15
  self._charles_actor_debug_output = ""
16
+ self._environment_state:EnvironmentState = EnvironmentState(episode=0, step=0)
17
  self._state = "Initializing"
18
  self._clip_transform = CLIPTransform()
19
 
 
246
  if not ray.is_initialized():
247
  init_ray()
248
 
249
+ charles_actor = CharlesApp()
 
 
 
 
 
250
  await charles_actor.start()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
251
 
252
 
253
  if __name__ == "__main__":