github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-08 11:47:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

bugfix(ChatExcel): ChatExcel Language confusion bug

Browse Source 
1.Fix ChatExcel Language confusion bug
This commit is contained in:
yhjun1026 2023-10-30 11:09:42 +08:00
parent eab1993540
commit 3dbf0cf34b
2 changed files with 1 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pilot/scene/chat_agent/coding/stock_price_chart.py
View File

@ -1,18 +0,0 @@
# filename: stock_price_chart.py
import yfinance as yf
import matplotlib.pyplot as plt
# Define the ticker symbols for NVDA and TESLA
tickers = ["NVDA", "TSLA"]
# Fetch the historical stock price data for YTD
data = yf.download(tickers, start="2021-01-01", end="2021-12-31")
# Extract the 'Close' price column for each ticker
close_prices = data["Close"]
# Plot the stock price change YTD
close_prices.plot(title="Stock Price Change YTD", ylabel="Price (USD)")
# Display the chart
plt.show()

2
pilot/scene/chat_data/chat_excel/excel_analyze/chat.py
View File

@ -62,7 +62,7 @@ class ChatExcel(BaseChat):
# ] # ]
return "\n".join(f"{i+1}. {item}" for i, item in enumerate(command_strings)) return "\n".join(f"{i+1}. {item}" for i, item in enumerate(command_strings))
def generate_input_values(self): async def generate_input_values(self) -> Dict:
input_values = { input_values = {
"user_input": self.current_user_input, "user_input": self.current_user_input,
"table_name": self.excel_reader.table_name, "table_name": self.excel_reader.table_name,