wizelit-sdk 0.1.24__py3-none-any.whl → 0.1.26__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

@@ -10,8 +10,8 @@ from typing import List, Optional, Awaitable, Any, TYPE_CHECKING
 from fastmcp import Context
 
 if TYPE_CHECKING:
-    from wizelit_sdk.database import DatabaseManager
-    from wizelit_sdk.agent_wrapper.streaming import LogStreamer
+    from ..database import DatabaseManager
+    from .streaming import LogStreamer
 
 
 class MemoryLogHandler(logging.Handler):
@@ -60,7 +60,7 @@ class DatabaseLogHandler(logging.Handler):
         """
         try:
             # Import here to avoid circular dependency
-            from models.job import JobLogModel
+            from ..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 ..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.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizelit-sdk
-Version: 0.1.24
+Version: 0.1.26
 Summary: Wizelit Agent Wrapper - Internal utility package
 Author-email: Your Name <your.email@company.com>
 Requires-Python: >=3.10
@@ -55,6 +55,18 @@ from wizelit_agent_wrapper import your_module
 result = your_module.function()
 ```
 
+## Configuration
+
+The SDK reads database configuration from the hosting application's environment. Provide these variables in the consuming project (e.g., via your app's .env or deployment secrets):
+
+- POSTGRES_USER
+- POSTGRES_PASSWORD
+- POSTGRES_HOST
+- POSTGRES_PORT
+- POSTGRES_DB
+
+You can also supply a full connection string via `DATABASE_URL` (overrides the individual fields). If using streaming/logging with Redis, set `REDIS_HOST`, `REDIS_PORT`, and optionally `REDIS_PASSWORD`.
+
 ## Development
 
 ### Setup Development Environment
@@ -103,9 +115,9 @@ make versions       # List all available versions
 ## Versioning
 
 We use [Semantic Versioning](https://semver.org/):
+
 - **MAJOR** version for incompatible API changes
 - **MINOR** version for new functionality (backward compatible)
 - **PATCH** version for bug fixes
 
 See [CHANGELOG.md](CHANGELOG.md) for version history.
-

wizelit_sdk-0.1.26.dist-info/RECORD

@@ -0,0 +1,12 @@
+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=Gk2CyLBQFzIosYuGrqUj6_rZ9kDIO6YLYQ7Pxk6ASuM,13776
+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.26.dist-info/METADATA,sha256=0QpbkJ8Z5Vl-2s7xrYHoTrutSAMUgQol9JNr9fgXkgo,3265
+wizelit_sdk-0.1.26.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+wizelit_sdk-0.1.26.dist-info/RECORD,,

wizelit_sdk/agent_wrapper/utils.py

@@ -1,45 +0,0 @@
-"""Utility functions for Wizelit Agent Wrapper."""
-
-
-def greet(name: str = "World") -> str:
-    """
-    Generate a greeting message.
-    
-    Args:
-        name: Name to greet. Defaults to "World".
-        
-    Returns:
-        A greeting string.
-        
-    Example:
-        >>> greet("Alice")
-        'Hello, Alice!'
-        >>> greet()
-        'Hello, World!'
-    """
-    return f"Hello, {name}!"
-
-def greet_many(names: list[str]) -> list[str]:
-    """
-    Generate a greeting message for each name.
-    
-    Args:
-        names: List of names to greet.
-        
-    Returns:
-        A list of greeting strings.
-    """
-    return [f"Hello, {name}!" for name in names]
-
-
-def greet_many3(names: list[str]) -> str:
-    """
-    Generate a greeting message for each name.
-    
-    Args:
-        names: List of names to greet.
-        
-    Returns:
-        A list of greeting strings.
-    """
-    return ', '.join([f"Hello, {name}!" for name in names])

wizelit_sdk-0.1.24.dist-info/RECORD

@@ -1,13 +0,0 @@
-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/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,,