Skip to content

Commit

Permalink
templates: update qdrant self query (langchain-ai#19218)
Browse files Browse the repository at this point in the history
## Description

This PR
- Updates the Qdrant self-query template to reflect the recent updates.
- Enables reading config values from `env` files as the README [mentions
it](https://github.com/Anush008/langchain/tree/self-query-qdrant/templates/self-query-qdrant#environment-setup).

Co-authored-by: Erick Friis <erick@langchain.dev>
  • Loading branch information
Anush008 and efriis committed Mar 18, 2024
1 parent 21f7599 commit aee5138
Show file tree
Hide file tree
Showing 3 changed files with 27 additions and 7 deletions.
20 changes: 18 additions & 2 deletions templates/self-query-qdrant/poetry.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion templates/self-query-qdrant/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -10,10 +10,10 @@ packages = [{include = "self_query_qdrant"}]
[tool.poetry.dependencies]
python = ">=3.9,<3.13"
langchain = "^0.1"
openai = "<2"
qdrant-client = ">=1.6"
lark = "^1.1.8"
tiktoken = "^0.5.1"
langchain-openai = "^0.0.8"

[tool.poetry.group.dev.dependencies]
langchain-cli = ">=0.0.21"
Expand Down
12 changes: 8 additions & 4 deletions templates/self-query-qdrant/self_query_qdrant/chain.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,14 +3,14 @@

from langchain.chains.query_constructor.schema import AttributeInfo
from langchain.retrievers import SelfQueryRetriever
from langchain_community.embeddings import OpenAIEmbeddings
from langchain_community.llms import BaseLLM
from langchain_community.llms.openai import OpenAI
from langchain_community.vectorstores.qdrant import Qdrant
from langchain_core.documents import Document, StrOutputParser
from langchain_core.documents import Document
from langchain_core.embeddings import Embeddings
from langchain_core.output_parsers.string import StrOutputParser
from langchain_core.pydantic_v1 import BaseModel
from langchain_core.runnables import RunnableParallel, RunnablePassthrough
from langchain_openai import OpenAI, OpenAIEmbeddings
from qdrant_client import QdrantClient

from self_query_qdrant import defaults, helper, prompts
Expand Down Expand Up @@ -84,7 +84,11 @@ def initialize(

# Set up a vector store to store your vectors and metadata
Qdrant.from_documents(
documents, embedding=embeddings, collection_name=collection_name
documents,
embedding=embeddings,
collection_name=collection_name,
url=os.environ.get("QDRANT_URL", "http://localhost:6333"),
api_key=os.environ.get("QDRANT_API_KEY"),
)


Expand Down

0 comments on commit aee5138

Please sign in to comment.