SamerPF commited on
Commit
55feaa9
·
verified ·
1 Parent(s): 6c20db5

Update rag_agent.py

Browse files
Files changed (1) hide show
  1. rag_agent.py +1 -1
rag_agent.py CHANGED
@@ -1,6 +1,6 @@
1
  import os
2
  from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, ServiceContext
3
- #from llama_index.llms.huggingface import HuggingFaceLLM
4
  from llama_index.core.tools import QueryEngineTool, ToolMetadata
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
  from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 
1
  import os
2
  from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, ServiceContext
3
+ from llama_index.llms.huggingface import HuggingFaceLLM
4
  from llama_index.core.tools import QueryEngineTool, ToolMetadata
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
  from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI