From 30101cf1f0191c61bd351dd001d060aa8f324013 Mon Sep 17 00:00:00 2001
From: webjoin111 <455457521@qq.com>
Date: Fri, 12 Dec 2025 17:04:09 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20feat(llm):=20=E6=94=AF=E6=8C=81?=
 =?UTF-8?q?=E7=BB=93=E6=9E=84=E5=8C=96=E7=94=9F=E6=88=90=E5=87=BD=E6=95=B0?=
 =?UTF-8?q?=E6=8E=A5=E6=94=B6=20UniMessage?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 zhenxun/services/llm/api.py     | 2 +-
 zhenxun/services/llm/session.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/zhenxun/services/llm/api.py b/zhenxun/services/llm/api.py
index 3bb45897..644c6de0 100644
--- a/zhenxun/services/llm/api.py
+++ b/zhenxun/services/llm/api.py
@@ -182,7 +182,7 @@ async def embed_documents(
 
 
 async def generate_structured(
-    message: str | LLMMessage | list[LLMContentPart],
+    message: str | UniMessage | LLMMessage | list[LLMContentPart],
     response_model: type[T],
     *,
     model: ModelName = None,
diff --git a/zhenxun/services/llm/session.py b/zhenxun/services/llm/session.py
index a07a7987..5f2e646b 100644
--- a/zhenxun/services/llm/session.py
+++ b/zhenxun/services/llm/session.py
@@ -449,7 +449,7 @@ class AI:
 
     async def generate_structured(
         self,
-        message: str | LLMMessage | list[LLMContentPart] | None,
+        message: str | UniMessage | LLMMessage | list[LLMContentPart] | None,
         response_model: type[T],
         *,
         model: ModelName = None,