process-gpt-agent-sdk 0.2.6__tar.gz → 0.2.8__tar.gz

{process_gpt_agent_sdk-0.2.6/process_gpt_agent_sdk.egg-info → process_gpt_agent_sdk-0.2.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: process-gpt-agent-sdk
-Version: 0.2.6
+Version: 0.2.8
 Summary: Supabase 기반 이벤트/작업 폴링으로 A2A AgentExecutor를 실행하는 SDK
 License: MIT
 Project-URL: Homepage, https://github.com/your-org/process-gpt-agent-sdk

{process_gpt_agent_sdk-0.2.6 → process_gpt_agent_sdk-0.2.8}/function.sql

@@ -100,6 +100,110 @@ BEGIN
 END;
 $$ LANGUAGE plpgsql VOLATILE;
 
+DROP FUNCTION IF EXISTS public.fetch_pending_task_dev(text, text, integer, text);
+
+CREATE OR REPLACE FUNCTION public.fetch_pending_task_dev(
+  p_agent_orch text,
+  p_consumer   text,
+  p_limit      integer,
+  p_tenant_id  text
+)
+RETURNS TABLE (
+  id uuid,
+  user_id text,
+  proc_inst_id text,
+  proc_def_id text,
+  activity_id text,
+  activity_name text,
+  start_date timestamp without time zone,
+  end_date timestamp without time zone,
+  description text,
+  tool text,
+  due_date timestamp without time zone,
+  tenant_id text,
+  reference_ids text[],
+  adhoc boolean,
+  assignees jsonb,
+  duration integer,
+  output jsonb,
+  retry integer,
+  consumer text,
+  log text,
+  draft jsonb,
+  project_id uuid,
+  feedback jsonb,
+  updated_at timestamp with time zone,
+  username text,
+  status public.todo_status,
+  agent_mode public.agent_mode,
+  agent_orch public.agent_orch,
+  temp_feedback text,
+  draft_status public.draft_status,
+  task_type public.draft_status
+)
+AS $$
+BEGIN
+  RETURN QUERY
+    WITH cte AS (
+      SELECT
+        t.*,
+        t.draft_status AS task_type
+      FROM todolist AS t
+      WHERE t.status = 'IN_PROGRESS'
+        AND t.tenant_id = p_tenant_id
+        AND (p_agent_orch IS NULL OR p_agent_orch = '' OR t.agent_orch::text = p_agent_orch)
+        AND (
+          (t.agent_mode IN ('DRAFT','COMPLETE') AND t.draft IS NULL AND t.draft_status IS NULL)
+          OR t.draft_status = 'FB_REQUESTED'
+        )
+      ORDER BY t.start_date
+      LIMIT p_limit
+      FOR UPDATE SKIP LOCKED
+    ),
+    upd AS (
+      UPDATE todolist AS t
+         SET draft_status = 'STARTED',
+             consumer     = p_consumer
+        FROM cte
+       WHERE t.id = cte.id
+       RETURNING
+         t.id,
+         t.user_id,
+         t.proc_inst_id,
+         t.proc_def_id,
+         t.activity_id,
+         t.activity_name,
+         t.start_date,
+         t.end_date,
+         t.description,
+         t.tool,
+         t.due_date,
+         t.tenant_id,
+         t.reference_ids,
+         t.adhoc,
+         t.assignees,
+         t.duration,
+         t.output,
+         t.retry,
+         t.consumer,
+         t.log,
+         t.draft,
+         t.project_id,
+         t.feedback,
+         t.updated_at,
+         t.username,
+         t.status,
+         t.agent_mode,
+         t.agent_orch,
+         t.temp_feedback,
+         t.draft_status,
+         cte.task_type
+    )
+    SELECT * FROM upd;
+END;
+$$ LANGUAGE plpgsql VOLATILE;
+
+
 
 -- 2) 완료된 데이터(output/feedback) 조회
 DROP FUNCTION IF EXISTS public.fetch_done_data(text);
@@ -160,5 +264,6 @@ $$ LANGUAGE plpgsql VOLATILE;
 
 -- 익명(anon) 역할에 실행 권한 부여
 GRANT EXECUTE ON FUNCTION public.fetch_pending_task(text, text, integer) TO anon;
+GRANT EXECUTE ON FUNCTION public.fetch_pending_task_dev(text, text, integer, text) TO anon;
 GRANT EXECUTE ON FUNCTION public.fetch_done_data(text) TO anon;
 GRANT EXECUTE ON FUNCTION public.save_task_result(uuid, jsonb, boolean) TO anon;

{process_gpt_agent_sdk-0.2.6 → process_gpt_agent_sdk-0.2.8/process_gpt_agent_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: process-gpt-agent-sdk
-Version: 0.2.6
+Version: 0.2.8
 Summary: Supabase 기반 이벤트/작업 폴링으로 A2A AgentExecutor를 실행하는 SDK
 License: MIT
 Project-URL: Homepage, https://github.com/your-org/process-gpt-agent-sdk

{process_gpt_agent_sdk-0.2.6 → process_gpt_agent_sdk-0.2.8}/processgpt_agent_sdk/core/database.py

@@ -103,16 +103,29 @@ def get_consumer_id() -> str:
 async def polling_pending_todos(agent_orch: str, consumer: str) -> Optional[Dict[str, Any]]:
     """TODOLIST 테이블에서 대기중인 워크아이템을 조회"""
     def _call():
-        client = get_db_client()
-        return client.rpc(
-            "fetch_pending_task",
-            {"p_agent_orch": agent_orch, "p_consumer": consumer, "p_limit": 1},
-        ).execute()
+        supabase = get_db_client()
+        env = (os.getenv("ENV") or "").lower()
+
+        if env == "dev":
+            # 개발 환경: 특정 테넌트(uengine)만 폴링
+            resp = supabase.rpc(
+                "fetch_pending_task_dev",
+                {"p_agent_orch": agent_orch, "p_limit": 1, "p_consumer": consumer, "p_tenant_id": "uengine"},
+            ).execute()
+        else:
+            # 운영/기타 환경: 기존 로직 유지
+            resp = supabase.rpc(
+                "fetch_pending_task",
+                {"p_agent_orch": agent_orch, "p_limit": 1, "p_consumer": consumer},
+            ).execute()
+
+        rows = resp.data or []
+        return rows[0] if rows else None
 
     resp = await _async_retry(_call, name="polling_pending_todos", fallback=lambda: None)
-    if not resp or not resp.data:
+    if not resp:
         return None
-    return resp.data[0]
+    return resp
 
 
 async def fetch_todo_by_id(todo_id: str) -> Optional[Dict[str, Any]]:

{process_gpt_agent_sdk-0.2.6 → process_gpt_agent_sdk-0.2.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "process-gpt-agent-sdk"
-version = "0.2.6"
+version = "0.2.8"
 description = "Supabase 기반 이벤트/작업 폴링으로 A2A AgentExecutor를 실행하는 SDK"
 readme = "README.md"
 requires-python = ">=3.9"