44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
"""
|
|
A2A LiteLLM Agent 主入口
|
|
支持从环境变量或请求传入 API key
|
|
"""
|
|
import os
|
|
import uvicorn
|
|
from a2a_server import create_app
|
|
|
|
# 环境变量配置
|
|
SERVICE_HOST = os.getenv("SERVICE_HOST", "0.0.0.0")
|
|
SERVICE_PORT = int(os.getenv("SERVICE_PORT", "8080"))
|
|
POD_NAME = os.getenv("POD_NAME", "a2a-litellm-agent")
|
|
TEMPLATE_TYPE = os.getenv("TEMPLATE_TYPE", "a2a_litellm_agent")
|
|
|
|
# 从环境变量获取默认配置（可选）
|
|
default_api_key = os.getenv("LITELLM_API_KEY")
|
|
default_model = os.getenv("MODEL_NAME") or os.getenv("LITELLM_MODEL")
|
|
|
|
# 创建应用
|
|
app = create_app(api_key=default_api_key, model=default_model)
|
|
|
|
|
|
def main():
|
|
"""主函数"""
|
|
print(f"🚀 启动 A2A LiteLLM Agent")
|
|
print(f" - Pod名称: {POD_NAME}")
|
|
print(f" - 模板类型: {TEMPLATE_TYPE}")
|
|
print(f" - 服务地址: http://{SERVICE_HOST}:{SERVICE_PORT}")
|
|
if default_api_key:
|
|
print(f" - 已配置默认 API key（可通过请求覆盖）")
|
|
else:
|
|
print(f" - 未配置默认 API key，需在请求中传入")
|
|
|
|
uvicorn.run(
|
|
app,
|
|
host=SERVICE_HOST,
|
|
port=SERVICE_PORT,
|
|
log_level="info"
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|