sunholo 0.60.9__tar.gz → 0.61.0__tar.gz

{sunholo-0.60.9 → sunholo-0.61.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.60.9
+Version: 0.61.0
 Summary: Large Language Model DevOps - a package to help deploy LLMs to the Cloud.
 Home-page: https://github.com/sunholo-data/sunholo-py
-Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.60.9.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.61.0.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0

{sunholo-0.60.9 → sunholo-0.61.0}/setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 # Define your base version
-version = '0.60.9'
+version = '0.61.0'
 
 setup(
     name='sunholo',

{sunholo-0.60.9 → sunholo-0.61.0}/sunholo.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.60.9
+Version: 0.61.0
 Summary: Large Language Model DevOps - a package to help deploy LLMs to the Cloud.
 Home-page: https://github.com/sunholo-data/sunholo-py
-Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.60.9.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.61.0.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0

{sunholo-0.60.9 → sunholo-0.61.0}/sunholo.egg-info/SOURCES.txt

@@ -18,7 +18,6 @@ sunholo/agents/langserve.py
 sunholo/agents/pubsub.py
 sunholo/agents/route.py
 sunholo/agents/special_commands.py
-sunholo/agents/test_chat_history.py
 sunholo/agents/fastapi/__init__.py
 sunholo/agents/fastapi/base.py
 sunholo/agents/fastapi/qna_routes.py
@@ -107,4 +106,6 @@ sunholo/utils/parsers.py
 sunholo/utils/user_ids.py
 sunholo/vertex/__init__.py
 sunholo/vertex/init_vertex.py
-test/test_dispatch_to_qa.py
+test/test_dispatch_to_qa.py
+tests/test_chat_history.py
+tests/test_config.py

sunholo-0.61.0/tests/test_chat_history.py

@@ -0,0 +1,50 @@
+import pytest
+from sunholo.agents.chat_history import extract_chat_history, embeds_to_json, create_message_element, is_human, is_bot, is_ai
+
+# Test cases for extract_chat_history function
+@pytest.mark.parametrize("chat_history,expected", [
+    ([], []),
+    ([{"name": "Human", "text": "Hello, AI!"}, {"name": "AI", "text": "Hello, Human! How can I help you today?"}], [("Hello, AI!", "Hello, Human! How can I help you today?")])
+])
+def test_extract_chat_history(chat_history, expected):
+    assert extract_chat_history(chat_history) == expected
+
+# Test cases for embeds_to_json function
+@pytest.mark.parametrize("message,expected", [
+    ({"embeds": []}, ""),
+    ({"embeds": [{"type": "image", "url": "https://example.com/image.png"}]}, '[{"type": "image", "url": "https://example.com/image.png"}]')
+])
+def test_embeds_to_json(message, expected):
+    assert embeds_to_json(message) == expected
+
+# Test cases for create_message_element function
+@pytest.mark.parametrize("message,expected", [
+    ({"text": "Hello, AI!"}, "Hello, AI!"),
+    ({"content": "Hello, AI!"}, "Hello, AI!")
+])
+def test_create_message_element(message, expected):
+    assert create_message_element(message) == expected
+
+# Test cases for is_human function
+@pytest.mark.parametrize("message,expected", [
+    ({"name": "Human"}, True),
+    ({"name": "AI"}, False)
+])
+def test_is_human(message, expected):
+    assert is_human(message) == expected
+
+# Test cases for is_bot function
+@pytest.mark.parametrize("message,expected", [
+    ({"name": "AI"}, True),
+    ({"name": "Human"}, False)
+])
+def test_is_bot(message, expected):
+    assert is_bot(message) == expected
+
+# Test cases for is_ai function
+@pytest.mark.parametrize("message,expected", [
+    ({"name": "AI"}, True),
+    ({"name": "Human"}, False)
+])
+def test_is_ai(message, expected):
+    assert is_ai(message) == expected

sunholo-0.61.0/tests/test_config.py

@@ -0,0 +1,40 @@
+import pytest
+from sunholo.utils import config
+
+# Test for the `load_config` function
+def test_load_config():
+    # Assuming the function is supposed to load a configuration from a file
+    # and return a dictionary with the configuration data
+    expected_config = {"key": "value"}
+    with pytest.raises(FileNotFoundError):
+        config.load_config("non_existent_file")
+    with pytest.mock.patch("builtins.open", pytest.mock.mock_open(read_data="key: value"), create=True):
+        assert config.load_config("mock_file.yaml") == expected_config
+
+# Test for the `load_config_key` function
+def test_load_config_key():
+    # Assuming the function is supposed to load a specific key from the configuration
+    expected_value = "value"
+    with pytest.raises(KeyError):
+        config.load_config_key("non_existent_key", "mock_vector_name")
+    with pytest.mock.patch("sunholo.utils.config.load_all_configs", return_value={"mock_vector_name": {"key": "value"}}):
+        assert config.load_config_key("key", "mock_vector_name") == expected_value
+
+# Test for the `get_module_filepath` function
+def test_get_module_filepath():
+    # Assuming the function is supposed to return the absolute path of a module file
+    expected_path = "/absolute/path/to/module/file"
+    with pytest.mock.patch("os.path.dirname", return_value="/absolute/path/to"):
+        assert config.get_module_filepath("module/file") == expected_path
+
+# Test for the `fetch_config` function
+def test_fetch_config():
+    # Assuming the function is supposed to fetch configuration from a cloud storage bucket
+    expected_update_time = "2022-01-01T00:00:00Z"
+    with pytest.mock.patch("google.cloud.storage.Client") as mock_client:
+        mock_bucket = mock_client.return_value.get_bucket.return_value
+        mock_blob = mock_bucket.get_blob.return_value
+        mock_blob.updated = expected_update_time
+        assert config.fetch_config("bucket_name", "blob_name") == expected_update_time
+        mock_bucket.get_blob.return_value = None
+        assert config.fetch_config("bucket_name", "blob_name") is None

sunholo-0.60.9/sunholo/agents/test_chat_history.py

@@ -1,119 +0,0 @@
-import pytest
-from .chat_history import *
-
-
-def test_extract_chat_history_null_input():
-    assert extract_chat_history(None) == [], 'Expected empty list for null input'
-
-
-def test_extract_chat_history_no_chat():
-    assert extract_chat_history([]) == [], 'Expected empty list for no chat history'
-
-
-def test_extract_chat_history_with_chat():
-    chat_history = [('User', 'Hello'), ('Bot', 'Hi'), ('User', 'How are you?'), ('Bot', 'I am fine.')]
-    expected_output = [('User', 'Hello'), ('Bot', 'Hi'), ('User', 'How are you?'), ('Bot', 'I am fine.')]
-    assert extract_chat_history(chat_history) == expected_output, 'Expected list of paired messages for chat history'
-
-
-# Test cases for embeds_to_json function
-
-def test_embeds_to_json_no_embeds():
-    message = 'Hello, world!'
-    assert embeds_to_json(message) == '', 'Expected empty string for message with no embeds'
-
-
-def test_embeds_to_json_one_embed():
-    message = 'Hello, world! [embed]'
-    expected_output = '{"embeds": ["embed"]}'
-    assert embeds_to_json(message) == expected_output, 'Expected JSON string with one embed for message with one embed'
-
-
-def test_embeds_to_json_multiple_embeds():
-    message = 'Hello, world! [embed1] [embed2]'
-    expected_output = '{"embeds": ["embed1", "embed2"]}'
-    assert embeds_to_json(message) == expected_output, 'Expected JSON string with multiple embeds for message with multiple embeds'
-
-
-# Test cases for create_message_element function
-
-def test_create_message_element_text():
-    message = {'text': 'Hello, world!'}
-    assert create_message_element(message) == 'Hello, world!', 'Expected text element for message with text'
-
-
-def test_create_message_element_content():
-    message = {'content': 'Hello, world!'}
-    assert create_message_element(message) == 'Hello, world!', 'Expected content element for message with content'
-
-
-def test_create_message_element_no_text_or_content():
-    message = {}
-    with pytest.raises(KeyError):
-        create_message_element(message)
-
-
-# Test cases for is_human function
-
-def test_is_human_name_human():
-    message = {'name': 'Human'}
-    assert is_human(message) == True, 'Expected True for message with name Human'
-
-
-def test_is_human_sender_type_human():
-    message = {'sender': {'type': 'HUMAN'}}
-    assert is_human(message) == True, 'Expected True for message with sender type HUMAN'
-
-
-def test_is_human_user_no_bot_id():
-    message = {'user': 'User1', 'bot_id': None}
-    assert is_human(message) == True, 'Expected True for message with user field and no bot_id field'
-
-
-def test_is_human_not_human():
-    message = {'name': 'Bot'}
-    assert is_human(message) == False, 'Expected False for message not from a human'
-
-
-# Test cases for is_bot function
-
-def test_is_bot_name_bot():
-    message = {'name': 'Bot'}
-    assert is_bot(message) == True, 'Expected True for message with name Bot'
-
-
-def test_is_bot_sender_type_bot():
-    message = {'sender': {'type': 'BOT'}}
-    assert is_bot(message) == True, 'Expected True for message with sender type BOT'
-
-
-def test_is_bot_with_bot_id():
-    message = {'bot_id': 'bot1'}
-    assert is_bot(message) == True, 'Expected True for message with bot_id field'
-
-
-def test_is_bot_not_bot():
-    message = {'name': 'Human'}
-    assert is_bot(message) == False, 'Expected False for message not from a bot'
-
-
-# Test cases for is_ai function
-
-def test_is_ai_name_ai():
-    message = {'name': 'AI'}
-    assert is_ai(message) == True, 'Expected True for message with name AI'
-
-
-def test_is_ai_sender_type_bot():
-    message = {'sender': {'type': 'BOT'}}
-    assert is_ai(message) == True, 'Expected True for message with sender type BOT'
-
-
-def test_is_ai_with_bot_id():
-    message = {'bot_id': 'bot1'}
-    assert is_ai(message) == True, 'Expected True for message with bot_id field'
-
-
-def test_is_ai_not_ai():
-    message = {'name': 'Human'}
-    assert is_ai(message) == False, 'Expected False for message not from an AI'