sunholo 0.62.11__tar.gz → 0.62.13__tar.gz

{sunholo-0.62.11 → sunholo-0.62.13}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.62.11
+Version: 0.62.13
 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.62.11.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.62.13.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0

{sunholo-0.62.11 → sunholo-0.62.13}/setup.py

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

{sunholo-0.62.11 → sunholo-0.62.13}/sunholo/agents/dispatch_to_qa.py

@@ -74,7 +74,7 @@ def prep_request_payload(user_input, chat_history, vector_name, stream, **kwargs
         # Update qna_data with optional values from kwargs
         qna_data.update(kwargs)
 
-        if not 'vector_name' not in qna_data:
+        if 'vector_name' not in qna_data:
             qna_data['vector_name'] = vector_name
 
     return qna_endpoint, qna_data

{sunholo-0.62.11 → sunholo-0.62.13}/sunholo/agents/flask/qna_routes.py

@@ -206,6 +206,8 @@ def prep_vac(request, vector_name):
     stream_wait_time = data.pop('stream_wait_time', 7)
     stream_timeout = data.pop('stream_timeout', 120)
     chat_history = data.pop('chat_history', None)
+    vector_name = data.pop('vector_name', vector_name)
+
     paired_messages = extract_chat_history(chat_history)
 
     all_input = {'user_input': user_input, 

{sunholo-0.62.11 → sunholo-0.62.13}/sunholo.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.62.11
+Version: 0.62.13
 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.62.11.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.62.13.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0

{sunholo-0.62.11 → sunholo-0.62.13}/sunholo.egg-info/SOURCES.txt

@@ -112,4 +112,5 @@ sunholo/vertex/__init__.py
 sunholo/vertex/init.py
 sunholo/vertex/memory_tools.py
 tests/test_chat_history.py
-tests/test_config.py
+tests/test_config.py
+tests/test_dispatch_to_qa.py

sunholo-0.62.13/tests/test_config.py

@@ -0,0 +1,34 @@
+import pytest
+from unittest.mock import patch, mock_open
+from sunholo.utils import config
+
+def test_load_config():
+    expected_config = {"key": "value"}
+    with pytest.raises(FileNotFoundError):
+        config.load_config("non_existent_file")
+    with patch("builtins.open", mock_open(read_data='{"key": "value"}'), create=True):
+        result, _ = config.load_config("mock_file.json")
+        assert result == expected_config
+
+# Test cases for load_config_key function
+@patch("sunholo.utils.config.load_all_configs")
+def test_load_config_key(mock_load_all_configs):
+    mock_load_all_configs.return_value = {
+        "vacConfig": {
+            "vac": {
+                "test_vector": {
+                    "key1": "value1",
+                    "key2": "value2"
+                }
+            }
+        }
+    }
+    # Test existing key
+    assert config.load_config_key("key1", "test_vector", "vacConfig") == "value1"
+    # Test non-existing key
+    result = config.load_config_key("non_existing_key", "test_vector", "vacConfig")
+    assert result is None
+
+    # Test invalid configuration
+    with pytest.raises(KeyError):
+        config.load_config_key("key1", "test_vector", "invalidConfig")

sunholo-0.62.13/tests/test_dispatch_to_qa.py

@@ -0,0 +1,18 @@
+import pytest
+from unittest.mock import patch
+from sunholo.agents.dispatch_to_qa import prep_request_payload
+
+@pytest.mark.parametrize("user_input, chat_history, vector_name, stream, expected_endpoint, expected_payload", [
+    ("What is AI?", [], "my_vector", False, "http://example.com/invoke", {"user_input": "What is AI?", "chat_history": [], "vector_name": "my_vector"}),
+    ("Tell me about ML", ["Previous chat message"], "another_vector", True, "http://example.com/stream", {"user_input": "Tell me about ML", "chat_history": ["Previous chat message"], "vector_name": "another_vector"})
+])
+@patch('sunholo.agents.dispatch_to_qa.load_config_key')
+@patch('sunholo.agents.dispatch_to_qa.route_endpoint')
+def test_prep_request_payload(mock_route_endpoint, mock_load_config_key, user_input, chat_history, vector_name, stream, expected_endpoint, expected_payload):
+    mock_route_endpoint.return_value = {"invoke": "http://example.com/invoke", "stream": "http://example.com/stream"}
+    mock_load_config_key.side_effect = lambda key, vector_name=None, kind=None: "my_vector" if key == "vector_name" else None
+
+    endpoint, payload = prep_request_payload(user_input, chat_history, vector_name, stream)
+
+    assert endpoint == expected_endpoint
+    assert payload == expected_payload

sunholo-0.62.11/tests/test_config.py

@@ -1,11 +0,0 @@
-import pytest
-from unittest.mock import patch, mock_open
-from sunholo.utils import config
-
-def test_load_config():
-    expected_config = {"key": "value"}
-    with pytest.raises(FileNotFoundError):
-        config.load_config("non_existent_file")
-    with patch("builtins.open", mock_open(read_data='{"key": "value"}'), create=True):
-        result, _ = config.load_config("mock_file.json")
-        assert result == expected_config