feat(api): update via SDK Studio

Author: stainless-app[bot]

api.md

@@ -91,13 +91,13 @@ Methods:
 Types:
 
 ```python
-from contextual.types.applications import QueryFeedbackResponse, QueryStartResponse
+from contextual.types.applications import QueryResponse, QueryFeedbackResponse
 ```
 
 Methods:
 
 - <code title="post /applications/{application_id}/feedback">client.applications.query.<a href="./src/contextual/resources/applications/query.py">feedback</a>(application_id, \*\*<a href="src/contextual/types/applications/query_feedback_params.py">params</a>) -> <a href="./src/contextual/types/applications/query_feedback_response.py">object</a></code>
-- <code title="post /applications/{application_id}/query">client.applications.query.<a href="./src/contextual/resources/applications/query.py">start</a>(application_id, \*\*<a href="src/contextual/types/applications/query_start_params.py">params</a>) -> <a href="./src/contextual/types/applications/query_start_response.py">QueryStartResponse</a></code>
+- <code title="post /applications/{application_id}/query">client.applications.query.<a href="./src/contextual/resources/applications/query.py">start</a>(application_id, \*\*<a href="src/contextual/types/applications/query_start_params.py">params</a>) -> <a href="./src/contextual/types/applications/query_response.py">QueryResponse</a></code>
 
 ## Evaluate
 

src/contextual/resources/applications/query.py

@@ -22,7 +22,7 @@
 )
 from ..._base_client import make_request_options
 from ...types.applications import query_start_params, query_feedback_params
-from ...types.applications.query_start_response import QueryStartResponse
+from ...types.applications.query_response import QueryResponse
 
 __all__ = ["QueryResource", "AsyncQueryResource"]
 
@@ -129,7 +129,7 @@ def start(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> QueryStartResponse:
+    ) -> QueryResponse:
         """
         Start a conversation with an `Application` and receive its generated response,
         along with relevant retrieved data and attributions.
@@ -177,7 +177,7 @@ def start(
                 timeout=timeout,
                 query=maybe_transform({"retrievals_only": retrievals_only}, query_start_params.QueryStartParams),
             ),
-            cast_to=QueryStartResponse,
+            cast_to=QueryResponse,
         )
 
 
@@ -283,7 +283,7 @@ async def start(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> QueryStartResponse:
+    ) -> QueryResponse:
         """
         Start a conversation with an `Application` and receive its generated response,
         along with relevant retrieved data and attributions.
@@ -333,7 +333,7 @@ async def start(
                     {"retrievals_only": retrievals_only}, query_start_params.QueryStartParams
                 ),
             ),
-            cast_to=QueryStartResponse,
+            cast_to=QueryResponse,
         )
 
 

src/contextual/types/applications/__init__.py

@@ -3,11 +3,11 @@
 from __future__ import annotations
 
 from .tune_response import TuneResponse as TuneResponse
+from .query_response import QueryResponse as QueryResponse
 from .query_start_params import QueryStartParams as QueryStartParams
 from .tune_create_params import TuneCreateParams as TuneCreateParams
 from .dataset_list_params import DatasetListParams as DatasetListParams
 from .get_dataset_response import GetDatasetResponse as GetDatasetResponse
-from .query_start_response import QueryStartResponse as QueryStartResponse
 from .dataset_create_params import DatasetCreateParams as DatasetCreateParams
 from .dataset_update_params import DatasetUpdateParams as DatasetUpdateParams
 from .list_dataset_response import ListDatasetResponse as ListDatasetResponse

src/contextual/types/applications/query_start_response.py renamed to src/contextual/types/applications/query_response.py

@@ -5,7 +5,7 @@
 
 from ..._models import BaseModel
 
-__all__ = ["QueryStartResponse", "RetrievalContent", "Attribution", "Message"]
+__all__ = ["QueryResponse", "RetrievalContent", "Attribution", "Message"]
 
 
 class RetrievalContent(BaseModel):
@@ -62,7 +62,7 @@ class Message(BaseModel):
     """Role of sender"""
 
 
-class QueryStartResponse(BaseModel):
+class QueryResponse(BaseModel):
     conversation_id: str
     """A unique identifier for the conversation.
 

tests/api_resources/applications/test_query.py

@@ -9,7 +9,7 @@
 
 from contextual import ContextualAI, AsyncContextualAI
 from tests.utils import assert_matches_type
-from contextual.types.applications import QueryStartResponse
+from contextual.types.applications import QueryResponse
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
@@ -85,7 +85,7 @@ def test_method_start(self, client: ContextualAI) -> None:
                 }
             ],
         )
-        assert_matches_type(QueryStartResponse, query, path=["response"])
+        assert_matches_type(QueryResponse, query, path=["response"])
 
     @parametrize
     def test_method_start_with_all_params(self, client: ContextualAI) -> None:
@@ -102,7 +102,7 @@ def test_method_start_with_all_params(self, client: ContextualAI) -> None:
             model_id="model_id",
             stream=True,
         )
-        assert_matches_type(QueryStartResponse, query, path=["response"])
+        assert_matches_type(QueryResponse, query, path=["response"])
 
     @parametrize
     def test_raw_response_start(self, client: ContextualAI) -> None:
@@ -119,7 +119,7 @@ def test_raw_response_start(self, client: ContextualAI) -> None:
         assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
         query = response.parse()
-        assert_matches_type(QueryStartResponse, query, path=["response"])
+        assert_matches_type(QueryResponse, query, path=["response"])
 
     @parametrize
     def test_streaming_response_start(self, client: ContextualAI) -> None:
@@ -136,7 +136,7 @@ def test_streaming_response_start(self, client: ContextualAI) -> None:
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             query = response.parse()
-            assert_matches_type(QueryStartResponse, query, path=["response"])
+            assert_matches_type(QueryResponse, query, path=["response"])
 
         assert cast(Any, response.is_closed) is True
 
@@ -225,7 +225,7 @@ async def test_method_start(self, async_client: AsyncContextualAI) -> None:
                 }
             ],
         )
-        assert_matches_type(QueryStartResponse, query, path=["response"])
+        assert_matches_type(QueryResponse, query, path=["response"])
 
     @parametrize
     async def test_method_start_with_all_params(self, async_client: AsyncContextualAI) -> None:
@@ -242,7 +242,7 @@ async def test_method_start_with_all_params(self, async_client: AsyncContextualA
             model_id="model_id",
             stream=True,
         )
-        assert_matches_type(QueryStartResponse, query, path=["response"])
+        assert_matches_type(QueryResponse, query, path=["response"])
 
     @parametrize
     async def test_raw_response_start(self, async_client: AsyncContextualAI) -> None:
@@ -259,7 +259,7 @@ async def test_raw_response_start(self, async_client: AsyncContextualAI) -> None
         assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
         query = await response.parse()
-        assert_matches_type(QueryStartResponse, query, path=["response"])
+        assert_matches_type(QueryResponse, query, path=["response"])
 
     @parametrize
     async def test_streaming_response_start(self, async_client: AsyncContextualAI) -> None:
@@ -276,7 +276,7 @@ async def test_streaming_response_start(self, async_client: AsyncContextualAI) -
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             query = await response.parse()
-            assert_matches_type(QueryStartResponse, query, path=["response"])
+            assert_matches_type(QueryResponse, query, path=["response"])
 
         assert cast(Any, response.is_closed) is True