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(
|