update firestore

app.py

@@ -3,6 +3,8 @@ import os
 from openai import AzureOpenAI
 
 from functions import call_function
+import firebase_admin
+from firebase_admin import credentials, firestore
 
 st.title("SupportFlow Demo")
 # when will my order be delivered?, colin.flueck@gmail.com W123123
@@ -72,6 +74,14 @@ functions = [
     }
 ]
 
+cred = credentials.Certificate("supportflow-4851d-firebase-adminsdk-cdrzu-bf620a4b52.json")
+try:
+    app = firebase_admin.initialize_app(cred)
+except Exception:
+    pass
+
+db = firestore.client()
+
 client = AzureOpenAI(
     api_key=os.environ['OPENAI_API_KEY'],
     api_version="2023-07-01-preview",
@@ -139,7 +149,14 @@ if prompt := st.chat_input("How can we help you today?"):
 
             if function_response["name"] is not None and function_response["name"] == "refer_to_human_agent":
                 print("connect to human agent")
+                print(function_response["name"])
                 st.info('You will be connected with an agent shortly', icon="ℹ️")
+
+                # Get the document to update
+                doc_ref = db.collection('handoffs').document('conversation')
+
+                # Update the document
+                doc_ref.update({'summary': str(function_response["content"]), 'message_history': st.session_state.messages})
             else:
                 message_placeholder = st.empty()
                 full_message = ""

functions.py

@@ -186,4 +186,4 @@ def get_product_listing(sku: str):
 
 
 def refer_to_human_agent(summary: str):
-    return {"role": "function", "content": summary, "name": "refer_to_human_agent"}
+    return summary

requirements.txt

@@ -2,4 +2,5 @@ openai==1.3.3
 streamlit~=1.28.2
 pandas~=2.1.3
 requests~=2.31.0
-fuzzywuzzy~=0.18.0
+fuzzywuzzy~=0.18.0
+firebase_admin==5.0.1

supportflow-4851d-firebase-adminsdk-cdrzu-bf620a4b52.json

@@ -0,0 +1,13 @@
+{
+  "type": "service_account",
+  "project_id": "supportflow-4851d",
+  "private_key_id": "bf620a4b5240ab4a898145aeb53735073d8f2020",
+  "private_key": "-----BEGIN PRIVATE KEY-----\nMIIEvQIBADANBgkqhkiG9w0BAQEFAASCBKcwggSjAgEAAoIBAQDvyurE/WJzsb08\nKTrn/i6688BiIDIqBgN3R71m+Y165M69KoqH/oJLi4PqmbOlv/zLmwjI4vyxUJo3\nKKE2pTETlOIC1hvxbufFnaryWS30f7rooWkxxJr2GmOAxgikp/3LDFjg1mVTaJQG\nTMPeJ9dMPUrGdVg2VWD9Pn0EhWjBt5595Yqz5CBFSFZ39iNWLUHhlblrr4Qc52AG\nxd/B/yso7FvzQcqVMKt6P8TRcGSAvtOr8Juq9d3vTpe3oRKbiJKiJHGDwRJjDvpb\nX5Ri66cfogH91YynwxggRnaKKZXfkhIoJeMRR8QaVJ4OKkRSJ5EGddMWpMf2/wXf\nta7BFndBAgMBAAECggEAL3vxgwpBH3NsK3hnqY6QTuWbbGL/RAqWT53FcBmNTm92\nUAfXUGhSAWcye3et1y3+P4xnEC7AxtUVAMS7LeDG59+E5PZkqKnjXCLleeG2g7Ly\nfoQOGnSaEg92Mkck6YQ5seZzSCy0enCcVWYIqFqf3AFMKCIPEo1ULDneVFGdxbQ2\nW/ux5zPIDKKyjI038+VFsRtnVoJnwvORC22ndQbZprJrQYEhrolnkWZ3nUCp0QgR\n4CKmLKGrVddwhG2NHgfh12NHbDTYUe4rDfg44rjfwHpkM7tgXvMFuP54d8lSKqEc\nUwkp3E50SO3Y5tQs84KcgIZqiwlSJpnBWMDhk1PWdQKBgQD7GQ/KDapf18T4F/1f\nrv06J2vmhSs3OEQLYgPnpFkXVAyBNjqnAHjU+SOtrmWTDYxQ3BtOd7aCKwTaErE0\nHWNvo1Z19NK6jj6Nqt6UrRSal/Z3uvIvZUti6xOLCOw4nUlJCe4DkfTwJwFyuhmI\ni464RPYxqaWQyRIT2xyRXBe1VQKBgQD0eVqcH6QOMIj2WDRJoBd27iyYr4QGFtbd\ndXIbBHysgqjjgNI64QgZ6YOh2ehFgmiJn4dMb2kAN8TW1Qhgu4XSVcUBj23DW4xi\nfQX0l0EruERxcslS6Cnsr+LlnxRdVXjpyBHfcFNVhgKgSSP5xqy694AStCMCfhvk\n+9mbju86PQKBgCD6Dp/LoaeKr9yhUe5iDfG4Qax5qs5Kvmz8R2UBqY2txSKnG5bZ\nMgVOXwgtUju42Me50IBCqp53xeBZ+hnNDkUYH3/IUnCbw81ZNngyLvWKpZzd2Xau\nB/OKyftFhbh9t6U2/6JKZe+wP7lwhW/wXCMJM4gcLplQPuaKrTrN7HMtAoGBAKDB\nI3nBvIiwFX+XJFq9cHB90MldsDyozAS3c56cZraw/AkM7jBVRJR1Xl8RAJCkXcUR\nJ1B1ed/wy/hQAMBKzoYua+rIu25YRomEmGPrCamL80V+Ky/fBaJjTZmb7zIoP+fJ\n7Sa1n7Z+9j8bmRJnN7GBia/Cz7qV3hXGHaMSa/ApAoGAKsESANQHpjF9i9ERFjOr\nmn3asPxwUcpn5Mgpj2mmuMz1pYy9iGPvZ7V+YefPT7SN5v/PrPnCrP2vz6zwe0pj\nFJWb19yVbccLXP0+k7XTEAXBfAQEUO4pw9HT2FNZmJxIUM84KRhU3BxMqhyW2dn3\n/IHB5Q6p7P2K3i8CxKBYM40=\n-----END PRIVATE KEY-----\n",
+  "client_email": "firebase-adminsdk-cdrzu@supportflow-4851d.iam.gserviceaccount.com",
+  "client_id": "105720102348271033803",
+  "auth_uri": "https://accounts.google.com/o/oauth2/auth",
+  "token_uri": "https://oauth2.googleapis.com/token",
+  "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
+  "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/firebase-adminsdk-cdrzu%40supportflow-4851d.iam.gserviceaccount.com",
+  "universe_domain": "googleapis.com"
+}