Spaces:
Sleeping
Sleeping
Kate Forsberg
commited on
Commit
·
36d83b1
1
Parent(s):
318ef9e
deletes memory if someone says done
Browse files
app.py
CHANGED
@@ -1,19 +1,18 @@
|
|
1 |
import glob
|
2 |
import gradio as gr
|
3 |
-
from uuid import uuid4 as uuid
|
4 |
-
from huggingface_hub import HfApi
|
5 |
from typing import Any
|
6 |
from dotenv import load_dotenv
|
7 |
from griptape.structures import Agent
|
8 |
-
from griptape.tasks import PromptTask
|
9 |
-
from griptape.drivers import LocalConversationMemoryDriver, GriptapeCloudStructureRunDriver,
|
10 |
from griptape.memory.structure import ConversationMemory
|
11 |
-
from griptape.tools import StructureRunClient,
|
12 |
from griptape.rules import Rule, Ruleset
|
13 |
from griptape.config import AnthropicStructureConfig
|
14 |
-
from griptape.events import EventListener, FinishStructureRunEvent
|
15 |
import time
|
16 |
import os
|
|
|
|
|
17 |
|
18 |
|
19 |
#Load environment variables
|
@@ -154,6 +153,9 @@ def build_agent(session_id:str,message:str) -> Agent:
|
|
154 |
rulesets=[ruleset],
|
155 |
)
|
156 |
|
|
|
|
|
|
|
157 |
|
158 |
|
159 |
def send_message(message:str, history, request:gr.Request) -> Any:
|
@@ -161,6 +163,8 @@ def send_message(message:str, history, request:gr.Request) -> Any:
|
|
161 |
session_hash = request.session_hash
|
162 |
agent = build_agent(session_hash,message)
|
163 |
response = agent.run(message)
|
|
|
|
|
164 |
return response.output.value
|
165 |
|
166 |
demo = gr.ChatInterface(
|
|
|
1 |
import glob
|
2 |
import gradio as gr
|
|
|
|
|
3 |
from typing import Any
|
4 |
from dotenv import load_dotenv
|
5 |
from griptape.structures import Agent
|
6 |
+
from griptape.tasks import PromptTask
|
7 |
+
from griptape.drivers import LocalConversationMemoryDriver, GriptapeCloudStructureRunDriver, LocalFileManagerDriver, LocalStructureRunDriver
|
8 |
from griptape.memory.structure import ConversationMemory
|
9 |
+
from griptape.tools import StructureRunClient, FileManager
|
10 |
from griptape.rules import Rule, Ruleset
|
11 |
from griptape.config import AnthropicStructureConfig
|
|
|
12 |
import time
|
13 |
import os
|
14 |
+
import re
|
15 |
+
|
16 |
|
17 |
|
18 |
#Load environment variables
|
|
|
153 |
rulesets=[ruleset],
|
154 |
)
|
155 |
|
156 |
+
def delete_json(session_id:str) -> None:
|
157 |
+
for file in glob.glob(f'{session_id}.json'):
|
158 |
+
os.remove(file)
|
159 |
|
160 |
|
161 |
def send_message(message:str, history, request:gr.Request) -> Any:
|
|
|
163 |
session_hash = request.session_hash
|
164 |
agent = build_agent(session_hash,message)
|
165 |
response = agent.run(message)
|
166 |
+
if re.search(r'\bdone[.,!?]?\b', message, re.IGNORECASE):
|
167 |
+
delete_json(session_hash)
|
168 |
return response.output.value
|
169 |
|
170 |
demo = gr.ChatInterface(
|