bump production demo

Author: aboutphilippe

production_demo/pyproject.toml

@@ -7,7 +7,7 @@ requires-python = ">=3.10,<3.14"
 readme = "README.md"
 dependencies = [
     "pydantic>=2.10.6",
-    "restack-ai==0.0.91",
+    "restack-ai==0.0.92",
     "watchfiles>=1.0.4",
     "python-dotenv==1.0.1",
     "openai>=1.61.0",

production_demo/src/functions/evaluate.py

@@ -1,4 +1,4 @@
-from restack_ai.function import function, FunctionFailure, log
+from restack_ai.function import function, NonRetryableError, log
 from openai import OpenAI
 from pydantic import BaseModel
 
@@ -11,7 +11,7 @@ async def llm_evaluate(input: EvaluateInput) -> str:
         client = OpenAI(base_url="http://192.168.205.1:1234/v1/",api_key="llmstudio")
     except Exception as e:
         log.error(f"Failed to create LLM client {e}")
-        raise FunctionFailure(f"Failed to create OpenAI client {e}", non_retryable=True) from e
+        raise NonRetryableError(message=f"Failed to create OpenAI client {e}") from e
 
     prompt = (
         f"Evaluate the following joke for humor, creativity, and originality. "

production_demo/src/functions/function.py

@@ -1,4 +1,4 @@
-from restack_ai.function import function, log, FunctionFailure
+from restack_ai.function import function, log, NonRetryableError
 
 tries = 0
 
@@ -14,7 +14,7 @@ async def example_function(input: ExampleFunctionInput) -> str:
 
         if tries == 0:
             tries += 1
-            raise FunctionFailure(message="Simulated failure", non_retryable=False)
+            raise NonRetryableError(message="Simulated failure")
 
         log.info("example function started", input=input)
         return f"Hello, {input.name}!"

production_demo/src/functions/generate.py

@@ -1,4 +1,4 @@
-from restack_ai.function import function, FunctionFailure, log
+from restack_ai.function import function, NonRetryableError, log
 from openai import OpenAI
 
 from pydantic import BaseModel
@@ -13,7 +13,7 @@ async def llm_generate(input: GenerateInput) -> str:
         client = OpenAI(base_url="http://192.168.205.1:1234/v1/",api_key="llmstudio")
     except Exception as e:
         log.error(f"Failed to create LLM client {e}")
-        raise FunctionFailure(f"Failed to create OpenAI client {e}", non_retryable=True) from e
+        raise NonRetryableError(message=f"Failed to create OpenAI client {e}") from e
 
     try:
         response = client.chat.completions.create(

production_demo/src/workflows/child.py

@@ -15,7 +15,7 @@ class ChildWorkflow:
     @workflow.run
     async def run(self, input: ChildWorkflowInput):
         log.info("ChildWorkflow started")
-        await workflow.step(example_function, input=ExampleFunctionInput(name='John Doe'), start_to_close_timeout=timedelta(minutes=2))
+        await workflow.step(function=example_function, function_input=ExampleFunctionInput(name='John Doe'), start_to_close_timeout=timedelta(minutes=2))
 
         await workflow.sleep(1)