Merge pull request #2 from lion-agi/modified-init

updated init, bump version to 0.1.1

README.md

@@ -1 +1,3 @@
-# lion-service
+# lion-service
+
+API services for lion framework

lion_service/__init__.py

@@ -1,5 +1,13 @@
-from .version import __version__
+from .rate_limiter import RateLimiter, RateLimitError
 from .service import Service, register_service
-from .rate_limiter import RateLimiter
+from .service_util import invoke_retry
+from .version import __version__
 
-__all__ = ["Service", "register_service", "RateLimiter", "__version__"]
+__all__ = [
+    "Service",
+    "register_service",
+    "RateLimiter",
+    "__version__",
+    "RateLimitError",
+    "invoke_retry",
+]

lion_service/rate_limiter.py

@@ -1,9 +1,10 @@
 from collections import deque
-from datetime import datetime, UTC, timezone
+from datetime import UTC, datetime, timezone
 
 from pydantic import BaseModel, Field
 
-from .complete_request_info import CompleteRequestInfo, CompleteRequestTokenInfo
+from .complete_request_info import (CompleteRequestInfo,
+                                    CompleteRequestTokenInfo)
 
 
 class RateLimiter(BaseModel):

lion_service/service_util.py

@@ -1,5 +1,6 @@
 import asyncio
 from functools import wraps
+
 from .rate_limiter import RateLimitError
 
 

lion_service/token_calculator.py

@@ -1,6 +1,6 @@
-import tiktoken
 from abc import abstractmethod
 
+import tiktoken
 from pydantic import BaseModel, Field, field_validator
 
 

lion_service/version.py

@@ -1 +1 @@
-__version__ = "0.1.0"
+__version__ = "0.1.1"

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "lion-service"
-version = "0.1.0"
+version = "0.1.1"
 description = ""
 authors = ["OceanLi <[email protected]>"]
 readme = "README.md"
@@ -14,6 +14,7 @@ tiktoken = "^0.7.0"
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.3.3"
 black = "^24.8.0"
+isort = "^5.13.2"
 
 [build-system]
 requires = ["poetry-core"]