AashitaK commited on
Commit
532b4ba
·
verified ·
1 Parent(s): ae6a546

Update openai_api.py

Browse files
Files changed (1) hide show
  1. openai_api.py +8 -1
openai_api.py CHANGED
@@ -4,10 +4,17 @@ import openai
4
  # Set your OpenAI API key
5
  openai.api_key = os.getenv("OPENAI_API_KEY")
6
 
 
 
 
 
 
 
 
 
7
  COMPLETIONS_MODEL = "gpt-4o-mini"# "gpt-3.5-turbo-instruct" used earlier .
8
  EMBEDDING_MODEL = "text-embedding-3-large" # "text-embedding-3-small"
9
 
10
- META_PROMPT="You are an AI assistant for the IT Helpdesk at Harvey Mudd College."
11
 
12
  def get_embedding(text: str, model: str=EMBEDDING_MODEL) -> list[float]:
13
  result = openai.Embedding.create(
 
4
  # Set your OpenAI API key
5
  openai.api_key = os.getenv("OPENAI_API_KEY")
6
 
7
+ def load_meta_prompt(file_path: str) -> str:
8
+ """Reads the META_PROMPT from a file."""
9
+ with open(file_path, "r") as file:
10
+ return file.read()
11
+
12
+ # Load the prompt from a file
13
+ META_PROMPT = load_meta_prompt("meta_prompt.txt")
14
+
15
  COMPLETIONS_MODEL = "gpt-4o-mini"# "gpt-3.5-turbo-instruct" used earlier .
16
  EMBEDDING_MODEL = "text-embedding-3-large" # "text-embedding-3-small"
17
 
 
18
 
19
  def get_embedding(text: str, model: str=EMBEDDING_MODEL) -> list[float]:
20
  result = openai.Embedding.create(