memorydb-langchain 0.2.1__py3-none-any.whl

memorydb_langchain/__init__.py

@@ -0,0 +1,4 @@
+"""LangChain integration for MemoryDB."""
+from memorydb_langchain.chat_history import MemoryDBChatMessageHistory
+
+__all__ = ["MemoryDBChatMessageHistory"]

memorydb_langchain/chat_history.py

@@ -0,0 +1,69 @@
+"""LangChain integration for MemoryDB: chat message history backed by the memory engine.
+
+Each session lives in one `chat` record; LangChain-style appends become new immutable
+revisions in the WAL, so `db.messages.get(pk).history()` is a free audit trail.
+
+Install: pip install ./sdk && pip install langchain-core
+"""
+import json
+
+from langchain_core.chat_history import BaseChatMessageHistory
+from langchain_core.messages import BaseMessage, messages_from_dict, message_to_dict
+
+from memorydb import MemoryDB, MemoryDBError
+
+
+class MemoryDBChatMessageHistory(BaseChatMessageHistory):
+    """Conversational memory stored as a versioned MemoryDB chat record."""
+
+    def __init__(self, session_id: str, *, client: MemoryDB | None = None,
+                 url: str = "http://127.0.0.1:8080", token: str | None = None,
+                 source: str = "langchain", max_retries: int = 3):
+        self.client = client or MemoryDB(url, token=token)
+        self.session_id = session_id
+        self.source = source
+        self.max_retries = max_retries
+
+    def _find(self):
+        listed = self.client.request("GET", f"memories/?kind=chat&source={self.source}&limit=500")
+        for hit in listed["results"]:
+            if hit["record"].get("metadata", {}).get("session_id") == self.session_id:
+                return hit["record"]
+        return None
+
+    @property
+    def messages(self):
+        record = self._find()
+        if record is None:
+            return []
+        return messages_from_dict(json.loads(record["content"]))
+
+    def add_messages(self, messages):
+        payload_new = [message_to_dict(m) for m in messages]
+        for attempt in range(self.max_retries):
+            record = self._find()
+            try:
+                if record is None:
+                    # Full-record PUT carries the version we read; a concurrent writer bumps it first,
+                    # so a conflict means re-read and retry rather than lose the other append.
+                    self.client.request("POST", "memories/", {
+                        "kind": "chat", "content": json.dumps(payload_new),
+                        "source": self.source, "layer": "raw",
+                        "metadata": {"session_id": self.session_id, "method": "langchain chat history"}})
+                else:
+                    updated = dict(record)
+                    updated["content"] = json.dumps(json.loads(record["content"]) + payload_new)
+                    self.client.request("PUT", "memories/" + record["id"] + "/", updated)
+                return
+            except MemoryDBError:
+                if attempt == self.max_retries - 1:
+                    raise
+
+    def add_message(self, message: BaseMessage):
+        self.add_messages([message])
+
+    def clear(self):
+        record = self._find()
+        if record is not None:
+            self.client.request("DELETE", "memories/" + record["id"] + "/",
+                                headers={"If-Match": str(record["version"])})

memorydb_langchain-0.2.1.dist-info/METADATA

@@ -0,0 +1,54 @@
+Metadata-Version: 2.4
+Name: memorydb-langchain
+Version: 0.2.1
+Summary: LangChain chat-message history backed by the MemoryDB agent-memory engine
+Author: humaniods
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/humaniods/MemoryDB
+Project-URL: Repository, https://github.com/humaniods/MemoryDB
+Project-URL: Issues, https://github.com/humaniods/MemoryDB/issues
+Keywords: langchain,agent-memory,chat-history,llm,database
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: POSIX
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: superb-memory
+Requires-Dist: langchain-core>=0.2
+
+# memorydb-langchain
+
+LangChain chat message history backed by
+[MemoryDB](https://github.com/humaniods/MemoryDB), so agent conversations survive process
+restarts and stay searchable alongside the facts you ingest from them.
+
+```bash
+pip install memorydb-langchain
+export MEMORYDB_URL=memorydb://TOKEN@127.0.0.1:8080
+```
+
+```python
+from memorydb import MemoryDB
+from memorydb_langchain import MemoryDBChatMessageHistory
+from langchain_core.messages import HumanMessage, AIMessage
+
+history = MemoryDBChatMessageHistory(session_id="user-42", client=MemoryDB())
+history.add_message(HumanMessage("What is the capital of Rajasthan?"))
+history.add_message(AIMessage("Jaipur."))
+history.messages          # [HumanMessage, AIMessage]
+history.clear()
+```
+
+Every append writes a new immutable revision of one record rather than rewriting a table, so the
+server keeps the full trail:
+
+```python
+db = MemoryDB()
+db.objects.history(pk=history.record_id)["results"]   # versions 1..n, a free audit log
+```
+
+Use one principal (token) per deployed product so sessions of different products land in
+different workspaces, and give each end-user their own `session_id`. Search the whole history
+with `db.objects.search("Jaipur")`, including hybrid BM25 + vector ranking.

memorydb_langchain-0.2.1.dist-info/RECORD

@@ -0,0 +1,6 @@
+memorydb_langchain/__init__.py,sha256=5dZLH8sPHnFE4X4ujGx_mK5q4WoPjmA8lMZ4ZutkrAs,155
+memorydb_langchain/chat_history.py,sha256=gnA3YjUL1z_6Ba80IBpL6ns_TxL4METtWWcbK0cEJr4,2984
+memorydb_langchain-0.2.1.dist-info/METADATA,sha256=g__gulqh4Hb-LD3BKKYzs9lmC-NgDsc1uSizZIYDDVk,2060
+memorydb_langchain-0.2.1.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
+memorydb_langchain-0.2.1.dist-info/top_level.txt,sha256=heoqqw4PcZpOa9tN7HsXrn0HZsMPbQW71bYbm8IFlOs,19
+memorydb_langchain-0.2.1.dist-info/RECORD,,

memorydb_langchain-0.2.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (84.2.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

memorydb_langchain-0.2.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+memorydb_langchain