tests fixed, added instructions in the readme

snowflake-cortex/README.md

@@ -1,8 +1,18 @@
+This Native App examplifies how to implement the Cortex Complete and to make it interact with user data.
+
+For this usecase, the dataset used is rather small, with only 10 entries, this is because the language model used in the Cortex function restricts the bytes of input data size and costs. YOu can change the language model used to a bigger or different style according to your necessities.
+For more information about it please visit **[this page](https://docs.snowflake.com/en/user-guide/snowflake-cortex/llm-functions#cost-considerations)**.
+
 To run this example first execute this command:
-```bash
+```sh
 snow sql -f 'prepare/prepare_data.sql'
 ```
 
 Then run `snow app run` on your terminal.
 
-Currently it has a very short dataset due to llm model bytes limit.
+To delete the database and the app run
+
+```sh
+snow sql -q 'DROP DATABASE SPOTIFY_CORTEX_DB;'
+snow app teardown
+```

snowflake-cortex/src/ui/cortexCaller.py

@@ -2,8 +2,13 @@
 
 class CortexCaller:
 
+   def call_complete(_self,input_with_table):
+      response = Complete('llama2-70b-chat', input_with_table)
+      return response
+
    def call_cortex(_self, df, input: str):
       sample_df = df.to_string()
       input_with_table = f"""The following data is a table that contains songs information it is wrapped in this tags <dataset>{sample_df}</dataset> {input}"""
-      response = Complete('llama2-70b-chat', input_with_table)
-      return response
+      response = _self.call_complete(input_with_table)
+      return response
+

snowflake-cortex/src/ui/dashboard.py

@@ -4,7 +4,7 @@
 
 class Dashboard:
 
-   def __init__(self, session: Session = Session.builder.getOrCreate(), cortex_caller: CortexCaller = CortexCaller()) -> None:
+   def __init__(self, session: Session, cortex_caller: CortexCaller = CortexCaller()) -> None:
       self.session = session
       self.cortex_caller = cortex_caller
 
@@ -23,4 +23,4 @@ def run_streamlit(self):
          cortex_chat.chat_message("assistant").write(cortex_fn)
 
 if __name__ == '__main__':
-   Dashboard().run_streamlit()
+   Dashboard(Session.builder.getOrCreate()).run_streamlit()

snowflake-cortex/test/test_cortex.py

@@ -0,0 +1,20 @@
+import unittest
+from unittest.mock import patch, MagicMock
+import pandas as pd
+import pytest as pytest
+from snowflake import cortex
+from snowflake.cortex import Complete
+
+class TestDataExporter(unittest.TestCase):
+
+    def test_cortex_response(self):
+        from cortexCaller import CortexCaller
+
+        mock_df = pd.DataFrame([{"TestColumn": "TestValue"}])
+        mock_input = 'Test input from user'
+        cortexCaller = CortexCaller()
+        cortexCaller.call_complete = MagicMock(name='call_complete')
+        cortexCaller.call_complete.return_value = 'test answer from cortex'
+        response = cortexCaller.call_cortex(mock_df, mock_input)
+        cortexCaller.call_complete.assert_called_once()
+        assert response == cortexCaller.call_complete.return_value

snowflake-cortex/test/test_ui.py

@@ -1,4 +1,3 @@
-from dashboard import Dashboard
 from cortexCaller import CortexCaller
 from streamlit.testing.v1 import AppTest
 from snowflake.snowpark import Session
@@ -30,18 +29,33 @@ def script(session, cortex_caller):
     at = AppTest.from_function(script, kwargs={ "session": session, "cortex_caller": cortex_caller }).run()
 
     at.chat_input(key='cortex_input').set_value('text for testing purposes').run()
-    assert at.chat_message[0].name == 'user'
+    call_cortex.assert_called_once()
     assert at.chat_message[1].name == 'assistant'
 
-def test_input_showed_in_chat():
-    dashboard = Dashboard(session)
-    at = AppTest.from_function(dashboard.run_streamlit).run()
+@patch('cortexCaller.CortexCaller.call_cortex')
+@patch('snowflake.snowpark.session.Session.table')
+def test_input_showed_in_chat(table: MagicMock, call_cortex: MagicMock, session, cortex_caller):
+
+    def script(session, cortex_caller):
+        from dashboard import Dashboard
+        dashboard = Dashboard(session, cortex_caller)
+        return dashboard.run_streamlit()
+
+    table.return_value=session.create_dataframe([{"TestColumn": "TestValue"}])
+    at = AppTest.from_function(script, kwargs={ "session": session, "cortex_caller": cortex_caller }).run()
     at.chat_input(key='cortex_input').set_value('text for testing purposes').run()
     assert at.chat_message[0].name == 'user'
 
-def test_input_empty():
-    dashboard = Dashboard(session)
-    at = AppTest.from_function(dashboard.run_streamlit).run()
+@patch('cortexCaller.CortexCaller.call_cortex')
+@patch('snowflake.snowpark.session.Session.table')
+def test_input_empty(table: MagicMock, call_cortex: MagicMock, session, cortex_caller):
+    def script(session, cortex_caller):
+        from dashboard import Dashboard
+        dashboard = Dashboard(session, cortex_caller)
+        return dashboard.run_streamlit()
+
+    table.return_value=session.create_dataframe([{"TestColumn": "TestValue"}])
+    at = AppTest.from_function(script, kwargs={ "session": session, "cortex_caller": cortex_caller }).run()
     at.chat_input(key='cortex_input').set_value(' ').run()
     assert not at.chat_message