Skip to content

Commit da3b244

Browse files
feat: Bump langchain dependencies to 1.0.x
- Update langchain from 0.1.16 to ^1.0.2 - Update langchain_core from 0.1.42 to ^1.0.0 - Add langchain_community ^0.4 as explicit dependency - Update imports to use new langchain package structure: - langchain_community.embeddings for embedding classes - langchain_text_splitters for text splitting - langchain_core.utils.strings for utility functions - Fix CohereEmbeddings initialization to include required user_agent parameter - All fast tests passing (3800 passed) Co-Authored-By: AJ Steers <[email protected]>
1 parent 20ae208 commit da3b244

File tree

4 files changed

+942
-234
lines changed

4 files changed

+942
-234
lines changed

airbyte_cdk/destinations/vector_db_based/document_processor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,9 @@
88
from typing import Any, Dict, List, Mapping, Optional, Tuple
99

1010
import dpath
11-
from langchain.text_splitter import Language, RecursiveCharacterTextSplitter
12-
from langchain.utils import stringify_dict
11+
from langchain_text_splitters import Language, RecursiveCharacterTextSplitter
1312
from langchain_core.documents.base import Document
13+
from langchain_core.utils.strings import stringify_dict
1414

1515
from airbyte_cdk.destinations.vector_db_based.config import (
1616
ProcessingConfigModel,

airbyte_cdk/destinations/vector_db_based/embedder.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,7 @@
77
from dataclasses import dataclass
88
from typing import List, Optional, Union, cast
99

10-
from langchain.embeddings.cohere import CohereEmbeddings
11-
from langchain.embeddings.fake import FakeEmbeddings
12-
from langchain.embeddings.localai import LocalAIEmbeddings
13-
from langchain.embeddings.openai import OpenAIEmbeddings
10+
from langchain_community.embeddings import CohereEmbeddings, FakeEmbeddings, LocalAIEmbeddings, OpenAIEmbeddings
1411

1512
from airbyte_cdk.destinations.vector_db_based.config import (
1613
AzureOpenAIEmbeddingConfigModel,
@@ -140,7 +137,7 @@ def __init__(self, config: CohereEmbeddingConfigModel):
140137
super().__init__()
141138
# Client is set internally
142139
self.embeddings = CohereEmbeddings(
143-
cohere_api_key=config.cohere_key, model="embed-english-light-v2.0"
140+
cohere_api_key=config.cohere_key, model="embed-english-light-v2.0", user_agent="airbyte-cdk"
144141
) # type: ignore
145142

146143
def check(self) -> Optional[str]:

0 commit comments

Comments
 (0)