Spaces:
Runtime error
Runtime error
File size: 4,242 Bytes
2a67e1c f86e7a9 a626dc8 2a67e1c f86e7a9 a626dc8 2a67e1c f86e7a9 a626dc8 f86e7a9 cf2a5aa f86e7a9 a626dc8 f86e7a9 a626dc8 f86e7a9 a626dc8 f86e7a9 a626dc8 f86e7a9 a626dc8 f86e7a9 a626dc8 |
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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import streamlit as st
import os
from openai import AzureOpenAI
from functions import call_function
st.title("Support Chat UI")
# when will my order be delivered?, [email protected] W123123
functions = [
{
"name": "order_tracking_status",
"description": "Retrieves the status of an order based on **both** the email address and order number.",
"parameters": {
"type": "object",
"properties": {
"email_address": {
"type": "string",
"description": "The email address associated with the order"
},
"order_number": {
"type": "string",
"description": "The order number. For online orders, this starts with either the letter W or C. "
"For in-store orders, this starts with the letter H. You can find your order "
"number in your confirmation email or your receipt."
},
},
"required": ["email_address, order_number"]
}
}
]
client = AzureOpenAI(
api_key=os.environ['OPENAI_API_KEY'],
api_version="2023-07-01-preview",
azure_endpoint=os.environ['AZURE_ENDPOINT'],
)
if "openai_model" not in st.session_state:
st.session_state["openai_model"] = "gpt-35-turbo"
if "messages" not in st.session_state:
st.session_state.messages = [{"role": "system", "content": "You are a helpful customer support agent for "
"The Home Depot. Be as helpful as possible and call "
"functions when necessary."},]
for message in st.session_state.messages:
if message["role"] == "assistant" or message["role"] == "user":
with st.chat_message(message["role"]):
st.markdown(message["content"])
if prompt := st.chat_input("How can we help you today?"):
st.session_state.messages.append({"role": "user", "content": prompt})
with st.chat_message("user"):
st.markdown(prompt)
with st.chat_message("assistant", avatar="🏠"): # avatar=st.image('Home-Depot-Logo.png', width=50)):
message_placeholder = st.empty()
full_message = ""
func_call = {
"name": None,
"arguments": "",
}
called_function = True
while called_function:
called_function = False
for response in client.chat.completions.create(
model=st.session_state["openai_model"],
messages=[
{"role": m["role"], "content": m["content"], "name": m["name"]} if "name" in m else
{"role": m["role"], "content": m["content"]}
for m in st.session_state.messages
],
functions=functions,
function_call="auto",
stream=True,
):
if len(response.choices) > 0:
delta = response.choices[0].delta
full_message += (delta.content or "")
if delta.function_call is not None:
if delta.function_call.name is not None:
func_call["name"] = delta.function_call.name
if delta.function_call.arguments is not None:
func_call["arguments"] += delta.function_call.arguments
if response.choices[0].finish_reason == "function_call" and func_call["name"] is not None:
print(f"Function generation requested, calling function")
function_response = call_function(st.session_state.messages, func_call)
print("function response")
print(function_response)
st.session_state.messages.append(function_response)
called_function = True
message_placeholder.markdown(full_message + "▌")
message_placeholder.markdown(full_message)
st.session_state.messages.append({"role": "assistant", "content": full_message})
|