beamlit 0.0.20rc5__py3-none-any.whl → 0.0.20rc7__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (164) hide show
  1. beamlit/agents/__init__.py +4 -0
  2. beamlit/agents/chat.py +87 -0
  3. beamlit/agents/decorator.py +147 -0
  4. beamlit/api/agents/delete_agent_history.py +6 -2
  5. beamlit/api/agents/get_agent_deployment_logs.py +11 -11
  6. beamlit/api/agents/get_agent_history.py +6 -2
  7. beamlit/api/agents/list_agent_deployment_history.py +11 -11
  8. beamlit/api/agents/list_agent_deployments.py +11 -11
  9. beamlit/api/agents/list_agents.py +11 -11
  10. beamlit/api/agents/put_agent_history.py +6 -2
  11. beamlit/api/environments/list_environments.py +11 -11
  12. beamlit/api/functions/get_function_deployment_logs.py +11 -11
  13. beamlit/api/functions/list_function_deployments.py +11 -11
  14. beamlit/api/functions/list_functions.py +11 -11
  15. beamlit/api/history/get_agents_history.py +11 -11
  16. beamlit/api/history/list_agents_history.py +11 -11
  17. beamlit/api/integrations/list_integration_connections.py +11 -11
  18. beamlit/api/invitations/list_all_pending_invitations.py +11 -11
  19. beamlit/api/locations/list_locations.py +11 -11
  20. beamlit/api/model_providers/list_model_providers.py +11 -11
  21. beamlit/api/models/get_model_deployment_logs.py +11 -11
  22. beamlit/api/models/list_model_deployments.py +11 -11
  23. beamlit/api/models/list_models.py +11 -11
  24. beamlit/api/policies/list_policies.py +11 -11
  25. beamlit/api/service_accounts/get_workspace_service_accounts.py +11 -11
  26. beamlit/api/service_accounts/list_api_keys_for_service_account.py +11 -11
  27. beamlit/api/store/list_store_agents.py +11 -11
  28. beamlit/api/store/list_store_functions.py +11 -11
  29. beamlit/api/workspaces/list_workspace_users.py +11 -11
  30. beamlit/api/workspaces/list_workspaces.py +11 -11
  31. beamlit/authentication/__init__.py +26 -10
  32. beamlit/authentication/apikey.py +4 -4
  33. beamlit/authentication/authentication.py +36 -5
  34. beamlit/authentication/clientcredentials.py +18 -20
  35. beamlit/authentication/credentials.py +18 -11
  36. beamlit/authentication/device_mode.py +15 -12
  37. beamlit/common/__init__.py +13 -0
  38. beamlit/common/generate.py +31 -19
  39. beamlit/common/logger.py +9 -12
  40. beamlit/common/secrets.py +11 -0
  41. beamlit/common/settings.py +78 -30
  42. beamlit/common/utils.py +4 -2
  43. beamlit/functions/__init__.py +5 -0
  44. beamlit/functions/decorator.py +90 -0
  45. beamlit/models/acl.py +2 -2
  46. beamlit/models/agent.py +3 -3
  47. beamlit/models/agent_chain.py +2 -2
  48. beamlit/models/agent_configuration.py +2 -2
  49. beamlit/models/agent_deployment.py +25 -25
  50. beamlit/models/agent_deployment_configuration.py +2 -2
  51. beamlit/models/agent_deployment_history.py +5 -5
  52. beamlit/models/agent_deployment_history_event.py +9 -9
  53. beamlit/models/agent_deployment_pod_template.py +2 -2
  54. beamlit/models/agent_release.py +2 -2
  55. beamlit/models/agent_spec.py +15 -5
  56. beamlit/models/agent_with_deployments.py +6 -6
  57. beamlit/models/api_key.py +2 -2
  58. beamlit/models/authentication_provider_model.py +6 -6
  59. beamlit/models/authentication_provider_organization.py +2 -2
  60. beamlit/models/configuration.py +10 -10
  61. beamlit/models/continent.py +2 -2
  62. beamlit/models/core_spec.py +9 -3
  63. beamlit/models/country.py +2 -2
  64. beamlit/models/create_api_key_for_service_account_body.py +2 -2
  65. beamlit/models/create_workspace_service_account_body.py +2 -2
  66. beamlit/models/create_workspace_service_account_response_200.py +2 -2
  67. beamlit/models/delete_workspace_service_account_response_200.py +2 -2
  68. beamlit/models/deployment_configuration.py +2 -2
  69. beamlit/models/deployment_configurations.py +2 -2
  70. beamlit/models/deployment_serverless_config.py +2 -2
  71. beamlit/models/deployment_serverless_config_type_0.py +3 -1
  72. beamlit/models/environment.py +7 -7
  73. beamlit/models/environment_metrics.py +8 -8
  74. beamlit/models/flavor.py +9 -9
  75. beamlit/models/function.py +3 -3
  76. beamlit/models/function_configuration.py +2 -2
  77. beamlit/models/function_deployment.py +24 -24
  78. beamlit/models/function_deployment_configuration.py +2 -2
  79. beamlit/models/function_deployment_pod_template.py +2 -2
  80. beamlit/models/function_kit.py +5 -5
  81. beamlit/models/function_provider_ref.py +2 -2
  82. beamlit/models/function_release.py +2 -2
  83. beamlit/models/function_spec.py +9 -3
  84. beamlit/models/function_with_deployments.py +6 -6
  85. beamlit/models/get_workspace_service_accounts_response_200_item.py +2 -2
  86. beamlit/models/increase_and_rate_metric.py +5 -5
  87. beamlit/models/integration.py +12 -12
  88. beamlit/models/integration_config.py +2 -2
  89. beamlit/models/integration_connection.py +5 -5
  90. beamlit/models/integration_connection_config.py +2 -2
  91. beamlit/models/integration_connection_secret.py +2 -2
  92. beamlit/models/integration_model.py +6 -6
  93. beamlit/models/integration_secret.py +2 -2
  94. beamlit/models/invite_workspace_user_body.py +2 -2
  95. beamlit/models/labels_type_0.py +2 -2
  96. beamlit/models/location.py +5 -5
  97. beamlit/models/location_response.py +5 -5
  98. beamlit/models/metric.py +2 -2
  99. beamlit/models/metrics.py +8 -8
  100. beamlit/models/model.py +3 -3
  101. beamlit/models/model_deployment.py +18 -18
  102. beamlit/models/model_deployment_log.py +2 -2
  103. beamlit/models/model_deployment_metrics.py +9 -9
  104. beamlit/models/model_deployment_metrics_inference_per_second_per_region.py +5 -5
  105. beamlit/models/model_deployment_metrics_query_per_second_per_region_per_code.py +3 -3
  106. beamlit/models/model_deployment_pod_template.py +2 -2
  107. beamlit/models/model_metrics.py +6 -6
  108. beamlit/models/model_provider.py +11 -11
  109. beamlit/models/model_provider_ref.py +2 -2
  110. beamlit/models/model_release.py +2 -2
  111. beamlit/models/model_spec.py +9 -3
  112. beamlit/models/model_with_deployments.py +6 -6
  113. beamlit/models/pending_invitation.py +2 -2
  114. beamlit/models/pending_invitation_accept.py +3 -3
  115. beamlit/models/pending_invitation_render.py +5 -5
  116. beamlit/models/pending_invitation_render_invited_by.py +2 -2
  117. beamlit/models/pending_invitation_render_workspace.py +2 -2
  118. beamlit/models/pending_invitation_workspace_details.py +6 -6
  119. beamlit/models/policy.py +20 -20
  120. beamlit/models/policy_location.py +9 -9
  121. beamlit/models/policy_spec.py +9 -3
  122. beamlit/models/provider_config.py +7 -7
  123. beamlit/models/qps.py +2 -2
  124. beamlit/models/resource_deployment_log.py +2 -2
  125. beamlit/models/resource_deployment_metrics.py +16 -16
  126. beamlit/models/resource_deployment_metrics_inference_per_region.py +5 -5
  127. beamlit/models/resource_deployment_metrics_inference_per_region_type_0.py +6 -2
  128. beamlit/models/resource_deployment_metrics_inference_per_second_per_region.py +5 -5
  129. beamlit/models/resource_deployment_metrics_inference_per_second_per_region_type_0.py +6 -2
  130. beamlit/models/resource_deployment_metrics_query_per_region_per_code.py +3 -3
  131. beamlit/models/resource_deployment_metrics_query_per_second_per_region_per_code.py +3 -3
  132. beamlit/models/resource_environment_metrics.py +62 -24
  133. beamlit/models/resource_environment_metrics_inference_per_second_per_region.py +6 -2
  134. beamlit/models/resource_metrics.py +10 -10
  135. beamlit/models/runtime.py +23 -23
  136. beamlit/models/runtime_readiness_probe.py +2 -2
  137. beamlit/models/runtime_resources.py +2 -2
  138. beamlit/models/serverless_config.py +2 -2
  139. beamlit/models/standard_fields_dynamo_db.py +2 -2
  140. beamlit/models/store_agent.py +6 -6
  141. beamlit/models/store_agent_configuration.py +2 -2
  142. beamlit/models/store_agent_labels.py +2 -2
  143. beamlit/models/store_configuration.py +16 -16
  144. beamlit/models/store_configuration_option.py +2 -2
  145. beamlit/models/store_function.py +12 -12
  146. beamlit/models/store_function_configuration.py +2 -2
  147. beamlit/models/store_function_kit.py +5 -5
  148. beamlit/models/store_function_labels.py +2 -2
  149. beamlit/models/store_function_parameter.py +9 -9
  150. beamlit/models/update_workspace_service_account_body.py +2 -2
  151. beamlit/models/update_workspace_service_account_response_200.py +2 -2
  152. beamlit/models/update_workspace_user_role_body.py +2 -2
  153. beamlit/models/workspace.py +3 -3
  154. beamlit/models/workspace_labels.py +2 -2
  155. beamlit/models/workspace_user.py +2 -2
  156. beamlit/run.py +6 -6
  157. beamlit/serve/app.py +73 -0
  158. beamlit/serve/middlewares/__init__.py +4 -0
  159. beamlit/serve/middlewares/accesslog.py +14 -0
  160. beamlit/serve/middlewares/processtime.py +12 -0
  161. {beamlit-0.0.20rc5.dist-info → beamlit-0.0.20rc7.dist-info}/METADATA +7 -1
  162. beamlit-0.0.20rc7.dist-info/RECORD +293 -0
  163. beamlit-0.0.20rc5.dist-info/RECORD +0 -282
  164. {beamlit-0.0.20rc5.dist-info → beamlit-0.0.20rc7.dist-info}/WHEEL +0 -0
beamlit/api/service_accounts/get_workspace_service_accounts.py CHANGED
@@ -1,5 +1,5 @@
1
1
  from http import HTTPStatus
2
- from typing import Any, List, Optional, Union
2
+ from typing import Any, Optional, Union
3
3
 
4
4
  import httpx
5
5
 
@@ -20,7 +20,7 @@ def _get_kwargs() -> dict[str, Any]:
20
20
 
21
21
  def _parse_response(
22
22
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
23
- ) -> Optional[List["GetWorkspaceServiceAccountsResponse200Item"]]:
23
+ ) -> Optional[list["GetWorkspaceServiceAccountsResponse200Item"]]:
24
24
  if response.status_code == 200:
25
25
  response_200 = []
26
26
  _response_200 = response.json()
@@ -38,7 +38,7 @@ def _parse_response(
38
38
 
39
39
  def _build_response(
40
40
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
41
- ) -> Response[List["GetWorkspaceServiceAccountsResponse200Item"]]:
41
+ ) -> Response[list["GetWorkspaceServiceAccountsResponse200Item"]]:
42
42
  return Response(
43
43
  status_code=HTTPStatus(response.status_code),
44
44
  content=response.content,
@@ -50,7 +50,7 @@ def _build_response(
50
50
  def sync_detailed(
51
51
  *,
52
52
  client: AuthenticatedClient,
53
- ) -> Response[List["GetWorkspaceServiceAccountsResponse200Item"]]:
53
+ ) -> Response[list["GetWorkspaceServiceAccountsResponse200Item"]]:
54
54
  """Get workspace service accounts
55
55
 
56
56
  Returns a list of all service accounts in the workspace.
@@ -60,7 +60,7 @@ def sync_detailed(
60
60
  httpx.TimeoutException: If the request takes longer than Client.timeout.
61
61
 
62
62
  Returns:
63
- Response[List['GetWorkspaceServiceAccountsResponse200Item']]
63
+ Response[list['GetWorkspaceServiceAccountsResponse200Item']]
64
64
  """
65
65
 
66
66
  kwargs = _get_kwargs()
@@ -75,7 +75,7 @@ def sync_detailed(
75
75
  def sync(
76
76
  *,
77
77
  client: AuthenticatedClient,
78
- ) -> Optional[List["GetWorkspaceServiceAccountsResponse200Item"]]:
78
+ ) -> Optional[list["GetWorkspaceServiceAccountsResponse200Item"]]:
79
79
  """Get workspace service accounts
80
80
 
81
81
  Returns a list of all service accounts in the workspace.
@@ -85,7 +85,7 @@ def sync(
85
85
  httpx.TimeoutException: If the request takes longer than Client.timeout.
86
86
 
87
87
  Returns:
88
- List['GetWorkspaceServiceAccountsResponse200Item']
88
+ list['GetWorkspaceServiceAccountsResponse200Item']
89
89
  """
90
90
 
91
91
  return sync_detailed(
@@ -96,7 +96,7 @@ def sync(
96
96
  async def asyncio_detailed(
97
97
  *,
98
98
  client: AuthenticatedClient,
99
- ) -> Response[List["GetWorkspaceServiceAccountsResponse200Item"]]:
99
+ ) -> Response[list["GetWorkspaceServiceAccountsResponse200Item"]]:
100
100
  """Get workspace service accounts
101
101
 
102
102
  Returns a list of all service accounts in the workspace.
@@ -106,7 +106,7 @@ async def asyncio_detailed(
106
106
  httpx.TimeoutException: If the request takes longer than Client.timeout.
107
107
 
108
108
  Returns:
109
- Response[List['GetWorkspaceServiceAccountsResponse200Item']]
109
+ Response[list['GetWorkspaceServiceAccountsResponse200Item']]
110
110
  """
111
111
 
112
112
  kwargs = _get_kwargs()
@@ -119,7 +119,7 @@ async def asyncio_detailed(
119
119
  async def asyncio(
120
120
  *,
121
121
  client: AuthenticatedClient,
122
- ) -> Optional[List["GetWorkspaceServiceAccountsResponse200Item"]]:
122
+ ) -> Optional[list["GetWorkspaceServiceAccountsResponse200Item"]]:
123
123
  """Get workspace service accounts
124
124
 
125
125
  Returns a list of all service accounts in the workspace.
@@ -129,7 +129,7 @@ async def asyncio(
129
129
  httpx.TimeoutException: If the request takes longer than Client.timeout.
130
130
 
131
131
  Returns:
132
- List['GetWorkspaceServiceAccountsResponse200Item']
132
+ list['GetWorkspaceServiceAccountsResponse200Item']
133
133
  """
134
134
 
135
135
  return (
beamlit/api/service_accounts/list_api_keys_for_service_account.py CHANGED
@@ -1,5 +1,5 @@
1
1
  from http import HTTPStatus
2
- from typing import Any, List, Optional, Union
2
+ from typing import Any, Optional, Union
3
3
 
4
4
  import httpx
5
5
 
@@ -22,7 +22,7 @@ def _get_kwargs(
22
22
 
23
23
  def _parse_response(
24
24
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
25
- ) -> Optional[List["ApiKey"]]:
25
+ ) -> Optional[list["ApiKey"]]:
26
26
  if response.status_code == 200:
27
27
  response_200 = []
28
28
  _response_200 = response.json()
@@ -40,7 +40,7 @@ def _parse_response(
40
40
 
41
41
  def _build_response(
42
42
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
43
- ) -> Response[List["ApiKey"]]:
43
+ ) -> Response[list["ApiKey"]]:
44
44
  return Response(
45
45
  status_code=HTTPStatus(response.status_code),
46
46
  content=response.content,
@@ -53,7 +53,7 @@ def sync_detailed(
53
53
  client_id: str,
54
54
  *,
55
55
  client: AuthenticatedClient,
56
- ) -> Response[List["ApiKey"]]:
56
+ ) -> Response[list["ApiKey"]]:
57
57
  """List API keys for service account
58
58
 
59
59
  Returns a list of all API keys for a service account.
@@ -66,7 +66,7 @@ def sync_detailed(
66
66
  httpx.TimeoutException: If the request takes longer than Client.timeout.
67
67
 
68
68
  Returns:
69
- Response[List['ApiKey']]
69
+ Response[list['ApiKey']]
70
70
  """
71
71
 
72
72
  kwargs = _get_kwargs(
@@ -84,7 +84,7 @@ def sync(
84
84
  client_id: str,
85
85
  *,
86
86
  client: AuthenticatedClient,
87
- ) -> Optional[List["ApiKey"]]:
87
+ ) -> Optional[list["ApiKey"]]:
88
88
  """List API keys for service account
89
89
 
90
90
  Returns a list of all API keys for a service account.
@@ -97,7 +97,7 @@ def sync(
97
97
  httpx.TimeoutException: If the request takes longer than Client.timeout.
98
98
 
99
99
  Returns:
100
- List['ApiKey']
100
+ list['ApiKey']
101
101
  """
102
102
 
103
103
  return sync_detailed(
@@ -110,7 +110,7 @@ async def asyncio_detailed(
110
110
  client_id: str,
111
111
  *,
112
112
  client: AuthenticatedClient,
113
- ) -> Response[List["ApiKey"]]:
113
+ ) -> Response[list["ApiKey"]]:
114
114
  """List API keys for service account
115
115
 
116
116
  Returns a list of all API keys for a service account.
@@ -123,7 +123,7 @@ async def asyncio_detailed(
123
123
  httpx.TimeoutException: If the request takes longer than Client.timeout.
124
124
 
125
125
  Returns:
126
- Response[List['ApiKey']]
126
+ Response[list['ApiKey']]
127
127
  """
128
128
 
129
129
  kwargs = _get_kwargs(
@@ -139,7 +139,7 @@ async def asyncio(
139
139
  client_id: str,
140
140
  *,
141
141
  client: AuthenticatedClient,
142
- ) -> Optional[List["ApiKey"]]:
142
+ ) -> Optional[list["ApiKey"]]:
143
143
  """List API keys for service account
144
144
 
145
145
  Returns a list of all API keys for a service account.
@@ -152,7 +152,7 @@ async def asyncio(
152
152
  httpx.TimeoutException: If the request takes longer than Client.timeout.
153
153
 
154
154
  Returns:
155
- List['ApiKey']
155
+ list['ApiKey']
156
156
  """
157
157
 
158
158
  return (
beamlit/api/store/list_store_agents.py CHANGED
@@ -1,5 +1,5 @@
1
1
  from http import HTTPStatus
2
- from typing import Any, List, Optional, Union
2
+ from typing import Any, Optional, Union
3
3
 
4
4
  import httpx
5
5
 
@@ -20,7 +20,7 @@ def _get_kwargs() -> dict[str, Any]:
20
20
 
21
21
  def _parse_response(
22
22
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
23
- ) -> Optional[List["StoreAgent"]]:
23
+ ) -> Optional[list["StoreAgent"]]:
24
24
  if response.status_code == 200:
25
25
  response_200 = []
26
26
  _response_200 = response.json()
@@ -38,7 +38,7 @@ def _parse_response(
38
38
 
39
39
  def _build_response(
40
40
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
41
- ) -> Response[List["StoreAgent"]]:
41
+ ) -> Response[list["StoreAgent"]]:
42
42
  return Response(
43
43
  status_code=HTTPStatus(response.status_code),
44
44
  content=response.content,
@@ -50,7 +50,7 @@ def _build_response(
50
50
  def sync_detailed(
51
51
  *,
52
52
  client: Union[AuthenticatedClient, Client],
53
- ) -> Response[List["StoreAgent"]]:
53
+ ) -> Response[list["StoreAgent"]]:
54
54
  """List all store agent
55
55
 
56
56
  Raises:
@@ -58,7 +58,7 @@ def sync_detailed(
58
58
  httpx.TimeoutException: If the request takes longer than Client.timeout.
59
59
 
60
60
  Returns:
61
- Response[List['StoreAgent']]
61
+ Response[list['StoreAgent']]
62
62
  """
63
63
 
64
64
  kwargs = _get_kwargs()
@@ -73,7 +73,7 @@ def sync_detailed(
73
73
  def sync(
74
74
  *,
75
75
  client: Union[AuthenticatedClient, Client],
76
- ) -> Optional[List["StoreAgent"]]:
76
+ ) -> Optional[list["StoreAgent"]]:
77
77
  """List all store agent
78
78
 
79
79
  Raises:
@@ -81,7 +81,7 @@ def sync(
81
81
  httpx.TimeoutException: If the request takes longer than Client.timeout.
82
82
 
83
83
  Returns:
84
- List['StoreAgent']
84
+ list['StoreAgent']
85
85
  """
86
86
 
87
87
  return sync_detailed(
@@ -92,7 +92,7 @@ def sync(
92
92
  async def asyncio_detailed(
93
93
  *,
94
94
  client: Union[AuthenticatedClient, Client],
95
- ) -> Response[List["StoreAgent"]]:
95
+ ) -> Response[list["StoreAgent"]]:
96
96
  """List all store agent
97
97
 
98
98
  Raises:
@@ -100,7 +100,7 @@ async def asyncio_detailed(
100
100
  httpx.TimeoutException: If the request takes longer than Client.timeout.
101
101
 
102
102
  Returns:
103
- Response[List['StoreAgent']]
103
+ Response[list['StoreAgent']]
104
104
  """
105
105
 
106
106
  kwargs = _get_kwargs()
@@ -113,7 +113,7 @@ async def asyncio_detailed(
113
113
  async def asyncio(
114
114
  *,
115
115
  client: Union[AuthenticatedClient, Client],
116
- ) -> Optional[List["StoreAgent"]]:
116
+ ) -> Optional[list["StoreAgent"]]:
117
117
  """List all store agent
118
118
 
119
119
  Raises:
@@ -121,7 +121,7 @@ async def asyncio(
121
121
  httpx.TimeoutException: If the request takes longer than Client.timeout.
122
122
 
123
123
  Returns:
124
- List['StoreAgent']
124
+ list['StoreAgent']
125
125
  """
126
126
 
127
127
  return (
beamlit/api/store/list_store_functions.py CHANGED
@@ -1,5 +1,5 @@
1
1
  from http import HTTPStatus
2
- from typing import Any, List, Optional, Union
2
+ from typing import Any, Optional, Union
3
3
 
4
4
  import httpx
5
5
 
@@ -20,7 +20,7 @@ def _get_kwargs() -> dict[str, Any]:
20
20
 
21
21
  def _parse_response(
22
22
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
23
- ) -> Optional[List["StoreFunction"]]:
23
+ ) -> Optional[list["StoreFunction"]]:
24
24
  if response.status_code == 200:
25
25
  response_200 = []
26
26
  _response_200 = response.json()
@@ -38,7 +38,7 @@ def _parse_response(
38
38
 
39
39
  def _build_response(
40
40
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
41
- ) -> Response[List["StoreFunction"]]:
41
+ ) -> Response[list["StoreFunction"]]:
42
42
  return Response(
43
43
  status_code=HTTPStatus(response.status_code),
44
44
  content=response.content,
@@ -50,7 +50,7 @@ def _build_response(
50
50
  def sync_detailed(
51
51
  *,
52
52
  client: Union[AuthenticatedClient, Client],
53
- ) -> Response[List["StoreFunction"]]:
53
+ ) -> Response[list["StoreFunction"]]:
54
54
  """List all store agent functions
55
55
 
56
56
  Raises:
@@ -58,7 +58,7 @@ def sync_detailed(
58
58
  httpx.TimeoutException: If the request takes longer than Client.timeout.
59
59
 
60
60
  Returns:
61
- Response[List['StoreFunction']]
61
+ Response[list['StoreFunction']]
62
62
  """
63
63
 
64
64
  kwargs = _get_kwargs()
@@ -73,7 +73,7 @@ def sync_detailed(
73
73
  def sync(
74
74
  *,
75
75
  client: Union[AuthenticatedClient, Client],
76
- ) -> Optional[List["StoreFunction"]]:
76
+ ) -> Optional[list["StoreFunction"]]:
77
77
  """List all store agent functions
78
78
 
79
79
  Raises:
@@ -81,7 +81,7 @@ def sync(
81
81
  httpx.TimeoutException: If the request takes longer than Client.timeout.
82
82
 
83
83
  Returns:
84
- List['StoreFunction']
84
+ list['StoreFunction']
85
85
  """
86
86
 
87
87
  return sync_detailed(
@@ -92,7 +92,7 @@ def sync(
92
92
  async def asyncio_detailed(
93
93
  *,
94
94
  client: Union[AuthenticatedClient, Client],
95
- ) -> Response[List["StoreFunction"]]:
95
+ ) -> Response[list["StoreFunction"]]:
96
96
  """List all store agent functions
97
97
 
98
98
  Raises:
@@ -100,7 +100,7 @@ async def asyncio_detailed(
100
100
  httpx.TimeoutException: If the request takes longer than Client.timeout.
101
101
 
102
102
  Returns:
103
- Response[List['StoreFunction']]
103
+ Response[list['StoreFunction']]
104
104
  """
105
105
 
106
106
  kwargs = _get_kwargs()
@@ -113,7 +113,7 @@ async def asyncio_detailed(
113
113
  async def asyncio(
114
114
  *,
115
115
  client: Union[AuthenticatedClient, Client],
116
- ) -> Optional[List["StoreFunction"]]:
116
+ ) -> Optional[list["StoreFunction"]]:
117
117
  """List all store agent functions
118
118
 
119
119
  Raises:
@@ -121,7 +121,7 @@ async def asyncio(
121
121
  httpx.TimeoutException: If the request takes longer than Client.timeout.
122
122
 
123
123
  Returns:
124
- List['StoreFunction']
124
+ list['StoreFunction']
125
125
  """
126
126
 
127
127
  return (
beamlit/api/workspaces/list_workspace_users.py CHANGED
@@ -1,5 +1,5 @@
1
1
  from http import HTTPStatus
2
- from typing import Any, List, Optional, Union
2
+ from typing import Any, Optional, Union
3
3
 
4
4
  import httpx
5
5
 
@@ -20,7 +20,7 @@ def _get_kwargs() -> dict[str, Any]:
20
20
 
21
21
  def _parse_response(
22
22
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
23
- ) -> Optional[List["WorkspaceUser"]]:
23
+ ) -> Optional[list["WorkspaceUser"]]:
24
24
  if response.status_code == 200:
25
25
  response_200 = []
26
26
  _response_200 = response.json()
@@ -38,7 +38,7 @@ def _parse_response(
38
38
 
39
39
  def _build_response(
40
40
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
41
- ) -> Response[List["WorkspaceUser"]]:
41
+ ) -> Response[list["WorkspaceUser"]]:
42
42
  return Response(
43
43
  status_code=HTTPStatus(response.status_code),
44
44
  content=response.content,
@@ -50,7 +50,7 @@ def _build_response(
50
50
  def sync_detailed(
51
51
  *,
52
52
  client: AuthenticatedClient,
53
- ) -> Response[List["WorkspaceUser"]]:
53
+ ) -> Response[list["WorkspaceUser"]]:
54
54
  """List users in workspace
55
55
 
56
56
  Returns a list of all users in the workspace.
@@ -60,7 +60,7 @@ def sync_detailed(
60
60
  httpx.TimeoutException: If the request takes longer than Client.timeout.
61
61
 
62
62
  Returns:
63
- Response[List['WorkspaceUser']]
63
+ Response[list['WorkspaceUser']]
64
64
  """
65
65
 
66
66
  kwargs = _get_kwargs()
@@ -75,7 +75,7 @@ def sync_detailed(
75
75
  def sync(
76
76
  *,
77
77
  client: AuthenticatedClient,
78
- ) -> Optional[List["WorkspaceUser"]]:
78
+ ) -> Optional[list["WorkspaceUser"]]:
79
79
  """List users in workspace
80
80
 
81
81
  Returns a list of all users in the workspace.
@@ -85,7 +85,7 @@ def sync(
85
85
  httpx.TimeoutException: If the request takes longer than Client.timeout.
86
86
 
87
87
  Returns:
88
- List['WorkspaceUser']
88
+ list['WorkspaceUser']
89
89
  """
90
90
 
91
91
  return sync_detailed(
@@ -96,7 +96,7 @@ def sync(
96
96
  async def asyncio_detailed(
97
97
  *,
98
98
  client: AuthenticatedClient,
99
- ) -> Response[List["WorkspaceUser"]]:
99
+ ) -> Response[list["WorkspaceUser"]]:
100
100
  """List users in workspace
101
101
 
102
102
  Returns a list of all users in the workspace.
@@ -106,7 +106,7 @@ async def asyncio_detailed(
106
106
  httpx.TimeoutException: If the request takes longer than Client.timeout.
107
107
 
108
108
  Returns:
109
- Response[List['WorkspaceUser']]
109
+ Response[list['WorkspaceUser']]
110
110
  """
111
111
 
112
112
  kwargs = _get_kwargs()
@@ -119,7 +119,7 @@ async def asyncio_detailed(
119
119
  async def asyncio(
120
120
  *,
121
121
  client: AuthenticatedClient,
122
- ) -> Optional[List["WorkspaceUser"]]:
122
+ ) -> Optional[list["WorkspaceUser"]]:
123
123
  """List users in workspace
124
124
 
125
125
  Returns a list of all users in the workspace.
@@ -129,7 +129,7 @@ async def asyncio(
129
129
  httpx.TimeoutException: If the request takes longer than Client.timeout.
130
130
 
131
131
  Returns:
132
- List['WorkspaceUser']
132
+ list['WorkspaceUser']
133
133
  """
134
134
 
135
135
  return (
beamlit/api/workspaces/list_workspaces.py CHANGED
@@ -1,5 +1,5 @@
1
1
  from http import HTTPStatus
2
- from typing import Any, List, Optional, Union
2
+ from typing import Any, Optional, Union
3
3
 
4
4
  import httpx
5
5
 
@@ -20,7 +20,7 @@ def _get_kwargs() -> dict[str, Any]:
20
20
 
21
21
  def _parse_response(
22
22
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
23
- ) -> Optional[List["Workspace"]]:
23
+ ) -> Optional[list["Workspace"]]:
24
24
  if response.status_code == 200:
25
25
  response_200 = []
26
26
  _response_200 = response.json()
@@ -38,7 +38,7 @@ def _parse_response(
38
38
 
39
39
  def _build_response(
40
40
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
41
- ) -> Response[List["Workspace"]]:
41
+ ) -> Response[list["Workspace"]]:
42
42
  return Response(
43
43
  status_code=HTTPStatus(response.status_code),
44
44
  content=response.content,
@@ -50,7 +50,7 @@ def _build_response(
50
50
  def sync_detailed(
51
51
  *,
52
52
  client: AuthenticatedClient,
53
- ) -> Response[List["Workspace"]]:
53
+ ) -> Response[list["Workspace"]]:
54
54
  """List workspaces
55
55
 
56
56
  Returns a list of all workspaces.
@@ -60,7 +60,7 @@ def sync_detailed(
60
60
  httpx.TimeoutException: If the request takes longer than Client.timeout.
61
61
 
62
62
  Returns:
63
- Response[List['Workspace']]
63
+ Response[list['Workspace']]
64
64
  """
65
65
 
66
66
  kwargs = _get_kwargs()
@@ -75,7 +75,7 @@ def sync_detailed(
75
75
  def sync(
76
76
  *,
77
77
  client: AuthenticatedClient,
78
- ) -> Optional[List["Workspace"]]:
78
+ ) -> Optional[list["Workspace"]]:
79
79
  """List workspaces
80
80
 
81
81
  Returns a list of all workspaces.
@@ -85,7 +85,7 @@ def sync(
85
85
  httpx.TimeoutException: If the request takes longer than Client.timeout.
86
86
 
87
87
  Returns:
88
- List['Workspace']
88
+ list['Workspace']
89
89
  """
90
90
 
91
91
  return sync_detailed(
@@ -96,7 +96,7 @@ def sync(
96
96
  async def asyncio_detailed(
97
97
  *,
98
98
  client: AuthenticatedClient,
99
- ) -> Response[List["Workspace"]]:
99
+ ) -> Response[list["Workspace"]]:
100
100
  """List workspaces
101
101
 
102
102
  Returns a list of all workspaces.
@@ -106,7 +106,7 @@ async def asyncio_detailed(
106
106
  httpx.TimeoutException: If the request takes longer than Client.timeout.
107
107
 
108
108
  Returns:
109
- Response[List['Workspace']]
109
+ Response[list['Workspace']]
110
110
  """
111
111
 
112
112
  kwargs = _get_kwargs()
@@ -119,7 +119,7 @@ async def asyncio_detailed(
119
119
  async def asyncio(
120
120
  *,
121
121
  client: AuthenticatedClient,
122
- ) -> Optional[List["Workspace"]]:
122
+ ) -> Optional[list["Workspace"]]:
123
123
  """List workspaces
124
124
 
125
125
  Returns a list of all workspaces.
@@ -129,7 +129,7 @@ async def asyncio(
129
129
  httpx.TimeoutException: If the request takes longer than Client.timeout.
130
130
 
131
131
  Returns:
132
- List['Workspace']
132
+ list['Workspace']
133
133
  """
134
134
 
135
135
  return (
beamlit/authentication/__init__.py CHANGED
@@ -1,12 +1,27 @@
1
1
  from .apikey import ApiKeyProvider
2
- from .authentication import (PublicProvider, RunClientWithCredentials,
3
- get_authentication_headers,
4
- new_client_from_settings,
5
- new_client_with_credentials)
6
- from .credentials import (Config, ContextConfig, Credentials, WorkspaceConfig,
7
- load_credentials, load_credentials_from_settings)
8
- from .device_mode import (BearerToken, DeviceLogin, DeviceLoginFinalizeRequest,
9
- DeviceLoginFinalizeResponse, DeviceLoginResponse)
2
+ from .authentication import (
3
+ PublicProvider,
4
+ RunClientWithCredentials,
5
+ get_authentication_headers,
6
+ new_client,
7
+ new_client_from_settings,
8
+ new_client_with_credentials,
9
+ )
10
+ from .credentials import (
11
+ Config,
12
+ ContextConfig,
13
+ Credentials,
14
+ WorkspaceConfig,
15
+ load_credentials,
16
+ load_credentials_from_settings,
17
+ )
18
+ from .device_mode import (
19
+ BearerToken,
20
+ DeviceLogin,
21
+ DeviceLoginFinalizeRequest,
22
+ DeviceLoginFinalizeResponse,
23
+ DeviceLoginResponse,
24
+ )
10
25
 
11
26
  __all__ = (
12
27
  "ApiKeyProvider",
@@ -14,6 +29,7 @@ __all__ = (
14
29
  "RunClientWithCredentials",
15
30
  "new_client_with_credentials",
16
31
  "new_client_from_settings",
32
+ "new_client",
17
33
  "get_authentication_headers",
18
34
  "Config",
19
35
  "ContextConfig",
@@ -25,5 +41,5 @@ __all__ = (
25
41
  "DeviceLogin",
26
42
  "DeviceLoginFinalizeRequest",
27
43
  "DeviceLoginFinalizeResponse",
28
- "DeviceLoginResponse"
29
- )
44
+ "DeviceLoginResponse",
45
+ )
beamlit/authentication/apikey.py CHANGED
@@ -10,11 +10,11 @@ class ApiKeyProvider(Auth):
10
10
 
11
11
  def get_headers(self):
12
12
  return {
13
- 'X-Beamlit-Api-Key': self.credentials.api_key,
14
- 'X-Beamlit-Workspace': self.workspace_name
13
+ "X-Beamlit-Api-Key": self.credentials.api_key,
14
+ "X-Beamlit-Workspace": self.workspace_name,
15
15
  }
16
16
 
17
17
  def auth_flow(self, request: Request) -> Generator[Request, Response, None]:
18
- request.headers['X-Beamlit-Api-Key'] = self.credentials.api_key
19
- request.headers['X-Beamlit-Workspace'] = self.workspace_name
18
+ request.headers["X-Beamlit-Api-Key"] = self.credentials.api_key
19
+ request.headers["X-Beamlit-Workspace"] = self.workspace_name
20
20
  yield request