Update utils.py

utils.py

@@ -23,11 +23,14 @@ def callAzure(prompt,text):
     url = "https://Meta-Llama-3-70B-Instruct-fkqip-serverless.eastus2.inference.ai.azure.com"
     api_key = "o5yaLhTIvg0s5zuYVInBpyneEZO8oonY"
     client = OpenAI(base_url=url, api_key=api_key)
+    msg = "{} {}".format(prompt, text)
+    msg = msg[:7000]
+    
     response = client.chat.completions.create(
         messages=[
             {
                 "role": "user",
-                "content": "{} {}".format(prompt, text),
+                "content": ,
             }
         ],
         model="azureai",
@@ -99,13 +102,19 @@ def summary(input_json):
     output["Date of time of receiving data from BSE"] = input_json["newsdate"]
     
     output["Stock Ticker"] = input_json['symbol']
-    
-    output['Short Summary'] = callAzure(promptShort[id], long_text)
 
-    output['Long summary'] = callAzure(promptLong[id], long_text)
+    answer = callAzure(promptShort[id], long_text)
+    idx = answer.index("\n")
+    output['Short Summary'] = answer[idx+2:]
+
+    answer = callAzure(promptLong[id], long_text)
+    idx = answer.index("\n")
+    output['Long summary'] = answer[idx+2:]
 
     prompt = "1 word Financial SEO tag for this news article"
-    output['Tag'] = callAzure(prompt, output['Short Summary'])
+    answer = callAzure(prompt, output['Short Summary'])
+    idx = answer.index("\n")
+    output['Tag'] = answer[idx+2:]
 
     prompt = "Give a single headline for this News Article"
     output['Headline'] = callAzure(prompt, output['Short Summary'])