consultant-agent 0.1.0__tar.gz

consultant_agent-0.1.0/CHANGELOG.md

@@ -0,0 +1,29 @@
+# 更新日志
+
+## [0.1.0] - 2024-07-08
+
+### 新增
+- 🎉 首次发布
+- ✨ 核心 Agent 功能
+  - 天气查询
+  - 数学计算
+  - 联网搜索
+  - 网页抓取
+- 📄 文档处理
+  - PDF 阅读
+  - Word 阅读
+  - Excel 阅读
+  - PPT 阅读
+- 📊 分析功能
+  - 内容总结
+  - 报告大纲生成
+  - 竞品对比
+  - 案例搜索
+- 🖥️ 命令行工具
+  - `consultant` 交互式对话
+  - `consultant ask` 单次提问
+  - `consultant clear` 清空记忆
+- 🔧 开发者 API
+  - `ConsultantAgent` 类
+  - `as_tool()` 方法，可嵌入其他 Agent
+- 💾 对话记忆功能

consultant_agent-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 xiaoboluo
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

consultant_agent-0.1.0/PKG-INFO

@@ -0,0 +1,195 @@
+Metadata-Version: 2.4
+Name: consultant-agent
+Version: 0.1.0
+Summary: 一个专业的咨询顾问 AI Agent，支持文档分析、联网搜索、报告生成
+Project-URL: Homepage, https://github.com/xiaoboluo/consultant-agent
+Project-URL: Repository, https://github.com/xiaoboluo/consultant-agent
+Author-email: xiaoboluo <your@email.com>
+License-Expression: MIT
+License-File: LICENSE
+Keywords: agent,ai,consultant,langchain,llm
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Requires-Dist: beautifulsoup4>=4.12.0
+Requires-Dist: ddgs>=6.0.0
+Requires-Dist: langchain-openai>=0.1.0
+Requires-Dist: langchain>=0.2.0
+Requires-Dist: langgraph>=0.1.0
+Requires-Dist: openpyxl>=3.1.0
+Requires-Dist: pandas>=2.0.0
+Requires-Dist: pdfplumber>=0.10.0
+Requires-Dist: python-docx>=1.0.0
+Requires-Dist: python-pptx>=0.6.0
+Requires-Dist: requests>=2.31.0
+Provides-Extra: dev
+Requires-Dist: black>=24.0.0; extra == 'dev'
+Requires-Dist: pytest>=8.0.0; extra == 'dev'
+Requires-Dist: ruff>=0.3.0; extra == 'dev'
+Provides-Extra: ocr
+Requires-Dist: pillow>=10.0.0; extra == 'ocr'
+Requires-Dist: pymupdf>=1.23.0; extra == 'ocr'
+Requires-Dist: pytesseract>=0.3.10; extra == 'ocr'
+Provides-Extra: web
+Requires-Dist: fastapi>=0.110.0; extra == 'web'
+Requires-Dist: gradio>=4.0.0; extra == 'web'
+Requires-Dist: uvicorn>=0.27.0; extra == 'web'
+Description-Content-Type: text/markdown
+
+# Consultant Agent 🤖
+
+一个专业的咨询顾问 AI Agent，一行命令安装，开箱即用。
+
+[![PyPI version](https://badge.fury.io/py/consultant-agent.svg)](https://pypi.org/project/consultant-agent/)
+[![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+## ✨ 特性
+
+- 🔍 **联网搜索** - 实时获取最新信息
+- 📄 **文档阅读** - 支持 PDF、Word、Excel、PPT
+- 📊 **数据分析** - Excel 数据处理与分析
+- 📝 **内容生成** - 报告大纲、竞品对比、内容总结
+- 🛠️ **易于集成** - 可作为工具嵌入其他 Agent
+- 💾 **对话记忆** - 自动保存上下文
+
+## 🚀 快速开始
+
+### 安装
+
+```bash
+pip install consultant-agent
+```
+
+### 设置 API Key
+
+```bash
+# Linux/Mac
+export DEEPSEEK_API_KEY="sk-your-api-key"
+
+# Windows
+set DEEPSEEK_API_KEY=sk-your-api-key
+```
+
+> 💡 去 [DeepSeek 开放平台](https://platform.deepseek.com/) 获取 API Key
+
+### 使用
+
+**命令行交互**
+```bash
+consultant
+```
+
+**单次提问**
+```bash
+consultant ask "北京今天天气怎么样"
+consultant ask "帮我生成一个 AI Agent 技术发展的报告大纲"
+```
+
+**Python 代码**
+```python
+from consultant_agent import ConsultantAgent
+
+agent = ConsultantAgent()
+reply = agent.ask("分析一下新能源汽车行业趋势")
+print(reply)
+```
+
+## 📖 功能详解
+
+### 命令行命令
+
+| 命令 | 说明 | 示例 |
+|------|------|------|
+| `consultant` | 交互式对话 | - |
+| `consultant ask "问题"` | 单次提问 | `consultant ask "上海天气"` |
+| `consultant clear` | 清空记忆 | - |
+
+### Python API
+
+```python
+from consultant_agent import ConsultantAgent
+
+# 基础用法
+agent = ConsultantAgent()
+reply = agent.ask("你的问题")
+
+# 自定义配置
+agent = ConsultantAgent(
+    api_key="sk-xxx",           # 可选，默认从环境变量读取
+    model="deepseek-chat",      # 模型名称
+    temperature=0,              # 温度参数
+    enable_memory=True,         # 启用记忆
+)
+
+# 清空记忆
+agent.clear_memory()
+```
+
+### 作为工具嵌入其他 Agent
+
+```python
+from consultant_agent import ConsultantAgent
+from langgraph.prebuilt import create_react_agent
+
+# 创建咨询顾问
+consultant = ConsultantAgent()
+
+# 转换为工具
+consultant_tool = consultant.as_tool()
+
+# 在你的 Agent 中使用
+your_agent = create_react_agent(
+    your_llm,
+    tools=[consultant_tool, your_other_tools...]
+)
+```
+
+## 🎯 支持的功能
+
+| 功能 | 说明 | 示例 |
+|------|------|------|
+| 天气查询 | 全球城市天气 | "北京天气" |
+| 数学计算 | 表达式计算 | "计算 123 * 456" |
+| 联网搜索 | DuckDuckGo 搜索 | "搜索 AI Agent 最新进展" |
+| 网页抓取 | 读取网页内容 | "读取 https://example.com" |
+| PDF 阅读 | 提取 PDF 文字 | "读取 report.pdf" |
+| Word 阅读 | 读取 docx 文档 | "读取 document.docx" |
+| Excel 阅读 | 读取表格数据 | "读取 data.xlsx" |
+| PPT 阅读 | 提取幻灯片文字 | "读取 slides.pptx" |
+| 内容总结 | 多种总结风格 | "总结这篇文章" |
+| 报告大纲 | 生成专业大纲 | "生成 XX 报告大纲" |
+| 竞品对比 | 多维度对比表 | "对比 A、B、C 产品" |
+| 案例搜索 | 查找行业案例 | "搜索数字化转型案例" |
+
+## ⚙️ 环境变量
+
+| 变量 | 必须 | 说明 |
+|------|------|------|
+| `DEEPSEEK_API_KEY` | ✅ | DeepSeek API Key |
+
+## 📦 可选依赖
+
+```bash
+# OCR 功能（识别图片/PDF 中的文字）
+pip install consultant-agent[ocr]
+
+# Web 界面
+pip install consultant-agent[web]
+
+# 完整安装
+pip install consultant-agent[ocr,web]
+```
+
+## 🤝 贡献
+
+欢迎 Issue 和 PR！
+
+## 📄 许可证
+
+MIT License

consultant_agent-0.1.0/README.md

@@ -0,0 +1,152 @@
+# Consultant Agent 🤖
+
+一个专业的咨询顾问 AI Agent，一行命令安装，开箱即用。
+
+[![PyPI version](https://badge.fury.io/py/consultant-agent.svg)](https://pypi.org/project/consultant-agent/)
+[![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+## ✨ 特性
+
+- 🔍 **联网搜索** - 实时获取最新信息
+- 📄 **文档阅读** - 支持 PDF、Word、Excel、PPT
+- 📊 **数据分析** - Excel 数据处理与分析
+- 📝 **内容生成** - 报告大纲、竞品对比、内容总结
+- 🛠️ **易于集成** - 可作为工具嵌入其他 Agent
+- 💾 **对话记忆** - 自动保存上下文
+
+## 🚀 快速开始
+
+### 安装
+
+```bash
+pip install consultant-agent
+```
+
+### 设置 API Key
+
+```bash
+# Linux/Mac
+export DEEPSEEK_API_KEY="sk-your-api-key"
+
+# Windows
+set DEEPSEEK_API_KEY=sk-your-api-key
+```
+
+> 💡 去 [DeepSeek 开放平台](https://platform.deepseek.com/) 获取 API Key
+
+### 使用
+
+**命令行交互**
+```bash
+consultant
+```
+
+**单次提问**
+```bash
+consultant ask "北京今天天气怎么样"
+consultant ask "帮我生成一个 AI Agent 技术发展的报告大纲"
+```
+
+**Python 代码**
+```python
+from consultant_agent import ConsultantAgent
+
+agent = ConsultantAgent()
+reply = agent.ask("分析一下新能源汽车行业趋势")
+print(reply)
+```
+
+## 📖 功能详解
+
+### 命令行命令
+
+| 命令 | 说明 | 示例 |
+|------|------|------|
+| `consultant` | 交互式对话 | - |
+| `consultant ask "问题"` | 单次提问 | `consultant ask "上海天气"` |
+| `consultant clear` | 清空记忆 | - |
+
+### Python API
+
+```python
+from consultant_agent import ConsultantAgent
+
+# 基础用法
+agent = ConsultantAgent()
+reply = agent.ask("你的问题")
+
+# 自定义配置
+agent = ConsultantAgent(
+    api_key="sk-xxx",           # 可选，默认从环境变量读取
+    model="deepseek-chat",      # 模型名称
+    temperature=0,              # 温度参数
+    enable_memory=True,         # 启用记忆
+)
+
+# 清空记忆
+agent.clear_memory()
+```
+
+### 作为工具嵌入其他 Agent
+
+```python
+from consultant_agent import ConsultantAgent
+from langgraph.prebuilt import create_react_agent
+
+# 创建咨询顾问
+consultant = ConsultantAgent()
+
+# 转换为工具
+consultant_tool = consultant.as_tool()
+
+# 在你的 Agent 中使用
+your_agent = create_react_agent(
+    your_llm,
+    tools=[consultant_tool, your_other_tools...]
+)
+```
+
+## 🎯 支持的功能
+
+| 功能 | 说明 | 示例 |
+|------|------|------|
+| 天气查询 | 全球城市天气 | "北京天气" |
+| 数学计算 | 表达式计算 | "计算 123 * 456" |
+| 联网搜索 | DuckDuckGo 搜索 | "搜索 AI Agent 最新进展" |
+| 网页抓取 | 读取网页内容 | "读取 https://example.com" |
+| PDF 阅读 | 提取 PDF 文字 | "读取 report.pdf" |
+| Word 阅读 | 读取 docx 文档 | "读取 document.docx" |
+| Excel 阅读 | 读取表格数据 | "读取 data.xlsx" |
+| PPT 阅读 | 提取幻灯片文字 | "读取 slides.pptx" |
+| 内容总结 | 多种总结风格 | "总结这篇文章" |
+| 报告大纲 | 生成专业大纲 | "生成 XX 报告大纲" |
+| 竞品对比 | 多维度对比表 | "对比 A、B、C 产品" |
+| 案例搜索 | 查找行业案例 | "搜索数字化转型案例" |
+
+## ⚙️ 环境变量
+
+| 变量 | 必须 | 说明 |
+|------|------|------|
+| `DEEPSEEK_API_KEY` | ✅ | DeepSeek API Key |
+
+## 📦 可选依赖
+
+```bash
+# OCR 功能（识别图片/PDF 中的文字）
+pip install consultant-agent[ocr]
+
+# Web 界面
+pip install consultant-agent[web]
+
+# 完整安装
+pip install consultant-agent[ocr,web]
+```
+
+## 🤝 贡献
+
+欢迎 Issue 和 PR！
+
+## 📄 许可证
+
+MIT License

consultant_agent-0.1.0/pyproject.toml

@@ -0,0 +1,77 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "consultant-agent"
+version = "0.1.0"
+description = "一个专业的咨询顾问 AI Agent，支持文档分析、联网搜索、报告生成"
+readme = "README.md"
+license = "MIT"
+requires-python = ">=3.10"
+authors = [
+    { name = "xiaoboluo", email = "your@email.com" }
+]
+keywords = ["ai", "agent", "langchain", "consultant", "llm"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+]
+
+dependencies = [
+    "langchain>=0.2.0",
+    "langchain-openai>=0.1.0",
+    "langgraph>=0.1.0",
+    "requests>=2.31.0",
+    "ddgs>=6.0.0",
+    "beautifulsoup4>=4.12.0",
+    "pdfplumber>=0.10.0",
+    "python-docx>=1.0.0",
+    "python-pptx>=0.6.0",
+    "pandas>=2.0.0",
+    "openpyxl>=3.1.0",
+]
+
+[project.optional-dependencies]
+ocr = [
+    "pytesseract>=0.3.10",
+    "Pillow>=10.0.0",
+    "PyMuPDF>=1.23.0",
+]
+web = [
+    "gradio>=4.0.0",
+    "fastapi>=0.110.0",
+    "uvicorn>=0.27.0",
+]
+dev = [
+    "pytest>=8.0.0",
+    "black>=24.0.0",
+    "ruff>=0.3.0",
+]
+
+# 命令行入口
+[project.scripts]
+consultant = "consultant_agent.cli:main"
+
+[project.urls]
+Homepage = "https://github.com/xiaoboluo/consultant-agent"
+Repository = "https://github.com/xiaoboluo/consultant-agent"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/consultant_agent"]
+
+[tool.hatch.build.targets.sdist]
+exclude = [
+    "venv",
+    ".venv",
+    "__pycache__",
+    "*.pyc",
+    ".git",
+    "dist",
+    "build",
+]

consultant_agent-0.1.0/src/consultant_agent/__init__.py

@@ -0,0 +1,15 @@
+"""
+Consultant Agent - 专业的咨询顾问 AI Agent
+
+快速开始：
+    from consultant_agent import ConsultantAgent
+
+    agent = ConsultantAgent(api_key="sk-xxx")
+    reply = agent.ask("帮我分析新能源汽车行业")
+    print(reply)
+"""
+
+from consultant_agent.agent import ConsultantAgent
+from consultant_agent.version import __version__
+
+__all__ = ["ConsultantAgent", "__version__"]

consultant_agent-0.1.0/src/consultant_agent/agent.py

@@ -0,0 +1,178 @@
+"""
+ConsultantAgent - 核心 Agent 类
+
+可以独立使用，也可以作为工具嵌入其他 Agent。
+"""
+
+import os
+import json
+from datetime import datetime
+from typing import Optional
+from langchain_openai import ChatOpenAI
+from langchain_core.tools import tool
+from langgraph.prebuilt import create_react_agent
+
+from consultant_agent.tools import get_all_tools
+
+
+class ConsultantAgent:
+    """
+    咨询顾问 AI Agent
+
+    Example:
+        >>> agent = ConsultantAgent(api_key="sk-xxx")
+        >>> reply = agent.ask("北京今天天气怎么样")
+        >>> print(reply)
+    """
+
+    def __init__(
+        self,
+        api_key: Optional[str] = None,
+        model: str = "deepseek-chat",
+        base_url: str = "https://api.deepseek.com",
+        temperature: float = 0,
+        memory_file: Optional[str] = None,
+        enable_memory: bool = True,
+    ):
+        """
+        初始化 Agent
+
+        Args:
+            api_key: DeepSeek API Key，不传则从环境变量 DEEPSEEK_API_KEY 读取
+            model: 模型名称
+            base_url: API 地址
+            temperature: 温度参数
+            memory_file: 记忆文件路径，None 则使用默认路径
+            enable_memory: 是否启用记忆功能
+        """
+        # API Key
+        self.api_key = api_key or os.environ.get("DEEPSEEK_API_KEY")
+        if not self.api_key:
+            raise ValueError(
+                "需要提供 api_key 或设置环境变量 DEEPSEEK_API_KEY\n"
+                "export DEEPSEEK_API_KEY='sk-xxx'"
+            )
+
+        # LLM
+        self.llm = ChatOpenAI(
+            model=model,
+            base_url=base_url,
+            api_key=self.api_key,
+            temperature=temperature,
+        )
+
+        # 记忆
+        self.enable_memory = enable_memory
+        if memory_file:
+            self.memory_file = memory_file
+        else:
+            home = os.path.expanduser("~")
+            self.memory_file = os.path.join(home, ".consultant_agent", "memory.json")
+
+        # 确保目录存在
+        os.makedirs(os.path.dirname(self.memory_file), exist_ok=True)
+
+        # 工具
+        self.tools = get_all_tools(self.llm)
+
+        # System Prompt
+        self.system_prompt = """你是一个专业的咨询顾问 AI 助手。
+
+你的能力：
+- 查询天气、计算数学
+- 联网搜索最新信息
+- 读取文档（PDF、Word、Excel、PPT）
+- 总结分析内容、生成报告大纲
+- 竞品对比分析
+
+请用中文回答，保持专业、简洁。
+"""
+
+        # 创建 Agent
+        self._agent = create_react_agent(
+            self.llm,
+            self.tools,
+            prompt=self.system_prompt,
+        )
+
+    def ask(self, question: str) -> str:
+        """
+        向 Agent 提问
+
+        Args:
+            question: 问题或任务
+
+        Returns:
+            Agent 的回复
+        """
+        # 加载记忆
+        memory = self._load_memory() if self.enable_memory else []
+
+        # 构建消息
+        messages = []
+        for msg in memory:
+            messages.append((msg["role"], msg["content"]))
+        messages.append(("user", question))
+
+        # 调用 Agent
+        result = self._agent.invoke({"messages": messages})
+        reply = result["messages"][-1].content
+
+        # 保存记忆
+        if self.enable_memory:
+            memory.append({"role": "user", "content": question})
+            memory.append({"role": "assistant", "content": reply})
+            self._save_memory(memory)
+
+        return reply
+
+    def clear_memory(self):
+        """清空记忆"""
+        self._save_memory([])
+
+    def as_tool(self):
+        """
+        将 Agent 转换为 LangChain Tool，可嵌入其他 Agent
+
+        Example:
+            >>> consultant = ConsultantAgent(api_key="sk-xxx")
+            >>> tool = consultant.as_tool()
+            >>> # 在其他 Agent 中使用
+            >>> other_agent = create_react_agent(llm, [tool, ...])
+        """
+
+        @tool
+        def consultant_agent(question: str) -> str:
+            """
+            向咨询顾问 Agent 提问。擅长：
+            - 读取文档（PDF/Word/Excel/PPT）
+            - 联网搜索最新信息
+            - 生成报告大纲、竞品分析
+            - 内容总结分析
+
+            Args:
+                question: 问题或任务描述
+            """
+            return self.ask(question)
+
+        return consultant_agent
+
+    def _load_memory(self) -> list:
+        """加载记忆"""
+        if os.path.exists(self.memory_file):
+            try:
+                with open(self.memory_file, "r", encoding="utf-8") as f:
+                    data = json.load(f)
+                    return data.get("messages", [])[-100:]  # 最多保留 100 条
+            except:
+                pass
+        return []
+
+    def _save_memory(self, messages: list):
+        """保存记忆"""
+        data = {
+            "messages": messages[-100:],  # 最多保留 100 条
+            "updated_at": datetime.now().isoformat(),
+        }
+        with open(self.memory_file, "w", encoding="utf-8") as f:
+            json.dump(data, f, ensure_ascii=False, indent=2)

consultant_agent-0.1.0/src/consultant_agent/cli.py

@@ -0,0 +1,128 @@
+"""
+命令行入口
+
+用法：
+    consultant              # 交互式对话
+    consultant ask "问题"    # 单次提问
+    consultant web          # 启动 Web 界面
+"""
+
+import argparse
+import sys
+import os
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        prog="consultant",
+        description="咨询顾问 AI Agent - 专业的文档分析和报告生成助手",
+    )
+    subparsers = parser.add_subparsers(dest="command", help="可用命令")
+
+    # ask 命令
+    ask_parser = subparsers.add_parser("ask", help="单次提问")
+    ask_parser.add_argument("question", help="问题内容")
+
+    # web 命令
+    web_parser = subparsers.add_parser("web", help="启动 Web 界面")
+    web_parser.add_argument("--port", type=int, default=7860, help="端口号")
+    web_parser.add_argument("--share", action="store_true", help="生成公网链接")
+
+    # clear 命令
+    clear_parser = subparsers.add_parser("clear", help="清空记忆")
+
+    args = parser.parse_args()
+
+    # 检查 API Key
+    api_key = os.environ.get("DEEPSEEK_API_KEY")
+    if not api_key:
+        print("❌ 错误：请设置环境变量 DEEPSEEK_API_KEY")
+        print("")
+        print("   export DEEPSEEK_API_KEY='sk-xxx'")
+        print("")
+        sys.exit(1)
+
+    if args.command == "ask":
+        # 单次提问
+        from consultant_agent import ConsultantAgent
+
+        agent = ConsultantAgent(api_key=api_key)
+        print(agent.ask(args.question))
+
+    elif args.command == "web":
+        # 启动 Web 界面
+        try:
+            from consultant_agent.web import launch_web
+            launch_web(port=args.port, share=args.share)
+        except ImportError:
+            print("❌ 需要安装 web 依赖：")
+            print("   pip install consultant-agent[web]")
+            sys.exit(1)
+
+    elif args.command == "clear":
+        # 清空记忆
+        from consultant_agent import ConsultantAgent
+
+        agent = ConsultantAgent(api_key=api_key)
+        agent.clear_memory()
+        print("✅ 记忆已清空")
+
+    else:
+        # 交互式对话
+        interactive_mode(api_key)
+
+
+def interactive_mode(api_key: str):
+    """交互式对话模式"""
+    from consultant_agent import ConsultantAgent
+
+    print("=" * 50)
+    print("🤖 咨询顾问 Agent")
+    print("=" * 50)
+    print("输入问题开始对话，输入 'quit' 退出")
+    print("=" * 50)
+    print("")
+
+    agent = ConsultantAgent(api_key=api_key)
+
+    while True:
+        try:
+            user_input = input("你: ").strip()
+
+            if not user_input:
+                continue
+
+            if user_input.lower() in ["quit", "exit", "退出", "bye"]:
+                print("👋 再见！")
+                break
+
+            if user_input == "/clear":
+                agent.clear_memory()
+                print("✅ 记忆已清空\n")
+                continue
+
+            if user_input == "/help":
+                print("""
+可用命令：
+  /clear  - 清空对话记忆
+  /help   - 显示帮助
+  quit    - 退出程序
+
+或者直接输入问题开始对话！
+""")
+                continue
+
+            # 调用 Agent
+            print("思考中...")
+            reply = agent.ask(user_input)
+            print(f"\n助手: {reply}\n")
+
+        except KeyboardInterrupt:
+            print("\n👋 再见！")
+            break
+        except Exception as e:
+            print(f"❌ 出错了：{e}\n")
+
+
+if __name__ == "__main__":
+    main()

consultant_agent-0.1.0/src/consultant_agent/tools/__init__.py

@@ -0,0 +1,21 @@
+"""
+工具集合
+"""
+
+from consultant_agent.tools.basic import get_basic_tools
+from consultant_agent.tools.search import get_search_tools
+from consultant_agent.tools.document import get_document_tools
+from consultant_agent.tools.analysis import get_analysis_tools
+
+
+def get_all_tools(llm=None):
+    """获取所有工具"""
+    tools = []
+    tools.extend(get_basic_tools())
+    tools.extend(get_search_tools())
+    tools.extend(get_document_tools())
+    tools.extend(get_analysis_tools(llm))
+    return tools
+
+
+__all__ = ["get_all_tools"]

consultant_agent-0.1.0/src/consultant_agent/tools/analysis.py

@@ -0,0 +1,111 @@
+"""
+分析工具：总结、报告大纲、竞品对比
+"""
+
+from langchain_core.tools import tool
+
+
+def get_analysis_tools(llm=None):
+    """获取分析工具列表，需要传入 LLM 实例"""
+
+    if llm is None:
+        return []
+
+    @tool
+    def summarize(content: str, style: str = "要点") -> str:
+        """
+        总结归纳内容。
+        content: 需要总结的文本
+        style: 总结风格 - "要点"(3-5个要点)、"简述"(一段话)、"详细"(分层次)
+        """
+        style_prompts = {
+            "要点": "请将以下内容总结为3-5个核心要点：",
+            "简述": "请用一段话（100字以内）概括以下内容：",
+            "详细": "请对以下内容进行分层次的详细总结：",
+        }
+
+        prompt = style_prompts.get(style, style_prompts["要点"])
+
+        try:
+            if len(content) > 6000:
+                content = content[:6000] + "..."
+
+            response = llm.invoke([("user", f"{prompt}\n\n{content}")])
+            return f"📋 总结（{style}模式）：\n\n{response.content}"
+        except Exception as e:
+            return f"总结失败：{e}"
+
+    @tool
+    def generate_outline(topic: str, report_type: str = "分析报告") -> str:
+        """
+        根据主题生成报告大纲。
+        topic: 报告主题
+        report_type: 报告类型 - "分析报告"、"研究报告"、"咨询方案"
+        """
+        prompt = f"""请为主题「{topic}」设计一个专业的{report_type}大纲框架。
+
+要求：
+1. 包含一级和二级标题
+2. 每个部分简要说明应包含的内容
+3. 逻辑严谨，结构完整"""
+
+        try:
+            response = llm.invoke([("user", prompt)])
+            return f"📑 报告大纲 - {topic}\n\n{response.content}"
+        except Exception as e:
+            return f"生成大纲失败：{e}"
+
+    @tool
+    def generate_comparison(topic: str, products: str, dimensions: str = "") -> str:
+        """
+        生成竞品对比表。
+        topic: 对比主题
+        products: 要对比的产品，逗号分隔
+        dimensions: 对比维度，逗号分隔（可选）
+        """
+        prompt = f"""请生成一份专业的竞品对比表。
+
+对比主题：{topic}
+对比产品：{products}
+对比维度：{dimensions if dimensions else "请根据主题自动选择5-8个关键维度"}
+
+要求：
+1. 使用 Markdown 表格格式
+2. 内容具体、有数据支撑
+3. 在表格后添加对比总结（3-5个要点）"""
+
+        try:
+            response = llm.invoke([("user", prompt)])
+            return f"📊 竞品对比 - {topic}\n\n{response.content}"
+        except Exception as e:
+            return f"生成对比表失败：{e}"
+
+    @tool
+    def search_cases(topic: str, industry: str = "") -> str:
+        """
+        搜索相关案例。
+        topic: 案例主题
+        industry: 行业领域（可选）
+        """
+        try:
+            from ddgs import DDGS
+
+            search_query = f"{topic} 案例 成功 企业 {industry}".strip()
+
+            with DDGS() as ddgs:
+                results = list(ddgs.text(search_query, max_results=5))
+
+            if not results:
+                return f"未找到关于 '{topic}' 的案例"
+
+            output = [f"🔍 案例搜索：{topic}"]
+            for i, r in enumerate(results, 1):
+                output.append(f"\n案例 {i}：{r['title']}")
+                output.append(f"来源：{r['href']}")
+                output.append(f"摘要：{r['body'][:200]}...")
+
+            return "\n".join(output)
+        except Exception as e:
+            return f"案例搜索失败：{e}"
+
+    return [summarize, generate_outline, generate_comparison, search_cases]

consultant_agent-0.1.0/src/consultant_agent/tools/basic.py

@@ -0,0 +1,45 @@
+"""
+基础工具：天气、计算
+"""
+
+import requests
+from langchain_core.tools import tool
+
+
+@tool
+def get_weather(city: str) -> str:
+    """查询指定城市的天气。传入城市名称（中文或英文）。"""
+    try:
+        url = f"https://wttr.in/{city}?format=j1"
+        response = requests.get(url, timeout=10)
+        data = response.json()
+
+        current = data["current_condition"][0]
+        weather_desc = current.get("lang_zh", [{}])[0].get(
+            "value", current["weatherDesc"][0]["value"]
+        )
+        temp = current["temp_C"]
+        humidity = current["humidity"]
+        wind = current["windspeedKmph"]
+
+        return f"{city}天气：{weather_desc}，温度 {temp}°C，湿度 {humidity}%，风速 {wind}km/h"
+    except Exception as e:
+        return f"查询天气失败：{e}"
+
+
+@tool
+def calculate(expression: str) -> str:
+    """计算数学表达式。例如：'2+3*4' 或 '100/5' 或 '2**10'"""
+    try:
+        allowed = set("0123456789+-*/.() ")
+        if not all(c in allowed for c in expression):
+            return "只支持数学运算"
+        result = eval(expression)
+        return f"{expression} = {result}"
+    except Exception as e:
+        return f"计算出错：{e}"
+
+
+def get_basic_tools():
+    """获取基础工具列表"""
+    return [get_weather, calculate]

consultant_agent-0.1.0/src/consultant_agent/tools/document.py

@@ -0,0 +1,216 @@
+"""
+文档工具：读取 PDF/Word/Excel/PPT
+"""
+
+import os
+from langchain_core.tools import tool
+
+
+@tool
+def read_file(file_path: str) -> str:
+    """读取本地文本文件内容。传入文件的完整路径。"""
+    try:
+        full_path = os.path.abspath(os.path.expanduser(file_path))
+
+        if not os.path.exists(full_path):
+            return f"文件不存在：{full_path}"
+
+        with open(full_path, "r", encoding="utf-8") as f:
+            content = f.read()
+
+        if len(content) > 5000:
+            return f"文件内容（前5000字）：\n{content[:5000]}..."
+        return f"文件内容：\n{content}"
+    except Exception as e:
+        return f"读取失败：{e}"
+
+
+@tool
+def read_pdf(file_path: str, max_pages: int = 50) -> str:
+    """
+    读取PDF文件内容。
+    file_path: PDF文件路径
+    max_pages: 最多读取的页数，默认50页
+    """
+    try:
+        import pdfplumber
+
+        full_path = os.path.abspath(os.path.expanduser(file_path))
+
+        if not os.path.exists(full_path):
+            return f"文件不存在：{full_path}"
+
+        text_parts = []
+
+        with pdfplumber.open(full_path) as pdf:
+            total_pages = len(pdf.pages)
+            pages_to_read = min(max_pages, total_pages) if max_pages > 0 else total_pages
+
+            for i, page in enumerate(pdf.pages[:pages_to_read]):
+                page_text = page.extract_text()
+                if page_text:
+                    text_parts.append(f"--- 第 {i+1} 页 ---\n{page_text}")
+
+        content = "\n\n".join(text_parts)
+        result = f"📑 PDF文件：{os.path.basename(full_path)}\n📄 共 {total_pages} 页（已读取 {pages_to_read} 页）\n\n{content}"
+
+        if len(result) > 10000:
+            result = result[:10000] + "\n\n...(内容过长，已截断)"
+
+        return result
+
+    except Exception as e:
+        return f"读取PDF失败：{e}"
+
+
+@tool
+def read_docx(file_path: str) -> str:
+    """读取Word文档(.docx)内容。"""
+    try:
+        from docx import Document
+
+        full_path = os.path.abspath(os.path.expanduser(file_path))
+
+        if not os.path.exists(full_path):
+            return f"文件不存在：{full_path}"
+
+        doc = Document(full_path)
+        text_parts = []
+
+        # 读取段落
+        for para in doc.paragraphs:
+            if para.text.strip():
+                text_parts.append(para.text)
+
+        # 读取表格
+        for table in doc.tables:
+            for row in table.rows:
+                row_text = " | ".join(cell.text.strip() for cell in row.cells)
+                if row_text.strip():
+                    text_parts.append(row_text)
+
+        content = "\n\n".join(text_parts)
+        result = f"📄 Word文档：{os.path.basename(full_path)}\n\n{content}"
+
+        if len(result) > 8000:
+            result = result[:8000] + "\n\n...(内容过长，已截断)"
+
+        return result
+
+    except Exception as e:
+        return f"读取Word文档失败：{e}"
+
+
+@tool
+def read_excel(file_path: str, sheet_name: str = None, max_rows: int = 100) -> str:
+    """
+    读取Excel文件内容。
+    file_path: Excel文件路径
+    sheet_name: 工作表名称，默认读取第一个
+    max_rows: 最多读取的行数
+    """
+    try:
+        import pandas as pd
+
+        full_path = os.path.abspath(os.path.expanduser(file_path))
+
+        if not os.path.exists(full_path):
+            return f"文件不存在：{full_path}"
+
+        xlsx = pd.ExcelFile(full_path)
+        sheet_names = xlsx.sheet_names
+
+        result = [f"📊 Excel文件：{os.path.basename(full_path)}"]
+        result.append(f"📑 工作表：{', '.join(sheet_names)}\n")
+
+        target_sheet = sheet_name if sheet_name else sheet_names[0]
+
+        if target_sheet not in sheet_names:
+            return f"工作表 '{target_sheet}' 不存在，可用：{', '.join(sheet_names)}"
+
+        df = pd.read_excel(full_path, sheet_name=target_sheet)
+
+        total_rows = len(df)
+        result.append(f"【{target_sheet}】共 {total_rows} 行 × {len(df.columns)} 列")
+        result.append(f"列名：{', '.join(str(col) for col in df.columns)}\n")
+
+        if total_rows > max_rows:
+            df = df.head(max_rows)
+            result.append(f"(仅显示前 {max_rows} 行)\n")
+
+        result.append(df.to_string(index=False))
+
+        content = "\n".join(result)
+        if len(content) > 10000:
+            content = content[:10000] + "\n\n...(内容过长，已截断)"
+
+        return content
+
+    except Exception as e:
+        return f"读取Excel失败：{e}"
+
+
+@tool
+def read_pptx(file_path: str) -> str:
+    """读取PPT文件内容。"""
+    try:
+        from pptx import Presentation
+
+        full_path = os.path.abspath(os.path.expanduser(file_path))
+
+        if not os.path.exists(full_path):
+            return f"文件不存在：{full_path}"
+
+        prs = Presentation(full_path)
+        output = [f"📊 PPT文件：{os.path.basename(full_path)}"]
+        output.append(f"📄 共 {len(prs.slides)} 页\n")
+
+        for i, slide in enumerate(prs.slides, 1):
+            slide_texts = []
+            for shape in slide.shapes:
+                if hasattr(shape, "text") and shape.text.strip():
+                    slide_texts.append(shape.text.strip())
+
+            output.append(f"\n【第 {i} 页】")
+            if slide_texts:
+                output.append("\n".join(slide_texts))
+            else:
+                output.append("(无文字内容)")
+
+        content = "\n".join(output)
+        if len(content) > 8000:
+            content = content[:8000] + "\n\n...(内容过长，已截断)"
+
+        return content
+
+    except Exception as e:
+        return f"读取PPT失败：{e}"
+
+
+@tool
+def list_files(directory: str) -> str:
+    """列出指定目录下的所有文件和文件夹。"""
+    try:
+        full_path = os.path.abspath(os.path.expanduser(directory))
+
+        if not os.path.exists(full_path):
+            return f"目录不存在：{full_path}"
+
+        items = os.listdir(full_path)
+        if not items:
+            return "目录为空"
+
+        output = []
+        for item in sorted(items):
+            item_path = os.path.join(full_path, item)
+            prefix = "📁" if os.path.isdir(item_path) else "📄"
+            output.append(f"{prefix} {item}")
+
+        return "\n".join(output)
+    except Exception as e:
+        return f"查看失败：{e}"
+
+
+def get_document_tools():
+    """获取文档工具列表"""
+    return [read_file, read_pdf, read_docx, read_excel, read_pptx, list_files]

consultant_agent-0.1.0/src/consultant_agent/tools/search.py

@@ -0,0 +1,78 @@
+"""
+搜索工具：联网搜索、网页抓取
+"""
+
+import requests
+from langchain_core.tools import tool
+
+
+@tool
+def web_search(query: str) -> str:
+    """搜索互联网获取最新信息。传入搜索关键词。"""
+    try:
+        from ddgs import DDGS
+
+        with DDGS() as ddgs:
+            results = list(ddgs.text(query, max_results=5))
+
+        if not results:
+            return f"未找到关于 '{query}' 的结果"
+
+        output = []
+        for i, r in enumerate(results, 1):
+            output.append(
+                f"{i}. {r['title']}\n   链接：{r['href']}\n   摘要：{r['body'][:150]}..."
+            )
+
+        return "\n\n".join(output)
+    except Exception as e:
+        return f"搜索失败：{e}"
+
+
+@tool
+def read_url(url: str) -> str:
+    """抓取网页正文内容。传入完整的URL地址。"""
+    try:
+        from bs4 import BeautifulSoup
+
+        headers = {
+            "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36"
+        }
+        response = requests.get(url, headers=headers, timeout=15)
+        response.encoding = response.apparent_encoding
+
+        soup = BeautifulSoup(response.text, "html.parser")
+
+        # 移除脚本和样式
+        for tag in soup(["script", "style", "nav", "header", "footer", "aside"]):
+            tag.decompose()
+
+        # 获取标题
+        title = soup.title.string if soup.title else "无标题"
+
+        # 获取正文
+        main_content = soup.find("article") or soup.find("main") or soup.find("body")
+
+        if main_content:
+            paragraphs = main_content.find_all(["p", "h1", "h2", "h3", "h4", "li"])
+            text_parts = []
+            for p in paragraphs:
+                text = p.get_text(strip=True)
+                if len(text) > 20:
+                    text_parts.append(text)
+            content = "\n\n".join(text_parts)
+        else:
+            content = soup.get_text(separator="\n", strip=True)
+
+        if len(content) > 5000:
+            content = content[:5000] + "\n\n...(内容过长，已截断)"
+
+        return f"📄 标题：{title}\n🔗 来源：{url}\n\n{content}"
+
+    except Exception as e:
+        return f"抓取网页失败：{e}"
+
+
+def get_search_tools():
+    """获取搜索工具列表"""
+    return [web_search, read_url]

consultant_agent-0.1.0/src/consultant_agent/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

consultant_agent-0.1.0/src/consultant_agent/web.py

@@ -0,0 +1,86 @@
+"""
+Gradio Web 界面
+
+用法：
+    from consultant_agent.web import launch_web
+    launch_web(port=7860)
+"""
+
+import os
+
+
+def launch_web(port: int = 7860, share: bool = False):
+    """启动 Gradio Web 界面"""
+    try:
+        import gradio as gr
+    except ImportError:
+        raise ImportError("需要安装 gradio：pip install consultant-agent[web]")
+
+    from consultant_agent import ConsultantAgent
+
+    # 创建 Agent
+    api_key = os.environ.get("DEEPSEEK_API_KEY")
+    if not api_key:
+        raise ValueError("请设置环境变量 DEEPSEEK_API_KEY")
+
+    agent = ConsultantAgent(api_key=api_key, memory_file=None, enable_memory=False)
+
+    def chat(message: str, history: list) -> str:
+        """Gradio 聊天函数"""
+        if not message.strip():
+            return ""
+
+        try:
+            # 使用历史构建上下文（简化版，不持久化）
+            return agent.ask(message)
+        except Exception as e:
+            return f"出错了：{e}"
+
+    # 示例问题
+    examples = [
+        "北京今天天气怎么样？",
+        "计算 123 * 456 + 789",
+        "搜索一下 2026年AI Agent发展趋势",
+        "帮我生成一个「新能源汽车行业分析」的报告大纲",
+    ]
+
+    # 创建界面
+    with gr.Blocks(title="咨询顾问 Agent", theme=gr.themes.Soft()) as demo:
+        gr.Markdown("""
+        # 🤖 咨询顾问 Agent
+
+        **能力**：天气查询 | 数学计算 | 联网搜索 | 文档阅读 | 内容总结 | 报告大纲 | 竞品对比
+
+        ---
+        """)
+
+        chatbot = gr.Chatbot(height=500, show_label=False)
+
+        with gr.Row():
+            msg = gr.Textbox(
+                placeholder="输入你的问题...",
+                show_label=False,
+                scale=9,
+            )
+            submit_btn = gr.Button("发送", scale=1, variant="primary")
+
+        with gr.Row():
+            clear_btn = gr.Button("🗑️ 清空对话")
+
+        gr.Examples(examples=examples, inputs=msg, label="💡 试试这些问题")
+
+        # 绑定事件
+        msg.submit(chat, [msg, chatbot], [chatbot]).then(lambda: "", None, msg)
+        submit_btn.click(chat, [msg, chatbot], [chatbot]).then(lambda: "", None, msg)
+        clear_btn.click(lambda: ([], ""), None, [chatbot, msg])
+
+    print("=" * 50)
+    print("🚀 咨询顾问 Agent Web 服务")
+    print("=" * 50)
+    print(f"访问地址：http://localhost:{port}")
+    if share:
+        print("正在生成公网链接...")
+    print("按 Ctrl+C 停止服务")
+    print("=" * 50)
+
+    demo.launch(server_name="0.0.0.0", server_port=port, share=share)