Skip to content

Commit 21dbcac

Browse files
updated deprecated imports and added collection_name parameter
1 parent d1e9b3d commit 21dbcac

File tree

1 file changed

+17
-4
lines changed

1 file changed

+17
-4
lines changed

knowledge/langchain.mdx

Lines changed: 17 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,21 +4,25 @@ sidebarTitle: LangChain
44
---
55

66
The **LangchainKnowledgeBase** allows us to use a LangChain retriever or vector store as a knowledge base.
7-
7+
Using ChromaDB as the vectorDB in this example.
88
## Usage
99

1010
```shell
1111
pip install langchain
1212
```
1313

14+
```shell
15+
pip install -qU "langchain-chroma>=0.1.2"
16+
```
17+
1418
```python langchain_kb.py
1519
from phi.agent import Agent
1620
from phi.knowledge.langchain import LangChainKnowledgeBase
1721

18-
from langchain.embeddings import OpenAIEmbeddings
22+
from langchain_openai import OpenAIEmbeddings
1923
from langchain.document_loaders import TextLoader
2024
from langchain.text_splitter import CharacterTextSplitter
21-
from langchain.vectorstores import Chroma
25+
from langchain_chroma import Chroma
2226

2327
chroma_db_dir = "./chroma_db"
2428

@@ -35,7 +39,13 @@ def load_vector_store():
3539

3640

3741
# -*- Get the vectordb
38-
db = Chroma(embedding_function=OpenAIEmbeddings(), persist_directory=str(chroma_db_dir))
42+
db = Chroma(
43+
collection_name="your_collection_name",
44+
persist_directory=chroma_db_dir,
45+
embedding_function=OpenAIEmbeddings(),
46+
47+
)
48+
3949
# -*- Create a retriever from the vector store
4050
retriever = db.as_retriever()
4151

@@ -53,3 +63,6 @@ conv.print_response("What did the president say about technology?")
5363
| `retriever` | `Any` | `None` | LangChain retriever. |
5464
| `vectorstore` | `Any` | `None` | LangChain vector store used to create a retriever. |
5565
| `search_kwargs` | `dict` | `None` | Search kwargs when creating a retriever using the langchain vector store. |
66+
67+
68+
You can also find more information about updates here [LangChain Chroma Vector Store Documentation](https://python.langchain.com/docs/integrations/vectorstores/chroma/)

0 commit comments

Comments
 (0)