heysho commited on
Commit
0d7428d
1 Parent(s): b3cce14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -16
app.py CHANGED
@@ -14,14 +14,6 @@ from langchain_google_genai import ChatGoogleGenerativeAI
14
  from tools.search_ddg import search_ddg
15
  from tools.fetch_page import fetch_page
16
 
17
- ###### dotenv を利用しない場合は消してください ######
18
- try:
19
- from dotenv import load_dotenv
20
- load_dotenv()
21
- except ImportError:
22
- import warnings
23
- warnings.warn("dotenv not found. Please make sure to set your environment variables manually.", ImportWarning)
24
- ################################################
25
 
26
  CUSTOM_SYSTEM_PROMPT = """
27
  あなたは、ユーザーのリクエストに基づいてインターネットで調べ物を行うアシスタントです。
@@ -58,10 +50,10 @@ CUSTOM_SYSTEM_PROMPT = """
58
 
59
  def init_page():
60
  st.set_page_config(
61
- page_title="Web Browsing Agent",
62
  page_icon="🤗"
63
  )
64
- st.header("Web Browsing Agent 🤗")
65
  # st.sidebar.title("Options")
66
 
67
 
@@ -77,11 +69,6 @@ def init_messages():
77
  k=10
78
  )
79
 
80
- # このようにも書ける
81
- # from langchain_community.chat_message_histories import StreamlitChatMessageHistory
82
- # msgs = StreamlitChatMessageHistory(key="special_app_key")
83
- # st.session_state['memory'] = ConversationBufferMemory(memory_key="history", chat_memory=msgs)
84
-
85
 
86
  def select_model(temperature=0):
87
  models = ("GPT-4o","GPT-4o-mini", "Claude 3.5 Sonnet", "Gemini 1.5 Pro")
@@ -122,7 +109,7 @@ def main():
122
  for msg in st.session_state['memory'].chat_memory.messages:
123
  st.chat_message(msg.type).write(msg.content)
124
 
125
- if prompt := st.chat_input(placeholder="2024年9月の東京のイベント情報を教えて"):
126
  st.chat_message("user").write(prompt)
127
 
128
  with st.chat_message("assistant"):
 
14
  from tools.search_ddg import search_ddg
15
  from tools.fetch_page import fetch_page
16
 
 
 
 
 
 
 
 
 
17
 
18
  CUSTOM_SYSTEM_PROMPT = """
19
  あなたは、ユーザーのリクエストに基づいてインターネットで調べ物を行うアシスタントです。
 
50
 
51
  def init_page():
52
  st.set_page_config(
53
+ page_title="Web調査エージェント",
54
  page_icon="🤗"
55
  )
56
+ st.header("Web調査エージェント 🤗")
57
  # st.sidebar.title("Options")
58
 
59
 
 
69
  k=10
70
  )
71
 
 
 
 
 
 
72
 
73
  def select_model(temperature=0):
74
  models = ("GPT-4o","GPT-4o-mini", "Claude 3.5 Sonnet", "Gemini 1.5 Pro")
 
109
  for msg in st.session_state['memory'].chat_memory.messages:
110
  st.chat_message(msg.type).write(msg.content)
111
 
112
+ if prompt := st.chat_input(placeholder="最近のサッカー日本代表の試合結果を教えて。"):
113
  st.chat_message("user").write(prompt)
114
 
115
  with st.chat_message("assistant"):