Merge branch 'main' of github.com:hatchet-dev/hatchet-python-quickstart into main

Author: abelanger5

fast-api-react/backend/src/workflows/simple.py

@@ -1,5 +1,6 @@
 from .hatchet import hatchet
 from hatchet_sdk import Context
+import openai
 
 
 @hatchet.workflow()

simple-examples/src/genai/simple.py

@@ -1,28 +1,32 @@
 from ..hatchet import hatchet
 from hatchet_sdk import Context
+import openai
 
-# This is a simple example of a workflow that has 3 steps.
+# This is a simple example of a workflow that uses the OpenAI GPT-3 model to generate an answer to a question.
 # The workflow is declared decorated with `@hatchet_workflow` and the steps are declared with `@hatchet_step`.
 
 
-@hatchet.workflow(on_events=["simpleai:create"])
+@hatchet.workflow()
 class SimpleGenAiWorkflow:
-
     @hatchet.step()
-    def step1(self, context: Context):
-        # The context object is passed to each step and contains the input data for the workflow.
-        messages = context.workflow_input()['request']['messages']
-        print("> starting step1", messages)
+    def start(self, ctx: Context):
+        message = ctx.workflow_input()["messages"][-1]
+
+        prompt = ctx.playground(
+            "prompt", "The user is asking the following question: {message}")
 
-        return {"status": "thinking"}
+        prompt = prompt.format(message=message['content'])
 
-    @hatchet.step(parents=["step1"])
-    def step2(self, context: Context):
-        print("starting step2")
+        model = ctx.playground("model", "gpt-3.5-turbo")
 
-        return {"status": "writing a response"}
+        completion = openai.chat.completions.create(
+            model=model,
+            messages=[
+                {"role": "system", "content": prompt},
+                message
+            ]
+        )
 
-    @hatchet.step(parents=["step2"], timeout='5m')
-    def step3(self, context: Context):
-        messages = context.workflow_input()['request']['messages']
-        return {"complete": "true", "status": "idle", "message": "response from step3"}
+        return {
+            "answer": completion.choices[0].message.content,
+        }