wei12138 commited on
Commit
ecac7df
·
1 Parent(s): 5f7af9b

Update interact_with_llm.py

Browse files
Files changed (1) hide show
  1. interact_with_llm.py +4 -4
interact_with_llm.py CHANGED
@@ -48,10 +48,10 @@ os.environ['OPENAI_API_KEY']= os.environ.get('OPENAI_API_KEY')
48
  class Agent():
49
  def __init__(self) -> None:
50
 
51
- embeddings = OpenAIEmbeddings()
52
- documents = TextLoader("document_0.txt").load()
53
- text_splitter = CharacterTextSplitter(chunk_size=500, chunk_overlap=0)
54
- docs = text_splitter.split_documents(documents)
55
  vectorstore = Chroma(embedding_function=OpenAIEmbeddings(), persist_directory="./chroma_db")
56
  retriever = vectorstore.as_retriever()
57
  llm = ChatOpenAI(model_name="gpt-4", temperature=0)
 
48
  class Agent():
49
  def __init__(self) -> None:
50
 
51
+ # embeddings = OpenAIEmbeddings()
52
+ # documents = TextLoader("document_0.txt").load()
53
+ # text_splitter = CharacterTextSplitter(chunk_size=500, chunk_overlap=0)
54
+ # docs = text_splitter.split_documents(documents)
55
  vectorstore = Chroma(embedding_function=OpenAIEmbeddings(), persist_directory="./chroma_db")
56
  retriever = vectorstore.as_retriever()
57
  llm = ChatOpenAI(model_name="gpt-4", temperature=0)