phenoml 0.0.2__py3-none-any.whl → 0.0.15__py3-none-any.whl

phenoml/agent/__init__.py

@@ -4,23 +4,22 @@
 
 from .types import (
     AgentChatResponse,
+    AgentCreateRequest,
+    AgentCreateRequestProvider,
     AgentDeleteResponse,
-    AgentFhirConfig,
     AgentGetChatMessagesRequestOrder,
     AgentGetChatMessagesResponse,
     AgentListResponse,
     AgentPromptsResponse,
-    AgentProvider,
     AgentResponse,
     AgentTemplate,
-    ChatFhirClientConfig,
+    AgentTemplateProvider,
     ChatMessageTemplate,
     ChatSessionTemplate,
     JsonPatch,
     JsonPatchOperation,
     JsonPatchOperationOp,
     PromptTemplate,
-    ProviderType,
     SuccessResponse,
 )
 from .errors import BadRequestError, ForbiddenError, InternalServerError, NotFoundError, UnauthorizedError
@@ -29,17 +28,17 @@ from .prompts import PromptsDeleteResponse, PromptsListResponse
 
 __all__ = [
     "AgentChatResponse",
+    "AgentCreateRequest",
+    "AgentCreateRequestProvider",
     "AgentDeleteResponse",
-    "AgentFhirConfig",
     "AgentGetChatMessagesRequestOrder",
     "AgentGetChatMessagesResponse",
     "AgentListResponse",
     "AgentPromptsResponse",
-    "AgentProvider",
     "AgentResponse",
     "AgentTemplate",
+    "AgentTemplateProvider",
     "BadRequestError",
-    "ChatFhirClientConfig",
     "ChatMessageTemplate",
     "ChatSessionTemplate",
     "ForbiddenError",
@@ -51,7 +50,6 @@ __all__ = [
     "PromptTemplate",
     "PromptsDeleteResponse",
     "PromptsListResponse",
-    "ProviderType",
     "SuccessResponse",
     "UnauthorizedError",
     "prompts",

phenoml/agent/client.py

@@ -7,14 +7,12 @@ from ..core.request_options import RequestOptions
 from .prompts.client import AsyncPromptsClient, PromptsClient
 from .raw_client import AsyncRawAgentClient, RawAgentClient
 from .types.agent_chat_response import AgentChatResponse
+from .types.agent_create_request_provider import AgentCreateRequestProvider
 from .types.agent_delete_response import AgentDeleteResponse
-from .types.agent_fhir_config import AgentFhirConfig
 from .types.agent_get_chat_messages_request_order import AgentGetChatMessagesRequestOrder
 from .types.agent_get_chat_messages_response import AgentGetChatMessagesResponse
 from .types.agent_list_response import AgentListResponse
-from .types.agent_provider import AgentProvider
 from .types.agent_response import AgentResponse
-from .types.chat_fhir_client_config import ChatFhirClientConfig
 from .types.json_patch import JsonPatch
 
 # this is used as the default value for optional parameters
@@ -46,8 +44,7 @@ class AgentClient:
         description: typing.Optional[str] = OMIT,
         tools: typing.Optional[typing.Sequence[str]] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        provider: typing.Optional[AgentProvider] = OMIT,
-        meta: typing.Optional[AgentFhirConfig] = OMIT,
+        provider: typing.Optional[AgentCreateRequestProvider] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentResponse:
         """
@@ -73,10 +70,8 @@ class AgentClient:
         tags : typing.Optional[typing.Sequence[str]]
             Tags for categorizing the agent
 
-        provider : typing.Optional[AgentProvider]
-            FHIR provider type - can be a single provider or array of providers
-
-        meta : typing.Optional[AgentFhirConfig]
+        provider : typing.Optional[AgentCreateRequestProvider]
+            FHIR provider ID(s) - must be valid UUIDs from existing FHIR providers
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -107,7 +102,6 @@ class AgentClient:
             tools=tools,
             tags=tags,
             provider=provider,
-            meta=meta,
             request_options=request_options,
         )
         return _response.data
@@ -145,7 +139,10 @@ class AgentClient:
         client = phenoml(
             token="YOUR_TOKEN",
         )
-        client.agent.list()
+        client.agent.list(
+            is_active=True,
+            tags="tags",
+        )
         """
         _response = self._raw_client.list(is_active=is_active, tags=tags, request_options=request_options)
         return _response.data
@@ -185,14 +182,13 @@ class AgentClient:
         self,
         id: str,
         *,
-        name: typing.Optional[str] = OMIT,
+        name: str,
+        prompts: typing.Sequence[str],
+        is_active: bool,
         description: typing.Optional[str] = OMIT,
-        prompts: typing.Optional[typing.Sequence[str]] = OMIT,
         tools: typing.Optional[typing.Sequence[str]] = OMIT,
-        is_active: typing.Optional[bool] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        provider: typing.Optional[AgentProvider] = OMIT,
-        meta: typing.Optional[AgentFhirConfig] = OMIT,
+        provider: typing.Optional[AgentCreateRequestProvider] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentResponse:
         """
@@ -203,28 +199,26 @@ class AgentClient:
         id : str
             Agent ID
 
-        name : typing.Optional[str]
+        name : str
             Agent name
 
+        prompts : typing.Sequence[str]
+            Array of prompt IDs to use for this agent
+
+        is_active : bool
+            Whether the agent is active
+
         description : typing.Optional[str]
             Agent description
 
-        prompts : typing.Optional[typing.Sequence[str]]
-            Array of prompt IDs to use for this agent
-
         tools : typing.Optional[typing.Sequence[str]]
             Array of MCP server tool IDs to use for this agent
 
-        is_active : typing.Optional[bool]
-            Whether the agent is active
-
         tags : typing.Optional[typing.Sequence[str]]
             Tags for categorizing the agent
 
-        provider : typing.Optional[AgentProvider]
-            FHIR provider type - can be a single provider or array of providers
-
-        meta : typing.Optional[AgentFhirConfig]
+        provider : typing.Optional[AgentCreateRequestProvider]
+            FHIR provider ID(s) - must be valid UUIDs from existing FHIR providers
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -243,18 +237,20 @@ class AgentClient:
         )
         client.agent.update(
             id="id",
+            name="name",
+            prompts=["prompt_123", "prompt_456"],
+            is_active=True,
         )
         """
         _response = self._raw_client.update(
             id,
             name=name,
-            description=description,
             prompts=prompts,
-            tools=tools,
             is_active=is_active,
+            description=description,
+            tools=tools,
             tags=tags,
             provider=provider,
-            meta=meta,
             request_options=request_options,
         )
         return _response.data
@@ -347,9 +343,10 @@ class AgentClient:
         *,
         message: str,
         agent_id: str,
+        phenoml_on_behalf_of: typing.Optional[str] = None,
+        phenoml_fhir_provider: typing.Optional[str] = None,
         context: typing.Optional[str] = OMIT,
         session_id: typing.Optional[str] = OMIT,
-        meta: typing.Optional[ChatFhirClientConfig] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentChatResponse:
         """
@@ -363,15 +360,20 @@ class AgentClient:
         agent_id : str
             The ID of the agent to chat with
 
+        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.
+
         context : typing.Optional[str]
             Optional context for the conversation
 
         session_id : typing.Optional[str]
             Optional session ID for conversation continuity
 
-        meta : typing.Optional[ChatFhirClientConfig]
-            Optional user-specific FHIR configuration overrides
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -388,6 +390,8 @@ class AgentClient:
             token="YOUR_TOKEN",
         )
         client.agent.chat(
+            phenoml_on_behalf_of="Patient/550e8400-e29b-41d4-a716-446655440000",
+            phenoml_fhir_provider="550e8400-e29b-41d4-a716-446655440000:eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c...",
             message="What is the patient's current condition?",
             agent_id="agent-123",
         )
@@ -395,9 +399,10 @@ class AgentClient:
         _response = self._raw_client.chat(
             message=message,
             agent_id=agent_id,
+            phenoml_on_behalf_of=phenoml_on_behalf_of,
+            phenoml_fhir_provider=phenoml_fhir_provider,
             context=context,
             session_id=session_id,
-            meta=meta,
             request_options=request_options,
         )
         return _response.data
@@ -445,6 +450,9 @@ class AgentClient:
         )
         client.agent.get_chat_messages(
             chat_session_id="chat_session_id",
+            num_messages=1,
+            role="role",
+            order="asc",
         )
         """
         _response = self._raw_client.get_chat_messages(
@@ -482,8 +490,7 @@ class AsyncAgentClient:
         description: typing.Optional[str] = OMIT,
         tools: typing.Optional[typing.Sequence[str]] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        provider: typing.Optional[AgentProvider] = OMIT,
-        meta: typing.Optional[AgentFhirConfig] = OMIT,
+        provider: typing.Optional[AgentCreateRequestProvider] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentResponse:
         """
@@ -509,10 +516,8 @@ class AsyncAgentClient:
         tags : typing.Optional[typing.Sequence[str]]
             Tags for categorizing the agent
 
-        provider : typing.Optional[AgentProvider]
-            FHIR provider type - can be a single provider or array of providers
-
-        meta : typing.Optional[AgentFhirConfig]
+        provider : typing.Optional[AgentCreateRequestProvider]
+            FHIR provider ID(s) - must be valid UUIDs from existing FHIR providers
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -551,7 +556,6 @@ class AsyncAgentClient:
             tools=tools,
             tags=tags,
             provider=provider,
-            meta=meta,
             request_options=request_options,
         )
         return _response.data
@@ -594,7 +598,10 @@ class AsyncAgentClient:
 
 
         async def main() -> None:
-            await client.agent.list()
+            await client.agent.list(
+                is_active=True,
+                tags="tags",
+            )
 
 
         asyncio.run(main())
@@ -645,14 +652,13 @@ class AsyncAgentClient:
         self,
         id: str,
         *,
-        name: typing.Optional[str] = OMIT,
+        name: str,
+        prompts: typing.Sequence[str],
+        is_active: bool,
         description: typing.Optional[str] = OMIT,
-        prompts: typing.Optional[typing.Sequence[str]] = OMIT,
         tools: typing.Optional[typing.Sequence[str]] = OMIT,
-        is_active: typing.Optional[bool] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        provider: typing.Optional[AgentProvider] = OMIT,
-        meta: typing.Optional[AgentFhirConfig] = OMIT,
+        provider: typing.Optional[AgentCreateRequestProvider] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentResponse:
         """
@@ -663,28 +669,26 @@ class AsyncAgentClient:
         id : str
             Agent ID
 
-        name : typing.Optional[str]
+        name : str
             Agent name
 
+        prompts : typing.Sequence[str]
+            Array of prompt IDs to use for this agent
+
+        is_active : bool
+            Whether the agent is active
+
         description : typing.Optional[str]
             Agent description
 
-        prompts : typing.Optional[typing.Sequence[str]]
-            Array of prompt IDs to use for this agent
-
         tools : typing.Optional[typing.Sequence[str]]
             Array of MCP server tool IDs to use for this agent
 
-        is_active : typing.Optional[bool]
-            Whether the agent is active
-
         tags : typing.Optional[typing.Sequence[str]]
             Tags for categorizing the agent
 
-        provider : typing.Optional[AgentProvider]
-            FHIR provider type - can be a single provider or array of providers
-
-        meta : typing.Optional[AgentFhirConfig]
+        provider : typing.Optional[AgentCreateRequestProvider]
+            FHIR provider ID(s) - must be valid UUIDs from existing FHIR providers
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -708,6 +712,9 @@ class AsyncAgentClient:
         async def main() -> None:
             await client.agent.update(
                 id="id",
+                name="name",
+                prompts=["prompt_123", "prompt_456"],
+                is_active=True,
             )
 
 
@@ -716,13 +723,12 @@ class AsyncAgentClient:
         _response = await self._raw_client.update(
             id,
             name=name,
-            description=description,
             prompts=prompts,
-            tools=tools,
             is_active=is_active,
+            description=description,
+            tools=tools,
             tags=tags,
             provider=provider,
-            meta=meta,
             request_options=request_options,
         )
         return _response.data
@@ -831,9 +837,10 @@ class AsyncAgentClient:
         *,
         message: str,
         agent_id: str,
+        phenoml_on_behalf_of: typing.Optional[str] = None,
+        phenoml_fhir_provider: typing.Optional[str] = None,
         context: typing.Optional[str] = OMIT,
         session_id: typing.Optional[str] = OMIT,
-        meta: typing.Optional[ChatFhirClientConfig] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentChatResponse:
         """
@@ -847,15 +854,20 @@ class AsyncAgentClient:
         agent_id : str
             The ID of the agent to chat with
 
+        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.
+
         context : typing.Optional[str]
             Optional context for the conversation
 
         session_id : typing.Optional[str]
             Optional session ID for conversation continuity
 
-        meta : typing.Optional[ChatFhirClientConfig]
-            Optional user-specific FHIR configuration overrides
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -877,6 +889,8 @@ class AsyncAgentClient:
 
         async def main() -> None:
             await client.agent.chat(
+                phenoml_on_behalf_of="Patient/550e8400-e29b-41d4-a716-446655440000",
+                phenoml_fhir_provider="550e8400-e29b-41d4-a716-446655440000:eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c...",
                 message="What is the patient's current condition?",
                 agent_id="agent-123",
             )
@@ -887,9 +901,10 @@ class AsyncAgentClient:
         _response = await self._raw_client.chat(
             message=message,
             agent_id=agent_id,
+            phenoml_on_behalf_of=phenoml_on_behalf_of,
+            phenoml_fhir_provider=phenoml_fhir_provider,
             context=context,
             session_id=session_id,
-            meta=meta,
             request_options=request_options,
         )
         return _response.data
@@ -942,6 +957,9 @@ class AsyncAgentClient:
         async def main() -> None:
             await client.agent.get_chat_messages(
                 chat_session_id="chat_session_id",
+                num_messages=1,
+                role="role",
+                order="asc",
             )