wizelit-sdk 0.1.28__py3-none-any.whl → 0.1.30__py3-none-any.whl

wizelit_sdk/agent_wrapper/agent_wrapper.py

@@ -3,7 +3,7 @@ import asyncio
 import inspect
 import logging
 import os
-from typing import Callable, Any, Optional, Literal, Dict, TYPE_CHECKING, Union
+from typing import Callable, Any, Optional, Literal, Dict, TYPE_CHECKING, Union, cast
 from contextvars import ContextVar
 from fastmcp import FastMCP, Context
 from fastmcp.dependencies import CurrentContext
@@ -14,6 +14,9 @@ from wizelit_sdk.agent_wrapper.signature_validation import (
     ensure_type_hints,
 )
 
+# Local Transport literal to avoid import issues when fastmcp.types is unavailable
+Transport = Literal["stdio", "http", "sse", "streamable-http"]
+
 if TYPE_CHECKING:
     from wizelit_sdk.database import DatabaseManager
 
@@ -73,7 +76,7 @@ class WizelitAgent:
         self._tools = {}
         self._jobs: Dict[str, Job] = {}  # Store jobs by job_id
         self._host = host
-        self._transport = transport
+        self._transport: Transport = cast(Transport, transport)
         self._port = port
         self._db_manager = db_manager
         self._log_streamer = None
@@ -257,9 +260,9 @@ class WizelitAgent:
                 )
 
             # Set the signature with ctx as last parameter with CurrentContext() default
-            tool_wrapper.__signature__ = new_sig
-            tool_wrapper.__name__ = tool_name
-            tool_wrapper.__doc__ = tool_description
+            cast(Any, tool_wrapper).__signature__ = new_sig
+            cast(Any, tool_wrapper).__name__ = tool_name
+            cast(Any, tool_wrapper).__doc__ = tool_description
 
             # Copy annotations and add Context
             # Note: We don't add job annotation here since we use Any and exclude it from schema
@@ -271,7 +274,7 @@ class WizelitAgent:
                 new_annotations["job"] = (
                     Any  # Use Any instead of Job to avoid Pydantic schema issues
                 )
-            tool_wrapper.__annotations__ = new_annotations
+            cast(Any, tool_wrapper).__annotations__ = new_annotations
 
             # Register with fast-mcp
             # Exclude ctx and job from schema generation since they're dependency-injected
@@ -392,7 +395,7 @@ class WizelitAgent:
 
     def run(
         self,
-        transport: Optional[str] = None,
+        transport: Optional[Transport] = None,
         host: Optional[str] = None,
         port: Optional[int] = None,
         **kwargs,
@@ -406,7 +409,7 @@ class WizelitAgent:
             port: Port to bind to (for HTTP transports)
             **kwargs: Additional arguments passed to fast-mcp
         """
-        transport = transport or self._transport
+        transport = cast(Transport, transport or self._transport)
         host = host or self._host
         port = port or self._port
         print(f"🚀 Starting {self._name} MCP Server")
@@ -516,12 +519,12 @@ class WizelitAgent:
                     "error": job_model.error,
                     "created_at": (
                         job_model.created_at.isoformat()
-                        if job_model.created_at
+                        if job_model.created_at is not None
                         else None
                     ),
                     "updated_at": (
                         job_model.updated_at.isoformat()
-                        if job_model.updated_at
+                        if job_model.updated_at is not None
                         else None
                     ),
                 }

wizelit_sdk/agent_wrapper/job.py

@@ -6,7 +6,7 @@ import asyncio
 import uuid
 import time
 from datetime import datetime, UTC
-from typing import List, Optional, Awaitable, Any, TYPE_CHECKING
+from typing import Optional, Awaitable, Any, TYPE_CHECKING
 from fastmcp import Context
 
 if TYPE_CHECKING:
@@ -19,7 +19,7 @@ class MemoryLogHandler(logging.Handler):
     Custom logging handler that stores log messages in a list.
     """
 
-    def __init__(self, logs_list: List[str]):
+    def __init__(self, logs_list: list[str]):
         super().__init__()
         self.logs_list = logs_list
         self.setFormatter(logging.Formatter('%(message)s'))
@@ -156,7 +156,7 @@ class Job:
         self._ctx = ctx
         self._id = job_id or f"JOB-{str(uuid.uuid4())[:8]}"
         self._status = "running"
-        self._logs: List[str] = []
+        self._logs: list[str] = []
         self._result: Optional[str] = None
         self._error: Optional[str] = None
         self._db_manager = db_manager
@@ -176,7 +176,7 @@ class Job:
         return self._logger
 
     @property
-    def logs(self) -> List[str]:
+    def logs(self) -> list[str]:
         """List of log messages (timestamped strings)."""
         return self._logs
 

wizelit_sdk/agent_wrapper/streaming.py

@@ -196,11 +196,11 @@ class LogStreamer:
             self._redis = None
             logger.info("Redis connection closed")
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> 'LogStreamer':
         """Async context manager entry."""
         await self._ensure_connected()
         return self
 
-    async def __aexit__(self, exc_type, exc_val, exc_tb):
+    async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
         """Async context manager exit."""
         await self.close()

wizelit_sdk/database.py

@@ -132,7 +132,7 @@ class DatabaseManager:
         finally:
             await session.close()
 
-    async def close(self):
+    async def close(self) -> None:
         """Dispose of the engine and close all connections."""
         await self.engine.dispose()
         logger.info("Database connections closed")

wizelit_sdk/models/base.py

@@ -3,6 +3,7 @@ from sqlalchemy import Column
 from sqlalchemy.dialects.postgresql import UUID
 import uuid
 from datetime import datetime
+from typing import Dict, Any
 
 Base = declarative_base()
 
@@ -10,7 +11,7 @@ class TimestampMixin:
     """Mixin for models that need timestamp functionality."""
 
     @staticmethod
-    def get_timestamp():
+    def get_timestamp() -> str:
         return datetime.utcnow().isoformat()
 
 
@@ -20,6 +21,6 @@ class BaseModel(Base):
 
     id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
 
-    def to_dict(self):
+    def to_dict(self) -> Dict[str, Any]:
         """Convert model to dictionary."""
         return {c.name: getattr(self, c.name) for c in self.__table__.columns}

wizelit_sdk/models/job.py

@@ -41,7 +41,7 @@ class JobModel(BaseModel):
         Index('idx_job_created_at', 'created_at'),
     )
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return f"<JobModel(id={self.id}, status={self.status})>"
 
 
@@ -68,5 +68,5 @@ class JobLogModel(BaseModel):
         Index('idx_job_log_job_id_timestamp', 'job_id', 'timestamp'),
     )
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return f"<JobLogModel(id={self.id}, job_id={self.job_id}, level={self.level})>"

{wizelit_sdk-0.1.28.dist-info → wizelit_sdk-0.1.30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizelit-sdk
-Version: 0.1.28
+Version: 0.1.30
 Summary: Wizelit Agent Wrapper - Internal utility package
 Author-email: Your Name <your.email@company.com>
 Requires-Python: >=3.10

wizelit_sdk-0.1.30.dist-info/RECORD

@@ -0,0 +1,13 @@
+wizelit_sdk/__init__.py,sha256=76Dt-WxNbEJwOCPqu2P9AooVSQoGmc8-CH1fxIWJ5Lo,471
+wizelit_sdk/database.py,sha256=HzvMY-KZQBkGs-Ey_wKAIPOJEb5BqGjEi98Tip7CQBo,4830
+wizelit_sdk/agent_wrapper/__init__.py,sha256=srVcHZfvpf3q8m2nzIST6Ofl2V8Vv5GRtcNXudOLiD8,333
+wizelit_sdk/agent_wrapper/agent_wrapper.py,sha256=qkskmRRjhIooXVeyZHbH5q1OIp30zw29bBRUoqTPYLE,23476
+wizelit_sdk/agent_wrapper/job.py,sha256=jW2Cl7QvSgZDUqWqZSND-V1XphGdISJuh9ViHgCn7iU,13770
+wizelit_sdk/agent_wrapper/signature_validation.py,sha256=Njplwofw36jXoWgBooaZCbwhkf72pR-UoRQ3Q6-biro,3729
+wizelit_sdk/agent_wrapper/streaming.py,sha256=l7SbvPuQYMdGO38t1upyI5wTVdd5eB-2BSwU-BQSlvc,6673
+wizelit_sdk/models/__init__.py,sha256=UB7nfHoE6hGcjfzy7w8AmuKVmYrTg9wIgGjG8GWziJ0,143
+wizelit_sdk/models/base.py,sha256=11XyxLUG0VjdiNxe4zmzGDUjKEOsjSvUjA8yaQgfCa0,758
+wizelit_sdk/models/job.py,sha256=HD8ohVqzakC8J7VAwmh8UCi_BLLiqx85TbR0FjO2rlg,2494
+wizelit_sdk-0.1.30.dist-info/METADATA,sha256=zCb8i8TbnbAVCtHzcBxhgvQOlNIyNehCKQUWVCYpeqA,3297
+wizelit_sdk-0.1.30.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+wizelit_sdk-0.1.30.dist-info/RECORD,,

wizelit_sdk-0.1.28.dist-info/RECORD

@@ -1,13 +0,0 @@
-wizelit_sdk/__init__.py,sha256=76Dt-WxNbEJwOCPqu2P9AooVSQoGmc8-CH1fxIWJ5Lo,471
-wizelit_sdk/database.py,sha256=6F2Nyt-sr5U5GkAB9OeSCU-iv2byZBDsg7aXqCl-ntk,4822
-wizelit_sdk/agent_wrapper/__init__.py,sha256=srVcHZfvpf3q8m2nzIST6Ofl2V8Vv5GRtcNXudOLiD8,333
-wizelit_sdk/agent_wrapper/agent_wrapper.py,sha256=LiSYpnKF-RTx_9UnVgW9KqyfTWtilK_Gft1yCuxujus,23204
-wizelit_sdk/agent_wrapper/job.py,sha256=Gk2CyLBQFzIosYuGrqUj6_rZ9kDIO6YLYQ7Pxk6ASuM,13776
-wizelit_sdk/agent_wrapper/signature_validation.py,sha256=Njplwofw36jXoWgBooaZCbwhkf72pR-UoRQ3Q6-biro,3729
-wizelit_sdk/agent_wrapper/streaming.py,sha256=f0VV3IzGAlfQY_cw2OHgxWjvM16Hs42_b700EUX2QpY,6633
-wizelit_sdk/models/__init__.py,sha256=UB7nfHoE6hGcjfzy7w8AmuKVmYrTg9wIgGjG8GWziJ0,143
-wizelit_sdk/models/base.py,sha256=aEPGMZVczKnlWz4Ps99e_xI5TcuSV3DxCigRMU5BnhE,704
-wizelit_sdk/models/job.py,sha256=P68Vb1k77Z_Tha4dE0GzrLPa0LJrnMCxyYMENZyD7Kc,2480
-wizelit_sdk-0.1.28.dist-info/METADATA,sha256=wBU5HBbuQBXynPvt_L2blH-JzArvPpDgU2jkRZ2kOWQ,3297
-wizelit_sdk-0.1.28.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-wizelit_sdk-0.1.28.dist-info/RECORD,,