Spaces:
Running
Running
Kate Forsberg
commited on
Commit
·
f787fa1
1
Parent(s):
49630fa
504 error?
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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)
|