2727from .._streaming import Stream , AsyncStream
2828from ..pagination import SyncCursorPage , AsyncCursorPage
2929from ..types .file import File
30+ from ..types .graph import Graph
3031from .._base_client import AsyncPaginator , make_request_options
3132from ..types .question import Question
32- from ..types .graph_list_response import GraphListResponse
3333from ..types .graph_create_response import GraphCreateResponse
3434from ..types .graph_delete_response import GraphDeleteResponse
3535from ..types .graph_update_response import GraphUpdateResponse
36- from ..types .graph_retrieve_response import GraphRetrieveResponse
3736from ..types .question_response_chunk import QuestionResponseChunk
3837from ..types .graph_remove_file_from_graph_response import GraphRemoveFileFromGraphResponse
3938
@@ -115,7 +114,7 @@ def retrieve(
115114 extra_query : Query | None = None ,
116115 extra_body : Body | None = None ,
117116 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
118- ) -> GraphRetrieveResponse :
117+ ) -> Graph :
119118 """
120119 Retrieve a Knowledge Graph.
121120
@@ -135,7 +134,7 @@ def retrieve(
135134 options = make_request_options (
136135 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
137136 ),
138- cast_to = GraphRetrieveResponse ,
137+ cast_to = Graph ,
139138 )
140139
141140 def update (
@@ -205,7 +204,7 @@ def list(
205204 extra_query : Query | None = None ,
206205 extra_body : Body | None = None ,
207206 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
208- ) -> SyncCursorPage [GraphListResponse ]:
207+ ) -> SyncCursorPage [Graph ]:
209208 """
210209 Retrieve a list of Knowledge Graphs.
211210
@@ -232,7 +231,7 @@ def list(
232231 """
233232 return self ._get_api_list (
234233 "/v1/graphs" ,
235- page = SyncCursorPage [GraphListResponse ],
234+ page = SyncCursorPage [Graph ],
236235 options = make_request_options (
237236 extra_headers = extra_headers ,
238237 extra_query = extra_query ,
@@ -248,7 +247,7 @@ def list(
248247 graph_list_params .GraphListParams ,
249248 ),
250249 ),
251- model = GraphListResponse ,
250+ model = Graph ,
252251 )
253252
254253 def delete (
@@ -620,7 +619,7 @@ async def retrieve(
620619 extra_query : Query | None = None ,
621620 extra_body : Body | None = None ,
622621 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
623- ) -> GraphRetrieveResponse :
622+ ) -> Graph :
624623 """
625624 Retrieve a Knowledge Graph.
626625
@@ -640,7 +639,7 @@ async def retrieve(
640639 options = make_request_options (
641640 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
642641 ),
643- cast_to = GraphRetrieveResponse ,
642+ cast_to = Graph ,
644643 )
645644
646645 async def update (
@@ -710,7 +709,7 @@ def list(
710709 extra_query : Query | None = None ,
711710 extra_body : Body | None = None ,
712711 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
713- ) -> AsyncPaginator [GraphListResponse , AsyncCursorPage [GraphListResponse ]]:
712+ ) -> AsyncPaginator [Graph , AsyncCursorPage [Graph ]]:
714713 """
715714 Retrieve a list of Knowledge Graphs.
716715
@@ -737,7 +736,7 @@ def list(
737736 """
738737 return self ._get_api_list (
739738 "/v1/graphs" ,
740- page = AsyncCursorPage [GraphListResponse ],
739+ page = AsyncCursorPage [Graph ],
741740 options = make_request_options (
742741 extra_headers = extra_headers ,
743742 extra_query = extra_query ,
@@ -753,7 +752,7 @@ def list(
753752 graph_list_params .GraphListParams ,
754753 ),
755754 ),
756- model = GraphListResponse ,
755+ model = Graph ,
757756 )
758757
759758 async def delete (
0 commit comments