Spaces:
Running
Running
| import re | |
| import helpers.helper as helper | |
| import requests | |
| import json | |
| import os | |
| from function_support import _function | |
| def extract_links(text): | |
| # Regular expression pattern to match URLs | |
| url_pattern = r'http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\\(\\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+' | |
| # Find all matches of the URL pattern in the text | |
| urls = re.findall(url_pattern, text) | |
| return urls | |
| def allocate(messages,api_keys,model,functs): | |
| helper.models=model | |
| if "gemini" not in model: | |
| for msg in messages: | |
| if isinstance(msg["content"],list): | |
| msg["content"]=msg["content"][0]["text"] | |
| # 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") | |
| for msg in messages: | |
| if "tool" in msg["role"]: | |
| msg["role"]="user" | |
| msg["content"]=f"Tool {msg['name']} returned response: {msg['content']}. Now you must output the next tool Call or respond to user in natural language after the task has been completed. " | |
| del msg['name'] | |
| del msg["tool_call_id"] | |
| if "tool_calls" in msg: | |
| add="" | |
| for tools in msg["tool_calls"]: | |
| add=f""" | |
| ```json | |
| [ | |
| {{ | |
| "tool":"{tools["function"]["name"]}", | |
| "tool_input":{tools["function"]["arguments"]} | |
| }} | |
| ] | |
| ```""" | |
| msg["content"]=add | |
| del msg["tool_calls"] | |
| if functs !=[]: | |
| print("ADDDDDDDDDDDDDDDDING TOOOOLS") | |
| function_call=_function(tools=functs) | |
| messages.insert(1,{"role": "system", "content": function_call}) | |
| # print(messages) | |
| # file=open("static/messages.json","a") | |
| # file.write(str(messages) ) | |
| # file.close() | |
| helper.filen=[] | |