Commit
·
2b51c5d
1
Parent(s):
d375321
Open AI update
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import streamlit as st
|
2 |
from sentence_transformers import SentenceTransformer
|
3 |
from sklearn.metrics.pairwise import cosine_similarity
|
4 |
-
import
|
5 |
import numpy as np
|
6 |
import pandas as pd
|
7 |
|
@@ -69,15 +69,17 @@ if st.button("Calculate Similarities"):
|
|
69 |
if not openai_api_key:
|
70 |
st.error("Please provide your OpenAI API key for this model.")
|
71 |
else:
|
72 |
-
openai.api_key = openai_api_key
|
73 |
st.info("Generating OpenAI embeddings...")
|
74 |
|
|
|
|
|
|
|
75 |
def get_openai_embedding(text):
|
76 |
-
response =
|
77 |
model="text-embedding-ada-002",
|
78 |
input=text
|
79 |
)
|
80 |
-
return np.array(response
|
81 |
|
82 |
primary_embedding = get_openai_embedding(primary_keyword)
|
83 |
keyword_embeddings = np.array([get_openai_embedding(kw) for kw in keyword_list])
|
|
|
1 |
import streamlit as st
|
2 |
from sentence_transformers import SentenceTransformer
|
3 |
from sklearn.metrics.pairwise import cosine_similarity
|
4 |
+
from openai import OpenAI
|
5 |
import numpy as np
|
6 |
import pandas as pd
|
7 |
|
|
|
69 |
if not openai_api_key:
|
70 |
st.error("Please provide your OpenAI API key for this model.")
|
71 |
else:
|
|
|
72 |
st.info("Generating OpenAI embeddings...")
|
73 |
|
74 |
+
# Initialize OpenAI client with new v1.0.0 interface
|
75 |
+
client = OpenAI(api_key=openai_api_key)
|
76 |
+
|
77 |
def get_openai_embedding(text):
|
78 |
+
response = client.embeddings.create(
|
79 |
model="text-embedding-ada-002",
|
80 |
input=text
|
81 |
)
|
82 |
+
return np.array(response.data[0].embedding)
|
83 |
|
84 |
primary_embedding = get_openai_embedding(primary_keyword)
|
85 |
keyword_embeddings = np.array([get_openai_embedding(kw) for kw in keyword_list])
|