sunholo 0.106.1__py3-none-any.whl → 0.106.3__py3-none-any.whl

sunholo/discovery_engine/discovery_engine_client.py

@@ -12,6 +12,7 @@ except ImportError:
 
 from ..custom_logging import log
 from typing import Optional, List
+import asyncio
 
 class DiscoveryEngineClient:
     """
@@ -69,9 +70,15 @@ class DiscoveryEngineClient:
         self.store_client  = discoveryengine.DataStoreServiceClient(client_options=client_options)
         self.doc_client    = discoveryengine.DocumentServiceClient(client_options=client_options)
         self.search_client = discoveryengine.SearchServiceClient(client_options=client_options)
-        self.async_search_client = discoveryengine.SearchServiceAsyncClient(client_options=client_options)
         self.engine_client = discoveryengine.EngineServiceClient(client_options=client_options)
-
+        # Initialize the async client only if there's an active event loop
+        try:
+            asyncio.get_running_loop()
+            self.async_search_client = discoveryengine.SearchServiceAsyncClient(client_options=client_options)
+        except RuntimeError:
+            # No event loop in non-async environment, set async client to None
+            log.info("No event loop detected; skipping async client initialization")
+            self.async_search_client = None
 
     @classmethod
     def my_retry(cls):

sunholo/discovery_engine/get_ai_search_chunks.py

@@ -35,8 +35,9 @@ def get_all_chunks(question:str, config:ConfigManager):
                         vector_name = new_vector_name
                 
                 num_chunks = value.get('num_chunks') or 3
-
-                chunk = get_chunks(question, vector_name, num_chunks)
+                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)
                 if chunk:
                     chunks.append(chunk)
     if chunks:
@@ -45,8 +46,8 @@ 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):
-    de = DiscoveryEngineClient(vector_name, project_id=get_gcp_project(include_config=True))
+def get_chunks(question, vector_name, num_chunks, project_id=None):
+    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)
     except Exception as err:

{sunholo-0.106.1.dist-info → sunholo-0.106.3.dist-info}/METADATA

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

{sunholo-0.106.1.dist-info → sunholo-0.106.3.dist-info}/RECORD

@@ -73,8 +73,8 @@ sunholo/database/sql/sb/setup.sql,sha256=CvoFvZQev2uWjmFa3aj3m3iuPFzAAJZ0S7Qi3L3
 sunholo/discovery_engine/__init__.py,sha256=hLgqRDJ22Aov9o2QjAEfsVgnL3kMdM-g5p8RJ9OyKdQ,130
 sunholo/discovery_engine/chunker_handler.py,sha256=Fv4BLOBi_7ap3AiAy4TlTN48CLZSMurJ3TkvC75Euro,5123
 sunholo/discovery_engine/create_new.py,sha256=NzhSh6nG6nQ5J9gZh8IDph4JiEVT_DC5GGvP0GuwTWs,943
-sunholo/discovery_engine/discovery_engine_client.py,sha256=QpTkKaoNiHgVCVcKbMyXAPtCXzpLOIpriQcZxs8zATM,21620
-sunholo/discovery_engine/get_ai_search_chunks.py,sha256=27fLdDNqeac657PibHJ0XaTuSZi47fDv9FCxpPV4CaQ,3919
+sunholo/discovery_engine/discovery_engine_client.py,sha256=jfIayVUOPM4svGF1S5Kk60rIG-xSo_e3zOHtBRg0nZA,22002
+sunholo/discovery_engine/get_ai_search_chunks.py,sha256=7a83FnEDb5hn9j167XC_mN-3l-Sa-p_Yvivmu_DDN_0,4090
 sunholo/embedder/__init__.py,sha256=sI4N_CqgEVcrMDxXgxKp1FsfsB4FpjoXgPGkl4N_u4I,44
 sunholo/embedder/embed_chunk.py,sha256=_FdO4-9frUJlDPqXv2Ai49ejUrrMTMGti3D7bfJGfy8,6984
 sunholo/excel/__init__.py,sha256=AqTMN9K4qJYi4maEgoORc5oxDVGO_eqmwzDaVP37JgY,56
@@ -149,9 +149,9 @@ sunholo/vertex/init.py,sha256=1OQwcPBKZYBTDPdyU7IM4X4OmiXLdsNV30C-fee2scQ,2875
 sunholo/vertex/memory_tools.py,sha256=tBZxqVZ4InTmdBvLlOYwoSEWu4-kGquc-gxDwZCC4FA,7667
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
 sunholo/vertex/type_dict_to_json.py,sha256=uTzL4o9tJRao4u-gJOFcACgWGkBOtqACmb6ihvCErL8,4694
-sunholo-0.106.1.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.106.1.dist-info/METADATA,sha256=LT-NrNweknkdB5XPQFE9yEPQ6Z421flnTD4utV8MjEM,8670
-sunholo-0.106.1.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-sunholo-0.106.1.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.106.1.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.106.1.dist-info/RECORD,,
+sunholo-0.106.3.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.106.3.dist-info/METADATA,sha256=BgiMmvgSwpUsbrTpeoaB2KDwBTiBIypHvBcvbgFklGw,8670
+sunholo-0.106.3.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+sunholo-0.106.3.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.106.3.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.106.3.dist-info/RECORD,,

{sunholo-0.106.1.dist-info → sunholo-0.106.3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.2.0)
+Generator: setuptools (75.3.0)
 Root-Is-Purelib: true
 Tag: py3-none-any