Merge pull request #37 from ks6088ts-labs/cosmetic-changes

Cosmetic changes

Author: ks6088ts

pyproject.toml

@@ -8,6 +8,7 @@ dependencies = [
     "azure-cosmos>=4.9.0",
     "elasticsearch>=9.1.0",
     "httpx>=0.28.1",
+    "langchain-azure-ai>=0.1.4",
     "langchain-community>=0.3.27",
     "langchain-openai>=0.3.28",
     "langchain-text-splitters>=0.3.9",

scripts/cosmosdb_operator.py

@@ -52,6 +52,7 @@ def add_documents(
     for id in ids:
         logger.debug(f"Added document ID: {id}")
 
+    # FIXME: deleting documents() is not working as expected
     # assert cosmosdb_client.delete_documents(ids=ids), "Failed to delete documents from Cosmos DB"
 
 

scripts/test_all.sh

@@ -7,22 +7,24 @@ set -euo pipefail
 docker compose up -d --wait
 
 # Qdrant
+COLLECTION_NAME="qa_kabuto"
 uv run python scripts/qdrant_operator.py --help
-uv run python scripts/qdrant_operator.py delete-collection --collection-name qa_kabuto --verbose
-uv run python scripts/qdrant_operator.py add-documents --collection-name qa_kabuto --verbose
-uv run python scripts/qdrant_operator.py search-documents --collection-name qa_kabuto --question "「鬼灯」を実行すると、KABUTOが急に停止します。原因と対策を教えてください。" --verbose
+uv run python scripts/qdrant_operator.py delete-collection --collection-name $COLLECTION_NAME --verbose
+uv run python scripts/qdrant_operator.py add-documents     --collection-name $COLLECTION_NAME --verbose
+uv run python scripts/qdrant_operator.py search-documents  --collection-name $COLLECTION_NAME --verbose --question "「鬼灯」を実行すると、KABUTOが急に停止します。原因と対策を教えてください。"
 
 # Elasticsearch
+INDEX_NAME="docs_kabuto"
 uv run python scripts/elasticsearch_operator.py --help
-uv run python scripts/elasticsearch_operator.py delete-index --index-name docs_kabuto --verbose
-uv run python scripts/elasticsearch_operator.py create-index --index-name docs_kabuto --verbose
-uv run python scripts/elasticsearch_operator.py add-documents --index-name docs_kabuto --verbose
-uv run python scripts/elasticsearch_operator.py search-documents --index-name docs_kabuto --query "禅モード" --verbose
+uv run python scripts/elasticsearch_operator.py delete-index     --index-name $INDEX_NAME --verbose
+uv run python scripts/elasticsearch_operator.py create-index     --index-name $INDEX_NAME --verbose
+uv run python scripts/elasticsearch_operator.py add-documents    --index-name $INDEX_NAME --verbose
+uv run python scripts/elasticsearch_operator.py search-documents --index-name $INDEX_NAME --verbose --query "禅モード"
 
 # Azure Cosmos DB NoSQL
 uv run python scripts/cosmosdb_operator.py --help
-uv run python scripts/cosmosdb_operator.py add-documents --verbose
-uv run python scripts/cosmosdb_operator.py similarity-search --query "禅モード" --k 3 --verbose
+uv run python scripts/cosmosdb_operator.py add-documents     --verbose
+uv run python scripts/cosmosdb_operator.py similarity-search --verbose --query "禅モード" --k 3
 
 # Agents
 

template_langgraph/tools/cosmosdb_tool.py

@@ -1,9 +1,7 @@
 from functools import lru_cache
 
 from azure.cosmos import CosmosClient, PartitionKey
-from langchain_community.vectorstores.azure_cosmos_db_no_sql import (
-    AzureCosmosDBNoSqlVectorSearch,
-)
+from langchain_azure_ai.vectorstores.azure_cosmos_db_no_sql import AzureCosmosDBNoSqlVectorSearch
 from langchain_core.documents import Document
 from langchain_core.tools import tool
 from pydantic import BaseModel, Field
@@ -74,6 +72,10 @@ def __init__(
                 "partition_key": PartitionKey(path=settings.cosmosdb_partition_key),
             },
             cosmos_database_properties={},
+            vector_search_fields={
+                "text_field": "text",
+                "embedding_field": "embedding",
+            },
             full_text_policy={
                 "defaultLanguage": "en-US",
                 "fullTextPaths": [