langroid 0.18.1__tar.gz → 0.18.2__tar.gz

{langroid-0.18.1 → langroid-0.18.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langroid
-Version: 0.18.1
+Version: 0.18.2
 Summary: Harness LLMs with Multi-Agent Programming
 License: MIT
 Author: Prasad Chalasani

{langroid-0.18.1 → langroid-0.18.2}/langroid/agent/base.py

@@ -91,6 +91,8 @@ class AgentConfig(BaseSettings):
     show_stats: bool = True  # show token usage/cost stats?
     add_to_registry: bool = True  # register agent in ObjectRegistry?
     respond_tools_only: bool = False  # respond only to tool messages (not plain text)?
+    # allow multiple tool messages in a single response?
+    allow_multiple_tools: bool = True
 
     @validator("name")
     def check_name_alphanum(cls, v: str) -> str:
@@ -1120,6 +1122,8 @@ class Agent(ABC):
             # as a response to the tool message even though the tool was not intended
             # for this agent.
             return None
+        if len(tools) > 1 and not self.config.allow_multiple_tools:
+            return self.to_ChatDocument("ERROR: Use ONE tool at a time!")
         if len(tools) == 0:
             fallback_result = self.handle_message_fallback(msg)
             if fallback_result is None:

{langroid-0.18.1 → langroid-0.18.2}/langroid/utils/system.py

@@ -10,7 +10,7 @@ import socket
 import traceback
 import uuid
 from pathlib import Path
-from typing import Any
+from typing import Any, Literal
 
 logger = logging.getLogger(__name__)
 
@@ -186,24 +186,48 @@ def generate_unique_id() -> str:
     return str(uuid.uuid4())
 
 
-def create_file(filepath: str | Path, content: str = "") -> None:
+def create_file(
+    filepath: str | Path,
+    content: str = "",
+    if_exists: Literal["overwrite", "skip", "error", "append"] = "overwrite",
+) -> None:
     """
-    Create a file with the given content in the specified directory.
+    Create, overwrite or append to a file, with the given content
+    at the specified filepath.
     If content is empty, it will simply touch to create an empty file.
 
     Args:
         filepath (str|Path): The relative path of the file to be created
         content (str): The content to be written to the file
+        if_exists (Literal["overwrite", "skip", "error", "append"]):
+            Action to take if file exists
     """
-    Path(filepath).parent.mkdir(parents=True, exist_ok=True)
-    if content == "":
-        Path(filepath).touch()
+    filepath = Path(filepath)
+    filepath.parent.mkdir(parents=True, exist_ok=True)
+
+    if filepath.exists():
+        if if_exists == "skip":
+            logger.warning(f"File already exists, skipping: {filepath}")
+            return
+        elif if_exists == "error":
+            raise FileExistsError(f"File already exists: {filepath}")
+        elif if_exists == "append":
+            mode = "a"
+        else:  # overwrite
+            mode = "w"
+    else:
+        mode = "w"
+
+    if content == "" and mode in ["a", "w"]:
+        filepath.touch()
+        logger.warning(f"Empty file created: {filepath}")
     else:
         # the newline = '\n` argument is used to ensure that
         # newlines in the content are written as actual line breaks
-        with open(filepath, "w", newline="\n") as f:
+        with open(filepath, mode, newline="\n") as f:
             f.write(content)
-    logger.warning(f"File created/updated: {filepath}")
+        action = "appended to" if mode == "a" else "created/updated in"
+        logger.warning(f"Content {action}: {filepath}")
 
 
 def read_file(path: str, line_numbers: bool = False) -> str:

{langroid-0.18.1 → langroid-0.18.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langroid"
-version = "0.18.1"
+version = "0.18.2"
 description = "Harness LLMs with Multi-Agent Programming"
 authors = ["Prasad Chalasani <pchalasani@gmail.com>"]
 readme = "README.md"