beamlit 0.0.54rc101__py3-none-any.whl → 0.0.55__py3-none-any.whl

beamlit/models/model_spec.py

@@ -32,6 +32,7 @@ class ModelSpec:
         runtime (Union[Unset, Runtime]): Set of configurations for a deployment
         sandbox (Union[Unset, bool]): Sandbox mode
         serverless_config (Union[Unset, ServerlessConfig]): Configuration for a serverless deployment
+        knowledgebase (Union[Unset, str]): Knowledgebase Name
     """
 
     configurations: Union[Unset, "CoreSpecConfigurations"] = UNSET
@@ -44,6 +45,7 @@ class ModelSpec:
     runtime: Union[Unset, "Runtime"] = UNSET
     sandbox: Union[Unset, bool] = UNSET
     serverless_config: Union[Unset, "ServerlessConfig"] = UNSET
+    knowledgebase: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -108,6 +110,8 @@ class ModelSpec:
         elif self.serverless_config and isinstance(self.serverless_config, dict):
             serverless_config = self.serverless_config
 
+        knowledgebase = self.knowledgebase
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
@@ -131,6 +135,8 @@ class ModelSpec:
             field_dict["sandbox"] = sandbox
         if serverless_config is not UNSET:
             field_dict["serverlessConfig"] = serverless_config
+        if knowledgebase is not UNSET:
+            field_dict["knowledgebase"] = knowledgebase
 
         return field_dict
 
@@ -196,6 +202,8 @@ class ModelSpec:
         else:
             serverless_config = ServerlessConfig.from_dict(_serverless_config)
 
+        knowledgebase = d.pop("knowledgebase", UNSET)
+
         model_spec = cls(
             configurations=configurations,
             enabled=enabled,
@@ -207,6 +215,7 @@ class ModelSpec:
             runtime=runtime,
             sandbox=sandbox,
             serverless_config=serverless_config,
+            knowledgebase=knowledgebase,
         )
 
         model_spec.additional_properties = d

beamlit/models/runtime.py

@@ -20,24 +20,29 @@ class Runtime:
     Attributes:
         args (Union[Unset, list[Any]]): The arguments to pass to the deployment runtime
         command (Union[Unset, list[Any]]): The command to run the deployment
+        endpoint_name (Union[Unset, str]): Endpoint Name of the model. In case of hf_private_endpoint, it is the
+            endpoint name. In case of hf_public_endpoint, it is not used.
         envs (Union[Unset, list[Any]]): The environment variables to set in the deployment. Should be a list of
             Kubernetes EnvVar types
         image (Union[Unset, str]): The Docker image for the deployment
         metric_port (Union[Unset, int]): The port to serve the metrics on
-        model (Union[Unset, str]): The slug name of the origin model. Only used if the deployment is a Deployment
+        model (Union[Unset, str]): The slug name of the origin model at HuggingFace.
+        organization (Union[Unset, str]): The organization of the model
         readiness_probe (Union[Unset, RuntimeReadinessProbe]): The readiness probe. Should be a Kubernetes Probe type
         resources (Union[Unset, RuntimeResources]): The resources for the deployment. Should be a Kubernetes
             ResourceRequirements type
         serving_port (Union[Unset, int]): The port to serve the model on
-        type_ (Union[Unset, str]): The type of origin for the deployment
+        type_ (Union[Unset, str]): The type of origin for the deployment (hf_private_endpoint, hf_public_endpoint)
     """
 
     args: Union[Unset, list[Any]] = UNSET
     command: Union[Unset, list[Any]] = UNSET
+    endpoint_name: Union[Unset, str] = UNSET
     envs: Union[Unset, list[Any]] = UNSET
     image: Union[Unset, str] = UNSET
     metric_port: Union[Unset, int] = UNSET
     model: Union[Unset, str] = UNSET
+    organization: Union[Unset, str] = UNSET
     readiness_probe: Union[Unset, "RuntimeReadinessProbe"] = UNSET
     resources: Union[Unset, "RuntimeResources"] = UNSET
     serving_port: Union[Unset, int] = UNSET
@@ -53,6 +58,8 @@ class Runtime:
         if not isinstance(self.command, Unset):
             command = self.command
 
+        endpoint_name = self.endpoint_name
+
         envs: Union[Unset, list[Any]] = UNSET
         if not isinstance(self.envs, Unset):
             envs = self.envs
@@ -63,6 +70,8 @@ class Runtime:
 
         model = self.model
 
+        organization = self.organization
+
         readiness_probe: Union[Unset, dict[str, Any]] = UNSET
         if (
             self.readiness_probe
@@ -90,6 +99,8 @@ class Runtime:
             field_dict["args"] = args
         if command is not UNSET:
             field_dict["command"] = command
+        if endpoint_name is not UNSET:
+            field_dict["endpointName"] = endpoint_name
         if envs is not UNSET:
             field_dict["envs"] = envs
         if image is not UNSET:
@@ -98,6 +109,8 @@ class Runtime:
             field_dict["metricPort"] = metric_port
         if model is not UNSET:
             field_dict["model"] = model
+        if organization is not UNSET:
+            field_dict["organization"] = organization
         if readiness_probe is not UNSET:
             field_dict["readinessProbe"] = readiness_probe
         if resources is not UNSET:
@@ -121,6 +134,8 @@ class Runtime:
 
         command = cast(list[Any], d.pop("command", UNSET))
 
+        endpoint_name = d.pop("endpointName", UNSET)
+
         envs = cast(list[Any], d.pop("envs", UNSET))
 
         image = d.pop("image", UNSET)
@@ -129,6 +144,8 @@ class Runtime:
 
         model = d.pop("model", UNSET)
 
+        organization = d.pop("organization", UNSET)
+
         _readiness_probe = d.pop("readinessProbe", UNSET)
         readiness_probe: Union[Unset, RuntimeReadinessProbe]
         if isinstance(_readiness_probe, Unset):
@@ -150,10 +167,12 @@ class Runtime:
         runtime = cls(
             args=args,
             command=command,
+            endpoint_name=endpoint_name,
             envs=envs,
             image=image,
             metric_port=metric_port,
             model=model,
+            organization=organization,
             readiness_probe=readiness_probe,
             resources=resources,
             serving_port=serving_port,

beamlit/models/store_agent.py

@@ -28,6 +28,7 @@ class StoreAgent:
         image (Union[Unset, str]): Store agent image
         labels (Union[Unset, StoreAgentLabels]): Store agent labels
         name (Union[Unset, str]): Store agent name
+        prompt (Union[Unset, str]): Store agent prompt, this is to define what the agent does
     """
 
     created_at: Union[Unset, str] = UNSET
@@ -40,6 +41,7 @@ class StoreAgent:
     image: Union[Unset, str] = UNSET
     labels: Union[Unset, "StoreAgentLabels"] = UNSET
     name: Union[Unset, str] = UNSET
+    prompt: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -72,6 +74,8 @@ class StoreAgent:
 
         name = self.name
 
+        prompt = self.prompt
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
@@ -95,6 +99,8 @@ class StoreAgent:
             field_dict["labels"] = labels
         if name is not UNSET:
             field_dict["name"] = name
+        if prompt is not UNSET:
+            field_dict["prompt"] = prompt
 
         return field_dict
 
@@ -136,6 +142,8 @@ class StoreAgent:
 
         name = d.pop("name", UNSET)
 
+        prompt = d.pop("prompt", UNSET)
+
         store_agent = cls(
             created_at=created_at,
             updated_at=updated_at,
@@ -147,6 +155,7 @@ class StoreAgent:
             image=image,
             labels=labels,
             name=name,
+            prompt=prompt,
         )
 
         store_agent.additional_properties = d

{beamlit-0.0.54rc101.dist-info → beamlit-0.0.55.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.54rc101
+Version: 0.0.55
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 License-File: LICENSE

{beamlit-0.0.54rc101.dist-info → beamlit-0.0.55.dist-info}/RECORD

@@ -5,18 +5,13 @@ beamlit/py.typed,sha256=8ZJUsxZiuOy1oJeVhsTWQhTG_6pTVHVXk5hJL79ebTk,25
 beamlit/run.py,sha256=ERZ0tA7TsocV_2LC81QbxPxCokNCSmrZv4kqh9sGEkc,3494
 beamlit/types.py,sha256=E1hhDh_zXfsSQ0NCt9-uw90_Mr5iIlsdfnfvxv5HarU,1005
 beamlit/agents/__init__.py,sha256=bWsFaXUbAps3IsL3Prti89m1s714vICXodbQi77h3vY,206
-beamlit/agents/chain.py,sha256=eAk1ylea1UKum2JkGH31TfuyddyEEY3sYZ0gt8VEhzg,4232
-beamlit/agents/chat.py,sha256=rbrb1gt6EXQyAgl51mi37eteEm7pZIltcqfTIFzRH9g,8602
-beamlit/agents/decorator.py,sha256=0uWiiNtpEN3tMv8LWkWgSGAGoZF0QyPVCR2MpxViPmg,8107
+beamlit/agents/chain.py,sha256=xtOTOFQR9Wml-ZwW06cKZB4dIPmkQxpBx1dPyaE3PKk,4333
+beamlit/agents/chat.py,sha256=ItuS9zXkALgXVa6IevNpCVKvsicwg9OcUf4_VqO1x5Q,8603
+beamlit/agents/decorator.py,sha256=KdOKKf_svJc7Pa-eyP-_sXqEndZrEIQCV5Yfgz-zoQE,8253
 beamlit/agents/thread.py,sha256=XNqj2WI3W2roQB7J5mpF1wEuGb3VtfvKQvWaEPx-sx8,1014
 beamlit/agents/voice/openai.py,sha256=-RDBwl16i4TbUhFo5-77Ci3zmI3Y8U2yf2MmvXR2haQ,9479
 beamlit/agents/voice/utils.py,sha256=tQidyM40Ewuy12wKqpvJLvfJgneQ0sZf50dqnerPGHg,836
 beamlit/api/__init__.py,sha256=zTSiG_ujSjAqWPyc435YXaX9XTlpMjiJWBbV-f-YtdA,45
-beamlit/api/accounts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-beamlit/api/accounts/delete_account.py,sha256=Mrbhuxl1IF1GDclSixIAvj3YHa7trlFg3JNmllqpBRA,3864
-beamlit/api/accounts/get_account.py,sha256=_H0IQUlUFuA92WrYDpCmXjsJIzuVqa3oTwtNqMvM7Gg,3640
-beamlit/api/accounts/list_accounts.py,sha256=rg55vATS7416ECC3ZrSJxXOif2sMwoqQXLt9FQuxv3o,3505
-beamlit/api/accounts/update_account.py,sha256=1iO87kO87zjy4qlEyHTPmBMGU6vPIj_TVy4Z7UrnHb4,4103
 beamlit/api/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 beamlit/api/agents/create_agent.py,sha256=t5Pr62My2EhQlcIY71MrI73-0_q5Djr3a_Ybt9MIiQQ,3587
 beamlit/api/agents/create_agent_release.py,sha256=cH_GLiSFyMblwN0tayVPugwyGNMVReoahYRF85moGeY,3685
@@ -34,7 +29,6 @@ beamlit/api/agents/update_agent.py,sha256=No9iJkjUFJBCHDth1_vpuDfl2Ev3us1OWHXrrR
 beamlit/api/configurations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 beamlit/api/configurations/get_configuration.py,sha256=BtazLwvVe1OGjt0dJ-NVVipqAqZt1HwemWTGK2iuN9w,3197
 beamlit/api/default/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-beamlit/api/default/create_account.py,sha256=NJgrP9BPHVFpUELj2r-BkrQDpsYs-xPN1zXi9_YzZ9Q,3721
 beamlit/api/default/get_trace.py,sha256=2x-qpeu8jpfbq_c4-aAHgliwBQfagLEWEgM77eB_ZiY,3621
 beamlit/api/default/get_trace_ids.py,sha256=qxT085F7pt4GI0Ir7fpFA19IIR9_NFy47fYL9bPAtJo,6712
 beamlit/api/default/get_trace_logs.py,sha256=R69_j9GFAwkujsNMp4VTxoZuzw8jFni56l0bRxACT4w,4770
@@ -63,14 +57,19 @@ beamlit/api/integrations/create_integration_connection.py,sha256=VoaP3qLQpHCyB-K
 beamlit/api/integrations/delete_integration_connection.py,sha256=DjyhnqA6hn8mTZ9FXI6VcAUaeLvfXMDxDp2iHLJreqc,4188
 beamlit/api/integrations/get_integration.py,sha256=SOuuztVMIES0TCg4LpVXCA7T9N0fPcks5xLx8bUQ7AI,2473
 beamlit/api/integrations/get_integration_connection.py,sha256=uKWMH0BuXdciDm0RCzWDmTecQOlNOY2yec78yovYKI4,4173
-beamlit/api/integrations/get_integration_connection_model.py,sha256=-aqokNnQ8khEDAYDSMzUYyVcfIx2CkSGIfX1YRA3Aw0,2686
-beamlit/api/integrations/get_integration_model.py,sha256=MZhqFQHRafNZXMYHnpSzFlX6F9a4z0BlTM6EJoHkGIs,2636
+beamlit/api/integrations/get_integration_connection_model.py,sha256=U3rSVTdCKxA7jfpAGv1HD1Yk3OkYE36fllPbb9LRbUU,2708
+beamlit/api/integrations/get_integration_connection_model_endpoint_configurations.py,sha256=cqK0tKRwacslh1GKvEohYElmjA9kaFvZGBVyevS71ZE,2588
 beamlit/api/integrations/list_integration_connection_models.py,sha256=xEND9JaVrqTIXQ5vlBUihfASFkj-P28oTKN-DYYpPeQ,2530
 beamlit/api/integrations/list_integration_connections.py,sha256=Ol96kXuANQA8DMnBQ_zFL1SLDD5qkgZGhFuU8Rvs9Hw,3900
-beamlit/api/integrations/list_integration_models.py,sha256=3oxo5R0w_C3Fy7QLI2hNX7UsNcXJnqAORyqrW0_eGlw,2484
 beamlit/api/integrations/update_integration_connection.py,sha256=aCVzuQjUeebipm9p79PVVBEkHtzubHFhN1gviCILXic,4894
 beamlit/api/invitations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 beamlit/api/invitations/list_all_pending_invitations.py,sha256=YZxQdPXGNJD7OGXngV1y6UmT6APl5RqFt12ARKO6H2Q,4111
+beamlit/api/knowledgebases/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+beamlit/api/knowledgebases/create_knowledgebase.py,sha256=WS3FEF0Y1bLVrVR7QOGa8uTMn3KyQta_xx-Tf3M-cs4,3923
+beamlit/api/knowledgebases/delete_knowledgebase.py,sha256=lMuMU5MBz7iALAVO3g6abzblL46sevDcshs7kXC6L8k,4572
+beamlit/api/knowledgebases/get_knowledgebase.py,sha256=S3jGssOkaWH4L_6Zp3LRvBMk3A9zM5ff4hLxMfiBgYs,4565
+beamlit/api/knowledgebases/list_knowledgebases.py,sha256=6jTx1D_Y5tY8g4cmq6Ypmov6HTdkkGqPDovYXd7S0ME,4485
+beamlit/api/knowledgebases/update_knowledgebase.py,sha256=e-JQiDCCAih2L6V3fk0lzP2LnfiuXnaRIcw8bLyzSvk,4389
 beamlit/api/locations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 beamlit/api/locations/list_locations.py,sha256=6hu06Yt8ky_xPk9BRbN3Y749J0qDfVnVSbX5IwRWF8I,3720
 beamlit/api/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -141,28 +140,23 @@ beamlit/common/settings.py,sha256=Zf43TPYGVIeqi2WiKXvIfbHVuR6V9cbTeMqjz4WN--Q,53
 beamlit/common/slugify.py,sha256=QPJqa1VrSjRle931RN37t24rUjAbyZEg1UFAp2fLgac,679
 beamlit/common/utils.py,sha256=eG201z9gMRnhoHkaZGNtfFUbCzfg_Y59JR4ciMgidW8,1465
 beamlit/deploy/__init__.py,sha256=uRsI_-gTbbki59LlvubeTfG6wfI3o2XqZODW0QXA-Ao,292
-beamlit/deploy/deploy.py,sha256=44WvXtv71Sins6bTK-eK-GWLJfh7Gy4_IQlnrzDRq10,11118
-beamlit/deploy/format.py,sha256=GpXpmM5Fzmy5E9yo2pu57NopXCXRZE2w7c8atWqQKrA,2879
+beamlit/deploy/deploy.py,sha256=V07m0GKEKvQKjyw5r5UR2IsXQxhgyH6OeSgexj3AAtY,11232
+beamlit/deploy/format.py,sha256=W3ESUHyFv-iZDjVnHOf9YFDDXZSXYIFFbwCoL1GInE0,1162
 beamlit/deploy/parser.py,sha256=gjRUhOVtfKnc1UNc_FhXsEfj9zrMNuq8W93pNsJBpo0,7586
 beamlit/functions/__init__.py,sha256=Mnoqpa1dm7TXwjodBbF_40JyD78aXsOYWmqjDSnA1lU,317
 beamlit/functions/common.py,sha256=IkdxQYrvu6H4cPgqc0OeXquCInMtGvmq6GP2FA1rh7w,9221
 beamlit/functions/decorator.py,sha256=iQbLwUo0K83DFJ3ub8O5jKtkbSINnku6GZcKJ9h7-5E,2292
 beamlit/functions/mcp/mcp.py,sha256=M6SAGnBo5xpZcPKB6YE0R2Xp73hHOlbZXc0Q2kFSYuU,5458
 beamlit/functions/remote/remote.py,sha256=g0M909Ow9GLPgtistrWkUuhcITUwA2iHIOQNpUsp28g,5989
-beamlit/models/__init__.py,sha256=ZAcQKwcTj25TEBiCdtU6O42PGEAcZGAbEQx1ixp71Ck,9033
-beamlit/models/account.py,sha256=A8A1nkT-CfkeDIMhQctAnSvVagM9by3S3LfP7QqTAag,7184
-beamlit/models/account_metadata.py,sha256=SnJxJCJJSH5DGGO3uKsPUWxbi0TXicq0MucpaYuaG_Y,3785
-beamlit/models/account_spec.py,sha256=2j2r6HMKLZBH9-H2m6lRmADrcLExin62JysEP8LiTwk,3672
-beamlit/models/account_spec_address.py,sha256=B7ZiP2mWJxZyFfAcScYV6_YYWLGIi9U5dDinNhdViLw,1263
+beamlit/models/__init__.py,sha256=2yOQDeO7hQ_85q0WTU7ikAe-IfGr7PwlnHXJnLRSuB0,9036
 beamlit/models/acl.py,sha256=tH67gsl_BMaviSbTaaIkO1g9cWZgJ6VgAnYVjQSzGZY,3952
 beamlit/models/agent.py,sha256=pkFemfg0OUAuiqebiT3PurXhjAKvgCa_YOPuyqFVE2o,4264
-beamlit/models/agent_chain.py,sha256=8PN8wVSayS-LoBN2nahZsOmr6r3t62H_LPDK_8fnkM8,2255
+beamlit/models/agent_chain.py,sha256=Vz6galRoT20JRM3mAfaJ_HbOX2pS1pL_fyjYM35b0NI,2566
 beamlit/models/agent_history.py,sha256=76ZHoaz7oq3-2ikNQj25NmzyXOzx5JIPxOlfUyljRQg,5124
 beamlit/models/agent_history_event.py,sha256=Ed_xaedwLEPNuM807Ldj5xqUBj4Eua9NqpCzpTRi0Og,3820
 beamlit/models/agent_release.py,sha256=AXtHX_Ze7cMh2tKnRw2usRWnLf2B9u_iFPTF6WYdJLM,1928
-beamlit/models/agent_spec.py,sha256=DDOkgjbKA8XHNIgZH0Sajm5xKYFlX1XdDVJczhUN17g,12557
+beamlit/models/agent_spec.py,sha256=B6TrOmGwQk8Mu-xNmjRqvaItNpT9mutejQL0M9Q6YLw,12879
 beamlit/models/api_key.py,sha256=oKuqDF0xe2Z2-6yJqulbzlXEQyM3W7lvQn6FXCktaaU,4278
-beamlit/models/billing_address.py,sha256=B4u5bZzBH7LLelRGTDjCWL9vJjK3FBcba14RctqYLx4,3844
 beamlit/models/configuration.py,sha256=KujTXl7iihrPcL0jX0hgxN0xAAwMgJsTaFKcMBpCVLs,2761
 beamlit/models/continent.py,sha256=_gQJci2_MUmCiGGG4iCMnUDTww9OqcTsNVqwPVigy3U,1880
 beamlit/models/core_event.py,sha256=L9GQh2rRPM1RC042IUJHEazha1gHqdNb9XN44zjawSs,2321
@@ -192,9 +186,13 @@ beamlit/models/integration_connection.py,sha256=w3FZOQTDFqPVMEiXcL5NeLVp_uvIFpa7
 beamlit/models/integration_connection_spec.py,sha256=p0YvnqdAXNDg4mR5hiT8WYr4kCJkMnOG8EokVw2ERMU,4047
 beamlit/models/integration_connection_spec_config.py,sha256=D-MuemiNwD6jhs-hsvGdazKuzB2akg1O7Mdro1g59pY,1360
 beamlit/models/integration_connection_spec_secret.py,sha256=UJCjioPQ6gHh5tbq1V2uKvWtQa0_YmmhhaiQoPDL274,1334
-beamlit/models/integration_model.py,sha256=gW6folMKJsIScQsV0IT5nQpO2sKzQb6d3E9fw4W4RIY,4469
+beamlit/models/integration_model.py,sha256=-2_UVeP_5xWqYfP42gU2tckirN43LKTqUyRapwke090,5086
 beamlit/models/integration_repository.py,sha256=Ac5c_UBdlUHvGflvCzqA5gXzX1_YEDHz3yBgMCGqqDg,2397
 beamlit/models/invite_workspace_user_body.py,sha256=ITmKPCuOGKTHn4VjnZ2EjIj6PGPoGh-M_rzgo0VMgNM,1612
+beamlit/models/knowledgebase.py,sha256=EBDrt1hZuObNHtZpLtYIFbgQjovYgpqiEBZID146Kfo,4392
+beamlit/models/knowledgebase_release.py,sha256=5LMllHTVxquiukS2_k33t_GxoFwgzfjIi0xhgaDILwM,1985
+beamlit/models/knowledgebase_spec.py,sha256=bWjB1x8If5-NZW27HEsuBhCFAWBCaKdTz0rsbI5Z_l0,5071
+beamlit/models/knowledgebase_spec_options.py,sha256=F0NfzFhP3bifGOBiyuvpVWh2iI7NABrcmFTg9A9a140,1316
 beamlit/models/last_n_requests_metric.py,sha256=BYaEhJAeoGE-NtdlT7EkUSkaNManeCsXeVryOcKH57s,2544
 beamlit/models/latency_metric.py,sha256=8-iz4JaxqI0Mi0a51A9ey_9yGS8gh2yoO_GJKnaAWJU,5527
 beamlit/models/location_response.py,sha256=Tvs5CdgyB7sPvogk4clT9jMJQa63Ivt3iIEfRP9UcKA,3388
@@ -208,7 +206,7 @@ beamlit/models/metrics_rps_per_code.py,sha256=CBhe8-hqpsK3lS4KepgqveAlnJRtjANoYn
 beamlit/models/model.py,sha256=wATVQPhQy8LLtRd2MlmWF0gt21ZoGadOwPhNPajvYiY,4366
 beamlit/models/model_private_cluster.py,sha256=EojXwmxgDLogsUlTDRw9PvhiQVk_b5nQUSBbWQosQpg,2281
 beamlit/models/model_release.py,sha256=ql8EDbKc9--kre7zp3o5_XoJjpAoXrsgJdYbjYST4vs,1928
-beamlit/models/model_spec.py,sha256=73ztq0F1zzNh8y_VYDwOQUjm-guYr3I12Wg37lDa7X0,9317
+beamlit/models/model_spec.py,sha256=5z2aZ4GCv3u76SH_w072aDbQ3vpQxOp7Bb6-0AiP_KY,9659
 beamlit/models/owner_fields.py,sha256=8LsT7inzzFI1hZQma798gYoOfJpdmwK9T8y_0JrKGSU,2022
 beamlit/models/pending_invitation.py,sha256=YJaUNHXoq-X1ZTBkOpmRQj31R99S6Y7Ik45dHDss6-M,3797
 beamlit/models/pending_invitation_accept.py,sha256=KbHS5L7SSPro45RK0_pORp6yhPKPao_izivB_FFYPjk,2550
@@ -236,12 +234,12 @@ beamlit/models/resource_environment_metrics.py,sha256=NOF-xh4CncVHKgDRalVhAmNrt_
 beamlit/models/resource_environment_metrics_request_total_per_code.py,sha256=HhGnCuajc2K4chKxNnB2_RfEtwdswG5GlEaHURUyTDI,1448
 beamlit/models/resource_environment_metrics_rps_per_code.py,sha256=mcAUfxNU759akRuLJIXvqzAJPIVEzspnzSYVm7Uo1vM,1411
 beamlit/models/resource_log.py,sha256=XV_DaMU4jf6pVk2XyWyWPtMeZ9uJJ9_LTDEYg3BtxHo,2253
-beamlit/models/runtime.py,sha256=bWnnLHlSow97NeD8KHStaCOsSRBIkCU91Fryk9JrIdw,6385
+beamlit/models/runtime.py,sha256=VA4fhHBgzYU3uu0J-yngA8SlrJlTsQ0Lj3Qofz_4vIM,7205
 beamlit/models/runtime_readiness_probe.py,sha256=iqNikZl5CffpNEwsUxVbs9k2_y2IUWpkEyZicf5T4Pk,1317
 beamlit/models/runtime_resources.py,sha256=6ioyGkvDPn93_MjJTJwB-J33R9vLJpagz4OvFuIExWE,1317
 beamlit/models/serverless_config.py,sha256=He_W4af-9QoBztQl3i3loSvn9pyHKhbb5fsQsuJSI-M,4989
 beamlit/models/spec_configuration.py,sha256=oGmMqoZSVEGedarAou-102vfw16bFevHUp4C_OW6h1k,1970
-beamlit/models/store_agent.py,sha256=V4h8DCVyaVjeDdp8AxvD2NMmuw25JftT40S_Dex-cco,5783
+beamlit/models/store_agent.py,sha256=8-sFVS8fMhD9cAM4_ZpsTti27Lyseom7QOxnO5RZnHI,6087
 beamlit/models/store_agent_labels.py,sha256=DDsYO0AXh8B4k1hgLBXHyHH1WwP2kKCyHwZqmd3qbOE,1256
 beamlit/models/store_configuration.py,sha256=8ppSJsmkC4yjPQXipt8hiju_bskYGEQ37FtmCLuYOiQ,4958
 beamlit/models/store_configuration_option.py,sha256=ErsfYkkG-b8MXf_xWS837FGdlbZWzTyDMkLtfXzmqQY,2269
@@ -265,8 +263,8 @@ beamlit/serve/app.py,sha256=lM59fdUtfkfAYNPWSCU9pkXIPBnhgVGvvgfoMkSVtks,4531
 beamlit/serve/middlewares/__init__.py,sha256=O7fyfE1DIYmajFY9WWdzxCgeAQWZzJfeUjzHGbpWaAk,309
 beamlit/serve/middlewares/accesslog.py,sha256=lcu33j4epFSHRBaeTpyt8deNb3kaM3K91-andw4fp80,1112
 beamlit/serve/middlewares/processtime.py,sha256=3x5w1yQexB0xFNKK6fgLbINxT-eLLunfZ6UDV0bIIF4,944
-beamlit-0.0.54rc101.dist-info/METADATA,sha256=JlyvBncCMQ34OQ6WXhOzECQpzh0ho2jpFbhJGgz1adE,3515
-beamlit-0.0.54rc101.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.54rc101.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
-beamlit-0.0.54rc101.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-beamlit-0.0.54rc101.dist-info/RECORD,,
+beamlit-0.0.55.dist-info/METADATA,sha256=hpUf_exI0FRoWST2SSvMilv1DjwhSjuJxU7HgdEPX0M,3510
+beamlit-0.0.55.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.55.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
+beamlit-0.0.55.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+beamlit-0.0.55.dist-info/RECORD,,

beamlit/api/integrations/get_integration_model.py

@@ -1,104 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Optional, Union
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...types import Response
-
-
-def _get_kwargs(
-    integration_name: str,
-    model_id: str,
-) -> dict[str, Any]:
-    _kwargs: dict[str, Any] = {
-        "method": "get",
-        "url": f"/integrations/{integration_name}/models/{model_id}",
-    }
-
-    return _kwargs
-
-
-def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Any]:
-    if response.status_code == 200:
-        return None
-    if client.raise_on_unexpected_status:
-        raise errors.UnexpectedStatus(response.status_code, response.content)
-    else:
-        return None
-
-
-def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Any]:
-    return Response(
-        status_code=HTTPStatus(response.status_code),
-        content=response.content,
-        headers=response.headers,
-        parsed=_parse_response(client=client, response=response),
-    )
-
-
-def sync_detailed(
-    integration_name: str,
-    model_id: str,
-    *,
-    client: AuthenticatedClient,
-) -> Response[Any]:
-    """Get integration model
-
-     Returns a model for an integration by ID.
-
-    Args:
-        integration_name (str):
-        model_id (str):
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        Response[Any]
-    """
-
-    kwargs = _get_kwargs(
-        integration_name=integration_name,
-        model_id=model_id,
-    )
-
-    response = client.get_httpx_client().request(
-        **kwargs,
-    )
-
-    return _build_response(client=client, response=response)
-
-
-async def asyncio_detailed(
-    integration_name: str,
-    model_id: str,
-    *,
-    client: AuthenticatedClient,
-) -> Response[Any]:
-    """Get integration model
-
-     Returns a model for an integration by ID.
-
-    Args:
-        integration_name (str):
-        model_id (str):
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        Response[Any]
-    """
-
-    kwargs = _get_kwargs(
-        integration_name=integration_name,
-        model_id=model_id,
-    )
-
-    response = await client.get_async_httpx_client().request(**kwargs)
-
-    return _build_response(client=client, response=response)

beamlit/models/account.py

@@ -1,224 +0,0 @@
-from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-if TYPE_CHECKING:
-    from ..models.account_spec_address import AccountSpecAddress
-    from ..models.metadata_labels import MetadataLabels
-
-
-T = TypeVar("T", bound="Account")
-
-
-@_attrs_define
-class Account:
-    """Account
-
-    Attributes:
-        created_at (Union[Unset, str]): The date and time when the resource was created
-        updated_at (Union[Unset, str]): The date and time when the resource was updated
-        created_by (Union[Unset, str]): The user or service account who created the resource
-        updated_by (Union[Unset, str]): The user or service account who updated the resource
-        display_name (Union[Unset, str]): Model display name
-        labels (Union[Unset, MetadataLabels]): Labels
-        name (Union[Unset, str]): Model name
-        workspace (Union[Unset, str]): Workspace name
-        address (Union[Unset, AccountSpecAddress]): Billing address
-        admins (Union[Unset, list[Any]]): Admins
-        currency (Union[Unset, str]): Currency
-        owner (Union[Unset, str]): Owner
-        status (Union[Unset, str]): Status
-        tax_id (Union[Unset, str]): Tax ID
-        metadata (Union[Unset, Any]):
-        spec (Union[Unset, Any]):
-    """
-
-    created_at: Union[Unset, str] = UNSET
-    updated_at: Union[Unset, str] = UNSET
-    created_by: Union[Unset, str] = UNSET
-    updated_by: Union[Unset, str] = UNSET
-    display_name: Union[Unset, str] = UNSET
-    labels: Union[Unset, "MetadataLabels"] = UNSET
-    name: Union[Unset, str] = UNSET
-    workspace: Union[Unset, str] = UNSET
-    address: Union[Unset, "AccountSpecAddress"] = UNSET
-    admins: Union[Unset, list[Any]] = UNSET
-    currency: Union[Unset, str] = UNSET
-    owner: Union[Unset, str] = UNSET
-    status: Union[Unset, str] = UNSET
-    tax_id: Union[Unset, str] = UNSET
-    metadata: Union[Unset, Any] = UNSET
-    spec: Union[Unset, Any] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        created_at = self.created_at
-
-        updated_at = self.updated_at
-
-        created_by = self.created_by
-
-        updated_by = self.updated_by
-
-        display_name = self.display_name
-
-        labels: Union[Unset, dict[str, Any]] = UNSET
-        if self.labels and not isinstance(self.labels, Unset) and not isinstance(self.labels, dict):
-            labels = self.labels.to_dict()
-        elif self.labels and isinstance(self.labels, dict):
-            labels = self.labels
-
-        name = self.name
-
-        workspace = self.workspace
-
-        address: Union[Unset, dict[str, Any]] = UNSET
-        if self.address and not isinstance(self.address, Unset) and not isinstance(self.address, dict):
-            address = self.address.to_dict()
-        elif self.address and isinstance(self.address, dict):
-            address = self.address
-
-        admins: Union[Unset, list[Any]] = UNSET
-        if not isinstance(self.admins, Unset):
-            admins = self.admins
-
-        currency = self.currency
-
-        owner = self.owner
-
-        status = self.status
-
-        tax_id = self.tax_id
-
-        metadata = self.metadata
-
-        spec = self.spec
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if created_at is not UNSET:
-            field_dict["createdAt"] = created_at
-        if updated_at is not UNSET:
-            field_dict["updatedAt"] = updated_at
-        if created_by is not UNSET:
-            field_dict["createdBy"] = created_by
-        if updated_by is not UNSET:
-            field_dict["updatedBy"] = updated_by
-        if display_name is not UNSET:
-            field_dict["displayName"] = display_name
-        if labels is not UNSET:
-            field_dict["labels"] = labels
-        if name is not UNSET:
-            field_dict["name"] = name
-        if workspace is not UNSET:
-            field_dict["workspace"] = workspace
-        if address is not UNSET:
-            field_dict["address"] = address
-        if admins is not UNSET:
-            field_dict["admins"] = admins
-        if currency is not UNSET:
-            field_dict["currency"] = currency
-        if owner is not UNSET:
-            field_dict["owner"] = owner
-        if status is not UNSET:
-            field_dict["status"] = status
-        if tax_id is not UNSET:
-            field_dict["tax_id"] = tax_id
-        if metadata is not UNSET:
-            field_dict["metadata"] = metadata
-        if spec is not UNSET:
-            field_dict["spec"] = spec
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.account_spec_address import AccountSpecAddress
-        from ..models.metadata_labels import MetadataLabels
-
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        created_at = d.pop("createdAt", UNSET)
-
-        updated_at = d.pop("updatedAt", UNSET)
-
-        created_by = d.pop("createdBy", UNSET)
-
-        updated_by = d.pop("updatedBy", UNSET)
-
-        display_name = d.pop("displayName", UNSET)
-
-        _labels = d.pop("labels", UNSET)
-        labels: Union[Unset, MetadataLabels]
-        if isinstance(_labels, Unset):
-            labels = UNSET
-        else:
-            labels = MetadataLabels.from_dict(_labels)
-
-        name = d.pop("name", UNSET)
-
-        workspace = d.pop("workspace", UNSET)
-
-        _address = d.pop("address", UNSET)
-        address: Union[Unset, AccountSpecAddress]
-        if isinstance(_address, Unset):
-            address = UNSET
-        else:
-            address = AccountSpecAddress.from_dict(_address)
-
-        admins = cast(list[Any], d.pop("admins", UNSET))
-
-        currency = d.pop("currency", UNSET)
-
-        owner = d.pop("owner", UNSET)
-
-        status = d.pop("status", UNSET)
-
-        tax_id = d.pop("tax_id", UNSET)
-
-        metadata = d.pop("metadata", UNSET)
-
-        spec = d.pop("spec", UNSET)
-
-        account = cls(
-            created_at=created_at,
-            updated_at=updated_at,
-            created_by=created_by,
-            updated_by=updated_by,
-            display_name=display_name,
-            labels=labels,
-            name=name,
-            workspace=workspace,
-            address=address,
-            admins=admins,
-            currency=currency,
-            owner=owner,
-            status=status,
-            tax_id=tax_id,
-            metadata=metadata,
-            spec=spec,
-        )
-
-        account.additional_properties = d
-        return account
-
-    @property
-    def additional_keys(self) -> list[str]:
-        return list(self.additional_properties.keys())
-
-    def __getitem__(self, key: str) -> Any:
-        return self.additional_properties[key]
-
-    def __setitem__(self, key: str, value: Any) -> None:
-        self.additional_properties[key] = value
-
-    def __delitem__(self, key: str) -> None:
-        del self.additional_properties[key]
-
-    def __contains__(self, key: str) -> bool:
-        return key in self.additional_properties