feat(api): model groups (#49)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
-configured_endpoints: 81
+configured_endpoints: 82
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-4bee800ad604492e6ee5776c10e1668037755aeaae6ff6faa2d09a0c3763eeee.yml

api.md

@@ -11,7 +11,13 @@ from asktable.types import Policy
 Types:
 
 ```python
-from asktable.types.sys import APIKey, Project, ProjectDeleteResponse
+from asktable.types.sys import (
+    APIKey,
+    ModelGroup,
+    Project,
+    ProjectDeleteResponse,
+    ProjectModelGroupsResponse,
+)
 ```
 
 Methods:
@@ -21,6 +27,7 @@ Methods:
 - <code title="patch /sys/projects/{project_id}">client.sys.projects.<a href="./src/asktable/resources/sys/projects/projects.py">update</a>(project_id, \*\*<a href="src/asktable/types/sys/project_update_params.py">params</a>) -> <a href="./src/asktable/types/sys/project.py">Project</a></code>
 - <code title="get /sys/projects">client.sys.projects.<a href="./src/asktable/resources/sys/projects/projects.py">list</a>(\*\*<a href="src/asktable/types/sys/project_list_params.py">params</a>) -> <a href="./src/asktable/types/sys/project.py">SyncPage[Project]</a></code>
 - <code title="delete /sys/projects/{project_id}">client.sys.projects.<a href="./src/asktable/resources/sys/projects/projects.py">delete</a>(project_id) -> <a href="./src/asktable/types/sys/project_delete_response.py">object</a></code>
+- <code title="get /sys/projects/model-groups">client.sys.projects.<a href="./src/asktable/resources/sys/projects/projects.py">model_groups</a>() -> <a href="./src/asktable/types/sys/project_model_groups_response.py">ProjectModelGroupsResponse</a></code>
 
 ### APIKeys
 

src/asktable/resources/sys/projects/projects.py

@@ -31,6 +31,7 @@
 from ....pagination import SyncPage, AsyncPage
 from ...._base_client import AsyncPaginator, make_request_options
 from ....types.sys.project import Project
+from ....types.sys.project_model_groups_response import ProjectModelGroupsResponse
 
 __all__ = ["ProjectsResource", "AsyncProjectsResource"]
 
@@ -260,6 +261,25 @@ def delete(
             cast_to=object,
         )
 
+    def model_groups(
+        self,
+        *,
+        # 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,
+    ) -> ProjectModelGroupsResponse:
+        """Get Llm Model Groups"""
+        return self._get(
+            "/sys/projects/model-groups",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ProjectModelGroupsResponse,
+        )
+
 
 class AsyncProjectsResource(AsyncAPIResource):
     @cached_property
@@ -486,6 +506,25 @@ async def delete(
             cast_to=object,
         )
 
+    async def model_groups(
+        self,
+        *,
+        # 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,
+    ) -> ProjectModelGroupsResponse:
+        """Get Llm Model Groups"""
+        return await self._get(
+            "/sys/projects/model-groups",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ProjectModelGroupsResponse,
+        )
+
 
 class ProjectsResourceWithRawResponse:
     def __init__(self, projects: ProjectsResource) -> None:
@@ -506,6 +545,9 @@ def __init__(self, projects: ProjectsResource) -> None:
         self.delete = to_raw_response_wrapper(
             projects.delete,
         )
+        self.model_groups = to_raw_response_wrapper(
+            projects.model_groups,
+        )
 
     @cached_property
     def api_keys(self) -> APIKeysResourceWithRawResponse:
@@ -531,6 +573,9 @@ def __init__(self, projects: AsyncProjectsResource) -> None:
         self.delete = async_to_raw_response_wrapper(
             projects.delete,
         )
+        self.model_groups = async_to_raw_response_wrapper(
+            projects.model_groups,
+        )
 
     @cached_property
     def api_keys(self) -> AsyncAPIKeysResourceWithRawResponse:
@@ -556,6 +601,9 @@ def __init__(self, projects: ProjectsResource) -> None:
         self.delete = to_streamed_response_wrapper(
             projects.delete,
         )
+        self.model_groups = to_streamed_response_wrapper(
+            projects.model_groups,
+        )
 
     @cached_property
     def api_keys(self) -> APIKeysResourceWithStreamingResponse:
@@ -581,6 +629,9 @@ def __init__(self, projects: AsyncProjectsResource) -> None:
         self.delete = async_to_streamed_response_wrapper(
             projects.delete,
         )
+        self.model_groups = async_to_streamed_response_wrapper(
+            projects.model_groups,
+        )
 
     @cached_property
     def api_keys(self) -> AsyncAPIKeysResourceWithStreamingResponse:

src/asktable/types/sys/__init__.py

@@ -4,6 +4,8 @@
 
 from .api_key import APIKey as APIKey
 from .project import Project as Project
+from .model_group import ModelGroup as ModelGroup
 from .project_list_params import ProjectListParams as ProjectListParams
 from .project_create_params import ProjectCreateParams as ProjectCreateParams
 from .project_update_params import ProjectUpdateParams as ProjectUpdateParams
+from .project_model_groups_response import ProjectModelGroupsResponse as ProjectModelGroupsResponse

src/asktable/types/sys/model_group.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+
+from ..._models import BaseModel
+
+__all__ = ["ModelGroup"]
+
+
+class ModelGroup(BaseModel):
+    image_models: List[str]
+    """图像模型列表"""
+
+    name: str
+    """模型组名称"""
+
+    text_models: List[str]
+    """文本模型列表"""

src/asktable/types/sys/project_model_groups_response.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import TypeAlias
+
+from .model_group import ModelGroup
+
+__all__ = ["ProjectModelGroupsResponse"]
+
+ProjectModelGroupsResponse: TypeAlias = List[ModelGroup]

tests/api_resources/sys/test_projects.py

@@ -9,7 +9,10 @@
 
 from asktable import Asktable, AsyncAsktable
 from tests.utils import assert_matches_type
-from asktable.types.sys import Project
+from asktable.types.sys import (
+    Project,
+    ProjectModelGroupsResponse,
+)
 from asktable.pagination import SyncPage, AsyncPage
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
@@ -207,6 +210,31 @@ def test_path_params_delete(self, client: Asktable) -> None:
                 "",
             )
 
+    @parametrize
+    def test_method_model_groups(self, client: Asktable) -> None:
+        project = client.sys.projects.model_groups()
+        assert_matches_type(ProjectModelGroupsResponse, project, path=["response"])
+
+    @parametrize
+    def test_raw_response_model_groups(self, client: Asktable) -> None:
+        response = client.sys.projects.with_raw_response.model_groups()
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        project = response.parse()
+        assert_matches_type(ProjectModelGroupsResponse, project, path=["response"])
+
+    @parametrize
+    def test_streaming_response_model_groups(self, client: Asktable) -> None:
+        with client.sys.projects.with_streaming_response.model_groups() as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            project = response.parse()
+            assert_matches_type(ProjectModelGroupsResponse, project, path=["response"])
+
+        assert cast(Any, response.is_closed) is True
+
 
 class TestAsyncProjects:
     parametrize = pytest.mark.parametrize("async_client", [False, True], indirect=True, ids=["loose", "strict"])
@@ -399,3 +427,28 @@ async def test_path_params_delete(self, async_client: AsyncAsktable) -> None:
             await async_client.sys.projects.with_raw_response.delete(
                 "",
             )
+
+    @parametrize
+    async def test_method_model_groups(self, async_client: AsyncAsktable) -> None:
+        project = await async_client.sys.projects.model_groups()
+        assert_matches_type(ProjectModelGroupsResponse, project, path=["response"])
+
+    @parametrize
+    async def test_raw_response_model_groups(self, async_client: AsyncAsktable) -> None:
+        response = await async_client.sys.projects.with_raw_response.model_groups()
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        project = await response.parse()
+        assert_matches_type(ProjectModelGroupsResponse, project, path=["response"])
+
+    @parametrize
+    async def test_streaming_response_model_groups(self, async_client: AsyncAsktable) -> None:
+        async with async_client.sys.projects.with_streaming_response.model_groups() as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            project = await response.parse()
+            assert_matches_type(ProjectModelGroupsResponse, project, path=["response"])
+
+        assert cast(Any, response.is_closed) is True