Create openai_api.py

openai_api.py

@@ -0,0 +1,29 @@
+import openai
+
+# Set your OpenAI API key
+openai.api_key = os.getenv("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" 
+
+META_PROMPT="You are an AI assistant for the IT Helpdesk at Harvey Mudd College."
+
+def get_embedding(text: str, model: str=EMBEDDING_MODEL) -> list[float]:
+    result = openai.Embedding.create(
+      model=model,
+      input=text
+    )
+    return result["data"][0]["embedding"]
+
+def get_response(prompt: str, model: str=COMPLETIONS_MODEL) -> list[float]:
+    """Chat completion using OpenAI's GPT models."""
+    response = openai.ChatCompletion.create(
+        model = COMPLETIONS_MODEL, 
+        messages=[
+        {"role": "system", "content": META_PROMPT},
+        {"role": "user", "content": prompt}
+        ],
+        temperature=0,
+        max_completion_tokens=800
+    )
+    return response["choices"][0]["message"]["content"]