런타임 강화: 긴 컨텍스트 윈도우에서 "지시문 부식" 방지하기

import datetime
import pytz

def get_temporal_context():
    # 1. 기업 타임존 정의 (데이터 정렬에 필수)
    target_tz = pytz.timezone("America/Los_Angeles")
    # 2. 특정 현재 시각 획득
    current_time = datetime.datetime.now(target_tz).strftime(
        "%Y-%m-%d %H:%M:%S %Z"
    )
    # 3. 시스템 지시문 형식으로 반환
    return f"[Timestamp]: The current date and time is {current_time}."

class ImprovePrompt:
    def before_model_callback(self, callback_context, llm_request) -> Optional[None]:
        target_tz = pytz.timezone("America/Los_Angeles")
        current_time = datetime.datetime.now(target_tz).strftime("%Y-%m-%d %H:%M:%S %Z")
        timestamp_prefix = f"[System Clock]: Current Time is {current_time}."

        reinforcement_prompt = """
        [1] DATA EXCLUSION RULES (MANDATORY):
        - IF querying 'Active Users', you MUST exclude 'test_%' IDs.
        [2] SYNTAX SAFETY:
        - ALWAYS use JSON_EXTRACT(data, '$.id') instead of dot notation.
        """

        if not llm_request.contents: return None
        current_parts = list(llm_request.contents[0].parts)
        injection_text = f"

--- INTERNAL SYSTEM INJECTION ---
{timestamp_prefix}
{reinforcement_prompt}
"
        
        injection_part = types.Part(text=injection_text)
        new_parts = [injection_part] + current_parts
        llm_request.contents[0].parts = new_parts
        return None