liqiang
b119135836
Build latest book artifacts / build (push) Canceled after 0s
dependency resolution / resolve (3.11) (push) Canceled after 0s
dependency resolution / resolve (3.13) (push) Canceled after 0s
deploy-pages / build (push) Canceled after 0s
deploy-pages / deploy (push) Canceled after 0s
i18n consistency check / check (push) Canceled after 0s
provider adoption tests / test (chapter2/context-compression) (push) Canceled after 0s
provider adoption tests / test (chapter2/prompt-injection) (push) Canceled after 0s
provider adoption tests / test (chapter2/system-hint) (push) Canceled after 0s
provider adoption tests / test (chapter3/log-sanitization) (push) Canceled after 0s
web-search-agent tests / test (push) Canceled after 0s
web-search-agent tests / agentbook (push) Canceled after 0s
32 lines
796 B
Python
32 lines
796 B
Python
from typing import Optional
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from aworldspace.base_agent import AworldBaseAgent
|
|
|
|
"""
|
|
Agent Space
|
|
"""
|
|
|
|
class AgentMeta(BaseModel):
|
|
name: str = None
|
|
desc: str = None
|
|
|
|
|
|
|
|
class AgentSpace(BaseModel):
|
|
agent_modules: Optional[dict] = Field(default_factory=dict, description="agent module")
|
|
agents_meta: Optional[dict] = Field(default_factory=dict, description="agents meta")
|
|
|
|
def register(self, agent_name: str, agent_instance: AworldBaseAgent, metadata: dict=None):
|
|
# Register agent metadata and instance
|
|
self.agent_modules[agent_name] = agent_instance
|
|
|
|
async def get_agent_modules(self):
|
|
return self.agent_modules
|
|
|
|
async def get_agents_meta(self):
|
|
return self.agents_meta
|
|
|
|
|
|
AGENT_SPACE = AgentSpace() |