mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-09 06:53:59 +00:00
embeddings: nomic embed vision (#22482)
Thank you for contributing to LangChain! **Description:** Adds Langchain support for Nomic Embed Vision **Twitter handle:** nomic_ai,zach_nussbaum - [x] **Add tests and docs**: If you're adding a new integration, please include 1. a test for the integration, preferably unit tests that do not rely on network access, 2. an example notebook showing its use. It lives in `docs/docs/integrations` directory. - [ ] **Lint and test**: Run `make format`, `make lint` and `make test` from the root of the package(s) you've modified. See contribution guidelines for more: https://python.langchain.com/docs/contributing/ Additional guidelines: - Make sure optional dependencies are imported within a function. - Please do not add dependencies to pyproject.toml files (even optional ones) unless they are required for unit tests. - Most PRs should not touch more than one package. - Changes should be backwards compatible. - If you are adding something to community, do not re-import it in langchain. If no one reviews your PR within a few days, please @-mention one of baskaryan, efriis, eyurtsev, ccurme, vbarda, hwchase17. --------- Co-authored-by: Lance Martin <122662504+rlancemartin@users.noreply.github.com> Co-authored-by: Bagatur <baskaryan@gmail.com>
This commit is contained in:
@@ -1,5 +1,3 @@
|
||||
from langchain_nomic.embeddings import NomicEmbeddings
|
||||
|
||||
__all__ = [
|
||||
"NomicEmbeddings",
|
||||
]
|
||||
__all__ = ["NomicEmbeddings"]
|
||||
|
@@ -22,6 +22,7 @@ class NomicEmbeddings(Embeddings):
|
||||
self,
|
||||
*,
|
||||
model: str,
|
||||
nomic_api_key: Optional[str] = ...,
|
||||
dimensionality: Optional[int] = ...,
|
||||
inference_mode: Literal["remote"] = ...,
|
||||
):
|
||||
@@ -32,6 +33,7 @@ class NomicEmbeddings(Embeddings):
|
||||
self,
|
||||
*,
|
||||
model: str,
|
||||
nomic_api_key: Optional[str] = ...,
|
||||
dimensionality: Optional[int] = ...,
|
||||
inference_mode: Literal["local", "dynamic"],
|
||||
device: Optional[str] = ...,
|
||||
@@ -43,6 +45,7 @@ class NomicEmbeddings(Embeddings):
|
||||
self,
|
||||
*,
|
||||
model: str,
|
||||
nomic_api_key: Optional[str] = ...,
|
||||
dimensionality: Optional[int] = ...,
|
||||
inference_mode: str,
|
||||
device: Optional[str] = ...,
|
||||
@@ -57,6 +60,7 @@ class NomicEmbeddings(Embeddings):
|
||||
dimensionality: Optional[int] = None,
|
||||
inference_mode: str = "remote",
|
||||
device: Optional[str] = None,
|
||||
vision_model: Optional[str] = None,
|
||||
):
|
||||
"""Initialize NomicEmbeddings model.
|
||||
|
||||
@@ -80,6 +84,7 @@ class NomicEmbeddings(Embeddings):
|
||||
self.dimensionality = dimensionality
|
||||
self.inference_mode = inference_mode
|
||||
self.device = device
|
||||
self.vision_model = vision_model
|
||||
|
||||
def embed(self, texts: List[str], *, task_type: str) -> List[List[float]]:
|
||||
"""Embed texts.
|
||||
@@ -121,3 +126,9 @@ class NomicEmbeddings(Embeddings):
|
||||
texts=[text],
|
||||
task_type="search_query",
|
||||
)[0]
|
||||
|
||||
def embed_image(self, uris: List[str]) -> List[List[float]]:
|
||||
return embed.image(
|
||||
images=uris,
|
||||
model=self.vision_model,
|
||||
)["embeddings"]
|
||||
|
14
libs/partners/nomic/poetry.lock
generated
14
libs/partners/nomic/poetry.lock
generated
@@ -1,4 +1,4 @@
|
||||
# This file is automatically @generated by Poetry 1.8.2 and should not be changed by hand.
|
||||
# This file is automatically @generated by Poetry 1.8.3 and should not be changed by hand.
|
||||
|
||||
[[package]]
|
||||
name = "annotated-types"
|
||||
@@ -276,7 +276,7 @@ files = [
|
||||
|
||||
[[package]]
|
||||
name = "langchain-core"
|
||||
version = "0.2.0rc1"
|
||||
version = "0.2.3"
|
||||
description = "Building applications with LLMs through composability"
|
||||
optional = false
|
||||
python-versions = ">=3.8.1,<4.0"
|
||||
@@ -285,7 +285,7 @@ develop = true
|
||||
|
||||
[package.dependencies]
|
||||
jsonpatch = "^1.33"
|
||||
langsmith = "^0.1.0"
|
||||
langsmith = "^0.1.65"
|
||||
packaging = "^23.2"
|
||||
pydantic = ">=1,<3"
|
||||
PyYAML = ">=5.3"
|
||||
@@ -300,13 +300,13 @@ url = "../../core"
|
||||
|
||||
[[package]]
|
||||
name = "langsmith"
|
||||
version = "0.1.58"
|
||||
version = "0.1.65"
|
||||
description = "Client library to connect to the LangSmith LLM Tracing and Evaluation Platform."
|
||||
optional = false
|
||||
python-versions = "<4.0,>=3.8.1"
|
||||
files = [
|
||||
{file = "langsmith-0.1.58-py3-none-any.whl", hash = "sha256:1148cc836ec99d1b2f37cd2fa3014fcac213bb6bad798a2b21bb9111c18c9768"},
|
||||
{file = "langsmith-0.1.58.tar.gz", hash = "sha256:a5060933c1fb3006b498ec849677993329d7e6138bdc2ec044068ab806e09c39"},
|
||||
{file = "langsmith-0.1.65-py3-none-any.whl", hash = "sha256:ab4487029240e69cca30da1065f1e9138e5a7ca2bbe8c697f0bd7d5839f71cf7"},
|
||||
{file = "langsmith-0.1.65.tar.gz", hash = "sha256:d3c2eb2391478bd79989f02652cf66e29a7959d677614b6993a47cef43f7f43b"},
|
||||
]
|
||||
|
||||
[package.dependencies]
|
||||
@@ -1309,4 +1309,4 @@ dev = ["black (>=19.3b0)", "pytest (>=4.6.2)"]
|
||||
[metadata]
|
||||
lock-version = "2.0"
|
||||
python-versions = ">=3.8.1,<4.0"
|
||||
content-hash = "369d2f7218d797a01a533380de9cce01037963f628dce10bc9927eac014edeeb"
|
||||
content-hash = "bf51336a3b4035385ddd68946aa5bbe699f4b805dd0503ba1dd1454a69248616"
|
||||
|
@@ -14,6 +14,7 @@ license = "MIT"
|
||||
python = ">=3.8.1,<4.0"
|
||||
langchain-core = ">=0.1.46,<0.3"
|
||||
nomic = "^3.0.29"
|
||||
pillow = "^10.3.0"
|
||||
|
||||
[tool.poetry.group.test]
|
||||
optional = true
|
||||
|
Reference in New Issue
Block a user