80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
"""GEO AgentKit Server 启动入口
|
||
|
||
工厂函数 create_geo_app() 使用 agentkit.yaml 统一配置，
|
||
初始化 LLM Gateway、Tool Registry、Skill Registry，然后创建 FastAPI 应用。
|
||
|
||
使用方式：
|
||
uvicorn configs.geo_server:create_geo_app --factory --host 0.0.0.0 --port 8001
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
import os
|
||
|
||
from fastapi import FastAPI
|
||
|
||
from agentkit.server.app import _build_llm_gateway, create_app
|
||
from agentkit.server.config import ServerConfig
|
||
from agentkit.skills.loader import SkillLoader
|
||
from agentkit.skills.registry import SkillRegistry
|
||
from agentkit.tools.registry import ToolRegistry
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# ─── 配置路径 ───
|
||
|
||
CONFIGS_DIR = os.path.dirname(os.path.abspath(__file__))
|
||
SKILLS_DIR = os.path.join(CONFIGS_DIR, "skills")
|
||
|
||
# 查找 agentkit.yaml：项目根目录 > configs 目录
|
||
_PROJECT_ROOT = os.path.dirname(CONFIGS_DIR)
|
||
_AGENTKIT_YAML = os.path.join(_PROJECT_ROOT, "agentkit.yaml")
|
||
|
||
|
||
def _load_server_config() -> ServerConfig:
|
||
"""Load ServerConfig from agentkit.yaml with env var resolution."""
|
||
if os.path.isfile(_AGENTKIT_YAML):
|
||
return ServerConfig.from_yaml(_AGENTKIT_YAML)
|
||
raise FileNotFoundError(f"agentkit.yaml not found at {_AGENTKIT_YAML}")
|
||
|
||
|
||
def _init_tool_registry() -> ToolRegistry:
|
||
"""初始化 Tool Registry 并注册 GEO Tools"""
|
||
registry = ToolRegistry()
|
||
from configs.geo_tools import register_geo_tools
|
||
|
||
register_geo_tools(registry)
|
||
return registry
|
||
|
||
|
||
def _init_skill_registry(tool_registry: ToolRegistry) -> SkillRegistry:
|
||
"""初始化 Skill Registry 并从 configs/skills/ 目录加载"""
|
||
registry = SkillRegistry()
|
||
loader = SkillLoader(registry, tool_registry)
|
||
skills = loader.load_from_directory(SKILLS_DIR)
|
||
logger.info(f"Loaded {len(skills)} skills from {SKILLS_DIR}")
|
||
return registry
|
||
|
||
|
||
def create_geo_app() -> FastAPI:
|
||
"""GEO AgentKit Server FastAPI 工厂函数"""
|
||
config = _load_server_config()
|
||
llm_gateway = _build_llm_gateway(config)
|
||
tool_registry = _init_tool_registry()
|
||
skill_registry = _init_skill_registry(tool_registry)
|
||
|
||
app = create_app(
|
||
llm_gateway=llm_gateway,
|
||
skill_registry=skill_registry,
|
||
tool_registry=tool_registry,
|
||
)
|
||
app.title = "GEO AgentKit Server"
|
||
|
||
logger.info(
|
||
f"GEO AgentKit Server initialized: {len(skill_registry.list_skills())} skills, "
|
||
f"{len(tool_registry.list_tools())} tools"
|
||
)
|
||
|
||
return app
|