use LLMNodeConfig

This commit is contained in:
2026-02-12 14:54:27 +08:00
parent 1972c182d8
commit 9363bd3442
8 changed files with 16 additions and 16 deletions

View File

@@ -12,7 +12,7 @@ import base64
import json
from loguru import logger
from lang_agent.config import LLMKeyConfig
from lang_agent.config import LLMNodeConfig
from lang_agent.components.tool_manager import ToolManager, ToolManagerConfig
from lang_agent.components.prompt_store import build_prompt_store
from lang_agent.base import GraphBase, ToolNodeBase
@@ -90,7 +90,7 @@ class VisionRoutingState(TypedDict):
@tyro.conf.configure(tyro.conf.SuppressFixed)
@dataclass
class VisionRoutingConfig(LLMKeyConfig):
class VisionRoutingConfig(LLMNodeConfig):
_target: Type = field(default_factory=lambda: VisionRoutingGraph)
tool_llm_name: str = "qwen-flash"