Update app.py

app.py

@@ -59,6 +59,48 @@ function_descriptions = [
     },
 ]
 
+def get_stock_news(performanceId):
+    url = "https://morning-star.p.rapidapi.com/news/list"
+
+    querystring = {"performanceId":performanceId}
+
+    headers = {
+        "X-RapidAPI-Key": rapid_api_key,
+        "X-RapidAPI-Host": "morning-star.p.rapidapi.com"
+    }
+
+    response = requests.get(url, headers=headers, params=querystring)
+
+    short_news_list = response.json()[:5]
+
+    print("response:", response, " json response:", short_news_list)
+
+    return short_news_list
+
+def get_stock_movers():
+    url = "https://morning-star.p.rapidapi.com/market/v2/get-movers"
+
+    headers = {
+        "X-RapidAPI-Key": rapid_api_key,
+        "X-RapidAPI-Host": "morning-star.p.rapidapi.com"
+    }
+
+    response = requests.get(url, headers=headers)
+    
+    return response.json()
+
+def function_call(ai_response):
+    function_call = ai_response["choices"][0]["message"]["function_call"]
+    function_name = function_call["name"]
+    arguments = function_call["arguments"]
+    if function_name == "get_stock_movers":
+        return get_stock_movers()
+    elif function_name == "get_stock_news":
+        performanceId = eval(arguments).get("performanceId")
+        return get_stock_news(performanceId)
+    else:
+        return
+
 query = "Give me a summary of what happend to the tesla stocks today?"
 messages = [{"role":"user","content":query}]