Merge pull request #182 from kyaukyuai/fix/project

Refactor creating human message with task information
kyaukyuai · Mar 16, 2024 · c4ba403 · c4ba403
2 parents 35a7dd6 + 2b589f5
commit c4ba403
Show file tree

Hide file tree

Showing 3 changed files with 662 additions and 2,103 deletions.
diff --git a/gpt_all_star/core/message.py b/gpt_all_star/core/message.py
@@ -11,7 +11,10 @@ def create_system_message(message: str) -> SystemMessage:
         return SystemMessage(content=message)
 
     @staticmethod
-    def create_human_message(message: str, name: Optional[str] = None) -> HumanMessage:
+    def create_human_message(
+        message: str,
+        name: Optional[str] = None,
+    ) -> HumanMessage:
         return HumanMessage(content=message, name=name)
 
     @staticmethod

diff --git a/gpt_all_star/core/project.py b/gpt_all_star/core/project.py
@@ -187,7 +187,13 @@ def chat(self, message: str) -> None:
             for task in step.additional_tasks():
                 tasks["plan"].append(task)
 
-            yield {"tasks": tasks}
+            yield {
+                "messages": [
+                    Message.create_human_message(
+                        message=str(tasks), name=self.supervisor.name
+                    )
+                ],
+            }
 
             count = 1
             while len(tasks["plan"]) > 0: