add kia ai chat tools13
CI/CD Pipeline / build-and-deploy (push) Successful in 17s Details

This commit is contained in:
vahidrezvani 2026-02-25 14:38:25 +03:30
parent 2650eb4776
commit 1be3dc5838
1 changed files with 2 additions and 2 deletions

View File

@ -459,7 +459,7 @@ async def send_message(
from .kiaai import KiaAIService from .kiaai import KiaAIService
# Push the human message into the graph state first # Push the human message into the graph state first
await app.aupdate_state(config, {"messages": [human_msg]}) await app.aupdate_state(config, {"messages": [human_msg]}, as_node="model")
state = await app.aget_state(config) state = await app.aget_state(config)
all_msgs = state.values.get("messages", [human_msg]) all_msgs = state.values.get("messages", [human_msg])
trimmed = await get_trimmer().ainvoke(all_msgs) trimmed = await get_trimmer().ainvoke(all_msgs)
@ -506,7 +506,7 @@ async def send_message(
"total_tokens": input_tokens + output_tokens, "total_tokens": input_tokens + output_tokens,
} }
ai_msg.additional_kwargs["created_at"] = ai_msg_created_at ai_msg.additional_kwargs["created_at"] = ai_msg_created_at
await app.aupdate_state(config, {"messages": [ai_msg]}) await app.aupdate_state(config, {"messages": [ai_msg]}, as_node="model")
await user.save() await user.save()
await Message.create( await Message.create(