phenoml 0.0.1__tar.gz → 0.0.2__tar.gz

{phenoml-0.0.1 → phenoml-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: phenoml
-Version: 0.0.1
+Version: 0.0.2
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers

{phenoml-0.0.1 → phenoml-0.0.2}/pyproject.toml

@@ -3,7 +3,7 @@ name = "phenoml"
 
 [tool.poetry]
 name = "phenoml"
-version = "0.0.1"
+version = "0.0.2"
 description = ""
 readme = "README.md"
 authors = []
@@ -31,7 +31,7 @@ packages = [
 ]
 
 [project.urls]
-Repository = 'https://github.com/fern-demo/phenoml-python-sdk'
+Repository = 'https://github.com/phenoml/phenoml-python-sdk'
 
 [tool.poetry.dependencies]
 python = "^3.8"

{phenoml-0.0.1 → phenoml-0.0.2}/src/phenoml/agent/__init__.py

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

{phenoml-0.0.1 → phenoml-0.0.2}/src/phenoml/agent/client.py

@@ -7,12 +7,13 @@ 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.agent_update_request_provider import AgentUpdateRequestProvider
 from .types.chat_fhir_client_config import ChatFhirClientConfig
 from .types.json_patch import JsonPatch
 
@@ -43,8 +44,9 @@ class AgentClient:
         prompts: typing.Sequence[str],
         is_active: bool,
         description: typing.Optional[str] = OMIT,
+        tools: typing.Optional[typing.Sequence[str]] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        provider: typing.Optional[AgentCreateRequestProvider] = OMIT,
+        provider: typing.Optional[AgentProvider] = OMIT,
         meta: typing.Optional[AgentFhirConfig] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentResponse:
@@ -65,10 +67,13 @@ class AgentClient:
         description : typing.Optional[str]
             Agent description
 
+        tools : typing.Optional[typing.Sequence[str]]
+            Array of MCP server tool IDs to use for this agent
+
         tags : typing.Optional[typing.Sequence[str]]
             Tags for categorizing the agent
 
-        provider : typing.Optional[AgentCreateRequestProvider]
+        provider : typing.Optional[AgentProvider]
             FHIR provider type - can be a single provider or array of providers
 
         meta : typing.Optional[AgentFhirConfig]
@@ -99,6 +104,7 @@ class AgentClient:
             prompts=prompts,
             is_active=is_active,
             description=description,
+            tools=tools,
             tags=tags,
             provider=provider,
             meta=meta,
@@ -182,9 +188,10 @@ class AgentClient:
         name: typing.Optional[str] = OMIT,
         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[AgentUpdateRequestProvider] = OMIT,
+        provider: typing.Optional[AgentProvider] = OMIT,
         meta: typing.Optional[AgentFhirConfig] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentResponse:
@@ -205,13 +212,16 @@ class AgentClient:
         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[AgentUpdateRequestProvider]
+        provider : typing.Optional[AgentProvider]
             FHIR provider type - can be a single provider or array of providers
 
         meta : typing.Optional[AgentFhirConfig]
@@ -240,6 +250,7 @@ class AgentClient:
             name=name,
             description=description,
             prompts=prompts,
+            tools=tools,
             is_active=is_active,
             tags=tags,
             provider=provider,
@@ -391,6 +402,60 @@ class AgentClient:
         )
         return _response.data
 
+    def get_chat_messages(
+        self,
+        *,
+        chat_session_id: str,
+        num_messages: typing.Optional[int] = None,
+        role: typing.Optional[str] = None,
+        order: typing.Optional[AgentGetChatMessagesRequestOrder] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AgentGetChatMessagesResponse:
+        """
+        Retrieves a list of chat messages for a given chat session
+
+        Parameters
+        ----------
+        chat_session_id : str
+            Chat session ID
+
+        num_messages : typing.Optional[int]
+            Number of messages to return
+
+        role : typing.Optional[str]
+            Filter by role
+
+        order : typing.Optional[AgentGetChatMessagesRequestOrder]
+            Order of messages
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AgentGetChatMessagesResponse
+            Chat messages retrieved successfully
+
+        Examples
+        --------
+        from phenoml import phenoml
+
+        client = phenoml(
+            token="YOUR_TOKEN",
+        )
+        client.agent.get_chat_messages(
+            chat_session_id="chat_session_id",
+        )
+        """
+        _response = self._raw_client.get_chat_messages(
+            chat_session_id=chat_session_id,
+            num_messages=num_messages,
+            role=role,
+            order=order,
+            request_options=request_options,
+        )
+        return _response.data
+
 
 class AsyncAgentClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -415,8 +480,9 @@ class AsyncAgentClient:
         prompts: typing.Sequence[str],
         is_active: bool,
         description: typing.Optional[str] = OMIT,
+        tools: typing.Optional[typing.Sequence[str]] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        provider: typing.Optional[AgentCreateRequestProvider] = OMIT,
+        provider: typing.Optional[AgentProvider] = OMIT,
         meta: typing.Optional[AgentFhirConfig] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentResponse:
@@ -437,10 +503,13 @@ class AsyncAgentClient:
         description : typing.Optional[str]
             Agent description
 
+        tools : typing.Optional[typing.Sequence[str]]
+            Array of MCP server tool IDs to use for this agent
+
         tags : typing.Optional[typing.Sequence[str]]
             Tags for categorizing the agent
 
-        provider : typing.Optional[AgentCreateRequestProvider]
+        provider : typing.Optional[AgentProvider]
             FHIR provider type - can be a single provider or array of providers
 
         meta : typing.Optional[AgentFhirConfig]
@@ -479,6 +548,7 @@ class AsyncAgentClient:
             prompts=prompts,
             is_active=is_active,
             description=description,
+            tools=tools,
             tags=tags,
             provider=provider,
             meta=meta,
@@ -578,9 +648,10 @@ class AsyncAgentClient:
         name: typing.Optional[str] = OMIT,
         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[AgentUpdateRequestProvider] = OMIT,
+        provider: typing.Optional[AgentProvider] = OMIT,
         meta: typing.Optional[AgentFhirConfig] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentResponse:
@@ -601,13 +672,16 @@ class AsyncAgentClient:
         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[AgentUpdateRequestProvider]
+        provider : typing.Optional[AgentProvider]
             FHIR provider type - can be a single provider or array of providers
 
         meta : typing.Optional[AgentFhirConfig]
@@ -644,6 +718,7 @@ class AsyncAgentClient:
             name=name,
             description=description,
             prompts=prompts,
+            tools=tools,
             is_active=is_active,
             tags=tags,
             provider=provider,
@@ -818,3 +893,65 @@ class AsyncAgentClient:
             request_options=request_options,
         )
         return _response.data
+
+    async def get_chat_messages(
+        self,
+        *,
+        chat_session_id: str,
+        num_messages: typing.Optional[int] = None,
+        role: typing.Optional[str] = None,
+        order: typing.Optional[AgentGetChatMessagesRequestOrder] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AgentGetChatMessagesResponse:
+        """
+        Retrieves a list of chat messages for a given chat session
+
+        Parameters
+        ----------
+        chat_session_id : str
+            Chat session ID
+
+        num_messages : typing.Optional[int]
+            Number of messages to return
+
+        role : typing.Optional[str]
+            Filter by role
+
+        order : typing.Optional[AgentGetChatMessagesRequestOrder]
+            Order of messages
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AgentGetChatMessagesResponse
+            Chat messages retrieved successfully
+
+        Examples
+        --------
+        import asyncio
+
+        from phenoml import Asyncphenoml
+
+        client = Asyncphenoml(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.agent.get_chat_messages(
+                chat_session_id="chat_session_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_chat_messages(
+            chat_session_id=chat_session_id,
+            num_messages=num_messages,
+            role=role,
+            order=order,
+            request_options=request_options,
+        )
+        return _response.data