use llm key config
This commit is contained in:
@@ -9,7 +9,7 @@ import commentjson
|
|||||||
import glob
|
import glob
|
||||||
import time
|
import time
|
||||||
|
|
||||||
from lang_agent.config import KeyConfig
|
from lang_agent.config import LLMKeyConfig
|
||||||
from lang_agent.components.tool_manager import ToolManager, ToolManagerConfig
|
from lang_agent.components.tool_manager import ToolManager, ToolManagerConfig
|
||||||
from lang_agent.base import GraphBase, ToolNodeBase
|
from lang_agent.base import GraphBase, ToolNodeBase
|
||||||
from lang_agent.graphs.graph_states import State
|
from lang_agent.graphs.graph_states import State
|
||||||
@@ -27,7 +27,7 @@ from langgraph.checkpoint.memory import MemorySaver
|
|||||||
|
|
||||||
@tyro.conf.configure(tyro.conf.SuppressFixed)
|
@tyro.conf.configure(tyro.conf.SuppressFixed)
|
||||||
@dataclass
|
@dataclass
|
||||||
class RoutingConfig(KeyConfig):
|
class RoutingConfig(LLMKeyConfig):
|
||||||
_target: Type = field(default_factory=lambda: RoutingGraph)
|
_target: Type = field(default_factory=lambda: RoutingGraph)
|
||||||
|
|
||||||
llm_name: str = "qwen-plus"
|
llm_name: str = "qwen-plus"
|
||||||
|
|||||||
Reference in New Issue
Block a user