worqhat 3.2.0__py3-none-any.whl → 3.4.0__py3-none-any.whl

worqhat/resources/flows.py

@@ -2,15 +2,15 @@
 
 from __future__ import annotations
 
-from typing import Union
+from typing import Dict, Union, Mapping, cast
 from datetime import date
 from typing_extensions import Literal
 
 import httpx
 
-from ..types import flow_retrieve_metrics_params
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from .._utils import maybe_transform, async_maybe_transform
+from ..types import flow_get_metrics_params, flow_trigger_with_file_params, flow_trigger_with_payload_params
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven, FileTypes
+from .._utils import extract_files, maybe_transform, deepcopy_minimal, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (
@@ -20,7 +20,9 @@ from .._response import (
     async_to_streamed_response_wrapper,
 )
 from .._base_client import make_request_options
-from ..types.flow_retrieve_metrics_response import FlowRetrieveMetricsResponse
+from ..types.flow_get_metrics_response import FlowGetMetricsResponse
+from ..types.flow_trigger_with_file_response import FlowTriggerWithFileResponse
+from ..types.flow_trigger_with_payload_response import FlowTriggerWithPayloadResponse
 
 __all__ = ["FlowsResource", "AsyncFlowsResource"]
 
@@ -45,7 +47,7 @@ class FlowsResource(SyncAPIResource):
         """
         return FlowsResourceWithStreamingResponse(self)
 
-    def retrieve_metrics(
+    def get_metrics(
         self,
         *,
         end_date: Union[str, date] | NotGiven = NOT_GIVEN,
@@ -58,7 +60,7 @@ class FlowsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> FlowRetrieveMetricsResponse:
+    ) -> FlowGetMetricsResponse:
         """Get metrics for workflows within a specified date range.
 
         This endpoint provides
@@ -99,10 +101,104 @@ class FlowsResource(SyncAPIResource):
                         "status": status,
                         "user_id": user_id,
                     },
-                    flow_retrieve_metrics_params.FlowRetrieveMetricsParams,
+                    flow_get_metrics_params.FlowGetMetricsParams,
                 ),
             ),
-            cast_to=FlowRetrieveMetricsResponse,
+            cast_to=FlowGetMetricsResponse,
+        )
+
+    def trigger_with_file(
+        self,
+        flow_id: str,
+        *,
+        file: FileTypes | NotGiven = NOT_GIVEN,
+        url: str | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> FlowTriggerWithFileResponse:
+        """Trigger a workflow by its ID with either a file upload or a URL to a file.
+
+        This
+        endpoint accepts multipart/form-data with either a file field or a URL field.
+        When a URL is provided, the server will download the file and process it as if
+        it were directly uploaded. The workflow ID must be specified in the URL path.
+
+        Args:
+          file: File to upload and process
+
+          url: URL to a file to download and process
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not flow_id:
+            raise ValueError(f"Expected a non-empty value for `flow_id` but received {flow_id!r}")
+        body = deepcopy_minimal(
+            {
+                "file": file,
+                "url": url,
+            }
+        )
+        files = extract_files(cast(Mapping[str, object], body), paths=[["file"]])
+        # It should be noted that the actual Content-Type header that will be
+        # sent to the server will contain a `boundary` parameter, e.g.
+        # multipart/form-data; boundary=---abc--
+        extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
+        return self._post(
+            f"/flows/file/{flow_id}",
+            body=maybe_transform(body, flow_trigger_with_file_params.FlowTriggerWithFileParams),
+            files=files,
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=FlowTriggerWithFileResponse,
+        )
+
+    def trigger_with_payload(
+        self,
+        flow_id: str,
+        *,
+        body: Dict[str, object],
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> FlowTriggerWithPayloadResponse:
+        """Trigger a workflow by its ID with a JSON payload.
+
+        This endpoint accepts any
+        valid JSON object as the request body and forwards it to the workflow. The
+        workflow ID must be specified in the URL path.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not flow_id:
+            raise ValueError(f"Expected a non-empty value for `flow_id` but received {flow_id!r}")
+        return self._post(
+            f"/flows/trigger/{flow_id}",
+            body=maybe_transform(body, flow_trigger_with_payload_params.FlowTriggerWithPayloadParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=FlowTriggerWithPayloadResponse,
         )
 
 
@@ -126,7 +222,7 @@ class AsyncFlowsResource(AsyncAPIResource):
         """
         return AsyncFlowsResourceWithStreamingResponse(self)
 
-    async def retrieve_metrics(
+    async def get_metrics(
         self,
         *,
         end_date: Union[str, date] | NotGiven = NOT_GIVEN,
@@ -139,7 +235,7 @@ class AsyncFlowsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> FlowRetrieveMetricsResponse:
+    ) -> FlowGetMetricsResponse:
         """Get metrics for workflows within a specified date range.
 
         This endpoint provides
@@ -180,10 +276,104 @@ class AsyncFlowsResource(AsyncAPIResource):
                         "status": status,
                         "user_id": user_id,
                     },
-                    flow_retrieve_metrics_params.FlowRetrieveMetricsParams,
+                    flow_get_metrics_params.FlowGetMetricsParams,
                 ),
             ),
-            cast_to=FlowRetrieveMetricsResponse,
+            cast_to=FlowGetMetricsResponse,
+        )
+
+    async def trigger_with_file(
+        self,
+        flow_id: str,
+        *,
+        file: FileTypes | NotGiven = NOT_GIVEN,
+        url: str | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> FlowTriggerWithFileResponse:
+        """Trigger a workflow by its ID with either a file upload or a URL to a file.
+
+        This
+        endpoint accepts multipart/form-data with either a file field or a URL field.
+        When a URL is provided, the server will download the file and process it as if
+        it were directly uploaded. The workflow ID must be specified in the URL path.
+
+        Args:
+          file: File to upload and process
+
+          url: URL to a file to download and process
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not flow_id:
+            raise ValueError(f"Expected a non-empty value for `flow_id` but received {flow_id!r}")
+        body = deepcopy_minimal(
+            {
+                "file": file,
+                "url": url,
+            }
+        )
+        files = extract_files(cast(Mapping[str, object], body), paths=[["file"]])
+        # It should be noted that the actual Content-Type header that will be
+        # sent to the server will contain a `boundary` parameter, e.g.
+        # multipart/form-data; boundary=---abc--
+        extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
+        return await self._post(
+            f"/flows/file/{flow_id}",
+            body=await async_maybe_transform(body, flow_trigger_with_file_params.FlowTriggerWithFileParams),
+            files=files,
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=FlowTriggerWithFileResponse,
+        )
+
+    async def trigger_with_payload(
+        self,
+        flow_id: str,
+        *,
+        body: Dict[str, object],
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> FlowTriggerWithPayloadResponse:
+        """Trigger a workflow by its ID with a JSON payload.
+
+        This endpoint accepts any
+        valid JSON object as the request body and forwards it to the workflow. The
+        workflow ID must be specified in the URL path.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not flow_id:
+            raise ValueError(f"Expected a non-empty value for `flow_id` but received {flow_id!r}")
+        return await self._post(
+            f"/flows/trigger/{flow_id}",
+            body=await async_maybe_transform(body, flow_trigger_with_payload_params.FlowTriggerWithPayloadParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=FlowTriggerWithPayloadResponse,
         )
 
 
@@ -191,8 +381,14 @@ class FlowsResourceWithRawResponse:
     def __init__(self, flows: FlowsResource) -> None:
         self._flows = flows
 
-        self.retrieve_metrics = to_raw_response_wrapper(
-            flows.retrieve_metrics,
+        self.get_metrics = to_raw_response_wrapper(
+            flows.get_metrics,
+        )
+        self.trigger_with_file = to_raw_response_wrapper(
+            flows.trigger_with_file,
+        )
+        self.trigger_with_payload = to_raw_response_wrapper(
+            flows.trigger_with_payload,
         )
 
 
@@ -200,8 +396,14 @@ class AsyncFlowsResourceWithRawResponse:
     def __init__(self, flows: AsyncFlowsResource) -> None:
         self._flows = flows
 
-        self.retrieve_metrics = async_to_raw_response_wrapper(
-            flows.retrieve_metrics,
+        self.get_metrics = async_to_raw_response_wrapper(
+            flows.get_metrics,
+        )
+        self.trigger_with_file = async_to_raw_response_wrapper(
+            flows.trigger_with_file,
+        )
+        self.trigger_with_payload = async_to_raw_response_wrapper(
+            flows.trigger_with_payload,
         )
 
 
@@ -209,8 +411,14 @@ class FlowsResourceWithStreamingResponse:
     def __init__(self, flows: FlowsResource) -> None:
         self._flows = flows
 
-        self.retrieve_metrics = to_streamed_response_wrapper(
-            flows.retrieve_metrics,
+        self.get_metrics = to_streamed_response_wrapper(
+            flows.get_metrics,
+        )
+        self.trigger_with_file = to_streamed_response_wrapper(
+            flows.trigger_with_file,
+        )
+        self.trigger_with_payload = to_streamed_response_wrapper(
+            flows.trigger_with_payload,
         )
 
 
@@ -218,6 +426,12 @@ class AsyncFlowsResourceWithStreamingResponse:
     def __init__(self, flows: AsyncFlowsResource) -> None:
         self._flows = flows
 
-        self.retrieve_metrics = async_to_streamed_response_wrapper(
-            flows.retrieve_metrics,
+        self.get_metrics = async_to_streamed_response_wrapper(
+            flows.get_metrics,
+        )
+        self.trigger_with_file = async_to_streamed_response_wrapper(
+            flows.trigger_with_file,
+        )
+        self.trigger_with_payload = async_to_streamed_response_wrapper(
+            flows.trigger_with_payload,
         )

worqhat/types/__init__.py

@@ -3,6 +3,20 @@
 from __future__ import annotations
 
 from .health_check_response import HealthCheckResponse as HealthCheckResponse
-from .flow_retrieve_metrics_params import FlowRetrieveMetricsParams as FlowRetrieveMetricsParams
-from .retrieve_server_info_response import RetrieveServerInfoResponse as RetrieveServerInfoResponse
-from .flow_retrieve_metrics_response import FlowRetrieveMetricsResponse as FlowRetrieveMetricsResponse
+from .db_execute_query_params import DBExecuteQueryParams as DBExecuteQueryParams
+from .db_insert_record_params import DBInsertRecordParams as DBInsertRecordParams
+from .flow_get_metrics_params import FlowGetMetricsParams as FlowGetMetricsParams
+from .db_delete_records_params import DBDeleteRecordsParams as DBDeleteRecordsParams
+from .db_update_records_params import DBUpdateRecordsParams as DBUpdateRecordsParams
+from .get_server_info_response import GetServerInfoResponse as GetServerInfoResponse
+from .db_execute_query_response import DBExecuteQueryResponse as DBExecuteQueryResponse
+from .db_insert_record_response import DBInsertRecordResponse as DBInsertRecordResponse
+from .flow_get_metrics_response import FlowGetMetricsResponse as FlowGetMetricsResponse
+from .db_delete_records_response import DBDeleteRecordsResponse as DBDeleteRecordsResponse
+from .db_process_nl_query_params import DBProcessNlQueryParams as DBProcessNlQueryParams
+from .db_update_records_response import DBUpdateRecordsResponse as DBUpdateRecordsResponse
+from .db_process_nl_query_response import DBProcessNlQueryResponse as DBProcessNlQueryResponse
+from .flow_trigger_with_file_params import FlowTriggerWithFileParams as FlowTriggerWithFileParams
+from .flow_trigger_with_file_response import FlowTriggerWithFileResponse as FlowTriggerWithFileResponse
+from .flow_trigger_with_payload_params import FlowTriggerWithPayloadParams as FlowTriggerWithPayloadParams
+from .flow_trigger_with_payload_response import FlowTriggerWithPayloadResponse as FlowTriggerWithPayloadResponse

worqhat/types/db_delete_records_params.py

@@ -0,0 +1,15 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+__all__ = ["DBDeleteRecordsParams"]
+
+
+class DBDeleteRecordsParams(TypedDict, total=False):
+    table: Required[str]
+    """Table name to delete from"""
+
+    where: Required[object]
+    """Where conditions"""

worqhat/types/db_delete_records_response.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from .._models import BaseModel
+
+__all__ = ["DBDeleteRecordsResponse"]
+
+
+class DBDeleteRecordsResponse(BaseModel):
+    count: Optional[int] = None
+    """Number of records deleted"""
+
+    data: Optional[List[object]] = None
+
+    message: Optional[str] = None
+
+    success: Optional[bool] = None

worqhat/types/db_execute_query_params.py

@@ -0,0 +1,12 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+__all__ = ["DBExecuteQueryParams"]
+
+
+class DBExecuteQueryParams(TypedDict, total=False):
+    query: Required[str]
+    """SQL query to execute"""

worqhat/types/db_execute_query_response.py

@@ -0,0 +1,21 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from pydantic import Field as FieldInfo
+
+from .._models import BaseModel
+
+__all__ = ["DBExecuteQueryResponse"]
+
+
+class DBExecuteQueryResponse(BaseModel):
+    data: Optional[List[object]] = None
+
+    execution_time: Optional[int] = FieldInfo(alias="executionTime", default=None)
+    """Query execution time in milliseconds"""
+
+    query: Optional[str] = None
+    """The executed SQL query"""
+
+    success: Optional[bool] = None

worqhat/types/db_insert_record_params.py

@@ -0,0 +1,15 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+__all__ = ["DBInsertRecordParams"]
+
+
+class DBInsertRecordParams(TypedDict, total=False):
+    data: Required[object]
+    """Data to insert"""
+
+    table: Required[str]
+    """Table name to insert into"""

worqhat/types/db_insert_record_response.py

@@ -0,0 +1,15 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from .._models import BaseModel
+
+__all__ = ["DBInsertRecordResponse"]
+
+
+class DBInsertRecordResponse(BaseModel):
+    data: Optional[object] = None
+
+    message: Optional[str] = None
+
+    success: Optional[bool] = None

worqhat/types/db_process_nl_query_params.py

@@ -0,0 +1,15 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+__all__ = ["DBProcessNlQueryParams"]
+
+
+class DBProcessNlQueryParams(TypedDict, total=False):
+    question: Required[str]
+    """Natural language question"""
+
+    context: object
+    """Additional context for the query"""

worqhat/types/db_process_nl_query_response.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from .._models import BaseModel
+
+__all__ = ["DBProcessNlQueryResponse"]
+
+
+class DBProcessNlQueryResponse(BaseModel):
+    data: Optional[List[object]] = None
+
+    message: Optional[str] = None
+
+    sql: Optional[str] = None
+    """The generated SQL query"""
+
+    success: Optional[bool] = None

worqhat/types/db_update_records_params.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+__all__ = ["DBUpdateRecordsParams"]
+
+
+class DBUpdateRecordsParams(TypedDict, total=False):
+    data: Required[object]
+    """Data to update"""
+
+    table: Required[str]
+    """Table name to update"""
+
+    where: Required[object]
+    """Where conditions"""

worqhat/types/db_update_records_response.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from .._models import BaseModel
+
+__all__ = ["DBUpdateRecordsResponse"]
+
+
+class DBUpdateRecordsResponse(BaseModel):
+    count: Optional[int] = None
+    """Number of records updated"""
+
+    data: Optional[List[object]] = None
+
+    message: Optional[str] = None
+
+    success: Optional[bool] = None

worqhat/types/{flow_retrieve_metrics_params.py → flow_get_metrics_params.py}

@@ -8,10 +8,10 @@ from typing_extensions import Literal, Annotated, TypedDict
 
 from .._utils import PropertyInfo
 
-__all__ = ["FlowRetrieveMetricsParams"]
+__all__ = ["FlowGetMetricsParams"]
 
 
-class FlowRetrieveMetricsParams(TypedDict, total=False):
+class FlowGetMetricsParams(TypedDict, total=False):
     end_date: Annotated[Union[str, date], PropertyInfo(format="iso8601")]
     """End date for filtering (YYYY-MM-DD format)"""
 

worqhat/types/{flow_retrieve_metrics_response.py → flow_get_metrics_response.py}

@@ -6,7 +6,7 @@ from typing_extensions import Literal
 
 from .._models import BaseModel
 
-__all__ = ["FlowRetrieveMetricsResponse", "Metrics", "MetricsMetricsByUser", "Workflow"]
+__all__ = ["FlowGetMetricsResponse", "Metrics", "MetricsMetricsByUser", "Workflow"]
 
 
 class MetricsMetricsByUser(BaseModel):
@@ -49,7 +49,7 @@ class Workflow(BaseModel):
     workflow_id: Optional[str] = None
 
 
-class FlowRetrieveMetricsResponse(BaseModel):
+class FlowGetMetricsResponse(BaseModel):
     metrics: Optional[Metrics] = None
 
     workflows: Optional[List[Workflow]] = None

worqhat/types/flow_trigger_with_file_params.py

@@ -0,0 +1,17 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import TypedDict
+
+from .._types import FileTypes
+
+__all__ = ["FlowTriggerWithFileParams"]
+
+
+class FlowTriggerWithFileParams(TypedDict, total=False):
+    file: FileTypes
+    """File to upload and process"""
+
+    url: str
+    """URL to a file to download and process"""

worqhat/types/flow_trigger_with_file_response.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, Optional
+from datetime import datetime
+
+from .._models import BaseModel
+
+__all__ = ["FlowTriggerWithFileResponse"]
+
+
+class FlowTriggerWithFileResponse(BaseModel):
+    data: Optional[Dict[str, object]] = None
+
+    message: Optional[str] = None
+
+    success: Optional[bool] = None
+
+    timestamp: Optional[datetime] = None

worqhat/types/flow_trigger_with_payload_params.py

@@ -0,0 +1,12 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict
+from typing_extensions import Required, TypedDict
+
+__all__ = ["FlowTriggerWithPayloadParams"]
+
+
+class FlowTriggerWithPayloadParams(TypedDict, total=False):
+    body: Required[Dict[str, object]]

worqhat/types/flow_trigger_with_payload_response.py

@@ -0,0 +1,20 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, Optional
+from datetime import datetime
+
+from .._models import BaseModel
+
+__all__ = ["FlowTriggerWithPayloadResponse"]
+
+
+class FlowTriggerWithPayloadResponse(BaseModel):
+    analytics_id: Optional[str] = None
+
+    data: Optional[Dict[str, object]] = None
+
+    message: Optional[str] = None
+
+    success: Optional[bool] = None
+
+    timestamp: Optional[datetime] = None

worqhat/types/{retrieve_server_info_response.py → get_server_info_response.py}

@@ -4,10 +4,10 @@ from typing import Optional
 
 from .._models import BaseModel
 
-__all__ = ["RetrieveServerInfoResponse"]
+__all__ = ["GetServerInfoResponse"]
 
 
-class RetrieveServerInfoResponse(BaseModel):
+class GetServerInfoResponse(BaseModel):
     environment: Optional[str] = None
 
     name: Optional[str] = None