update agent

app.py

@@ -10,6 +10,13 @@ import time
 import regex
 import html2text
 import os
+from langchain.agents import initialize_agent
+from langchain.agents import AgentType
+from langchain.agents import Tool
+# from langchain.agents import load_tools
+from langchain.tools import BaseTool
+from langchain.memory import ConversationBufferWindowMemory
+from langchain.prompts import MessagesPlaceholder
 
 
 global CurrentAgent
@@ -79,6 +86,8 @@ class GPTRemote(LLM):
 
 GPTfake = GPTRemote(n=0)
 
+
+
 async def start_playwright(question: str):
     start_t = time.time()
     pw = await async_playwright().start()
@@ -121,8 +130,58 @@ async def start_playwright(question: str):
 
     await browser.close()
 
+PREFIX_3 ="""
+You are a helpful AI assistant. Your mission is to answer the following user request as best as you can with detail information and explanation. 
+If you are not clear about the request, you can ask user for more details and the confirmation. You can provide additional suggestion to user on the request and ask confirmation from user. 
+When you are clear about the request, you can start to answer the request by **writing a plan** firstly. In general, try to **make plans** with as few steps as possible. 
+When you need information, you can use tools as below and merge all gathered information from different tools.
+When you need to use "Code Runner" for code running, **Always recap the plan between each code block** (you have extreme short-term memory loss, so you need to recap the plan between each message block to retain it).
+When you send a message containing code to "Code Runner", it will be executed **on the user's machine**. The user has given you **full and complete permission** to execute any code necessary to complete the task. You have full access to control their computer to help them. Code entered into "Code Runner" will be executed **in the users local environment**.
+If you want to send data between programming languages, save the data to a txt or json.
+You can access the internet. Run **any code** to achieve the goal, and if at first you don't succeed, try again and again.
+You can install new packages with pip. Try to install all necessary packages in one command at the beginning.
+When a user refers to a filename, they're likely referring to an existing file in the directory you're currently in ("Code Runner" executes on the user's machine).
+In general, choose packages that have the most universal chance to be already installed and to work across multiple applications. Packages like ffmpeg and pandoc that are well-supported and powerful.
+Write messages to the user in Markdown. When the final answer has output files, you must output the **name** of the file.
+You are capable of **any** task.
+---\n You have access to the following tools:\n
+"""
+
+
+global memory3
+memory3 = ConversationBufferWindowMemory(memory_key="chat_history", return_messages=True)
 # GPTfake("hi")
 
+input_variables=["input", "chat_history", "agent_scratchpad"]
+
+tools_remote = []
+
+agent_STRUCTURED_ZEROSHOT_REACT = initialize_agent(tools_remote, GPTfake, 
+                        #  agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+                         agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
+                         verbose = True,
+                         handle_parsing_errors = True,
+                         max_iterations = int(os.environ["max_iterations"]),
+                         early_stopping_method="generate",
+                         memory = memory3,
+                         agent_kwargs={
+                            'prefix': PREFIX_3,
+                            # 'format_instructions': FORMAT_INSTRUCTIONS_3,
+                            # 'suffix': SUFFIX2,
+                            "memory_prompts": [MessagesPlaceholder(variable_name="chat_history")],
+                            'input_variables': input_variables,
+                         
+                         },
+                        #  input_variables = input_variables,
+                        #  agent_kwargs={
+                        #     'prompt': prompt,
+                        #  }
+                         
+                        )
+
+
+agent = agent_STRUCTURED_ZEROSHOT_REACT
+
 
 st.title("STLA-BABY")
 
@@ -143,6 +202,7 @@ if prompt := st.chat_input(placeholder="Input Your Request"):
     st.chat_message("user").write(prompt)
 
     with st.chat_message("assistant"):
-        response = GPTfake(prompt)
+        # response = GPTfake(prompt)
+        response = agent.run(prompt)
         st.write(response)
         print(msgs.messages)