process-gpt-agent-sdk 0.2.11__py3-none-any.whl → 0.3.4__py3-none-any.whl

processgpt_agent_sdk/utils/crewai_event_listener.py

@@ -1,205 +0,0 @@
-from __future__ import annotations
-
-import os
-import json
-import uuid
-from datetime import datetime, timezone
-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_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
-
-
-class CrewAIEventLogger:
-    """CrewAI 이벤트 로거 - Supabase 전용"""
-
-    # =============================================================================
-    # Initialization
-    # =============================================================================
-    def __init__(self):
-        """Supabase 클라이언트를 초기화한다."""
-        initialize_db()
-        self.supabase = get_db_client()
-        write_log_message("CrewAIEventLogger 초기화 완료")
-
-    # =============================================================================
-    # Job ID Generation
-    # =============================================================================
-    def _generate_job_id(self, event_obj: Any, source: Any = None) -> str:
-        """이벤트 객체에서 Job ID 생성"""
-        try:
-            if hasattr(event_obj, "task") and hasattr(event_obj.task, "id"):
-                return str(event_obj.task.id)
-            if source and hasattr(source, "task") and hasattr(source.task, "id"):
-                return str(source.task.id)
-        except Exception:
-            pass
-        return "unknown"
-
-    # =============================================================================
-    # Record Creation
-    # =============================================================================
-    def _create_event_record(
-        self,
-        event_type: str,
-        data: Dict[str, Any],
-        job_id: str,
-        crew_type: str,
-        todo_id: Optional[str],
-        proc_inst_id: Optional[str],
-    ) -> Dict[str, Any]:
-        """이벤트 레코드 생성"""
-        return {
-            "id": str(uuid.uuid4()),
-            "job_id": job_id,
-            "todo_id": todo_id,
-            "proc_inst_id": proc_inst_id,
-            "event_type": event_type,
-            "crew_type": crew_type,
-            "data": data,
-            "timestamp": datetime.now(timezone.utc).isoformat(),
-        }
-
-    # =============================================================================
-    # Parsing Helpers
-    # =============================================================================
-    def _parse_json_text(self, text: str) -> Any:
-        """JSON 문자열을 객체로 파싱하거나 원본 반환"""
-        try:
-            return json.loads(text)
-        except:
-            return text
-
-    def _parse_output(self, output: Any) -> Any:
-        """output 또는 raw 텍스트를 파싱해 반환"""
-        if not output:
-            return ""
-        text = getattr(output, "raw", None) or (output if isinstance(output, str) else "")
-        return self._parse_json_text(text)
-
-    def _parse_tool_args(self, args_text: str) -> Optional[str]:
-        """tool_args에서 query 키 추출"""
-        try:
-            args = json.loads(args_text or "{}")
-            return args.get("query")
-        except Exception:
-            return None
-
-    # =============================================================================
-    # Formatting Helpers
-    # =============================================================================
-    def _format_plans_md(self, plans: List[Dict[str, Any]]) -> str:
-        """list_of_plans_per_task 형식을 Markdown 문자열로 변환"""
-        lines: List[str] = []
-        for idx, item in enumerate(plans or [], 1):
-            task = item.get("task", "")
-            plan = item.get("plan", "")
-            lines.append(f"## {idx}. {task}")
-            lines.append("")
-            if isinstance(plan, list):
-                for line in plan:
-                    lines.append(str(line))
-            elif isinstance(plan, str):
-                for line in plan.split("\n"):
-                    lines.append(line)
-            else:
-                lines.append(str(plan))
-            lines.append("")
-        return "\n".join(lines).strip()
-
-    # =============================================================================
-    # Data Extraction
-    # =============================================================================
-    def _extract_event_data(self, event_obj: Any, source: Any = None) -> Dict[str, Any]:
-        """이벤트 타입별 데이터 추출"""
-        etype = getattr(event_obj, "type", None) or type(event_obj).__name__
-        if etype == "task_started":
-            agent = getattr(getattr(event_obj, "task", None), "agent", None)
-            return {
-                "role": getattr(agent, "role", "Unknown"),
-                "goal": getattr(agent, "goal", "Unknown"),
-                "agent_profile": getattr(agent, "profile", None) or "/images/chat-icon.png",
-                "name": getattr(agent, "name", "Unknown"),
-            }
-        if etype == "task_completed":
-            result = self._parse_output(getattr(event_obj, "output", None))
-            if isinstance(result, dict) and "list_of_plans_per_task" in result:
-                md = self._format_plans_md(result.get("list_of_plans_per_task") or [])
-                return {"plans": md}
-            return {"result": result}
-
-        if etype in ("tool_usage_started", "tool_usage_finished") or str(etype).startswith("tool_"):
-            return {
-                "tool_name": getattr(event_obj, "tool_name", None),
-                "query": self._parse_tool_args(getattr(event_obj, "tool_args", "")),
-            }
-        return {"info": f"Event type: {etype}"}
-
-    # =============================================================================
-    # Event Saving
-    # =============================================================================
-    def _save_event(self, record: Dict[str, Any]) -> None:
-        """Supabase에 이벤트 레코드 저장 (간단 재시도 포함)"""
-        payload = json.loads(json.dumps(record, default=str))
-        for attempt in range(1, 4):
-            try:
-                self.supabase.table("events").insert(payload).execute()
-                return
-            except Exception as e:
-                if attempt < 3:
-                    handle_application_error("이벤트저장오류(재시도)", e, raise_error=False)
-                    import time
-                    time.sleep(0.3 * attempt)
-                    continue
-                handle_application_error("이벤트저장오류(최종)", e, raise_error=False)
-                return
-
-    # =============================================================================
-    # Event Handling
-    # =============================================================================
-    def on_event(self, event_obj: Any, source: Any = None) -> None:
-        """이벤트 수신부터 DB 저장까지 처리"""
-        etype = getattr(event_obj, "type", None) or type(event_obj).__name__
-        ALLOWED = {"task_started", "task_completed", "tool_usage_started", "tool_usage_finished"}
-        if etype not in ALLOWED:
-            return
-        try:
-            job_id = self._generate_job_id(event_obj, source)
-            data = self._extract_event_data(event_obj, source)
-            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)
-            write_log_message(f"[{etype}] [{job_id[:8]}] 저장 완료")
-        except Exception as e:
-            handle_application_error("이벤트처리오류", e, raise_error=False)
-
-
-# =============================================================================
-# CrewConfigManager
-# 설명: 이벤트 리스너를 프로세스 단위로 1회 등록
-# =============================================================================
-class CrewConfigManager:
-    """글로벌 CrewAI 이벤트 리스너 등록 매니저"""
-    _registered_by_pid: set[int] = set()
-
-    def __init__(self) -> None:
-        self.logger = CrewAIEventLogger()
-        self._register_once_per_process()
-
-    def _register_once_per_process(self) -> None:
-        """현재 프로세스에만 한 번 이벤트 리스너를 등록한다."""
-        try:
-            pid = os.getpid()
-            if pid in self._registered_by_pid:
-                return
-            bus = CrewAIEventsBus()
-            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)
-            write_log_message("CrewAI event listeners 등록 완료")
-        except Exception as e:
-            handle_application_error("CrewAI 이벤트 버스 등록 실패", e, raise_error=False)

processgpt_agent_sdk/utils/event_handler.py

@@ -1,72 +0,0 @@
-from __future__ import annotations
-
-from typing import Any, Dict
-import uuid
-
-from a2a.server.events import Event
-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
-
-
-# =============================================================================
-# 이벤트 변환: Event 또는 dict를 표준 dict로 통일
-# =============================================================================
-
-def convert_event_to_dictionary(event: Event) -> Dict[str, Any]:
-	"""Event/dict를 표준 dict로 변환한다."""
-	try:
-		# 이미 dict로 전달된 경우 그대로 사용
-		if isinstance(event, dict):
-			return event
-		# Event 객체면 공개 필드만 추출
-		if hasattr(event, "__dict__"):
-			return {k: v for k, v in event.__dict__.items() if not k.startswith("_")}
-		# 알 수 없는 타입은 문자열로 보존
-		return {"type": "event", "data": str(event)}
-	except Exception as e:
-		handle_application_error("event dict 변환 실패", e, raise_error=False)
-		return {"type": "event", "data": str(event)}
-
-
-# =============================================================================
-# 이벤트 처리: type에 따라 저장 위치 분기
-# =============================================================================
-
-async def process_event_message(todo: Dict[str, Any], event: Event) -> None:
-	"""이벤트 타입별로 todolist/events에 저장하거나 리소스 정리."""
-	try:
-		data = convert_event_to_dictionary(event)
-		evt_type = str(data.get("type") or data.get("event_type") or "").lower()
-
-		# done: 종료 이벤트 → 기록 후 MCP 정리
-		if evt_type == "done":
-			payload = data.get("data") or {}
-			if isinstance(payload, dict) and "id" not in payload:
-				payload["id"] = str(uuid.uuid4())
-			await record_event(payload)
-			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 {}
-			is_final = bool(payload.get("final") or payload.get("is_final")) if isinstance(payload, dict) else False
-			content = payload.get("content") or payload.get("data") if isinstance(payload, dict) else payload
-			await save_task_result(str(todo.get("id")), content, final=is_final)
-			return
-		
-		# event : 일반 이벤트 저장 (워커 데이터 그대로 보존)
-		if evt_type == "event":
-			payload = data.get("data") or {}
-			if isinstance(payload, dict) and "id" not in payload:
-				payload["id"] = str(uuid.uuid4())
-			await record_event(payload)
-			return
-
-	except Exception as e:
-		handle_application_error("process_event_message 처리 실패", e, raise_error=False)

processgpt_agent_sdk/utils/logger.py

@@ -1,97 +0,0 @@
-import logging
-import os
-import traceback
-from typing import Optional, Dict
-
-
-# Configure root logger only once (idempotent)
-if not logging.getLogger().handlers:
-	# LOG_LEVEL 환경변수 읽기
-	log_level = os.getenv("LOG_LEVEL", "INFO").upper()
-	if log_level == "DEBUG":
-		level = logging.DEBUG
-	elif log_level == "INFO":
-		level = logging.INFO
-	elif log_level == "WARNING":
-		level = logging.WARNING
-	elif log_level == "ERROR":
-		level = logging.ERROR
-	else:
-		level = logging.INFO
-	
-	logging.basicConfig(
-		level=level,
-		format="%(asctime)s %(levelname)s %(name)s - %(message)s",
-	)
-
-LOGGER_NAME = os.getenv("LOGGER_NAME") or "processgpt"
-APPLICATION_LOGGER = logging.getLogger(LOGGER_NAME)
-
-# Application logger도 같은 레벨로 설정
-log_level = os.getenv("LOG_LEVEL", "INFO").upper()
-if log_level == "DEBUG":
-	APPLICATION_LOGGER.setLevel(logging.DEBUG)
-elif log_level == "INFO":
-	APPLICATION_LOGGER.setLevel(logging.INFO)
-elif log_level == "WARNING":
-	APPLICATION_LOGGER.setLevel(logging.WARNING)
-elif log_level == "ERROR":
-	APPLICATION_LOGGER.setLevel(logging.ERROR)
-
-# 디버그 레벨 상수 정의
-DEBUG_LEVEL_NONE = 0      # 디버그 로그 없음
-DEBUG_LEVEL_BASIC = 1     # 기본 디버그 로그 (INFO 레벨)
-DEBUG_LEVEL_DETAILED = 2  # 상세 디버그 로그 (DEBUG 레벨)
-DEBUG_LEVEL_VERBOSE = 3   # 매우 상세한 디버그 로그 (DEBUG 레벨 + 추가 정보)
-
-# 환경변수에서 디버그 레벨 읽기
-DEBUG_LEVEL = int(os.getenv("DEBUG_LEVEL", "1"))
-
-
-def set_application_logger_name(name: str) -> None:
-	"""애플리케이션 로거 이름을 런타임에 변경한다."""
-	global APPLICATION_LOGGER
-	APPLICATION_LOGGER = logging.getLogger(name or "processgpt")
-
-
-def write_log_message(message: str, level: int = logging.INFO, debug_level: int = DEBUG_LEVEL_BASIC) -> None:
-	"""로그 메시지를 쓴다. 디버그 레벨에 따라 출력 여부를 결정한다."""
-	# 디버그 레벨 체크
-	if debug_level > DEBUG_LEVEL:
-		return
-	
-	# DEBUG 레벨 로그의 경우 로거 레벨도 확인
-	if level == logging.DEBUG and DEBUG_LEVEL < DEBUG_LEVEL_DETAILED:
-		return
-	
-	spaced = os.getenv("LOG_SPACED", "1") != "0"
-	suffix = "\n" if spaced else ""
-	APPLICATION_LOGGER.log(level, f"{message}{suffix}")
-
-
-def write_debug_message(message: str, debug_level: int = DEBUG_LEVEL_BASIC) -> None:
-	"""디버그 전용 로그 메시지를 쓴다."""
-	write_log_message(message, logging.DEBUG, debug_level)
-
-
-def write_info_message(message: str, debug_level: int = DEBUG_LEVEL_BASIC) -> None:
-	"""정보 로그 메시지를 쓴다."""
-	write_log_message(message, logging.INFO, debug_level)
-
-
-def set_debug_level(level: int) -> None:
-	"""런타임에 디버그 레벨을 설정한다."""
-	global DEBUG_LEVEL
-	DEBUG_LEVEL = level
-	write_info_message(f"디버그 레벨이 {level}로 설정되었습니다.", DEBUG_LEVEL_BASIC)
-
-
-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 ""
-	APPLICATION_LOGGER.error(f"{title}: {error}{context}{suffix}")
-	APPLICATION_LOGGER.error(traceback.format_exc())
-	if raise_error:
-		raise error

processgpt_agent_sdk/utils/summarizer.py

@@ -1,146 +0,0 @@
-from __future__ import annotations
-
-
-import os
-import json
-import asyncio
-from typing import Any, Tuple
-
-import openai
-from .logger import handle_application_error, write_log_message
-
-# =============================================================================
-# 요약기(Summarizer)
-# 설명: 출력/피드백/현재 내용으로부터 OpenAI를 사용해 간단 요약을 생성한다.
-# =============================================================================
-
-async def summarize_async(outputs: Any, feedbacks: Any, contents: Any = None) -> Tuple[str, str]:
-	"""(output_summary, feedback_summary)를 비동기로 생성해 반환한다.
-	키 없음/오류 시 빈 문자열 폴백, 취소는 상위로 전파."""
-	outputs_str = _convert_to_string(outputs).strip()
-	feedbacks_str = _convert_to_string(feedbacks).strip()
-	contents_str = _convert_to_string(contents).strip()
-
-	output_summary = ""
-	feedback_summary = ""
-
-	if outputs_str and outputs_str not in ("[]", "{}", "[{}]"):
-		write_log_message("요약 호출(이전결과물)")
-		output_prompt = _create_output_summary_prompt(outputs_str)
-		output_summary = await _call_openai_api_async(output_prompt, task_name="output")
-
-	if feedbacks_str and feedbacks_str not in ("[]", "{}"):
-		write_log_message("요약 호출(피드백)")
-		feedback_prompt = _create_feedback_summary_prompt(feedbacks_str, contents_str)
-		feedback_summary = await _call_openai_api_async(feedback_prompt, task_name="feedback")
-
-	return output_summary or "", feedback_summary or ""
-
-
-# =============================================================================
-# 헬퍼: 문자열 변환
-# =============================================================================
-
-def _convert_to_string(data: Any) -> str:
-	"""임의 데이터를 안전하게 문자열로 변환한다."""
-	if data is None:
-		return ""
-	if isinstance(data, str):
-		return data
-	try:
-		return json.dumps(data, ensure_ascii=False)
-	except Exception:
-		return str(data)
-
-
-# =============================================================================
-# 헬퍼: 프롬프트 생성
-# =============================================================================
-
-def _create_output_summary_prompt(outputs_str: str) -> str:
-	"""결과물 요약용 사용자 프롬프트를 생성한다."""
-	return (
-		"다음 작업 결과를 정리해주세요:\n\n"
-		f"{outputs_str}\n\n"
-		"처리 방식:\n"
-		"- 짧은 내용은 요약하지 말고 그대로 유지 (정보 손실 방지)\n"
-		"- 긴 내용만 적절히 요약하여 핵심 정보 전달\n"
-		"- 수치, 목차, 인물명, 물건명, 날짜, 시간 등 객관적 정보는 반드시 포함\n"
-		"- 왜곡이나 의미 변경 금지, 원본 의미 보존\n"
-		"- 중복만 정리하고 핵심 내용은 모두 보존\n"
-		"- 하나의 통합된 문맥으로 작성"
-	)
-
-
-def _create_feedback_summary_prompt(feedbacks_str: str, contents_str: str = "") -> str:
-	"""피드백/현재 결과물을 통합 요약하는 사용자 프롬프트를 생성한다."""
-	feedback_section = (
-		f"=== 피드백 내용 ===\n{feedbacks_str}" if feedbacks_str and feedbacks_str.strip() else ""
-	)
-	content_section = (
-		f"=== 현재 결과물/작업 내용 ===\n{contents_str}" if contents_str and contents_str.strip() else ""
-	)
-	return (
-		"다음은 사용자의 피드백과 결과물입니다. 이를 분석하여 통합된 피드백을 작성해주세요:\n\n"
-		f"{feedback_section}\n\n{content_section}\n\n"
-		"상황 분석 및 처리 방식:\n"
-		"- 현재 결과물을 보고 문제/개선 필요점 판단\n"
-		"- 최신 피드백을 최우선으로 반영\n"
-		"- 실행 가능한 개선사항 제시\n"
-		"- 하나의 통합된 문장으로 작성 (최대 2500자)"
-	)
-
-
-# =============================================================================
-# 헬퍼: 시스템 프롬프트 선택
-# =============================================================================
-
-def _get_system_prompt(task_name: str) -> str:
-	"""작업 종류에 맞는 시스템 프롬프트를 반환한다."""
-	if task_name == "feedback":
-		return (
-			"당신은 피드백 정리 전문가입니다. 최신 피드백을 우선 반영하고,"
-			" 문맥을 연결하여 하나의 완전한 요청으로 통합해 주세요."
-		)
-	return (
-		"당신은 결과물 요약 전문가입니다. 긴 내용만 요약하고,"
-		" 수치/고유명/날짜 등 객관 정보를 보존해 주세요."
-	)
-
-
-# =============================================================================
-# 외부 호출: OpenAI API
-# =============================================================================
-
-async def _call_openai_api_async(prompt: str, task_name: str) -> str:
-	"""OpenAI 비동기 API를 호출해 요약 텍스트를 생성한다."""
-	
-	if not (os.getenv("OPENAI_API_KEY") or os.getenv("OPENAI_API_KEY_BETA")):
-		write_log_message("요약 비활성화: OPENAI_API_KEY 미설정")
-		return ""
-
-	client = openai.AsyncOpenAI()
-	system_prompt = _get_system_prompt(task_name)
-	model = os.getenv("OPENAI_SUMMARY_MODEL", "gpt-4o-mini")
-
-	for attempt in range(1, 4):
-		try:
-			resp = await client.chat.completions.create(
-				model=model,
-				messages=[
-					{"role": "system", "content": system_prompt},
-					{"role": "user", "content": prompt},
-				],
-				temperature=0.1,
-				timeout=30.0,
-			)
-			return (resp.choices[0].message.content or "").strip()
-		except asyncio.CancelledError:
-			raise
-		except Exception as e:
-			if attempt < 3:
-				handle_application_error("요약 호출 오류(재시도)", e, raise_error=False, extra={"attempt": attempt})
-				await asyncio.sleep(0.8 * (2 ** (attempt - 1)))
-				continue
-			handle_application_error("요약 호출 최종 실패", e, raise_error=False)
-			return ""