new tool added

data_ingetion/data_api.py

@@ -1,6 +1,6 @@
 from fastapi import APIRouter
 from pydantic import BaseModel
-from .market_data import price_change, earning_summary, portfolio_data
+from .market_data import price_change, earning_summary, portfolio_data, get_update
 from .vectroDB import get_relevant_chunks
 
 app = APIRouter()
@@ -43,3 +43,8 @@ def get_portfolio_data(req: PortfolioReq):
 @app.post("/get_knowledge")
 def get_knowledge(req: KnowledgeReq):
     return {"response": get_relevant_chunks(req.query)}
+
+
+@app.post("/get_update")
+def get_ticker_update(req: EarningReq):
+    return {"response": get_update(req.symbol)}

data_ingetion/market_data.py

@@ -5,6 +5,7 @@ import os
 
 
 def portfolio_data(region: str = "IND"):
+    region = "IND"
     if region == "IND":
         portfolio = "/data_ingetion/portfolios/IND.csv"
     else:
@@ -82,3 +83,19 @@ def earning_summary(symbol):
     response = f"Earning metrics for {symbol} are following in {currency} currency in {units}: \n {selected_metric}"
 
     return response
+
+
+def get_update(symbol):
+    stock = yf.Ticker(ticker=symbol)
+    data = stock.news
+    news = ""
+
+    for info in data[:5]:
+        news += info["content"]["title"] + "\n"
+        news += info["content"]["summary"] + "\n\n"
+
+    pc = price_change(symbol, 7)
+
+    response = f"Price change for {symbol} in past 7 days is {pc} \n\n NEWS :\n\n{news}"
+
+    return response

orchestrator/orchestrator.py

@@ -47,6 +47,11 @@ def get_orchertration_resposne(query, history):
             url="http://127.0.0.1:7860/data/get_knowledge", json=data["parameters"]
         ).json()
 
+    elif data["tool"] == "get_update":
+        result = requests.post(
+            url="http://127.0.0.1:7860/data/get_update", json=data["parameters"]
+        ).json()
+
     elif data["tool"] == None:
         return data["parameters"]
 

orchestrator/prompts.py

@@ -19,6 +19,8 @@ Tools available:
         get_knowledge() -> use this tool to get prior knowledge as context about the firm, its a RAG based tool ie. you will give the augmented user query for better retrival results.
             parameters -> query:str User query augmented/expanded by you for better retrival results
 
+        get_update() -> use this tool to get latest news and price movement in the stock/index. This information can be helpful in providing sentiments around the stock or index. Can also be used to tell general market trends by getting info on indexes like NIFTY50 (yfinance - ^NSEI).
+            parameters -> symbol:str the symbol of the stock/index yfinance style.
 
 
 You have to respond in structured json format, mentioning tool name and prameter json.
@@ -47,9 +49,19 @@ response :
         }
     }
 
+Query : How does general market looks like ?
+
+response : 
+    {
+    "tool":"get_update",
+    "parameters" : {
+        "response" : "^NSEI"
+        }
+    }
+
 
 Dont add any comments around json, you should only respond in valid json format only.
 """
 FINAL_SYS_PROMPT = """
-You task to to generate final response of a long workflow/reseach. You will be provided with Query that is the original query and some context that is derived from different tools your task is to create a condensed output to effectively answer the user query. Try to be consise and to the point. Also not necessarily but do add disclamers if there is any uncertanity. Stay respectful and official tone. Talk like a financial expert.
+You task to to generate final response of a long workflow/reseach. You will be provided with Query that is the original query and some context that is derived from different tools your task is to create a condensed output to effectively answer the user query. Try to be consise and to the point. Also not necessarily but every now and then add disclamers if there is any uncertanity. Stay respectful and official tone. Talk like a financial expert.
 """