ChatAnthropic

app/services/ai_tool.py

@@ -4,7 +4,7 @@ import re
 from typing import Any
 
 from json_repair import repair_json
-from langchain_openai import ChatOpenAI
+from langchain_core.language_models import BaseChatModel
 from langchain_core.messages import SystemMessage, HumanMessage
 
 from app.core.logger import log
@@ -28,7 +28,7 @@ def parse_llm_json(text: str) -> Any:
     return repair_json(cleaned, return_objects=True)
 
 
-async def ai_chat(llm: ChatOpenAI, system_prompt: str, user_message: str) -> str:
+async def ai_chat(llm: BaseChatModel, system_prompt: str, user_message: str) -> str:
     """异步调用 LLM，返回原始文本"""
     messages = [
         SystemMessage(content=system_prompt),
@@ -38,7 +38,7 @@ async def ai_chat(llm: ChatOpenAI, system_prompt: str, user_message: str) -> str
     return response.content
 
 
-async def ai_chat_json(llm: ChatOpenAI, system_prompt: str, user_message: str) -> Any:
+async def ai_chat_json(llm: BaseChatModel, system_prompt: str, user_message: str) -> Any:
     """异步调用 LLM，返回解析后的 JSON 对象"""
     raw = await ai_chat(llm, system_prompt, user_message)
     if not raw or not raw.strip():