Update main.py

main.py

@@ -93,38 +93,21 @@ async def core():
 def getPrompt(prompt, modelID, attempts=1):
     response = {}
     print(modelID)
-    if modelID != magic_prompt_model:
-        chat = [
-            {"role": "user", "content": prompt_base},
-            {"role": "assistant", "content": prompt_assistant},
-            {"role": "user", "content": prompt},
-            ]
-        try:
+    try:  
+        if modelID != magic_prompt_model:
+            chat = [
+                {"role": "user", "content": prompt_base},
+                {"role": "assistant", "content": prompt_assistant},
+                {"role": "user", "content": prompt},
+                ]
             response = groqClient.chat.completions.create(messages=chat, temperature=1, max_tokens=2048, top_p=1, stream=False, stop=None, model=modelID) 
-        except Exception as e:
-            print(f"An error occurred: {e}")
-            if attempts < 3:
-                getPrompt(prompt, modelID, attempts + 1) 
-    else:
-        try:    
+        else:
             apiData={"inputs":prompt, "parameters": parameters, "options": options, "timeout": 45}
             response = requests.post(API_URL + modelID, headers=headers, data=json.dumps(apiData))
-            print(response)
-            if response.status_code == 200:
-                try:
-                    responseData = response.json()
-                    return responseData
-                except ValueError as e:
-                    print(f"Error parsing JSON: {e}")
-            else:
-                print(f"Error from API: {response.status_code} - {response.text}")
-                if attempts < 3:
-                    getPrompt(prompt, modelID, attempts + 1)
-        except Exception as e:
-            print(f"An error occurred: {e}")
-            if attempts < 3:
-                getPrompt(prompt, modelID, attempts + 1)
-    print(str(response.json()))
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        if attempts < 3:
+            getPrompt(prompt, modelID, attempts + 1)
     return response.json()
 
 @app.post("/inferencePrompt")
@@ -133,12 +116,7 @@ def inferencePrompt(item: Core):
     try:
         plain_response_data = getPrompt(item.itemString, prompt_model)
         magic_response_data = getPrompt(item.itemString, magic_prompt_model)
-        print(plain_response_data['choices'])
-        print(plain_response_data['choices'][0])
-        print(plain_response_data['choices'][0]['message'])
         plain_response = plain_response_data['choices'][0]['message']['content']
-        print(plain_response)
-        print(magic_response_data[0]["generated_text"])
         returnJson = {"plain": plain_response, "magic": item.itemString + magic_response_data[0]["generated_text"]}
         print(f'Return Json {returnJson}')
         return returnJson