unitypredict 1.1.79__tar.gz → 1.1.80__tar.gz

{unitypredict-1.1.79 → unitypredict-1.1.80}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unitypredict
-Version: 1.1.79
+Version: 1.1.80
 Summary:  
 Home-page: https://unitypredict.com
 Author: UnityPredict

{unitypredict-1.1.79 → unitypredict-1.1.80}/unitypredict/UnityPredictClient.py

@@ -57,19 +57,21 @@ class UnityPredictRequest:
         InputValues (dict): A dictionary of input values for the model, where the key is the input variable name and the value of the variable's value.
         DesiredOutcomes (list): A list of desired output variable names.
         OutputFolderPath (str): The target path for saving model's output files (this is optional/not used if the model doesn't produce any output files).
+        CallbackUrl (str): (Optional) consumer-provided endpoint that will be involved (POST) with a copy of the InferenceResponse. Note: endpoint must be publicly accessible.
     """
 
     ContextId: str = ''
     InputValues: dict
     DesiredOutcomes: list
     OutputFolderPath: str
+    CallbackUrl: str = ''
 
-    def __init__(self, ContextId='', InputValues={}, DesiredOutcomes=[], OutputFolderPath=""):
+    def __init__(self, ContextId='', InputValues={}, DesiredOutcomes=[], OutputFolderPath="", CallbackUrl=''):
         self.ContextId = ContextId 
         self.InputValues = InputValues 
         self.DesiredOutcomes = DesiredOutcomes 
         self.OutputFolderPath = OutputFolderPath 
-
+        self.CallbackUrl = CallbackUrl
     
 class UnityPredictResponse:
 
@@ -87,7 +89,10 @@ class UnityPredictResponse:
 
     ContextId: str = ''
     RequestId: str = ''
+    LogMessages: str = ''
     ErrorMessages: str = ''
+    ComputeTime: str = ''
+    ComputeTimeCharged: str = ''
     ComputeCost: float = 0.0
     Outcomes: dict = {}
     Status: str|None = None
@@ -108,7 +113,7 @@ class UnityPredictClient:
     ApiMaxTimeout: int = 12 * 60 * 60
 
 
-    def __init__(self, apiKey, apiMaxTimeoutInSec = 12 * 60 * 60):
+    def __init__(self, apiKey, apiMaxTimeoutInSec = 12 * 60 * 60, apiEnv: str = 'prod', apiCustomUrl: str = ''):
         
         """
         Initializes a UnityPredictClient instance.
@@ -117,6 +122,18 @@ class UnityPredictClient:
             apiKey (str): The UnityPredict API key for authentication.
             apiMaxTimeoutInSec (int, optional): The maximum wait time (in seconds) for synchronous responses. Defaults to 12 hours.
         """
+
+        if apiEnv.casefold() == 'prod':
+            self.ApiBaseUrl = 'https://api.prod.unitypredict.com/api'
+        elif apiEnv.casefold() == 'dev':
+            self.ApiBaseUrl = 'https://api.dev.unitypredict.net/api'
+        elif apiEnv.casefold() == 'custom':
+            if apiCustomUrl == '':
+                raise ValueError(f"Invalid API URL: {apiCustomUrl}")
+            self.ApiBaseUrl = apiCustomUrl
+        else:
+            raise ValueError(f"Invalid API environment: {apiEnv}")
+
         self.ApiKey = f"APIKEY@{apiKey}"
         self.ApiMaxTimeout = apiMaxTimeoutInSec
 
@@ -232,10 +249,13 @@ class UnityPredictClient:
 
         try:
             results.Status = finalResponseJson.get('status')
+            results.ComputeTime = finalResponseJson.get('computeTime')
+            results.ComputeTimeCharged = finalResponseJson.get('computeTimeCharged')
             results.ComputeCost = finalResponseJson.get('computeCost')
             results.Outcomes = outcomes
             results.RequestId = finalResponseRequestId
             results.ContextId = finalResponseJson.get('contextId')
+            results.LogMessages = finalResponseJson.get('logMessages')
             results.ErrorMessages = finalResponseJson.get('errorMessages')
 
         except Exception as e:

{unitypredict-1.1.79 → unitypredict-1.1.80}/unitypredict.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unitypredict
-Version: 1.1.79
+Version: 1.1.80
 Summary:  
 Home-page: https://unitypredict.com
 Author: UnityPredict