diff --git a/templates/self-query-qdrant/poetry.lock b/templates/self-query-qdrant/poetry.lock index 03d0834992..729f9ab44a 100644 --- a/templates/self-query-qdrant/poetry.lock +++ b/templates/self-query-qdrant/poetry.lock @@ -1,4 +1,4 @@ -# This file is automatically @generated by Poetry 1.7.1 and should not be changed by hand. +# This file is automatically @generated by Poetry 1.8.2 and should not be changed by hand. [[package]] name = "aiohttp" @@ -939,6 +939,22 @@ tenacity = ">=8.1.0,<9.0.0" [package.extras] extended-testing = ["jinja2 (>=3,<4)"] +[[package]] +name = "langchain-openai" +version = "0.0.8" +description = "An integration package connecting OpenAI and LangChain" +optional = false +python-versions = ">=3.8.1,<4.0" +files = [ + {file = "langchain_openai-0.0.8-py3-none-any.whl", hash = "sha256:4862fc72cecbee0240aaa6df0234d5893dd30cd33ca23ac5cfdd86c11d2c44df"}, + {file = "langchain_openai-0.0.8.tar.gz", hash = "sha256:b7aba7fcc52305e78b08197ebc54fc45cc06dbc40ba5b913bc48a22b30a4f5c9"}, +] + +[package.dependencies] +langchain-core = ">=0.1.27,<0.2.0" +openai = ">=1.10.0,<2.0.0" +tiktoken = ">=0.5.2,<1" + [[package]] name = "langchain-text-splitters" version = "0.0.1" @@ -2207,4 +2223,4 @@ multidict = ">=4.0" [metadata] lock-version = "2.0" python-versions = ">=3.9,<3.13" -content-hash = "a50f00122a31808a22a6301cbd6b2f6f9db67bae3305c93a69e6d7ee451df628" +content-hash = "0dacdc7b4d1ed6ebdfd86b15135142c3d4ed8d8728dd9cc09c8504aff590917e" diff --git a/templates/self-query-qdrant/pyproject.toml b/templates/self-query-qdrant/pyproject.toml index 1e41e7161a..69850e30ae 100644 --- a/templates/self-query-qdrant/pyproject.toml +++ b/templates/self-query-qdrant/pyproject.toml @@ -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" diff --git a/templates/self-query-qdrant/self_query_qdrant/chain.py b/templates/self-query-qdrant/self_query_qdrant/chain.py index ccc30be636..0e82817743 100644 --- a/templates/self-query-qdrant/self_query_qdrant/chain.py +++ b/templates/self-query-qdrant/self_query_qdrant/chain.py @@ -3,14 +3,14 @@ from typing import List, Optional 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 @@ -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"), )