54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
Data Analysis Agent Package
|
||
|
||
一个基于LLM的智能数据分析代理，专门为Jupyter Notebook环境设计。
|
||
"""
|
||
|
||
from .core.notebook_agent import NotebookAgent
|
||
from .config.llm_config import LLMConfig
|
||
from .utils.code_executor import CodeExecutor
|
||
|
||
__version__ = "1.0.0"
|
||
__author__ = "Data Analysis Agent Team"
|
||
|
||
# 主要导出类
|
||
__all__ = [
|
||
"NotebookAgent",
|
||
"LLMConfig",
|
||
"CodeExecutor",
|
||
]
|
||
|
||
# 便捷函数
|
||
def create_agent(config=None, output_dir="outputs", max_rounds=20, session_dir=None):
|
||
"""
|
||
创建一个数据分析智能体实例
|
||
|
||
Args:
|
||
config: LLM配置，如果为None则使用默认配置
|
||
output_dir: 输出目录
|
||
max_rounds: 最大分析轮数
|
||
session_dir: 指定会话目录（可选）
|
||
|
||
Returns:
|
||
NotebookAgent: 智能体实例
|
||
"""
|
||
if config is None:
|
||
config = LLMConfig()
|
||
return NotebookAgent(config=config, output_dir=output_dir, max_rounds=max_rounds, session_dir=session_dir)
|
||
|
||
def quick_analysis(query, files=None, output_dir="outputs", max_rounds=10):
|
||
"""
|
||
快速数据分析函数
|
||
|
||
Args:
|
||
query: 分析需求（自然语言）
|
||
files: 数据文件路径列表
|
||
output_dir: 输出目录
|
||
max_rounds: 最大分析轮数
|
||
|
||
Returns:
|
||
dict: 分析结果
|
||
"""
|
||
agent = create_agent(output_dir=output_dir, max_rounds=max_rounds)
|
||
return agent.analyze(query, files) |