Update app.py

app.py

@@ -7,30 +7,6 @@ from smolagents import CodeAgent, HfApiModel, load_tool, tool
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# A dummy tool that does nothing (for demonstration)
-@tool
-def my_custom_tool(arg1: str, arg2: int) -> str:
-    """A tool that does nothing yet.
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return "What magic will you build ?"
-
-# Tool to get the current time in a given timezone.
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """Fetches the current local time in the specified timezone.
-    Args:
-        timezone: A valid timezone string (e.g., 'America/New_York').
-    """
-    try:
-        tz = pytz.timezone(timezone)
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
 # Tool to fetch the current stock price using yfinance.
 @tool
 def get_current_stock_price(symbol: str) -> str:
@@ -51,13 +27,13 @@ def get_current_stock_price(symbol: str) -> str:
     except Exception as e:
         return f"Error fetching current price for {symbol}: {str(e)}"
 
-# Tool to fetch and summarize historical stock data using yfinance.
+# Tool to fetch historical stock data for a given timeline using yfinance.
 @tool
 def get_stock_historical_data(symbol: str, timeline: str) -> str:
     """Fetches historical stock data for a given symbol and timeline.
     
     Args:
-        symbol: The stock ticker symbol (e.g., 'AAPL').
+        symbol: The stock ticker symbol (e.g., 'TSLA').
         timeline: The timeframe for data. Acceptable values: '1h', '1day', '1week', '1year'.
         
     Returns:
@@ -95,34 +71,35 @@ def get_stock_historical_data(symbol: str, timeline: str) -> str:
     except Exception as e:
         return f"Error fetching historical data for {symbol} over {timeline}: {str(e)}"
 
-# Final answer tool that should always be included.
+# Final answer tool (must be included)
 final_answer = FinalAnswerTool()
 
-# Define the model with appropriate configuration.
+# Define the model configuration.
+# Here we reduce max_tokens to help keep the total token count below the limit.
 model = HfApiModel(
-    max_tokens=2096,
+    max_tokens=1000,  # Reduced from 2096 to reduce total token count
     temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # if overloaded, consider an alternative endpoint
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
     custom_role_conversions=None,
 )
 
-# Optional: Load an additional tool from the Hub (e.g., a text-to-image tool)
+# Optionally, load an additional tool from the Hub (e.g., a text-to-image tool)
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Load prompt templates from a YAML file.
+# Load prompt templates from YAML.
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
-# Initialize the CodeAgent with the new yfinance tools along with other tools.
+# Initialize the CodeAgent with the new yfinance tools.
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, get_current_stock_price, get_stock_historical_data, my_custom_tool],
-    max_steps=6,
+    tools=[final_answer, get_current_stock_price, get_stock_historical_data],
+    max_steps=3,  # Reduced steps to limit the chain-of-thought length
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name="TradingAgent",
-    description="An AI agent that analyzes stock data using yfinance and advises on long-term trading decisions (buy, sell, or hold).",
+    description="An AI agent that analyzes stock data using yfinance and advises on long-term trading decisions.",
     prompt_templates=prompt_templates
 )