unitypredict-engines 1.1.76__tar.gz → 1.1.78__tar.gz

{unitypredict_engines-1.1.76 → unitypredict_engines-1.1.78}/PKG-INFO

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

{unitypredict_engines-1.1.76 → unitypredict_engines-1.1.78}/unitypredict_engines/Platform.py

@@ -466,7 +466,7 @@ class IPlatform:
         raise NotImplementedError
 
     @abstractmethod
-    def invokeUnityPredictModel(self, modelId: str, request: ChainedInferenceRequest, waitForResponse: bool = True) -> ChainedInferenceResponse: 
+    def invokeUnityPredictModel(self, modelId: str, request: ChainedInferenceRequest, waitForResponse: bool = True, timeout: int = None) -> ChainedInferenceResponse: 
         """
         Invokes another UnityPredict model from the current engine, enabling model chaining.
 
@@ -484,13 +484,16 @@ class IPlatform:
             waitForResponse (bool, optional): If True, the method will wait for the inference to complete
                 before returning. If False, it will return immediately with a response containing the
                 status URL that can be used to check the inference progress. Defaults to True.
+            timeout (int, optional): The maximum time in seconds to wait for the inference to complete
+                when waitForResponse is True. If None, the method will wait indefinitely. Defaults to None.
 
         Returns:
             ChainedInferenceResponse: The response from the chained model, including outcomes,
                 errors, and compute costs. If waitForResponse is False and the inference is still
                 processing, the response will contain a status URL that can be used to check the
                 inference progress. In this case, you can use checkChainedInferenceJobStatus to
-                periodically check the status of the inference until it completes.
+                periodically check the status of the inference until it completes. If a timeout
+                occurs, the response will have status 'Error' with an appropriate error message.
 
         Note:
             When waitForResponse is False, you can use checkChainedInferenceJobStatus to poll

{unitypredict_engines-1.1.76 → unitypredict_engines-1.1.78}/unitypredict_engines/unitypredictcli.py

@@ -351,6 +351,9 @@ class UnityPredictCli:
         apiKey = self._uptApiKeyDict[uptProfile]["UPT_API_KEY"]
         endpointHeader = {"Authorization": f"Bearer {self._uptApiKeyHeaderPrefix}{apiKey}"}
 
+        if self._uptVerbose:
+            print (f"Fetching upload url for the file: {fileName} of type {fileType}: {fileToUploadUrl} ...")
+
         try:
             response = requests.get(fileToUploadUrl, headers=endpointHeader, timeout=self._uptResponseTimeout)
             
@@ -584,7 +587,7 @@ class UnityPredictCli:
         # Upload source files
         mainFile: str = "main.py"
         excludedExtensions: list = [".env", ".md", ".gitignore", ".DS_Store", "Thumbs.db"]
-        excludedFiles: list = ["config.json"]
+        excludedFiles: list = ["config.json", "main.py", "AdditionalDockerCommands.txt", "requirements.txt", "DebugLogs.txt"]
 
         fileUploadList: list = []
 
@@ -602,6 +605,7 @@ class UnityPredictCli:
             
             # Check if file has an excluded extension
             fileExtension = os.path.splitext(items)[1].lower()
+
             if fileExtension in excludedExtensions:
                 continue
 

{unitypredict_engines-1.1.76 → unitypredict_engines-1.1.78}/unitypredict_engines.egg-info/PKG-INFO

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