Merge pull request #8 from pokedim13/dev

Dev

pyproject.toml

@@ -1,18 +1,47 @@
 [tool.poetry]
 name = "wombo"
-version = "0.2.1"
-description = ""
-authors = ["H3try <skinxedovich@vk.com>"]
+version = "0.3.0"
+description = "AI picture generated tool"
+authors = ["Алексей Баев <baev@cumproject.ru>"]
 readme = "README.md"
 
 [tool.poetry.dependencies]
 python = "^3.8"
-pydantic = "2.1.1"
-httpx = "^0.24.1"
-pillow = {version = "^10.0.0", optional = true}
+httpx = "^0.27.2"
+pydantic = "^2.9.2"
 
-[tool.poetry.extras]
-gif = ["pillow"]
+[tool.poetry.group.dev.dependencies]
+ruff = "^0.6.8"
+
+[tool.poetry.group.test.dependencies]
+pytest = "^8.3.3"
+pytest-asyncio = "^0.24.0"
+
+
+[tool.ruff]
+line-length = 120
+exclude = [
+    "migrations",
+    "manage.py",
+]
+
+[tool.ruff.lint]
+unfixable = ["B"]
+select = [
+    "F",
+    "B",
+    "E",
+    "I",
+    "UP",
+    "N",
+    "ANN",
+    "Q"
+]
+ignore = ["ANN002", "ANN003", "ANN101", "ANN102", "ANN204"]
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["E402"]
+"**/{tests,docs,tools}/*" = ["E402"]
 
 
 [build-system]

tests/example.py

@@ -0,0 +1,11 @@
+from wombo import AsyncDream, Dream
+import asyncio
+
+async def main():
+    dream = AsyncDream()
+    print(await dream.generate("anime waifu"))
+
+
+dream = Dream()
+print(dream.generate("anime waifu"))
+asyncio.run(main())

tests/test_async.py

@@ -0,0 +1,13 @@
+from wombo import AsyncDream, models
+
+import pytest
+
+dream = AsyncDream()
+
+@pytest.mark.asyncio
+async def test_generate():
+    assert isinstance(await dream.generate("anime waifu"), models.TaskModel)
+
+@pytest.mark.asyncio
+async def test_styles():
+    assert isinstance(await dream.style._get_styles(), models.StyleModel)

tests/test_sync.py

@@ -0,0 +1,9 @@
+from wombo import Dream, models
+
+dream = Dream()
+
+def test_generate():
+    assert isinstance(dream.generate("anime waifu"), models.TaskModel)
+
+def test_styles():
+    assert isinstance(dream.style._get_styles(), models.StyleModel)

wombo/__init__.py

@@ -1,4 +1,4 @@
+from wombo import models
 from wombo.api import AsyncDream, Dream
-from wombo.base_models import Style
 
-__all__ = ["AsyncDream", "Dream", "Style"]
+__all__ = ["models", "AsyncDream", "Dream"]

wombo/api/__init__.py

@@ -1,4 +1,4 @@
-from .async_dream import AsyncDream
-from .dream import Dream
+from wombo.api.async_dream import AsyncDream
+from wombo.api.dream import Dream
 
-__all__ = ["Dream", "AsyncDream"]
+__all__ = ["AsyncDream", "Dream"]