Spaces:
Running
Running
File size: 1,977 Bytes
399b80c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | """
Utility functions for extracting model information from LangChain LLMs.
This module provides utilities to extract provider and model information
from LangChain language models for telemetry purposes.
"""
import importlib.metadata
from typing import Any
try:
from langchain_core.language_models.base import BaseLanguageModel # type: ignore[import-untyped]
except ImportError:
BaseLanguageModel = None # type: ignore[misc, assignment]
def get_package_version() -> str:
"""Get the current mcp-use package version."""
try:
return importlib.metadata.version("mcp-use")
except importlib.metadata.PackageNotFoundError:
return "unknown"
def get_model_provider(llm: Any) -> str:
"""Extract the model provider from LangChain LLM using BaseChatModel standards."""
if BaseLanguageModel is None:
return getattr(llm, "__class__", type(llm)).__name__.lower()
# Use LangChain's standard _llm_type property for identification
return getattr(llm, "_llm_type", llm.__class__.__name__.lower())
def get_model_name(llm: Any) -> str:
"""Extract the model name from LangChain LLM using BaseChatModel standards."""
# First try _identifying_params which may contain model info
if hasattr(llm, "_identifying_params"):
identifying_params = llm._identifying_params
if isinstance(identifying_params, dict):
# Common keys that contain model names
for key in ["model", "model_name", "model_id", "deployment_name"]:
if key in identifying_params:
return str(identifying_params[key])
# Fallback to direct model attributes
return getattr(llm, "model", getattr(llm, "model_name", llm.__class__.__name__))
def extract_model_info(llm: Any) -> tuple[str, str]:
"""Extract both provider and model name from LangChain LLM.
Returns:
Tuple of (provider, model_name)
"""
return get_model_provider(llm), get_model_name(llm)
|