process-gpt-agent-sdk 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

processgpt_agent_sdk/utils/crewai_event_listener.py

@@ -9,7 +9,7 @@ from typing import Any, Optional, Dict, List
 from crewai.utilities.events import CrewAIEventsBus, ToolUsageStartedEvent, ToolUsageFinishedEvent
 from crewai.utilities.events.task_events import TaskStartedEvent, TaskCompletedEvent
 
-from .logger import handle_error as _err, log as _log
+from .logger import handle_application_error, write_log_message
 from .context_manager import todo_id_var, proc_id_var, crew_type_var, form_id_var, form_key_var
 from ..core.database import initialize_db, get_db_client
 
@@ -23,7 +23,7 @@ class CrewAIEventLogger:
     def __init__(self):
         initialize_db()
         self.supabase = get_db_client()
-        _log("CrewAIEventLogger 초기화 완료")
+        write_log_message("CrewAIEventLogger 초기화 완료")
 
     # =============================================================================
     # Job ID Generation
@@ -150,12 +150,12 @@ class CrewAIEventLogger:
                 return
             except Exception as e:
                 if attempt < 3:
-                    _err("이벤트저장오류(재시도)", e, raise_error=False)
+                    handle_application_error("이벤트저장오류(재시도)", e, raise_error=False)
                     # 지수 백오프: 0.3s, 0.6s
                     import time
                     time.sleep(0.3 * attempt)
                     continue
-                _err("이벤트저장오류(최종)", e, raise_error=False)
+                handle_application_error("이벤트저장오류(최종)", e, raise_error=False)
                 return
 
     # =============================================================================
@@ -173,9 +173,9 @@ class CrewAIEventLogger:
             crew_type = crew_type_var.get() or "action"
             rec = self._create_event_record(etype, data, job_id, crew_type, todo_id_var.get(), proc_id_var.get())
             self._save_event(rec)
-            _log(f"[{etype}] [{job_id[:8]}] 저장 완료")
+            write_log_message(f"[{etype}] [{job_id[:8]}] 저장 완료")
         except Exception as e:
-            _err("이벤트처리오류", e, raise_error=False)
+            handle_application_error("이벤트처리오류", e, raise_error=False)
 
 
 
@@ -196,6 +196,6 @@ class CrewConfigManager:
             for evt in (TaskStartedEvent, TaskCompletedEvent, ToolUsageStartedEvent, ToolUsageFinishedEvent):
                 bus.on(evt)(lambda source, event, logger=self.logger: logger.on_event(event, source))
             self._registered_by_pid.add(pid)
-            _log("CrewAI event listeners 등록 완료")
+            write_log_message("CrewAI event listeners 등록 완료")
         except Exception as e:
-            _err("CrewAI 이벤트 버스 등록 실패", e, raise_error=False)
+            handle_application_error("CrewAI 이벤트 버스 등록 실패", e, raise_error=False)

processgpt_agent_sdk/utils/event_handler.py

@@ -6,43 +6,61 @@ from datetime import datetime, timezone
 from typing import Any, Dict
 
 from a2a.server.events import Event
-from .logger import handle_error as _emit_error, log as _emit_log
+from .logger import handle_application_error, write_log_message
 from ..core.database import record_event, save_task_result
+from ..tools.safe_tool_loader import SafeToolLoader
 
 
-def _event_to_dict(event: Event) -> Dict[str, Any]:
+def convert_event_to_dictionary(event: Event) -> Dict[str, Any]:
 	try:
 		if hasattr(event, "__dict__"):
 			return {k: v for k, v in event.__dict__.items() if not k.startswith("_")}
 		return {"event": str(event)}
 	except Exception as e:
-		_emit_error("event dict 변환 실패", e, raise_error=False)
+		handle_application_error("event dict 변환 실패", e, raise_error=False)
 		return {"event": str(event)}
 
 
-async def route_event(todo: Dict[str, Any], event: Event) -> None:
+async def process_event_message(todo: Dict[str, Any], event: Event) -> None:
 	"""이벤트(dict)와 출력(output)을 구분해 처리.
 
-	- "event": CrewAI 등에서 발생한 실행 이벤트 → events 테이블 저장
-	- "output": 실행 결과 → save_task_result를 통해 중간/최종 여부에 따라 저장
+	- "event": 일반 이벤트 → events 테이블 저장
+	- "output": 실행 결과 → save_task_result로 중간/최종 여부에 따라 저장
+	- "done": 실행 종료 신호 → 이벤트 기록 후 MCP 리소스 정리
 	"""
 	try:
-		data = _event_to_dict(event)
+		data = convert_event_to_dictionary(event)
+		evt_type = str(data.get("type") or data.get("event_type") or "").lower()
 
-		# output 이벤트 처리
-		if data.get("type") == "output" or data.get("event_type") == "output":
+		# done: 종료 이벤트 → 기록 후 MCP 정리
+		if evt_type == "done":
+			normalized = {
+				"id": str(uuid.uuid4()),
+				"timestamp": datetime.now(timezone.utc).isoformat(),
+				**data,
+			}
+			await record_event(todo, normalized, event_type="done")
+			try:
+				SafeToolLoader.shutdown_all_adapters()
+				write_log_message("MCP 리소스 정리 완료")
+			except Exception as ce:
+				handle_application_error("MCP 리소스 정리 실패", ce, raise_error=False)
+			return
+
+		# output: 결과 저장만 수행
+		if evt_type == "output":
 			payload = data.get("data") or data.get("payload") or {}
 			is_final = bool(payload.get("final") or payload.get("is_final"))
 			content = payload.get("content") if isinstance(payload, dict) else payload
 			await save_task_result(str(todo.get("id")), content, final=is_final)
 			return
 
-		# 일반 event 처리 (원형 + 타임스탬프)
+		# event: 일반 이벤트 저장
 		normalized = {
 			"id": str(uuid.uuid4()),
 			"timestamp": datetime.now(timezone.utc).isoformat(),
 			**data,
 		}
-		await record_event(todo, normalized, event_type=str(data.get("type") or data.get("event_type") or "event"))
+		await record_event(todo, normalized, event_type="event")
 	except Exception as e:
-		_emit_error("route_event 처리 실패", e, raise_error=False)
+		handle_application_error("process_event_message 처리 실패", e, raise_error=False)

processgpt_agent_sdk/utils/logger.py

@@ -11,20 +11,20 @@ if not logging.getLogger().handlers:
 		format="%(asctime)s %(levelname)s %(name)s - %(message)s",
 	)
 
-_logger = logging.getLogger("process-gpt-agent-framework")
+APPLICATION_LOGGER = logging.getLogger("process-gpt-agent-framework")
 
 
-def log(message: str, level: int = logging.INFO) -> None:
+def write_log_message(message: str, level: int = logging.INFO) -> None:
 	spaced = os.getenv("LOG_SPACED", "1") != "0"
 	suffix = "\n" if spaced else ""
-	_logger.log(level, f"{message}{suffix}")
+	APPLICATION_LOGGER.log(level, f"{message}{suffix}")
 
 
-def handle_error(title: str, error: Exception, *, raise_error: bool = True, extra: Optional[Dict] = None) -> None:
+def handle_application_error(title: str, error: Exception, *, raise_error: bool = True, extra: Optional[Dict] = None) -> None:
 	spaced = os.getenv("LOG_SPACED", "1") != "0"
 	suffix = "\n" if spaced else ""
 	context = f" | extra={extra}" if extra else ""
-	_logger.error(f"{title}: {error}{context}{suffix}")
-	_logger.error(traceback.format_exc())
+	APPLICATION_LOGGER.error(f"{title}: {error}{context}{suffix}")
+	APPLICATION_LOGGER.error(traceback.format_exc())
 	if raise_error:
 		raise error

process_gpt_agent_sdk-0.1.4.dist-info/RECORD

@@ -1,17 +0,0 @@
-processgpt_agent_sdk/__init__.py,sha256=IvAL5WBZhI83LYQogRP6-i04bxZkhmkgmES4FRQY888,185
-processgpt_agent_sdk/server.py,sha256=cMta6gk0NVxJSN778skGaT0cCTlpK7UKiMzER2UTRcU,8115
-processgpt_agent_sdk/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-processgpt_agent_sdk/core/database.py,sha256=a87Y1DLxMIm3Yu_-6LWWANnZKJgU265Nn7cKjOB1s8Y,13667
-processgpt_agent_sdk/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-processgpt_agent_sdk/tools/knowledge_tools.py,sha256=-IPQB6A59wMTBsZUDM-7q10IHMz30X-m_vp6RgB6yHA,8995
-processgpt_agent_sdk/tools/safe_tool_loader.py,sha256=a2Twnldu4VliHk41OY03lxqM91cK6-MRmjXuSHbcKqk,5647
-processgpt_agent_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-processgpt_agent_sdk/utils/context_manager.py,sha256=Vqx3PUSNR_V3t7CbK6_W0xU5eb2IJPpn-YFLlfncIL4,1174
-processgpt_agent_sdk/utils/crewai_event_listener.py,sha256=-YIJCvrTCsS1Tz0JBqgXRcj2ziv3Pqkak_iez-pQUBI,8686
-processgpt_agent_sdk/utils/event_handler.py,sha256=Kga4uc4qSTywkFA0dJy699wcbogVkjbx90-psLXNnv0,1718
-processgpt_agent_sdk/utils/logger.py,sha256=JmA5_A1DO3oRqZpiT4ZpNrPjYGWRAR38V-aBNr4-R1c,878
-processgpt_agent_sdk/utils/summarizer.py,sha256=XV7e4pyKEK7NiGK_PAc1xorKSO2MKlSHCAiV7slOdus,4527
-process_gpt_agent_sdk-0.1.4.dist-info/METADATA,sha256=9gqChy6DknuWhMSliDzN1_n5usQFKGpUzZ0a8ezOOP8,12935
-process_gpt_agent_sdk-0.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-process_gpt_agent_sdk-0.1.4.dist-info/top_level.txt,sha256=Xe6zrj3_3Vv7d0pl5RRtenVUckwOVBVLQn2P03j5REo,21
-process_gpt_agent_sdk-0.1.4.dist-info/RECORD,,