flowtask 5.13.4__tar.gz → 5.13.5__tar.gz

{flowtask-5.13.4 → flowtask-5.13.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flowtask
-Version: 5.13.4
+Version: 5.13.5
 Summary: Framework for Task orchestration
 Author-email: Jesus Lara <jesuslarag@gmail.com>
 License: Apache-2.0

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/components/NetworkNinja/models/forms.py

@@ -7,6 +7,7 @@ from datamodel import BaseModel, Field
 from datamodel.parsers.json import json_encoder
 from querysource.conf import default_dsn, async_default_dsn
 from querysource.outputs.tables import PgOutput
+from ....conf import pg_server_settings
 from .abstract import AbstractPayload
 from .organization import Organization
 from .client import Client
@@ -465,7 +466,9 @@ class FormData(AbstractPayload):
 
         logger.debug(f"Processing form update: {previous_form} -> {new_form}")
 
-        async with await AsyncDB('pg', dsn=default_dsn).connection() as conn:
+        async with await AsyncDB(
+            'pg', dsn=default_dsn, server_settings=pg_server_settings('NetworkNinja')
+        ).connection() as conn:
             try:
                 # Use PL/pgSQL to handle the UPDATE exception properly
                 transaction = f"""

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/components/NetworkNinja/router.py

@@ -34,7 +34,8 @@ from ...interfaces.http import HTTPService
 from ...conf import (
     NETWORKNINJA_API_KEY,
     NETWORKNINJA_BASE_URL,
-    NETWORKNINJA_ENV
+    NETWORKNINJA_ENV,
+    pg_pool_options,
 )
 from .models import NetworkNinja_Map, NN_Order
 from .models.abstract import AbstractPayload
@@ -491,8 +492,13 @@ class NetworkNinja(HTTPService, FlowComponent):
         return final_form_ids
 
     async def close(self):
-        if self._pool:
-            await self._pool.close()
+        """Release the PostgreSQL pool (safe to call more than once)."""
+        pool, self._pool = self._pool, None
+        if pool:
+            try:
+                await pool.close()
+            except Exception as exc:  # pylint: disable=W0718
+                self._logger.warning(f"Error closing NetworkNinja pool: {exc!s}")
 
     def _evaluate_input(self):
         if self.previous or self.input is not None:
@@ -526,13 +532,31 @@ class NetworkNinja(HTTPService, FlowComponent):
         return True
 
     async def run(self):
-        """Run NetworkNinja Router."""
+        """Run NetworkNinja Router.
+
+        Opens a short-lived PostgreSQL pool for the duration of the action.
+        The task runner only calls ``close()`` when ``run()`` returns, so the
+        pool is released here as well whenever the action raises (for example
+        ``DataNotFound`` from ``get_batches`` on an empty NetworkNinja queue).
+        """
+        self._pool = AsyncPool(
+            'pg',
+            dsn=default_dsn,
+            **pg_pool_options(suffix='NetworkNinja'),
+        )
+        await self._pool.connect()
+        try:
+            return await self._run_action()
+        except BaseException:
+            await self.close()
+            raise
+
+    async def _run_action(self):
+        """Execute the configured NetworkNinja action and return its result."""
         tasks = []
         fn = getattr(self, self._action)
         self.add_metric("ACTION", self._action)
         self._result = {}
-        self._pool = AsyncPool('pg', dsn=default_dsn, max_clients=1000)
-        await self._pool.connect()
         if self._action == 'get_batch':
             try:
                 result = await fn()

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/components/OpenWithBase.py

@@ -19,7 +19,7 @@ from ..exceptions import FileNotFound, ComponentError
 from ..utils.mail import MailMessage
 from ..utils import check_empty
 from ..parsers.maps import open_map, open_model
-from ..conf import TASK_PATH, DEFAULT_ENCODING
+from ..conf import TASK_PATH, DEFAULT_ENCODING, pg_server_settings
 from ..interfaces.flow import FlowComponent
 from ..utils.constants import excel_based
 
@@ -224,7 +224,12 @@ class OpenWithBase(FlowComponent):
                         event_loop = asyncio.get_running_loop()
                     except RuntimeError:
                         event_loop = asyncio.get_event_loop()
-                    db = AsyncDB("pg", dsn=asyncpg_url, loop=event_loop)
+                    db = AsyncDB(
+                        "pg",
+                        dsn=asyncpg_url,
+                        loop=event_loop,
+                        server_settings=pg_server_settings(self.__class__.__name__),
+                    )
                     async with await db.connection() as conn:
                         result, error = await conn.query(discover)
                         if error:

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/components/PlanogramCompliance.py

@@ -8,10 +8,12 @@ from pathlib import Path
 from PIL import Image
 from asyncdb import AsyncDB
 from querysource.conf import default_dsn
+from ..conf import pg_server_settings
 import re
 from parrot.conf import PLANOGRAM_FOLDER
 from parrot.clients.google import GoogleGenAIClient, GoogleModel
 from parrot_pipelines.models import PlanogramConfig, EndcapGeometry
+from parrot_pipelines.planogram.plan import PlanogramCompliance as PlanogramPipeline
 from ..interfaces.flow import FlowComponent
 from ..interfaces.pipelines.parrot import AIPipeline
 from ..exceptions import ComponentError, ConfigError
@@ -54,6 +56,7 @@ class PlanogramCompliance(AIPipeline, FlowComponent):
     | id_column              | No  | DataFrame column used as per-row image id (default ``photo_id``)|
     | reference_images       | No  | Mapping key -> image file or directory under ``images/``;       |
     |                        |     | overrides the ``reference_images`` column of the row            |
+    |                        |     | (ignored for types without references, e.g. ``ink_wall``)       |
     | llm_backend            | No  | LLM backend as ``provider:model``; overrides the ``llm_backend``|
     |                        |     | column of the row (default: the ai-parrot package default)      |
     | llm_config             | No  | Arguments of an explicit Google GenAI client; when given it wins|
@@ -200,7 +203,7 @@ class PlanogramCompliance(AIPipeline, FlowComponent):
                 "planogram_config_table is not specified"
             )
         planogram_query = f"SELECT * FROM {self._planogram_config_table} WHERE config_name = $1"
-        db = AsyncDB('pg', dsn=default_dsn)
+        db = AsyncDB('pg', dsn=default_dsn, server_settings=pg_server_settings('PlanogramCompliance'))
         async with await db.connection() as conn:
             result = await conn.fetch_one(planogram_query, self._planogram_name)
         if not result:
@@ -255,19 +258,27 @@ class PlanogramCompliance(AIPipeline, FlowComponent):
             self.confidence_threshold = row['confidence_threshold']
         if row.get('detection_model'):
             self.detection_model = row['detection_model']
-        # Task reference images win; otherwise use the ones stored in the row.
-        references = reference_images or self._reference_images
-        if not references:
-            references = self._db_reference_images(
-                self._decode_json_column(row, 'reference_images')
+        # DB value takes precedence, the task value is the fallback.
+        planogram_type = row.get('planogram_type') or self._planogram_type
+        layout_config = self._decode_json_column(row, 'planogram_config') or {}
+        references: Dict[str, Union[Path, List[Path]]] = {}
+        if self._uses_reference_images(planogram_type, layout_config):
+            # Task reference images win; otherwise use the ones stored in the row.
+            references = reference_images or self._prepare_task_references()
+            if not references:
+                references = self._db_reference_images(
+                    self._decode_json_column(row, 'reference_images')
+                )
+        else:
+            self._logger.debug(
+                f"Planogram type '{planogram_type}' does not use reference images; skipping them"
             )
         try:
             planogram_config = PlanogramConfig(
                 planogram_id=row.get('planogram_id'),
                 config_name=self._planogram_name,
-                # DB value takes precedence, the task value is the fallback.
-                planogram_type=row.get('planogram_type') or self._planogram_type,
-                planogram_config=self._decode_json_column(row, 'planogram_config') or {},
+                planogram_type=planogram_type,
+                planogram_config=layout_config,
                 endcap_geometry=geometry,
                 roi_detection_prompt=row.get('roi_detection_prompt') or None,
                 object_identification_prompt=row.get('object_identification_prompt') or None,
@@ -284,6 +295,40 @@ class PlanogramCompliance(AIPipeline, FlowComponent):
         self._logger.debug('Planogram config loaded: %s', planogram_config)
         return planogram_config
 
+    def _uses_reference_images(self, planogram_type: str, layout_config: Dict[str, Any]) -> bool:
+        """Ask the pipeline whether this planogram type consumes reference images.
+
+        Args:
+            planogram_type: Resolved planogram type of the configuration.
+            layout_config: The ``planogram_config`` column (may override ``layout_profile.references``).
+
+        Returns:
+            False for types that identify products without references (e.g. ``ink_wall``); True
+            when the installed ai-parrot predates ``uses_reference_images``.
+
+        Raises:
+            ConfigError: Unknown planogram type or an invalid ``layout_profile`` override.
+        """
+        uses_references = getattr(PlanogramPipeline, 'uses_reference_images', None)
+        if uses_references is None:
+            return True
+        try:
+            return bool(uses_references(planogram_type, layout_config))
+        except ValueError as e:
+            raise ConfigError(
+                f"Invalid planogram configuration: {e}"
+            ) from e
+
+    def _prepare_task_references(self) -> Dict[str, Union[Path, List[Path]]]:
+        """Resolve the task ``reference_images`` once, only for types that use them.
+
+        Returns:
+            The cached mapping of catalogue key to a Path or a list of Paths.
+        """
+        if not self._reference_images and self.reference_images:
+            self._reference_images = self._prepare_reference_images()
+        return self._reference_images
+
     @staticmethod
     def _decode_json_column(row: Dict[str, Any], column: str) -> Any:
         """Return a JSONB column of the row as a Python object.
@@ -381,16 +426,12 @@ class PlanogramCompliance(AIPipeline, FlowComponent):
                 f"Image column '{self.image_column}' not found in DataFrame"
             )
 
-        # Prepare reference images list - convert to string paths as expected by pipeline
-        self._reference_images = self._prepare_reference_images()
-
         # Initialize LLM for pipeline
         self._llm = self._initialize_llm()
 
-        # Generate Planogram Configuration:
-        self.planogram_config = await self.get_planogram_config(
-            reference_images=self._reference_images
-        )
+        # Generate Planogram Configuration; reference images are resolved there,
+        # only when the planogram type uses them.
+        self.planogram_config = await self.get_planogram_config()
 
         self._logger.info(
             f"PlanogramPipeline started with {len(self.data)} rows to process"

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/components/ProductReportBot.py

@@ -17,6 +17,7 @@ from parrot.models.google import ConversationalScriptConfig, FictionalSpeaker
 from parrot.conf import STATIC_DIR
 from asyncdb import AsyncDB
 from querysource.conf import default_dsn
+from ..conf import pg_server_settings
 from parrot.models.responses import AgentResponse, AIMessage
 from parrot.models.basic import CompletionUsage
 from pathlib import Path
@@ -826,7 +827,7 @@ class ProductReportBot(ParrotAgentConfigMixin, ParrotBot, FlowComponent):
             )
 
         query = query_file.read_text()
-        db = AsyncDB('pg', dsn=default_dsn)
+        db = AsyncDB('pg', dsn=default_dsn, server_settings=pg_server_settings('ProductReportBot'))
         async with await db.connection() as conn:
             product_data, error = await conn.query(query, model)
             if error:
@@ -992,7 +993,7 @@ class ProductReportBot(ParrotAgentConfigMixin, ParrotBot, FlowComponent):
                 response_dict.pop(field, None)
             
            
-            db = AsyncDB('pg', dsn=default_dsn)
+            db = AsyncDB('pg', dsn=default_dsn, server_settings=pg_server_settings('ProductReportBot'))
             async with await db.connection() as conn:
                 ProductResponse.Meta.connection = conn
                 ProductResponse.Meta.schema = program_slug

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/components/tMap/tMap.py

@@ -9,6 +9,7 @@ from navconfig.logging import logging
 from asyncdb import AsyncDB
 from asyncdb.exceptions import NoDataFound
 from querysource.conf import asyncpg_url
+from ...conf import pg_server_settings
 from querysource.types.dt import transforms as qsdfunctions
 from ...exceptions import ComponentError
 from ...parsers.maps import open_map, open_model
@@ -142,7 +143,12 @@ class tMap(FlowComponent):
                         event_loop = asyncio.get_running_loop()
                     except RuntimeError:
                         event_loop = asyncio.get_event_loop()
-                    db = AsyncDB("pg", dsn=asyncpg_url, loop=event_loop)
+                    db = AsyncDB(
+                        "pg",
+                        dsn=asyncpg_url,
+                        loop=event_loop,
+                        server_settings=pg_server_settings("tMap"),
+                    )
                     async with await db.connection() as conn:
                         result, error = await conn.query(discover)
                         if error:

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/conf.py

@@ -47,6 +47,77 @@ default_pg = f"postgres://{DBUSER}:{DBPWD}@{DBHOST}:{DBPORT}/{DBNAME}"
 # sqlalchemy+asyncpg connector:
 default_sqlalchemy_pg = f"postgresql+asyncpg://{DBUSER}:{DBPWD}@{DBHOST}:{DBPORT}/{DBNAME}"
 
+# Application name reported to PostgreSQL (pg_stat_activity.application_name).
+# asyncdb falls back to the "NAV" name when nothing is passed, so every
+# connection/pool created by Flowtask must forward this value explicitly.
+APP_NAME = config.get("APP_NAME", fallback="Flowtask")
+
+# Defaults for short-lived PostgreSQL pools opened by components.
+# asyncdb's own defaults (min_size=10, 60-minute session/statement timeouts,
+# 100-hour inactive lifetime) are meant for long-running services, not for a
+# pool that lives only while one component runs.
+DB_POOL_MIN_SIZE = config.getint("DB_POOL_MIN_SIZE", fallback=2)
+DB_POOL_MAX_SIZE = config.getint("DB_POOL_MAX_SIZE", fallback=100)
+# idle_session_timeout / idle_in_transaction_session_timeout (PostgreSQL interval)
+DB_POOL_IDLE_TIMEOUT = config.get("DB_POOL_IDLE_TIMEOUT", fallback="5min")
+# statement_timeout (PostgreSQL interval)
+DB_POOL_STATEMENT_TIMEOUT = config.get("DB_POOL_STATEMENT_TIMEOUT", fallback="10min")
+# max_inactive_connection_lifetime, in seconds
+DB_POOL_MAX_INACTIVE = config.getint("DB_POOL_MAX_INACTIVE", fallback=300)
+
+
+def pg_application_name(suffix: str | None = None) -> str:
+    """Build the ``application_name`` reported to PostgreSQL.
+
+    Args:
+        suffix: Optional component/service suffix appended as ``APP_NAME.suffix``.
+
+    Returns:
+        The application name to put in ``server_settings``.
+    """
+    return f"{APP_NAME}.{suffix}" if suffix else APP_NAME
+
+
+def pg_server_settings(suffix: str | None = None, **extra: str) -> dict[str, str]:
+    """Build the ``server_settings`` dict for a short-lived PostgreSQL connection.
+
+    Args:
+        suffix: Optional suffix for the application name (see ``pg_application_name``).
+        **extra: Additional server settings; they override the defaults.
+
+    Returns:
+        A ``server_settings`` mapping ready to pass to ``AsyncDB``/``AsyncPool``.
+    """
+    settings: dict[str, str] = {
+        "application_name": pg_application_name(suffix),
+        "idle_session_timeout": DB_POOL_IDLE_TIMEOUT,
+        "idle_in_transaction_session_timeout": DB_POOL_IDLE_TIMEOUT,
+        "statement_timeout": DB_POOL_STATEMENT_TIMEOUT,
+    }
+    settings.update(extra)
+    return settings
+
+
+def pg_pool_options(suffix: str | None = None, **overrides: Any) -> dict[str, Any]:
+    """Build the keyword arguments for a short-lived ``AsyncPool('pg', ...)``.
+
+    Args:
+        suffix: Optional suffix for the application name (see ``pg_application_name``).
+        **overrides: Pool options (``min_size``, ``max_clients``,
+            ``max_inactive_timeout``, ``server_settings``) that override the defaults.
+
+    Returns:
+        Keyword arguments for ``AsyncPool``.
+    """
+    options: dict[str, Any] = {
+        "min_size": DB_POOL_MIN_SIZE,
+        "max_clients": DB_POOL_MAX_SIZE,
+        "max_inactive_timeout": DB_POOL_MAX_INACTIVE,
+        "server_settings": pg_server_settings(suffix),
+    }
+    options.update(overrides)
+    return options
+
 
 ### InfluxDB configuration:
 ## INFLUXDB

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/documentation/generated/components/PlanogramCompliance.doc.json

@@ -34,7 +34,7 @@
       },
       "reference_images": {
         "type": "string",
-        "description": "Mapping key -> image file or directory under ``images/``; overrides the ``reference_images`` column of the row"
+        "description": "Mapping key -> image file or directory under ``images/``; overrides the ``reference_images`` column of the row (ignored for types without references, e.g. ``ink_wall``)"
       },
       "llm_backend": {
         "type": "string",

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/documentation/generated/index.json

@@ -1,5 +1,5 @@
 {
-  "updated_at": "2026-10-07T23:19:44.163509",
+  "updated_at": "2026-10-09T16:44:22.374579",
   "components": {
     "ASPX": {
       "file": "components/ASPX.doc.json",

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/hooks/actions/zoom_event.py

@@ -8,7 +8,7 @@ from typing import ClassVar, Optional
 from asyncdb import AsyncPool
 from pydantic import BaseModel, Field
 
-from flowtask.conf import default_dsn
+from flowtask.conf import default_dsn, pg_application_name
 from .abstract import AbstractAction
 
 
@@ -40,7 +40,13 @@ class PersistZoomEvent(AbstractAction):
             self.__class__._pool_lock = asyncio.Lock()
         async with self.__class__._pool_lock:
             if self.__class__._pool is None:
-                pool = AsyncPool("pg", dsn=self.dsn or default_dsn, min_size=1, max_size=self.pool_size)
+                pool = AsyncPool(
+                    "pg",
+                    dsn=self.dsn or default_dsn,
+                    min_size=1,
+                    max_size=self.pool_size,
+                    server_settings={"application_name": pg_application_name("ZoomEvent")},
+                )
                 await pool.connect()
                 self.__class__._pool = pool
 

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/hooks/service.py

@@ -12,7 +12,7 @@ from navigator.types import WebApp
 from navigator.applications.base import BaseApplication
 from navigator.views import ModelView
 from ..exceptions import ConfigError, FlowTaskError
-from ..conf import TASK_STORAGES, default_dsn
+from ..conf import TASK_STORAGES, default_dsn, pg_server_settings
 from .types.base import BaseTrigger
 from .hook import Hook
 from .models import HookObject
@@ -285,7 +285,7 @@ class HookService:
     async def load_db_hooks(self, app: WebApp) -> None:
         """Load all Hooks saved into the Database."""
         try:
-            db = AsyncDB("pg", dsn=default_dsn)
+            db = AsyncDB("pg", dsn=default_dsn, server_settings=pg_server_settings("Hooks"))
             async with await db.connection() as conn:
                 HookObject.Meta.connection = conn
                 hooks = await HookObject.all()

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/interfaces/databases/db.py

@@ -4,6 +4,7 @@ import pandas
 from asyncdb import AsyncDB
 from navconfig.logging import logging
 from querysource.conf import asyncpg_url, default_dsn
+from ...conf import APP_NAME
 from ..credentials import CredentialsInterface
 from ...utils.functions import is_empty, as_boolean
 from ...exceptions import DataNotFound
@@ -14,7 +15,7 @@ class DBSupport(CredentialsInterface):
 
         Interface for adding AsyncbDB-based Database Support to Components.
     """
-    _service_name: str = 'Flowtask'
+    _service_name: str = APP_NAME
     _credentials = {
         "user": str,
         "password": str,

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/interfaces/db.py

@@ -52,7 +52,7 @@ from querysource.conf import (
 from asyncdb import AsyncDB
 from asyncdb.drivers.base import BaseDriver
 from asyncdb.exceptions import ProviderError
-from ..conf import default_dsn
+from ..conf import default_dsn, pg_application_name
 from ..exceptions import ComponentError
 from .credentials import CredentialsInterface
 
@@ -116,7 +116,7 @@ class DBInterface(CredentialsInterface):
     def pg_connection(self, event_loop: asyncio.AbstractEventLoop = None) -> BaseDriver:
         pgargs: dict = {
             "server_settings": {
-                "application_name": "Flowtask",
+                "application_name": pg_application_name(),
                 "client_min_messages": "notice",
                 "max_parallel_workers": "48",
                 "jit": "off",

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/interfaces/qs.py

@@ -33,6 +33,7 @@ from querysource.conf import (
 )
 import pandas as pd
 import numpy as np
+from ..conf import pg_application_name
 from ..exceptions import (
     ConfigError,
     ComponentError,
@@ -123,7 +124,7 @@ class QSSupport(ABC):
             args: dict = {
                 "min_size": 2,
                 "server_settings": {
-                    "application_name": "FlowTask:QS",
+                    "application_name": pg_application_name("QS"),
                     "client_min_messages": "notice",
                     "jit": "off",
                     "statement_timeout": f"{DB_STATEMENT_TIMEOUT}",
@@ -163,7 +164,7 @@ class QSSupport(ABC):
                 kwargs: dict = {
                     "min_size": 2,
                     "server_settings": {
-                        "application_name": "FlowTask:CopyToPg",
+                        "application_name": pg_application_name("CopyToPg"),
                         "client_min_messages": "notice",
                         "jit": "off",
                         "statement_timeout": f"{DB_STATEMENT_TIMEOUT}",

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/interfaces/workday/handlers/custom_report.py

@@ -503,6 +503,11 @@ class CustomReportType(WorkdayTypeBase):
                     "Set WORKDAY_REPORT_USERNAME and WORKDAY_REPORT_PASSWORD"
                 )
 
+            # Request timeout comes from the component's ``timeout`` attribute
+            # (WorkdayConfig.timeout, default 300s).
+            config = getattr(self.service, '_config', None)
+            timeout = int(getattr(config, 'timeout', None) or 120)
+
             # Create HTTP client with Basic Auth
             self._http_client = HTTPService(
                 credentials={
@@ -511,10 +516,13 @@ class CustomReportType(WorkdayTypeBase):
                 },
                 auth_type="basic",
                 accept=accept_mime,
-                timeout=120
+                timeout=timeout
             )
-            # Set logger for HTTP client
+            # HTTPService.__init__ ignores the ``timeout`` kwarg (hardcoded 30s),
+            # so set it explicitly on the instance.
+            self._http_client.timeout = timeout
             self._http_client._logger = self._logger
+            self._logger.debug(f"Custom report HTTP timeout: {timeout}s")
 
         # Execute the REST API request
         try:

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/models.py

@@ -14,6 +14,7 @@ from asyncdb import AsyncDB
 from asyncdb.exceptions import NoDataFound, ModelError
 from querysource.types.validators import Entity
 from querysource.conf import default_dsn
+from .conf import pg_server_settings
 
 
 class TaskState(Enum):
@@ -136,7 +137,7 @@ async def setTaskState(task, message, **kwargs):
         except RuntimeError:
             evt = asyncio.new_event_loop()
             _new = True
-        db = AsyncDB("pg", dsn=dsn, loop=evt)
+        db = AsyncDB("pg", dsn=dsn, loop=evt, server_settings=pg_server_settings("TaskModel"))
         try:
             async with await db.connection() as conn:  # pylint: disable=E1101
                 TaskModel.Meta.schema = schema
@@ -159,7 +160,7 @@ async def setTaskState(task, message, **kwargs):
                 return None
             # Another operation is in progress, create a new Connection:
             logging.error(err)
-            db = AsyncDB("pg", dsn=dsn, loop=evt)
+            db = AsyncDB("pg", dsn=dsn, loop=evt, server_settings=pg_server_settings("TaskModel"))
             async with await db.connection() as conn:  # pylint: disable=E1101
                 ## Running Update manually:
                 program = task.getProgram()

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/scheduler/functions.py

@@ -16,6 +16,7 @@ from querysource.types.validators import Entity
 from ..conf import (
     DEBUG,
     default_pg,
+    pg_application_name,
     SCHEDULER_WORKER_TIMEOUT,
     SCHEDULER_RETRY_ENQUEUE,
     SCHEDULER_MAX_RETRY_ENQUEUE,
@@ -102,7 +103,7 @@ class TaskScheduler:
         result = None
         options = {
             "server_settings": {
-                "application_name": "Flowtask.Scheduler",
+                "application_name": pg_application_name("Scheduler"),
                 "client_min_messages": "notice",
             }
         }

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/scheduler/scheduler.py

@@ -75,6 +75,7 @@ from ..conf import (
     WORKER_HIGH_LIST,
     WORKER_LIST,
     default_dsn,
+    pg_application_name,
     USE_TIMEZONE,
     PUBSUB_REDIS,
     ERROR_CHANNEL,
@@ -875,7 +876,7 @@ class NavScheduler:
         result = None
         options = {
             "server_settings": {
-                "application_name": "Flowtask.Scheduler",
+                "application_name": pg_application_name("Scheduler"),
                 "client_min_messages": "notice",
                 "jit": "on",
             },

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/services/files/service.py

@@ -28,7 +28,7 @@ from ...exceptions import (
 )
 # tasks
 from ...services.tasks import launch_task
-from ...conf import FILES_PATH
+from ...conf import FILES_PATH, pg_server_settings
 from .model import FileModel
 
 
@@ -484,7 +484,7 @@ class FileService(BaseView):
                     'X-MESSAGE': f'File Slug doesnt exists or is disabled {file_slug}',
                     'X-FILENAME': f.filename
                 }
-                db = AsyncDB('pg', dsn=asyncpg_url)
+                db = AsyncDB('pg', dsn=asyncpg_url, server_settings=pg_server_settings('Files'))
                 async with await db.connection() as conn:
                     FileModel.Meta.connection = conn
                     # getting from model:

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/tasks/abstract.py

@@ -22,7 +22,7 @@ from navconfig import DEBUG, config
 from navconfig.logging import logging
 
 # Flowtask Dependencies
-from ..conf import FILE_STORAGES, SYSTEM_LOCALE, TASK_STORAGES, default_dsn
+from ..conf import FILE_STORAGES, SYSTEM_LOCALE, TASK_STORAGES, default_dsn, pg_server_settings
 from ..events import LogError, LogEvent
 from ..events.manager import EventManager
 from ..exceptions import TaskError, TaskParseError
@@ -361,7 +361,7 @@ class AbstractTask(ABC):
 
     async def get_task(self):
         try:
-            db = AsyncDB("pg", dsn=default_dsn, loop=self._loop)
+            db = AsyncDB("pg", dsn=default_dsn, loop=self._loop, server_settings=pg_server_settings("Task"))
             async with await db.connection() as conn:
                 # first, check if a Tasks table exists on tenant:
                 sql = f"""SELECT EXISTS (

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask/version.py

@@ -5,7 +5,7 @@ __description__ = (
     "Framework for running Tasks and from CLI and API for orchestation. "
     "Component-based Task builder/Runner for non-programmers."
 )
-__version__ = "5.13.4"
+__version__ = "5.13.5"
 __copyright__ = "Copyright (c) 2018-2025 Jesus Lara"
 __author__ = "Jesus Lara"
 __author_email__ = "jesuslarag@gmail.com"

{flowtask-5.13.4 → flowtask-5.13.5}/flowtask.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flowtask
-Version: 5.13.4
+Version: 5.13.5
 Summary: Framework for Task orchestration
 Author-email: Jesus Lara <jesuslarag@gmail.com>
 License: Apache-2.0