Files
FiDA_Python/src/server/agent/tools.py

26 lines
1.1 KiB
Python
Raw Normal View History

2026-02-04 17:57:49 +08:00
from langchain_core.tools import tool
# --- 模拟你已经开发好的报告生成功能 ---
@tool
def generate_2025_report_tool(topic: str) -> str:
"""
专门用于收集信息并生成报告
当用户询问关于趋势市场分析年度报告如2025家具报告时调用此工具
"""
print(f"\n[系统日志] 正在调用外部模块生成关于 '{topic}' 的报告...")
# 这里对接你实际的代码比如return my_existing_module.run(topic)
return f"【报告生成成功】已生成关于 {topic} 的 PDF 报告。核心洞察2025年趋势倾向于生物嗜好设计(Biophilic Design)和可持续软木材质。"
# --- 绘图工具 ---
@tool
def generate_furniture_sketch(prompt: str) -> str:
"""
用于生成家具草图输入必须是详细的英文绘画提示词(Prompt)
"""
print(f"\n[系统日志] 正在调用 Gemini/Imagen 绘图 APIPrompt: {prompt}...")
# 在真实场景中,这里调用 Google Imagen API 或 Midjourney API
# 示例返回一个模拟的图片链接
return "https://furniture-design-db.com/generated_sketch_v1.jpg"