feat:新增投顾agent和nl2sqlagent
This commit is contained in:
@@ -0,0 +1,36 @@
|
||||
"""投顾 Agent 运行时依赖组装。"""
|
||||
from __future__ import annotations
|
||||
|
||||
from dataclasses import dataclass
|
||||
|
||||
from agent.advisor_agent.memory import (
|
||||
EmptyMemoryProvider,
|
||||
MemoryProvider,
|
||||
MemoryServiceProvider,
|
||||
)
|
||||
|
||||
|
||||
@dataclass
|
||||
class AdvisorAgentRuntime:
|
||||
memory_provider: MemoryProvider
|
||||
llm_client: object | None = None
|
||||
graph_tool: object | None = None
|
||||
redis: object | None = None
|
||||
|
||||
|
||||
def build_default_runtime(
|
||||
*,
|
||||
memory_provider: MemoryProvider | None = None,
|
||||
llm_client: object | None = None,
|
||||
graph_tool: object | None = None,
|
||||
redis: object | None = None,
|
||||
memory_service: object | None = None,
|
||||
) -> AdvisorAgentRuntime:
|
||||
if memory_provider is None and memory_service is not None:
|
||||
memory_provider = MemoryServiceProvider(memory_service)
|
||||
return AdvisorAgentRuntime(
|
||||
memory_provider=memory_provider or EmptyMemoryProvider(),
|
||||
llm_client=llm_client,
|
||||
graph_tool=graph_tool,
|
||||
redis=redis,
|
||||
)
|
||||
Reference in New Issue
Block a user