minor fixes

unifyai · Apr 1, 2024 · 145872b · 145872b
1 parent 60d157b
commit 145872b
Show file tree

Hide file tree

Showing 7 changed files with 47 additions and 24 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -14,7 +14,7 @@ mypy = "^1.1.1"
 isort = "^5.11.4"
 pre-commit = "^3.0.1"
 wemake-python-styleguide = "^0.17.0"
-black = "^22.12.0"
+black = "^24.3.0"
 autoflake = "^1.6.1"
 
 [tool.poetry.group.dev.dependencies]

diff --git a/unify/__init__.py b/unify/__init__.py
diff --git a/unifyai/__init__.py b/unifyai/__init__.py
@@ -0,0 +1,3 @@
+"""Unify python module."""
+
+from unifyai.clients import AsyncUnify, Unify  # noqa: F403
diff --git a/unify/clients.py → unifyai/clients.py b/unify/clients.py → unifyai/clients.py
@@ -2,7 +2,7 @@
 from typing import AsyncGenerator, Dict, Generator, List, Optional, Union
 
 import openai
-from unify.exceptions import UnifyError, status_error_map
+from unifyai.exceptions import UnifyError, status_error_map
 
 
 def _validate_api_key(api_key: Optional[str]) -> str:

diff --git a/unify/exceptions.py → unifyai/exceptions.py b/unify/exceptions.py → unifyai/exceptions.py
diff --git a/unify/tests.py → unifyai/tests.py b/unify/tests.py → unifyai/tests.py
@@ -3,8 +3,8 @@
 from types import AsyncGeneratorType, GeneratorType
 from unittest.mock import MagicMock, patch
 
-from unify.clients import AsyncUnify, Unify
-from unify.exceptions import AuthenticationError, InternalServerError
+from unifyai.clients import AsyncUnify, Unify
+from unifyai.exceptions import AuthenticationError, InternalServerError
 
 
 class TestUnify(unittest.TestCase):