Update openai_api.py

openai_api.py

@@ -2,14 +2,16 @@ import os
 from openai import OpenAI
 from file_utils import load_file
 
-# Initialize the OpenAI client using the API keys
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+# Ensure you set your API token in your environment variables or pass it directly
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")  # Make sure it's set in your environment
 
-# Load the prompt from a file
-metaprompt_filepath = "meta_prompt.txt"
-META_PROMPT = load_file(metaprompt_filepath)
+if OPENAI_API_KEY is None:
+    raise ValueError("Hugging Face API token is missing. Set it as an environment variable: OPENAI_API_KEY")
+
+
+# Initialize the OpenAI client using the API keys
+client = OpenAI(api_key=OPENAI_API_KEY)
 
-COMPLETIONS_MODEL = "gpt-4o-mini"# "gpt-3.5-turbo-instruct" used earlier .
 EMBEDDING_MODEL = "text-embedding-3-large" # "text-embedding-3-small" 
 
 def get_embedding(text: str, model: str=EMBEDDING_MODEL) -> list[float]:
@@ -20,6 +22,7 @@ def get_embedding(text: str, model: str=EMBEDDING_MODEL) -> list[float]:
     )
     return result.data[0].embedding
 
+COMPLETIONS_MODEL = "gpt-4o-mini"# "gpt-3.5-turbo-instruct" used earlier .
 
 def get_response(messages: list[dict], model: str=COMPLETIONS_MODEL, 
                 temperature=0, max_completion_tokens=800) -> str: