langchain-b12 0.1.6__tar.gz → 0.1.7__tar.gz

{langchain_b12-0.1.6 → langchain_b12-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-b12
-Version: 0.1.6
+Version: 0.1.7
 Summary: A reusable collection of tools and implementations for Langchain
 Author-email: Vincent Min <vincent.min@b12-consulting.com>
 Requires-Python: >=3.11

{langchain_b12-0.1.6 → langchain_b12-0.1.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langchain-b12"
-version = "0.1.6"
+version = "0.1.7"
 description = "A reusable collection of tools and implementations for Langchain"
 readme = "README.md"
 authors = [

{langchain_b12-0.1.6 → langchain_b12-0.1.7}/src/langchain_b12/genai/embeddings.py

@@ -1,6 +1,7 @@
 import os
 
 from google.genai import Client
+from google.genai.types import EmbedContentConfigOrDict
 from google.oauth2 import service_account
 from langchain_core.embeddings import Embeddings
 from pydantic import BaseModel, ConfigDict, Field
@@ -19,6 +20,7 @@ class GenAIEmbeddings(Embeddings, BaseModel):
         ),
         exclude=True,
     )
+    embed_content_config: EmbedContentConfigOrDict | None = Field(default=None)
     model_config = ConfigDict(
         arbitrary_types_allowed=True,
     )
@@ -33,19 +35,17 @@ class GenAIEmbeddings(Embeddings, BaseModel):
             list[list[float]]: The embedding vectors.
         """
         embeddings = []
-        for text in texts:
-            response = self.client.models.embed_content(
-                model=self.model_name,
-                contents=[text],
-            )
+        response = self.client.models.embed_content(
+            model=self.model_name,
+            contents=texts,
+            config=self.embed_content_config,
+        )
+        assert response.embeddings is not None, "No embeddings found in the response."
+        for embedding in response.embeddings:
             assert (
-                response.embeddings is not None
-            ), "No embeddings found in the response."
-            for embedding in response.embeddings:
-                assert (
-                    embedding.values is not None
-                ), "No embedding values found in the response."
-                embeddings.append(embedding.values)
+                embedding.values is not None
+            ), "No embedding values found in the response."
+            embeddings.append(embedding.values)
         assert len(embeddings) == len(
             texts
         ), "The number of embeddings does not match the number of texts."
@@ -75,6 +75,7 @@ class GenAIEmbeddings(Embeddings, BaseModel):
         response = await self.client.aio.models.embed_content(
             model=self.model_name,
             contents=texts,
+            config=self.embed_content_config,
         )
         assert response.embeddings is not None, "No embeddings found in the response."
         for embedding in response.embeddings:

{langchain_b12-0.1.6 → langchain_b12-0.1.7}/src/langchain_b12/genai/genai.py

@@ -419,6 +419,7 @@ class ChatGenAI(BaseChatModel):
             # add model name if final chunk
             if top_candidate.finish_reason is not None:
                 message.response_metadata["model_name"] = self.model_name
+                message.response_metadata["tags"] = self.tags or []
 
         return (
             ChatGenerationChunk(

{langchain_b12-0.1.6 → langchain_b12-0.1.7}/uv.lock

@@ -295,7 +295,7 @@ wheels = [
 
 [[package]]
 name = "langchain-b12"
-version = "0.1.4"
+version = "0.1.6"
 source = { editable = "." }
 dependencies = [
     { name = "langchain-core" },