Add yf_selected_expiration

app.py

@@ -63,6 +63,24 @@ def yf_underlying_price(ticker: str, dte: int = 0) -> float:
     underlying_price, _, _, _, _ = _get_atm_option_data(ticker, dte)
     return underlying_price
 
+@tool
+def yf_selected_expiration(ticker: str, dte: int = 0) -> str:
+    """
+    Return the option expiration date selected for the given stock ticker and
+    desired days-to-expiration (DTE). The expiration chosen is the one whose
+    days-to-expiration is closest to the requested DTE.
+
+    Args:
+        ticker (str): The stock ticker symbol, e.g., AAPL.
+        dte (int): Desired days-to-expiration. A value of 0 selects the nearest
+            available expiration. Defaults to 0.
+
+    Returns:
+        str: The selected expiration date in YYYY-MM-DD format.
+    """
+    _, expiration, _, _, _ = _get_atm_option_data(ticker, dte)
+    return expiration
+
 @tool
 def yf_atm_strike(ticker: str, dte: int = 0) -> float:
     """
@@ -155,7 +173,6 @@ def yf_atm_straddle_price(ticker: str, dte: int = 0) -> float:
     put_price = float(put_row.get("lastPrice", 0.0))
     return call_price + put_price
 
-
 ddgs = DuckDuckGoSearchTool();
 
 final_answer = FinalAnswerTool()
@@ -170,7 +187,6 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may
 custom_role_conversions=None,
 )
 
-
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -189,5 +205,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-
 GradioUI(agent).launch()