Kate Forsberg commited on
Commit
36d83b1
·
1 Parent(s): 318ef9e

deletes memory if someone says done

Browse files
Files changed (1) hide show
  1. app.py +10 -6
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, StructureRunTask, ToolkitTask
9
- from griptape.drivers import LocalConversationMemoryDriver, GriptapeCloudStructureRunDriver, GriptapeCloudEventListenerDriver, LocalFileManagerDriver, LocalStructureRunDriver
10
  from griptape.memory.structure import ConversationMemory
11
- from griptape.tools import StructureRunClient, TaskMemoryClient, FileManager
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(