Update app.py

app.py

@@ -31,6 +31,31 @@ def difference_of_two_numbers(arg1:int, arg2:int)-> int:
     diff = arg1 - arg2
     return diff
 
+@tool
+def nse_stock_price_tool(stock_ticker:str)-> str: 
+    """A tool that is used to retrieve the latest closing price of a stock based on the stock ticker from the Indian Stock Exchange (NSE).
+    example: 'RELIANCE.NS'
+    Args:
+        stock_ticker: The stock ticker/symbol for the stock listed in NSE. The exact string input needs to be provided to the function.
+    Returns:
+        result_string: A string containing the stock_ticker along with the amount in Rupees.
+        example: "RELIANCE NS: 1237.22 Rupees."
+    """
+    try:
+        import math
+        import yfinance as yf
+        stock_symbol = stock_ticker
+        data = yf.download(tickers=stock_symbol, period='1d', interval='1m')
+        if len(data):
+            latest_data = data.tail(1)
+            latest_price = latest_data['Close'].values[0]   
+            return (f"{stock_symbol}: {round(latest_price, 2)} Rupees.")
+        else:
+            return "INVALID TICKER/STOCK"
+    except Exception as e:
+        traceback.print_exc()
+        return "INVALID REQUEST"
+        
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -50,7 +75,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
 final_answer = FinalAnswerTool()
 model = HfApiModel(
 max_tokens=2096,
-temperature=0,
+temperature=0.5,
 model_id='meta-llama/Meta-Llama-3-8B-Instruct',
 custom_role_conversions=None,
 )
@@ -62,10 +87,16 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
+
 agent = CodeAgent(
     model=model,
-    tools=[sum_of_two_numbers, difference_of_two_numbers, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
+    tools=[nse_stock_price_tool, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
+    verbosity_level=1,
+    grammar=None,
+    planning_interval=None,
+    name=None,
+    description=None,
     prompt_templates=prompt_templates
 )