Spaces:
Sleeping
Sleeping
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)
|