atlasai-dstoolkit-client 0.0.14__py3-none-any.whl → 0.0.15__py3-none-any.whl

atlasai/toolkit/workflows.py

@@ -11,6 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+import ast
 import logging
 import os
 import time
@@ -74,6 +75,11 @@ class Workflow:
         return result
 
     def logs(self, only_errors=True):
+        def cleanup_logs(logs):
+            if not logs:
+                return "-"
+            return ast.literal_eval(logs).decode('utf-8')
+
         if not self.id:
             raise Exception('No valid execution of the workflow. Please use the `run` method first.')
 
@@ -86,7 +92,6 @@ class Workflow:
         o.print_subtitle(f'Message: {workflow["data"]["message"] or "-"}')
 
         o.print_body('---------------------------------------------------')
-
         if not result['data']:
             if only_errors:
                 o.print_title('No errors detected.')
@@ -96,7 +101,7 @@ class Workflow:
         for pod, data in result['data'].items():
             o.print_title(f'Pod: {pod}')
             o.print_subtitle(f'Message: {data["message"] or "-"}')
-            o.print_subtitle(f'Logs: {data["logs"] or "-"}')
+            o.print_subtitle(f'Logs: {cleanup_logs(data.get("logs"))}')
             o.print_body('---------------------------------------------------')
 
     def run(self, wait_until_complete=False):

{atlasai_dstoolkit_client-0.0.14.dist-info → atlasai_dstoolkit_client-0.0.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: atlasai-dstoolkit-client
-Version: 0.0.14
+Version: 0.0.15
 Summary: UNKNOWN
 Home-page: UNKNOWN
 Author: AtlasAI SWE

{atlasai_dstoolkit_client-0.0.14.dist-info → atlasai_dstoolkit_client-0.0.15.dist-info}/RECORD

@@ -10,9 +10,9 @@ atlasai/toolkit/model.py,sha256=RUe0HbDpzvHOV9A4rzG3PgN9boMWDHQ2tR7IKHXzbx8,4126
 atlasai/toolkit/output.py,sha256=FyDjrpVlbrEyfHfwOpxp8H57jx_qXahDjO1qpHIeuYM,473
 atlasai/toolkit/requests.py,sha256=X86nIo07hAjUlilZcZ1lV8RB7KOsTKbTGtcY_SpFEXY,1223
 atlasai/toolkit/utils.py,sha256=lYh3P2XOshRgHCjFeXJ0FOJWQW64sddgx8c2kL6Wqwc,1566
-atlasai/toolkit/workflows.py,sha256=HlmLNma1P4xyd74gzM5moBXPGjVQqziGOtyRTXBXUhw,6994
-atlasai_dstoolkit_client-0.0.14.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-atlasai_dstoolkit_client-0.0.14.dist-info/METADATA,sha256=vWbApYiDX_XA2mFvwXKRWDCV6KW5OwOYL-jqJN4S56A,1471
-atlasai_dstoolkit_client-0.0.14.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-atlasai_dstoolkit_client-0.0.14.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
-atlasai_dstoolkit_client-0.0.14.dist-info/RECORD,,
+atlasai/toolkit/workflows.py,sha256=9NTqPZqoT46tdjMuFApYsGUpbNNDP_Gp2gYMXgVLo_s,7158
+atlasai_dstoolkit_client-0.0.15.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+atlasai_dstoolkit_client-0.0.15.dist-info/METADATA,sha256=Ada-4ks6hKHOq1GmPEJJKnO9TlATP2XVRyAat70spbo,1471
+atlasai_dstoolkit_client-0.0.15.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+atlasai_dstoolkit_client-0.0.15.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
+atlasai_dstoolkit_client-0.0.15.dist-info/RECORD,,