graphlit-client 1.0.20250426001__tar.gz → 1.0.20250428002__tar.gz

{graphlit_client-1.0.20250426001/graphlit_client.egg-info → graphlit_client-1.0.20250428002}/PKG-INFO

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

{graphlit_client-1.0.20250426001 → graphlit_client-1.0.20250428002}/graphlit_api/__init__.py

@@ -515,6 +515,7 @@ from .enums import (
     AzureDocumentIntelligenceModels,
     AzureDocumentIntelligenceVersions,
     AzureOpenAIModels,
+    BedrockModels,
     BillableMetrics,
     CategoryFacetTypes,
     CerebrasModels,
@@ -924,6 +925,7 @@ from .get_specification import (
     GetSpecificationSpecificationAnthropic,
     GetSpecificationSpecificationAzureAi,
     GetSpecificationSpecificationAzureOpenAi,
+    GetSpecificationSpecificationBedrock,
     GetSpecificationSpecificationCerebras,
     GetSpecificationSpecificationCohere,
     GetSpecificationSpecificationDeepseek,
@@ -1098,6 +1100,8 @@ from .input_types import (
     AzureOpenAIModelPropertiesInput,
     AzureOpenAIModelPropertiesUpdateInput,
     AzureTextExtractionPropertiesInput,
+    BedrockModelPropertiesInput,
+    BedrockModelPropertiesUpdateInput,
     BoundingBoxInput,
     BoxFeedPropertiesInput,
     BoxFeedPropertiesUpdateInput,
@@ -1644,6 +1648,7 @@ from .operations import (
     QUERY_SLACK_CHANNELS_GQL,
     QUERY_SOFTWARES_GQL,
     QUERY_SPECIFICATIONS_GQL,
+    QUERY_TOKENS_GQL,
     QUERY_USAGE_GQL,
     QUERY_USERS_GQL,
     QUERY_WORKFLOWS_GQL,
@@ -2129,6 +2134,7 @@ from .query_specifications import (
     QuerySpecificationsSpecificationsResultsAnthropic,
     QuerySpecificationsSpecificationsResultsAzureAi,
     QuerySpecificationsSpecificationsResultsAzureOpenAi,
+    QuerySpecificationsSpecificationsResultsBedrock,
     QuerySpecificationsSpecificationsResultsCerebras,
     QuerySpecificationsSpecificationsResultsCohere,
     QuerySpecificationsSpecificationsResultsDeepseek,
@@ -2147,6 +2153,7 @@ from .query_specifications import (
     QuerySpecificationsSpecificationsResultsStrategy,
     QuerySpecificationsSpecificationsResultsVoyage,
 )
+from .query_tokens import QueryTokens, QueryTokensTokens
 from .query_usage import QueryUsage, QueryUsageUsage
 from .query_users import (
     QueryUsers,
@@ -2561,6 +2568,9 @@ __all__ = [
     "AzureOpenAIModels",
     "AzureTextExtractionPropertiesInput",
     "BaseModel",
+    "BedrockModelPropertiesInput",
+    "BedrockModelPropertiesUpdateInput",
+    "BedrockModels",
     "BillableMetrics",
     "BoundingBoxInput",
     "BoxFeedPropertiesInput",
@@ -3549,6 +3559,7 @@ __all__ = [
     "GetSpecificationSpecificationAnthropic",
     "GetSpecificationSpecificationAzureAi",
     "GetSpecificationSpecificationAzureOpenAi",
+    "GetSpecificationSpecificationBedrock",
     "GetSpecificationSpecificationCerebras",
     "GetSpecificationSpecificationCohere",
     "GetSpecificationSpecificationDeepseek",
@@ -4018,6 +4029,7 @@ __all__ = [
     "QUERY_SLACK_CHANNELS_GQL",
     "QUERY_SOFTWARES_GQL",
     "QUERY_SPECIFICATIONS_GQL",
+    "QUERY_TOKENS_GQL",
     "QUERY_USAGE_GQL",
     "QUERY_USERS_GQL",
     "QUERY_WORKFLOWS_GQL",
@@ -4312,6 +4324,7 @@ __all__ = [
     "QuerySpecificationsSpecificationsResultsAnthropic",
     "QuerySpecificationsSpecificationsResultsAzureAi",
     "QuerySpecificationsSpecificationsResultsAzureOpenAi",
+    "QuerySpecificationsSpecificationsResultsBedrock",
     "QuerySpecificationsSpecificationsResultsCerebras",
     "QuerySpecificationsSpecificationsResultsCohere",
     "QuerySpecificationsSpecificationsResultsDeepseek",
@@ -4329,6 +4342,8 @@ __all__ = [
     "QuerySpecificationsSpecificationsResultsRevisionStrategy",
     "QuerySpecificationsSpecificationsResultsStrategy",
     "QuerySpecificationsSpecificationsResultsVoyage",
+    "QueryTokens",
+    "QueryTokensTokens",
     "QueryUsage",
     "QueryUsageUsage",
     "QueryUsers",

{graphlit_client-1.0.20250426001 → graphlit_client-1.0.20250428002}/graphlit_api/client.py

@@ -559,6 +559,7 @@ from .operations import (
     QUERY_SLACK_CHANNELS_GQL,
     QUERY_SOFTWARES_GQL,
     QUERY_SPECIFICATIONS_GQL,
+    QUERY_TOKENS_GQL,
     QUERY_USAGE_GQL,
     QUERY_USERS_GQL,
     QUERY_WORKFLOWS_GQL,
@@ -656,6 +657,7 @@ from .query_share_point_libraries import QuerySharePointLibraries
 from .query_slack_channels import QuerySlackChannels
 from .query_softwares import QuerySoftwares
 from .query_specifications import QuerySpecifications
+from .query_tokens import QueryTokens
 from .query_usage import QueryUsage
 from .query_users import QueryUsers
 from .query_workflows import QueryWorkflows
@@ -4857,11 +4859,25 @@ class Client(AsyncBaseClient):
         data = self.get_data(response)
         return QueryCredits.model_validate(data)
 
+    async def query_tokens(
+        self, start_date: Any, duration: Any, **kwargs: Any
+    ) -> QueryTokens:
+        variables: Dict[str, object] = {"startDate": start_date, "duration": duration}
+        response = await self.execute(
+            query=QUERY_TOKENS_GQL,
+            operation_name="QueryTokens",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return QueryTokens.model_validate(data)
+
     async def query_usage(
         self,
         start_date: Any,
         duration: Any,
         names: Union[Optional[List[str]], UnsetType] = UNSET,
+        excluded_names: Union[Optional[List[str]], UnsetType] = UNSET,
         offset: Union[Optional[int], UnsetType] = UNSET,
         limit: Union[Optional[int], UnsetType] = UNSET,
         **kwargs: Any
@@ -4870,6 +4886,7 @@ class Client(AsyncBaseClient):
             "startDate": start_date,
             "duration": duration,
             "names": names,
+            "excludedNames": excluded_names,
             "offset": offset,
             "limit": limit,
         }

{graphlit_client-1.0.20250426001 → graphlit_client-1.0.20250428002}/graphlit_api/enums.py

@@ -246,6 +246,7 @@ class ModelServiceTypes(str, Enum):
     DEEPSEEK = "DEEPSEEK"
     JINA = "JINA"
     VOYAGE = "VOYAGE"
+    BEDROCK = "BEDROCK"
 
 
 class MedicalStudyFacetTypes(str, Enum):
@@ -537,6 +538,10 @@ class OccurrenceTypes(str, Enum):
     TEXT = "TEXT"
 
 
+class BedrockModels(str, Enum):
+    CUSTOM = "CUSTOM"
+
+
 class ContentSourceTypes(str, Enum):
     FRAME = "FRAME"
     DOCUMENT = "DOCUMENT"

{graphlit_client-1.0.20250426001 → graphlit_client-1.0.20250428002}/graphlit_api/get_specification.py

@@ -9,6 +9,7 @@ from .base_model import BaseModel
 from .enums import (
     AnthropicModels,
     AzureOpenAIModels,
+    BedrockModels,
     CerebrasModels,
     CohereModels,
     ConversationSearchTypes,
@@ -78,6 +79,7 @@ class GetSpecificationSpecification(BaseModel):
     google: Optional["GetSpecificationSpecificationGoogle"]
     replicate: Optional["GetSpecificationSpecificationReplicate"]
     mistral: Optional["GetSpecificationSpecificationMistral"]
+    bedrock: Optional["GetSpecificationSpecificationBedrock"]
     groq: Optional["GetSpecificationSpecificationGroq"]
     cerebras: Optional["GetSpecificationSpecificationCerebras"]
     deepseek: Optional["GetSpecificationSpecificationDeepseek"]
@@ -222,6 +224,17 @@ class GetSpecificationSpecificationMistral(BaseModel):
     chunk_token_limit: Optional[int] = Field(alias="chunkTokenLimit")
 
 
+class GetSpecificationSpecificationBedrock(BaseModel):
+    token_limit: Optional[int] = Field(alias="tokenLimit")
+    completion_token_limit: Optional[int] = Field(alias="completionTokenLimit")
+    model: BedrockModels
+    access_key: Optional[str] = Field(alias="accessKey")
+    secret_access_key: Optional[str] = Field(alias="secretAccessKey")
+    model_name: Optional[str] = Field(alias="modelName")
+    temperature: Optional[float]
+    probability: Optional[float]
+
+
 class GetSpecificationSpecificationGroq(BaseModel):
     token_limit: Optional[int] = Field(alias="tokenLimit")
     completion_token_limit: Optional[int] = Field(alias="completionTokenLimit")

{graphlit_client-1.0.20250426001 → graphlit_client-1.0.20250428002}/graphlit_api/input_types.py

@@ -14,6 +14,7 @@ from .enums import (
     AzureDocumentIntelligenceModels,
     AzureDocumentIntelligenceVersions,
     AzureOpenAIModels,
+    BedrockModels,
     CategoryFacetTypes,
     CerebrasModels,
     CohereModels,
@@ -852,6 +853,7 @@ class SpecificationInput(BaseModel):
     google: Optional["GoogleModelPropertiesInput"] = None
     replicate: Optional["ReplicateModelPropertiesInput"] = None
     mistral: Optional["MistralModelPropertiesInput"] = None
+    bedrock: Optional["BedrockModelPropertiesInput"] = None
     groq: Optional["GroqModelPropertiesInput"] = None
     cerebras: Optional["CerebrasModelPropertiesInput"] = None
     deepseek: Optional["DeepseekModelPropertiesInput"] = None
@@ -1028,6 +1030,19 @@ class IntegrationConnectorUpdateInput(BaseModel):
     twitter: Optional["TwitterIntegrationPropertiesInput"] = None
 
 
+class BedrockModelPropertiesInput(BaseModel):
+    model: BedrockModels
+    model_name: Optional[str] = Field(alias="modelName", default=None)
+    access_key: Optional[str] = Field(alias="accessKey", default=None)
+    secret_access_key: Optional[str] = Field(alias="secretAccessKey", default=None)
+    temperature: Optional[float] = None
+    probability: Optional[float] = None
+    token_limit: Optional[int] = Field(alias="tokenLimit", default=None)
+    completion_token_limit: Optional[int] = Field(
+        alias="completionTokenLimit", default=None
+    )
+
+
 class MedicalDeviceFilter(BaseModel):
     search: Optional[str] = None
     order_by: Optional[OrderByTypes] = Field(alias="orderBy", default=None)
@@ -1095,6 +1110,19 @@ class AzureFileFeedPropertiesUpdateInput(BaseModel):
     prefix: Optional[str] = None
 
 
+class BedrockModelPropertiesUpdateInput(BaseModel):
+    model: Optional[BedrockModels] = None
+    model_name: Optional[str] = Field(alias="modelName", default=None)
+    access_key: Optional[str] = Field(alias="accessKey", default=None)
+    secret_access_key: Optional[str] = Field(alias="secretAccessKey", default=None)
+    temperature: Optional[float] = None
+    probability: Optional[float] = None
+    token_limit: Optional[int] = Field(alias="tokenLimit", default=None)
+    completion_token_limit: Optional[int] = Field(
+        alias="completionTokenLimit", default=None
+    )
+
+
 class PromptClassificationRuleInput(BaseModel):
     if_: Optional[str] = Field(alias="if", default=None)
     then: Optional[str] = None
@@ -1487,6 +1515,7 @@ class SpecificationUpdateInput(BaseModel):
     google: Optional["GoogleModelPropertiesUpdateInput"] = None
     replicate: Optional["ReplicateModelPropertiesUpdateInput"] = None
     mistral: Optional["MistralModelPropertiesUpdateInput"] = None
+    bedrock: Optional["BedrockModelPropertiesUpdateInput"] = None
     groq: Optional["GroqModelPropertiesUpdateInput"] = None
     cerebras: Optional["CerebrasModelPropertiesUpdateInput"] = None
     deepseek: Optional["DeepseekModelPropertiesUpdateInput"] = None

{graphlit_client-1.0.20250426001 → graphlit_client-1.0.20250428002}/graphlit_api/operations.py

@@ -245,6 +245,7 @@ __all__ = [
     "QUERY_SLACK_CHANNELS_GQL",
     "QUERY_SOFTWARES_GQL",
     "QUERY_SPECIFICATIONS_GQL",
+    "QUERY_TOKENS_GQL",
     "QUERY_USAGE_GQL",
     "QUERY_USERS_GQL",
     "QUERY_WORKFLOWS_GQL",
@@ -8015,12 +8016,36 @@ query QueryCredits($startDate: DateTime!, $duration: TimeSpan!) {
 }
 """
 
+QUERY_TOKENS_GQL = """
+query QueryTokens($startDate: DateTime!, $duration: TimeSpan!) {
+  tokens(startDate: $startDate, duration: $duration) {
+    correlationId
+    ownerId
+    embeddingInputTokens
+    embeddingModelServices
+    completionInputTokens
+    completionOutputTokens
+    completionModelServices
+    preparationInputTokens
+    preparationOutputTokens
+    preparationModelServices
+    extractionInputTokens
+    extractionOutputTokens
+    extractionModelServices
+    generationInputTokens
+    generationOutputTokens
+    generationModelServices
+  }
+}
+"""
+
 QUERY_USAGE_GQL = """
-query QueryUsage($startDate: DateTime!, $duration: TimeSpan!, $names: [String!], $offset: Int, $limit: Int) {
+query QueryUsage($startDate: DateTime!, $duration: TimeSpan!, $names: [String!], $excludedNames: [String!], $offset: Int, $limit: Int) {
   usage(
     startDate: $startDate
     duration: $duration
     names: $names
+    excludedNames: $excludedNames
     offset: $offset
     limit: $limit
   ) {
@@ -8470,6 +8495,16 @@ query GetSpecification($id: ID!, $correlationId: String) {
       probability
       chunkTokenLimit
     }
+    bedrock {
+      tokenLimit
+      completionTokenLimit
+      model
+      accessKey
+      secretAccessKey
+      modelName
+      temperature
+      probability
+    }
     groq {
       tokenLimit
       completionTokenLimit
@@ -8818,6 +8853,16 @@ query QuerySpecifications($filter: SpecificationFilter, $correlationId: String)
         probability
         chunkTokenLimit
       }
+      bedrock {
+        tokenLimit
+        completionTokenLimit
+        model
+        accessKey
+        secretAccessKey
+        modelName
+        temperature
+        probability
+      }
       groq {
         tokenLimit
         completionTokenLimit

{graphlit_client-1.0.20250426001 → graphlit_client-1.0.20250428002}/graphlit_api/query_specifications.py

@@ -9,6 +9,7 @@ from .base_model import BaseModel
 from .enums import (
     AnthropicModels,
     AzureOpenAIModels,
+    BedrockModels,
     CerebrasModels,
     CohereModels,
     ConversationSearchTypes,
@@ -86,6 +87,7 @@ class QuerySpecificationsSpecificationsResults(BaseModel):
     google: Optional["QuerySpecificationsSpecificationsResultsGoogle"]
     replicate: Optional["QuerySpecificationsSpecificationsResultsReplicate"]
     mistral: Optional["QuerySpecificationsSpecificationsResultsMistral"]
+    bedrock: Optional["QuerySpecificationsSpecificationsResultsBedrock"]
     groq: Optional["QuerySpecificationsSpecificationsResultsGroq"]
     cerebras: Optional["QuerySpecificationsSpecificationsResultsCerebras"]
     deepseek: Optional["QuerySpecificationsSpecificationsResultsDeepseek"]
@@ -230,6 +232,17 @@ class QuerySpecificationsSpecificationsResultsMistral(BaseModel):
     chunk_token_limit: Optional[int] = Field(alias="chunkTokenLimit")
 
 
+class QuerySpecificationsSpecificationsResultsBedrock(BaseModel):
+    token_limit: Optional[int] = Field(alias="tokenLimit")
+    completion_token_limit: Optional[int] = Field(alias="completionTokenLimit")
+    model: BedrockModels
+    access_key: Optional[str] = Field(alias="accessKey")
+    secret_access_key: Optional[str] = Field(alias="secretAccessKey")
+    model_name: Optional[str] = Field(alias="modelName")
+    temperature: Optional[float]
+    probability: Optional[float]
+
+
 class QuerySpecificationsSpecificationsResultsGroq(BaseModel):
     token_limit: Optional[int] = Field(alias="tokenLimit")
     completion_token_limit: Optional[int] = Field(alias="completionTokenLimit")

graphlit_client-1.0.20250428002/graphlit_api/query_tokens.py

@@ -0,0 +1,44 @@
+# Generated by ariadne-codegen
+# Source: ./documents
+
+from typing import List, Optional
+
+from pydantic import Field
+
+from .base_model import BaseModel
+
+
+class QueryTokens(BaseModel):
+    tokens: Optional["QueryTokensTokens"]
+
+
+class QueryTokensTokens(BaseModel):
+    correlation_id: Optional[str] = Field(alias="correlationId")
+    owner_id: Optional[str] = Field(alias="ownerId")
+    embedding_input_tokens: Optional[int] = Field(alias="embeddingInputTokens")
+    embedding_model_services: Optional[List[Optional[str]]] = Field(
+        alias="embeddingModelServices"
+    )
+    completion_input_tokens: Optional[int] = Field(alias="completionInputTokens")
+    completion_output_tokens: Optional[int] = Field(alias="completionOutputTokens")
+    completion_model_services: Optional[List[Optional[str]]] = Field(
+        alias="completionModelServices"
+    )
+    preparation_input_tokens: Optional[int] = Field(alias="preparationInputTokens")
+    preparation_output_tokens: Optional[int] = Field(alias="preparationOutputTokens")
+    preparation_model_services: Optional[List[Optional[str]]] = Field(
+        alias="preparationModelServices"
+    )
+    extraction_input_tokens: Optional[int] = Field(alias="extractionInputTokens")
+    extraction_output_tokens: Optional[int] = Field(alias="extractionOutputTokens")
+    extraction_model_services: Optional[List[Optional[str]]] = Field(
+        alias="extractionModelServices"
+    )
+    generation_input_tokens: Optional[int] = Field(alias="generationInputTokens")
+    generation_output_tokens: Optional[int] = Field(alias="generationOutputTokens")
+    generation_model_services: Optional[List[Optional[str]]] = Field(
+        alias="generationModelServices"
+    )
+
+
+QueryTokens.model_rebuild()

{graphlit_client-1.0.20250426001 → graphlit_client-1.0.20250428002/graphlit_client.egg-info}/PKG-INFO

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

{graphlit_client-1.0.20250426001 → graphlit_client-1.0.20250428002}/graphlit_client.egg-info/SOURCES.txt

@@ -255,6 +255,7 @@ graphlit_api/query_share_point_libraries.py
 graphlit_api/query_slack_channels.py
 graphlit_api/query_softwares.py
 graphlit_api/query_specifications.py
+graphlit_api/query_tokens.py
 graphlit_api/query_usage.py
 graphlit_api/query_users.py
 graphlit_api/query_workflows.py