langroid 0.6.4__py3-none-any.whl → 0.6.6__py3-none-any.whl

langroid/agent/special/neo4j/neo4j_chat_agent.py

@@ -6,6 +6,7 @@ from rich import print
 from rich.console import Console
 
 from langroid.agent import ToolMessage
+from langroid.parsing.parse_json import datetime_to_json
 from langroid.pydantic_v1 import BaseModel, BaseSettings
 
 if TYPE_CHECKING:
@@ -279,11 +280,17 @@ class Neo4jChatAgent(ChatAgent):
         response = self.read_query(query)
         if isinstance(response.data, list) and len(response.data) == 0:
             return """
-            No results found; perhaps try an approximate or case-insensitive match,
-            or the other RETRY-SUGGESTIONS in your instructions.
+            No results found; check if your query used the right label names -- 
+            remember these are case sensitive, so you have to use the exact label
+            names you found in the schema. 
+            Or retry using one of the  RETRY-SUGGESTIONS in your instructions. 
             """
         if response.success:
-            return json.dumps(response.data)
+            try:
+                json_str = json.dumps(response.data, default=datetime_to_json)
+                return json_str
+            except TypeError:
+                return str(response.data)
         else:
             return str(response.data)
 

langroid/agent/special/neo4j/utils/system_message.py

@@ -41,7 +41,8 @@ If you receive a null or other unexpected result,
 (a) make sure you use the available TOOLs correctly,
 (b) USE `get_schema` tool/function-call to get all the node labels, relationship 
  types, and property keys available in your Neo4j database. 
-(c) LABELS may be CASE-SENSITIVE, so try using CASE-INSENSITIVE MATCHES,
+(c) LABELS are CASE-SENSITIVE -- make sure you adhere to the exact label name
+   you found in the schema.
 (d) see if you have made an assumption in your Neo4j query, and try another way, 
    or use `retrieval_query` to explore the database contents before submitting your 
    final query. 

langroid/parsing/parse_json.py

@@ -1,5 +1,6 @@
 import ast
 import json
+from datetime import datetime
 from typing import Any, Dict, Iterator, List, Union
 
 import yaml
@@ -167,3 +168,10 @@ def top_level_json_field(s: str, f: str) -> Any:
             return json_data[f]
 
     return ""
+
+
+def datetime_to_json(obj: Any) -> Any:
+    if isinstance(obj, datetime):
+        return obj.isoformat()
+    # Let json.dumps() handle the raising of TypeError for non-serializable objects
+    return obj

{langroid-0.6.4.dist-info → langroid-0.6.6.dist-info}/METADATA

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

{langroid-0.6.4.dist-info → langroid-0.6.6.dist-info}/RECORD

@@ -23,9 +23,9 @@ langroid/agent/special/lance_rag_new/query_planner_agent.py,sha256=JqO_5fKW8HPn-
 langroid/agent/special/lance_tools.py,sha256=BznV_r3LAFyybvBRa9KQ0oU7mPM3uQVfri7PFp7M_qc,1894
 langroid/agent/special/neo4j/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/agent/special/neo4j/csv_kg_chat.py,sha256=dRsAgMBa1H_EMI2YYgJR2Xyv1D7e4o3G9M64mTewq_c,6409
-langroid/agent/special/neo4j/neo4j_chat_agent.py,sha256=ic09EvAxSWo6q-h-HpPKwFMzaeuD8iGbbJYDvHnRonA,13301
+langroid/agent/special/neo4j/neo4j_chat_agent.py,sha256=W-nXoqfOwZTl6p7ZLRQw017OARWKUzz_LSwFTjqpE04,13651
 langroid/agent/special/neo4j/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langroid/agent/special/neo4j/utils/system_message.py,sha256=XNEP1jWQNYqjxMAVJpBkpfHHK0x537rj882sgV111vE,2887
+langroid/agent/special/neo4j/utils/system_message.py,sha256=_2NKX2Sx_7nLGNk_0rUyDkgTHZJuKm_DgtG7M_BFPkY,2920
 langroid/agent/special/relevance_extractor_agent.py,sha256=zIx8GUdVo1aGW6ASla0NPQjYYIpmriK_TYMijqAx3F8,4796
 langroid/agent/special/retriever_agent.py,sha256=lvMvf-u9rSosg4YASuFdUbGLgkzLPknXAbJZfZ1LZCc,1868
 langroid/agent/special/sql/__init__.py,sha256=mWfmm1QpXCezpFOS2eI57M0L_Ok3q5_ukG8tXBnBrEA,319
@@ -87,7 +87,7 @@ langroid/parsing/config.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/parsing/document_parser.py,sha256=WGnA5ADwMHliGJt6WW9rc4RiFXQcKU33b5zdPiGrtEY,24265
 langroid/parsing/image_text.py,sha256=sbLIQ5nHe2UnYUksBaQsmZGaX-X0qgEpPd7CEzi_z5M,910
 langroid/parsing/para_sentence_split.py,sha256=AJBzZojP3zpB-_IMiiHismhqcvkrVBQ3ZINoQyx_bE4,2000
-langroid/parsing/parse_json.py,sha256=Rcy0iCyMMICk2QstrlJtW7zDPs9lOZ3N_mlaJGWkC40,5033
+langroid/parsing/parse_json.py,sha256=Ivh6YcEOESAi9y8E55E7qhFqfncvnluyb6sZGWK0--U,5268
 langroid/parsing/parser.py,sha256=AgtmlVUvrkSG1l7-YZPX8rlldgXjh_HqXAMqpXkBxUo,11746
 langroid/parsing/repo_loader.py,sha256=3GjvPJS6Vf5L6gV2zOU8s-Tf1oq_fZm-IB_RL_7CTsY,29373
 langroid/parsing/routing.py,sha256=-FcnlqldzL4ZoxuDwXjQPNHgBe9F9-F4R6q7b_z9CvI,1232
@@ -134,8 +134,8 @@ langroid/vector_store/meilisearch.py,sha256=6frB7GFWeWmeKzRfLZIvzRjllniZ1cYj3Hmh
 langroid/vector_store/momento.py,sha256=qR-zBF1RKVHQZPZQYW_7g-XpTwr46p8HJuYPCkfJbM4,10534
 langroid/vector_store/qdrant_cloud.py,sha256=3im4Mip0QXLkR6wiqVsjV1QvhSElfxdFSuDKddBDQ-4,188
 langroid/vector_store/qdrantdb.py,sha256=v88lqFkepADvlN6lByUj9I4NEKa9X9lWH16uTPPbYrE,17457
-pyproject.toml,sha256=MUbx-I1YZJEr0r177JeAZTH0YRO_TyN5UQIDWsiNCeI,7063
-langroid-0.6.4.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.6.4.dist-info/METADATA,sha256=h2pVxowaNGKIl57xLgzNdahRrcNly8W2koXNk1Mg2pM,54402
-langroid-0.6.4.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-langroid-0.6.4.dist-info/RECORD,,
+pyproject.toml,sha256=s7qSfD_4EuQqtVpxQ-3fAGwEiJjb6b3tRtXh4ZQgKNU,7063
+langroid-0.6.6.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.6.6.dist-info/METADATA,sha256=-xsYzS3TfrdtgozB_TNT5IZFV_RlG3z1eLFItYVAwRg,54402
+langroid-0.6.6.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+langroid-0.6.6.dist-info/RECORD,,

pyproject.toml

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