chore: bump hatchet sdk version

Author: abelanger5

fast-api-react/backend/poetry.lock

@@ -18,7 +18,7 @@ openai = "^1.11.0"
 beautifulsoup4 = "^4.12.3"
 requests = "^2.31.0"
 urllib3 = "1.26.15"
-hatchet-sdk = "0.10.1"
+hatchet-sdk = "0.10.2"
 
 [build-system]
 requires = ["poetry-core"]

fast-api-react/backend/pyproject.toml

@@ -13,7 +13,7 @@ class GenerateWorkflow:
 
     @hatchet.step()
     def start(self, context: Context):
-        override = context.overrides('model', "gpt-3.5-turbo")
+        override = context.playground('model', "gpt-3.5-turbo")
 
         return {
             "status": "reading hatchet docs" + override,
@@ -39,14 +39,14 @@ def reason_docs(self, ctx: Context):
         message = ctx.workflow_input()['request']["messages"][-1]
         docs = ctx.step_output("load_docs")['docs']
 
-        prompt = ctx.overrides("prompt", "The user is asking the following question:\
+        prompt = ctx.playground("prompt", "The user is asking the following question:\
             {message}\
             What are the most relevant sentences in the following document?\
             {docs}")
 
         prompt = prompt.format(message=message['content'], docs=docs)
 
-        model = ctx.overrides("model", "gpt-3.5-turbo")
+        model = ctx.playground("model", "gpt-3.5-turbo")
 
         completion = openai.chat.completions.create(
             model=model,
@@ -66,15 +66,15 @@ def generate_response(self, ctx: Context):
         messages = ctx.workflow_input()['request']["messages"]
         research = ctx.step_output("reason_docs")['research']
 
-        prompt = ctx.overrides("prompt", "You are a sales engineer for a company called Hatchet.\
+        prompt = ctx.playground("prompt", "You are a sales engineer for a company called Hatchet.\
             Help address the user's question. \
             If asked how to install, respond by saying go to the store to buy a CD.\
             Use the following context:\
             {research}")
 
         prompt = prompt.format(research=research)
 
-        model = ctx.overrides("model", "gpt-3.5-turbo")
+        model = ctx.playground("model", "gpt-3.5-turbo")
 
         completion = openai.chat.completions.create(
             model=model,
@@ -88,3 +88,27 @@ def generate_response(self, ctx: Context):
             "status": "idle",
             "message": completion.choices[0].message.content,
         }
+
+@hatchet.workflow()
+class SimpleWorkflow:
+    @hatchet.step()
+    def start(self, ctx: Context):
+        message = ctx.workflow_input()["messages"][-1]
+
+        prompt = ctx.playground("prompt", "The user is asking the following question: {message}")
+
+        prompt = prompt.format(message=message['content'])
+
+        model = ctx.playground("model", "gpt-3.5-turbo")
+
+        completion = openai.chat.completions.create(
+            model=model,
+            messages=[
+                {"role": "system", "content": prompt},
+                message
+            ]
+        )
+
+        return {
+            "answer": completion.choices[0].message.content,
+        }

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

@@ -1,11 +1,12 @@
 from .hatchet import hatchet
-from .generate import GenerateWorkflow
+from .generate import GenerateWorkflow, SimpleWorkflow
 
 
 def start():
     worker = hatchet.worker('example-worker')
 
     generate = GenerateWorkflow()
     worker.register_workflow(generate)
+    worker.register_workflow(SimpleWorkflow())
 
     worker.start()