sunholo 0.109.6__tar.gz → 0.110.1__tar.gz

{sunholo-0.109.6 → sunholo-0.110.1}/PKG-INFO

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

{sunholo-0.109.6 → sunholo-0.110.1}/setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-version = '0.109.6'
+version = '0.110.1'
 
 setup(
     name='sunholo',

{sunholo-0.109.6 → sunholo-0.110.1}/sunholo/discovery_engine/chunker_handler.py

@@ -36,10 +36,18 @@ def do_discovery_engine(message_data:str, metadata:dict, config:ConfigManager=No
                 log.info(f"Found vectorstore {vectorstore}")
                 if value.get('read_only'):
                     continue
+                gcp_config = config.vacConfig("gcp_config")
+                if not gcp_config:
+                    project_id = get_gcp_project()
+                else:
+                    project_id = gcp_config.get("project_id")
+
+                if not project_id:
+                    raise ValueError("Couldn't retrieve project_id for vertex_ai_search")        
                 #location = gcp_config.get('location') 
                 corpus = DiscoveryEngineClient(
                     data_store_id=config.vector_name, 
-                    project_id=get_gcp_project(),
+                    project_id=project_id
                     # location needs to be 'eu' or 'us' which doesn't work with other configurations
                     #location=location or global_location
                     )

{sunholo-0.109.6 → sunholo-0.110.1}/sunholo/discovery_engine/create_new.py

@@ -11,7 +11,8 @@ def create_new_discovery_engine(config:ConfigManager):
         if "chunk_size" in chunker_config:
             chunk_size = chunker_config["chunk_size"]    
 
-    project_id = get_gcp_project()
+    gcp_config = config.vacConfig("gcp_config")
+    project_id = gcp_config.get("project_id") or get_gcp_project()
     if not project_id:
         raise ValueError("Could not find project_id in gcp_config")
     

{sunholo-0.109.6 → sunholo-0.110.1}/sunholo/discovery_engine/get_ai_search_chunks.py

@@ -37,7 +37,9 @@ def get_all_chunks(question:str, config:ConfigManager):
                 num_chunks = value.get('num_chunks') or 3
                 gcp_config = config.vacConfig("gcp_config")
                 project_id = gcp_config.get('project_id')
-                chunk = get_chunks(question, vector_name, num_chunks, project_id=project_id)
+                serving_config = value.get('serving_config')
+
+                chunk = get_chunks(question, vector_name, num_chunks, project_id=project_id, serving_config=serving_config)
                 if chunk:
                     chunks.append(chunk)
     if chunks:
@@ -46,10 +48,12 @@ def get_all_chunks(question:str, config:ConfigManager):
     log.warning(f"No chunks found for {vector_name}")
     return None
 
-def get_chunks(question, vector_name, num_chunks, project_id=None):
+def get_chunks(question, vector_name, num_chunks, project_id=None, serving_config=None):
+    if serving_config is None:
+        serving_config = "default_serving_config"
     de = DiscoveryEngineClient(vector_name, project_id=project_id or get_gcp_project(include_config=True))
     try:
-        return de.get_chunks(question, num_previous_chunks=num_chunks, num_next_chunks=num_chunks)
+        return de.get_chunks(question, num_previous_chunks=num_chunks, num_next_chunks=num_chunks, serving_config=serving_config)
     except Exception as err:
         log.error(f"No discovery engine chunks found: {str(err)}")
     

{sunholo-0.109.6 → sunholo-0.110.1}/sunholo/utils/config.py

@@ -211,6 +211,8 @@ def load_config_key(key: str, vector_name: str, kind: str):
     config = configs_by_kind[kind]
 
     apiVersion = config.get('apiVersion')
+    if apiVersion is None:
+        raise ValueError("Could not find apiVersion for config")
 
     log.debug(f"Fetching '{key}' for '{vector_name}' from '{kind}/{apiVersion}'")
     

{sunholo-0.109.6 → sunholo-0.110.1}/sunholo.egg-info/PKG-INFO

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

{sunholo-0.109.6 → sunholo-0.110.1}/tests/test_config.py

@@ -15,6 +15,7 @@ def test_load_config():
 def test_load_config_key(mock_load_all_configs):
     mock_load_all_configs.return_value = {
         "vacConfig": {
+            "apiVersion": "test",
             "vac": {
                 "test_vector": {
                     "key1": "value1",