Spaces:
Build error
Build error
Update openai_api.py
Browse files- 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(
|