openai error handling

aiproductguy · Nov 10, 2024 · c756ad2 · c756ad2
1 parent 571622d
commit c756ad2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/notebooks/streamlit-app-lightrag.py b/notebooks/streamlit-app-lightrag.py
@@ -714,7 +714,11 @@ def rewrite_prompt(prompt: str) -> str:
     """Rewrite the user prompt into a templated format using OpenAI."""
     try:
         from openai import OpenAI
-        client = OpenAI(api_key=st.session_state.settings["api_key"])
+        api_key = get_api_key()  # Get API key using our utility function
+        if not api_key:
+            raise ValueError("OpenAI API key not found")
+
+        client = OpenAI(api_key=api_key)
 
         system_instruction = f"""
         You are a prompt engineering assistant. Your task is to rewrite user prompts into a templated format.