graphlit-client 1.0.20250423001__py3-none-any.whl → 1.0.20250428001__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
graphlit_api/__init__.py CHANGED
@@ -515,6 +515,7 @@ from .enums import (
515
515
  AzureDocumentIntelligenceModels,
516
516
  AzureDocumentIntelligenceVersions,
517
517
  AzureOpenAIModels,
518
+ BedrockModels,
518
519
  BillableMetrics,
519
520
  CategoryFacetTypes,
520
521
  CerebrasModels,
@@ -698,6 +699,7 @@ from .get_alert import (
698
699
  GetAlertAlertFilterOrObservationsObservable,
699
700
  GetAlertAlertFilterOrUsers,
700
701
  GetAlertAlertFilterOrWorkflows,
702
+ GetAlertAlertFilterSimilarContents,
701
703
  GetAlertAlertFilterUsers,
702
704
  GetAlertAlertFilterWorkflows,
703
705
  GetAlertAlertIntegration,
@@ -778,6 +780,7 @@ from .get_conversation import (
778
780
  GetConversationConversationAugmentedFilterOrObservationsObservable,
779
781
  GetConversationConversationAugmentedFilterOrUsers,
780
782
  GetConversationConversationAugmentedFilterOrWorkflows,
783
+ GetConversationConversationAugmentedFilterSimilarContents,
781
784
  GetConversationConversationAugmentedFilterUsers,
782
785
  GetConversationConversationAugmentedFilterWorkflows,
783
786
  GetConversationConversationFallbacks,
@@ -803,6 +806,7 @@ from .get_conversation import (
803
806
  GetConversationConversationFilterOrObservationsObservable,
804
807
  GetConversationConversationFilterOrUsers,
805
808
  GetConversationConversationFilterOrWorkflows,
809
+ GetConversationConversationFilterSimilarContents,
806
810
  GetConversationConversationFilterUsers,
807
811
  GetConversationConversationFilterWorkflows,
808
812
  GetConversationConversationMessages,
@@ -921,6 +925,7 @@ from .get_specification import (
921
925
  GetSpecificationSpecificationAnthropic,
922
926
  GetSpecificationSpecificationAzureAi,
923
927
  GetSpecificationSpecificationAzureOpenAi,
928
+ GetSpecificationSpecificationBedrock,
924
929
  GetSpecificationSpecificationCerebras,
925
930
  GetSpecificationSpecificationCohere,
926
931
  GetSpecificationSpecificationDeepseek,
@@ -1095,6 +1100,8 @@ from .input_types import (
1095
1100
  AzureOpenAIModelPropertiesInput,
1096
1101
  AzureOpenAIModelPropertiesUpdateInput,
1097
1102
  AzureTextExtractionPropertiesInput,
1103
+ BedrockModelPropertiesInput,
1104
+ BedrockModelPropertiesUpdateInput,
1098
1105
  BoundingBoxInput,
1099
1106
  BoxFeedPropertiesInput,
1100
1107
  BoxFeedPropertiesUpdateInput,
@@ -1641,6 +1648,7 @@ from .operations import (
1641
1648
  QUERY_SLACK_CHANNELS_GQL,
1642
1649
  QUERY_SOFTWARES_GQL,
1643
1650
  QUERY_SPECIFICATIONS_GQL,
1651
+ QUERY_TOKENS_GQL,
1644
1652
  QUERY_USAGE_GQL,
1645
1653
  QUERY_USERS_GQL,
1646
1654
  QUERY_WORKFLOWS_GQL,
@@ -1801,6 +1809,7 @@ from .query_alerts import (
1801
1809
  QueryAlertsAlertsResultsFilterOrObservationsObservable,
1802
1810
  QueryAlertsAlertsResultsFilterOrUsers,
1803
1811
  QueryAlertsAlertsResultsFilterOrWorkflows,
1812
+ QueryAlertsAlertsResultsFilterSimilarContents,
1804
1813
  QueryAlertsAlertsResultsFilterUsers,
1805
1814
  QueryAlertsAlertsResultsFilterWorkflows,
1806
1815
  QueryAlertsAlertsResultsIntegration,
@@ -1901,6 +1910,7 @@ from .query_conversations import (
1901
1910
  QueryConversationsConversationsResultsAugmentedFilterOrObservationsObservable,
1902
1911
  QueryConversationsConversationsResultsAugmentedFilterOrUsers,
1903
1912
  QueryConversationsConversationsResultsAugmentedFilterOrWorkflows,
1913
+ QueryConversationsConversationsResultsAugmentedFilterSimilarContents,
1904
1914
  QueryConversationsConversationsResultsAugmentedFilterUsers,
1905
1915
  QueryConversationsConversationsResultsAugmentedFilterWorkflows,
1906
1916
  QueryConversationsConversationsResultsFallbacks,
@@ -1926,6 +1936,7 @@ from .query_conversations import (
1926
1936
  QueryConversationsConversationsResultsFilterOrObservationsObservable,
1927
1937
  QueryConversationsConversationsResultsFilterOrUsers,
1928
1938
  QueryConversationsConversationsResultsFilterOrWorkflows,
1939
+ QueryConversationsConversationsResultsFilterSimilarContents,
1929
1940
  QueryConversationsConversationsResultsFilterUsers,
1930
1941
  QueryConversationsConversationsResultsFilterWorkflows,
1931
1942
  QueryConversationsConversationsResultsMessages,
@@ -2123,6 +2134,7 @@ from .query_specifications import (
2123
2134
  QuerySpecificationsSpecificationsResultsAnthropic,
2124
2135
  QuerySpecificationsSpecificationsResultsAzureAi,
2125
2136
  QuerySpecificationsSpecificationsResultsAzureOpenAi,
2137
+ QuerySpecificationsSpecificationsResultsBedrock,
2126
2138
  QuerySpecificationsSpecificationsResultsCerebras,
2127
2139
  QuerySpecificationsSpecificationsResultsCohere,
2128
2140
  QuerySpecificationsSpecificationsResultsDeepseek,
@@ -2141,6 +2153,7 @@ from .query_specifications import (
2141
2153
  QuerySpecificationsSpecificationsResultsStrategy,
2142
2154
  QuerySpecificationsSpecificationsResultsVoyage,
2143
2155
  )
2156
+ from .query_tokens import QueryTokens, QueryTokensTokens
2144
2157
  from .query_usage import QueryUsage, QueryUsageUsage
2145
2158
  from .query_users import (
2146
2159
  QueryUsers,
@@ -2555,6 +2568,9 @@ __all__ = [
2555
2568
  "AzureOpenAIModels",
2556
2569
  "AzureTextExtractionPropertiesInput",
2557
2570
  "BaseModel",
2571
+ "BedrockModelPropertiesInput",
2572
+ "BedrockModelPropertiesUpdateInput",
2573
+ "BedrockModels",
2558
2574
  "BillableMetrics",
2559
2575
  "BoundingBoxInput",
2560
2576
  "BoxFeedPropertiesInput",
@@ -3331,6 +3347,7 @@ __all__ = [
3331
3347
  "GetAlertAlertFilterOrObservationsObservable",
3332
3348
  "GetAlertAlertFilterOrUsers",
3333
3349
  "GetAlertAlertFilterOrWorkflows",
3350
+ "GetAlertAlertFilterSimilarContents",
3334
3351
  "GetAlertAlertFilterUsers",
3335
3352
  "GetAlertAlertFilterWorkflows",
3336
3353
  "GetAlertAlertIntegration",
@@ -3406,6 +3423,7 @@ __all__ = [
3406
3423
  "GetConversationConversationAugmentedFilterOrObservationsObservable",
3407
3424
  "GetConversationConversationAugmentedFilterOrUsers",
3408
3425
  "GetConversationConversationAugmentedFilterOrWorkflows",
3426
+ "GetConversationConversationAugmentedFilterSimilarContents",
3409
3427
  "GetConversationConversationAugmentedFilterUsers",
3410
3428
  "GetConversationConversationAugmentedFilterWorkflows",
3411
3429
  "GetConversationConversationFallbacks",
@@ -3431,6 +3449,7 @@ __all__ = [
3431
3449
  "GetConversationConversationFilterOrObservationsObservable",
3432
3450
  "GetConversationConversationFilterOrUsers",
3433
3451
  "GetConversationConversationFilterOrWorkflows",
3452
+ "GetConversationConversationFilterSimilarContents",
3434
3453
  "GetConversationConversationFilterUsers",
3435
3454
  "GetConversationConversationFilterWorkflows",
3436
3455
  "GetConversationConversationMessages",
@@ -3540,6 +3559,7 @@ __all__ = [
3540
3559
  "GetSpecificationSpecificationAnthropic",
3541
3560
  "GetSpecificationSpecificationAzureAi",
3542
3561
  "GetSpecificationSpecificationAzureOpenAi",
3562
+ "GetSpecificationSpecificationBedrock",
3543
3563
  "GetSpecificationSpecificationCerebras",
3544
3564
  "GetSpecificationSpecificationCohere",
3545
3565
  "GetSpecificationSpecificationDeepseek",
@@ -4009,6 +4029,7 @@ __all__ = [
4009
4029
  "QUERY_SLACK_CHANNELS_GQL",
4010
4030
  "QUERY_SOFTWARES_GQL",
4011
4031
  "QUERY_SPECIFICATIONS_GQL",
4032
+ "QUERY_TOKENS_GQL",
4012
4033
  "QUERY_USAGE_GQL",
4013
4034
  "QUERY_USERS_GQL",
4014
4035
  "QUERY_WORKFLOWS_GQL",
@@ -4037,6 +4058,7 @@ __all__ = [
4037
4058
  "QueryAlertsAlertsResultsFilterOrObservationsObservable",
4038
4059
  "QueryAlertsAlertsResultsFilterOrUsers",
4039
4060
  "QueryAlertsAlertsResultsFilterOrWorkflows",
4061
+ "QueryAlertsAlertsResultsFilterSimilarContents",
4040
4062
  "QueryAlertsAlertsResultsFilterUsers",
4041
4063
  "QueryAlertsAlertsResultsFilterWorkflows",
4042
4064
  "QueryAlertsAlertsResultsIntegration",
@@ -4125,6 +4147,7 @@ __all__ = [
4125
4147
  "QueryConversationsConversationsResultsAugmentedFilterOrObservationsObservable",
4126
4148
  "QueryConversationsConversationsResultsAugmentedFilterOrUsers",
4127
4149
  "QueryConversationsConversationsResultsAugmentedFilterOrWorkflows",
4150
+ "QueryConversationsConversationsResultsAugmentedFilterSimilarContents",
4128
4151
  "QueryConversationsConversationsResultsAugmentedFilterUsers",
4129
4152
  "QueryConversationsConversationsResultsAugmentedFilterWorkflows",
4130
4153
  "QueryConversationsConversationsResultsFallbacks",
@@ -4150,6 +4173,7 @@ __all__ = [
4150
4173
  "QueryConversationsConversationsResultsFilterOrObservationsObservable",
4151
4174
  "QueryConversationsConversationsResultsFilterOrUsers",
4152
4175
  "QueryConversationsConversationsResultsFilterOrWorkflows",
4176
+ "QueryConversationsConversationsResultsFilterSimilarContents",
4153
4177
  "QueryConversationsConversationsResultsFilterUsers",
4154
4178
  "QueryConversationsConversationsResultsFilterWorkflows",
4155
4179
  "QueryConversationsConversationsResultsMessages",
@@ -4300,6 +4324,7 @@ __all__ = [
4300
4324
  "QuerySpecificationsSpecificationsResultsAnthropic",
4301
4325
  "QuerySpecificationsSpecificationsResultsAzureAi",
4302
4326
  "QuerySpecificationsSpecificationsResultsAzureOpenAi",
4327
+ "QuerySpecificationsSpecificationsResultsBedrock",
4303
4328
  "QuerySpecificationsSpecificationsResultsCerebras",
4304
4329
  "QuerySpecificationsSpecificationsResultsCohere",
4305
4330
  "QuerySpecificationsSpecificationsResultsDeepseek",
@@ -4317,6 +4342,8 @@ __all__ = [
4317
4342
  "QuerySpecificationsSpecificationsResultsRevisionStrategy",
4318
4343
  "QuerySpecificationsSpecificationsResultsStrategy",
4319
4344
  "QuerySpecificationsSpecificationsResultsVoyage",
4345
+ "QueryTokens",
4346
+ "QueryTokensTokens",
4320
4347
  "QueryUsage",
4321
4348
  "QueryUsageUsage",
4322
4349
  "QueryUsers",
graphlit_api/client.py CHANGED
@@ -559,6 +559,7 @@ from .operations import (
559
559
  QUERY_SLACK_CHANNELS_GQL,
560
560
  QUERY_SOFTWARES_GQL,
561
561
  QUERY_SPECIFICATIONS_GQL,
562
+ QUERY_TOKENS_GQL,
562
563
  QUERY_USAGE_GQL,
563
564
  QUERY_USERS_GQL,
564
565
  QUERY_WORKFLOWS_GQL,
@@ -656,6 +657,7 @@ from .query_share_point_libraries import QuerySharePointLibraries
656
657
  from .query_slack_channels import QuerySlackChannels
657
658
  from .query_softwares import QuerySoftwares
658
659
  from .query_specifications import QuerySpecifications
660
+ from .query_tokens import QueryTokens
659
661
  from .query_usage import QueryUsage
660
662
  from .query_users import QueryUsers
661
663
  from .query_workflows import QueryWorkflows
@@ -4857,6 +4859,19 @@ class Client(AsyncBaseClient):
4857
4859
  data = self.get_data(response)
4858
4860
  return QueryCredits.model_validate(data)
4859
4861
 
4862
+ async def query_tokens(
4863
+ self, start_date: Any, duration: Any, **kwargs: Any
4864
+ ) -> QueryTokens:
4865
+ variables: Dict[str, object] = {"startDate": start_date, "duration": duration}
4866
+ response = await self.execute(
4867
+ query=QUERY_TOKENS_GQL,
4868
+ operation_name="QueryTokens",
4869
+ variables=variables,
4870
+ **kwargs
4871
+ )
4872
+ data = self.get_data(response)
4873
+ return QueryTokens.model_validate(data)
4874
+
4860
4875
  async def query_usage(
4861
4876
  self,
4862
4877
  start_date: Any,
graphlit_api/enums.py CHANGED
@@ -246,6 +246,7 @@ class ModelServiceTypes(str, Enum):
246
246
  DEEPSEEK = "DEEPSEEK"
247
247
  JINA = "JINA"
248
248
  VOYAGE = "VOYAGE"
249
+ BEDROCK = "BEDROCK"
249
250
 
250
251
 
251
252
  class MedicalStudyFacetTypes(str, Enum):
@@ -537,6 +538,10 @@ class OccurrenceTypes(str, Enum):
537
538
  TEXT = "TEXT"
538
539
 
539
540
 
541
+ class BedrockModels(str, Enum):
542
+ CUSTOM = "CUSTOM"
543
+
544
+
540
545
  class ContentSourceTypes(str, Enum):
541
546
  FRAME = "FRAME"
542
547
  DOCUMENT = "DOCUMENT"
graphlit_api/get_alert.py CHANGED
@@ -61,6 +61,9 @@ class GetAlertAlertFilter(BaseModel):
61
61
  file_types: Optional[List[Optional[FileTypes]]] = Field(alias="fileTypes")
62
62
  formats: Optional[List[Optional[str]]]
63
63
  file_extensions: Optional[List[str]] = Field(alias="fileExtensions")
64
+ similar_contents: Optional[List["GetAlertAlertFilterSimilarContents"]] = Field(
65
+ alias="similarContents"
66
+ )
64
67
  contents: Optional[List["GetAlertAlertFilterContents"]]
65
68
  feeds: Optional[List["GetAlertAlertFilterFeeds"]]
66
69
  workflows: Optional[List["GetAlertAlertFilterWorkflows"]]
@@ -81,6 +84,10 @@ class GetAlertAlertFilterCreationDateRange(BaseModel):
81
84
  to: Optional[Any]
82
85
 
83
86
 
87
+ class GetAlertAlertFilterSimilarContents(BaseModel):
88
+ id: str
89
+
90
+
84
91
  class GetAlertAlertFilterContents(BaseModel):
85
92
  id: str
86
93
 
@@ -208,6 +208,9 @@ class GetConversationConversationFilter(BaseModel):
208
208
  file_types: Optional[List[Optional[FileTypes]]] = Field(alias="fileTypes")
209
209
  formats: Optional[List[Optional[str]]]
210
210
  file_extensions: Optional[List[str]] = Field(alias="fileExtensions")
211
+ similar_contents: Optional[
212
+ List["GetConversationConversationFilterSimilarContents"]
213
+ ] = Field(alias="similarContents")
211
214
  contents: Optional[List["GetConversationConversationFilterContents"]]
212
215
  feeds: Optional[List["GetConversationConversationFilterFeeds"]]
213
216
  workflows: Optional[List["GetConversationConversationFilterWorkflows"]]
@@ -228,6 +231,10 @@ class GetConversationConversationFilterCreationDateRange(BaseModel):
228
231
  to: Optional[Any]
229
232
 
230
233
 
234
+ class GetConversationConversationFilterSimilarContents(BaseModel):
235
+ id: str
236
+
237
+
231
238
  class GetConversationConversationFilterContents(BaseModel):
232
239
  id: str
233
240
 
@@ -339,6 +346,9 @@ class GetConversationConversationAugmentedFilter(BaseModel):
339
346
  file_types: Optional[List[Optional[FileTypes]]] = Field(alias="fileTypes")
340
347
  formats: Optional[List[Optional[str]]]
341
348
  file_extensions: Optional[List[str]] = Field(alias="fileExtensions")
349
+ similar_contents: Optional[
350
+ List["GetConversationConversationAugmentedFilterSimilarContents"]
351
+ ] = Field(alias="similarContents")
342
352
  contents: Optional[List["GetConversationConversationAugmentedFilterContents"]]
343
353
  feeds: Optional[List["GetConversationConversationAugmentedFilterFeeds"]]
344
354
  workflows: Optional[List["GetConversationConversationAugmentedFilterWorkflows"]]
@@ -365,6 +375,10 @@ class GetConversationConversationAugmentedFilterCreationDateRange(BaseModel):
365
375
  to: Optional[Any]
366
376
 
367
377
 
378
+ class GetConversationConversationAugmentedFilterSimilarContents(BaseModel):
379
+ id: str
380
+
381
+
368
382
  class GetConversationConversationAugmentedFilterContents(BaseModel):
369
383
  id: str
370
384
 
@@ -9,6 +9,7 @@ from .base_model import BaseModel
9
9
  from .enums import (
10
10
  AnthropicModels,
11
11
  AzureOpenAIModels,
12
+ BedrockModels,
12
13
  CerebrasModels,
13
14
  CohereModels,
14
15
  ConversationSearchTypes,
@@ -78,6 +79,7 @@ class GetSpecificationSpecification(BaseModel):
78
79
  google: Optional["GetSpecificationSpecificationGoogle"]
79
80
  replicate: Optional["GetSpecificationSpecificationReplicate"]
80
81
  mistral: Optional["GetSpecificationSpecificationMistral"]
82
+ bedrock: Optional["GetSpecificationSpecificationBedrock"]
81
83
  groq: Optional["GetSpecificationSpecificationGroq"]
82
84
  cerebras: Optional["GetSpecificationSpecificationCerebras"]
83
85
  deepseek: Optional["GetSpecificationSpecificationDeepseek"]
@@ -222,6 +224,17 @@ class GetSpecificationSpecificationMistral(BaseModel):
222
224
  chunk_token_limit: Optional[int] = Field(alias="chunkTokenLimit")
223
225
 
224
226
 
227
+ class GetSpecificationSpecificationBedrock(BaseModel):
228
+ token_limit: Optional[int] = Field(alias="tokenLimit")
229
+ completion_token_limit: Optional[int] = Field(alias="completionTokenLimit")
230
+ model: BedrockModels
231
+ access_key: Optional[str] = Field(alias="accessKey")
232
+ secret_access_key: Optional[str] = Field(alias="secretAccessKey")
233
+ model_name: Optional[str] = Field(alias="modelName")
234
+ temperature: Optional[float]
235
+ probability: Optional[float]
236
+
237
+
225
238
  class GetSpecificationSpecificationGroq(BaseModel):
226
239
  token_limit: Optional[int] = Field(alias="tokenLimit")
227
240
  completion_token_limit: Optional[int] = Field(alias="completionTokenLimit")
@@ -14,6 +14,7 @@ from .enums import (
14
14
  AzureDocumentIntelligenceModels,
15
15
  AzureDocumentIntelligenceVersions,
16
16
  AzureOpenAIModels,
17
+ BedrockModels,
17
18
  CategoryFacetTypes,
18
19
  CerebrasModels,
19
20
  CohereModels,
@@ -146,6 +147,9 @@ class ContentCriteriaInput(BaseModel):
146
147
  )
147
148
  types: Optional[List[ContentTypes]] = None
148
149
  file_types: Optional[List[FileTypes]] = Field(alias="fileTypes", default=None)
150
+ similar_contents: Optional[List["EntityReferenceInput"]] = Field(
151
+ alias="similarContents", default=None
152
+ )
149
153
  contents: Optional[List["EntityReferenceInput"]] = None
150
154
  feeds: Optional[List["EntityReferenceInput"]] = None
151
155
  workflows: Optional[List["EntityReferenceInput"]] = None
@@ -399,7 +403,9 @@ class ProductFilter(BaseModel):
399
403
  manufacturer: Optional[str] = None
400
404
  brand: Optional[str] = None
401
405
  model: Optional[str] = None
402
- products: Optional[List["EntityReferenceFilter"]] = None
406
+ similar_products: Optional[List["EntityReferenceFilter"]] = Field(
407
+ alias="similarProducts", default=None
408
+ )
403
409
 
404
410
 
405
411
  class OneDriveFeedPropertiesInput(BaseModel):
@@ -777,7 +783,9 @@ class MedicalStudyFilter(BaseModel):
777
783
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
778
784
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
779
785
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
780
- studies: Optional[List["EntityReferenceFilter"]] = None
786
+ similar_studies: Optional[List["EntityReferenceFilter"]] = Field(
787
+ alias="similarStudies", default=None
788
+ )
781
789
 
782
790
 
783
791
  class H3IndexFilter(BaseModel):
@@ -845,6 +853,7 @@ class SpecificationInput(BaseModel):
845
853
  google: Optional["GoogleModelPropertiesInput"] = None
846
854
  replicate: Optional["ReplicateModelPropertiesInput"] = None
847
855
  mistral: Optional["MistralModelPropertiesInput"] = None
856
+ bedrock: Optional["BedrockModelPropertiesInput"] = None
848
857
  groq: Optional["GroqModelPropertiesInput"] = None
849
858
  cerebras: Optional["CerebrasModelPropertiesInput"] = None
850
859
  deepseek: Optional["DeepseekModelPropertiesInput"] = None
@@ -995,7 +1004,9 @@ class RepoFilter(BaseModel):
995
1004
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
996
1005
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
997
1006
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
998
- repos: Optional[List["EntityReferenceFilter"]] = None
1007
+ similar_repos: Optional[List["EntityReferenceFilter"]] = Field(
1008
+ alias="similarRepos", default=None
1009
+ )
999
1010
 
1000
1011
 
1001
1012
  class GoogleDriveFeedPropertiesInput(BaseModel):
@@ -1019,6 +1030,19 @@ class IntegrationConnectorUpdateInput(BaseModel):
1019
1030
  twitter: Optional["TwitterIntegrationPropertiesInput"] = None
1020
1031
 
1021
1032
 
1033
+ class BedrockModelPropertiesInput(BaseModel):
1034
+ model: BedrockModels
1035
+ model_name: Optional[str] = Field(alias="modelName", default=None)
1036
+ access_key: Optional[str] = Field(alias="accessKey", default=None)
1037
+ secret_access_key: Optional[str] = Field(alias="secretAccessKey", default=None)
1038
+ temperature: Optional[float] = None
1039
+ probability: Optional[float] = None
1040
+ token_limit: Optional[int] = Field(alias="tokenLimit", default=None)
1041
+ completion_token_limit: Optional[int] = Field(
1042
+ alias="completionTokenLimit", default=None
1043
+ )
1044
+
1045
+
1022
1046
  class MedicalDeviceFilter(BaseModel):
1023
1047
  search: Optional[str] = None
1024
1048
  order_by: Optional[OrderByTypes] = Field(alias="orderBy", default=None)
@@ -1039,7 +1063,9 @@ class MedicalDeviceFilter(BaseModel):
1039
1063
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
1040
1064
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
1041
1065
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
1042
- devices: Optional[List["EntityReferenceFilter"]] = None
1066
+ similar_devices: Optional[List["EntityReferenceFilter"]] = Field(
1067
+ alias="similarDevices", default=None
1068
+ )
1043
1069
 
1044
1070
 
1045
1071
  class MedicalContraindicationFilter(BaseModel):
@@ -1062,7 +1088,9 @@ class MedicalContraindicationFilter(BaseModel):
1062
1088
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
1063
1089
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
1064
1090
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
1065
- contraindications: Optional[List["EntityReferenceFilter"]] = None
1091
+ similar_contraindications: Optional[List["EntityReferenceFilter"]] = Field(
1092
+ alias="similarContraindications", default=None
1093
+ )
1066
1094
 
1067
1095
 
1068
1096
  class DiscordFeedPropertiesUpdateInput(BaseModel):
@@ -1082,6 +1110,19 @@ class AzureFileFeedPropertiesUpdateInput(BaseModel):
1082
1110
  prefix: Optional[str] = None
1083
1111
 
1084
1112
 
1113
+ class BedrockModelPropertiesUpdateInput(BaseModel):
1114
+ model: Optional[BedrockModels] = None
1115
+ model_name: Optional[str] = Field(alias="modelName", default=None)
1116
+ access_key: Optional[str] = Field(alias="accessKey", default=None)
1117
+ secret_access_key: Optional[str] = Field(alias="secretAccessKey", default=None)
1118
+ temperature: Optional[float] = None
1119
+ probability: Optional[float] = None
1120
+ token_limit: Optional[int] = Field(alias="tokenLimit", default=None)
1121
+ completion_token_limit: Optional[int] = Field(
1122
+ alias="completionTokenLimit", default=None
1123
+ )
1124
+
1125
+
1085
1126
  class PromptClassificationRuleInput(BaseModel):
1086
1127
  if_: Optional[str] = Field(alias="if", default=None)
1087
1128
  then: Optional[str] = None
@@ -1117,7 +1158,9 @@ class MedicalIndicationFilter(BaseModel):
1117
1158
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
1118
1159
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
1119
1160
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
1120
- indications: Optional[List["EntityReferenceFilter"]] = None
1161
+ similar_indications: Optional[List["EntityReferenceFilter"]] = Field(
1162
+ alias="similarIndications", default=None
1163
+ )
1121
1164
 
1122
1165
 
1123
1166
  class IndexingWorkflowJobInput(BaseModel):
@@ -1299,6 +1342,9 @@ class ConversationFilter(BaseModel):
1299
1342
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
1300
1343
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
1301
1344
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
1345
+ similar_conversations: Optional[List["EntityReferenceFilter"]] = Field(
1346
+ alias="similarConversations", default=None
1347
+ )
1302
1348
  conversations: Optional[List["EntityReferenceFilter"]] = None
1303
1349
 
1304
1350
 
@@ -1469,6 +1515,7 @@ class SpecificationUpdateInput(BaseModel):
1469
1515
  google: Optional["GoogleModelPropertiesUpdateInput"] = None
1470
1516
  replicate: Optional["ReplicateModelPropertiesUpdateInput"] = None
1471
1517
  mistral: Optional["MistralModelPropertiesUpdateInput"] = None
1518
+ bedrock: Optional["BedrockModelPropertiesUpdateInput"] = None
1472
1519
  groq: Optional["GroqModelPropertiesUpdateInput"] = None
1473
1520
  cerebras: Optional["CerebrasModelPropertiesUpdateInput"] = None
1474
1521
  deepseek: Optional["DeepseekModelPropertiesUpdateInput"] = None
@@ -1534,7 +1581,9 @@ class EventFilter(BaseModel):
1534
1581
  alias="isAccessibleForFree", default=None
1535
1582
  )
1536
1583
  typical_age_range: Optional[str] = Field(alias="typicalAgeRange", default=None)
1537
- events: Optional[List["EntityReferenceFilter"]] = None
1584
+ similar_events: Optional[List["EntityReferenceFilter"]] = Field(
1585
+ alias="similarEvents", default=None
1586
+ )
1538
1587
 
1539
1588
 
1540
1589
  class ClassificationWorkflowStageInput(BaseModel):
@@ -1605,7 +1654,9 @@ class MedicalGuidelineFilter(BaseModel):
1605
1654
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
1606
1655
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
1607
1656
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
1608
- guidelines: Optional[List["EntityReferenceFilter"]] = None
1657
+ similar_guidelines: Optional[List["EntityReferenceFilter"]] = Field(
1658
+ alias="similarGuidelines", default=None
1659
+ )
1609
1660
 
1610
1661
 
1611
1662
  class EmailFeedPropertiesInput(BaseModel):
@@ -1768,7 +1819,9 @@ class MedicalDrugFilter(BaseModel):
1768
1819
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
1769
1820
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
1770
1821
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
1771
- drugs: Optional[List["EntityReferenceFilter"]] = None
1822
+ similar_drugs: Optional[List["EntityReferenceFilter"]] = Field(
1823
+ alias="similarDrugs", default=None
1824
+ )
1772
1825
 
1773
1826
 
1774
1827
  class SlackIntegrationPropertiesInput(BaseModel):
@@ -1846,7 +1899,6 @@ class ContentFilter(BaseModel):
1846
1899
  location: Optional["PointFilter"] = None
1847
1900
  h_3: Optional["H3Filter"] = Field(alias="h3", default=None)
1848
1901
  boundaries: Optional[List[Optional[str]]] = None
1849
- contents: Optional[List["EntityReferenceFilter"]] = None
1850
1902
  in_last: Optional[Any] = Field(alias="inLast", default=None)
1851
1903
  original_date_range: Optional["DateRangeFilter"] = Field(
1852
1904
  alias="originalDateRange", default=None
@@ -1856,6 +1908,10 @@ class ContentFilter(BaseModel):
1856
1908
  file_size_range: Optional["Int64RangeFilter"] = Field(
1857
1909
  alias="fileSizeRange", default=None
1858
1910
  )
1911
+ similar_contents: Optional[List["EntityReferenceFilter"]] = Field(
1912
+ alias="similarContents", default=None
1913
+ )
1914
+ contents: Optional[List["EntityReferenceFilter"]] = None
1859
1915
  feeds: Optional[List["EntityReferenceFilter"]] = None
1860
1916
  workflows: Optional[List["EntityReferenceFilter"]] = None
1861
1917
  collections: Optional[List["EntityReferenceFilter"]] = None
@@ -2100,7 +2156,9 @@ class MedicalTherapyFilter(BaseModel):
2100
2156
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
2101
2157
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
2102
2158
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
2103
- therapies: Optional[List["EntityReferenceFilter"]] = None
2159
+ similar_therapies: Optional[List["EntityReferenceFilter"]] = Field(
2160
+ alias="similarTherapies", default=None
2161
+ )
2104
2162
 
2105
2163
 
2106
2164
  class ElevenLabsPublishingPropertiesInput(BaseModel):
@@ -2200,7 +2258,9 @@ class SoftwareFilter(BaseModel):
2200
2258
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
2201
2259
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
2202
2260
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
2203
- softwares: Optional[List["EntityReferenceFilter"]] = None
2261
+ similar_softwares: Optional[List["EntityReferenceFilter"]] = Field(
2262
+ alias="similarSoftwares", default=None
2263
+ )
2204
2264
 
2205
2265
 
2206
2266
  class AzureAIModelPropertiesUpdateInput(BaseModel):
@@ -2323,7 +2383,9 @@ class PersonFilter(BaseModel):
2323
2383
  phone_number: Optional[str] = Field(alias="phoneNumber", default=None)
2324
2384
  email: Optional[str] = None
2325
2385
  uri: Optional[Any] = None
2326
- persons: Optional[List["EntityReferenceFilter"]] = None
2386
+ similar_persons: Optional[List["EntityReferenceFilter"]] = Field(
2387
+ alias="similarPersons", default=None
2388
+ )
2327
2389
 
2328
2390
 
2329
2391
  class CerebrasModelPropertiesUpdateInput(BaseModel):
@@ -2654,7 +2716,9 @@ class MedicalDrugClassFilter(BaseModel):
2654
2716
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
2655
2717
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
2656
2718
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
2657
- classes: Optional[List["EntityReferenceFilter"]] = None
2719
+ similar_classes: Optional[List["EntityReferenceFilter"]] = Field(
2720
+ alias="similarClasses", default=None
2721
+ )
2658
2722
 
2659
2723
 
2660
2724
  class MedicalStudyInput(BaseModel):
@@ -2837,7 +2901,9 @@ class MedicalTestFilter(BaseModel):
2837
2901
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
2838
2902
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
2839
2903
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
2840
- tests: Optional[List["EntityReferenceFilter"]] = None
2904
+ similar_tests: Optional[List["EntityReferenceFilter"]] = Field(
2905
+ alias="similarTests", default=None
2906
+ )
2841
2907
 
2842
2908
 
2843
2909
  class IndexingWorkflowStageInput(BaseModel):
@@ -2924,7 +2990,9 @@ class MedicalProcedureFilter(BaseModel):
2924
2990
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
2925
2991
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
2926
2992
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
2927
- procedures: Optional[List["EntityReferenceFilter"]] = None
2993
+ similar_procedures: Optional[List["EntityReferenceFilter"]] = Field(
2994
+ alias="similarProcedures", default=None
2995
+ )
2928
2996
 
2929
2997
 
2930
2998
  class ExtractionWorkflowJobInput(BaseModel):
@@ -3065,7 +3133,9 @@ class MedicalConditionFilter(BaseModel):
3065
3133
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
3066
3134
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
3067
3135
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
3068
- conditions: Optional[List["EntityReferenceFilter"]] = None
3136
+ similar_conditions: Optional[List["EntityReferenceFilter"]] = Field(
3137
+ alias="similarConditions", default=None
3138
+ )
3069
3139
 
3070
3140
 
3071
3141
  class EmailMetadataInput(BaseModel):
@@ -3448,7 +3518,9 @@ class OrganizationFilter(BaseModel):
3448
3518
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
3449
3519
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
3450
3520
  uri: Optional[Any] = None
3451
- organizations: Optional[List["EntityReferenceFilter"]] = None
3521
+ similar_organizations: Optional[List["EntityReferenceFilter"]] = Field(
3522
+ alias="similarOrganizations", default=None
3523
+ )
3452
3524
 
3453
3525
 
3454
3526
  class SoftwareFacetInput(BaseModel):
@@ -3563,7 +3635,9 @@ class PlaceFilter(BaseModel):
3563
3635
  search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
3564
3636
  query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
3565
3637
  number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
3566
- places: Optional[List["EntityReferenceFilter"]] = None
3638
+ similar_places: Optional[List["EntityReferenceFilter"]] = Field(
3639
+ alias="similarPlaces", default=None
3640
+ )
3567
3641
 
3568
3642
 
3569
3643
  class H3Filter(BaseModel):
@@ -245,6 +245,7 @@ __all__ = [
245
245
  "QUERY_SLACK_CHANNELS_GQL",
246
246
  "QUERY_SOFTWARES_GQL",
247
247
  "QUERY_SPECIFICATIONS_GQL",
248
+ "QUERY_TOKENS_GQL",
248
249
  "QUERY_USAGE_GQL",
249
250
  "QUERY_USERS_GQL",
250
251
  "QUERY_WORKFLOWS_GQL",
@@ -396,6 +397,9 @@ query GetAlert($id: ID!, $correlationId: String) {
396
397
  fileTypes
397
398
  formats
398
399
  fileExtensions
400
+ similarContents {
401
+ id
402
+ }
399
403
  contents {
400
404
  id
401
405
  }
@@ -532,6 +536,9 @@ query QueryAlerts($filter: AlertFilter, $correlationId: String) {
532
536
  fileTypes
533
537
  formats
534
538
  fileExtensions
539
+ similarContents {
540
+ id
541
+ }
535
542
  contents {
536
543
  id
537
544
  }
@@ -3820,6 +3827,9 @@ query GetConversation($id: ID!, $correlationId: String) {
3820
3827
  fileTypes
3821
3828
  formats
3822
3829
  fileExtensions
3830
+ similarContents {
3831
+ id
3832
+ }
3823
3833
  contents {
3824
3834
  id
3825
3835
  }
@@ -3900,6 +3910,9 @@ query GetConversation($id: ID!, $correlationId: String) {
3900
3910
  fileTypes
3901
3911
  formats
3902
3912
  fileExtensions
3913
+ similarContents {
3914
+ id
3915
+ }
3903
3916
  contents {
3904
3917
  id
3905
3918
  }
@@ -4697,6 +4710,9 @@ query QueryConversations($filter: ConversationFilter, $correlationId: String) {
4697
4710
  fileTypes
4698
4711
  formats
4699
4712
  fileExtensions
4713
+ similarContents {
4714
+ id
4715
+ }
4700
4716
  contents {
4701
4717
  id
4702
4718
  }
@@ -4777,6 +4793,9 @@ query QueryConversations($filter: ConversationFilter, $correlationId: String) {
4777
4793
  fileTypes
4778
4794
  formats
4779
4795
  fileExtensions
4796
+ similarContents {
4797
+ id
4798
+ }
4780
4799
  contents {
4781
4800
  id
4782
4801
  }
@@ -7997,6 +8016,29 @@ query QueryCredits($startDate: DateTime!, $duration: TimeSpan!) {
7997
8016
  }
7998
8017
  """
7999
8018
 
8019
+ QUERY_TOKENS_GQL = """
8020
+ query QueryTokens($startDate: DateTime!, $duration: TimeSpan!) {
8021
+ tokens(startDate: $startDate, duration: $duration) {
8022
+ correlationId
8023
+ ownerId
8024
+ embeddingInputTokens
8025
+ embeddingModelServices
8026
+ completionInputTokens
8027
+ completionOutputTokens
8028
+ completionModelServices
8029
+ preparationInputTokens
8030
+ preparationOutputTokens
8031
+ preparationModelServices
8032
+ extractionInputTokens
8033
+ extractionOutputTokens
8034
+ extractionModelServices
8035
+ generationInputTokens
8036
+ generationOutputTokens
8037
+ generationModelServices
8038
+ }
8039
+ }
8040
+ """
8041
+
8000
8042
  QUERY_USAGE_GQL = """
8001
8043
  query QueryUsage($startDate: DateTime!, $duration: TimeSpan!, $names: [String!], $offset: Int, $limit: Int) {
8002
8044
  usage(
@@ -8452,6 +8494,16 @@ query GetSpecification($id: ID!, $correlationId: String) {
8452
8494
  probability
8453
8495
  chunkTokenLimit
8454
8496
  }
8497
+ bedrock {
8498
+ tokenLimit
8499
+ completionTokenLimit
8500
+ model
8501
+ accessKey
8502
+ secretAccessKey
8503
+ modelName
8504
+ temperature
8505
+ probability
8506
+ }
8455
8507
  groq {
8456
8508
  tokenLimit
8457
8509
  completionTokenLimit
@@ -8800,6 +8852,16 @@ query QuerySpecifications($filter: SpecificationFilter, $correlationId: String)
8800
8852
  probability
8801
8853
  chunkTokenLimit
8802
8854
  }
8855
+ bedrock {
8856
+ tokenLimit
8857
+ completionTokenLimit
8858
+ model
8859
+ accessKey
8860
+ secretAccessKey
8861
+ modelName
8862
+ temperature
8863
+ probability
8864
+ }
8803
8865
  groq {
8804
8866
  tokenLimit
8805
8867
  completionTokenLimit
@@ -67,6 +67,9 @@ class QueryAlertsAlertsResultsFilter(BaseModel):
67
67
  file_types: Optional[List[Optional[FileTypes]]] = Field(alias="fileTypes")
68
68
  formats: Optional[List[Optional[str]]]
69
69
  file_extensions: Optional[List[str]] = Field(alias="fileExtensions")
70
+ similar_contents: Optional[
71
+ List["QueryAlertsAlertsResultsFilterSimilarContents"]
72
+ ] = Field(alias="similarContents")
70
73
  contents: Optional[List["QueryAlertsAlertsResultsFilterContents"]]
71
74
  feeds: Optional[List["QueryAlertsAlertsResultsFilterFeeds"]]
72
75
  workflows: Optional[List["QueryAlertsAlertsResultsFilterWorkflows"]]
@@ -87,6 +90,10 @@ class QueryAlertsAlertsResultsFilterCreationDateRange(BaseModel):
87
90
  to: Optional[Any]
88
91
 
89
92
 
93
+ class QueryAlertsAlertsResultsFilterSimilarContents(BaseModel):
94
+ id: str
95
+
96
+
90
97
  class QueryAlertsAlertsResultsFilterContents(BaseModel):
91
98
  id: str
92
99
 
@@ -224,6 +224,9 @@ class QueryConversationsConversationsResultsFilter(BaseModel):
224
224
  file_types: Optional[List[Optional[FileTypes]]] = Field(alias="fileTypes")
225
225
  formats: Optional[List[Optional[str]]]
226
226
  file_extensions: Optional[List[str]] = Field(alias="fileExtensions")
227
+ similar_contents: Optional[
228
+ List["QueryConversationsConversationsResultsFilterSimilarContents"]
229
+ ] = Field(alias="similarContents")
227
230
  contents: Optional[List["QueryConversationsConversationsResultsFilterContents"]]
228
231
  feeds: Optional[List["QueryConversationsConversationsResultsFilterFeeds"]]
229
232
  workflows: Optional[List["QueryConversationsConversationsResultsFilterWorkflows"]]
@@ -252,6 +255,10 @@ class QueryConversationsConversationsResultsFilterCreationDateRange(BaseModel):
252
255
  to: Optional[Any]
253
256
 
254
257
 
258
+ class QueryConversationsConversationsResultsFilterSimilarContents(BaseModel):
259
+ id: str
260
+
261
+
255
262
  class QueryConversationsConversationsResultsFilterContents(BaseModel):
256
263
  id: str
257
264
 
@@ -373,6 +380,9 @@ class QueryConversationsConversationsResultsAugmentedFilter(BaseModel):
373
380
  file_types: Optional[List[Optional[FileTypes]]] = Field(alias="fileTypes")
374
381
  formats: Optional[List[Optional[str]]]
375
382
  file_extensions: Optional[List[str]] = Field(alias="fileExtensions")
383
+ similar_contents: Optional[
384
+ List["QueryConversationsConversationsResultsAugmentedFilterSimilarContents"]
385
+ ] = Field(alias="similarContents")
376
386
  contents: Optional[
377
387
  List["QueryConversationsConversationsResultsAugmentedFilterContents"]
378
388
  ]
@@ -405,6 +415,10 @@ class QueryConversationsConversationsResultsAugmentedFilterCreationDateRange(Bas
405
415
  to: Optional[Any]
406
416
 
407
417
 
418
+ class QueryConversationsConversationsResultsAugmentedFilterSimilarContents(BaseModel):
419
+ id: str
420
+
421
+
408
422
  class QueryConversationsConversationsResultsAugmentedFilterContents(BaseModel):
409
423
  id: str
410
424
 
@@ -9,6 +9,7 @@ from .base_model import BaseModel
9
9
  from .enums import (
10
10
  AnthropicModels,
11
11
  AzureOpenAIModels,
12
+ BedrockModels,
12
13
  CerebrasModels,
13
14
  CohereModels,
14
15
  ConversationSearchTypes,
@@ -86,6 +87,7 @@ class QuerySpecificationsSpecificationsResults(BaseModel):
86
87
  google: Optional["QuerySpecificationsSpecificationsResultsGoogle"]
87
88
  replicate: Optional["QuerySpecificationsSpecificationsResultsReplicate"]
88
89
  mistral: Optional["QuerySpecificationsSpecificationsResultsMistral"]
90
+ bedrock: Optional["QuerySpecificationsSpecificationsResultsBedrock"]
89
91
  groq: Optional["QuerySpecificationsSpecificationsResultsGroq"]
90
92
  cerebras: Optional["QuerySpecificationsSpecificationsResultsCerebras"]
91
93
  deepseek: Optional["QuerySpecificationsSpecificationsResultsDeepseek"]
@@ -230,6 +232,17 @@ class QuerySpecificationsSpecificationsResultsMistral(BaseModel):
230
232
  chunk_token_limit: Optional[int] = Field(alias="chunkTokenLimit")
231
233
 
232
234
 
235
+ class QuerySpecificationsSpecificationsResultsBedrock(BaseModel):
236
+ token_limit: Optional[int] = Field(alias="tokenLimit")
237
+ completion_token_limit: Optional[int] = Field(alias="completionTokenLimit")
238
+ model: BedrockModels
239
+ access_key: Optional[str] = Field(alias="accessKey")
240
+ secret_access_key: Optional[str] = Field(alias="secretAccessKey")
241
+ model_name: Optional[str] = Field(alias="modelName")
242
+ temperature: Optional[float]
243
+ probability: Optional[float]
244
+
245
+
233
246
  class QuerySpecificationsSpecificationsResultsGroq(BaseModel):
234
247
  token_limit: Optional[int] = Field(alias="tokenLimit")
235
248
  completion_token_limit: Optional[int] = Field(alias="completionTokenLimit")
@@ -0,0 +1,44 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: ./documents
3
+
4
+ from typing import List, Optional
5
+
6
+ from pydantic import Field
7
+
8
+ from .base_model import BaseModel
9
+
10
+
11
+ class QueryTokens(BaseModel):
12
+ tokens: Optional["QueryTokensTokens"]
13
+
14
+
15
+ class QueryTokensTokens(BaseModel):
16
+ correlation_id: Optional[str] = Field(alias="correlationId")
17
+ owner_id: Optional[str] = Field(alias="ownerId")
18
+ embedding_input_tokens: Optional[int] = Field(alias="embeddingInputTokens")
19
+ embedding_model_services: Optional[List[Optional[str]]] = Field(
20
+ alias="embeddingModelServices"
21
+ )
22
+ completion_input_tokens: Optional[int] = Field(alias="completionInputTokens")
23
+ completion_output_tokens: Optional[int] = Field(alias="completionOutputTokens")
24
+ completion_model_services: Optional[List[Optional[str]]] = Field(
25
+ alias="completionModelServices"
26
+ )
27
+ preparation_input_tokens: Optional[int] = Field(alias="preparationInputTokens")
28
+ preparation_output_tokens: Optional[int] = Field(alias="preparationOutputTokens")
29
+ preparation_model_services: Optional[List[Optional[str]]] = Field(
30
+ alias="preparationModelServices"
31
+ )
32
+ extraction_input_tokens: Optional[int] = Field(alias="extractionInputTokens")
33
+ extraction_output_tokens: Optional[int] = Field(alias="extractionOutputTokens")
34
+ extraction_model_services: Optional[List[Optional[str]]] = Field(
35
+ alias="extractionModelServices"
36
+ )
37
+ generation_input_tokens: Optional[int] = Field(alias="generationInputTokens")
38
+ generation_output_tokens: Optional[int] = Field(alias="generationOutputTokens")
39
+ generation_model_services: Optional[List[Optional[str]]] = Field(
40
+ alias="generationModelServices"
41
+ )
42
+
43
+
44
+ QueryTokens.model_rebuild()
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: graphlit-client
3
- Version: 1.0.20250423001
3
+ Version: 1.0.20250428001
4
4
  Summary: Graphlit API Python Client
5
5
  Home-page: https://github.com/graphlit/graphlit-client-python
6
6
  Author: Unstruk Data Inc.
@@ -1,12 +1,12 @@
1
1
  graphlit/__init__.py,sha256=4AyigTlFQWP40lnaaQ1H1iRT_B1hIXW9bgPanbwmTvs,32
2
2
  graphlit/graphlit.py,sha256=56G5Am72og5QHIjEh9cy4vkfos7YVpGvzRuIPhPbCEU,1949
3
- graphlit_api/__init__.py,sha256=nWEoolCar39jof6YP6SGGUQxkAwjMLMF2TM8M-sQHVI,188267
3
+ graphlit_api/__init__.py,sha256=7W_QO12-Eju3hYoDS8N_DSeY3twYqVsrNcl-WRbRl0c,189502
4
4
  graphlit_api/add_contents_to_collections.py,sha256=K7tNpLn8-lRVaVT39iKr-VtCKRWVONyL_h6cC0L606Y,888
5
5
  graphlit_api/ask_graphlit.py,sha256=F-mgXMfUqGYCOP4WFQXmZDp5RgK7Cej-nA1RkmRvp7o,6301
6
6
  graphlit_api/async_base_client.py,sha256=v0KUVwe2_RIQa8Mn7l_yD5McUe7B03vhclJ9SP4XGgw,12578
7
7
  graphlit_api/base_model.py,sha256=o2d-DixASFCGztr3rTiGX0AwgFu7Awr7EgD70FI8a-I,620
8
8
  graphlit_api/clear_conversation.py,sha256=5GOmc2wfupV-7EHWyi3v6LA0pSVLtFNCzxPJm42Dp6Y,531
9
- graphlit_api/client.py,sha256=qEFAYuk-3ugdImWf5qkU22sW9Cz4pPjNbYQ9SJMf1-k,200821
9
+ graphlit_api/client.py,sha256=JjuzYZmSui0sQGZZCIPv9qLAFliJLFmzb_TSOYtaPeI,201347
10
10
  graphlit_api/close_conversation.py,sha256=HcIUUiNf7hnuLZ7Fy6IcgfuHMSyWyJ7uOEy1EEETy_4,531
11
11
  graphlit_api/complete_conversation.py,sha256=9YmZZ6Gs7MkApNP1eu8tXG4wqOd3FWgPMF6BiJOGo3o,16468
12
12
  graphlit_api/continue_conversation.py,sha256=wy_ARPwt_mmmwianxaHCNvK53p_Sbeqkwnoo9d2nTco,16468
@@ -157,17 +157,17 @@ graphlit_api/disable_user.py,sha256=qfFXMq691gPVdFjF-mYzjcx6DIvZnsp5Nny5pl0qI9Q,
157
157
  graphlit_api/enable_alert.py,sha256=KctjIbnyZDoDynbuD6IR2fHuNHKAtM07VVgKzTL479g,398
158
158
  graphlit_api/enable_feed.py,sha256=BhtLlEGV9p8ICodywdN9-sASWSX3w_4c3zYtf-bdaGE,390
159
159
  graphlit_api/enable_user.py,sha256=_0rQnpwVYhBrRcnpfqrKTOiANgOMe-ibUAHmS-XC0qo,390
160
- graphlit_api/enums.py,sha256=9Fjc6z5j6geecv9XTYAWvSHdEu6JCf2Ql0z6mkIuvCM,27482
160
+ graphlit_api/enums.py,sha256=-NaXt2ArbkKKYdvZgxfJuI5OySbwliY7zKllqppjE54,27562
161
161
  graphlit_api/exceptions.py,sha256=33pJ_dDonmIx_D-43jvhuQYJAXz5HoaPf1DoxXmNsAw,2432
162
162
  graphlit_api/extract_contents.py,sha256=qsSy8UOJuqsUnCcuduaNPyaK2mZk_1noECT0Jgx3zRA,961
163
163
  graphlit_api/extract_text.py,sha256=jfkL6ep5wzXeoelZ5nRwNRT0t7oDCUt1xrigqkMj-7M,887
164
164
  graphlit_api/feed_exists.py,sha256=NlvWhqVaV-coIV_orR-XsaXhG0Gll3bPVS3tPF0zEt0,351
165
165
  graphlit_api/format_conversation.py,sha256=Bx9bN6zXF8CS7xjUcWz2D4nM6rrS7eKdVn_wHqW93IA,16082
166
- graphlit_api/get_alert.py,sha256=sYNx0_hvvhc1RxMBLkEFggP2idLnpGJG9bCYQY7tx00,6796
166
+ graphlit_api/get_alert.py,sha256=6cz0M1dGSgblAfAQSS7RrZW32XzudrhhCtHrzFYLsQ8,6985
167
167
  graphlit_api/get_category.py,sha256=r4BhKYOT49VeBrbNkAd8Hs8ndXnvUV0nPr5CurbI_Bk,439
168
168
  graphlit_api/get_collection.py,sha256=de2AaJQRkMEZoRhmYWnXlXlb1c76cF2Z8v6PwaL7wTk,830
169
169
  graphlit_api/get_content.py,sha256=nMUyT7FE52rcqZWee4voconMMAp3yHP820eQMwY3zHo,11061
170
- graphlit_api/get_conversation.py,sha256=q1CenM3uT_yYPBB5OlDaFCfmziDWXlsFe0Y1ryMl-xA,17057
170
+ graphlit_api/get_conversation.py,sha256=TZbFw32pzMrasT8b0nmbU42djXcGgW7zLLih5LEyZpk,17509
171
171
  graphlit_api/get_event.py,sha256=saVoCHle91eNLagCX8AZwcSDikEi9OSnImx-lGx3n9A,1523
172
172
  graphlit_api/get_feed.py,sha256=thNnL79fN00-hO5uJxMcDn3FYfQPuND_WsrQ4glVCkg,11843
173
173
  graphlit_api/get_label.py,sha256=0SmukTbSktKnVlx_FpVf3lxFOQVQ6RAiRj4kglnt3eI,418
@@ -190,7 +190,7 @@ graphlit_api/get_project.py,sha256=KDuQprWSINZrzhb_K1bnUY9MEdqhXl51x7uwclPCcJA,1
190
190
  graphlit_api/get_repo.py,sha256=4ngiYmVFEeKe7zK0daSppsbvRwXlwYpbB4HMU2xsl78,578
191
191
  graphlit_api/get_share_point_consent_uri.py,sha256=QaZxlq7Lkx29ryWk66F6ii_JRr7vao0xiObhK-1Ig3o,462
192
192
  graphlit_api/get_software.py,sha256=oFpWsAFCQfclVj7kdIzBxaIuKXavff4IP4jqVMDOzDI,696
193
- graphlit_api/get_specification.py,sha256=IQoSAuvStKHLms8InMHWb8lhL2EnpH-gCHXXDMXSsVE,10161
193
+ graphlit_api/get_specification.py,sha256=3zQhdMJ2bDffSYC0Gc_fJ5bISyUBFI9Nf8-ExCK7W_s,10713
194
194
  graphlit_api/get_user.py,sha256=pdfR920veI3hTBWqbIp8PP_1LCYi8KWTJ2Has8t8URM,2569
195
195
  graphlit_api/get_workflow.py,sha256=BWKzz03q51AmFnE0Cvxkm0WqIS2Es42x67r6KZxHuxo,14478
196
196
  graphlit_api/ingest_batch.py,sha256=pmO_rAZdG8dPid40h8lnTfKSa5r0EAOmFF7PIg3a_r4,2366
@@ -199,26 +199,26 @@ graphlit_api/ingest_memory.py,sha256=YF7sn_jvpk_iACg8encyp_gd0wrK0Om4blYzPDI-B8c
199
199
  graphlit_api/ingest_text.py,sha256=D4lpV9LTC_u586_ILVrB2rVpHG0-8HivqeOA1GpQuFs,2286
200
200
  graphlit_api/ingest_text_batch.py,sha256=gSD1bH3mAPwJzy5TeMJ6UguEgI9yrPUXyz1soExSttM,2521
201
201
  graphlit_api/ingest_uri.py,sha256=f71kMRyMoAhef6sJU85ZgGz4aPq_5CDLaDvCeQnLY5A,2248
202
- graphlit_api/input_types.py,sha256=WS4eFDzOcYF5kJCnlQuvRGYVPccKodpYoBsU5FZcBz0,141793
202
+ graphlit_api/input_types.py,sha256=A3eTGnqM_W2qCamjroK5u8jJCWRT1NIwNWVqfMBdQJQ,144583
203
203
  graphlit_api/is_content_done.py,sha256=X8uevsTD6oFMbC8I3E9Emg-_yrFTWnnrVL5LpruSB6Q,390
204
204
  graphlit_api/is_feed_done.py,sha256=-FQS2vtDMnNW75K_9jR6IUutvpwLmtoS5yY8wD17CaM,352
205
205
  graphlit_api/lookup_credits.py,sha256=WsV7fGbg29WWOjPRIaL2bnhISGsb0SqUlQxL7rBfNTo,1464
206
206
  graphlit_api/lookup_usage.py,sha256=D_cbO0KmXDqRYqQIhNwWXNFGjwNLEy2_5aVa-SYgRzw,1711
207
207
  graphlit_api/map_web.py,sha256=2rp4jFD1vDqcQ98mCVTeC0RzPqQxmmcRvHNRl8HJfFA,346
208
- graphlit_api/operations.py,sha256=2X6Ucld2Ll5MOSM0SHhPS2CjoArAKKFMIJPy0-eQoCk,185108
208
+ graphlit_api/operations.py,sha256=bbgw2cPI_Jtcsg3SpnYin0ptJBjLA-IDV9AFLeWIirk,186323
209
209
  graphlit_api/prompt.py,sha256=4A0P-8hk98sNFNfRovzDBXXCHAGXOs2zThsXY07Q5KI,6015
210
210
  graphlit_api/prompt_conversation.py,sha256=zdVzHhZdF2CFL87_8EUcugWYzSFSWc0nCv4JFjhN7lI,16082
211
211
  graphlit_api/prompt_specifications.py,sha256=z8JQpfCUfgPeiuolLEGY83Pr32VaI2Ng1Wq0pG-gv2w,6902
212
212
  graphlit_api/publish_contents.py,sha256=A584z9d98MTLhjRjQ5WCbfQXPN65V0433ws9HK4lVbE,5530
213
213
  graphlit_api/publish_conversation.py,sha256=KdJr0QD9pG6cpvZG7Ej5getOsGy1icLAxPv0d2ZkV10,4657
214
214
  graphlit_api/publish_text.py,sha256=DSKPRiEgrqkdJ2E_E9GPeEpnlj60lGconC1TlgCaUCQ,4435
215
- graphlit_api/query_alerts.py,sha256=vVihPXo7A5Sd8I4McuADzHbZbuEe9OWiHmpKHP1ad1E,7823
215
+ graphlit_api/query_alerts.py,sha256=7XgYquw5AEyiE6wJ_xCCGNZ55al1quAi4EOySO_EZDs,8034
216
216
  graphlit_api/query_categories.py,sha256=5AiZlgWRycLmr7-ID-EIoI2Q8ss6C5vGttr_ECKIG-w,636
217
217
  graphlit_api/query_collections.py,sha256=lQoUD4-ZiQYWa2ZeyOqVXokEA9IbkqJhrG56O0p8QVU,1075
218
218
  graphlit_api/query_contents.py,sha256=e5vWK6LR5BE_eNx2cmwjtopOWUb1bz_Z3OdxDeONPUA,11697
219
219
  graphlit_api/query_contents_facets.py,sha256=xKbOkq-F9-AYUulUQNPrR9PnQyq7M2j_u7fpQ8HYk4c,1368
220
220
  graphlit_api/query_contents_graph.py,sha256=RzQHZEQOOzB0yLTvU6rBJu_Str3Gc9m8BG8fY8xVrYY,975
221
- graphlit_api/query_conversations.py,sha256=rnCDtowAR0SSNCzqFTSQAcsfFshQyPI6F2cqRNaWA2o,19141
221
+ graphlit_api/query_conversations.py,sha256=s7q9vVRyvYGcYLo5neiRwFZJ8OF1DP0Avr5Lnn-XOQM,19637
222
222
  graphlit_api/query_credits.py,sha256=6uJqn4iYfBfsUtSMTCSuV4d2VTiYR4gR5tEeDH5Cf9o,1396
223
223
  graphlit_api/query_events.py,sha256=-YWXAy3hjSHlrIk9GtiKaPow3aWppE2XO60sToSjxSc,1716
224
224
  graphlit_api/query_feeds.py,sha256=lux15n2tARERMXgWwa-iuEvpv1MS_LcGmgc2Seq5Z60,12889
@@ -250,7 +250,8 @@ graphlit_api/query_share_point_folders.py,sha256=vu7XjVlAcZhIok3wvIvkzqbdxpcsbWB
250
250
  graphlit_api/query_share_point_libraries.py,sha256=u7EGJ91YnZ_MMIixtqJKu-2YhLl6ZyQtkeRLfac3XhI,961
251
251
  graphlit_api/query_slack_channels.py,sha256=QnG46uW5OIjh1S0nsXggNVKMAd2DqHc5JNXF9B5fd_8,421
252
252
  graphlit_api/query_softwares.py,sha256=WSPS-Q2Ol7bCqgqc5Q066wwa_afZEQESuLV1aganBkc,874
253
- graphlit_api/query_specifications.py,sha256=9C7TwJGZbrDB-g7yfmw49Cty9zvRdwES51kfffd326k,10850
253
+ graphlit_api/query_specifications.py,sha256=3rszim0N4hrCfr8R6GCAUySw5p4g85_dpLJtzhqBpgo,11424
254
+ graphlit_api/query_tokens.py,sha256=8lvoWuoTOGVxiJT089EgzwUt78UPc1dcvN1rMveO07M,1739
254
255
  graphlit_api/query_usage.py,sha256=VUKzjpaZjkcQFXJlEKIQT9I87HTgcYzX1CeimAzqeIM,1645
255
256
  graphlit_api/query_users.py,sha256=b50tNVR79XOcVqAGjxcjEsHe0Ka9rhXJzJHrEfaCP7Y,2965
256
257
  graphlit_api/query_workflows.py,sha256=AqkmFsfGCrXqN4jAHBqLkwupqRyOSqLIlf2QBB6Qfx0,16250
@@ -302,8 +303,8 @@ graphlit_api/upsert_label.py,sha256=_bVWrISvyt4G4IcjAKqu8c5P6FDgaODdIGtSToJfNOY,
302
303
  graphlit_api/upsert_specification.py,sha256=23eLTL8OLAYE-j_nhjT5NgaCrSUs9Q40rGW_VhDrDoM,643
303
304
  graphlit_api/upsert_workflow.py,sha256=QL5Gi8jqF4wG9_qReoW8E5X7VsSIsLzgxIY1aAK0ebo,15692
304
305
  graphlit_api/workflow_exists.py,sha256=1XVcqCW_KZ3BwUFx08lwqQdf1ZpJ6Vmi8jBqcrMqYRI,397
305
- graphlit_client-1.0.20250423001.dist-info/licenses/LICENSE,sha256=ivF8XnUYrNZFQ1wZFMrxWshDb1h7TdSK6Qk8_3WPkhM,1095
306
- graphlit_client-1.0.20250423001.dist-info/METADATA,sha256=sYjhcwrfDPNLvcgsKJuYD5xjfuQcSAYvNzlHhVB27Nw,3408
307
- graphlit_client-1.0.20250423001.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
308
- graphlit_client-1.0.20250423001.dist-info/top_level.txt,sha256=HUVfNzJrxWuHS-4M5I7XjLa8-mxYQwfx01A4YKJZSYM,22
309
- graphlit_client-1.0.20250423001.dist-info/RECORD,,
306
+ graphlit_client-1.0.20250428001.dist-info/licenses/LICENSE,sha256=ivF8XnUYrNZFQ1wZFMrxWshDb1h7TdSK6Qk8_3WPkhM,1095
307
+ graphlit_client-1.0.20250428001.dist-info/METADATA,sha256=durRTygATQO-gvG9w0NFpR-I476imgtllDSrdp4IcfY,3408
308
+ graphlit_client-1.0.20250428001.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
309
+ graphlit_client-1.0.20250428001.dist-info/top_level.txt,sha256=HUVfNzJrxWuHS-4M5I7XjLa8-mxYQwfx01A4YKJZSYM,22
310
+ graphlit_client-1.0.20250428001.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (79.0.1)
2
+ Generator: setuptools (80.0.0)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5