d

app.py

@@ -65,7 +65,10 @@ def extract_json(gen_text, n_shot_learning=0):
         for i in range(0, n_shot_learning):
             gen_text = gen_text[start_index:]
             start_index = gen_text.index("### Response:\n{") + 14
-    end_index = gen_text.find("}\n\n### ") + 1
+    end_index = gen_text.find("}\n\n### ")
+    if(end_index != -1) :
+        end_index = end_index + 1
+
     return gen_text[start_index:end_index]
 
 def score_model(model_uri, databricks_token, prompt):
@@ -87,7 +90,16 @@ def score_model(model_uri, databricks_token, prompt):
   response = requests.request(method='POST', headers=headers, url=model_uri, data=data_json)
   if response.status_code != 200:
       raise Exception(f"Request failed with status {response.status_code}, {response.text}")
-  return response.json()
+
+  response_json = response.json()
+
+  if(response_json["choices"] == "") :
+      print("Start***No choices!!!")
+      print(response.status_code)
+      print(response.txt)
+      print("End***No choices!!!")
+
+  return response_json
 
 def get_completion(prompt):
     return score_model(model_uri, databricks_token, prompt)