akshare-gpt

akshare-gpt

akshare-gpt 是一个开源工具，旨在将 Akshare 集成到 GPT 的工具中，实现自然语言问答。通过这个工具，用户可以方便地获取金融数据、经济指标和其他相关信息。

特性

• 自然语言处理：用户可以使用自然语言提出问题，系统将自动解析并返回相应的数据。
• 金融数据访问：直接从 Akshare 获取股票、期货、外汇等多种金融数据。
• 简化接口：封装了 Akshare 的复杂调用，使得数据获取更加简单直观。

安装

git clone https://github.com/chenwr727/akshare-gpt.git
cp config-template.toml config.toml
pip install -r requirements.txt

运行

streamlit run main.py

使用方法

定义 tool

from typing import Annotated

import akshare as ak

from modules.tools.tool_register import register_tool


@register_tool
def get_stock_sse_summary() -> str:
    """
    获取上海证券交易所最近交易日的股票数据总貌
    """
    df = ak.stock_sse_summary()
    return df.to_json(orient="records", force_ascii=False)

参考./modules/tools/akshare/stock.py

调用 openai

from datetime import datetime

from modules.config import load_config
from modules.openai_assistant import OpenAIAssistant

openai_config = load_config()["openai"]
assistant = OpenAIAssistant(
    openai_config["api_key"], openai_config["base_url"], openai_config["model"]
)

messages = [
    {
        "role": "system",
        "content": "你是一个智能助手，今天日期是{}".format(
            datetime.now().strftime("%Y-%m-%d")
        ),
    },
    {
        "role": "user",
        "content": "获取000001的行情报价数据",
    },
]
print(assistant.run_conversation(messages))