Skip to content

Commit 02226a9

Browse files
fix: Apply ruff formatting and lint fixes
Co-Authored-By: AJ Steers <[email protected]>
1 parent da3b244 commit 02226a9

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

airbyte_cdk/destinations/vector_db_based/document_processor.py

Lines changed: 1 addition & 1 deletion
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_splitters import Language, RecursiveCharacterTextSplitter
1211
from langchain_core.documents.base import Document
1312
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 & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,12 @@
77
from dataclasses import dataclass
88
from typing import List, Optional, Union, cast
99

10-
from langchain_community.embeddings import CohereEmbeddings, FakeEmbeddings, LocalAIEmbeddings, OpenAIEmbeddings
10+
from langchain_community.embeddings import (
11+
CohereEmbeddings,
12+
FakeEmbeddings,
13+
LocalAIEmbeddings,
14+
OpenAIEmbeddings,
15+
)
1116

1217
from airbyte_cdk.destinations.vector_db_based.config import (
1318
AzureOpenAIEmbeddingConfigModel,
@@ -137,7 +142,9 @@ def __init__(self, config: CohereEmbeddingConfigModel):
137142
super().__init__()
138143
# Client is set internally
139144
self.embeddings = CohereEmbeddings(
140-
cohere_api_key=config.cohere_key, model="embed-english-light-v2.0", user_agent="airbyte-cdk"
145+
cohere_api_key=config.cohere_key,
146+
model="embed-english-light-v2.0",
147+
user_agent="airbyte-cdk",
141148
) # type: ignore
142149

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

0 commit comments

Comments
 (0)