phenoml 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

phenoml/summary/types/__init__.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .create_summary_request_fhir_resources import CreateSummaryRequestFhirResources
+from .create_summary_request_mode import CreateSummaryRequestMode
+from .create_summary_response import CreateSummaryResponse
+from .create_summary_template_response import CreateSummaryTemplateResponse
+from .fhir_bundle import FhirBundle
+from .fhir_bundle_entry_item import FhirBundleEntryItem
+from .fhir_resource import FhirResource
+from .summary_delete_template_response import SummaryDeleteTemplateResponse
+from .summary_get_template_response import SummaryGetTemplateResponse
+from .summary_list_templates_response import SummaryListTemplatesResponse
+from .summary_template import SummaryTemplate
+from .summary_update_template_response import SummaryUpdateTemplateResponse
+
+__all__ = [
+    "CreateSummaryRequestFhirResources",
+    "CreateSummaryRequestMode",
+    "CreateSummaryResponse",
+    "CreateSummaryTemplateResponse",
+    "FhirBundle",
+    "FhirBundleEntryItem",
+    "FhirResource",
+    "SummaryDeleteTemplateResponse",
+    "SummaryGetTemplateResponse",
+    "SummaryListTemplatesResponse",
+    "SummaryTemplate",
+    "SummaryUpdateTemplateResponse",
+]

phenoml/summary/types/create_summary_request_fhir_resources.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .fhir_bundle import FhirBundle
+from .fhir_resource import FhirResource
+
+CreateSummaryRequestFhirResources = typing.Union[FhirResource, FhirBundle]

phenoml/summary/types/create_summary_request_mode.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+CreateSummaryRequestMode = typing.Union[typing.Literal["narrative", "flatten"], typing.Any]

phenoml/{construe/types/construe_cohort_response_queries_item_code_extract_results_item_codes_item.py → summary/types/create_summary_response.py}

@@ -6,20 +6,17 @@ import pydantic
 from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
-class ConstrueCohortResponseQueriesItemCodeExtractResultsItemCodesItem(UniversalBaseModel):
-    code: typing.Optional[str] = pydantic.Field(default=None)
+class CreateSummaryResponse(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: typing.Optional[str] = None
+    summary: typing.Optional[str] = pydantic.Field(default=None)
     """
-    Medical code.
+    Generated summary text
     """
 
-    description: typing.Optional[str] = pydantic.Field(default=None)
+    warnings: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
     """
-    Description of the medical code.
-    """
-
-    reason: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    Reason for selecting the code.
+    Unresolved placeholders or issues (narrative mode only)
     """
 
     if IS_PYDANTIC_V2:

phenoml/summary/types/create_summary_template_response.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .summary_template import SummaryTemplate
+
+
+class CreateSummaryTemplateResponse(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: typing.Optional[str] = None
+    template_id: typing.Optional[str] = None
+    template: typing.Optional[SummaryTemplate] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

phenoml/summary/types/fhir_bundle.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ...core.serialization import FieldMetadata
+from .fhir_bundle_entry_item import FhirBundleEntryItem
+
+
+class FhirBundle(UniversalBaseModel):
+    resource_type: typing_extensions.Annotated[typing.Literal["Bundle"], FieldMetadata(alias="resourceType")] = "Bundle"
+    entry: typing.List[FhirBundleEntryItem]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

phenoml/{construe/types/bad_request_error_body.py → summary/types/fhir_bundle_entry_item.py}

@@ -4,18 +4,11 @@ import typing
 
 import pydantic
 from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .fhir_resource import FhirResource
 
 
-class BadRequestErrorBody(UniversalBaseModel):
-    status: typing.Optional[int] = pydantic.Field(default=None)
-    """
-    HTTP status code.
-    """
-
-    message: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    Error message.
-    """
+class FhirBundleEntryItem(UniversalBaseModel):
+    resource: typing.Optional[FhirResource] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

phenoml/{construe/types/unauthorized_error_body.py → summary/types/fhir_resource.py}

@@ -3,18 +3,15 @@
 import typing
 
 import pydantic
+import typing_extensions
 from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ...core.serialization import FieldMetadata
 
 
-class UnauthorizedErrorBody(UniversalBaseModel):
-    status: typing.Optional[int] = pydantic.Field(default=None)
+class FhirResource(UniversalBaseModel):
+    resource_type: typing_extensions.Annotated[str, FieldMetadata(alias="resourceType")] = pydantic.Field()
     """
-    HTTP status code.
-    """
-
-    message: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    Error message.
+    FHIR resource type
     """
 
     if IS_PYDANTIC_V2:

phenoml/{construe/types/internal_server_error_body.py → summary/types/summary_delete_template_response.py}

@@ -6,16 +6,9 @@ import pydantic
 from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
-class InternalServerErrorBody(UniversalBaseModel):
-    status: typing.Optional[int] = pydantic.Field(default=None)
-    """
-    HTTP status code.
-    """
-
-    message: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    Error message.
-    """
+class SummaryDeleteTemplateResponse(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

phenoml/summary/types/summary_get_template_response.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .summary_template import SummaryTemplate
+
+
+class SummaryGetTemplateResponse(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    template: typing.Optional[SummaryTemplate] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

phenoml/summary/types/summary_list_templates_response.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .summary_template import SummaryTemplate
+
+
+class SummaryListTemplatesResponse(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    templates: typing.Optional[typing.List[SummaryTemplate]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

phenoml/summary/types/summary_template.py

@@ -0,0 +1,41 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class SummaryTemplate(UniversalBaseModel):
+    id: typing.Optional[str] = None
+    user_id: typing.Optional[str] = None
+    name: typing.Optional[str] = None
+    description: typing.Optional[str] = None
+    template: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Template with {{resource.field}} placeholders
+    """
+
+    target_resources: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    List of FHIR resource types/profiles
+    """
+
+    mode: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Template mode (stored value)
+    """
+
+    metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    created_at: typing.Optional[dt.datetime] = None
+    updated_at: typing.Optional[dt.datetime] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

phenoml/summary/types/summary_update_template_response.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .summary_template import SummaryTemplate
+
+
+class SummaryUpdateTemplateResponse(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: typing.Optional[str] = None
+    template: typing.Optional[SummaryTemplate] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

phenoml/tools/client.py

@@ -36,6 +36,8 @@ class ToolsClient:
         *,
         resource: Lang2FhirAndCreateRequestResource,
         text: str,
+        phenoml_on_behalf_of: typing.Optional[str] = None,
+        phenoml_fhir_provider: typing.Optional[str] = None,
         provider: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Lang2FhirAndCreateResponse:
@@ -50,6 +52,14 @@ class ToolsClient:
         text : str
             Natural language text to convert to FHIR resource
 
+        phenoml_on_behalf_of : typing.Optional[str]
+            Optional header for on-behalf-of authentication. Used when making requests on behalf of another user or entity.
+            Must be in the format: Patient/{uuid} or Practitioner/{uuid}
+
+        phenoml_fhir_provider : typing.Optional[str]
+            Optional header for FHIR provider authentication. Contains credentials in the format {fhir_provider_id}:{oauth2_token}.
+            Multiple FHIR provider integrations can be provided as comma-separated values.
+
         provider : typing.Optional[str]
             FHIR provider ID - must be a valid UUID from existing FHIR providers. also supports provider by name (e.g. medplum)
 
@@ -69,12 +79,19 @@ class ToolsClient:
             token="YOUR_TOKEN",
         )
         client.tools.create_fhir_resource(
+            phenoml_on_behalf_of="Patient/550e8400-e29b-41d4-a716-446655440000",
+            phenoml_fhir_provider="550e8400-e29b-41d4-a716-446655440000:eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c...",
             resource="auto",
             text="Patient John Doe has severe asthma with acute exacerbation",
         )
         """
         _response = self._raw_client.create_fhir_resource(
-            resource=resource, text=text, provider=provider, request_options=request_options
+            resource=resource,
+            text=text,
+            phenoml_on_behalf_of=phenoml_on_behalf_of,
+            phenoml_fhir_provider=phenoml_fhir_provider,
+            provider=provider,
+            request_options=request_options,
         )
         return _response.data
 
@@ -82,6 +99,8 @@ class ToolsClient:
         self,
         *,
         text: str,
+        phenoml_on_behalf_of: typing.Optional[str] = None,
+        phenoml_fhir_provider: typing.Optional[str] = None,
         patient_id: typing.Optional[str] = OMIT,
         practitioner_id: typing.Optional[str] = OMIT,
         count: typing.Optional[int] = OMIT,
@@ -96,6 +115,14 @@ class ToolsClient:
         text : str
             Natural language text to convert to FHIR search parameters
 
+        phenoml_on_behalf_of : typing.Optional[str]
+            Optional header for on-behalf-of authentication. Used when making requests on behalf of another user or entity.
+            Must be in the format: Patient/{uuid} or Practitioner/{uuid}
+
+        phenoml_fhir_provider : typing.Optional[str]
+            Optional header for FHIR provider authentication. Contains credentials in the format {fhir_provider_id}:{oauth2_token}.
+            Multiple FHIR provider integrations can be provided as comma-separated values.
+
         patient_id : typing.Optional[str]
             Patient ID to filter results
 
@@ -124,11 +151,15 @@ class ToolsClient:
             token="YOUR_TOKEN",
         )
         client.tools.search_fhir_resources(
+            phenoml_on_behalf_of="Patient/550e8400-e29b-41d4-a716-446655440000",
+            phenoml_fhir_provider="550e8400-e29b-41d4-a716-446655440000:eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c...",
             text="Find all appointments for patient John Doe next week",
         )
         """
         _response = self._raw_client.search_fhir_resources(
             text=text,
+            phenoml_on_behalf_of=phenoml_on_behalf_of,
+            phenoml_fhir_provider=phenoml_fhir_provider,
             patient_id=patient_id,
             practitioner_id=practitioner_id,
             count=count,
@@ -138,7 +169,13 @@ class ToolsClient:
         return _response.data
 
     def analyze_cohort(
-        self, *, text: str, provider: str, request_options: typing.Optional[RequestOptions] = None
+        self,
+        *,
+        text: str,
+        provider: str,
+        phenoml_on_behalf_of: typing.Optional[str] = None,
+        phenoml_fhir_provider: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> CohortResponse:
         """
         Uses LLM to extract search concepts from natural language and builds patient cohorts with inclusion/exclusion criteria
@@ -151,6 +188,14 @@ class ToolsClient:
         provider : str
             FHIR provider ID - must be a valid UUID from existing FHIR providers. also supports provider by name (e.g. medplum)
 
+        phenoml_on_behalf_of : typing.Optional[str]
+            Optional header for on-behalf-of authentication. Used when making requests on behalf of another user or entity.
+            Must be in the format: Patient/{uuid} or Practitioner/{uuid}
+
+        phenoml_fhir_provider : typing.Optional[str]
+            Optional header for FHIR provider authentication. Contains credentials in the format {fhir_provider_id}:{oauth2_token}.
+            Multiple FHIR provider integrations can be provided as comma-separated values.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -167,11 +212,19 @@ class ToolsClient:
             token="YOUR_TOKEN",
         )
         client.tools.analyze_cohort(
+            phenoml_on_behalf_of="Patient/550e8400-e29b-41d4-a716-446655440000",
+            phenoml_fhir_provider="550e8400-e29b-41d4-a716-446655440000:eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c...",
             text="female patients over 20 with diabetes but not hypertension",
             provider="550e8400-e29b-41d4-a716-446655440000",
         )
         """
-        _response = self._raw_client.analyze_cohort(text=text, provider=provider, request_options=request_options)
+        _response = self._raw_client.analyze_cohort(
+            text=text,
+            provider=provider,
+            phenoml_on_behalf_of=phenoml_on_behalf_of,
+            phenoml_fhir_provider=phenoml_fhir_provider,
+            request_options=request_options,
+        )
         return _response.data
 
 
@@ -196,6 +249,8 @@ class AsyncToolsClient:
         *,
         resource: Lang2FhirAndCreateRequestResource,
         text: str,
+        phenoml_on_behalf_of: typing.Optional[str] = None,
+        phenoml_fhir_provider: typing.Optional[str] = None,
         provider: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Lang2FhirAndCreateResponse:
@@ -210,6 +265,14 @@ class AsyncToolsClient:
         text : str
             Natural language text to convert to FHIR resource
 
+        phenoml_on_behalf_of : typing.Optional[str]
+            Optional header for on-behalf-of authentication. Used when making requests on behalf of another user or entity.
+            Must be in the format: Patient/{uuid} or Practitioner/{uuid}
+
+        phenoml_fhir_provider : typing.Optional[str]
+            Optional header for FHIR provider authentication. Contains credentials in the format {fhir_provider_id}:{oauth2_token}.
+            Multiple FHIR provider integrations can be provided as comma-separated values.
+
         provider : typing.Optional[str]
             FHIR provider ID - must be a valid UUID from existing FHIR providers. also supports provider by name (e.g. medplum)
 
@@ -234,6 +297,8 @@ class AsyncToolsClient:
 
         async def main() -> None:
             await client.tools.create_fhir_resource(
+                phenoml_on_behalf_of="Patient/550e8400-e29b-41d4-a716-446655440000",
+                phenoml_fhir_provider="550e8400-e29b-41d4-a716-446655440000:eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c...",
                 resource="auto",
                 text="Patient John Doe has severe asthma with acute exacerbation",
             )
@@ -242,7 +307,12 @@ class AsyncToolsClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.create_fhir_resource(
-            resource=resource, text=text, provider=provider, request_options=request_options
+            resource=resource,
+            text=text,
+            phenoml_on_behalf_of=phenoml_on_behalf_of,
+            phenoml_fhir_provider=phenoml_fhir_provider,
+            provider=provider,
+            request_options=request_options,
         )
         return _response.data
 
@@ -250,6 +320,8 @@ class AsyncToolsClient:
         self,
         *,
         text: str,
+        phenoml_on_behalf_of: typing.Optional[str] = None,
+        phenoml_fhir_provider: typing.Optional[str] = None,
         patient_id: typing.Optional[str] = OMIT,
         practitioner_id: typing.Optional[str] = OMIT,
         count: typing.Optional[int] = OMIT,
@@ -264,6 +336,14 @@ class AsyncToolsClient:
         text : str
             Natural language text to convert to FHIR search parameters
 
+        phenoml_on_behalf_of : typing.Optional[str]
+            Optional header for on-behalf-of authentication. Used when making requests on behalf of another user or entity.
+            Must be in the format: Patient/{uuid} or Practitioner/{uuid}
+
+        phenoml_fhir_provider : typing.Optional[str]
+            Optional header for FHIR provider authentication. Contains credentials in the format {fhir_provider_id}:{oauth2_token}.
+            Multiple FHIR provider integrations can be provided as comma-separated values.
+
         patient_id : typing.Optional[str]
             Patient ID to filter results
 
@@ -297,6 +377,8 @@ class AsyncToolsClient:
 
         async def main() -> None:
             await client.tools.search_fhir_resources(
+                phenoml_on_behalf_of="Patient/550e8400-e29b-41d4-a716-446655440000",
+                phenoml_fhir_provider="550e8400-e29b-41d4-a716-446655440000:eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c...",
                 text="Find all appointments for patient John Doe next week",
             )
 
@@ -305,6 +387,8 @@ class AsyncToolsClient:
         """
         _response = await self._raw_client.search_fhir_resources(
             text=text,
+            phenoml_on_behalf_of=phenoml_on_behalf_of,
+            phenoml_fhir_provider=phenoml_fhir_provider,
             patient_id=patient_id,
             practitioner_id=practitioner_id,
             count=count,
@@ -314,7 +398,13 @@ class AsyncToolsClient:
         return _response.data
 
     async def analyze_cohort(
-        self, *, text: str, provider: str, request_options: typing.Optional[RequestOptions] = None
+        self,
+        *,
+        text: str,
+        provider: str,
+        phenoml_on_behalf_of: typing.Optional[str] = None,
+        phenoml_fhir_provider: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> CohortResponse:
         """
         Uses LLM to extract search concepts from natural language and builds patient cohorts with inclusion/exclusion criteria
@@ -327,6 +417,14 @@ class AsyncToolsClient:
         provider : str
             FHIR provider ID - must be a valid UUID from existing FHIR providers. also supports provider by name (e.g. medplum)
 
+        phenoml_on_behalf_of : typing.Optional[str]
+            Optional header for on-behalf-of authentication. Used when making requests on behalf of another user or entity.
+            Must be in the format: Patient/{uuid} or Practitioner/{uuid}
+
+        phenoml_fhir_provider : typing.Optional[str]
+            Optional header for FHIR provider authentication. Contains credentials in the format {fhir_provider_id}:{oauth2_token}.
+            Multiple FHIR provider integrations can be provided as comma-separated values.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -348,6 +446,8 @@ class AsyncToolsClient:
 
         async def main() -> None:
             await client.tools.analyze_cohort(
+                phenoml_on_behalf_of="Patient/550e8400-e29b-41d4-a716-446655440000",
+                phenoml_fhir_provider="550e8400-e29b-41d4-a716-446655440000:eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c...",
                 text="female patients over 20 with diabetes but not hypertension",
                 provider="550e8400-e29b-41d4-a716-446655440000",
             )
@@ -355,5 +455,11 @@ class AsyncToolsClient:
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.analyze_cohort(text=text, provider=provider, request_options=request_options)
+        _response = await self._raw_client.analyze_cohort(
+            text=text,
+            provider=provider,
+            phenoml_on_behalf_of=phenoml_on_behalf_of,
+            phenoml_fhir_provider=phenoml_fhir_provider,
+            request_options=request_options,
+        )
         return _response.data