langroid 0.33.7__tar.gz → 0.33.9__tar.gz

{langroid-0.33.7 → langroid-0.33.9}/PKG-INFO

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

{langroid-0.33.7 → langroid-0.33.9}/langroid/agent/special/sql/sql_chat_agent.py

@@ -72,7 +72,8 @@ Start by asking what I would like to know about the data.
 
 ADDRESSING_INSTRUCTION = """
 IMPORTANT - Whenever you are NOT writing a SQL query, make sure you address the user
-using {prefix}User. You MUST use the EXACT syntax {prefix} !!!
+using {prefix}User (NO SPACE between {prefix} and User). 
+You MUST use the EXACT syntax {prefix}User !!!
 
 In other words, you ALWAYS write EITHER:
  - a SQL query using the `run_query` tool, 

{langroid-0.33.7 → langroid-0.33.9}/langroid/agent/task.py

@@ -200,7 +200,11 @@ class Task:
                 control of task termination.
             system_message (str): if not empty, overrides agent's system_message
             user_message (str): if not empty, overrides agent's user_message
-            restart (bool): if true, resets the agent's message history *at every run*.
+            restart (bool): if true (default), resets the agent's message history
+                *at every run* when it is the top-level task. Ignored when
+                the task is a subtask of another task. Restart behavior of a subtask's
+                `run()` can be controlled via the `TaskConfig.restart_as_subtask`
+                setting.
             default_human_response (str|None): default response from user; useful for
                 testing, to avoid interactive input from user.
                 [Instead of this, setting `interactive` usually suffices]

{langroid-0.33.7 → langroid-0.33.9}/langroid/parsing/url_loader.py

@@ -52,10 +52,14 @@ class URLLoader:
                     or url.lower().endswith(".docx")
                     or url.lower().endswith(".doc")
                 ):
-                    doc_parser = DocumentParser.create(
-                        url,
-                        self.parser.config,
-                    )
+                    try:
+                        doc_parser = DocumentParser.create(
+                            url,
+                            self.parser.config,
+                        )
+                    except Exception as e:
+                        logging.error(f"Error parsing {url}: {e}")
+                        continue
                     new_chunks = doc_parser.get_doc_chunks()
                     if len(new_chunks) == 0:
                         # If the document is empty, try to extract images

{langroid-0.33.7 → langroid-0.33.9}/langroid/vector_store/qdrantdb.py

@@ -2,6 +2,7 @@ import hashlib
 import json
 import logging
 import os
+import time
 import uuid
 from typing import Dict, List, Optional, Sequence, Tuple, TypeVar
 
@@ -323,6 +324,26 @@ class QdrantDB(VectorStore):
             }
             if self.config.use_sparse_embeddings:
                 vectors["text-sparse"] = sparse_embedding_vecs[i : i + b]
+            coll_found: bool = False
+            for _ in range(3):
+                # poll until collection is ready
+                if (
+                    self.client.collection_exists(self.config.collection_name)
+                    and self.client.get_collection(self.config.collection_name).status
+                    == CollectionStatus.GREEN
+                ):
+                    coll_found = True
+                    break
+                time.sleep(1)
+
+            if not coll_found:
+                raise ValueError(
+                    f"""
+                    QdrantDB Collection {self.config.collection_name} 
+                    not found or not ready
+                    """
+                )
+
             self.client.upsert(
                 collection_name=self.config.collection_name,
                 points=Batch(

{langroid-0.33.7 → langroid-0.33.9}/pyproject.toml

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