legolasyiu commited on
Commit
c37da7c
·
verified ·
1 Parent(s): 27e94c7

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +4 -1
src/streamlit_app.py CHANGED
@@ -3,11 +3,14 @@ import streamlit as st
3
  from typing import Literal
4
  from tavily import TavilyClient
5
  from deepagents import create_deep_agent
6
- from langchain.callbacks.streamlit import StreamlitCallbackHandler
7
  from langgraph.store.memory import InMemoryStore
8
  from langchain.chat_models import init_chat_model
9
  from langchain_huggingface import ChatHuggingFace, HuggingFacePipeline
10
 
 
 
 
 
11
 
12
  # --- Sidebar: API Keys ---
13
  st.sidebar.header("🔑 API Keys")
 
3
  from typing import Literal
4
  from tavily import TavilyClient
5
  from deepagents import create_deep_agent
 
6
  from langgraph.store.memory import InMemoryStore
7
  from langchain.chat_models import init_chat_model
8
  from langchain_huggingface import ChatHuggingFace, HuggingFacePipeline
9
 
10
+ from langchain_community.callbacks.streamlit import (
11
+ StreamlitCallbackHandler,
12
+ )
13
+
14
 
15
  # --- Sidebar: API Keys ---
16
  st.sidebar.header("🔑 API Keys")