Nullpointer-KK commited on
Commit
8a88917
·
verified ·
1 Parent(s): 5252971

Update scripts/main.py

Browse files
Files changed (1) hide show
  1. scripts/main.py +2 -1
scripts/main.py CHANGED
@@ -4,7 +4,8 @@ import pdb
4
  import gradio as gr
5
  import logfire
6
  from custom_retriever import CustomRetriever
7
- from llama_index.agent.openai import OpenAIAgent
 
8
  from llama_index.core.llms import MessageRole
9
  from llama_index.core.memory import ChatSummaryMemoryBuffer
10
  from llama_index.core.tools import RetrieverTool, ToolMetadata
 
4
  import gradio as gr
5
  import logfire
6
  from custom_retriever import CustomRetriever
7
+ #from llama_index.agent.openai import OpenAIAgent
8
+ from llama_index.core.agent import OpenAIAgent
9
  from llama_index.core.llms import MessageRole
10
  from llama_index.core.memory import ChatSummaryMemoryBuffer
11
  from llama_index.core.tools import RetrieverTool, ToolMetadata