chat / utilities.py
aiXpert's picture
v1
5316f23
raw
history blame
1.23 kB
import requests, openai
from tenacity import retry, wait_random_exponential, stop_after_attempt
client = openai.OpenAI()
@retry(wait=wait_random_exponential(multiplier=1, max=40), stop=stop_after_attempt(3))
def aichat(messages, openai_api_key):
try:
client = openai.OpenAI(api_key = openai_api_key)
response = client.chat.completions.create(
messages=messages,
model="gpt-3.5-turbo-0125",
stream=True,
)
return response
except Exception as e:
print("Unable to generate ChatCompletion response")
print(f"Exception: {e}")
return e
def get_products():
# url = "https://hypech.com/StoreSpark/product_short.json"
url = "https://hypech.com/StoreSpark/products.json"
response = requests.get(url)
if response.status_code == 200:
data = response.text
return data
else:
print(f"The store is closed:{response.status_code}")
def get_embedding(text, model="text-embedding-3-small"):
text = text.replace("\n", " ")
return client.embeddings.create(input = [text], model=model).data[0].embedding
# text = "test embedding"
# embeddings = get_embedding(text)