langroid 0.52.5__tar.gz → 0.52.6__tar.gz

{langroid-0.52.5 → langroid-0.52.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langroid
-Version: 0.52.5
+Version: 0.52.6
 Summary: Harness LLMs with Multi-Agent Programming
 Author-email: Prasad Chalasani <pchalasani@gmail.com>
 License: MIT

{langroid-0.52.5 → langroid-0.52.6}/langroid/agent/chat_agent.py

@@ -735,7 +735,11 @@ class ChatAgent(Agent):
                 self.llm_tools_usable.discard(t)
                 self.llm_functions_usable.discard(t)
 
-        # Set tool instructions and JSON format instructions
+        self._update_tool_instructions()
+
+    def _update_tool_instructions(self) -> None:
+        # Set tool instructions and JSON format instructions,
+        # in case Tools have been enabled/disabled.
         if self.config.use_tools:
             self.system_tool_format_instructions = self.tool_format_rules()
         self.system_tool_instructions = self.tool_instructions()
@@ -977,6 +981,8 @@ class ChatAgent(Agent):
             self.llm_tools_usable.discard(t)
             self.llm_functions_usable.discard(t)
 
+        self._update_tool_instructions()
+
     def disable_message_use_except(self, message_class: Type[ToolMessage]) -> None:
         """
         Disable this agent from USING ALL messages EXCEPT a message class (Tool)
@@ -988,6 +994,7 @@ class ChatAgent(Agent):
         for r in to_remove:
             self.llm_tools_usable.discard(r)
             self.llm_functions_usable.discard(r)
+        self._update_tool_instructions()
 
     def _load_output_format(self, message: ChatDocument) -> None:
         """

{langroid-0.52.5 → langroid-0.52.6}/langroid/agent/special/doc_chat_agent.py

@@ -123,6 +123,9 @@ class DocChatAgentConfig(ChatAgentConfig):
         None  # filter condition for various lexical/semantic search fns
     )
     conversation_mode: bool = True  # accumulate message history?
+    # retain retrieved context? Setting to True increases token consumption, but
+    # helps LLM fix citation errors and improve accuracy of follow-up questions.
+    retain_context: bool = False
     # In assistant mode, DocChatAgent receives questions from another Agent,
     # and those will already be in stand-alone form, so in this mode
     # there is no need to convert them to stand-alone form.
@@ -861,8 +864,11 @@ class DocChatAgent(ChatAgent):
         # one for `final_prompt`, and one for the LLM response
 
         if self.config.conversation_mode:
-            # respond with temporary context
-            answer_doc = super()._llm_response_temp_context(question, final_prompt)
+            if self.config.retain_context:
+                answer_doc = super().llm_response(final_prompt)
+            else:
+                # respond with temporary context
+                answer_doc = super()._llm_response_temp_context(question, final_prompt)
         else:
             answer_doc = super().llm_response_forget(final_prompt)
 

{langroid-0.52.5 → langroid-0.52.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langroid"
-version = "0.52.5"
+version = "0.52.6"
 authors = [
     {name = "Prasad Chalasani", email = "pchalasani@gmail.com"},
 ]