Added add_messages to make bulk load from LibreChat.

just_agents/llm_session.py

@@ -51,16 +51,17 @@ def instruct(self, prompt: str):
         self.memory.add_message(system_instruction, True)
         return system_instruction
 
-    def query(self, prompt: str, stream: bool = False, run_callbacks: bool = True, output: Optional[Path] = None) -> str:
+    def query(self, prompt: str = None, stream: bool = False, run_callbacks: bool = True, output: Optional[Path] = None) -> str:
         """
         Query large language model
         :param prompt:
         :param stream:
         :param run_callbacks:
         :return:
         """
-        question = Message(role="user", content=prompt)
-        self.memory.add_message(question)
+        if prompt is not None:
+            question = Message(role="user", content=prompt)
+            self.memory.add_message(question)
         options: Dict = self.llm_options
         response: ModelResponse = completion(messages=self.memory.messages, stream=stream, **options)
         self._process_response(response)

just_agents/memory.py

@@ -34,4 +34,13 @@ def add_message(self, message: Message, run_callbacks: bool = True):
 
     @property
     def last_message(self) -> Optional[Message]:
-        return self.messages[-1] if len(self.messages) > 0 else None
+        return self.messages[-1] if len(self.messages) > 0 else None
+
+
+    def add_messages(self, messages: list, run_callbacks: bool = True):
+        for message in messages:
+            msg = Message(content=message["content"], role=message["role"])
+            self.messages.append(msg)
+            if run_callbacks:
+                for handler in self.on_message:
+                    handler(message)