vayuapi 0.2.2__tar.gz → 0.2.4__tar.gz

{vayuapi-0.2.2 → vayuapi-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: vayuapi
-Version: 0.2.2
+Version: 0.2.4
 Summary: The fastest Python async API framework for rapid development with AI/ML support
 Home-page: https://github.com/vayuapi/vayuapi
 Author: VayuAPI Team
@@ -53,6 +53,7 @@ Requires-Dist: jinja2>=3.1.2
 Requires-Dist: httpx>=0.26.0
 Requires-Dist: itsdangerous
 Requires-Dist: pydantic[email]
+Requires-Dist: sse-starlette>=2.1.0
 Provides-Extra: django
 Requires-Dist: django<6.0.0,>=4.2.0; extra == "django"
 Requires-Dist: psycopg2-binary>=2.9.0; extra == "django"
@@ -80,18 +81,41 @@ Requires-Dist: langchain-community>=0.0.20; extra == "ai"
 Requires-Dist: pydantic-ai>=0.0.5; extra == "ai"
 Requires-Dist: openai>=1.10.0; extra == "ai"
 Requires-Dist: anthropic>=0.18.0; extra == "ai"
+Provides-Extra: rag
+Requires-Dist: langchain>=0.1.0; extra == "rag"
+Requires-Dist: langchain-chroma>=0.1.0; extra == "rag"
+Requires-Dist: langchain-pinecone>=0.1.0; extra == "rag"
+Requires-Dist: sentence-transformers>=2.3.0; extra == "rag"
+Requires-Dist: tiktoken>=0.6.0; extra == "rag"
 Provides-Extra: scheduler
 Requires-Dist: apscheduler>=3.10.0; extra == "scheduler"
+Provides-Extra: celery
+Requires-Dist: celery>=5.3.0; extra == "celery"
+Requires-Dist: redis>=5.0.0; extra == "celery"
+Requires-Dist: kombu>=5.3.0; extra == "celery"
+Provides-Extra: arq
+Requires-Dist: arq>=0.25.0; extra == "arq"
+Requires-Dist: redis>=5.0.0; extra == "arq"
+Provides-Extra: grpc
+Requires-Dist: grpcio>=1.60.0; extra == "grpc"
+Requires-Dist: grpcio-tools>=1.60.0; extra == "grpc"
+Requires-Dist: grpcio-reflection>=1.60.0; extra == "grpc"
+Requires-Dist: protobuf>=4.25.0; extra == "grpc"
 Provides-Extra: security
 Requires-Dist: cryptography>=41.0.0; extra == "security"
 Requires-Dist: pyjwt>=2.8.0; extra == "security"
 Requires-Dist: passlib>=1.7.4; extra == "security"
 Requires-Dist: bcrypt>=4.1.0; extra == "security"
+Provides-Extra: http
+Requires-Dist: httpx>=0.26.0; extra == "http"
+Requires-Dist: aiohttp>=3.9.0; extra == "http"
 Provides-Extra: test
 Requires-Dist: pytest>=7.4.0; extra == "test"
 Requires-Dist: pytest-asyncio>=0.23.0; extra == "test"
 Requires-Dist: pytest-cov>=4.1.0; extra == "test"
 Requires-Dist: httpx>=0.26.0; extra == "test"
+Provides-Extra: all
+Requires-Dist: vayuapi[ai,arq,celery,django,grpc,http,nosql,orm,rag,scheduler,security,sqlalchemy,vector]; extra == "all"
 Provides-Extra: dev
 Requires-Dist: vayuapi[test]; extra == "dev"
 Requires-Dist: black>=24.0.0; extra == "dev"
@@ -104,29 +128,10 @@ Requires-Dist: black>=24.0.0; extra == "dev-all"
 Requires-Dist: ruff>=0.2.0; extra == "dev-all"
 Requires-Dist: mypy>=1.8.0; extra == "dev-all"
 Requires-Dist: pre-commit>=3.6.0; extra == "dev-all"
-Provides-Extra: all
-Requires-Dist: vayuapi[ai,arq,celery,django,grpc,http,nosql,orm,rag,scheduler,security,sqlalchemy,vector]; extra == "all"
-Provides-Extra: rag
-Requires-Dist: langchain>=0.1.0; extra == "rag"
-Requires-Dist: langchain-chroma>=0.1.0; extra == "rag"
-Requires-Dist: langchain-pinecone>=0.1.0; extra == "rag"
-Requires-Dist: sentence-transformers>=2.3.0; extra == "rag"
-Requires-Dist: tiktoken>=0.6.0; extra == "rag"
-Provides-Extra: celery
-Requires-Dist: celery>=5.3.0; extra == "celery"
-Requires-Dist: redis>=5.0.0; extra == "celery"
-Requires-Dist: kombu>=5.3.0; extra == "celery"
-Provides-Extra: arq
-Requires-Dist: arq>=0.25.0; extra == "arq"
-Requires-Dist: redis>=5.0.0; extra == "arq"
-Provides-Extra: grpc
-Requires-Dist: grpcio>=1.60.0; extra == "grpc"
-Requires-Dist: grpcio-tools>=1.60.0; extra == "grpc"
-Requires-Dist: grpcio-reflection>=1.60.0; extra == "grpc"
-Requires-Dist: protobuf>=4.25.0; extra == "grpc"
-Provides-Extra: http
-Requires-Dist: httpx>=0.26.0; extra == "http"
-Requires-Dist: aiohttp>=3.9.0; extra == "http"
+Dynamic: author
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
 
 # VayuAPI Framework 🔥
 

{vayuapi-0.2.2 → vayuapi-0.2.4}/pyproject.toml

@@ -1,12 +1,12 @@
 [build-system]
-requires = ["setuptools>=61.0,<69.0", "wheel"]
+requires = ["setuptools>=61.0", "wheel"]
 build-backend = "setuptools.build_meta"
 [tool.setuptools.packages.find]
 include = ["vayuapi*"]
 
 [project]
 name = "vayuapi"
-version = "0.2.2"
+version = "0.2.4"
 description = "The fastest Python async API framework for rapid development with AI/ML support"
 readme = "README.md"
 requires-python = ">=3.11,<3.16"
@@ -62,7 +62,8 @@ dependencies = [
     "jinja2>=3.1.2",
     "httpx>=0.26.0",
     "itsdangerous",
-    "pydantic[email]"
+    "pydantic[email]",
+    "sse-starlette>=2.1.0"
 
 ]
 

{vayuapi-0.2.2 → vayuapi-0.2.4}/requirements.txt

@@ -55,3 +55,4 @@ black>=24.0.0
 ruff>=0.2.0
 itsdangerous
 pydantic[email]
+sse-starlette>=2.1.0

{vayuapi-0.2.2 → vayuapi-0.2.4}/setup.py

@@ -7,7 +7,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="vayuapi",
-    version="0.2.2",
+    version="0.2.4",
     author="VayuAPI Team",
     author_email="codeswithalok@gmail.com",
     description="The fastest Python async API framework for rapid development with AI/ML support",

{vayuapi-0.2.2 → vayuapi-0.2.4}/vayuapi/__init__.py

@@ -3,12 +3,11 @@
 VayuAPI - The fastest Python async API framework
 """
 
-__version__ = "0.1.0"
+__version__ = "0.2.3"
 __author__ = "VayuAPI Team"
 __license__ = "MIT"
 
 from vayuapi.core.application import VayuAPI
-from vayuapi.core.responses import JSONResponse
 from vayuapi.core.routing import Router, Route
 from vayuapi.core.middleware import Middleware
 from vayuapi.core.websocket import WebSocket, WebSocketManager
@@ -59,6 +58,7 @@ from vayuapi.core.concurrency import (
 
 # Exception handling
 from vayuapi.core.exceptions import (
+    ExceptionHandler,
     HTTPException,
     RequestValidationError,
     WebSocketException,
@@ -75,11 +75,15 @@ from vayuapi.core.exceptions import (
 
 # Request/Response types
 from starlette.requests import Request
-from starlette.responses import (
+from vayuapi.response import (
     Response,
+    JSONResponse,
     HTMLResponse,
+    PlainTextResponse,
+    RedirectResponse,
     StreamingResponse,
     FileResponse,
+    EventSourceResponse,
 )
 
 # Decorators
@@ -109,8 +113,11 @@ __all__ = [
     "Response",
     "JSONResponse",
     "HTMLResponse",
+    "PlainTextResponse",
+    "RedirectResponse",
     "StreamingResponse",
     "FileResponse",
+    "EventSourceResponse",
     # Parameters
     "Path",
     "Query",

{vayuapi-0.2.2 → vayuapi-0.2.4}/vayuapi/core/application.py

@@ -61,6 +61,7 @@ class VayuAPI:
         allowed_origins: List[str] = None,
         middleware: List[Middleware] = None,
         lifespan: Optional[Callable] = None,
+        exception_handlers: Optional[Dict[Type[Exception], Callable]] = None,
     ):
         """
         Initialize VayuAPI application.
@@ -81,6 +82,7 @@ class VayuAPI:
             allowed_origins: List of allowed origins for CORS
             middleware: List of middleware to add
             lifespan: Lifespan context manager
+            exception_handlers: Dict mapping exception classes to handler callables
         """
         self.title = title
         self.version = version
@@ -154,7 +156,10 @@ class VayuAPI:
 
         # Exception handlers
         from vayuapi.core.exceptions import ExceptionHandler
-        self.exception_handler = ExceptionHandler()
+        self._exception_handler = ExceptionHandler()
+        if exception_handlers:
+            for exc_class, handler in exception_handlers.items():
+                self._exception_handler.add_handler(exc_class, handler)
 #---------------------Routes------------------------------Added-------#
     @property
     def routes(self):
@@ -343,7 +348,27 @@ class VayuAPI:
             exc_class: Exception class to handle
             handler: Async function that handles the exception
         """
-        self.exception_handler.add_handler(exc_class, handler)
+        self._exception_handler.add_handler(exc_class, handler)
+
+    def exception_handler(self, exc_class: Type[Exception]):
+        """
+        Decorator to register a custom exception handler.
+
+        Example:
+            ```python
+            @app.exception_handler(HTTPException)
+            async def http_error_handler(request, exc):
+                return JSONResponse(status_code=exc.status_code, content={"error": exc.detail})
+
+            @app.exception_handler(RedirectToLogin)
+            async def redirect_handler(request, exc):
+                return RedirectResponse(url="/login")
+            ```
+        """
+        def decorator(handler: Callable):
+            self._exception_handler.add_handler(exc_class, handler)
+            return handler
+        return decorator
 
     def route(
         self,
@@ -649,7 +674,7 @@ class VayuAPI:
                             return JSONResponse({"message": "Success"})
                         return JSONResponse({"data": result})
                     except Exception as e:
-                        return await self.exception_handler.handle(request, e)
+                        return await self._exception_handler.handle(request, e)
                 return simple_wrapped
             else:
                 async def simple_wrapped_sync(request: Request) -> Response:
@@ -668,7 +693,7 @@ class VayuAPI:
                             return JSONResponse({"message": "Success"})
                         return JSONResponse({"data": result})
                     except Exception as e:
-                        return await self.exception_handler.handle(request, e)
+                        return await self._exception_handler.handle(request, e)
                 return simple_wrapped_sync
 
 #------------------------------------------------------------------------------------------------------------#
@@ -867,10 +892,10 @@ class VayuAPI:
             except ValidationError as e:
                 from vayuapi.core.exceptions import RequestValidationError
                 validation_error = RequestValidationError(e.errors())
-                return await self.exception_handler.handle(request, validation_error)
+                return await self._exception_handler.handle(request, validation_error)
             except Exception as e:
                 # Use exception handler
-                return await self.exception_handler.handle(request, e)
+                return await self._exception_handler.handle(request, e)
 
         return wrapped
 
@@ -952,7 +977,7 @@ class VayuAPI:
             from starlette.exceptions import HTTPException as StarletteHTTPException
 
             mw_stack = self.middleware_stack  # capture reference
-            exc_handler = self.exception_handler  # capture reference
+            exc_handler = self._exception_handler  # capture reference
 
             class _VayuMiddlewareWrapper(BaseHTTPMiddleware):
                 async def dispatch(self, request, call_next):

{vayuapi-0.2.2 → vayuapi-0.2.4}/vayuapi/core/responses.py

@@ -9,7 +9,16 @@ from uuid import UUID
 from typing import Any
 from enum import Enum
 
-from starlette.responses import JSONResponse as StarletteJSONResponse
+from starlette.responses import (
+    JSONResponse as StarletteJSONResponse,
+    Response,
+    HTMLResponse,
+    PlainTextResponse,
+    RedirectResponse,
+    StreamingResponse,
+    FileResponse,
+)
+from sse_starlette.sse import EventSourceResponse
 from pydantic import BaseModel
 
 
@@ -71,3 +80,15 @@ class JSONResponse(StarletteJSONResponse):
             separators=(",", ":"),
             cls=AgniJSONEncoder,
         ).encode("utf-8")
+
+
+__all__ = [
+    "Response",
+    "JSONResponse",
+    "HTMLResponse",
+    "PlainTextResponse",
+    "RedirectResponse",
+    "StreamingResponse",
+    "FileResponse",
+    "EventSourceResponse",
+]

vayuapi-0.2.4/vayuapi/response.py

@@ -0,0 +1,25 @@
+"""
+VayuAPI response classes — importable from vayuapi.response
+"""
+
+from vayuapi.core.responses import (
+    Response,
+    JSONResponse,
+    HTMLResponse,
+    PlainTextResponse,
+    RedirectResponse,
+    StreamingResponse,
+    FileResponse,
+    EventSourceResponse,
+)
+
+__all__ = [
+    "EventSourceResponse",
+    "FileResponse",
+    "HTMLResponse",
+    "JSONResponse",
+    "PlainTextResponse",
+    "RedirectResponse",
+    "Response",
+    "StreamingResponse",
+]

{vayuapi-0.2.2 → vayuapi-0.2.4}/vayuapi.egg-info/SOURCES.txt

@@ -5,6 +5,7 @@ pyproject.toml
 requirements.txt
 setup.py
 vayuapi/__init__.py
+vayuapi/response.py
 vayuapi/status.py
 vayuapi/admin/__init__.py
 vayuapi/admin/enhanced_panel.py