mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-20 20:39:55 +00:00
zhangtao
b0ae6af3b1
- Added *.pyc and *.pyo to .gitignore to prevent compiled Python files from being tracked. - Improved docstrings in various modules, providing clearer descriptions of functions, parameters, and return values to enhance code readability and maintainability.
74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
from typing import Any
|
|
|
|
from agno.agent import Agent
|
|
from agno.models.openai.like import OpenAILike
|
|
from agno.team import Team
|
|
|
|
from app.config.setting import settings
|
|
|
|
|
|
class AgnoFactory:
|
|
"""Agno 工厂类 - 统一管理 Agent、Team 创建逻辑"""
|
|
# 配置常量
|
|
AGENT_DESCRIPTION = "你是一个有用的AI助手，可以帮助用户回答问题和提供帮助。"
|
|
AGENT_INSTRUCTIONS = ["保持回答简洁明了", "如果不确定，请说明"]
|
|
AGENT_EXPECTED_OUTPUT = "中文回答"
|
|
AGENT_TEMPERATURE = 0.7
|
|
NUM_HISTORY_RUNS = 3
|
|
|
|
def create_agent(
|
|
self,
|
|
user_id: str,
|
|
dept_id: str,
|
|
session_id: str,
|
|
db: Any | None = None
|
|
) -> Team:
|
|
"""
|
|
创建带 Agent 的 Team 实例。
|
|
|
|
参数:
|
|
- user_id (str): 用户标识。
|
|
- dept_id (str): 部门/团队标识。
|
|
- session_id (str): 会话 ID。
|
|
- db (Any | None): Agno 持久化数据库实例，可选。
|
|
|
|
返回:
|
|
- Team: 配置好的 Team。
|
|
"""
|
|
|
|
# 创建 Agent
|
|
fastapiadmin_agent = Agent(
|
|
id=user_id,
|
|
name="fastapiadmin_agent",
|
|
role="You are a helpful AI assistant",
|
|
description=self.AGENT_DESCRIPTION,
|
|
tools=[],
|
|
)
|
|
|
|
# 创建 Team
|
|
fastapiadmin_team = Team(
|
|
id=dept_id,
|
|
user_id=user_id,
|
|
session_id=session_id,
|
|
model=OpenAILike(
|
|
id=settings.OPENAI_MODEL,
|
|
api_key=settings.OPENAI_API_KEY,
|
|
base_url=settings.OPENAI_BASE_URL,
|
|
temperature=self.AGENT_TEMPERATURE,
|
|
),
|
|
members=[fastapiadmin_agent],
|
|
instructions=self.AGENT_INSTRUCTIONS,
|
|
expected_output=self.AGENT_EXPECTED_OUTPUT,
|
|
add_datetime_to_context=True,
|
|
add_history_to_context=True,
|
|
markdown=True,
|
|
num_history_runs=self.NUM_HISTORY_RUNS,
|
|
input_schema=None,
|
|
output_schema=None,
|
|
parse_response=True,
|
|
read_chat_history=True,
|
|
db=db,
|
|
)
|
|
|
|
return fastapiadmin_team
|