graphlit-client 1.0.20260129001__tar.gz → 1.0.20260129002__tar.gz

{graphlit_client-1.0.20260129001/graphlit_client.egg-info → graphlit_client-1.0.20260129002}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphlit-client
-Version: 1.0.20260129001
+Version: 1.0.20260129002
 Summary: Graphlit API Python Client
 Home-page: https://github.com/graphlit/graphlit-client-python
 Author: Unstruk Data Inc.

{graphlit_client-1.0.20260129001 → graphlit_client-1.0.20260129002}/graphlit_api/add_contents_to_collections.py

@@ -1,7 +1,7 @@
 # Generated by ariadne-codegen
 # Source: ./documents
 
-from typing import List, Optional
+from typing import Optional
 
 from pydantic import Field
 
@@ -11,7 +11,7 @@ from .enums import CollectionTypes, EntityState
 
 class AddContentsToCollections(BaseModel):
     add_contents_to_collections: Optional[
-        List[Optional["AddContentsToCollectionsAddContentsToCollections"]]
+        list[Optional["AddContentsToCollectionsAddContentsToCollections"]]
     ] = Field(alias="addContentsToCollections")
 
 
@@ -21,7 +21,7 @@ class AddContentsToCollectionsAddContentsToCollections(BaseModel):
     state: EntityState
     type: Optional[CollectionTypes]
     contents: Optional[
-        List[Optional["AddContentsToCollectionsAddContentsToCollectionsContents"]]
+        list[Optional["AddContentsToCollectionsAddContentsToCollectionsContents"]]
     ]
 
 

{graphlit_client-1.0.20260129001 → graphlit_client-1.0.20260129002}/graphlit_api/ask_graphlit.py

@@ -1,7 +1,7 @@
 # Generated by ariadne-codegen
 # Source: ./documents
 
-from typing import Any, List, Optional
+from typing import Any, Optional
 
 from pydantic import Field
 
@@ -35,8 +35,8 @@ class AskGraphlitAskGraphlitMessage(BaseModel):
     role: ConversationRoleTypes
     author: Optional[str]
     message: Optional[str]
-    citations: Optional[List[Optional["AskGraphlitAskGraphlitMessageCitations"]]]
-    tool_calls: Optional[List[Optional["AskGraphlitAskGraphlitMessageToolCalls"]]] = (
+    citations: Optional[list[Optional["AskGraphlitAskGraphlitMessageCitations"]]]
+    tool_calls: Optional[list[Optional["AskGraphlitAskGraphlitMessageToolCalls"]]] = (
         Field(alias="toolCalls")
     )
     tokens: Optional[int]
@@ -88,13 +88,13 @@ class AskGraphlitAskGraphlitMessageCitationsContent(BaseModel):
     snapshot_count: Optional[int] = Field(alias="snapshotCount")
     summary: Optional[str]
     custom_summary: Optional[str] = Field(alias="customSummary")
-    keywords: Optional[List[str]]
-    bullets: Optional[List[str]]
-    headlines: Optional[List[str]]
-    posts: Optional[List[str]]
-    chapters: Optional[List[str]]
-    questions: Optional[List[str]]
-    quotes: Optional[List[str]]
+    keywords: Optional[list[str]]
+    bullets: Optional[list[str]]
+    headlines: Optional[list[str]]
+    posts: Optional[list[str]]
+    chapters: Optional[list[str]]
+    questions: Optional[list[str]]
+    quotes: Optional[list[str]]
     video: Optional["AskGraphlitAskGraphlitMessageCitationsContentVideo"]
     audio: Optional["AskGraphlitAskGraphlitMessageCitationsContentAudio"]
     image: Optional["AskGraphlitAskGraphlitMessageCitationsContentImage"]
@@ -110,12 +110,12 @@ class AskGraphlitAskGraphlitMessageCitationsContentVideo(BaseModel):
     software: Optional[str]
     title: Optional[str]
     description: Optional[str]
-    keywords: Optional[List[Optional[str]]]
+    keywords: Optional[list[Optional[str]]]
     author: Optional[str]
 
 
 class AskGraphlitAskGraphlitMessageCitationsContentAudio(BaseModel):
-    keywords: Optional[List[Optional[str]]]
+    keywords: Optional[list[Optional[str]]]
     author: Optional[str]
     series: Optional[str]
     episode: Optional[str]
@@ -163,7 +163,7 @@ class AskGraphlitAskGraphlitMessageCitationsContentDocument(BaseModel):
     last_modified_by: Optional[str] = Field(alias="lastModifiedBy")
     publisher: Optional[str]
     description: Optional[str]
-    keywords: Optional[List[Optional[str]]]
+    keywords: Optional[list[Optional[str]]]
     page_count: Optional[int] = Field(alias="pageCount")
     worksheet_count: Optional[int] = Field(alias="worksheetCount")
     slide_count: Optional[int] = Field(alias="slideCount")

{graphlit_client-1.0.20260129001 → graphlit_client-1.0.20260129002}/graphlit_api/async_base_client.py

@@ -2,7 +2,8 @@
 
 import enum
 import json
-from typing import IO, Any, AsyncIterator, Dict, List, Optional, Tuple, TypeVar, cast
+from collections.abc import AsyncIterator
+from typing import IO, Any, Optional, TypeVar, cast
 from uuid import uuid4
 
 import httpx
@@ -21,7 +22,7 @@ try:
     from websockets import (  # type: ignore[import-not-found,unused-ignore]
         ClientConnection,
     )
-    from websockets import (
+    from websockets import (  # type: ignore[import-not-found,unused-ignore]
         connect as ws_connect,
     )
     from websockets.typing import (  # type: ignore[import-not-found,unused-ignore]
@@ -65,12 +66,12 @@ class AsyncBaseClient:
     def __init__(
         self,
         url: str = "",
-        headers: Optional[Dict[str, str]] = None,
+        headers: Optional[dict[str, str]] = None,
         http_client: Optional[httpx.AsyncClient] = None,
         ws_url: str = "",
-        ws_headers: Optional[Dict[str, Any]] = None,
+        ws_headers: Optional[dict[str, Any]] = None,
         ws_origin: Optional[str] = None,
-        ws_connection_init_payload: Optional[Dict[str, Any]] = None,
+        ws_connection_init_payload: Optional[dict[str, Any]] = None,
     ) -> None:
         self.url = url
         self.headers = headers
@@ -98,7 +99,7 @@ class AsyncBaseClient:
         self,
         query: str,
         operation_name: Optional[str] = None,
-        variables: Optional[Dict[str, Any]] = None,
+        variables: Optional[dict[str, Any]] = None,
         **kwargs: Any,
     ) -> httpx.Response:
         processed_variables, files, files_map = self._process_variables(variables)
@@ -120,7 +121,7 @@ class AsyncBaseClient:
             **kwargs,
         )
 
-    def get_data(self, response: httpx.Response) -> Dict[str, Any]:
+    def get_data(self, response: httpx.Response) -> dict[str, Any]:
         if not response.is_success:
             raise GraphQLClientHttpError(
                 status_code=response.status_code, response=response
@@ -144,19 +145,19 @@ class AsyncBaseClient:
                 errors_dicts=errors, data=data
             )
 
-        return cast(Dict[str, Any], data)
+        return cast(dict[str, Any], data)
 
     async def execute_ws(
         self,
         query: str,
         operation_name: Optional[str] = None,
-        variables: Optional[Dict[str, Any]] = None,
+        variables: Optional[dict[str, Any]] = None,
         **kwargs: Any,
-    ) -> AsyncIterator[Dict[str, Any]]:
+    ) -> AsyncIterator[dict[str, Any]]:
         headers = self.ws_headers.copy()
         headers.update(kwargs.get("extra_headers", {}))
 
-        merged_kwargs: Dict[str, Any] = {"origin": self.ws_origin}
+        merged_kwargs: dict[str, Any] = {"origin": self.ws_origin}
         merged_kwargs.update(kwargs)
         merged_kwargs["extra_headers"] = headers
 
@@ -187,9 +188,9 @@ class AsyncBaseClient:
                     yield data
 
     def _process_variables(
-        self, variables: Optional[Dict[str, Any]]
-    ) -> Tuple[
-        Dict[str, Any], Dict[str, Tuple[str, IO[bytes], str]], Dict[str, List[str]]
+        self, variables: Optional[dict[str, Any]]
+    ) -> tuple[
+        dict[str, Any], dict[str, tuple[str, IO[bytes], str]], dict[str, list[str]]
     ]:
         if not variables:
             return {}, {}, {}
@@ -198,8 +199,8 @@ class AsyncBaseClient:
         return self._get_files_from_variables(serializable_variables)
 
     def _convert_dict_to_json_serializable(
-        self, dict_: Dict[str, Any]
-    ) -> Dict[str, Any]:
+        self, dict_: dict[str, Any]
+    ) -> dict[str, Any]:
         return {
             key: self._convert_value(value)
             for key, value in dict_.items()
@@ -214,12 +215,12 @@ class AsyncBaseClient:
         return value
 
     def _get_files_from_variables(
-        self, variables: Dict[str, Any]
-    ) -> Tuple[
-        Dict[str, Any], Dict[str, Tuple[str, IO[bytes], str]], Dict[str, List[str]]
+        self, variables: dict[str, Any]
+    ) -> tuple[
+        dict[str, Any], dict[str, tuple[str, IO[bytes], str]], dict[str, list[str]]
     ]:
-        files_map: Dict[str, List[str]] = {}
-        files_list: List[Upload] = []
+        files_map: dict[str, list[str]] = {}
+        files_list: list[Upload] = []
 
         def separate_files(path: str, obj: Any) -> Any:
             if isinstance(obj, list):
@@ -249,7 +250,7 @@ class AsyncBaseClient:
             return obj
 
         nulled_variables = separate_files("variables", variables)
-        files: Dict[str, Tuple[str, IO[bytes], str]] = {
+        files: dict[str, tuple[str, IO[bytes], str]] = {
             str(i): (file_.filename, cast(IO[bytes], file_.content), file_.content_type)
             for i, file_ in enumerate(files_list)
         }
@@ -259,9 +260,9 @@ class AsyncBaseClient:
         self,
         query: str,
         operation_name: Optional[str],
-        variables: Dict[str, Any],
-        files: Dict[str, Tuple[str, IO[bytes], str]],
-        files_map: Dict[str, List[str]],
+        variables: dict[str, Any],
+        files: dict[str, tuple[str, IO[bytes], str]],
+        files_map: dict[str, list[str]],
         **kwargs: Any,
     ) -> httpx.Response:
         data = {
@@ -284,13 +285,13 @@ class AsyncBaseClient:
         self,
         query: str,
         operation_name: Optional[str],
-        variables: Dict[str, Any],
+        variables: dict[str, Any],
         **kwargs: Any,
     ) -> httpx.Response:
-        headers: Dict[str, str] = {"Content-Type": "application/json"}
+        headers: dict[str, str] = {"Content-type": "application/json"}
         headers.update(kwargs.get("headers", {}))
 
-        merged_kwargs: Dict[str, Any] = kwargs.copy()
+        merged_kwargs: dict[str, Any] = kwargs.copy()
         merged_kwargs["headers"] = headers
 
         return await self.http_client.post(
@@ -307,7 +308,7 @@ class AsyncBaseClient:
         )
 
     async def _send_connection_init(self, websocket: ClientConnection) -> None:
-        payload: Dict[str, Any] = {
+        payload: dict[str, Any] = {
             "type": GraphQLTransportWSMessageType.CONNECTION_INIT.value
         }
         if self.ws_connection_init_payload:
@@ -320,9 +321,9 @@ class AsyncBaseClient:
         operation_id: str,
         query: str,
         operation_name: Optional[str] = None,
-        variables: Optional[Dict[str, Any]] = None,
+        variables: Optional[dict[str, Any]] = None,
     ) -> None:
-        payload: Dict[str, Any] = {
+        payload: dict[str, Any] = {
             "id": operation_id,
             "type": GraphQLTransportWSMessageType.SUBSCRIBE.value,
             "payload": {"query": query, "operationName": operation_name},
@@ -338,7 +339,7 @@ class AsyncBaseClient:
         message: Data,
         websocket: ClientConnection,
         expected_type: Optional[GraphQLTransportWSMessageType] = None,
-    ) -> Optional[Dict[str, Any]]:
+    ) -> Optional[dict[str, Any]]:
         try:
             message_dict = json.loads(message)
         except json.JSONDecodeError as exc:
@@ -358,7 +359,7 @@ class AsyncBaseClient:
         if type_ == GraphQLTransportWSMessageType.NEXT:
             if "data" not in payload:
                 raise GraphQLClientInvalidMessageFormat(message=message)
-            return cast(Dict[str, Any], payload["data"])
+            return cast(dict[str, Any], payload["data"])
 
         if type_ == GraphQLTransportWSMessageType.COMPLETE:
             await websocket.close()