Merge branch 'release/v1.0.1'

sad-zero · May 20, 2024 · 9c88ce4 · 9c88ce4
2 parents 7d78cd2 + dd7edae
commit 9c88ce4
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 7 deletions.
diff --git a/chat.py b/chat.py
@@ -14,7 +14,7 @@
 
 if __name__ == "__main__":
     embeddings = OllamaEmbeddings(model="all-minilm")
-    client = PersistentClient(path="./resources/vector_store.chroma", settings=Settings(anonymized_telemetry=False))
+    client = PersistentClient(path="resources/chroma_db", settings=Settings(anonymized_telemetry=False))
     collection = client.get_collection("books")
     model = ChatOllama(model="phi3")
 

diff --git a/etl.py b/etl.py
@@ -2,7 +2,6 @@
 ETL Pipeline
 """
 
-# TODO: ETL Pipeline 완성
 import logging
 from chromadb import PersistentClient
 from chromadb.config import Settings
@@ -13,9 +12,7 @@
 from langchain_community.embeddings.ollama import OllamaEmbeddings
 
 
-def get_vector_store(
-    coll_name: str = "books", persistent_path: str = "/Users/dev/Documents/simple-chatboot/resources/chroma_db"
-) -> Chroma:
+def get_vector_store(coll_name: str = "books", persistent_path: str = "resources/chroma_db") -> Chroma:
     embeddings = OllamaEmbeddings(model="all-minilm")
     store = InMemoryByteStore()
     embeddings_func = CacheBackedEmbeddings.from_bytes_store(

diff --git a/pyproject.toml b/pyproject.toml
@@ -7,7 +7,7 @@ dynamic = ["dependencies"]
 
 name = "simple_chatbot"
 authors = [{name = "sad-zero", email = "[email protected]"}]
-version = "1.0.0"
+version = "1.0.1"
 readme = "README.md"
 license = {file = "LICENSE"}
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,5 @@
 pypdf
 chromadb
 langchain
-langchain-chroma
+langchain-chroma
+langchain-community