sunholo 0.64.19__py3-none-any.whl → 0.66.1__py3-none-any.whl

sunholo/agents/swagger.py

@@ -158,9 +158,24 @@ def generate_swagger(vac_config, agent_config):
             log.warning(f"Failed to find URL stem for {vector_name}/{agent_type} - using {stem} instead")
         
         for method, endpoints in agent_config_paths.items():
-            if method not in ['get', 'post']:
-                log.warning(f"Skipping {endpoints}")
+            # controls if get requests are protected or not
+            do_auth = True
+
+            # default get: noauth, post: auth
+            if method not in ['get', 'post', 'get-auth', 'post-noauth']:
                 continue
+
+            if method == 'get-auth':
+                do_auth = True
+                method = 'get'
+            elif method == 'post-noauth':
+                do_auth = False
+                method = 'post'
+            elif method == 'get':
+                do_auth = False
+            elif method == 'post':
+                do_auth = True # not needed but to be clear whats happening
+
             for endpoint_key, endpoint_template in endpoints.items():
                 endpoint_template = endpoint_template.strip()
                 endpoint_address = endpoint_template.replace("{stem}", stem).replace("{vector_name}", vector_name).strip()
@@ -173,14 +188,17 @@ def generate_swagger(vac_config, agent_config):
 
                 operation_id = f"{method}_{agent_type}_{endpoint_key}_{vector_name}"
 
+                security = [{'ApiKeyAuth': []}] if do_auth else [{'None': []}]
+
                 swagger_template['paths'][endpoint_path][method] = {
                     'summary': f"{method.capitalize()} {vector_name}",
                     'operationId': operation_id,
                     'x-google-backend': {
                         'address': endpoint_address,
-                        'protocol': 'h2'
+                        'protocol': 'h2',
+                        'deadline':  180000  # Timeout of 3 minutes (1,800,00 milliseconds)
                     },
-                    'security': [{'ApiKeyAuth': []}] if method == 'post' else [{'None': []}],
+                    'security': security,
                     'responses': copy.deepcopy(agent_config_paths.get('response', {}).get(endpoint_key, {
                         '200': {
                             'description': 'Default - A successful response',
@@ -204,9 +222,26 @@ def generate_swagger(vac_config, agent_config):
             stem = f"${{{agent_type.upper()}_BACKEND_URL}}"
             log.warning(f"Failed to find URL stem for {vector_name}/{agent_type} - using {stem} instead")
         
+
         for method, endpoints in default_agent_config.items():
-            if method not in ['get', 'post']:
+            # controls if get requests are protected or not
+            do_auth = True
+
+            # default get: noauth, post: auth
+            if method not in ['get', 'post', 'get-auth', 'post-noauth']:
                 continue
+
+            if method == 'get-auth':
+                do_auth = True
+                method = 'get'
+            elif method == 'post-noauth':
+                do_auth = False
+                method = 'post'
+            elif method == 'get':
+                do_auth = False
+            elif method == 'post':
+                do_auth = True # not needed but to be clear whats happening
+
             for endpoint_key, endpoint_template in endpoints.items():
                 endpoint_template = endpoint_template.strip()
                 endpoint_address = endpoint_template.replace("{stem}", stem).replace("{vector_name}", vector_name).strip()
@@ -219,14 +254,17 @@ def generate_swagger(vac_config, agent_config):
                 
                 operation_id = f"{method}_{agent_type}_{endpoint_key}_{vector_name}"
 
+                security = [{'ApiKeyAuth': []}] if do_auth else [{'None': []}]
+
                 swagger_template['paths'][endpoint_path][method] = {
                     'summary': f"{method.capitalize()} {agent_type}",
                     'operationId': operation_id,
                     'x-google-backend': {
                         'address': endpoint_address,
-                        'protocol': 'h2'
+                        'protocol': 'h2',
+                        'deadline': 180000  # Timeout of 3 minutes (1,800,00 milliseconds)
                     },
-                    'security': [{'ApiKeyAuth': []}] if method == 'post' else [{'None': []}],
+                    'security': security,
                     'responses': copy.deepcopy(default_agent_config.get('response', {}).get(endpoint_key, {
                         '200': {
                             'description': 'Default - A successful response',

{sunholo-0.64.19.dist-info → sunholo-0.66.1.dist-info}/METADATA

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

{sunholo-0.64.19.dist-info → sunholo-0.66.1.dist-info}/RECORD

@@ -7,7 +7,7 @@ sunholo/agents/langserve.py,sha256=FdhQjorAY2bMn2rpuabNT6bU3uqSKWrl8DjpH3L_V7k,4
 sunholo/agents/pubsub.py,sha256=5hbbhbBGyVWRpt2sAGC5FEheYH1mCCwVUhZEB1S7vGg,1337
 sunholo/agents/route.py,sha256=V1HKXTvaNuYgjmT5_QgIQum4O7O0Mw497m6YiMpEzX0,2383
 sunholo/agents/special_commands.py,sha256=ecD5jrBVXo170sdgPILi0m_m_4nRFEv6qKn5zYEvEK8,6494
-sunholo/agents/swagger.py,sha256=nnaf9JtHeJwdzlpXmgFBpI71nf5oE7tSLG1VoZZj0SY,10261
+sunholo/agents/swagger.py,sha256=muFMQ1-9MtVlvSyUrLKjmULe4inC_QMb0Iofa14QnXw,11540
 sunholo/agents/fastapi/__init__.py,sha256=S_pj4_bTUmDGoq_exaREHlOKThi0zTuGT0VZY0YfODQ,88
 sunholo/agents/fastapi/base.py,sha256=clk76cHbUAvU0OYJrRfCWX_5f0ACbhDsIzYBhI3wyoE,2514
 sunholo/agents/fastapi/qna_routes.py,sha256=DgK4Btu5XriOC1JaRQ4G_nWEjJfnQ0J5pyLanF6eF1g,3857
@@ -104,9 +104,9 @@ sunholo/vertex/__init__.py,sha256=JvHcGFuv6R_nAhY2AdoqqhMpJ5ugeWPZ_svGhWrObBk,13
 sunholo/vertex/init.py,sha256=JDMUaBRdednzbKF-5p33qqLit2LMsvgvWW-NRz0AqO0,1801
 sunholo/vertex/memory_tools.py,sha256=8F1iTWnqEK9mX4W5RzCVKIjydIcNp6OFxjn_dtQ3GXo,5379
 sunholo/vertex/safety.py,sha256=3meAX0HyGZYrH7rXPUAHxtI_3w_zoy_RX7Shtkoa660,1275
-sunholo-0.64.19.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.64.19.dist-info/METADATA,sha256=8J1J8ApDb-Q3I3afrKM4RTck6U2_Lgb8K8ogxUZ8hTU,6000
-sunholo-0.64.19.dist-info/WHEEL,sha256=cpQTJ5IWu9CdaPViMhC9YzF8gZuS5-vlfoFihTBC86A,91
-sunholo-0.64.19.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.64.19.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.64.19.dist-info/RECORD,,
+sunholo-0.66.1.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.66.1.dist-info/METADATA,sha256=Mv0TUMCU4fHkOPMw2fCWKfiL6sJtkWG8MPDlA7JstiI,5998
+sunholo-0.66.1.dist-info/WHEEL,sha256=cpQTJ5IWu9CdaPViMhC9YzF8gZuS5-vlfoFihTBC86A,91
+sunholo-0.66.1.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.66.1.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.66.1.dist-info/RECORD,,