language-model-common 2.0.74__tar.gz → 2.0.76__tar.gz

{language_model_common-2.0.74/language_model_common.egg-info → language_model_common-2.0.76}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: language-model-common
-Version: 2.0.74
+Version: 2.0.76
 Summary: Provides the underlying framework to enhance langchain and add loading configurations
 Author-email: Imran Qureshi <imran.qureshi@bwell.com>
 License: Apache-2.0

language_model_common-2.0.76/VERSION

@@ -0,0 +1 @@
+2.0.76

{language_model_common-2.0.74 → language_model_common-2.0.76/language_model_common.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: language-model-common
-Version: 2.0.74
+Version: 2.0.76
 Summary: Provides the underlying framework to enhance langchain and add loading configurations
 Author-email: Imran Qureshi <imran.qureshi@bwell.com>
 License: Apache-2.0

{language_model_common-2.0.74 → language_model_common-2.0.76}/languagemodelcommon/configs/config_reader/config_reader.py

@@ -35,6 +35,8 @@ logger.setLevel(SRC_LOG_LEVELS.CONFIG)
 
 
 class ConfigReader:
+    SCHEMA_VERSION = 1
+
     def __init__(
         self,
         *,
@@ -147,8 +149,7 @@ class ConfigReader:
             return models
 
     def _model_config_cache_key(self, *, model_name: str) -> str:
-        version = self._environment_variables.model_config_cache_schema_version
-        return f"v{version}:{model_name}"
+        return f"v{self.SCHEMA_VERSION}:{model_name}"
 
     async def _read_from_model_config_cache(self) -> List[ChatModelConfig] | None:
         """Load model configs from the model config cache (one row per model).
@@ -162,8 +163,7 @@ class ConfigReader:
         """
         if not self._model_config_cache_store:
             return None
-        version = self._environment_variables.model_config_cache_schema_version
-        prefix = f"v{version}:"
+        prefix = f"v{self.SCHEMA_VERSION}:"
         model_keys = await self._get_cache_keys_by_prefix(prefix=prefix)
         if not model_keys:
             return None
@@ -694,8 +694,7 @@ class ConfigReader:
 
     async def clear_cache(self) -> None:
         if self._model_config_cache_store:
-            version = self._environment_variables.model_config_cache_schema_version
-            prefix = f"v{version}:"
+            prefix = f"v{self.SCHEMA_VERSION}:"
             model_keys = await self._get_cache_keys_by_prefix(prefix=prefix)
             if model_keys:
                 await self._model_config_cache_store.delete_many(

{language_model_common-2.0.74 → language_model_common-2.0.76}/languagemodelcommon/configs/prompt_library/prompt_store.py

@@ -15,6 +15,8 @@ class PromptStore:
     filesystem race conditions during multi-worker startup.
     """
 
+    SCHEMA_VERSION = 1
+
     def __init__(self, *, store: BaseStore, collection: str = "prompts") -> None:
         self._store = store
         self._collection = collection
@@ -26,13 +28,19 @@ class PromptStore:
         if result is None:
             return None
 
+        if result.get("schema_version") != self.SCHEMA_VERSION:
+            return None
+
         content = result.get("content")
         if not isinstance(content, str):
             return None
         return content
 
     async def put_prompt(self, *, name: str, content: str) -> None:
-        value: dict[str, Any] = {"content": content}
+        value: dict[str, Any] = {
+            "schema_version": self.SCHEMA_VERSION,
+            "content": content,
+        }
         await self._store.put(name, value, collection=self._collection)
 
     async def delete_prompt(self, *, name: str) -> None:
@@ -41,6 +49,7 @@ class PromptStore:
     async def clear(self) -> None:
         if isinstance(self._store, BaseDestroyCollectionStore):
             await self._store.destroy_collection(collection=self._collection)
+            self._store._setup_collection_complete[self._collection] = False
         else:
             logger.warning(
                 "Prompt store does not support destroy_collection; "

{language_model_common-2.0.74 → language_model_common-2.0.76}/languagemodelcommon/context/request_context.py

@@ -1,8 +1,5 @@
 from contextvars import ContextVar
-from dataclasses import dataclass
-from typing import TYPE_CHECKING, Dict
-
-from oidcauthlib.auth.models.auth import AuthInformation
+from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from languagemodelcommon.converters.stream_buffer import StreamBufferManager
@@ -11,16 +8,6 @@ if TYPE_CHECKING:
     )
 
 
-@dataclass(frozen=True)
-class McpRequestContext:
-    headers: Dict[str, str]
-    auth_information: AuthInformation
-    user_id: str | None = None
-
-
-_request_context_var: ContextVar["McpRequestContext | None"] = ContextVar(
-    "mcp_request_context", default=None
-)
 _stream_debug_output_var: ContextVar["StreamDebugOutputManager | None"] = ContextVar(
     "stream_debug_output", default=None
 )
@@ -29,15 +16,6 @@ _stream_buffer_var: ContextVar["StreamBufferManager | None"] = ContextVar(
 )
 
 
-def get_request_context() -> "McpRequestContext":
-    ctx = _request_context_var.get()
-    if ctx is None:
-        raise RuntimeError(
-            "McpRequestContext not set — call init_request_context() first"
-        )
-    return ctx
-
-
 def get_stream_debug_output_manager() -> "StreamDebugOutputManager":
     mgr = _stream_debug_output_var.get()
     if mgr is None:
@@ -58,24 +36,13 @@ def get_stream_buffer_manager() -> "StreamBufferManager":
 
 def init_request_context(
     *,
-    headers: Dict[str, str],
-    auth_information: "AuthInformation",
-    user_id: str | None = None,
     stream_debug_output_manager: "StreamDebugOutputManager",
     stream_buffer_manager: "StreamBufferManager",
 ) -> None:
-    _request_context_var.set(
-        McpRequestContext(
-            headers=headers,
-            auth_information=auth_information,
-            user_id=user_id,
-        )
-    )
     _stream_debug_output_var.set(stream_debug_output_manager)
     _stream_buffer_var.set(stream_buffer_manager)
 
 
 def reset_request_context() -> None:
-    _request_context_var.set(None)
     _stream_debug_output_var.set(None)
     _stream_buffer_var.set(None)

{language_model_common-2.0.74 → language_model_common-2.0.76}/languagemodelcommon/mcp/interceptors/auth.py

@@ -54,24 +54,8 @@ class AuthMcpCallInterceptor:
         self._tool_configs_by_server_name: Dict[str, AgentConfig] = {
             tc.name: tc for tc in tool_configs
         }
-        self._static_auth_information = auth_information
-        self._static_headers = headers
-
-    @property
-    def _headers(self) -> Dict[str, str]:
-        if self._static_headers is not None:
-            return self._static_headers
-        from languagemodelcommon.context.request_context import get_request_context
-
-        return get_request_context().headers
-
-    @property
-    def _auth_information(self) -> AuthInformation:
-        if self._static_auth_information is not None:
-            return self._static_auth_information
-        from languagemodelcommon.context.request_context import get_request_context
-
-        return get_request_context().auth_information
+        self._auth_information = auth_information or AuthInformation(redirect_uri="")
+        self._headers: Dict[str, str] = headers or {}
 
     async def resolve_auth_header_for_discovery(
         self, tool_config: AgentConfig

{language_model_common-2.0.74 → language_model_common-2.0.76}/languagemodelcommon/mcp/mcp_tool_provider.py

@@ -298,7 +298,7 @@ class MCPToolProvider:
         self,
         *,
         tool_config: AgentConfig,
-        headers: Dict[str, str] | None = None,
+        headers: Dict[str, str],
         auth_interceptor: AuthMcpCallInterceptor,
         session_pool: McpSessionPool | None = None,
     ) -> List[BaseTool]:
@@ -308,16 +308,11 @@ class MCPToolProvider:
         Args:
             tool_config: An AgentConfig instance containing the tool's configuration.
             headers: A dictionary of headers to include in the request, such as Authorization.
-                If None, reads from request context contextvar.
             auth_interceptor: An AuthMcpCallInterceptor instance.
             session_pool: An optional session pool instance.
         Returns:
             A list of BaseTool instances retrieved from the MCP.
         """
-        if headers is None:
-            from languagemodelcommon.context.request_context import get_request_context
-
-            headers = get_request_context().headers
         if tool_config.lazy_load:
             return self.get_lazy_tools(
                 tool_config=tool_config,
@@ -733,15 +728,11 @@ class MCPToolProvider:
         self,
         *,
         tools: list[AgentConfig],
-        headers: Dict[str, str] | None = None,
+        headers: Dict[str, str],
         auth_interceptor: AuthMcpCallInterceptor,
         session_pool: McpSessionPool | None = None,
     ) -> list[BaseTool]:
         """Fetch tools from all configured MCP servers concurrently."""
-        if headers is None:
-            from languagemodelcommon.context.request_context import get_request_context
-
-            headers = get_request_context().headers
         url_tools = [t for t in tools if t.url is not None]
         if not url_tools:
             return []
@@ -977,7 +968,7 @@ class MCPToolProvider:
     def create_tool_resolver(
         self,
         *,
-        headers: Dict[str, str] | None = None,
+        headers: Dict[str, str],
         auth_interceptor: AuthMcpCallInterceptor,
     ) -> ToolResolverProtocol:
         """Create a resolver that lazily fetches tools from MCP servers.
@@ -1124,21 +1115,13 @@ class _BoundToolResolver:
         self,
         *,
         provider: MCPToolProvider,
-        headers: Dict[str, str] | None = None,
+        headers: Dict[str, str],
         auth_interceptor: AuthMcpCallInterceptor,
     ) -> None:
         self._provider = provider
-        self._static_headers = headers
+        self._headers = headers
         self._auth_interceptor = auth_interceptor
 
-    @property
-    def _headers(self) -> Dict[str, str]:
-        if self._static_headers is not None:
-            return self._static_headers
-        from languagemodelcommon.context.request_context import get_request_context
-
-        return get_request_context().headers
-
     async def resolve_tools(
         self,
         agent_config: AgentConfig,

{language_model_common-2.0.74 → language_model_common-2.0.76}/languagemodelcommon/utilities/environment/language_model_common_environment_variables.py

@@ -158,16 +158,6 @@ class LanguageModelCommonEnvironmentVariables(
         """
         return int(os.environ.get("MODEL_CONFIG_CACHE_TTL_SECONDS", "3600"))
 
-    @property
-    def model_config_cache_schema_version(self) -> str:
-        """Schema version for model config cache entries.
-
-        Changing this value automatically obsoletes all existing cache
-        entries without migration — queries use the version as part of
-        the cache key, so old-version entries are never found.
-        """
-        return os.environ.get("MODEL_CONFIG_CACHE_SCHEMA_VERSION", "1")
-
     @property
     def mcp_tool_cache_db_name(self) -> str:
         """MongoDB database name for MCP tool list cache.

{language_model_common-2.0.74 → language_model_common-2.0.76}/languagemodelcommon/utilities/message_preprocessing/composite_message_preprocessor.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Sequence
+from typing import Dict, Sequence
 
 from langchain_core.messages import AnyMessage
 
@@ -13,8 +13,10 @@ class CompositeMessagePreprocessor:
     def __init__(self, *, preprocessors: Sequence[MessagePreprocessor]) -> None:
         self._preprocessors = list(preprocessors)
 
-    async def preprocess(self, *, messages: list[AnyMessage]) -> list[AnyMessage]:
+    async def preprocess(
+        self, *, messages: list[AnyMessage], headers: Dict[str, str] | None = None
+    ) -> list[AnyMessage]:
         result = messages
         for preprocessor in self._preprocessors:
-            result = await preprocessor.preprocess(messages=result)
+            result = await preprocessor.preprocess(messages=result, headers=headers)
         return result

language_model_common-2.0.76/languagemodelcommon/utilities/message_preprocessing/message_preprocessor.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+from typing import Dict, Protocol
+
+from langchain_core.messages import AnyMessage
+
+
+class MessagePreprocessor(Protocol):
+    async def preprocess(
+        self, *, messages: list[AnyMessage], headers: Dict[str, str] | None = None
+    ) -> list[AnyMessage]: ...

language_model_common-2.0.74/VERSION

@@ -1 +0,0 @@
-2.0.74

language_model_common-2.0.74/languagemodelcommon/context/__init__.py

@@ -1,17 +0,0 @@
-from languagemodelcommon.context.request_context import (
-    McpRequestContext,
-    get_request_context,
-    get_stream_buffer_manager,
-    get_stream_debug_output_manager,
-    init_request_context,
-    reset_request_context,
-)
-
-__all__ = [
-    "McpRequestContext",
-    "get_request_context",
-    "get_stream_buffer_manager",
-    "get_stream_debug_output_manager",
-    "init_request_context",
-    "reset_request_context",
-]

language_model_common-2.0.74/languagemodelcommon/utilities/message_preprocessing/__init__.py

@@ -1,6 +0,0 @@
-from languagemodelcommon.utilities.message_preprocessing.message_preprocessor import (
-    MessagePreprocessor as MessagePreprocessor,
-)
-from languagemodelcommon.utilities.message_preprocessing.composite_message_preprocessor import (
-    CompositeMessagePreprocessor as CompositeMessagePreprocessor,
-)

language_model_common-2.0.74/languagemodelcommon/utilities/message_preprocessing/message_preprocessor.py

@@ -1,9 +0,0 @@
-from __future__ import annotations
-
-from typing import Protocol
-
-from langchain_core.messages import AnyMessage
-
-
-class MessagePreprocessor(Protocol):
-    async def preprocess(self, *, messages: list[AnyMessage]) -> list[AnyMessage]: ...

language_model_common-2.0.74/languagemodelcommon/utilities/slash_command/__init__.py

@@ -1,21 +0,0 @@
-from languagemodelcommon.utilities.slash_command.slash_command_parser import (
-    SlashCommandMatch as SlashCommandMatch,
-    parse_slash_command as parse_slash_command,
-)
-from languagemodelcommon.utilities.slash_command.slash_command_handler import (
-    SlashCommandContext as SlashCommandContext,
-    SlashCommandEffect as SlashCommandEffect,
-    SlashCommandHandler as SlashCommandHandler,
-)
-from languagemodelcommon.utilities.slash_command.slash_command_processor import (
-    SlashCommandProcessor as SlashCommandProcessor,
-)
-from languagemodelcommon.utilities.slash_command.handlers.debug_command_handler import (
-    DebugCommandEffect as DebugCommandEffect,
-    DebugCommandHandler as DebugCommandHandler,
-)
-from languagemodelcommon.utilities.slash_command.handlers.skill_command_handler import (
-    SkillCommandEffect as SkillCommandEffect,
-    SkillCommandHandler as SkillCommandHandler,
-    SkillContentResolver as SkillContentResolver,
-)