Skip to content

Commit

Permalink
Solve pylint complains
Browse files Browse the repository at this point in the history
  • Loading branch information
Maclenn77 committed Dec 9, 2023
1 parent ca67687 commit 04e2ec8
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 1 deletion.
12 changes: 11 additions & 1 deletion app.py
Original file line number Diff line number Diff line change
@@ -1,15 +1,16 @@
""" A simple example of Streamlit. """
from datetime import datetime as Date
import textwrap
import os
import tiktoken
import chromadb
from chromadb.utils.embedding_functions import OpenAIEmbeddingFunction
import fitz
import streamlit as st
import openai
import os
from dotenv import load_dotenv
from openai import OpenAI
from langchain.vectorstores import Chroma

load_dotenv()

Expand Down Expand Up @@ -94,3 +95,12 @@
chroma_client.delete_collection(collection.name)
except AttributeError:
st.error("Collection erased.")

if chroma_client.get_collection(collection.name) is not None:
langchain_agent = Chroma(client=chroma_client,
collection_name=collection.name,
embedding_function=embedding_function
)



2 changes: 2 additions & 0 deletions wk_flow_requirements.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
streamlit
pymupdf
openai
tiktoken
pylint
chromadb>='0.4.18'

0 comments on commit 04e2ec8

Please sign in to comment.