use LLMNodeConfig
This commit is contained in:
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user