fountain-life-service-clients 3.28.2__py3-none-any.whl

fountain_life_service_clients/_base_client.py

@@ -0,0 +1,30 @@
+from typing import NotRequired, Unpack, cast
+
+from lifeomic_chatbot_tools.aws.alpha import (
+    Alpha,
+)
+from lifeomic_chatbot_tools.aws.alpha import (
+    AlphaConfig as BaseAlphaConfig,
+)
+from lifeomic_chatbot_tools.aws.alpha import (
+    AlphaResponse as BaseAlphaResponse,
+)
+
+
+class AlphaConfig(BaseAlphaConfig):
+    # Pull target out as another keyword arg instead of a positional
+    # arg, to make it easier to work with.
+    target: NotRequired[str]
+
+
+class BaseClient:
+    def __init__(self, **cfg: Unpack[AlphaConfig]):
+        self.client = Alpha(**cfg)
+
+
+# A typed version of AlphaResponse that allows us to access the
+# response body as a typed object.
+class AlphaResponse[T](BaseAlphaResponse):
+    @property
+    def body(self):
+        return cast(T, super().body)

fountain_life_service_clients/account_service.py

@@ -0,0 +1,116 @@
+# This file was generated automatically. Do not edit it directly.
+# generated by datamodel-codegen:
+#   filename:  account-service.json
+from typing import List, Literal, NotRequired, Optional, TypedDict, Unpack, cast
+from urllib.parse import quote
+
+from fountain_life_service_clients._base_client import (
+    AlphaConfig,
+    AlphaResponse,
+    BaseClient,
+)
+
+
+class Account(TypedDict):
+    id: str
+    name: str
+    owner: str
+    type: Literal["FREE", "PAID", "ENTERPRISE"]
+
+
+class ListAllAccountsResponse(TypedDict):
+    accounts: List[Account]
+
+
+class CreateAnAccountRequest(TypedDict):
+    id: str
+    name: str
+    owner: str
+    type: Literal["FREE", "PAID", "ENTERPRISE"]
+
+
+class CreateAnAccountResponse(TypedDict):
+    id: str
+    name: str
+    owner: str
+    type: Literal["FREE", "PAID", "ENTERPRISE"]
+
+
+class RetrieveAnAccountParams(TypedDict):
+    include: NotRequired[Literal["groups"]]
+
+
+class RetrieveAnAccountResponse(TypedDict):
+    id: str
+    name: str
+    owner: str
+    type: Literal["FREE", "PAID", "ENTERPRISE"]
+
+
+class UpdateAnAccountRequest(TypedDict):
+    name: NotRequired[str]
+    deletionDate: NotRequired[Optional[str]]
+
+
+class UpdateAnAccountResponse(TypedDict):
+    id: str
+    name: str
+    owner: str
+    type: Literal["FREE", "PAID", "ENTERPRISE"]
+
+
+class DeleteAnAccountParams(TypedDict):
+    force: NotRequired[bool]
+
+
+class DeleteAnAccountResponse1(TypedDict):
+    """
+    The account object, or null if the account was not found/already deleted.
+    """
+
+    id: str
+    name: str
+    owner: str
+    type: Literal["FREE", "PAID", "ENTERPRISE"]
+
+
+DeleteAnAccountResponse = Optional[DeleteAnAccountResponse1]
+
+
+class AccountServiceClient(BaseClient):
+    def __init__(self, **cfg: Unpack[AlphaConfig]):
+        kwargs = {"target": "lambda://account-service:deployed", **(cfg or {})}
+        super().__init__(**kwargs)
+
+    async def list_all_accounts(self):
+        """Returns a list of accounts that the user has access to."""
+        res = await self.client.request(path="/v1/accounts", method="GET")
+        return cast(AlphaResponse[ListAllAccountsResponse], res)
+
+    async def create_an_account(self, body: CreateAnAccountRequest):
+        """Creates an account, returning the account object."""
+        res = await self.client.request(
+            path="/v1/accounts", method="POST", body=cast(dict, body)
+        )
+        return cast(AlphaResponse[CreateAnAccountResponse], res)
+
+    async def retrieve_an_account(self, id: str, params: RetrieveAnAccountParams):
+        """Retrieves details about an account. Returns the account object."""
+        res = await self.client.request(
+            path=f"/v1/accounts/{quote(id)}", method="GET", params=cast(dict, params)
+        )
+        return cast(AlphaResponse[RetrieveAnAccountResponse], res)
+
+    async def update_an_account(self, id: str, body: UpdateAnAccountRequest):
+        """Update an account by changing the name or clearing out a pending deletion date. Returns the account object."""
+        res = await self.client.request(
+            path=f"/v1/accounts/{quote(id)}", method="PATCH", body=cast(dict, body)
+        )
+        return cast(AlphaResponse[UpdateAnAccountResponse], res)
+
+    async def delete_an_account(self, id: str, params: DeleteAnAccountParams):
+        """Deletes an account. By default, the account will not be deleted for 14 days. During this time, the pending deletion can be cancelled by using the `PATCH` method. After the 14 day grace period, the account and all of its data will be removed."""
+        res = await self.client.request(
+            path=f"/v1/accounts/{quote(id)}", method="DELETE", params=cast(dict, params)
+        )
+        return cast(AlphaResponse[DeleteAnAccountResponse], res)

fountain_life_service_clients/account_service_policy_attributes.py

@@ -0,0 +1,46 @@
+# This file was generated automatically. Do not edit it directly.
+# generated by datamodel-codegen:
+#   filename:  account-service-policy-attributes.json
+from typing import Any, Dict, TypedDict, Unpack, cast
+from urllib.parse import quote
+
+from fountain_life_service_clients._base_client import (
+    AlphaConfig,
+    AlphaResponse,
+    BaseClient,
+)
+
+
+class PolicyAttributesObject(TypedDict):
+    attributes: Dict[str, Any]
+
+
+class AccountServicePolicyAttributesClient(BaseClient):
+    def __init__(self, **cfg: Unpack[AlphaConfig]):
+        kwargs = {"target": "lambda://account-service:deployed", **(cfg or {})}
+        super().__init__(**kwargs)
+
+    async def update_policy_attributes_for_user(
+        self, username: str, body: PolicyAttributesObject
+    ):
+        """Updates the policy attributes for the specified user, returning the updated
+        attributes.
+
+        Using this API requires access to the `accessAdmin` operation.
+        """
+        res = await self.client.request(
+            path=f"/v1/policy-attributes/users/{quote(username)}",
+            method="PUT",
+            body=cast(dict, body),
+        )
+        return cast(AlphaResponse[PolicyAttributesObject], res)
+
+    async def get_policy_attributes_for_user(self, username: str):
+        """Fetches the policy attributes for the specified user.
+
+        Using this API requires access to the `accessAdmin` operation.
+        """
+        res = await self.client.request(
+            path=f"/v1/policy-attributes/users/{quote(username)}", method="GET"
+        )
+        return cast(AlphaResponse[PolicyAttributesObject], res)

fountain_life_service_clients/agents_api_service.py

@@ -0,0 +1,197 @@
+# This file was generated automatically. Do not edit it directly.
+# generated by datamodel-codegen:
+#   filename:  agents-api-service.json
+from typing import Any, Dict, List, Literal, NotRequired, TypedDict, Unpack, cast
+from urllib.parse import quote
+
+from fountain_life_service_clients._base_client import (
+    AlphaConfig,
+    AlphaResponse,
+    BaseClient,
+)
+
+
+class InvokeBasicAgentRequest(TypedDict):
+    input: str
+
+
+class InvokeBasicAgentResponse(TypedDict):
+    output: str
+
+
+class InvokeHealthSummaryAgentRequest(TypedDict):
+    subject_id: str
+    project_id: str
+    mutate: NotRequired[bool]
+    should_extract_data: NotRequired[bool]
+    should_include_synopsis_layout: NotRequired[bool]
+
+
+class InvokeHealthSummaryAgentResponse(TypedDict):
+    task_id: str
+
+
+class GetHealthSummaryInvocationsParams(TypedDict):
+    subject_id: str
+
+
+class GetHealthSummaryInvocationsResponseItem(TypedDict):
+    created: float
+    expires: NotRequired[float]
+    id: str
+    scheduled: NotRequired[float]
+    status: Literal[
+        "scheduled", "pending", "processing", "failed", "completed", "expired"
+    ]
+    accountId: str
+    agent: str
+    agentInput: Dict[str, Any]
+    subjectId: str
+
+
+GetHealthSummaryInvocationsResponse = List[GetHealthSummaryInvocationsResponseItem]
+
+
+class GetHealthSummaryScheduledTaskResponse(TypedDict):
+    created: float
+    expires: NotRequired[float]
+    id: str
+    scheduled: NotRequired[float]
+    status: Literal[
+        "scheduled", "pending", "processing", "failed", "completed", "expired"
+    ]
+    accountId: str
+    agent: str
+    agentInput: Dict[str, Any]
+    subjectId: str
+
+
+class FlushHealthSummaryScheduledTaskRequest(TypedDict):
+    pass
+
+
+class ProvideMessageFeedbackRequest(TypedDict):
+    agent_name: str
+    trace_id: str
+    feedback: NotRequired[str]
+    score: NotRequired[float]
+    emoji: NotRequired[str]
+
+
+class InvokeActionPlanNudgeRequest(TypedDict):
+    account_id: str
+    project_id: str
+    subject_id: str
+    user_id: str
+    correlation_id: NotRequired[str]
+
+
+class InvokeActionPlanNudgeResponse(TypedDict):
+    subject_id: str
+    nudge: Dict[str, Any]
+
+
+class GetAgentTokenResponse(TypedDict):
+    AccessKeyId: str
+    SecretAccessKey: str
+    SessionToken: str
+    Expiration: str
+
+
+class ValidateFileOwnershipRequest(TypedDict):
+    subject_id: str
+    file_id: str
+    project_id: str
+    document_reference_id: str
+
+
+class Result(TypedDict):
+    status: Literal["valid", "invalid", "inconclusive"]
+    reasoning: str
+
+
+class ValidateFileOwnershipResponse(TypedDict):
+    subject_id: str
+    file_id: str
+    document_reference_id: str
+    result: Result
+
+
+class AgentsApiServiceClient(BaseClient):
+    def __init__(self, **cfg: Unpack[AlphaConfig]):
+        kwargs = {"target": "lambda://agents-api-service-v2:deployed", **(cfg or {})}
+        super().__init__(**kwargs)
+
+    async def invoke_basic_agent(self, body: InvokeBasicAgentRequest):
+        """Invoke the basic agent"""
+        res = await self.client.request(
+            path="/v1/agents-v2/basic/invoke", method="POST", body=cast(dict, body)
+        )
+        return cast(AlphaResponse[InvokeBasicAgentResponse], res)
+
+    async def invoke_health_summary_agent(self, body: InvokeHealthSummaryAgentRequest):
+        """Invoke the health summary agent"""
+        res = await self.client.request(
+            path="/v1/agents-v2/health-summary/invoke",
+            method="POST",
+            body=cast(dict, body),
+        )
+        return cast(AlphaResponse[InvokeHealthSummaryAgentResponse], res)
+
+    async def get_health_summary_invocations(
+        self, params: GetHealthSummaryInvocationsParams
+    ):
+        """Get the health summary invocations"""
+        res = await self.client.request(
+            path="/v1/agents-v2/health-summary/invocations",
+            method="GET",
+            params=cast(dict, params),
+        )
+        return cast(AlphaResponse[GetHealthSummaryInvocationsResponse], res)
+
+    async def get_health_summary_scheduled_task(self, task_id: str):
+        """Get a health summary scheduled task"""
+        res = await self.client.request(
+            path=f"/v1/agents-v2/health-summary/invocations/{quote(task_id)}",
+            method="GET",
+        )
+        return cast(AlphaResponse[GetHealthSummaryScheduledTaskResponse], res)
+
+    async def flush_health_summary_scheduled_task(
+        self, task_id: str, body: FlushHealthSummaryScheduledTaskRequest
+    ):
+        """Flush the health summary scheduled task so it is run immediately"""
+        await self.client.request(
+            path=f"/v1/agents-v2/health-summary/scheduled/{quote(task_id)}",
+            method="PATCH",
+            body=cast(dict, body),
+        )
+
+    async def provide_message_feedback(self, body: ProvideMessageFeedbackRequest):
+        """Provide feedback on a message"""
+        await self.client.request(
+            path="/v1/agents-v2/feedback", method="POST", body=cast(dict, body)
+        )
+
+    async def invoke_action_plan_nudge(self, body: InvokeActionPlanNudgeRequest):
+        """Invoke the action plan nudge agent"""
+        res = await self.client.request(
+            path="/v1/private/agents-v2/action-plan-nudge/invoke",
+            method="POST",
+            body=cast(dict, body),
+        )
+        return cast(AlphaResponse[InvokeActionPlanNudgeResponse], res)
+
+    async def get_agent_token(self):
+        """Get an agent token for the current user"""
+        res = await self.client.request(path="/v1/agents-v2/token", method="GET")
+        return cast(AlphaResponse[GetAgentTokenResponse], res)
+
+    async def validate_file_ownership(self, body: ValidateFileOwnershipRequest):
+        """Validate file ownership"""
+        res = await self.client.request(
+            path="/v1/private/agents-v2/validate-ownership/invoke",
+            method="POST",
+            body=cast(dict, body),
+        )
+        return cast(AlphaResponse[ValidateFileOwnershipResponse], res)

fountain_life_service_clients/claimed_domains_service.py

@@ -0,0 +1,70 @@
+# This file was generated automatically. Do not edit it directly.
+# generated by datamodel-codegen:
+#   filename:  claimed-domains-service.json
+from typing import List, Literal, NotRequired, TypedDict, Unpack, cast
+
+from fountain_life_service_clients._base_client import (
+    AlphaConfig,
+    AlphaResponse,
+    BaseClient,
+)
+
+
+class PrivateListDomainsParams(TypedDict):
+    domain: str
+    pageSize: NotRequired[str]
+    nextPageToken: NotRequired[str]
+
+
+class Basic(TypedDict):
+    """
+    State of basic functionality of a claimed domain such as SSO login suggestions.
+    """
+
+    status: Literal["verified", "unverified"]
+
+
+class Email(TypedDict):
+    """
+    State of being able to send messages from an email address of the domain.
+    """
+
+    status: Literal["verified", "unverified"]
+
+
+class Verifications(TypedDict):
+    """
+    Describes the current state of the claimed domain's verifications.
+    """
+
+    basic: Basic
+    email: Email
+
+
+class Item(TypedDict):
+    account: str
+    domain: str
+    eligibleForLoginMethods: bool
+    verifications: Verifications
+
+
+class Links(TypedDict):
+    self: str
+    next: NotRequired[str]
+
+
+class PrivateListDomainsResponse(TypedDict):
+    items: List[Item]
+    links: Links
+
+
+class ClaimedDomainsServiceClient(BaseClient):
+    def __init__(self, **cfg: Unpack[AlphaConfig]):
+        kwargs = {"target": "lambda://claimed-domains-service:deployed", **(cfg or {})}
+        super().__init__(**kwargs)
+
+    async def private_list_domains(self, params: PrivateListDomainsParams):
+        res = await self.client.request(
+            path="/v1/private/claimed-domains", method="GET", params=cast(dict, params)
+        )
+        return cast(AlphaResponse[PrivateListDomainsResponse], res)

fountain_life_service_clients/cohorts_service.py

@@ -0,0 +1,184 @@
+# This file was generated automatically. Do not edit it directly.
+# generated by datamodel-codegen:
+#   filename:  cohorts-service.json
+from typing import Any, List, Literal, NotRequired, TypedDict, Unpack, cast
+from urllib.parse import quote
+
+from fountain_life_service_clients._base_client import (
+    AlphaConfig,
+    AlphaResponse,
+    BaseClient,
+)
+
+
+class GetCohortsParams(TypedDict):
+    projectId: str
+    name: NotRequired[str]
+    pageSize: NotRequired[float]
+    nextPageToken: NotRequired[str]
+
+
+class Query(TypedDict):
+    """
+    A query used to define a cohort
+    """
+
+    project: str
+    queryType: str
+    query: NotRequired[Any]
+
+
+class Item(TypedDict):
+    """
+    Response body for a cohort
+    """
+
+    id: str
+    name: str
+    description: NotRequired[str]
+    ownerProject: str
+    resultCount: NotRequired[float]
+    creatorUser: NotRequired[str]
+    creationTime: NotRequired[str]
+    queries: List[Query]
+    subjectIds: List[str]
+
+
+class Links(TypedDict):
+    self: str
+    next: NotRequired[str]
+
+
+class GetCohortsResponse(TypedDict):
+    """
+    Response body for listing cohorts
+    """
+
+    items: List[Item]
+    links: Links
+
+
+class CreateCohortRequest(TypedDict):
+    """
+    Request body for creating or updating a cohort
+    """
+
+    name: str
+    description: NotRequired[str]
+    ownerProject: str
+    queries: List[Query]
+
+
+class CreateCohortResponse(TypedDict):
+    """
+    Response body for a cohort
+    """
+
+    id: str
+    name: str
+    description: NotRequired[str]
+    ownerProject: str
+    resultCount: NotRequired[float]
+    creatorUser: NotRequired[str]
+    creationTime: NotRequired[str]
+    queries: List[Query]
+    subjectIds: List[str]
+
+
+class GetCohortResponse(TypedDict):
+    """
+    Response body for a cohort
+    """
+
+    id: str
+    name: str
+    description: NotRequired[str]
+    ownerProject: str
+    resultCount: NotRequired[float]
+    creatorUser: NotRequired[str]
+    creationTime: NotRequired[str]
+    queries: List[Query]
+    subjectIds: List[str]
+
+
+class UpdateCohortRequest(TypedDict):
+    """
+    Request body for creating or updating a cohort
+    """
+
+    name: str
+    description: NotRequired[str]
+    ownerProject: str
+    queries: List[Query]
+
+
+class UpdateCohortResponse(TypedDict):
+    """
+    Response body for a cohort
+    """
+
+    id: str
+    name: str
+    description: NotRequired[str]
+    ownerProject: str
+    resultCount: NotRequired[float]
+    creatorUser: NotRequired[str]
+    creationTime: NotRequired[str]
+    queries: List[Query]
+    subjectIds: List[str]
+
+
+class UpdateCohortPatientsRequest(TypedDict):
+    subjectIds: List[str]
+    action: Literal["ADD", "REMOVE"]
+
+
+class UpdateCohortPatientsResponse(TypedDict):
+    pass
+
+
+class CohortsServiceClient(BaseClient):
+    def __init__(self, **cfg: Unpack[AlphaConfig]):
+        kwargs = {"target": "lambda://cohorts-service:deployed", **(cfg or {})}
+        super().__init__(**kwargs)
+
+    async def get_cohorts(self, params: GetCohortsParams):
+        """Returns a list of cohorts that the user has access to."""
+        res = await self.client.request(
+            path="/v1/cohorts", method="GET", params=cast(dict, params)
+        )
+        return cast(AlphaResponse[GetCohortsResponse], res)
+
+    async def create_cohort(self, body: CreateCohortRequest):
+        """Create a cohort, representing a subset of a project based on one or more queries"""
+        res = await self.client.request(
+            path="/v1/cohorts", method="POST", body=cast(dict, body)
+        )
+        return cast(AlphaResponse[CreateCohortResponse], res)
+
+    async def get_cohort(self, cohort_id: str):
+        """Returns a cohort."""
+        res = await self.client.request(
+            path=f"/v1/cohorts/{quote(cohort_id)}", method="GET"
+        )
+        return cast(AlphaResponse[GetCohortResponse], res)
+
+    async def update_cohort(self, cohort_id: str, body: UpdateCohortRequest):
+        """Returns the updated cohort."""
+        res = await self.client.request(
+            path=f"/v1/cohorts/{quote(cohort_id)}",
+            method="PATCH",
+            body=cast(dict, body),
+        )
+        return cast(AlphaResponse[UpdateCohortResponse], res)
+
+    async def update_cohort_patients(
+        self, project_id: str, cohort_id: str, body: UpdateCohortPatientsRequest
+    ):
+        """Adds or removes a patient from a cohort. Only works in the Fountain Life account + project."""
+        res = await self.client.request(
+            path=f"/v1/private/cohorts/projects/{quote(project_id)}/cohorts/{quote(cohort_id)}/patients",
+            method="PATCH",
+            body=cast(dict, body),
+        )
+        return cast(AlphaResponse[UpdateCohortPatientsResponse], res)