genai.extensions.langchain.utils module#

class genai.extensions.langchain.utils.CustomAIMessageChunk[source]#

Bases: AIMessageChunk

Replaces LangChain’s ‘merge_dicts’ with our simplified ‘merge_objects’ utility

class genai.extensions.langchain.utils.CustomChatGenerationChunk[source]#

Bases: ChatGenerationChunk

class genai.extensions.langchain.utils.CustomGenerationChunk[source]#

Bases: GenerationChunk

genai.extensions.langchain.utils.create_llm_output(*, model, token_usages=None, **kwargs)[source]#
Parameters:

  • model (str) –

  • token_usages (list[dict | None] | None) –

Return type:

dict[str, Any]

genai.extensions.langchain.utils.dump_optional_model(model)[source]#
Parameters:

model (BaseModel | None) –

Return type:

Mapping[str, Any] | None

genai.extensions.langchain.utils.load_config(file)[source]#
Parameters:

file (str | Path) –

Return type:

dict

genai.extensions.langchain.utils.update_token_usage(*, target, source)[source]#
Parameters:

  • target (dict[str, Any]) –

  • source (dict[str, Any] | None) –

genai.extensions.langchain.utils.update_token_usage_stream(*, target, source)[source]#
Parameters:

  • target (dict[str, Any]) –

  • source (dict | None) –