Updated to latest API release.

Author: kirk-marple

graphlit_api/__init__.py

@@ -725,6 +725,7 @@
     GetSpecification,
     GetSpecificationSpecification,
     GetSpecificationSpecificationAnthropic,
+    GetSpecificationSpecificationAzureAi,
     GetSpecificationSpecificationAzureOpenAi,
     GetSpecificationSpecificationCerebras,
     GetSpecificationSpecificationCohere,
@@ -1709,6 +1710,7 @@
     QuerySpecificationsSpecifications,
     QuerySpecificationsSpecificationsResults,
     QuerySpecificationsSpecificationsResultsAnthropic,
+    QuerySpecificationsSpecificationsResultsAzureAi,
     QuerySpecificationsSpecificationsResultsAzureOpenAi,
     QuerySpecificationsSpecificationsResultsCerebras,
     QuerySpecificationsSpecificationsResultsCohere,
@@ -2729,6 +2731,7 @@
     "GetSpecification",
     "GetSpecificationSpecification",
     "GetSpecificationSpecificationAnthropic",
+    "GetSpecificationSpecificationAzureAi",
     "GetSpecificationSpecificationAzureOpenAi",
     "GetSpecificationSpecificationCerebras",
     "GetSpecificationSpecificationCohere",
@@ -3346,6 +3349,7 @@
     "QuerySpecificationsSpecifications",
     "QuerySpecificationsSpecificationsResults",
     "QuerySpecificationsSpecificationsResultsAnthropic",
+    "QuerySpecificationsSpecificationsResultsAzureAi",
     "QuerySpecificationsSpecificationsResultsAzureOpenAi",
     "QuerySpecificationsSpecificationsResultsCerebras",
     "QuerySpecificationsSpecificationsResultsCohere",

graphlit_api/enums.py

@@ -603,6 +603,7 @@ class MistralModels(str, Enum):
     MISTRAL_SMALL = "MISTRAL_SMALL"
     MISTRAL_MEDIUM = "MISTRAL_MEDIUM"
     MISTRAL_LARGE = "MISTRAL_LARGE"
+    PIXTRAL_12B_2409 = "PIXTRAL_12B_2409"
     CUSTOM = "CUSTOM"
 
 

graphlit_api/get_specification.py

@@ -64,6 +64,7 @@ class GetSpecificationSpecification(BaseModel):
     revision_strategy: Optional["GetSpecificationSpecificationRevisionStrategy"] = (
         Field(alias="revisionStrategy")
     )
+    azure_ai: Optional["GetSpecificationSpecificationAzureAi"] = Field(alias="azureAI")
     open_ai: Optional["GetSpecificationSpecificationOpenAi"] = Field(alias="openAI")
     azure_open_ai: Optional["GetSpecificationSpecificationAzureOpenAi"] = Field(
         alias="azureOpenAI"
@@ -118,6 +119,15 @@ class GetSpecificationSpecificationRevisionStrategy(BaseModel):
     count: Optional[int]
 
 
+class GetSpecificationSpecificationAzureAi(BaseModel):
+    token_limit: int = Field(alias="tokenLimit")
+    completion_token_limit: Optional[int] = Field(alias="completionTokenLimit")
+    key: str
+    endpoint: Any
+    temperature: Optional[float]
+    probability: Optional[float]
+
+
 class GetSpecificationSpecificationOpenAi(BaseModel):
     token_limit: Optional[int] = Field(alias="tokenLimit")
     completion_token_limit: Optional[int] = Field(alias="completionTokenLimit")

graphlit_api/operations.py

@@ -5162,6 +5162,14 @@
       customRevision
       count
     }
+    azureAI {
+      tokenLimit
+      completionTokenLimit
+      key
+      endpoint
+      temperature
+      probability
+    }
     openAI {
       tokenLimit
       completionTokenLimit
@@ -5441,6 +5449,14 @@
         customRevision
         count
       }
+      azureAI {
+        tokenLimit
+        completionTokenLimit
+        key
+        endpoint
+        temperature
+        probability
+      }
       openAI {
         tokenLimit
         completionTokenLimit

graphlit_api/query_specifications.py

@@ -68,6 +68,9 @@ class QuerySpecificationsSpecificationsResults(BaseModel):
     revision_strategy: Optional[
         "QuerySpecificationsSpecificationsResultsRevisionStrategy"
     ] = Field(alias="revisionStrategy")
+    azure_ai: Optional["QuerySpecificationsSpecificationsResultsAzureAi"] = Field(
+        alias="azureAI"
+    )
     open_ai: Optional["QuerySpecificationsSpecificationsResultsOpenAi"] = Field(
         alias="openAI"
     )
@@ -124,6 +127,15 @@ class QuerySpecificationsSpecificationsResultsRevisionStrategy(BaseModel):
     count: Optional[int]
 
 
+class QuerySpecificationsSpecificationsResultsAzureAi(BaseModel):
+    token_limit: int = Field(alias="tokenLimit")
+    completion_token_limit: Optional[int] = Field(alias="completionTokenLimit")
+    key: str
+    endpoint: Any
+    temperature: Optional[float]
+    probability: Optional[float]
+
+
 class QuerySpecificationsSpecificationsResultsOpenAi(BaseModel):
     token_limit: Optional[int] = Field(alias="tokenLimit")
     completion_token_limit: Optional[int] = Field(alias="completionTokenLimit")