phenoml 0.0.1__py3-none-any.whl → 0.0.5__py3-none-any.whl

phenoml/agent/prompts/raw_client.py

@@ -29,35 +29,86 @@ class RawPromptsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def list(self, *, request_options: typing.Optional[RequestOptions] = None) -> HttpResponse[PromptsListResponse]:
+    def create(
+        self,
+        *,
+        name: str,
+        content: str,
+        is_active: bool,
+        description: typing.Optional[str] = OMIT,
+        is_default: typing.Optional[bool] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[AgentPromptsResponse]:
         """
-        Retrieves a list of agent prompts belonging to the authenticated user
+        Creates a new agent prompt
 
         Parameters
         ----------
+        name : str
+            Prompt name
+
+        content : str
+            Prompt content
+
+        is_active : bool
+            Whether the prompt is active
+
+        description : typing.Optional[str]
+            Prompt description
+
+        is_default : typing.Optional[bool]
+            Whether this is a default prompt
+
+        tags : typing.Optional[typing.Sequence[str]]
+            Tags for categorizing the prompt
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        HttpResponse[PromptsListResponse]
-            Prompts retrieved successfully
+        HttpResponse[AgentPromptsResponse]
+            Prompt created successfully
         """
         _response = self._client_wrapper.httpx_client.request(
             "agent/prompts",
-            method="GET",
+            method="POST",
+            json={
+                "name": name,
+                "description": description,
+                "content": content,
+                "is_default": is_default,
+                "is_active": is_active,
+                "tags": tags,
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
+            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    PromptsListResponse,
+                    AgentPromptsResponse,
                     parse_obj_as(
-                        type_=PromptsListResponse,  # type: ignore
+                        type_=AgentPromptsResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
                 return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             if _response.status_code == 401:
                 raise UnauthorizedError(
                     headers=dict(_response.headers),
@@ -96,86 +147,35 @@ class RawPromptsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
-    def create(
-        self,
-        *,
-        name: str,
-        content: str,
-        is_active: bool,
-        description: typing.Optional[str] = OMIT,
-        is_default: typing.Optional[bool] = OMIT,
-        tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[AgentPromptsResponse]:
+    def list(self, *, request_options: typing.Optional[RequestOptions] = None) -> HttpResponse[PromptsListResponse]:
         """
-        Creates a new agent prompt
+        Retrieves a list of agent prompts belonging to the authenticated user
 
         Parameters
         ----------
-        name : str
-            Prompt name
-
-        content : str
-            Prompt content
-
-        is_active : bool
-            Whether the prompt is active
-
-        description : typing.Optional[str]
-            Prompt description
-
-        is_default : typing.Optional[bool]
-            Whether this is a default prompt
-
-        tags : typing.Optional[typing.Sequence[str]]
-            Tags for categorizing the prompt
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        HttpResponse[AgentPromptsResponse]
-            Prompt created successfully
+        HttpResponse[PromptsListResponse]
+            Prompts retrieved successfully
         """
         _response = self._client_wrapper.httpx_client.request(
-            "agent/prompts",
-            method="POST",
-            json={
-                "name": name,
-                "description": description,
-                "content": content,
-                "is_default": is_default,
-                "is_active": is_active,
-                "tags": tags,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+            "agent/prompts/list",
+            method="GET",
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    AgentPromptsResponse,
+                    PromptsListResponse,
                     parse_obj_as(
-                        type_=AgentPromptsResponse,  # type: ignore
+                        type_=PromptsListResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
                 return HttpResponse(response=_response, data=_data)
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
             if _response.status_code == 401:
                 raise UnauthorizedError(
                     headers=dict(_response.headers),
@@ -688,37 +688,86 @@ class AsyncRawPromptsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def list(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> AsyncHttpResponse[PromptsListResponse]:
+    async def create(
+        self,
+        *,
+        name: str,
+        content: str,
+        is_active: bool,
+        description: typing.Optional[str] = OMIT,
+        is_default: typing.Optional[bool] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[AgentPromptsResponse]:
         """
-        Retrieves a list of agent prompts belonging to the authenticated user
+        Creates a new agent prompt
 
         Parameters
         ----------
+        name : str
+            Prompt name
+
+        content : str
+            Prompt content
+
+        is_active : bool
+            Whether the prompt is active
+
+        description : typing.Optional[str]
+            Prompt description
+
+        is_default : typing.Optional[bool]
+            Whether this is a default prompt
+
+        tags : typing.Optional[typing.Sequence[str]]
+            Tags for categorizing the prompt
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AsyncHttpResponse[PromptsListResponse]
-            Prompts retrieved successfully
+        AsyncHttpResponse[AgentPromptsResponse]
+            Prompt created successfully
         """
         _response = await self._client_wrapper.httpx_client.request(
             "agent/prompts",
-            method="GET",
+            method="POST",
+            json={
+                "name": name,
+                "description": description,
+                "content": content,
+                "is_default": is_default,
+                "is_active": is_active,
+                "tags": tags,
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
+            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    PromptsListResponse,
+                    AgentPromptsResponse,
                     parse_obj_as(
-                        type_=PromptsListResponse,  # type: ignore
+                        type_=AgentPromptsResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
                 return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             if _response.status_code == 401:
                 raise UnauthorizedError(
                     headers=dict(_response.headers),
@@ -757,86 +806,37 @@ class AsyncRawPromptsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
-    async def create(
-        self,
-        *,
-        name: str,
-        content: str,
-        is_active: bool,
-        description: typing.Optional[str] = OMIT,
-        is_default: typing.Optional[bool] = OMIT,
-        tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[AgentPromptsResponse]:
+    async def list(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[PromptsListResponse]:
         """
-        Creates a new agent prompt
+        Retrieves a list of agent prompts belonging to the authenticated user
 
         Parameters
         ----------
-        name : str
-            Prompt name
-
-        content : str
-            Prompt content
-
-        is_active : bool
-            Whether the prompt is active
-
-        description : typing.Optional[str]
-            Prompt description
-
-        is_default : typing.Optional[bool]
-            Whether this is a default prompt
-
-        tags : typing.Optional[typing.Sequence[str]]
-            Tags for categorizing the prompt
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AsyncHttpResponse[AgentPromptsResponse]
-            Prompt created successfully
+        AsyncHttpResponse[PromptsListResponse]
+            Prompts retrieved successfully
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "agent/prompts",
-            method="POST",
-            json={
-                "name": name,
-                "description": description,
-                "content": content,
-                "is_default": is_default,
-                "is_active": is_active,
-                "tags": tags,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+            "agent/prompts/list",
+            method="GET",
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    AgentPromptsResponse,
+                    PromptsListResponse,
                     parse_obj_as(
-                        type_=AgentPromptsResponse,  # type: ignore
+                        type_=PromptsListResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
                 return AsyncHttpResponse(response=_response, data=_data)
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
             if _response.status_code == 401:
                 raise UnauthorizedError(
                     headers=dict(_response.headers),