PandasAI接入星火大模型

原创

已于 2025-02-20 13:11:19 修改 · 324 阅读

标签

#语言模型

收录于

于 2025-02-19 14:39:17 首次发布

import pandas as pd
from pandasai.llm.openai import OpenAI
from pandasai import Agent
# 示例表格
sales_by_country = pd.DataFrame({
            "国家": ["United States", "United Kingdom", "France", "Germany", "Italy", "Spain", "Canada", "Australia", "Japan", "China"],
                        "GDP": [19294482071552, 2891615567872, 2411255037952, 3435817336832, 1745433788416, 1181205135360, 1607402389504, 1490967855104, 4380756541440, 14631844184064],
                                        "幸福度": [6.94, 7.16, 6.66, 7.07, 6.38, 6.4, 7.23, 7.22, 5.87, 5.12]
                                                        })
url='https://spark-api-open.xf-yun.com/v1/'
API_PASSWD ='xxxx'
local_model='SparkDesk'
llm = OpenAI(api_base=url,
            api_token=API_PASSWD,
            model=local_model)
agent = Agent(dfs = [sales_by_country],config={'llm':llm } )
ans = agent.chat('哪五个国家GDP最高的？')
print(ans)