langroid 0.15.0__tar.gz → 0.15.2__tar.gz

{langroid-0.15.0 → langroid-0.15.2}/PKG-INFO

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

{langroid-0.15.0 → langroid-0.15.2}/langroid/agent/special/neo4j/neo4j_chat_agent.py

@@ -91,6 +91,11 @@ class Neo4jChatAgent(ChatAgent):
         self._import_neo4j()
         self._initialize_connection()
         self._init_tool_messages()
+        self.init_state()
+
+    def init_state(self) -> None:
+        super().init_state()
+        self.current_retrieval_cypher_query: str = ""
 
     def handle_message_fallback(
         self, msg: str | ChatDocument
@@ -274,7 +279,7 @@ class Neo4jChatAgent(ChatAgent):
             the database.
             """
         query = msg.cypher_query
-
+        self.current_retrieval_cypher_query = query
         logger.info(f"Executing Cypher query: {query}")
         response = self.read_query(query)
         if isinstance(response.data, list) and len(response.data) == 0:

{langroid-0.15.0 → langroid-0.15.2}/pyproject.toml

@@ -1,12 +1,11 @@
 [tool.poetry]
 name = "langroid"
-version = "0.15.0"
+version = "0.15.2"
 description = "Harness LLMs with Multi-Agent Programming"
 authors = ["Prasad Chalasani <pchalasani@gmail.com>"]
 readme = "README.md"
 license = "MIT"
-include = ["pyproject.toml"]
-
+include = ["pyproject.toml", "langroid/py.typed"]
 
 # =============== MAIN DEPS ==============
 [tool.poetry.dependencies]