From c5c5230520cf763d0f3d024c7774a5699457ebda Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabrielf.almeida90@gmail.com>
Date: Sat, 11 Feb 2023 06:11:12 +0000
Subject: [PATCH] feat: adding dictable interfaces JSON interface for Langchain
 #1

---
 src/interface.py | 45 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 src/interface.py

diff --git a/src/interface.py b/src/interface.py
new file mode 100644
index 000000000..e1afefe8e
--- /dev/null
+++ b/src/interface.py
@@ -0,0 +1,45 @@
+from langchain.chains import ConversationChain
+from langchain.chains.conversation.memory import ConversationBufferMemory
+from langchain.agents import Agent, ConversationalAgent, Tool, initialize_agent
+
+
+class Dictable(object):
+    """A mixin that allows an object to be converted to and from a dict"""
+
+    @classmethod
+    def from_dict(cls, d):
+        """Convert a dict to an object"""
+        return cls(**d)
+
+    def to_dict(self):
+        return self.__dict__
+
+
+class DictableChain(Dictable, ConversationChain):
+    """A ConversationChain that is also Dictable"""
+
+    pass
+
+
+class DictableMemory(Dictable, ConversationBufferMemory):
+    """A ConversationBufferMemory that is also Dictable"""
+
+    pass
+
+
+class DictableAgent(Dictable, Agent):
+    """An Agent that is also Dictable"""
+
+    pass
+
+
+class DictableConversationalAgent(Dictable, ConversationalAgent):
+    """A ConversationalAgent that is also Dictable"""
+
+    pass
+
+
+class DictableTool(Dictable, Tool):
+    """A Tool that is also Dictable"""
+
+    pass