ReithBjarkan commited on
Commit
2b51c5d
·
1 Parent(s): d375321

Open AI update

Browse files
Files changed (1) hide show
  1. app.py +6 -4
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 openai
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 = openai.Embedding.create(
77
  model="text-embedding-ada-002",
78
  input=text
79
  )
80
- return np.array(response['data'][0]['embedding'])
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])