Kate Forsberg commited on
Commit
f787fa1
·
1 Parent(s): 49630fa

504 error?

Browse files
Files changed (1) hide show
  1. app.py +13 -2
app.py CHANGED
@@ -2,7 +2,8 @@ import gradio as gr
2
  from uuid import uuid4 as uuid
3
  from huggingface_hub import HfApi
4
  from typing import Any
5
- from griptape.structures import Agent
 
6
  from griptape.drivers import LocalConversationMemoryDriver
7
  from griptape.memory.structure import ConversationMemory
8
  from griptape.tools import Calculator
@@ -41,7 +42,17 @@ agent = Agent(
41
  ) ]
42
  )
43
 
44
- gr.State()
 
 
 
 
 
 
 
 
 
 
45
 
46
  def send_message(message:str, history,) -> Any:
47
  response = agent.run(message)
 
2
  from uuid import uuid4 as uuid
3
  from huggingface_hub import HfApi
4
  from typing import Any
5
+ from griptape.structures import Agent, Workflow
6
+ from griptape.tasks import PromptTask
7
  from griptape.drivers import LocalConversationMemoryDriver
8
  from griptape.memory.structure import ConversationMemory
9
  from griptape.tools import Calculator
 
42
  ) ]
43
  )
44
 
45
+ workflow = Workflow()
46
+ start_task = PromptTask("I will provide you two dog breeds to compare", id="START")
47
+ end_task = PromptTask("How are the dog breeds different", id="END")
48
+ workflow.add_task(start_task)
49
+ workflow.add_task(end_task)
50
+ dog_task = PromptTask("Poodle and a Labrador", id="DOG")
51
+ workflow.insert_tasks(start_task,[dog_task],end_task)
52
+
53
+ def use_workflow(message:str, history) -> Any:
54
+ response = workflow.run()
55
+ return response.output.value
56
 
57
  def send_message(message:str, history,) -> Any:
58
  response = agent.run(message)