ChienChung commited on
Commit
d6cf0e3
·
verified ·
1 Parent(s): 55d0f4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -25,12 +25,15 @@ os.environ["HF_HUB_CACHE"] = "/tmp/huggingface/hf_cache"
25
  os.environ["HF_HUB_DOWNLOAD_TIMEOUT"] = "60"
26
 
27
  # Load Required Modules
28
- from langchain.embeddings import HuggingFaceEmbeddings
 
29
  from langchain.vectorstores import Chroma, FAISS
30
  from langchain.chains import RetrievalQA
31
  from langchain.prompts import PromptTemplate
32
- from langchain.llms import HuggingFacePipeline
33
- from langchain.chat_models import ChatOpenAI
 
 
34
  from langchain.chains import ConversationalRetrievalChain
35
  from langchain.memory import ConversationBufferMemory
36
  from langchain_community.document_loaders import PyPDFLoader, TextLoader, UnstructuredWordDocumentLoader
@@ -408,6 +411,7 @@ def get_time_tool(query: str) -> str:
408
 
409
  @tool("time_tl")
410
  def time_tool(query: str) -> str:
 
411
  # use GPT to find location keyword
412
  try:
413
  location_prompt = f"""
 
25
  os.environ["HF_HUB_DOWNLOAD_TIMEOUT"] = "60"
26
 
27
  # Load Required Modules
28
+ # from langchain.embeddings import HuggingFaceEmbeddings
29
+ from langchain_huggingface import HuggingFaceEmbeddings
30
  from langchain.vectorstores import Chroma, FAISS
31
  from langchain.chains import RetrievalQA
32
  from langchain.prompts import PromptTemplate
33
+ # from langchain.llms import HuggingFacePipeline
34
+ from langchain_huggingface import HuggingFacePipeline
35
+ # from langchain.chat_models import ChatOpenAI
36
+ from langchain_openai import ChatOpenAI
37
  from langchain.chains import ConversationalRetrievalChain
38
  from langchain.memory import ConversationBufferMemory
39
  from langchain_community.document_loaders import PyPDFLoader, TextLoader, UnstructuredWordDocumentLoader
 
411
 
412
  @tool("time_tl")
413
  def time_tool(query: str) -> str:
414
+ """Time Agent: Answer time or date queries worldwide using LLM + GeoLocator + TimezoneFinder."""
415
  # use GPT to find location keyword
416
  try:
417
  location_prompt = f"""