testing

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+protected-areas.duckdb filter=lfs diff=lfs merge=lfs -text
+pad.duckdb filter=lfs diff=lfs merge=lfs -text

README.md

@@ -5,7 +5,7 @@ colorFrom: indigo
 colorTo: gray
 sdk: streamlit
 sdk_version: 1.32.2
-app_file: app.py
+app_file: minimal_agent.py
 pinned: false
 license: bsd-2-clause
 ---

minimal_agent.py

@@ -0,0 +1,18 @@
+# +
+from langchain.llms import OpenAI
+from langchain.agents import AgentType, initialize_agent, load_tools
+from langchain.callbacks import StreamlitCallbackHandler
+import streamlit as st
+
+llm = OpenAI(temperature=0, streaming=True)
+tools = load_tools(["ddg-search"])
+agent = initialize_agent(
+ tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True
+)
+
+if prompt := st.chat_input():
+ st.chat_message("user").write(prompt)
+ with st.chat_message("assistant"):
+ st_callback = StreamlitCallbackHandler(st.container())
+ response = agent.run(prompt, callbacks=[st_callback])
+ st.write(response)

requirements.txt

@@ -6,4 +6,4 @@ leafmap
 ibis-framework[duckdb]
 altair
 openai
-
+langchain