langchain-b12 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

langchain_b12/citations/citations.py

@@ -3,7 +3,6 @@ from collections.abc import Sequence
 from typing import Any, Literal, TypedDict
 from uuid import UUID
 
-from fuzzysearch import find_near_matches
 from langchain_core.callbacks import Callbacks
 from langchain_core.language_models import BaseChatModel
 from langchain_core.messages import AIMessage, BaseMessage, SystemMessage
@@ -155,6 +154,8 @@ def validate_citations(
     sentences: list[str],
 ) -> list[tuple[Citation, Match | None]]:
     """Validate the citations. Invalid citations are dropped."""
+    from fuzzysearch import find_near_matches
+
     n_sentences = len(sentences)
 
     all_text = "\n".join(
@@ -244,18 +245,25 @@ def create_citation_model(
     e.g. `<context key="abc">Today is a sunny day</context>`.
     The returned AIMessage will have the following structure:
     AIMessage(
-        content= {
-            "citations": [
-                {
-                    "cited_text": "the color of the grass is green",
-                    "generated_cited_text": "the color of the grass is green",
-                    "key": "abc",
-                    "dist": 0,
-                }
-            ],
-            "text": "The grass is green",
-            "type": "text",
-        },
+        content=[
+            {
+                "citations": [
+                    {
+                        "cited_text": "the color of the grass is green",
+                        "generated_cited_text": "the color of the grass is green",
+                        "key": "abc",
+                        "dist": 0,
+                    }
+                ],
+                "text": "The grass is green",
+                "type": "text",
+            },
+            {
+                "citations": None,
+                "text": "Is there anything else I can help you with?",
+                "type": "text",
+            }
+        ]
     )
 
     Args:

langchain_b12/genai/genai.py

@@ -90,6 +90,8 @@ class ChatGenAI(BaseChatModel):
                 HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
             }
             """  # noqa: E501
+    thinking_config: types.ThinkingConfig | None = None
+    "The thinking configuration to use for the model."
 
     model_config = ConfigDict(
         arbitrary_types_allowed=True,
@@ -208,6 +210,10 @@ class ChatGenAI(BaseChatModel):
                 candidate_count=self.n,
                 stop_sequences=stop or self.stop,
                 safety_settings=self.safety_settings,
+                thinking_config=self.thinking_config,
+                automatic_function_calling=types.AutomaticFunctionCallingConfig(
+                    disable=True,
+                ),
                 **kwargs,
             ),
         )
@@ -240,6 +246,10 @@ class ChatGenAI(BaseChatModel):
                 candidate_count=self.n,
                 stop_sequences=stop or self.stop,
                 safety_settings=self.safety_settings,
+                thinking_config=self.thinking_config,
+                automatic_function_calling=types.AutomaticFunctionCallingConfig(
+                    disable=True,
+                ),
                 **kwargs,
             ),
         )
@@ -362,6 +372,12 @@ class ChatGenAI(BaseChatModel):
                 input_tokens=usage_metadata.prompt_token_count or 0,
                 output_tokens=usage_metadata.candidates_token_count or 0,
                 total_tokens=usage_metadata.total_token_count or 0,
+                input_token_details={
+                    "cache_read": usage_metadata.cached_content_token_count or 0
+                },
+                output_token_details={
+                    "reasoning": usage_metadata.thoughts_token_count or 0
+                },
             )
 
         total_lc_usage: UsageMetadata | None = (
@@ -392,7 +408,12 @@ class ChatGenAI(BaseChatModel):
                 "finish_reason": top_candidate.finish_reason,
                 "finish_message": top_candidate.finish_message,
             }
-            message = parse_response_candidate(top_candidate)
+            try:
+                message = parse_response_candidate(top_candidate)
+            except Exception as e:
+                raise ValueError(
+                    f"Failed to parse model response: {top_candidate.finish_message}"
+                ) from e
             if lc_usage:
                 message.usage_metadata = lc_usage
             # add model name if final chunk

{langchain_b12-0.1.4.dist-info → langchain_b12-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-b12
-Version: 0.1.4
+Version: 0.1.6
 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.dist-info/RECORD

@@ -0,0 +1,9 @@
+langchain_b12/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langchain_b12/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langchain_b12/citations/citations.py,sha256=ZQvYayjQXIUaRosJ0qwL3Nc7kC8sBzmaIkE-BOslaVI,12261
+langchain_b12/genai/embeddings.py,sha256=od2bVIgt7v9aNAHG0PVypVF1H_XgHto2nTd8vwfvyN8,3355
+langchain_b12/genai/genai.py,sha256=r7v_Z97N_Vd0zIR5mcQrlY3eWCPWhThvOvnXg59Ls8c,16868
+langchain_b12/genai/genai_utils.py,sha256=tA6UiJURK25-11vtaX4768UV47jDCYwVKIIWydD4Egw,10736
+langchain_b12-0.1.6.dist-info/METADATA,sha256=c2_hq-9spfRCb1nNRT6ztrCf1s1KNl8lkg-tLiB0mho,1204
+langchain_b12-0.1.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langchain_b12-0.1.6.dist-info/RECORD,,

langchain_b12-0.1.4.dist-info/RECORD

@@ -1,9 +0,0 @@
-langchain_b12/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langchain_b12/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langchain_b12/citations/citations.py,sha256=FO9ErybQws082JvV-MtTj81fVzdUWxrALcS81ElRsMw,12023
-langchain_b12/genai/embeddings.py,sha256=od2bVIgt7v9aNAHG0PVypVF1H_XgHto2nTd8vwfvyN8,3355
-langchain_b12/genai/genai.py,sha256=gzkgtvs3wNjcslS_KFZYCajUZIsJkVN2Tq2Q1RMIPyc,15910
-langchain_b12/genai/genai_utils.py,sha256=tA6UiJURK25-11vtaX4768UV47jDCYwVKIIWydD4Egw,10736
-langchain_b12-0.1.4.dist-info/METADATA,sha256=x659l7J9-4XSfjgZgGvR-cVoiCtTqq7cIaujV9JsTrE,1204
-langchain_b12-0.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langchain_b12-0.1.4.dist-info/RECORD,,