kenken999 commited on
Commit
be85c6e
β€’
1 Parent(s): 2866b32
Files changed (2) hide show
  1. babyagi/babyagi.py +3 -0
  2. chroma/chroma.sqlite3 +0 -0
babyagi/babyagi.py CHANGED
@@ -868,6 +868,7 @@ async def completion(message: str, history, c=None, d=None, prompt="あγͺたは
868
  # Step 1: Pull the first incomplete task
869
  task = tasks_storage.popleft()
870
  print("\033[92m\033[1m" + "\n*****NEXT TASK*****\n" + "\033[0m\033[0m")
 
871
  print(str(task["task_name"]))
872
  yield str(task["task_name"])
873
  result_all += str(task["task_name"])
@@ -875,9 +876,11 @@ async def completion(message: str, history, c=None, d=None, prompt="あγͺたは
875
  # Send to execution function to complete the task based on the context
876
  result = execution_agent(OBJECTIVE, str(task["task_name"]))
877
  print("\033[93m\033[1m" + "\n*****TASK RESULT*****\n" + "\033[0m\033[0m")
 
878
  print(result)
879
  yield result
880
  result_all += result
 
881
 
882
  # Step 2: Enrich result and store in the results storage
883
  # This is where you should enrich the result if needed
 
868
  # Step 1: Pull the first incomplete task
869
  task = tasks_storage.popleft()
870
  print("\033[92m\033[1m" + "\n*****NEXT TASK*****\n" + "\033[0m\033[0m")
871
+ yield "\033[92m\033[1m" + "\n*****NEXT TASK*****\n" + "\033[0m\033[0m"
872
  print(str(task["task_name"]))
873
  yield str(task["task_name"])
874
  result_all += str(task["task_name"])
 
876
  # Send to execution function to complete the task based on the context
877
  result = execution_agent(OBJECTIVE, str(task["task_name"]))
878
  print("\033[93m\033[1m" + "\n*****TASK RESULT*****\n" + "\033[0m\033[0m")
879
+ yield "\033[93m\033[1m" + "\n*****TASK RESULT*****\n" + "\033[0m\033[0m"
880
  print(result)
881
  yield result
882
  result_all += result
883
+ yield result_all
884
 
885
  # Step 2: Enrich result and store in the results storage
886
  # This is where you should enrich the result if needed
chroma/chroma.sqlite3 CHANGED
Binary files a/chroma/chroma.sqlite3 and b/chroma/chroma.sqlite3 differ