Merge pull request #51 from b0g3r/master

Add support for conversations/models APIs

Author: kishorenc

README.md

@@ -20,7 +20,7 @@ See detailed [API documentation](https://typesense.org/api).
 
 | Typesense Server | typesense-python |
 |------------------|------------------|
-| \>= v0.26.0      | \>= v0.20.0      |
+| \>= v26.0        | \>= v0.20.0      |
 | \>= v0.25.0      | \>= v0.16.0      |
 | \>= v0.23.0      | \>= v0.14.0      |
 | \>= v0.21.0      | \>= v0.13.0      |

src/typesense/__init__.py

@@ -1,5 +1,5 @@
 from .client import Client  # NOQA
 
 
-__version__ = '0.20.0'
+__version__ = '0.21.0'
 

src/typesense/client.py

@@ -1,4 +1,5 @@
 from .aliases import Aliases
+from .conversations_models import ConversationsModels
 from .debug import Debug
 from .collections import Collections
 from .multi_search import MultiSearch
@@ -21,3 +22,4 @@ def __init__(self, config_dict):
         self.operations = Operations(self.api_call)
         self.debug = Debug(self.api_call)
         self.stopwords = Stopwords(self.api_call)
+        self.conversations_models = ConversationsModels(self.api_call)

src/typesense/conversation_model.py

@@ -0,0 +1,17 @@
+class ConversationModel(object):
+    def __init__(self, api_call, model_id):
+        self.model_id = model_id
+        self.api_call = api_call
+
+    def _endpoint_path(self):
+        from .conversations_models import ConversationsModels
+        return u"{0}/{1}".format(ConversationsModels.RESOURCE_PATH, self.model_id)
+
+    def retrieve(self):
+        return self.api_call.get(self._endpoint_path())
+
+    def update(self, model):
+        return self.api_call.put(self._endpoint_path(), model)
+
+    def delete(self):
+        return self.api_call.delete(self._endpoint_path())

src/typesense/conversations_models.py

@@ -0,0 +1,21 @@
+from .conversation_model import ConversationModel
+
+
+class ConversationsModels(object):
+    RESOURCE_PATH = '/conversations/models'
+
+    def __init__(self, api_call):
+        self.api_call = api_call
+        self.conversations_models = {}
+
+    def __getitem__(self, model_id):
+        if model_id not in self.conversations_models:
+            self.conversations_models[model_id] = ConversationModel(self.api_call, model_id)
+
+        return self.conversations_models.get(model_id)
+
+    def create(self, model):
+        return self.api_call.post(ConversationsModels.RESOURCE_PATH, model)
+
+    def retrieve(self):
+        return self.api_call.get(ConversationsModels.RESOURCE_PATH)