langroid 0.9.3__tar.gz → 0.9.5__tar.gz

{langroid-0.9.3 → langroid-0.9.5}/PKG-INFO

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

{langroid-0.9.3 → langroid-0.9.5}/langroid/agent/base.py

@@ -45,7 +45,13 @@ from langroid.mytypes import Entity
 from langroid.parsing.parse_json import extract_top_level_json
 from langroid.parsing.parser import Parser, ParsingConfig
 from langroid.prompts.prompts_config import PromptsConfig
-from langroid.pydantic_v1 import BaseSettings, Field, ValidationError, validator
+from langroid.pydantic_v1 import (
+    BaseModel,
+    BaseSettings,
+    Field,
+    ValidationError,
+    validator,
+)
 from langroid.utils.configuration import settings
 from langroid.utils.constants import DONE, NO_ANSWER, PASS, PASS_TO, SEND_TO
 from langroid.utils.object_registry import ObjectRegistry
@@ -1300,8 +1306,25 @@ class Agent(ABC):
                     # else wrap it in an agent response and return it so
                     # orchestrator can find a respondent
                     result = self.create_agent_response(tool_messages=[maybe_result])
-            else:
+            elif isinstance(maybe_result, (ChatDocument, str)):
                 result = maybe_result
+            elif isinstance(maybe_result, BaseModel):
+                result = maybe_result.json()
+            else:
+                # last resort: use json.dumps() or str() to make it a str
+                try:
+                    result = json.dumps(maybe_result)
+                except Exception:
+                    try:
+                        result = str(maybe_result)
+                    except Exception as e:
+                        logger.error(
+                            f"""
+                            Error converting result of {tool_name} to str: {e}", 
+                            """,
+                            exc_info=True,
+                        )
+                        result = None
         except Exception as e:
             # raise the error here since we are sure it's
             # not a pydantic validation error,

{langroid-0.9.3 → langroid-0.9.5}/langroid/language_models/prompt_formatter/hf_formatter.py

@@ -18,7 +18,7 @@ from langroid.language_models.prompt_formatter.base import PromptFormatter
 logger = logging.getLogger(__name__)
 
 
-def try_import_hf_modules() -> Tuple[Type[Any], Type[Any], Type[Any]]:
+def try_import_hf_modules() -> Tuple[Type[Any], Type[Any]]:
     """
     Attempts to import the AutoTokenizer class from the transformers package.
     Returns:
@@ -27,17 +27,16 @@ def try_import_hf_modules() -> Tuple[Type[Any], Type[Any], Type[Any]]:
         ImportError: If the transformers package is not installed.
     """
     try:
-        from huggingface_hub import HfApi, ModelFilter
+        from huggingface_hub import HfApi
         from transformers import AutoTokenizer
 
-        return AutoTokenizer, HfApi, ModelFilter
+        return AutoTokenizer, HfApi
     except ImportError:
         raise ImportError(
             """
             You are trying to use some/all of:
             HuggingFace transformers.AutoTokenizer,
-            huggingface_hub.HfApi, 
-            huggingface_hub.ModelFilter,
+            huggingface_hub.HfApi,
             but these are not not installed 
             by default with Langroid. Please install langroid using the 
             `transformers` extra, like so:
@@ -48,7 +47,7 @@ def try_import_hf_modules() -> Tuple[Type[Any], Type[Any], Type[Any]]:
 
 
 def find_hf_formatter(model_name: str) -> str:
-    AutoTokenizer, HfApi, ModelFilter = try_import_hf_modules()
+    AutoTokenizer, HfApi = try_import_hf_modules()
     hf_api = HfApi()
     # try to find a matching model, with progressivly shorter prefixes of model_name
     model_name = model_name.lower().split("/")[-1]
@@ -57,19 +56,19 @@ def find_hf_formatter(model_name: str) -> str:
     for i in range(len(parts), 0, -1):
         prefix = "-".join(parts[:i])
         models = hf_api.list_models(
-            filter=ModelFilter(
-                task="text-generation",
-                model_name=prefix,
-            )
+            task="text-generation",
+            model_name=prefix,
         )
         try:
             mdl = next(models)
-        except StopIteration:
+            tokenizer = AutoTokenizer.from_pretrained(mdl.id)
+            if tokenizer.chat_template is not None:
+                return str(mdl.id)
+            else:
+                continue
+        except Exception:
             continue
 
-        tokenizer = AutoTokenizer.from_pretrained(mdl.id)
-        if tokenizer.chat_template is not None:
-            return str(mdl.id)
     return ""
 
 
@@ -78,14 +77,12 @@ class HFFormatter(PromptFormatter):
 
     def __init__(self, config: HFPromptFormatterConfig):
         super().__init__(config)
-        AutoTokenizer, HfApi, ModelFilter = try_import_hf_modules()
+        AutoTokenizer, HfApi = try_import_hf_modules()
         self.config: HFPromptFormatterConfig = config
         hf_api = HfApi()
         models = hf_api.list_models(
-            filter=ModelFilter(
-                task="text-generation",
-                model_name=config.model_name,
-            )
+            task="text-generation",
+            model_name=config.model_name,
         )
         try:
             mdl = next(models)

{langroid-0.9.3 → langroid-0.9.5}/pyproject.toml

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