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

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 WizelitAgent
 from .job import Job
 from .streaming import LogStreamer
 
-__all__ = ["greet", "greet_many", "greet_many3", "WizelitAgent", "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
@@ -362,9 +362,10 @@ class WizelitAgent:
                 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 WizelitAgent:
             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 WizelitAgent:
             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 WizelitAgent:
         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.24.dist-info → wizelit_sdk-0.1.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizelit-sdk
-Version: 0.1.24
+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.24.dist-info → wizelit_sdk-0.1.25.dist-info}/RECORD

@@ -1,13 +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=JuFeLxaANm1klxeQ14n6UjbpHNFQCzofa-FlcDVyo5Q,307
-wizelit_sdk/agent_wrapper/agent_wrapper.py,sha256=6UAJjNrdPIpDHU2ipF5M42umddM77CXU5imSn2tTv0Q,22630
-wizelit_sdk/agent_wrapper/job.py,sha256=Vvmm6eTX1cbXpFuwPMLaom6hEgdk3ovitnaj9d9p02o,13804
+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.24.dist-info/METADATA,sha256=FnRm-NWPMsTIc6v-SbF8lo7zv8s24lPmCWUaYL6sbE4,2778
-wizelit_sdk-0.1.24.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-wizelit_sdk-0.1.24.dist-info/RECORD,,
+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,,