File size: 1,904 Bytes
ad87194 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
import base64
import string
import requests
import yaml
import configparser
from core.models.utls import FollowUps
from langchain_core.output_parsers import JsonOutputParser
def load_config(file_path):
with open(file_path, 'r') as file:
config = yaml.safe_load(file)
return config
def load_ini_config(file_path):
config = configparser.ConfigParser()
config.read(file_path)
return config
def config_read():
conf = load_ini_config('config.ini')
return conf
def encode_to_base64(dct: dict):
for key in dct:
if type(dct[key]) == str:
dct[key] = base64.b64encode(dct[key].encode("utf-8", errors="replace")).decode("utf-8", errors="replace")
elif type(dct[key]) == dict:
dct[key] = encode_to_base64(dct[key])
return dct
def decode_base64(dct: dict):
if type(dct["output"]) == str:
dct["output"] = base64.b64decode(dct["output"].encode("utf-8", errors="replace")).decode("utf-8",
errors="replace")
else:
for key in dct["output"]:
dct["output"][key] = base64.b64decode(dct["output"][key].encode("utf-8", errors="replace")).decode("utf-8",
errors="replace")
return dct
def get_ip_info(ip: str):
try:
response = requests.get(f"https://ipinfo.io/{ip}/json")
data = response.json()
return data.get("city", "Unknown")
except Exception as e:
return "Unknown"
def json_parser():
json_parser = JsonOutputParser(pydantic_object=FollowUps)
return json_parser
def clean_text(text: str):
text = text.replace("\n", " ")
text = text.translate(str.maketrans('', '', string.punctuation.replace(".", "")))
return text
|