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

{language_model_common-2.0.75/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.75
+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.75 → 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.75
+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.75 → 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.75 → 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.75 → 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.75 → 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.75/VERSION

@@ -1 +0,0 @@
-2.0.75

language_model_common-2.0.75/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.75/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.75/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.75/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,
-)