sunholo 0.82.2__tar.gz → 0.83.0__tar.gz

{sunholo-0.82.2 → sunholo-0.83.0}/PKG-INFO

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

{sunholo-0.82.2 → sunholo-0.83.0}/setup.py

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

{sunholo-0.82.2 → sunholo-0.83.0}/sunholo/agents/flask/qna_routes.py

@@ -23,6 +23,7 @@ from ...streaming import start_streaming_chat
 from ...archive import archive_qa
 from ...custom_logging import log
 from ...utils.config import load_config
+from ...utils import ConfigManager
 from ...utils.version import sunholo_version
 import os
 from ...gcs.add_file import add_file_to_gcs, handle_base64_image
@@ -183,6 +184,15 @@ def register_qna_routes(app, stream_interpreter, vac_interpreter):
             generation.end(output=response)
             span.end(output=response)
             trace.update(output=response)
+        
+        #if 'user_id' in all_input["kwargs"]:
+        #    kwargs = all_input["kwargs"]
+        #    config = ConfigManager(vector_name)
+        #    add_user_history_rag(kwargs.pop('user_id'), 
+        #                         config, 
+        #                         question=all_input.pop("user_input"), 
+        #                         answer=response.get('answer'), 
+        #                         metadata=all_input)
 
         return response
 

{sunholo-0.82.2 → sunholo-0.83.0}/sunholo/cli/chat_vac.py

@@ -411,7 +411,7 @@ def resolve_service_url(args, no_config=False):
 
 def vac_command(args):
 
-    config = ConfigManager(args.vac_name)
+    
 
     if args.action == 'list':
 
@@ -426,6 +426,7 @@ def vac_command(args):
         return
     
     elif args.action == 'chat':
+        config = ConfigManager(args.vac_name)
         service_url = resolve_service_url(args)
         agent_name   = config.vacConfig("agent")
 
@@ -466,7 +467,6 @@ def vac_command(args):
 
         invoke_vac(service_url, args.data, is_file=args.is_file)
 
-
 def list_cloud_run_services(project, region):
     """
     Lists all Cloud Run services the user has access to in a specific project and region.
@@ -583,4 +583,5 @@ def setup_vac_subparser(subparsers):
     invoke_parser.add_argument('data', help='Data to send to the VAC service (as JSON string).')
     invoke_parser.add_argument('--is-file', action='store_true', help='Indicate if the data argument is a file path')
 
-    vac_parser.set_defaults(func=vac_command)
+    # If no subcommand is provided, print the help message
+    vac_parser.set_defaults(func=lambda args: vac_parser.print_help() if args.action is None else vac_command(args))

{sunholo-0.82.2 → sunholo-0.83.0}/sunholo/cli/cli.py

@@ -13,6 +13,7 @@ from .vertex import setup_vertex_subparser
 from ..llamaindex import setup_llamaindex_subparser
 
 from ..utils import ConfigManager
+from ..utils.version import sunholo_version
 
 from ..custom_logging import log
 
@@ -63,7 +64,8 @@ def main(args=None):
     parser.add_argument('--debug', action='store_true', help='Enable debug output')
     parser.add_argument('--project', default=default_project, help='GCP project to list Cloud Run services from.')
     parser.add_argument('--region', default=default_region, help='Region to list Cloud Run services from.')
-    
+    parser.add_argument('--version', action='store_true', help='Show the version and exit') 
+
     subparsers = parser.add_subparsers(title='commands', 
                                        description='Valid commands', 
                                        help='Commands', 
@@ -102,6 +104,10 @@ def main(args=None):
         log.setLevel(logging.WARNING)
         logging.getLogger().setLevel(logging.WARNING)
 
+    if args.version:
+        sunholo_version()
+        return
+
     if hasattr(args, 'func'):
         args.func(args)
     else:

{sunholo-0.82.2 → sunholo-0.83.0}/sunholo/database/alloydb_client.py

@@ -152,6 +152,20 @@ class AlloyDBClient:
         documents = await self.execute_sql_async(query)
         return documents
 
+    def get_sources_from_docstore(self, sources, vector_name, search_type="OR", just_source_name=False):
+        """Fetches sources from the docstore asynchronously."""
+        if just_source_name:
+            query = self._list_sources_from_docstore(sources, vector_name=vector_name, search_type=search_type)
+        else:
+            query = self._get_sources_from_docstore(sources, vector_name=vector_name, search_type=search_type)
+
+        if not query:
+            return []
+
+        documents = self.execute_sql(query)
+        
+        return documents
+
     def _get_sources_from_docstore(self, sources, vector_name, search_type="OR"):
         """Helper function to build the SQL query for fetching sources."""
         if not sources:

{sunholo-0.82.2 → sunholo-0.83.0}/sunholo/streaming/streaming.py

@@ -22,7 +22,7 @@ from .content_buffer import ContentBuffer, BufferStreamingStdOutCallbackHandler
 from ..qna.parsers import parse_output
 
 from ..custom_logging import log
-from ..utils import load_config_key
+from ..utils import load_config_key, ConfigManager
 from ..utils.parsers import check_kwargs_support
 
 from .langserve import parse_langserve_token, parse_langserve_token_async
@@ -248,8 +248,9 @@ def generate_proxy_stream(stream_to_f, user_input, vector_name, chat_history, ge
         ):
             print(output)  # Process each streaming output chunk
     """
-    agent = load_config_key("agent", vector_name=vector_name, kind="vacConfig")
-    agent_type = load_config_key("agent_type", vector_name=vector_name, kind="vacConfig")
+    config = ConfigManager(vector_name)
+    agent = config.vacConfig("agent")
+    agent_type = config.vacConfig("agent_type")
 
     def generate():
         json_buffer = ""
@@ -306,9 +307,10 @@ async def generate_proxy_stream_async(stream_to_f, user_input, vector_name, chat
         ):
             print(output)  # Process each streaming output chunk
     """
-    agent = load_config_key("agent", vector_name=vector_name, kind="vacConfig")
-    agent_type = load_config_key("agent_type", vector_name=vector_name, kind="vacConfig")
-
+    config = ConfigManager(vector_name)
+    agent = config.vacConfig("agent")
+    agent_type = config.vacConfig("agent_type")
+    
     async def generate():
         json_buffer = ""
         inside_json = False

{sunholo-0.82.2 → sunholo-0.83.0}/sunholo.egg-info/PKG-INFO

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