37 lines
990 B
Python
37 lines
990 B
Python
"""投顾 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,
|
|
)
|