Update openai_.py

openai_.py

@@ -2,21 +2,16 @@ import openai
 
 from IPython.display import display, HTML
 
-openai.api_key = OPENAI_API_KEY
+openai.api_key = os.environ["OPENAI_API_KEY"]
 
 def get_embedding(text):
-    """Generate an embedding for the given text using OpenAI's API."""
-
-    # Check for valid input
     if not text or not isinstance(text, str):
         return None
 
     try:
-        # Call OpenAI API to get the embedding
-        embedding = openai.embeddings.create(
+        return openai.embeddings.create(
             input=text,
             model="text-embedding-3-small", dimensions=1536).data[0].embedding
-        return embedding
     except Exception as e:
         print(f"Error in get_embedding: {e}")
         return None