clean up

Author: iamNarcisse

tests/test_embedding_async.py

@@ -0,0 +1,23 @@
+from unittest.mock import MagicMock
+import unittest
+from jigsawstack.exceptions import JigsawStackError
+from jigsawstack import AsyncJigsawStack
+import pytest
+import asyncio
+import logging
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
+def test_async_embedding_generation_response():
+    async def _test():
+        client = AsyncJigsawStack()
+        try:
+            result = await client.embedding({"text": "Hello, World!", "type": "text"})
+            logger.info(result)
+            assert result["success"] == True
+        except JigsawStackError as e:
+            pytest.fail(f"Unexpected JigsawStackError: {e}")
+
+    asyncio.run(_test())