Merge pull request #4 from ks6088ts-labs/feature/issue-3_qdrant-tool

ks6088ts · web-flow · commit f1794717251d · 2025-08-01T14:37:34.000+09:00
implement an indexer for adding documents to Qdrant
diff --git a/.env.template b/.env.template
@@ -1,2 +1,15 @@
 # Project environment variables
 PROJECT_NAME=template-langgraph
+
+# Azure OpenAI Service
+AZURE_OPENAI_ENDPOINT="https://YOUR_AZURE_OPENAI_ENDPOINT/"
+AZURE_OPENAI_API_KEY="YOUR_AZURE_OPENAI_API_KEY"
+AZURE_OPENAI_API_VERSION="2024-10-21"
+AZURE_OPENAI_MODEL_CHAT="gpt-4o"
+AZURE_OPENAI_MODEL_EMBEDDING="text-embedding-3-small"
+
+# CSV Loader Settings
+CSV_LOADER_DATA_DIR_PATH="./data"
+
+# Qdrant Settings
+QDRANT_URL="http://localhost:6333"
diff --git a/.gitignore b/.gitignore
@@ -162,3 +162,5 @@ cython_debug/
 # Project
 *.env
 requirements.txt
+assets/
+data/
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -0,0 +1,8 @@
+services:
+  qdrant:
+    image: qdrant/qdrant:v1.15.1
+    container_name: qdrant
+    ports:
+      - "6333:6333" # Dashboard: http://localhost:6333/dashboard
+    volumes:
+      - ./assets/qdrant_data:/qdrant/storage
diff --git a/docs/index.md b/docs/index.md
@@ -1 +1,19 @@
 # template-langgraph
+
+## Operations
+
+```bash
+# Add documents to Qdrant
+uv run python -m template_langgraph.tasks.add_documents_to_qdrant
+```
+
+## References
+
+### Models
+
+- [AzureOpenAIEmbeddings](https://python.langchain.com/docs/integrations/text_embedding/azureopenai/)
+
+### Tools
+
+- [CSVLoader](https://python.langchain.com/docs/how_to/document_loader_csv/)
+- [Qdrant](https://github.com/qdrant/qdrant)
diff --git a/pyproject.toml b/pyproject.toml
@@ -5,8 +5,14 @@ description = "A GitHub template repository for Python"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
+    "langchain-community>=0.3.27",
+    "langchain-openai>=0.3.28",
+    "langchain-text-splitters>=0.3.9",
+    "langgraph>=0.6.2",
+    "openai>=1.98.0",
     "pydantic-settings>=2.9.1",
     "python-dotenv>=1.1.0",
+    "qdrant-client>=1.15.1",
     "typer>=0.16.0",
 ]
 
@@ -52,3 +58,7 @@ show_missing = true
 
 [tool.ty]
 environment = { python-version = "3.10" }
+
+[tool.ty.rules]
+unknown-argument = "ignore"
+invalid-parameter-default = "ignore"
diff --git a/template_langgraph/models/__init__.py b/template_langgraph/models/__init__.py
diff --git a/template_langgraph/models/azure_openais.py b/template_langgraph/models/azure_openais.py
@@ -0,0 +1,48 @@
+from functools import lru_cache
+
+from langchain_openai import AzureChatOpenAI, AzureOpenAIEmbeddings
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class Settings(BaseSettings):
+    azure_openai_endpoint: str = "https://<YOUR_AOAI_NAME>.openai.azure.com/"
+    azure_openai_api_key: str = "<YOUR_API_KEY>"
+    azure_openai_api_version: str = "2024-10-21"
+    azure_openai_model_chat: str = "gpt-4o"
+    azure_openai_model_embedding: str = "text-embedding-3-small"
+
+    model_config = SettingsConfigDict(
+        env_file=".env",
+        env_ignore_empty=True,
+        extra="ignore",
+    )
+
+
+@lru_cache
+def get_azure_openai_settings() -> Settings:
+    return Settings()
+
+
+class AzureOpenAiWrapper:
+    def __init__(self, settings: Settings = None):
+        if settings is None:
+            settings = get_azure_openai_settings()
+
+        self.chat_model = AzureChatOpenAI(
+            azure_endpoint=settings.azure_openai_endpoint,
+            api_key=settings.azure_openai_api_key,
+            api_version=settings.azure_openai_api_version,
+            azure_deployment=settings.azure_openai_model_chat,
+            temperature=0.0,
+            streaming=True,
+        )
+        self.embedding_model = AzureOpenAIEmbeddings(
+            azure_endpoint=settings.azure_openai_endpoint,
+            api_key=settings.azure_openai_api_key,
+            api_version=settings.azure_openai_api_version,
+            azure_deployment=settings.azure_openai_model_embedding,
+        )
+
+    def create_embedding(self, text: str):
+        """Create an embedding for the given text."""
+        return self.embedding_model.embed_query(text)
diff --git a/template_langgraph/tasks/__init__.py b/template_langgraph/tasks/__init__.py
diff --git a/template_langgraph/tasks/add_documents_to_qdrant.py b/template_langgraph/tasks/add_documents_to_qdrant.py
@@ -0,0 +1,50 @@
+import logging
+
+from qdrant_client.models import PointStruct
+
+from template_langgraph.loggers import get_logger
+from template_langgraph.models.azure_openais import AzureOpenAiWrapper
+from template_langgraph.tools.csv_loaders import CsvLoaderWrapper
+from template_langgraph.tools.qdrants import QdrantClientWrapper
+
+logger = get_logger(__name__)
+logger.setLevel(logging.INFO)
+COLLECTION_NAME = "documents"
+
+if __name__ == "__main__":
+    # Load documents from CSV files
+    documents = CsvLoaderWrapper().load_csv_docs()
+    logger.info(f"Loaded {len(documents)} documents from CSV.")
+
+    # hardcoded collection name for demonstration purposes
+
+    logger.info(f"Upserting {len(documents)} documents into Qdrant collection: {COLLECTION_NAME}")
+    points = []
+    for i, doc in enumerate(documents):
+        logger.debug(f"Processing document {i}: {doc.metadata.get('source', 'unknown')}")
+        content = doc.page_content
+        content = content.replace(" ", "")
+        embedding = AzureOpenAiWrapper().create_embedding(content)
+        points.append(
+            PointStruct(
+                id=i,
+                vector=embedding,
+                payload={
+                    "file_name": doc.metadata.get("source", f"doc_{i}"),
+                    "content": content,
+                },
+            )
+        )
+
+    qdrant_client = QdrantClientWrapper()
+    qdrant_client.create_collection(
+        collection_name=COLLECTION_NAME,
+        vector_size=len(points[0].vector) if points else 1536,  # default vector size
+    )
+
+    logger.info(f"Created Qdrant collection: {COLLECTION_NAME}")
+    operation_info = qdrant_client.upsert_points(
+        collection_name=COLLECTION_NAME,
+        points=points,
+    )
+    logger.info(f"Upserted {len(points)} points into Qdrant collection: {COLLECTION_NAME}")
diff --git a/template_langgraph/tools/__init__.py b/template_langgraph/tools/__init__.py
diff --git a/template_langgraph/tools/csv_loaders.py b/template_langgraph/tools/csv_loaders.py
@@ -0,0 +1,47 @@
+import os
+from functools import lru_cache
+from glob import glob
+
+from langchain_community.document_loaders.csv_loader import CSVLoader
+from langchain_core.documents import Document
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class Settings(BaseSettings):
+    csv_loader_data_dir_path: str = "./data"
+
+    model_config = SettingsConfigDict(
+        env_file=".env",
+        env_ignore_empty=True,
+        extra="ignore",
+    )
+
+
+@lru_cache
+def get_csv_loader_settings() -> Settings:
+    """Get CSV loader settings."""
+    return Settings()
+
+
+class CsvLoaderWrapper:
+    def __init__(
+        self,
+        settings: Settings = None,
+    ):
+        if settings is None:
+            settings = get_csv_loader_settings()
+        self.settings = settings
+
+    def load_csv_docs(self) -> list[Document]:
+        """Load CSV documents from the specified directory."""
+        csv_path = glob(
+            os.path.join(self.settings.csv_loader_data_dir_path, "**", "*.csv"),
+            recursive=True,
+        )
+        docs = []
+
+        for path in csv_path:
+            loader = CSVLoader(file_path=path)
+            docs.extend(loader.load())
+
+        return docs
diff --git a/template_langgraph/tools/qdrants.py b/template_langgraph/tools/qdrants.py
@@ -0,0 +1,61 @@
+from functools import lru_cache
+
+from pydantic_settings import BaseSettings, SettingsConfigDict
+from qdrant_client import QdrantClient
+from qdrant_client.http.models import UpdateResult
+from qdrant_client.models import Distance, PointStruct, VectorParams
+
+
+class Settings(BaseSettings):
+    qdrant_url: str = "http://localhost:6333"
+
+    model_config = SettingsConfigDict(
+        env_file=".env",
+        env_ignore_empty=True,
+        extra="ignore",
+    )
+
+
+@lru_cache
+def get_qdrant_settings() -> Settings:
+    """Get Qdrant settings."""
+    return Settings()
+
+
+class QdrantClientWrapper:
+    def __init__(
+        self,
+        settings: Settings = None,
+    ):
+        if settings is None:
+            settings = get_qdrant_settings()
+        self.client = QdrantClient(
+            url=settings.qdrant_url,
+        )
+
+    def create_collection(
+        self,
+        collection_name: str,
+        vector_size: int = 1536,
+    ) -> bool:
+        """Create a collection in Qdrant."""
+        result = self.client.create_collection(
+            collection_name=collection_name,
+            vectors_config=VectorParams(
+                size=vector_size,
+                distance=Distance.COSINE,
+            ),
+        )
+        return result
+
+    def upsert_points(
+        self,
+        collection_name: str,
+        points: list[PointStruct],
+    ) -> UpdateResult:
+        """Upsert points into a Qdrant collection."""
+        return self.client.upsert(
+            collection_name=collection_name,
+            points=points,
+            wait=True,
+        )
diff --git a/uv.lock b/uv.lock
