feat(document): add document post as sugar (#283)

docarray · Apr 19, 2022 · 446507d · 446507d
1 parent d822fe8
commit 446507d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/docarray/array/mixins/post.py b/docarray/array/mixins/post.py
@@ -70,7 +70,8 @@ def post(
         elif _scheme in ('grpc', 'http', 'ws', 'websocket'):
             from jina import Client
 
-            c = Client(host=r.hostname)
+            standardized_host = standardized_host + (f':{_port}' or '')
+            c = Client(host=standardized_host)
             return c.post(
                 _on,
                 inputs=self,

diff --git a/docarray/document/mixins/sugar.py b/docarray/document/mixins/sugar.py
@@ -84,3 +84,9 @@ def embed(self: 'T', *args, **kwargs) -> 'T':
         _tmp = DocumentArray(self)
         _tmp.embed(*args, **kwargs)
         return self
+
+    def post(self: 'T', *args, **kwargs) -> 'T':
+        from ... import DocumentArray
+
+        _tmp = DocumentArray(self)
+        return _tmp.post(*args, **kwargs)[0]
diff --git a/tests/unit/array/mixins/test_post.py b/tests/unit/array/mixins/test_post.py
@@ -3,7 +3,7 @@
 
 import pytest
 
-from docarray import DocumentArray
+from docarray import DocumentArray, Document
 from docarray.helper import random_port
 
 
@@ -39,6 +39,8 @@ def test_post_with_jinahub(hub_uri):
     da = DocumentArray.empty(100)
     da.post(hub_uri)
 
+    assert isinstance(Document().post(hub_uri), Document)
+
 
 def test_post_bad_scheme():
     da = DocumentArray.empty(100)