彻底重构状态系统:整合所有旧状态到 MainGraphState,修复所有节点
Some checks failed
构建并部署 AI Agent 服务 / deploy (push) Failing after 6m35s
Some checks failed
构建并部署 AI Agent 服务 / deploy (push) Failing after 6m35s
This commit is contained in:
@@ -7,14 +7,6 @@ from app.config import ENABLE_GRAPH_TRACE
|
||||
from app.logger import debug, info
|
||||
|
||||
|
||||
def _get_attr(state, attr_name, default=None):
|
||||
"""通用方法:兼容 dict 和 dataclass 两种状态格式"""
|
||||
if isinstance(state, dict):
|
||||
return state.get(attr_name, default)
|
||||
else:
|
||||
return getattr(state, attr_name, default)
|
||||
|
||||
|
||||
def log_state_change(node_name: str, state, prefix: str = "进入"):
|
||||
"""
|
||||
记录状态变化日志
|
||||
@@ -26,7 +18,13 @@ def log_state_change(node_name: str, state, prefix: str = "进入"):
|
||||
"""
|
||||
from app.logger import info
|
||||
|
||||
messages = _get_attr(state, "messages", [])
|
||||
# 获取 messages
|
||||
messages = []
|
||||
if isinstance(state, dict):
|
||||
messages = state.get("messages", [])
|
||||
else:
|
||||
messages = getattr(state, "messages", [])
|
||||
|
||||
msg_count = len(messages)
|
||||
last_msg = messages[-1] if messages else None
|
||||
last_info = ""
|
||||
@@ -48,7 +46,7 @@ def print_llm_input(prompt_value):
|
||||
|
||||
Args:
|
||||
prompt_value: ChatPromptValue 对象,包含格式化后的消息列表
|
||||
|
||||
|
||||
Returns:
|
||||
原样返回 prompt_value,不影响链式调用
|
||||
"""
|
||||
@@ -56,14 +54,14 @@ def print_llm_input(prompt_value):
|
||||
return prompt_value
|
||||
|
||||
messages = prompt_value.messages # ChatPromptValue 提供 .messages 属性
|
||||
|
||||
debug("\n" + "=" * 80)
|
||||
debug("📤 [LLM输入] 格式化后发送给大模型的完整消息:")
|
||||
|
||||
debug("\n" + "="*80)
|
||||
debug("📥 [LLM输入] 格式化后发送给大模型的完整消息:")
|
||||
debug(f" 总消息数: {len(messages)}")
|
||||
debug("-" * 80)
|
||||
debug("-"*80)
|
||||
for i, msg in enumerate(messages):
|
||||
content_preview = str(msg.content) # 完整输出
|
||||
debug(f" [{i}] {msg.type.upper():10s}: {content_preview}")
|
||||
debug("\n" + "=" * 80 + "\n")
|
||||
debug("\n" + "="*80 + "\n")
|
||||
|
||||
return prompt_value
|
||||
return prompt_value
|
||||
Reference in New Issue
Block a user