-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
streamlit_app.py
34 lines (29 loc) · 1.15 KB
/
streamlit_app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import openai, streamlit as st
from langchain.llms.openai import OpenAI
openai.api_base = "https://oai.hconeai.com/v1"
# Streamlit app
st.title('LLM Observability Demo')
# Get OpenAI API key, Helicone API key, and user query
openai_api_key = st.text_input("OpenAI API Key", type="password")
helicone_api_key = st.text_input("Helicone API Key", type="password")
user_query = st.text_input("Your Query")
# If the 'Submit' button is clicked
if st.button("Submit"):
# Validate inputs
if not openai_api_key.strip() or not helicone_api_key.strip() or not user_query.strip():
st.error(f"Please provide the missing fields.")
else:
try:
# Initialize OpenAI model with Helicone integration
llm = OpenAI(
temperature=0.9,
openai_api_key=openai_api_key,
headers={
"Helicone-Auth": f"Bearer {helicone_api_key}",
"Helicone-Cache-Enabled": "true"
}
)
# Run user query and display response
st.success(llm(user_query))
except Exception as e:
st.error(f"An error occurred: {e}")