Merge pull request #57 from ks6088ts-labs/feature/issue-56_add-tests

add smoke test

Author: ks6088ts

apps/4_streamlit_chat_history/main.py

@@ -38,9 +38,6 @@ def store_chat_history(container: ContainerProxy):
     pprint(response)
 
 
-# To reuse the session for Cosmos DB, just define the container globally
-container = get_container_client()
-
 with st.sidebar:
     azure_openai_endpoint = st.text_input(
         label="AZURE_OPENAI_ENDPOINT",
@@ -116,5 +113,7 @@ def store_chat_history(container: ContainerProxy):
         }
     )
     # !! Here's the only difference from apps/2_streamlit_chat/main.py
-    store_chat_history(container)
+    store_chat_history(
+        container=get_container_client(),
+    )
     st.chat_message("assistant").write(msg)

apps/5_streamlit_query_chat_history/main.py

@@ -21,8 +21,6 @@ def get_container_client() -> ContainerProxy:
     )
 
 
-container = get_container_client()
-
 with st.sidebar:
     "[Azure Portal](https://portal.azure.com/)"
     "[Azure OpenAI Studio](https://oai.azure.com/resource/overview)"
@@ -57,7 +55,7 @@ def convert_to_epoch(d: datetime.date) -> int:
     ]
 
     with st.spinner("Retrieving chat history. Please wait."):
-        items = container.query_items(
+        items = get_container_client().query_items(
             query=query,
             parameters=parameters,
             enable_cross_partition_query=True,

apps/8_streamlit_azure_openai_batch/main.py

@@ -36,12 +36,6 @@
     "[Azure OpenAI Studio](https://oai.azure.com/resource/overview)"
     "[View the source code](https://github.com/ks6088ts-labs/workshop-azure-openai/blob/main/apps/8_streamlit_azure_openai_batch/main.py)"
 
-client = AzureOpenAI(
-    api_key=azure_openai_api_key,
-    api_version=azure_openai_api_version,
-    azure_endpoint=azure_openai_endpoint,
-)
-
 st.title("8_streamlit_azure_openai_batch")
 
 if not azure_openai_api_key or not azure_openai_endpoint or not azure_openai_api_version or not azure_openai_gpt_model:
@@ -53,6 +47,11 @@
 # ---------------
 st.header("Upload batch file")
 st.info("Upload a file in JSON lines format (.jsonl)")
+client = AzureOpenAI(
+    api_key=azure_openai_api_key,
+    api_version=azure_openai_api_version,
+    azure_endpoint=azure_openai_endpoint,
+)
 uploaded_file = st.file_uploader("Upload an input file in JSON lines format", type=("jsonl"))
 if uploaded_file:
     bytes_data = uploaded_file.read()

tests/test_placeholder.py

@@ -0,0 +1,19 @@
+from streamlit.testing.v1 import AppTest
+
+
+def test_smoke():
+    paths = [
+        "apps/2_streamlit_chat/main.py",
+        "apps/4_streamlit_chat_history/main.py",
+        "apps/5_streamlit_query_chat_history/main.py",
+        # "apps/7_streamlit_chat_rag/main.py",
+        "apps/8_streamlit_azure_openai_batch/main.py",
+        "apps/99_streamlit_examples/main.py",
+    ]
+    for path in paths:
+        at = AppTest(
+            script_path=path,
+            default_timeout=60,
+        )
+        at.run()
+        assert not at.exception