sunholo 0.84.5__py3-none-any.whl → 0.84.7__py3-none-any.whl

sunholo/genai/__init__.py

@@ -0,0 +1,2 @@
+from .process_funcs_cls import GenAIFunctionProcessor
+from .safety import genai_safety

sunholo/genai/process_funcs_cls.py

@@ -28,24 +28,21 @@ class GenAIFunctionProcessor:
         funcs (dict): A dictionary of function names mapped to their implementations.
 
     Example usage:
-        # Subclassing GenAIFunctionProcessor for a specific application (e.g., AlloyDB)
-        class AlloyDBFunctionProcessor(GenAIFunctionProcessor):
-            def construct_tools(self) -> dict:
-                # Implement function mapping specific to AlloyDB
-                ...
+    ```python
+    class AlloyDBFunctionProcessor(GenAIFunctionProcessor):
+        def construct_tools(self) -> dict:
+            ...
 
-        # Creating an instance of the subclass and using it
-        config = ConfigManager()
-        alloydb_processor = AlloyDBFunctionProcessor(config)
+    config = ConfigManager()
+    alloydb_processor = AlloyDBFunctionProcessor(config)
 
-        # Processing functions based on a generative AI response
-        results = alloydb_processor.process_funcs(full_response)
+    results = alloydb_processor.process_funcs(full_response)
 
-        # Getting the configured model
-        alloydb_model = alloydb_processor.get_model(
-            model_name="gemini-1.5-pro",
-            system_instruction="You are a helpful AlloyDB agent that helps users search and extract documents from the database."
-        )
+    alloydb_model = alloydb_processor.get_model(
+        model_name="gemini-1.5-pro",
+        system_instruction="You are a helpful AlloyDB agent that helps users search and extract documents from the database."
+    )
+    ```
     """
 
     def __init__(self, config: ConfigManager):
@@ -60,6 +57,7 @@ class GenAIFunctionProcessor:
         
         self.config = config
         self.funcs = self.construct_tools()
+        self.model_name = config.vacConfig("model") or "gemini-1.5-flash"
         self._validate_functions()
 
     def construct_tools(self) -> dict:
@@ -93,7 +91,7 @@ class GenAIFunctionProcessor:
                 log.error(f"Function {func_name} is missing a docstring.")
                 raise ValueError(f"Function {func_name} must have a docstring to be used as a genai tool.")
 
-    def process_funcs(self, full_response, output_parts=True) -> list[Part] | str:
+    def process_funcs(self, full_response, output_parts=True) -> list['Part'] | str:
         """
         Processes the functions based on the full_response from the generative model.
 
@@ -109,7 +107,9 @@ class GenAIFunctionProcessor:
             list[Part] | str: A list of Part objects or a formatted string with the results.
 
         Example usage:
-            results = alloydb_processor.process_funcs(full_response)
+        ```python
+        results = alloydb_processor.process_funcs(full_response)
+        ```
         """
         api_requests_and_responses = []
 
@@ -164,7 +164,7 @@ class GenAIFunctionProcessor:
 
         return strings
 
-    def get_model(self, model_name: str, system_instruction: str, generation_config=None):
+    def get_model(self, system_instruction: str, generation_config=None, model_name: str=None):
         """
         Constructs and returns the generative AI model configured with the tools.
 
@@ -180,10 +180,12 @@ class GenAIFunctionProcessor:
             GenerativeModel: An instance of the GenerativeModel configured with the provided tools.
 
         Example usage:
-            alloydb_model = alloydb_processor.get_model(
-                model_name="gemini-1.5-pro",
-                system_instruction="You are a helpful AlloyDB agent that helps users search and extract documents from the database."
-            )
+        ```python
+        alloydb_model = alloydb_processor.get_model(
+            model_name="gemini-1.5-pro",
+            system_instruction="You are a helpful AlloyDB agent that helps users search and extract documents from the database."
+        )
+        ```
         """
         if generation_config is None:
             generation_config = {
@@ -196,7 +198,7 @@ class GenAIFunctionProcessor:
 
         try:
             model = genai.GenerativeModel(
-                model_name=model_name,
+                model_name=model_name or self.model_name,
                 tools=tools,
                 generation_config=generation_config,
                 safety_settings=genai_safety(),

{sunholo-0.84.5.dist-info → sunholo-0.84.7.dist-info}/METADATA

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

{sunholo-0.84.5.dist-info → sunholo-0.84.7.dist-info}/RECORD

@@ -82,8 +82,8 @@ sunholo/gcs/add_file.py,sha256=TRbdEd3j-lFnyC7kqs5IWEPMPlDMFLR68fdTWga06Wo,8096
 sunholo/gcs/download_folder.py,sha256=ijJTnS595JqZhBH8iHFErQilMbkuKgL-bnTCMLGuvlA,1614
 sunholo/gcs/download_url.py,sha256=q1NiJSvEhdNrmU5ZJ-sBCMC_J5CxzrajY8LRgdPOV_M,6130
 sunholo/gcs/metadata.py,sha256=oQLcXi4brsZ74aegWyC1JZmhlaEV270HS5_UWtAYYWE,898
-sunholo/genai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sunholo/genai/process_funcs_cls.py,sha256=mmbBlUjGbxvx8paf0sLa98fbboCoLwaSEjQV5R44DA8,8345
+sunholo/genai/__init__.py,sha256=o2c17UbgzIDxNKKF8deCLoctNpatAYgm9Cf9XDfpqvg,87
+sunholo/genai/process_funcs_cls.py,sha256=h0W767nXrwPesrc9iDEJ1bzesvXeze7YV8U7cASkKdc,8150
 sunholo/genai/safety.py,sha256=mkFDO_BeEgiKjQd9o2I4UxB6XI7a9U-oOFjZ8LGRUC4,1238
 sunholo/invoke/__init__.py,sha256=bELcqIjzKvaupcIN5OQmDgGx_8jARtH9T6PCe8UgcvE,99
 sunholo/invoke/async_class.py,sha256=J-qMgarhdi6nPQyOIFM5k0wF7nqr8aJGaz99-Ye6XGQ,3507
@@ -138,9 +138,9 @@ sunholo/vertex/init.py,sha256=1OQwcPBKZYBTDPdyU7IM4X4OmiXLdsNV30C-fee2scQ,2875
 sunholo/vertex/memory_tools.py,sha256=q_phxgGX2TG2j2MXNULF2xGzQnQPENwjPN9nZ_A9Gh0,7526
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
 sunholo/vertex/type_dict_to_json.py,sha256=uTzL4o9tJRao4u-gJOFcACgWGkBOtqACmb6ihvCErL8,4694
-sunholo-0.84.5.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.84.5.dist-info/METADATA,sha256=xXnKIJEmFg5fMhGrdA5Sc3cbyaUdJkLG8cxCDAVdH0A,7412
-sunholo-0.84.5.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-sunholo-0.84.5.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.84.5.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.84.5.dist-info/RECORD,,
+sunholo-0.84.7.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.84.7.dist-info/METADATA,sha256=7k9YTlJrqeB25j52O1Qr3kCfTAmu2hy5W_Tn62IepEE,7412
+sunholo-0.84.7.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+sunholo-0.84.7.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.84.7.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.84.7.dist-info/RECORD,,