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

langroid/agent/special/neo4j/neo4j_chat_agent.py

@@ -265,10 +265,23 @@ class Neo4jChatAgent(ChatAgent):
         Returns:
             str: The result of executing the cypher_query.
         """
+        if not self.tried_schema:
+            return """
+            You did not yet use the `get_schema` tool to get the schema 
+            of the neo4j knowledge-graph db. Use that tool first before using 
+            the `retrieval_query` tool, to ensure you know all the correct
+            node labels, relationship types, and property keys available in
+            the database.
+            """
         query = msg.cypher_query
 
         logger.info(f"Executing Cypher query: {query}")
         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.
+            """
         if response.success:
             return json.dumps(response.data)
         else:
@@ -316,6 +329,7 @@ class Neo4jChatAgent(ChatAgent):
             behavior of 'self.read_query' method, which might raise exceptions related
              to database connectivity or query execution.
         """
+        self.tried_schema = True
         schema_result = self.read_query("CALL db.schema.visualization()")
         if schema_result.success:
             # ther is a possibility that the schema is empty, which is a valid response
@@ -326,6 +340,7 @@ class Neo4jChatAgent(ChatAgent):
 
     def _init_tool_messages(self) -> None:
         """Initialize message tools used for chatting."""
+        self.tried_schema = False
         message = self._format_message()
         self.config.system_message = self.config.system_message.format(mode=message)
         if self.config.addressing_prefix != "":

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

@@ -1,24 +1,29 @@
 DEFAULT_SYS_MSG = """You are a data scientist and expert in Knowledge Graphs, 
 with expertise in answering questions by interacting with a Neo4j graph database.
 
-The schema maps the Neo4j database structure. node labels, relationship types,
-and property keys available in your Neo4j database. 
+The schema below describes the Neo4j database structure, node labels, 
+relationship types, and property keys available in your Neo4j database. 
 {schema}
 Do not make assumptions about the database schema before using the tools.
-Use the tool/function to learn more about the database schema."""
+Use the tools/functions to learn more about the database schema."""
 
 SCHEMA_TOOLS_SYS_MSG = """You are a data scientist and expert in Knowledge Graphs, 
 with expertise in answering questions by querying Neo4j database.
-You have access to the following tools: 
- - `retrieval_query` tool/function-call to retreive infomration from the graph database
+You have access to the following tools:
+
+- `get_schema` tool/function-call to get all the node labels, relationship 
+ types, and property keys available in your Neo4j database. You MUST use
+ this tool BEFORE attempting to use the `retrieval_query` tool/function-call,
+ to ensure that you are using the correct node labels, relationship types, and
+ property keys in your `retrieval_query` tool/function-call.
+ 
+ - `retrieval_query` tool/function-call to retrieve infomration from the graph database
 to answer questions.
 
  - `create_query` tool/function-call to execute cypher query that creates
    entities/relationships in the graph database.
 
- - `get_schema` tool/function-call to get all the node labels, relationship 
- types, and property keys available in your Neo4j database.
-
+ 
 You must be smart about using the right node labels, relationship types, and property
 keys based on the english description. If you are thinking of using a node label,
 relationship type, or property key that does not exist, you are probably on the wrong 
@@ -30,7 +35,9 @@ DEFAULT_NEO4J_CHAT_SYSTEM_MESSAGE = """
 
 You do not need to attempt answering a question with just one query. 
 You could make a sequence of Neo4j queries to help you write the final query.
-Also if you receive a null or other unexpected result,
+
+RETRY-SUGGESTIONS:
+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. 

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

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

{langroid-0.6.3.dist-info → langroid-0.6.4.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=B5y4he__9QjNrjNyngv4rlfVx7QezQ4Turom-l_zjCg,12572
+langroid/agent/special/neo4j/neo4j_chat_agent.py,sha256=ic09EvAxSWo6q-h-HpPKwFMzaeuD8iGbbJYDvHnRonA,13301
 langroid/agent/special/neo4j/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langroid/agent/special/neo4j/utils/system_message.py,sha256=TxxA5RIYaTUIv6p2xubqUBuVgBLCfZNrHGOdX--OLVg,2625
+langroid/agent/special/neo4j/utils/system_message.py,sha256=XNEP1jWQNYqjxMAVJpBkpfHHK0x537rj882sgV111vE,2887
 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
@@ -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=A63IQrki6kjMdP3gnIaZJmd500tm1rYcajBojMoC3xE,7063
-langroid-0.6.3.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.6.3.dist-info/METADATA,sha256=nQGYokvPMKhMUENLf7qw4IpW7tCgu4RvssqQQ20m-so,54402
-langroid-0.6.3.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-langroid-0.6.3.dist-info/RECORD,,
+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

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