fixed error

app.py

@@ -9,6 +9,33 @@ import os
 from dotenv import load_dotenv
 from transformers import AutoModelForCausalLM, AutoTokenizer,pipeline
 import torch
+from langchain.prompts import PromptTemplate
+
+prompt_template = PromptTemplate(
+    input_variables=["input"],
+    template="""Answer the following question as best you can. You have access to the following tools:
+
+Stock Data Fetcher(ticker) - Fetch recent stock data for a valid stock ticker (e.g., AAPL for Apple).
+Stock News Fetcher(ticker) - Fetch recent news articles about a stock ticker.
+Moving Average Calculator(ticker, window=5) - Calculate the moving average of a stock over a 5-day window.
+
+Use the following format:
+
+Question: the input question you must answer
+Thought: you should always think about what to do
+Action: the action to take, should be one of [Stock Data Fetcher, Stock News Fetcher, Moving Average Calculator]
+Action Input: the input to the action
+Observation: the result of the action
+... (this Thought/Action/Action Input/Observation can repeat N times)
+Thought: I now know the final answer
+Final Answer: the final answer to the original input question
+
+Begin!
+
+Question: {input}
+"""
+)
+
 
 load_dotenv()
 NEWSAPI_KEY = os.getenv("NEWSAPI_KEY")
@@ -28,10 +55,16 @@ def validate_ticker(ticker):
 
 
 def fetch_stock_data(ticker):
-    ticker = validate_ticker(ticker)  # Validate and clean input
-    stock = yf.Ticker(ticker)
-    hist = stock.history(period="1mo")
-    return hist.tail(5)
+    try:
+        ticker = ticker.strip().upper()
+        stock = yf.Ticker(ticker)
+        hist = stock.history(period="1mo")
+        if hist.empty:
+            return {"error": f"No data found for ticker {ticker}"}
+        return hist.tail(5).to_dict()
+    except Exception as e:
+        return {"error": str(e)}
+
 
 def fetch_stock_news(ticker, NEWSAPI_KEY):
     api_url = f"https://newsapi.org/v2/everything?q={ticker}&apiKey={NEWSAPI_KEY}"
@@ -53,9 +86,10 @@ llm = HuggingFacePipeline(pipeline=pipe)
 stock_data_tool = Tool(
     name="Stock Data Fetcher",
     func=fetch_stock_data,
-    description="Fetch recent stock data for a valid stock ticker (e.g., AAPL for Apple)."
+    description="Fetch recent stock data for a valid stock ticker symbol (e.g., AAPL for Apple)."
 )
 
+
 stock_news_tool = Tool(
     name="Stock News Fetcher",
     func=lambda ticker: fetch_stock_news(ticker, NEWSAPI_KEY),
@@ -74,10 +108,12 @@ agent = initialize_agent(
     tools=tools,
     llm=llm,
     agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+    prompt=prompt_template,
     verbose=True,
-    handle_parsing_errors=True  # Enables automatic handling of parsing issues
+    handle_parsing_errors=True
 )
 
+
 print(fetch_stock_data("AAPL"))
 print(fetch_stock_news("AAPL", NEWSAPI_KEY))
 print(calculate_moving_average("AAPL"))