langroid 0.6.5__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:
@@ -285,7 +286,11 @@ class Neo4jChatAgent(ChatAgent):
             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/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.5.dist-info → langroid-0.6.6.dist-info}/METADATA

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

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

@@ -23,7 +23,7 @@ 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=xlyIxG3-OQhp6l-dW01HxkehPKMzv2Laqp0DcBJRSrM,13439
+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=_2NKX2Sx_7nLGNk_0rUyDkgTHZJuKm_DgtG7M_BFPkY,2920
 langroid/agent/special/relevance_extractor_agent.py,sha256=zIx8GUdVo1aGW6ASla0NPQjYYIpmriK_TYMijqAx3F8,4796
@@ -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=DzLYdTaZkEq7skUJyKoSIG57xLpZlLmL7ws2KwYPBe4,7063
-langroid-0.6.5.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.6.5.dist-info/METADATA,sha256=rSOgUJQHwzIdMef_inx5Up7kT9BUUrbroPVr-fg_9zo,54402
-langroid-0.6.5.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-langroid-0.6.5.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.5"
+version = "0.6.6"
 description = "Harness LLMs with Multi-Agent Programming"
 authors = ["Prasad Chalasani <pchalasani@gmail.com>"]
 readme = "README.md"