SamerPF commited on
Commit
ecf22a7
·
verified ·
1 Parent(s): ef15726

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.core.llms import OpenAI
4
  from llama_index.core.tools import QueryEngineTool, ToolMetadata
5
 
6
  # Build Service Context with preferred model
 
1
  import os
2
  from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, ServiceContext
3
+ from llama_index.llms.openai import OpenAI
4
  from llama_index.core.tools import QueryEngineTool, ToolMetadata
5
 
6
  # Build Service Context with preferred model