avtomatika 1.0b1__py3-none-any.whl → 1.0b2__py3-none-any.whl

avtomatika/api.html

@@ -305,6 +305,20 @@
                         responses: [
                             { code: '200 OK', description: 'Successful response.', body: "{...}" }
                         ]
+                    },
+                    {
+                        id: 'post-reload-worker-configs',
+                        name: 'Reload Worker Configurations',
+                        method: 'POST',
+                        path: '/api/{version}/admin/reload-workers',
+                        description: 'Triggers a dynamic reload of worker configurations from the TOML file. Requires client authentication.',
+                        parameters: [
+                            { name: 'version', type: 'string', description: 'API Version', example: 'v1' }
+                        ],
+                        request: { body: null },
+                        responses: [
+                            { code: '200 OK', description: 'Successful response.', body: { "status": "worker_configs_reloaded" } }
+                        ]
                     }
                 ]
             },

avtomatika/blueprint.py

@@ -178,10 +178,13 @@ class StateMachineBlueprint:
     def render_graph(self, output_filename: Optional[str] = None, output_format: str = "png"):
         import ast
         import inspect
+        import logging
         import textwrap
 
         from graphviz import Digraph  # type: ignore[import]
 
+        logger = logging.getLogger(__name__)
+
         dot = Digraph(comment=f"State Machine for {self.name}")
         dot.attr("node", shape="box", style="rounded")
         all_handlers = list(self.handlers.items()) + [(ch.state, ch.func) for ch in self.conditional_handlers]
@@ -222,7 +225,11 @@ class StateMachineBlueprint:
                                                 value,
                                                 label=f"on {key}",
                                             )
-            except (TypeError, OSError):
+            except (TypeError, OSError) as e:
+                logger.warning(
+                    f"Could not parse handler '{handler_func.__name__}' for state '{handler_state}'. "
+                    f"Graph may be incomplete. Error: {e}"
+                )
                 pass
         for state in states:
             dot.node(state, state)

avtomatika/engine.py

@@ -584,6 +584,18 @@ class OrchestratorEngine:
         jobs = await self.storage.get_quarantined_jobs()
         return web.json_response(jobs)
 
+    async def _reload_worker_configs_handler(self, request: web.Request) -> web.Response:
+        """Handles the dynamic reloading of worker configurations."""
+        logger.info("Received request to reload worker configurations.")
+        if not self.config.WORKERS_CONFIG_PATH:
+            return web.json_response(
+                {"error": "WORKERS_CONFIG_PATH is not set, cannot reload configs."},
+                status=400,
+            )
+
+        await load_worker_configs_to_redis(self.storage, self.config.WORKERS_CONFIG_PATH)
+        return web.json_response({"status": "worker_configs_reloaded"})
+
     async def _flush_db_handler(self, request: web.Request) -> web.Response:
         logger.warning("Received request to flush the database.")
         await self.storage.flush_all()
@@ -643,6 +655,7 @@ class OrchestratorEngine:
             app.router.add_get("/workers", self._get_workers_handler)
             app.router.add_get("/jobs", self._get_jobs_handler)
             app.router.add_get("/dashboard", self._get_dashboard_handler)
+            app.router.add_post("/admin/reload-workers", self._reload_worker_configs_handler)
 
         if has_unversioned_routes:
             self.app.add_subapp("/api/", protected_app)

avtomatika/security.py

@@ -1,3 +1,4 @@
+from hashlib import sha256
 from typing import Any, Awaitable, Callable
 
 from aiohttp import web
@@ -89,9 +90,10 @@ def worker_auth_middleware_factory(
                 )
 
         # --- Individual Token Check ---
-        expected_token = await storage.get_worker_token(worker_id)
-        if expected_token:
-            if provided_token == expected_token:
+        expected_token_hash = await storage.get_worker_token(worker_id)
+        if expected_token_hash:
+            hashed_provided_token = sha256(provided_token.encode()).hexdigest()
+            if hashed_provided_token == expected_token_hash:
                 request["worker_id"] = worker_id  # Attach authenticated worker_id
                 return await handler(request)
             else:

avtomatika/worker_config_loader.py

@@ -1,3 +1,4 @@
+from hashlib import sha256
 from logging import getLogger
 from os.path import exists
 from tomllib import load
@@ -36,8 +37,10 @@ async def load_worker_configs_to_redis(storage: StorageBackend, config_path: str
             continue
 
         try:
+            # Hash the token before storing it
+            hashed_token = sha256(token.encode()).hexdigest()
             # Store the token in a way that's easily retrievable by worker_id
-            await storage.set_worker_token(worker_id, token)
+            await storage.set_worker_token(worker_id, hashed_token)
             logger.info(f"Loaded token for worker_id '{worker_id}'.")
         except Exception as e:
             logger.error(f"Failed to store token for worker_id '{worker_id}' in Redis: {e}")

{avtomatika-1.0b1.dist-info → avtomatika-1.0b2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: avtomatika
-Version: 1.0b1
+Version: 1.0b2
 Summary: A state-machine based orchestrator for long-running jobs.
 Project-URL: Homepage, https://github.com/avtomatika-ai/avtomatika
 Project-URL: Bug Tracker, https://github.com/avtomatika-ai/avtomatika/issues
@@ -251,6 +251,11 @@ async def handle_normal(actions):
     actions.transition_to("normal_processing")
 ```
 
+> **Note on Limitations:** The current version of `.when()` uses a simple parser with the following limitations:
+> *   **No Nested Attributes:** You can only access direct fields of `context.initial_data` or `context.state_history` (e.g., `context.initial_data.field`). Nested objects (e.g., `context.initial_data.area.field`) are not supported.
+> *   **Simple Comparisons Only:** Only the following operators are supported: `==`, `!=`, `>`, `<`, `>=`, `<=`. Complex logical expressions with `AND`, `OR`, or `NOT` are not allowed.
+> *   **Limited Value Types:** The parser only recognizes strings (in quotes), integers, and floats. Boolean values (`True`, `False`) and `None` are not correctly parsed and will be treated as strings.
+
 ### 2. Delegating Tasks to Workers (`dispatch_task`)
 
 This is the primary function for delegating work. The orchestrator will queue the task and wait for a worker to pick it up and return a result.
@@ -368,7 +373,9 @@ The orchestrator uses tokens to authenticate API requests.
 *   **Client Authentication**: All API clients must provide a token in the `X-Avtomatika-Token` header. The orchestrator validates this token against client configurations.
 *   **Worker Authentication**: Workers must provide a token in the `X-Worker-Token` header.
     *   `GLOBAL_WORKER_TOKEN`: You can set a global token for all workers using this environment variable. For development and testing, it defaults to `"secure-worker-token"`.
-    *   **Individual Tokens**: For production, it is recommended to define individual tokens for each worker in a separate configuration file and provide its path via the `WORKERS_CONFIG_PATH` environment variable.
+    *   **Individual Tokens**: For production, it is recommended to define individual tokens for each worker in a separate configuration file and provide its path via the `WORKERS_CONFIG_PATH` environment variable. Tokens from this file are stored in a hashed format for security.
+
+> **Note on Dynamic Reloading:** The worker configuration file can be reloaded without restarting the orchestrator by sending an authenticated `POST` request to the `/api/v1/admin/reload-workers` endpoint. This allows for dynamic updates of worker tokens.
 
 ### Observability
 

{avtomatika-1.0b1.dist-info → avtomatika-1.0b2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 avtomatika/__init__.py,sha256=BtYVoi9xy3yuMDMmzvLTO3AfbL2xeEjVz2bP9GhQeHU,675
-avtomatika/api.html,sha256=BrsaJDctd6c2gGKGO7-ZyvjyXW7tLdxD0lVhwYR0AvA,32796
-avtomatika/blueprint.py,sha256=zJySegRNnIA8IFR6mDKqNgcAYDERhw6N9bVUvggSUuc,9159
+avtomatika/api.html,sha256=Z-Ikqrle7YPXagx2D-C5ylVZicLQFSsIzPsHCQgqMHM,33628
+avtomatika/blueprint.py,sha256=RvDidn2xkziZOOnJlkV30SMp59GuscASz11GolxK7fE,9445
 avtomatika/client_config_loader.py,sha256=N41juqx9CF2XZ3-WRXspkLG2eoGhj9LkWdaMCCCk33E,1962
 avtomatika/compression.py,sha256=bhA1kw4YrCR3I3kdquZSY0fAzCrRrjtz55uepzLUDKI,2498
 avtomatika/config.py,sha256=0vlMfVMjxwVUC8m_NglGocC_EoklzAc0qmt3UJbxm10,2087
@@ -8,7 +8,7 @@ avtomatika/context.py,sha256=rnF09jqQGkaKlax8P5ku9USwijSm6dommDGZbeVrzLk,4295
 avtomatika/data_types.py,sha256=g-g5hPnCpzeATgOn5v7EvDm5ps314owFJD5iWJ6IPR0,1425
 avtomatika/datastore.py,sha256=ERMyiFYQpAhVYijxzTrrdm6jtIPFf4dngWIa0qod3Wc,551
 avtomatika/dispatcher.py,sha256=2YW-A0QXD4T6RF3KiNHflGs8LeLNrMlo5NScE4T49-o,10015
-avtomatika/engine.py,sha256=cHkUrNU-Oi6wyhyeZ56Z7XFL8-hb8PUXIXkxMuvJnug,36778
+avtomatika/engine.py,sha256=IeBy7pnWjERZwTIAp02wcaKggRMi5Jof-02kB2LfAKc,37482
 avtomatika/executor.py,sha256=-gbLfluZfwtLnAYI8GdK1xswA_DYp_-yoLnP4AWlcyQ,21157
 avtomatika/health_checker.py,sha256=WXwvRJ-3cZC2Udc_ogsyIQp7VzcvJjq_IaqzkTdE0TE,1265
 avtomatika/logging_config.py,sha256=e0-eEEGHw1zz9ZshzXaxfavV0uZfamRNdcAeHnrgBYQ,1370
@@ -17,10 +17,10 @@ avtomatika/py.typed,sha256=CT_L7gw2MLcQY-X0vs-xB5Vr0wzvGo7GuQYPI_qwJE8,65
 avtomatika/quota.py,sha256=DNcaL6k0J1REeP8sVqbY9FprY_3BSr2SxM2Vf4mEqdw,1612
 avtomatika/ratelimit.py,sha256=bZSmdCHviSGMVDNOKTBSswDl6P9Dc63BKLqKU50twpg,1579
 avtomatika/reputation.py,sha256=XFEqDW4TEqjrWgGB2KwnBBJF1RpwLlZPgFwxjPl2z8w,3221
-avtomatika/security.py,sha256=XFHlurSxsqyTJPIAp6R1RmE3htQ5Pq6wemsyyO8K-vw,4312
+avtomatika/security.py,sha256=afj28O3xB20EmA75DAQCQm_QKzx_tX2Qv9zE9TlcFvM,4441
 avtomatika/telemetry.py,sha256=FIbbYjNX0JhpfT7UHjx6EdcFUGF1DpQDrbicIp0ZGvA,2353
 avtomatika/watcher.py,sha256=djqwgK8wu73LL2Rtxa6mV-EXimFO7T-iRQV5U_26dc8,2571
-avtomatika/worker_config_loader.py,sha256=QNjC1_y4ZvykKUDW3rmMNnNPZD2dNvomRZtiuPs8DdQ,1645
+avtomatika/worker_config_loader.py,sha256=2Fu2gU3O5H9VJ62D_Yc4gT19vV_be8DJZWzrTTovq8E,1788
 avtomatika/ws_manager.py,sha256=X420oV9_vnujV78VWuOVNqiMZAMQzfw6_0Ep694LJjQ,3069
 avtomatika/history/base.py,sha256=p0zItsdxFzd889LujV8py6GwK4CUfqAt8QL915mrT4k,1680
 avtomatika/history/noop.py,sha256=Hk5yJsS4S5G5A7NRRMEafIV_IFI9hddSwEvRg2Reh0M,982
@@ -30,8 +30,8 @@ avtomatika/storage/__init__.py,sha256=TFb3_Ab6K1tlGaBihMqhzwdwCarMG2uRpATDW372YN
 avtomatika/storage/base.py,sha256=rcHbTB_0s4ZhkxIACnp9f6Ow7Jvr9pHx96t3VUf4lIc,9581
 avtomatika/storage/memory.py,sha256=Wqi879hw1cZBIhMDk_bKPaWhnH33QEl4aUIxu9g3M0M,10282
 avtomatika/storage/redis.py,sha256=eT65_tvRufy-txiKfSSs5UmKBeEio8aWG0UTBfj9DdY,17088
-avtomatika-1.0b1.dist-info/licenses/LICENSE,sha256=tqCjw9Y1vbU-hLcWi__7wQstLbt2T1XWPdbQYqCxuWY,1072
-avtomatika-1.0b1.dist-info/METADATA,sha256=YQWPU2WEdZ9XpWYInig-5FJNYZ7mW_2EgM8GzKguyn8,17026
-avtomatika-1.0b1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-avtomatika-1.0b1.dist-info/top_level.txt,sha256=gLDWhA_wxHj0I6fG5X8vw9fE0HSN4hTE2dEJzeVS2x8,11
-avtomatika-1.0b1.dist-info/RECORD,,
+avtomatika-1.0b2.dist-info/licenses/LICENSE,sha256=tqCjw9Y1vbU-hLcWi__7wQstLbt2T1XWPdbQYqCxuWY,1072
+avtomatika-1.0b2.dist-info/METADATA,sha256=IofXNi3-o-YDIfLKrBIT0mV5osHPvRyYX95b_9WI8dY,18083
+avtomatika-1.0b2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+avtomatika-1.0b2.dist-info/top_level.txt,sha256=gLDWhA_wxHj0I6fG5X8vw9fE0HSN4hTE2dEJzeVS2x8,11
+avtomatika-1.0b2.dist-info/RECORD,,