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

beamlit/api/{accounts/get_account.py → knowledgebases/get_knowledgebase.py}

@@ -5,24 +5,33 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.account import Account
-from ...types import Response
+from ...models.knowledgebase import Knowledgebase
+from ...types import UNSET, Response, Unset
 
 
 def _get_kwargs(
-    account_id: str,
+    knowledgebase_id: str,
+    *,
+    environment: Union[Unset, str] = UNSET,
 ) -> dict[str, Any]:
+    params: dict[str, Any] = {}
+
+    params["environment"] = environment
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": f"/accounts/{account_id}",
+        "url": f"/knowledgebases/{knowledgebase_id}",
+        "params": params,
     }
 
     return _kwargs
 
 
-def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Account]:
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Knowledgebase]:
     if response.status_code == 200:
-        response_200 = Account.from_dict(response.json())
+        response_200 = Knowledgebase.from_dict(response.json())
 
         return response_200
     if client.raise_on_unexpected_status:
@@ -31,7 +40,7 @@ def _parse_response(*, client: Union[AuthenticatedClient, Client], response: htt
         return None
 
 
-def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Account]:
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Knowledgebase]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -41,27 +50,30 @@ def _build_response(*, client: Union[AuthenticatedClient, Client], response: htt
 
 
 def sync_detailed(
-    account_id: str,
+    knowledgebase_id: str,
     *,
     client: AuthenticatedClient,
-) -> Response[Account]:
-    """Get account by name
+    environment: Union[Unset, str] = UNSET,
+) -> Response[Knowledgebase]:
+    """Get knowledgebase
 
-     Returns an account by name.
+     Returns an knowledgebase by ID.
 
     Args:
-        account_id (str):
+        knowledgebase_id (str):
+        environment (Union[Unset, 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[Account]
+        Response[Knowledgebase]
     """
 
     kwargs = _get_kwargs(
-        account_id=account_id,
+        knowledgebase_id=knowledgebase_id,
+        environment=environment,
     )
 
     response = client.get_httpx_client().request(
@@ -72,53 +84,59 @@ def sync_detailed(
 
 
 def sync(
-    account_id: str,
+    knowledgebase_id: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[Account]:
-    """Get account by name
+    environment: Union[Unset, str] = UNSET,
+) -> Optional[Knowledgebase]:
+    """Get knowledgebase
 
-     Returns an account by name.
+     Returns an knowledgebase by ID.
 
     Args:
-        account_id (str):
+        knowledgebase_id (str):
+        environment (Union[Unset, 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:
-        Account
+        Knowledgebase
     """
 
     return sync_detailed(
-        account_id=account_id,
+        knowledgebase_id=knowledgebase_id,
         client=client,
+        environment=environment,
     ).parsed
 
 
 async def asyncio_detailed(
-    account_id: str,
+    knowledgebase_id: str,
     *,
     client: AuthenticatedClient,
-) -> Response[Account]:
-    """Get account by name
+    environment: Union[Unset, str] = UNSET,
+) -> Response[Knowledgebase]:
+    """Get knowledgebase
 
-     Returns an account by name.
+     Returns an knowledgebase by ID.
 
     Args:
-        account_id (str):
+        knowledgebase_id (str):
+        environment (Union[Unset, 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[Account]
+        Response[Knowledgebase]
     """
 
     kwargs = _get_kwargs(
-        account_id=account_id,
+        knowledgebase_id=knowledgebase_id,
+        environment=environment,
     )
 
     response = await client.get_async_httpx_client().request(**kwargs)
@@ -127,28 +145,31 @@ async def asyncio_detailed(
 
 
 async def asyncio(
-    account_id: str,
+    knowledgebase_id: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[Account]:
-    """Get account by name
+    environment: Union[Unset, str] = UNSET,
+) -> Optional[Knowledgebase]:
+    """Get knowledgebase
 
-     Returns an account by name.
+     Returns an knowledgebase by ID.
 
     Args:
-        account_id (str):
+        knowledgebase_id (str):
+        environment (Union[Unset, 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:
-        Account
+        Knowledgebase
     """
 
     return (
         await asyncio_detailed(
-            account_id=account_id,
+            knowledgebase_id=knowledgebase_id,
             client=client,
+            environment=environment,
         )
     ).parsed

beamlit/api/{accounts/list_accounts.py → knowledgebases/list_knowledgebases.py}

@@ -5,14 +5,24 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.account import Account
-from ...types import Response
+from ...models.knowledgebase import Knowledgebase
+from ...types import UNSET, Response, Unset
 
 
-def _get_kwargs() -> dict[str, Any]:
+def _get_kwargs(
+    *,
+    environment: Union[Unset, str] = UNSET,
+) -> dict[str, Any]:
+    params: dict[str, Any] = {}
+
+    params["environment"] = environment
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": "/accounts",
+        "url": "/knowledgebases",
+        "params": params,
     }
 
     return _kwargs
@@ -20,12 +30,12 @@ def _get_kwargs() -> dict[str, Any]:
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[list["Account"]]:
+) -> Optional[list["Knowledgebase"]]:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
         for response_200_item_data in _response_200:
-            response_200_item = Account.from_dict(response_200_item_data)
+            response_200_item = Knowledgebase.from_dict(response_200_item_data)
 
             response_200.append(response_200_item)
 
@@ -38,7 +48,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[list["Account"]]:
+) -> Response[list["Knowledgebase"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -50,20 +60,26 @@ def _build_response(
 def sync_detailed(
     *,
     client: AuthenticatedClient,
-) -> Response[list["Account"]]:
-    """List accounts
+    environment: Union[Unset, str] = UNSET,
+) -> Response[list["Knowledgebase"]]:
+    """List knowledgebases
+
+     Returns a list of all knowledgebases in the workspace.
 
-     Returns a list of all accounts.
+    Args:
+        environment (Union[Unset, 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[list['Account']]
+        Response[list['Knowledgebase']]
     """
 
-    kwargs = _get_kwargs()
+    kwargs = _get_kwargs(
+        environment=environment,
+    )
 
     response = client.get_httpx_client().request(
         **kwargs,
@@ -75,41 +91,52 @@ def sync_detailed(
 def sync(
     *,
     client: AuthenticatedClient,
-) -> Optional[list["Account"]]:
-    """List accounts
+    environment: Union[Unset, str] = UNSET,
+) -> Optional[list["Knowledgebase"]]:
+    """List knowledgebases
+
+     Returns a list of all knowledgebases in the workspace.
 
-     Returns a list of all accounts.
+    Args:
+        environment (Union[Unset, 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:
-        list['Account']
+        list['Knowledgebase']
     """
 
     return sync_detailed(
         client=client,
+        environment=environment,
     ).parsed
 
 
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
-) -> Response[list["Account"]]:
-    """List accounts
+    environment: Union[Unset, str] = UNSET,
+) -> Response[list["Knowledgebase"]]:
+    """List knowledgebases
+
+     Returns a list of all knowledgebases in the workspace.
 
-     Returns a list of all accounts.
+    Args:
+        environment (Union[Unset, 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[list['Account']]
+        Response[list['Knowledgebase']]
     """
 
-    kwargs = _get_kwargs()
+    kwargs = _get_kwargs(
+        environment=environment,
+    )
 
     response = await client.get_async_httpx_client().request(**kwargs)
 
@@ -119,21 +146,26 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: AuthenticatedClient,
-) -> Optional[list["Account"]]:
-    """List accounts
+    environment: Union[Unset, str] = UNSET,
+) -> Optional[list["Knowledgebase"]]:
+    """List knowledgebases
+
+     Returns a list of all knowledgebases in the workspace.
 
-     Returns a list of all accounts.
+    Args:
+        environment (Union[Unset, 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:
-        list['Account']
+        list['Knowledgebase']
     """
 
     return (
         await asyncio_detailed(
             client=client,
+            environment=environment,
         )
     ).parsed

beamlit/api/{accounts/update_account.py → knowledgebases/update_knowledgebase.py}

@@ -5,20 +5,20 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.account import Account
+from ...models.knowledgebase import Knowledgebase
 from ...types import Response
 
 
 def _get_kwargs(
-    account_id: str,
+    knowledgebase_id: str,
     *,
-    body: Account,
+    body: Knowledgebase,
 ) -> dict[str, Any]:
     headers: dict[str, Any] = {}
 
     _kwargs: dict[str, Any] = {
         "method": "put",
-        "url": f"/accounts/{account_id}",
+        "url": f"/knowledgebases/{knowledgebase_id}",
     }
 
     _body = body.to_dict()
@@ -30,9 +30,9 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Account]:
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Knowledgebase]:
     if response.status_code == 200:
-        response_200 = Account.from_dict(response.json())
+        response_200 = Knowledgebase.from_dict(response.json())
 
         return response_200
     if client.raise_on_unexpected_status:
@@ -41,7 +41,7 @@ def _parse_response(*, client: Union[AuthenticatedClient, Client], response: htt
         return None
 
 
-def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Account]:
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Knowledgebase]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -51,29 +51,29 @@ def _build_response(*, client: Union[AuthenticatedClient, Client], response: htt
 
 
 def sync_detailed(
-    account_id: str,
+    knowledgebase_id: str,
     *,
     client: AuthenticatedClient,
-    body: Account,
-) -> Response[Account]:
-    """Update account
+    body: Knowledgebase,
+) -> Response[Knowledgebase]:
+    """Update knowledgebase
 
-     Updates an account by name.
+     Updates an knowledgebase.
 
     Args:
-        account_id (str):
-        body (Account): Account
+        knowledgebase_id (str):
+        body (Knowledgebase): Knowledgebase
 
     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[Account]
+        Response[Knowledgebase]
     """
 
     kwargs = _get_kwargs(
-        account_id=account_id,
+        knowledgebase_id=knowledgebase_id,
         body=body,
     )
 
@@ -85,58 +85,58 @@ def sync_detailed(
 
 
 def sync(
-    account_id: str,
+    knowledgebase_id: str,
     *,
     client: AuthenticatedClient,
-    body: Account,
-) -> Optional[Account]:
-    """Update account
+    body: Knowledgebase,
+) -> Optional[Knowledgebase]:
+    """Update knowledgebase
 
-     Updates an account by name.
+     Updates an knowledgebase.
 
     Args:
-        account_id (str):
-        body (Account): Account
+        knowledgebase_id (str):
+        body (Knowledgebase): Knowledgebase
 
     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:
-        Account
+        Knowledgebase
     """
 
     return sync_detailed(
-        account_id=account_id,
+        knowledgebase_id=knowledgebase_id,
         client=client,
         body=body,
     ).parsed
 
 
 async def asyncio_detailed(
-    account_id: str,
+    knowledgebase_id: str,
     *,
     client: AuthenticatedClient,
-    body: Account,
-) -> Response[Account]:
-    """Update account
+    body: Knowledgebase,
+) -> Response[Knowledgebase]:
+    """Update knowledgebase
 
-     Updates an account by name.
+     Updates an knowledgebase.
 
     Args:
-        account_id (str):
-        body (Account): Account
+        knowledgebase_id (str):
+        body (Knowledgebase): Knowledgebase
 
     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[Account]
+        Response[Knowledgebase]
     """
 
     kwargs = _get_kwargs(
-        account_id=account_id,
+        knowledgebase_id=knowledgebase_id,
         body=body,
     )
 
@@ -146,30 +146,30 @@ async def asyncio_detailed(
 
 
 async def asyncio(
-    account_id: str,
+    knowledgebase_id: str,
     *,
     client: AuthenticatedClient,
-    body: Account,
-) -> Optional[Account]:
-    """Update account
+    body: Knowledgebase,
+) -> Optional[Knowledgebase]:
+    """Update knowledgebase
 
-     Updates an account by name.
+     Updates an knowledgebase.
 
     Args:
-        account_id (str):
-        body (Account): Account
+        knowledgebase_id (str):
+        body (Knowledgebase): Knowledgebase
 
     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:
-        Account
+        Knowledgebase
     """
 
     return (
         await asyncio_detailed(
-            account_id=account_id,
+            knowledgebase_id=knowledgebase_id,
             client=client,
             body=body,
         )

beamlit/deploy/deploy.py

@@ -77,6 +77,8 @@ def get_beamlit_deployment_from_resource(
                 metadata = EnvironmentMetadata(**value.get("metadata", {}))
                 spec = AgentSpec(**value.get("spec", {}))
                 agent = Agent(metadata=metadata, spec=spec)
+                if not agent.spec.prompt:
+                    agent.spec.prompt = get_description(None, resource)
                 return set_default_values(resource, agent)
         if arg.arg == "function":
             if isinstance(arg.value, ast.Dict):

beamlit/deploy/format.py

@@ -5,8 +5,6 @@ It includes functions to convert arguments, parameters, dictionaries, and agent
 
 import ast
 
-from beamlit.models import AgentChain, StoreFunctionParameter
-
 
 def arg_to_list(arg: ast.List):
     value = []
@@ -45,69 +43,4 @@ def arg_to_dict(arg: ast.keyword):
     for k, v in zip(arg.keys, arg.values):
         if isinstance(k, ast.Constant):
             value[k.value] = format_value(v)
-    return value
-
-def format_parameters(parameters: list[StoreFunctionParameter]) -> str:
-    """
-    Formats function parameters into a YAML-compatible string.
-
-    Args:
-        parameters (list[StoreFunctionParameter]): List of parameter objects.
-
-    Returns:
-        str: YAML-formatted string of parameters.
-    """
-    if not parameters:
-        return "[]"
-
-    formatted = []
-    for param in parameters:
-        formatted.append(f"""
-    - name: {param.name}
-      type: {param.type_}
-      required: {str(param.required).lower()}
-      description: {param.description}""")
-
-    return "\n".join(formatted)
-
-def format_dict(obj: dict) -> str:
-    """
-    Converts a dictionary to a YAML-compatible string.
-
-    Args:
-        obj (dict): The dictionary to format.
-
-    Returns:
-        str: YAML-formatted string representation of the dictionary.
-    """
-    if not obj:
-        return "null"
-    ret = ""
-    for k, v in obj.items():
-        if not v:
-            ret += f"{k}: null\n"
-        else:
-            ret += f"{k}: {v}\n"
-    return ret
-
-def format_agent_chain(agentChain: list[AgentChain]) -> str:
-    """
-    Formats agent chain configuration into a YAML-compatible string.
-
-    Args:
-        agentChain (list[AgentChain]): List of agent chain configurations.
-
-    Returns:
-        str: YAML-formatted string of agent chain.
-    """
-    if not agentChain:
-        return "[]"
-    formatted = []
-
-    for agent in agentChain:
-        formatted.append(f"""
-  - agent: {agent.name}
-    enabled: {agent.enabled}""")
-        if agent.description:
-            formatted.append(f"        description: {agent.description}")
-    return "\n".join(formatted)
+    return value