Skip to content

Commit c0e506f

Browse files
feat: Bump langchain dependencies to 1.0.x (#810)
Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
1 parent 95ff3b8 commit c0e506f

File tree

4 files changed

+625
-235
lines changed

4 files changed

+625
-235
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
1311
from langchain_core.documents.base import Document
12+
from langchain_core.utils.strings import stringify_dict
13+
from langchain_text_splitters import Language, RecursiveCharacterTextSplitter
1414

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

airbyte_cdk/destinations/vector_db_based/embedder.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,12 @@
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 (
11+
CohereEmbeddings,
12+
FakeEmbeddings,
13+
LocalAIEmbeddings,
14+
OpenAIEmbeddings,
15+
)
1416

1517
from airbyte_cdk.destinations.vector_db_based.config import (
1618
AzureOpenAIEmbeddingConfigModel,
@@ -140,7 +142,9 @@ def __init__(self, config: CohereEmbeddingConfigModel):
140142
super().__init__()
141143
# Client is set internally
142144
self.embeddings = CohereEmbeddings(
143-
cohere_api_key=config.cohere_key, model="embed-english-light-v2.0"
145+
cohere_api_key=config.cohere_key,
146+
model="embed-english-light-v2.0",
147+
user_agent="airbyte-cdk",
144148
) # type: ignore
145149

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

0 commit comments

Comments
 (0)