wizelit-sdk 0.1.23__tar.gz → 0.1.25__tar.gz

{wizelit_sdk-0.1.23 → wizelit_sdk-0.1.25}/PKG-INFO

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

{wizelit_sdk-0.1.23 → wizelit_sdk-0.1.25}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "wizelit-sdk"
-version = "0.1.23"
+version = "0.1.25"
 description = "Wizelit Agent Wrapper - Internal utility package"
 readme = "README.md"
 requires-python = ">=3.10"

{wizelit_sdk-0.1.23 → wizelit_sdk-0.1.25}/src/wizelit_sdk/__init__.py

@@ -1,11 +1,11 @@
 """Wizelit SDK package."""
 
-from wizelit_sdk.agent_wrapper import WizelitAgentWrapper
+from wizelit_sdk.agent_wrapper import WizelitAgent
 from wizelit_sdk.database import DatabaseManager
 from wizelit_sdk.agent_wrapper.job import Job
 from wizelit_sdk.agent_wrapper.streaming import LogStreamer
 from wizelit_sdk.models.base import BaseModel
 from wizelit_sdk.models.job import JobModel, JobLogModel, JobStatus
 
-__all__ = ["WizelitAgentWrapper", "DatabaseManager", "Job", "LogStreamer", "BaseModel", "JobModel", "JobLogModel", "JobStatus"]
+__all__ = ["WizelitAgent", "DatabaseManager", "Job", "LogStreamer", "BaseModel", "JobModel", "JobLogModel", "JobStatus"]
 

wizelit_sdk-0.1.25/src/wizelit_sdk/agent_wrapper/__init__.py

@@ -0,0 +1,8 @@
+"""Agent Wrapper - Internal utility package."""
+
+# Import main functions
+from .agent_wrapper import WizelitAgent
+from .job import Job
+from .streaming import LogStreamer
+
+__all__ = ["WizelitAgent", "Job", "LogStreamer"]

{wizelit_sdk-0.1.23 → wizelit_sdk-0.1.25}/src/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
+from typing import Callable, Any, Optional, Literal, Dict, TYPE_CHECKING, Union
 from contextvars import ContextVar
 from fastmcp import FastMCP, Context
 from fastmcp.dependencies import CurrentContext
@@ -34,7 +34,7 @@ class CurrentJob:
         return _current_job.get()
 
 
-class WizelitAgentWrapper:
+class WizelitAgent:
     """
     Main wrapper class that converts Python functions into MCP server tools.
     Built on top of fast-mcp with enhanced streaming and agent framework support.
@@ -87,7 +87,7 @@ class WizelitAgentWrapper:
                 print(f"Warning: Failed to initialize log streamer: {e}")
 
         print(
-            f"WizelitAgentWrapper initialized with name: {name}, transport: {transport}, host: {host}, port: {port}"
+            f"WizelitAgent initialized with name: {name}, transport: {transport}, host: {host}, port: {port}"
         )
 
     def ingest(
@@ -362,9 +362,10 @@ class WizelitAgentWrapper:
                 return result
 
             except Exception as e:
-                # Mark job as failed (only if job exists)
-                if job is not None:
-                    job.status = "failed"
+                # Mark job as failed when a job instance exists
+                active_job = job or _current_job.get()
+                if active_job is not None:
+                    active_job.status = "failed"
 
                 # Stream error information
                 await ctx.report_progress(
@@ -483,7 +484,7 @@ class WizelitAgentWrapper:
             return None
 
         try:
-            from models.job import JobModel
+            from wizelit_sdk.models.job import JobModel
             from sqlalchemy import select
 
             async with self._db_manager.get_session() as session:
@@ -532,7 +533,7 @@ class WizelitAgentWrapper:
             return None
 
         try:
-            from models.job import JobLogModel
+            from wizelit_sdk.models.job import JobLogModel
             from sqlalchemy import select
 
             async with self._db_manager.get_session() as session:
@@ -579,7 +580,7 @@ class WizelitAgentWrapper:
         return True
 
     def set_job_result(
-        self, job_id: str, result: Optional[str | dict[str, Any]]
+        self, job_id: str, result: Optional[Union[str, dict[str, Any]]]
     ) -> bool:
         """
         Set the result of a job by job_id.

{wizelit_sdk-0.1.23 → wizelit_sdk-0.1.25}/src/wizelit_sdk/agent_wrapper/job.py

@@ -60,7 +60,7 @@ class DatabaseLogHandler(logging.Handler):
         """
         try:
             # Import here to avoid circular dependency
-            from models.job import JobLogModel
+            from wizelit_sdk.models.job import JobLogModel
             from datetime import datetime
 
             async def write_log():
@@ -274,7 +274,7 @@ class Job:
             try:
                 result = await coro
                 # Store string results for convenience
-                if isinstance(result, str|dict):
+                if isinstance(result, (str, dict)):
                     self.result = result
                 if self._status == "running":
                     self.status = "completed"
@@ -351,7 +351,7 @@ class Job:
             return
 
         try:
-            from models.job import JobModel
+            from wizelit_sdk.models.job import JobModel
 
             async with self._db_manager.get_session() as session:
                 # Check if job already exists

wizelit_sdk-0.1.25/tests/test_agent_wrapper_import.py

@@ -0,0 +1,6 @@
+def test_imports():
+    # Simple import smoke test to ensure public symbols are available
+    from wizelit_sdk.agent_wrapper import WizelitAgent, Job  # type: ignore
+
+    assert WizelitAgent is not None
+    assert Job is not None

wizelit_sdk-0.1.25/tests/test_utils.py

@@ -0,0 +1,18 @@
+"""Public API tests for agent_wrapper."""
+
+import wizelit_sdk.agent_wrapper as agent_wrapper
+
+
+def test_agent_wrapper_public_api_is_minimal():
+    """Ensure placeholder utilities are not exposed via the public API."""
+    assert not hasattr(agent_wrapper, "greet")
+    assert not hasattr(agent_wrapper, "greet_many")
+    assert not hasattr(agent_wrapper, "greet_many3")
+
+
+def test_agent_wrapper_exports_core_symbols():
+    """Ensure core classes remain publicly available."""
+    assert agent_wrapper.WizelitAgent is not None
+    assert agent_wrapper.Job is not None
+    assert agent_wrapper.LogStreamer is not None
+

{wizelit_sdk-0.1.23 → wizelit_sdk-0.1.25}/uv.lock

@@ -2038,7 +2038,7 @@ wheels = [
 
 [[package]]
 name = "wizelit-sdk"
-version = "0.1.23"
+version = "0.1.25"
 source = { editable = "." }
 dependencies = [
     { name = "asyncpg" },

wizelit_sdk-0.1.23/src/wizelit_sdk/agent_wrapper/__init__.py

@@ -1,9 +0,0 @@
-"""Agent Wrapper - Internal utility package."""
-
-# Import main functions
-from .utils import greet, greet_many, greet_many3
-from .agent_wrapper import WizelitAgentWrapper
-from .job import Job
-from .streaming import LogStreamer
-
-__all__ = ["greet", "greet_many", "greet_many3", "WizelitAgentWrapper", "Job", "LogStreamer"]

wizelit_sdk-0.1.23/tests/test_agent_wrapper_import.py

@@ -1,6 +0,0 @@
-def test_imports():
-    # Simple import smoke test to ensure public symbols are available
-    from wizelit_sdk.agent_wrapper import WizelitAgentWrapper, Job  # type: ignore
-
-    assert WizelitAgentWrapper is not None
-    assert Job is not None

wizelit_sdk-0.1.23/tests/test_utils.py

@@ -1,23 +0,0 @@
-"""Tests for utility functions."""
-
-import pytest
-from wizelit_sdk.agent_wrapper import greet
-
-
-def test_greet_with_name():
-    """Test greet function with a name."""
-    result = greet("Alice")
-    assert result == "Hello, Alice!"
-
-
-def test_greet_default():
-    """Test greet function with default name."""
-    result = greet()
-    assert result == "Hello, World!"
-
-
-def test_greet_empty_string():
-    """Test greet function with empty string."""
-    result = greet("")
-    assert result == "Hello, !"
-