Upload llm_config.py

llm_config.py

@@ -0,0 +1,32 @@
+# llm_config.py
+import json
+import os
+from dotenv import load_dotenv
+from openai import OpenAI
+
+# Load environment variables
+load_dotenv()
+
+# Get the API key from the environment variable
+OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
+
+# Check if the API key is set
+if not OPENAI_API_KEY:
+    raise ValueError("OPENAI_API_KEY environment variable is not set")
+
+# Initialize the OpenAI client
+client = OpenAI(api_key=OPENAI_API_KEY)
+
+def generate_llm_response(prompt):
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.01
+        )
+        content = response.choices[0].message.content.strip()
+        #print(f"LLM Response: {content}")  # For debugging
+        return content
+    except Exception as e:
+        print(f"Error generating LLM response: {str(e)}")
+        return None