admin
1741f7b1aa
- 实现ProviderFactory用于构建不同类型的provider适配器 - 添加EnvironmentCredentialResolver用于解析环境变量中的凭证 - 实现OllamaProvider支持本地模型调用 - 实现OpenAICompatibleProvider支持OpenAI兼容接口 - 在AgentRuntime中添加对ProviderError的处理 - 更新Message结构体添加tool_calls字段 - 实现provider配置的增删改查API端点 - 添加provider注册表的replace方法 - 添加HTTP基础类和工具参数解码功能 - 更新依赖添加httpx库 - 添加相关单元测试验证provider适配器功能 ```
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
from dataclasses import dataclass
|
|
|
|
from app.agent import AgentRuntime, PermissionManager, PermissionPolicy, ToolRegistry
|
|
from app.agent.builtin_tools import register_builtin_tools
|
|
from app.contracts import ModelCapability, ProviderConfig, ProviderType
|
|
from app.providers import MockProvider, ProviderFactory, ProviderRegistry
|
|
from app.providers.credentials import EnvironmentCredentialResolver
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class ApplicationContainer:
|
|
providers: ProviderRegistry
|
|
provider_factory: ProviderFactory
|
|
tools: ToolRegistry
|
|
permissions: PermissionManager
|
|
agent: AgentRuntime
|
|
|
|
|
|
def build_container() -> ApplicationContainer:
|
|
provider_factory = ProviderFactory(EnvironmentCredentialResolver())
|
|
providers = ProviderRegistry()
|
|
providers.register(
|
|
ProviderConfig(
|
|
provider_id="mock",
|
|
provider_type=ProviderType.mock,
|
|
name="Mock Provider",
|
|
default_model="mock-1",
|
|
enabled=True,
|
|
capabilities=[
|
|
ModelCapability.chat,
|
|
ModelCapability.tool_calling,
|
|
ModelCapability.streaming,
|
|
],
|
|
),
|
|
MockProvider(),
|
|
)
|
|
|
|
tools = ToolRegistry()
|
|
register_builtin_tools(tools)
|
|
|
|
policy = PermissionPolicy()
|
|
permissions = PermissionManager(policy)
|
|
agent = AgentRuntime(providers=providers, tools=tools, permissions=permissions)
|
|
return ApplicationContainer(
|
|
providers=providers,
|
|
provider_factory=provider_factory,
|
|
tools=tools,
|
|
permissions=permissions,
|
|
agent=agent,
|
|
)
|
|
|
|
|
|
container = build_container()
|