wizelit-sdk 0.1.23__py3-none-any.whl → 0.1.25__py3-none-any.whl

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/agent_wrapper/__init__.py

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

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/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.23.dist-info → wizelit_sdk-0.1.25.dist-info}/METADATA

@@ -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.25.dist-info/RECORD

@@ -0,0 +1,13 @@
+wizelit_sdk/__init__.py,sha256=76Dt-WxNbEJwOCPqu2P9AooVSQoGmc8-CH1fxIWJ5Lo,471
+wizelit_sdk/database.py,sha256=39Vu5A8CV7u7ItIwRgrkAugD9UqNbG5vQrCsMOgTamc,4631
+wizelit_sdk/agent_wrapper/__init__.py,sha256=OAF36jAmI0A3ByppsYMzJDHeDG9X0Ac4LbFzNp0b7jw,219
+wizelit_sdk/agent_wrapper/agent_wrapper.py,sha256=e9KCQSpC7hox3q6sqzhOLDIq37btw7IT4ejeuljc6Gk,22742
+wizelit_sdk/agent_wrapper/job.py,sha256=oG7Ev-EW0GiLGM7fHhQrJJCdr_pJJsaDj9Tjg31_eW0,13831
+wizelit_sdk/agent_wrapper/streaming.py,sha256=f0VV3IzGAlfQY_cw2OHgxWjvM16Hs42_b700EUX2QpY,6633
+wizelit_sdk/agent_wrapper/utils.py,sha256=OUihj3Kk6uI0y7bDr_L_YQs33vSK0GAJRl0c4km4DNs,951
+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.25.dist-info/METADATA,sha256=-7bPTthsvDaNfjR_nPDoOLsIAclM_N2TA0-ND5ln1-k,2778
+wizelit_sdk-0.1.25.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+wizelit_sdk-0.1.25.dist-info/RECORD,,

wizelit_sdk-0.1.23.dist-info/RECORD

@@ -1,13 +0,0 @@
-wizelit_sdk/__init__.py,sha256=6UyHRL8Wu0CAWHIErlmDZFxqI_BZwsco_DWnoMr_H6A,485
-wizelit_sdk/database.py,sha256=39Vu5A8CV7u7ItIwRgrkAugD9UqNbG5vQrCsMOgTamc,4631
-wizelit_sdk/agent_wrapper/__init__.py,sha256=q4xsoiAyI_KM9TSL9rFfxPzGrONJt3GF6MlEx6vDkis,321
-wizelit_sdk/agent_wrapper/agent_wrapper.py,sha256=r1N47KT4Sh6Gudt6ZEFNXvDDD8g4kOkuwTm19K9Np9c,22644
-wizelit_sdk/agent_wrapper/job.py,sha256=Vvmm6eTX1cbXpFuwPMLaom6hEgdk3ovitnaj9d9p02o,13804
-wizelit_sdk/agent_wrapper/streaming.py,sha256=f0VV3IzGAlfQY_cw2OHgxWjvM16Hs42_b700EUX2QpY,6633
-wizelit_sdk/agent_wrapper/utils.py,sha256=OUihj3Kk6uI0y7bDr_L_YQs33vSK0GAJRl0c4km4DNs,951
-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.23.dist-info/METADATA,sha256=ta0J_LwnCa4NGBO2q-v0c23_G-b-SfK6Okhhzr-3zjM,2778
-wizelit_sdk-0.1.23.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-wizelit_sdk-0.1.23.dist-info/RECORD,,