sunholo 0.71.6__py3-none-any.whl → 0.71.10__py3-none-any.whl

sunholo/discovery_engine/__init__.py

@@ -0,0 +1 @@
+from .discovery_engine_client import DiscoveryEngineClient

sunholo/discovery_engine/discovery_engine_client.py

@@ -11,7 +11,7 @@ except ImportError:
     SearchResponse = None
 
 from ..logging import log
-from typing import Optional
+from typing import Optional, List
 
 class DiscoveryEngineClient:
     """
@@ -69,6 +69,7 @@ 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.engine_client = discoveryengine.EngineServiceClient(client_options=client_options)
 
     @classmethod
     def my_retry(cls):
@@ -261,6 +262,59 @@ class DiscoveryEngineClient:
         result_string = "\n".join(all_chunks)
 
         return result_string
+    
+    def create_engine(self,
+        engine_id: str, 
+        data_store_ids: List[str],
+        solution_type=None,
+        search_tier=None,
+        search_add_ons=None,
+    ) -> str:
+
+        # The full resource name of the collection
+        # e.g. projects/{project}/locations/{location}/collections/default_collection
+        parent = self.data_store_path()
+
+        engine = discoveryengine.Engine(
+            display_name=engine_id,
+            # Options: GENERIC, MEDIA, HEALTHCARE_FHIR
+            industry_vertical=discoveryengine.IndustryVertical.GENERIC,
+            # Options: SOLUTION_TYPE_RECOMMENDATION, SOLUTION_TYPE_SEARCH, SOLUTION_TYPE_CHAT, SOLUTION_TYPE_GENERATIVE_CHAT
+            solution_type=solution_type or discoveryengine.SolutionType.SOLUTION_TYPE_SEARCH,
+            # For search apps only
+            search_engine_config=discoveryengine.Engine.SearchEngineConfig(
+                # Options: SEARCH_TIER_STANDARD, SEARCH_TIER_ENTERPRISE
+                search_tier=search_tier or discoveryengine.SearchTier.SEARCH_TIER_ENTERPRISE,
+                # Options: SEARCH_ADD_ON_LLM, SEARCH_ADD_ON_UNSPECIFIED
+                search_add_ons=search_add_ons or [discoveryengine.SearchAddOn.SEARCH_ADD_ON_UNSPECIFIED],
+            ),
+            # For generic recommendation apps only
+            # similar_documents_config=discoveryengine.Engine.SimilarDocumentsEngineConfig,
+            data_store_ids=data_store_ids,
+        )
+
+        request = discoveryengine.CreateEngineRequest(
+            parent=parent,
+            engine=engine,
+            engine_id=engine_id,
+        )
+
+        # Make the request
+        operation = self.engine_client.create_engine(request=request)
+
+        log.info(f"Waiting for create vertex ai search operation to complete: {operation.operation.name}")
+        response = operation.result()
+
+        # Once the operation is complete,
+        # get information from operation metadata
+        metadata = discoveryengine.CreateEngineMetadata(operation.metadata)
+
+        # Handle the response
+        log.info(f"{response=} {metadata=}")
+
+        return operation.operation.name
+
+
 
     def import_documents(self,
         gcs_uri: Optional[str] = None,

{sunholo-0.71.6.dist-info → sunholo-0.71.10.dist-info}/METADATA

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

{sunholo-0.71.6.dist-info → sunholo-0.71.10.dist-info}/RECORD

@@ -60,10 +60,10 @@ sunholo/database/sql/sb/create_table.sql,sha256=SbcOrf5tUiVKGUohu1lau7IsbDRbTFbr
 sunholo/database/sql/sb/delete_source_row.sql,sha256=r6fEuUKdbiLHCDGKSbKINDCpJjsmfHZNNOo1ptwLLSo,75
 sunholo/database/sql/sb/return_sources.sql,sha256=89KAnxfK8n_qGK9jy1OQT8f9n4uYUtYL5cCxbC2mj_c,255
 sunholo/database/sql/sb/setup.sql,sha256=CvoFvZQev2uWjmFa3aj3m3iuPFzAAJZ0S7Qi3L3-zZI,89
-sunholo/discovery_engine/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sunholo/discovery_engine/__init__.py,sha256=qUKWzuHApDRJIUoynukVdGRBEq8eC9T7l9a3bWckgI0,59
 sunholo/discovery_engine/chunker_handler.py,sha256=H1HHDqWMCkchJER1_oU9TOLxqf2PygiMO6CL3uKZP64,4563
 sunholo/discovery_engine/create_new.py,sha256=7oZG78T6lW0EspRzlo7-qRyXFSuFxDn2dfSAVEaqlqY,978
-sunholo/discovery_engine/discovery_engine_client.py,sha256=cUdyXdXvFS_IxNqwJkGdGpXH-k18MQDKuUFn_7E7pKo,14922
+sunholo/discovery_engine/discovery_engine_client.py,sha256=-e3rzmqYcVmb-Q7scq-ae1NF7hQGfJCIU1ym2Cu2u6g,17167
 sunholo/embedder/__init__.py,sha256=sI4N_CqgEVcrMDxXgxKp1FsfsB4FpjoXgPGkl4N_u4I,44
 sunholo/embedder/embed_chunk.py,sha256=d_dIzeNF630Q0Ar-u1hxos60s0tLIImJccAvuo_LTIw,6814
 sunholo/gcs/__init__.py,sha256=DtVw_AZwQn-IguR5BJuIi2XJeF_FQXizhJikzRNrXiE,50
@@ -113,9 +113,9 @@ sunholo/vertex/extensions_class.py,sha256=0-XMrMvfhMN380ZdGXl11Mt7R9kCu9rB4Vduif
 sunholo/vertex/init.py,sha256=RLjQppTUwubWgwf2PoAke-EtcwlVkFPaPMYvUsMw1KQ,2029
 sunholo/vertex/memory_tools.py,sha256=sipBI7TFttbYzobSWS_1TzWFVTPnJckz3NvLVbTepMc,6345
 sunholo/vertex/safety.py,sha256=3meAX0HyGZYrH7rXPUAHxtI_3w_zoy_RX7Shtkoa660,1275
-sunholo-0.71.6.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.71.6.dist-info/METADATA,sha256=mF32bponOP0fNr7dvWPiIWOAlTf1-csaIxTIy_oSBDA,6613
-sunholo-0.71.6.dist-info/WHEEL,sha256=mguMlWGMX-VHnMpKOjjQidIo1ssRlCFu4a4mBpz1s2M,91
-sunholo-0.71.6.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.71.6.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.71.6.dist-info/RECORD,,
+sunholo-0.71.10.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.71.10.dist-info/METADATA,sha256=LIiaYAQSZmRw9v6rhqq3i-GZjIP_K9zZItlJ-j9cxW0,6615
+sunholo-0.71.10.dist-info/WHEEL,sha256=mguMlWGMX-VHnMpKOjjQidIo1ssRlCFu4a4mBpz1s2M,91
+sunholo-0.71.10.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.71.10.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.71.10.dist-info/RECORD,,