@@ -297,8 +297,6 @@ async def create_agent_config(
297297 }
298298 system_prompt = Template (prompt_template ["system_prompt" ], undefined = StrictUndefined ).render (render_kwargs )
299299
300- _print_prompt_with_token_count (system_prompt , agent_id , "BEFORE_INJECTION" )
301-
302300 if agent_info .get ("model_id" ) is not None :
303301 model_info = get_model_by_model_id (agent_info .get ("model_id" ))
304302 model_name = model_info ["display_name" ] if model_info is not None else "main_model"
@@ -452,25 +450,9 @@ async def prepare_prompt_templates(
452450 prompt_templates = get_agent_prompt_template (is_manager , language )
453451 prompt_templates ["system_prompt" ] = system_prompt
454452
455- # Print final prompt with all injections
456- _print_prompt_with_token_count (prompt_templates ["system_prompt" ], agent_id , "FINAL_PROMPT" )
457-
458453 return prompt_templates
459454
460455
461- def _print_prompt_with_token_count (prompt : str , agent_id : int = None , stage : str = "PROMPT" ):
462- """Print prompt content and estimate token count using tiktoken."""
463- try :
464- import tiktoken
465- encoding = tiktoken .get_encoding ("cl100k_base" )
466- token_count = len (encoding .encode (prompt ))
467- logger .info (f"[Skill Debug][{ stage } ] Agent { agent_id } token count: { token_count } " )
468- logger .info (f"[Skill Debug][{ stage } ] Agent { agent_id } prompt:\n { prompt } " )
469- except Exception as e :
470- logger .warning (f"[Skill Debug][{ stage } ] Failed to count tokens: { e } " )
471- logger .info (f"[Skill Debug][{ stage } ] Agent { agent_id } prompt:\n { prompt } " )
472-
473-
474456async def join_minio_file_description_to_query (minio_files , query ):
475457 final_query = query
476458 if minio_files and isinstance (minio_files , list ):
0 commit comments