langroid 0.6.4__tar.gz → 0.6.6__tar.gz

{langroid-0.6.4 → langroid-0.6.6}/PKG-INFO

@@ -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 → langroid-0.6.6}/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-0.6.4 → langroid-0.6.6}/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-0.6.4 → langroid-0.6.6}/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 → langroid-0.6.6}/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"