Pass analysis meta data to model

src/functions.py

@@ -4,6 +4,7 @@ import requests
 import pandas as pd
 import yfinance as yf
 import concurrent.futures
+from datetime import datetime
 
 from typing import List
 from bs4 import BeautifulSoup
@@ -34,7 +35,7 @@ def get_analysis(query: str) -> dict:
     results = az_search_client.search(
             query_type="simple", 
             search_text=query, 
-            select="title,content", 
+            select="title,content,asset_name,write_date", 
             include_total_count=True, 
             top=config.az_search_top_k
         )
@@ -42,7 +43,12 @@ def get_analysis(query: str) -> dict:
     output = []
     for x in results:
         if x["@search.score"] >= config.az_search_min_score:
-            output.append({"title": x["title"], "content": x["content"]})
+            output.append({
+                "security": x["asset_name"],
+                "date written": datetime.strptime(x["write_date"], "%Y%m%d").date(),
+                "title": x["title"], 
+                "content": x["content"]
+            })
     return output
 
 @tool