Update utils/functions.py

utils/functions.py

@@ -17,31 +17,22 @@ def extract_links(text):
 
 def allocate(messages,api_keys,model,functs):
     helper.models=model
-    try:
-        del helper.data["imageURL"]
-    except:
-        pass
-
-    try:
-        del helper.data["context"]
-    except:
-        pass
-
     for msg in messages:
         if isinstance(msg["content"],list):
-            for msgs in msg["content"]:
-                if msgs["type"]=="image_url":
-                    if "base64," in msgs["image_url"]:
-                        helper.data["imageBase64"]=msgs["image_url"]
-                        print(helper.data["imageBase64"]+"base")
+            msg["content"]=msg["content"][0]["text"]
 
-                    else:
-                        helper.data["imageURL"]=msgs["image_url"]["url"]
-                        print(helper.data["imageURL"]+"a")
+    #         for msgs in msg["content"]:
+    #             if msgs["type"]=="image_url":
+    #                 if "base64," in msgs["image_url"]:
+    #                     helper.data["imageBase64"]=msgs["image_url"]
+    #                     print(helper.data["imageBase64"]+"base")
+
+    #                 else:
+    #                     helper.data["imageURL"]=msgs["image_url"]["url"]
+    #                     print(helper.data["imageURL"]+"a")
 
 
 
-            msg["content"]=msg["content"][0]["text"]
 
 
     for msg in messages:
@@ -85,31 +76,3 @@ def allocate(messages,api_keys,model,functs):
 
 
 
-
-def ask(query,prompt,api_endpoint,output={}):
-  if output=={}:
-    python_boolean_to_json = {
-      "true": True,
-    }
-
-    data = {
-        'jailbreakConversationId': json.dumps(python_boolean_to_json['true']),
-        "systemMessage":prompt,
-        "message":query,
-        "toneStyle":"turbo",
-        "plugins":{"search":False},
-        #     "persona":"sydney"
-
-    }
-    try:
-        data["imageURL"]=helper.data["imageURL"]
-    except:
-        pass
-
-    resp=requests.post(api_endpoint, json=data,timeout=80) 
-
-    return resp.json()["response"]
-  else:
-    resp=requests.post(api_endpoint, json=output) 
-    return resp.json()
-