langroid 0.1.185__tar.gz → 0.1.186__tar.gz

{langroid-0.1.185 → langroid-0.1.186}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langroid
-Version: 0.1.185
+Version: 0.1.186
 Summary: Harness LLMs with Multi-Agent Programming
 License: MIT
 Author: Prasad Chalasani
@@ -37,7 +37,7 @@ Requires-Dist: google-api-python-client (>=2.95.0,<3.0.0)
 Requires-Dist: halo (>=0.0.31,<0.0.32)
 Requires-Dist: jinja2 (>=3.1.2,<4.0.0)
 Requires-Dist: lancedb (>=0.4.1,<0.5.0)
-Requires-Dist: litellm (>=1.22.3,<2.0.0) ; extra == "litellm"
+Requires-Dist: litellm (>=1.22.5,<2.0.0) ; extra == "litellm"
 Requires-Dist: lxml (>=4.9.3,<5.0.0)
 Requires-Dist: meilisearch (>=0.28.3,<0.29.0)
 Requires-Dist: meilisearch-python-sdk (>=2.2.3,<3.0.0)
@@ -75,6 +75,7 @@ Requires-Dist: pytest-postgresql (>=5.0.0,<6.0.0) ; extra == "postgres"
 Requires-Dist: pytest-redis (>=3.0.2,<4.0.0)
 Requires-Dist: python-docx (>=1.1.0,<2.0.0)
 Requires-Dist: python-dotenv (>=1.0.0,<2.0.0)
+Requires-Dist: python-socketio (>=5.11.0,<6.0.0) ; extra == "chainlit"
 Requires-Dist: qdrant-client (>=1.7.0,<2.0.0)
 Requires-Dist: rank-bm25 (>=0.2.2,<0.3.0)
 Requires-Dist: redis (>=5.0.1,<6.0.0)

{langroid-0.1.185 → langroid-0.1.186}/langroid/agent/base.py

@@ -105,6 +105,7 @@ class Agent(ABC):
             get_user_response=None,
             get_last_step=noop_fn,
             set_parent_agent=noop_fn,
+            show_error_message=noop_fn,
         )
 
     def entity_responders(

{langroid-0.1.185 → langroid-0.1.186}/langroid/agent/callbacks/chainlit.py

@@ -244,20 +244,6 @@ async def ask_user_step(
     return res["output"]
 
 
-async def show_first_user_message(message: cl.Message, agent_name: str):
-    """Display first user message, which is not yet part of an Agent step,
-    using the Agent's name, so it is displayed with a title
-    consistent with subsequent user responses within the Agent's steps."""
-
-    step = cl.Step(
-        name=f"{agent_name} (You 😃)",
-        type="run",
-        id=message.id,
-    )
-    step.output = message.content
-    await step.update()
-
-
 def wrap_text_preserving_structure(text: str, width: int = 90) -> str:
     """Wrap text preserving paragraph breaks. Typically used to
     format an agent_response output, which may have long lines
@@ -285,7 +271,10 @@ class ChainlitAgentCallbacks:
     stream: Optional[cl.Step] = None  # pushed into openai_gpt.py to stream tokens
     parent_agent: Optional[lr.Agent] = None  # used to get parent id, for step nesting
 
-    def __init__(self, agent: lr.Agent):
+    def __init__(self, agent: lr.Agent, msg: cl.Message = None):
+        """Add callbacks to the agent, and save the initial message,
+        so we can alter the display of the first user message.
+        """
         agent.callbacks.start_llm_stream = self.start_llm_stream
         agent.callbacks.cancel_llm_stream = self.cancel_llm_stream
         agent.callbacks.finish_llm_stream = self.finish_llm_stream
@@ -294,8 +283,11 @@ class ChainlitAgentCallbacks:
         agent.callbacks.get_user_response = self.get_user_response
         agent.callbacks.get_last_step = self.get_last_step
         agent.callbacks.set_parent_agent = self.set_parent_agent
+        agent.callbacks.show_error_message = self.show_error_message
         self.agent: lr.Agent = agent
         self.name = agent.config.name
+        if msg is not None:
+            self.show_first_user_message(msg)
 
     def _get_parent_id(self) -> str | None:
         """Get step id under which we need to nest the current step:
@@ -387,6 +379,18 @@ class ChainlitAgentCallbacks:
         step.output = content or NO_ANSWER
         run_sync(step.send())  # type: ignore
 
+    def show_error_message(self, error: str) -> None:
+        """Show error message as a step."""
+        step = cl.Step(
+            name=self.agent.config.name + "(Error 🚫)",
+            type="run",
+            parent_id=self._get_parent_id(),
+            language="text",
+        )
+        self.last_step = step
+        step.output = error
+        run_sync(step.send())
+
     def show_agent_response(self, content: str) -> None:
         """Show message from agent (typically tool handler).
         Agent response can be considered as a "step"
@@ -456,19 +460,33 @@ class ChainlitAgentCallbacks:
         step.output = message
         run_sync(step.send())
 
+    def show_first_user_message(self, msg: cl.Message):
+        """Show first user message as a step."""
+        step = cl.Step(
+            id=msg.id,
+            name=self.agent.config.name + "(You 😃)",
+            type="run",
+            parent_id=self._get_parent_id(),
+        )
+        self.last_step = step
+        step.output = msg.content
+        run_sync(step.update())
+
 
 class ChainlitTaskCallbacks:
     """
-    Inject ChainlitCallbacks into a Langroid Task's agent and
+    Recursively inject ChainlitAgentCallbacks into a Langroid Task's agent and
     agents of sub-tasks.
     """
 
-    def __init__(self, task: lr.Task):
-        ChainlitTaskCallbacks._inject_callbacks(task)
+    def __init__(self, task: lr.Task, msg: cl.Message = None):
+        """Inject callbacks recursively, ensuring msg is passed to the
+        top-level agent"""
+        ChainlitTaskCallbacks._inject_callbacks(task, msg)
 
     @staticmethod
-    def _inject_callbacks(task: lr.Task) -> None:
-        # recursively apply ChainlitCallbacks to agents of sub-tasks
-        ChainlitAgentCallbacks(task.agent)
+    def _inject_callbacks(task: lr.Task, msg: cl.Message = None) -> None:
+        # recursively apply ChainlitAgentCallbacks to agents of sub-tasks
+        ChainlitAgentCallbacks(task.agent, msg)
         for t in task.sub_tasks:
             ChainlitTaskCallbacks._inject_callbacks(t)

{langroid-0.1.185 → langroid-0.1.186}/langroid/language_models/utils.py

@@ -134,11 +134,3 @@ def async_retry_with_exponential_backoff(
                 raise e
 
     return wrapper
-
-
-# @retry_with_exponential_backoff
-# def completions_with_backoff(**kwargs):
-#     return openai.Completion.create(**kwargs)
-
-
-# completions_with_backoff(model="text-davinci-002", prompt="Once upon a time,")

{langroid-0.1.185 → langroid-0.1.186}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langroid"
-version = "0.1.185"
+version = "0.1.186"
 description = "Harness LLMs with Multi-Agent Programming"
 authors = ["Prasad Chalasani <pchalasani@gmail.com>"]
 readme = "README.md"
@@ -75,7 +75,7 @@ docstring-parser = "^0.15"
 farm-haystack = {extras = ["ocr", "preprocessing", "file-conversion", "pdf"], version = "^1.21.1"}
 meilisearch = "^0.28.3"
 meilisearch-python-sdk = "^2.2.3"
-litellm = {version = "^1.22.3", optional = true}
+litellm = {version = "^1.22.5", optional = true}
 scrapy = "^2.11.0"
 async-generator = "^1.10"
 lancedb = "^0.4.1"
@@ -86,6 +86,7 @@ python-docx = "^1.1.0"
 aiohttp = "^3.9.1"
 metaphor-python = {version = "^0.1.23", optional = true}
 chainlit = {version = "^1.0.200", optional = true}
+python-socketio = {version="^5.11.0", optional=true}
 
 [tool.poetry.extras]
 # install these using `poetry install -E [...]` where [...] is one of the extras below
@@ -97,7 +98,7 @@ litellm = ["litellm"]
 neo4j = ["neo4j"]
 sciphi = ["agent-search"]
 metaphor = ["metaphor-python"]
-chainlit = ["chainlit"]
+chainlit = ["chainlit", "python-socketio"]
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.3.1"