h2ogpte 1.6.43rc9__py3-none-any.whl → 1.6.44rc2__py3-none-any.whl

h2ogpte/__init__.py

@@ -3,7 +3,7 @@ from h2ogpte.h2ogpte import H2OGPTE
 from h2ogpte.h2ogpte_async import H2OGPTEAsync
 from h2ogpte.session_async import SessionAsync
 
-__version__ = "1.6.43rc9"
+__version__ = "1.6.44rc2"
 
 __all__ = [
     "H2OGPTE",

h2ogpte/h2ogpte.py

@@ -6927,6 +6927,37 @@ class H2OGPTE(H2OGPTESyncBase):
             )
 
     def add_agent_key(self, agent_keys: List[dict]) -> List[dict]:
+        """Create one or more agent keys for use with agent tools.
+
+        Processes a list of agent key configurations and creates each key.
+        Continues processing remaining keys if individual key creation fails.
+
+        Args:
+            agent_keys: List of key configuration dictionaries.
+
+                Expected structure::
+
+                    [
+                        {
+                            "name": str,
+                                # Display name for the key
+
+                            "value": str,
+                                # The actual key/token value
+
+                            "key_type": str,
+                                # Type of key ("private" or "shared")
+
+                            "description": str,
+                                # (Optional) Description of the key's purpose
+                        }
+                    ]
+
+        Returns:
+            List[dict]: List of created key results. Each successful creation
+                returns {"agent_key_id": str}. Failed creations are logged but
+                don't appear in results.
+        """
         result = []
         header = self._get_auth_header()
         with self._RESTClient(self) as rest_client:
@@ -6993,6 +7024,29 @@ class H2OGPTE(H2OGPTESyncBase):
     def assign_agent_key_for_tool(
         self, tool_dict_list: List[dict]
     ) -> Optional[List[Tuple]]:
+        """Assign agent keys to tools by creating associations between them.
+
+        Args:
+            tool_dict_list: List of dictionaries containing tool association data.
+                Each dictionary should have a "tool_dict" key with the association
+                configuration data for creating agent tool key associations.
+
+                Expected tool_dict structure::
+
+                    {
+                        "tool": str,        # Name of the tool (for example, "test_tool").
+                        "keys": list[dict], # List of key definitions. Each item is a dictionary with:
+                                           #   - "name": str
+                                           #       Environment variable name (for example, "TEST_KEY").
+                                           #   - "key_id": Any
+                                           #       Identifier assigned to the key (for example, agent_key_id).
+                    }
+
+        Returns:
+            Optional[List[Tuple]]: List of tuples containing association details.
+                Each tuple contains (associate_id, tool, key_name, key_id, user_id).
+                Returns None if no associations were created.
+        """
         result = []
         header = self._get_auth_header()
         with self._RESTClient(self) as rest_client:

h2ogpte/h2ogpte_async.py

@@ -7184,6 +7184,37 @@ class H2OGPTEAsync:
             )
 
     async def add_agent_key(self, agent_keys: List[dict]) -> List[dict]:
+        """Create one or more agent keys for use with agent tools.
+
+        Processes a list of agent key configurations and creates each key.
+        Continues processing remaining keys if individual key creation fails.
+
+        Args:
+            agent_keys: List of key configuration dictionaries.
+
+                Expected structure::
+
+                    [
+                        {
+                            "name": str,
+                                # Display name for the key
+
+                            "value": str,
+                                # The actual key/token value
+
+                            "key_type": str,
+                                # Type of key ("private" or "shared")
+
+                            "description": str,
+                                # (Optional) Description of the key's purpose
+                        }
+                    ]
+
+        Returns:
+            List[dict]: List of created key results. Each successful creation
+                returns {"agent_key_id": str}. Failed creations are logged but
+                don't appear in results.
+        """
         result = []
         header = await self._get_auth_header()
         async with self._RESTClient(self) as rest_client:
@@ -7250,6 +7281,29 @@ class H2OGPTEAsync:
     async def assign_agent_key_for_tool(
         self, tool_dict_list: List[dict]
     ) -> Optional[List[Tuple]]:
+        """Assign agent keys to tools by creating associations between them.
+
+        Args:
+            tool_dict_list: List of dictionaries containing tool association data.
+                Each dictionary should have a "tool_dict" key with the association
+                configuration data for creating agent tool key associations.
+
+                Expected tool_dict structure::
+
+                    {
+                        "tool": str,        # Name of the tool (for example, "test_tool").
+                        "keys": list[dict], # List of key definitions. Each item is a dictionary with:
+                                           #   - "name": str
+                                           #       Environment variable name (for example, "TEST_KEY").
+                                           #   - "key_id": Any
+                                           #       Identifier assigned to the key (for example, agent_key_id).
+                    }
+
+        Returns:
+            Optional[List[Tuple]]: List of tuples containing association details.
+                Each tuple contains (associate_id, tool, key_name, key_id, user_id).
+                Returns None if no associations were created.
+        """
         result = []
         header = await self._get_auth_header()
         async with self._RESTClient(self) as rest_client:

h2ogpte/rest_async/__init__.py

@@ -14,7 +14,7 @@
 """  # noqa: E501
 
 
-__version__ = "1.6.43-dev9"
+__version__ = "1.6.44-dev2"
 
 # import apis into sdk package
 from h2ogpte.rest_async.api.api_keys_api import APIKeysApi

h2ogpte/rest_async/api_client.py

@@ -90,7 +90,7 @@ class ApiClient:
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = 'OpenAPI-Generator/1.6.43-dev9/python'
+        self.user_agent = 'OpenAPI-Generator/1.6.44-dev2/python'
         self.client_side_validation = configuration.client_side_validation
 
     async def __aenter__(self):

h2ogpte/rest_async/configuration.py

@@ -499,7 +499,7 @@ class Configuration:
                "OS: {env}\n"\
                "Python Version: {pyversion}\n"\
                "Version of the API: v1.0.0\n"\
-               "SDK Package Version: 1.6.43-dev9".\
+               "SDK Package Version: 1.6.44-dev2".\
                format(env=sys.platform, pyversion=sys.version)
 
     def get_host_settings(self) -> List[HostSetting]:

h2ogpte/rest_sync/__init__.py

@@ -14,7 +14,7 @@
 """  # noqa: E501
 
 
-__version__ = "1.6.43-dev9"
+__version__ = "1.6.44-dev2"
 
 # import apis into sdk package
 from h2ogpte.rest_sync.api.api_keys_api import APIKeysApi

h2ogpte/rest_sync/api_client.py

@@ -90,7 +90,7 @@ class ApiClient:
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = 'OpenAPI-Generator/1.6.43-dev9/python'
+        self.user_agent = 'OpenAPI-Generator/1.6.44-dev2/python'
         self.client_side_validation = configuration.client_side_validation
 
     def __enter__(self):

h2ogpte/rest_sync/configuration.py

@@ -503,7 +503,7 @@ class Configuration:
                "OS: {env}\n"\
                "Python Version: {pyversion}\n"\
                "Version of the API: v1.0.0\n"\
-               "SDK Package Version: 1.6.43-dev9".\
+               "SDK Package Version: 1.6.44-dev2".\
                format(env=sys.platform, pyversion=sys.version)
 
     def get_host_settings(self) -> List[HostSetting]:

h2ogpte/session.py

@@ -149,6 +149,7 @@ class Session:
                     ssl_context=ssl_context,
                     open_timeout=self._open_timeout,
                     close_timeout=self._close_timeout,
+                    max_size=5 * 1024 * 1024,  # 5 MB limit for large responses
                 )
                 return self._connection
             except (ConnectionClosedError, InvalidURI, InvalidHandshake) as e:

h2ogpte/session_async.py

@@ -478,6 +478,7 @@ class SessionAsync:
                     open_timeout=self._open_timeout,
                     close_timeout=self._close_timeout,
                     ssl=ssl_context,
+                    max_size=5 * 1024 * 1024,  # 5 MB limit for large responses
                 )
                 return self._websocket
             except (ConnectionClosedError, InvalidURI, InvalidHandshake) as e:

h2ogpte/types.py

@@ -123,6 +123,8 @@ class ChatMessageFull(BaseModel):
     type_list: Optional[List[ChatMessageMeta]] = []
     has_references: bool
     total_references: int
+    collection_id: Optional[str] = None
+    collection_name: Optional[str] = None
     error: Optional[str] = None
 
 

{h2ogpte-1.6.43rc9.dist-info → h2ogpte-1.6.44rc2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: h2ogpte
-Version: 1.6.43rc9
+Version: 1.6.44rc2
 Summary: Client library for Enterprise h2oGPTe
 Author-email: "H2O.ai, Inc." <support@h2o.ai>
 Project-URL: Source, https://github.com/h2oai/h2ogpte

{h2ogpte-1.6.43rc9.dist-info → h2ogpte-1.6.44rc2.dist-info}/RECORD

@@ -1,13 +1,13 @@
-h2ogpte/__init__.py,sha256=j_E2gXwZIGy4HT31L3Z-hEB3wMFdCmCQiHS8N6__ePA,1524
+h2ogpte/__init__.py,sha256=41dhYDoY-nWEyPAY5eTqGt0RzE7T7bIpzr__XAnWyuI,1524
 h2ogpte/connectors.py,sha256=vkILsfW-tsLUn6KB6zgu_kjps4NMGgJpZ3OOaIjji04,8057
 h2ogpte/errors.py,sha256=XgLdfJO1fZ9Bf9rhUKpnvRzzvkNyan3Oc6WzGS6hCUA,1248
-h2ogpte/h2ogpte.py,sha256=DQC3Y1Lb9oef51hLK065g6-TSKQ7qiSICUYTwG7DCow,305935
-h2ogpte/h2ogpte_async.py,sha256=tzWqm9H92Jg01Ut3PdtbRUpysxbsa7uJS-9pplzbRXA,325832
+h2ogpte/h2ogpte.py,sha256=qUGUWHllcS-yYYIoR1TKW_8AuIOLqxWQ1aEfbYzAPPw,308265
+h2ogpte/h2ogpte_async.py,sha256=f2LQUXhuW7UjqQJ6kwgB0fF1XhQyW5sGem5xOU_IY3Q,328162
 h2ogpte/h2ogpte_sync_base.py,sha256=ftsVzpMqEsyi0UACMI-7H_EIYEx9JEdEUImbyjWy_Hc,15285
-h2ogpte/session.py,sha256=skOQa5XmCwdH1k7XdzKmc6jlwn9zcih-RVbAXnrwJ4g,32528
-h2ogpte/session_async.py,sha256=-V4rRh1diu5JD8IV9NYIKT_OGo7w-iYyHSnODygOp54,31274
+h2ogpte/session.py,sha256=BqJg3mWVeyz_ZLUJC_olzZzeLnRSaJwCH1NkXXfhg54,32608
+h2ogpte/session_async.py,sha256=UzNijTn3kZjAUYl_Jn5Oji4QrPTOpdX9KKByTmhLlK8,31354
 h2ogpte/shared_client.py,sha256=Zh24myL--5JDdrKoJPW4aeprHX6a_oB9o461Ho3hnU8,14691
-h2ogpte/types.py,sha256=umPY5ymhpLBxfLSqKY5cTDhmPxeDRIw_GT-yyXY3yng,15226
+h2ogpte/types.py,sha256=IRpRYrwKAGxiET3T-15R1heCRzpF0PXHwClC31GOuzM,15308
 h2ogpte/utils.py,sha256=Z9n57xxPu0KtsCzkJ9V_VgTW--oG_aXTLBgmXDWSdnM,3201
 h2ogpte/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h2ogpte/cli/main.py,sha256=Upf3t_5m1RqLh1jKGB6Gbyp3n9sujVny7sY-qxh2PYo,2722
@@ -41,10 +41,10 @@ h2ogpte/cli/ui/prompts.py,sha256=bJvRe_32KppQTK5bqnsrPh0RS4JaY9KkiV7y-3v8PMQ,538
 h2ogpte/cli/ui/status_bar.py,sha256=hs2MLvkg-y3Aiu3gWRtgMXf3jv3DGe7Y47ucgoBAP7Y,3852
 h2ogpte/cli/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h2ogpte/cli/utils/file_manager.py,sha256=ghNDX6G3Dr0vFvBYjbqx5o7qxq-pN8Vo2Rp1vyITfLo,13988
-h2ogpte/rest_async/__init__.py,sha256=pn91KQsrprFtMAZkD6u57t_atCeEuuHOfYcyBfO79MM,15203
-h2ogpte/rest_async/api_client.py,sha256=JOX7s12FP_qMr7tr2C093tvXrIROkuyCdA2jDagxVbU,29510
+h2ogpte/rest_async/__init__.py,sha256=TvbG7vh_bWZ1DXrKsyNXLdA7ko8Ma2mbt98XKZE10RQ,15203
+h2ogpte/rest_async/api_client.py,sha256=9weDOIQYDGenS3a7Hz09cwEnX4y8iD1Bmy0dW1Y_iFg,29510
 h2ogpte/rest_async/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
-h2ogpte/rest_async/configuration.py,sha256=0I0JtgDTl63q0EfEpSbAz7vMi0gHHVnmMXQna6so56E,19567
+h2ogpte/rest_async/configuration.py,sha256=TP67x9i9MlSQHOhFWjIGbvgwV_suJnOr-r9NEEC70Lg,19567
 h2ogpte/rest_async/exceptions.py,sha256=aSDc-0lURtyQjf5HGa7_Ta0nATxKxfHW3huDA2Zdj6o,8370
 h2ogpte/rest_async/rest.py,sha256=mdjDwzJ1kiaYtONUfDRqKsRPw5-tG6eyZV2P1yBuwRo,9147
 h2ogpte/rest_async/api/__init__.py,sha256=R_x57GGyaSgxZyrJOyOt551TodbRSQf3T7VrraQc-84,973
@@ -201,10 +201,10 @@ h2ogpte/rest_async/models/user_deletion_request.py,sha256=z7gD8XKOGwwg782TRzXJii
 h2ogpte/rest_async/models/user_info.py,sha256=ef59Eh9k42JUY3X2RnCrwYR7sc_8lXT1vRLGoNz3uTU,4489
 h2ogpte/rest_async/models/user_job_details.py,sha256=kzu8fLxVsRMgnyt6dLr0VWjlIoE3i1VRpGR9nDxFyk4,4985
 h2ogpte/rest_async/models/user_permission.py,sha256=9ffijaF3U3SYz_T_kcqHPJUfIZFkpCH0vBGboPjsg2o,4646
-h2ogpte/rest_sync/__init__.py,sha256=qHr8xGePQrfAz4fdPfhKqt7J2kMIU59I49Bve36tfog,15042
-h2ogpte/rest_sync/api_client.py,sha256=gHhM3KKT9qZmCOdeR2ZQIJ7YDXmnG3PwbWAfekYGX-A,29397
+h2ogpte/rest_sync/__init__.py,sha256=tZObsHV1yFgX_Q1x3VvUWH2pxXRHIr7rY0XSuri7hGw,15042
+h2ogpte/rest_sync/api_client.py,sha256=2wYQyyFN94Kcuged62M8yZJwSJwyVVQhorpgCGF85KE,29397
 h2ogpte/rest_sync/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
-h2ogpte/rest_sync/configuration.py,sha256=iztT8c9fqaUpXeilq3GHQl99WqCATvUv4oQpf7kOleg,19850
+h2ogpte/rest_sync/configuration.py,sha256=3nixVjOO5LmNhl-jNGn99ex8MksePUsNZhlAyBY3xxw,19850
 h2ogpte/rest_sync/exceptions.py,sha256=aSDc-0lURtyQjf5HGa7_Ta0nATxKxfHW3huDA2Zdj6o,8370
 h2ogpte/rest_sync/rest.py,sha256=evRzviTYC_fsrpTtFlGvruXmquH9C0jDn-oQrGrE5A0,11314
 h2ogpte/rest_sync/api/__init__.py,sha256=ZuLQQtyiXnP5UOwTlIOYLGLQq1BG_0PEkzC9s698vjM,958
@@ -361,8 +361,8 @@ h2ogpte/rest_sync/models/user_deletion_request.py,sha256=z7gD8XKOGwwg782TRzXJiiP
 h2ogpte/rest_sync/models/user_info.py,sha256=ef59Eh9k42JUY3X2RnCrwYR7sc_8lXT1vRLGoNz3uTU,4489
 h2ogpte/rest_sync/models/user_job_details.py,sha256=9cbhpgLMDpar-aTOaY5Ygud-8Kbi23cLNldTGab0Sd8,4984
 h2ogpte/rest_sync/models/user_permission.py,sha256=9ffijaF3U3SYz_T_kcqHPJUfIZFkpCH0vBGboPjsg2o,4646
-h2ogpte-1.6.43rc9.dist-info/METADATA,sha256=5ebWY4JvX7TTv6Lnt16lshefF8oout7gdWRECge83pw,8615
-h2ogpte-1.6.43rc9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-h2ogpte-1.6.43rc9.dist-info/entry_points.txt,sha256=BlaqX2SXJanrOGqNYwnzvCxHGNadM7RBI4pW4rVo5z4,54
-h2ogpte-1.6.43rc9.dist-info/top_level.txt,sha256=vXV4JnNwFWFAqTWyHrH-cGIQqbCcEDG9-BbyNn58JpM,8
-h2ogpte-1.6.43rc9.dist-info/RECORD,,
+h2ogpte-1.6.44rc2.dist-info/METADATA,sha256=qs4of6FD7lu3W-hS-ZgnlsCReV6WjInTBuX8YNqnbAQ,8615
+h2ogpte-1.6.44rc2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+h2ogpte-1.6.44rc2.dist-info/entry_points.txt,sha256=BlaqX2SXJanrOGqNYwnzvCxHGNadM7RBI4pW4rVo5z4,54
+h2ogpte-1.6.44rc2.dist-info/top_level.txt,sha256=vXV4JnNwFWFAqTWyHrH-cGIQqbCcEDG9-BbyNn58JpM,8
+h2ogpte-1.6.44rc2.dist-info/RECORD,,